Rainbond/api/controller/upload/dir.go

package upload

import (
	"math/rand"
	"os"
	"path/filepath"
	"strconv"
	"time"
)

// Directory mananger
type dirManager struct {
	Root string
	Path string
}

// Prepare dirManager given root, mime.
func createDir(root, mime string) (*dirManager, error) {
	dm := newDirManager(root)

	dm.CalcPath(mime)
	if err := dm.create(); err != nil {
		return nil, err
	}

	return dm, nil
}

// newDirManager returns a new dirManager given a root.
func newDirManager(root string) *dirManager {
	return &dirManager{Root: root}
}

// Return absolute path for directory
func (dm *dirManager) Abs() string {
	return filepath.Join(dm.Root, dm.Path)
}

// Create directory obtained by concatenating the root and path.
func (dm *dirManager) create() error {
	return os.MkdirAll(dm.Root+dm.Path, 0755)
}

// Generate path given mime and date.
func (dm *dirManager) CalcPath(mime string) {
	dm.Path = ""
}

func yearDay(t time.Time) string {
	return strconv.FormatInt(int64(t.YearDay()), 36)
}

func containerName(t time.Time) string {
	r := rand.New(rand.NewSource(time.Now().UnixNano())).Intn(1000)
	seconds := t.Hour()*3600 + t.Minute()*60 + t.Second()

	return strconv.FormatInt(int64(seconds*1000+r), 36)
}