api/model/file/file.go

package file

import (
	"context"
	"crypto/rand"
	"encoding/binary"
	"errors"
	"io"
	"strconv"
	"time"

	"git.aiterp.net/rpdata/api/internal/store"
	"github.com/globalsign/mgo"
	"github.com/globalsign/mgo/bson"
)

var fileCollection *mgo.Collection

// A File is a record of a file stored in the Space.
type File struct {
	ID       string    `bson:"_id" json:"id"`
	Time     time.Time `bson:"time" json:"time"`
	Kind     string    `bson:"kind" json:"kind"`
	Public   bool      `bson:"public" json:"public"`
	Name     string    `bson:"name" json:"name"`
	MimeType string    `bson:"mimeType" json:"mimeType"`
	Size     int64     `bson:"size" json:"size"`
	Author   string    `bson:"author" json:"author"`
	URL      string    `bson:"url,omitempty" json:"url,omitempty"`
}

// Edit edits the file, changing up to both the two mutable properties
func (file *File) Edit(name *string, public *bool) error {
	changes := bson.M{}
	changedFile := *file

	if name != nil && *name != file.Name {
		changes["name"] = *name
		changedFile.Name = *name
	}
	if public != nil && *public != file.Public {
		changes["public"] = *public
		changedFile.Public = *public
	}

	if len(changes) == 0 {
		return nil
	}

	err := fileCollection.UpdateId(file.ID, bson.M{"$set": changes})
	if err != nil {
		return err
	}

	*file = changedFile

	return nil
}

// Delete removes the file information from the database, and deletes the file.
func (file *File) Delete() error {
	err := fileCollection.RemoveId(file.ID)
	if err != nil {
		return err
	}

	if file.Kind == "upload" {
		err = store.RemoveFile("files", file.ID)
		if err != nil {
			return err
		}
	}

	file.URL = ""

	return nil
}

// Insert manually inserts file information into the database. This should never, ever be API accessible
func Insert(name, kind, mimeType, author string, time time.Time, size int64, url string) (File, error) {
	file := File{
		ID:       makeFileID(),
		Kind:     kind,
		Time:     time,
		Public:   false,
		Author:   author,
		Name:     name,
		MimeType: mimeType,
		Size:     size,
		URL:      url,
	}

	err := fileCollection.Insert(file)
	if err != nil {
		return File{}, err
	}

	return file, nil
}

// Upload adds a file to the space.
func Upload(ctx context.Context, name, mimeType, author string, size int64, input io.Reader) (File, error) {
	if !allowdMimeTypes[mimeType] {
		return File{}, errors.New("File type not allowed:" + mimeType)
	}

	if name == "" {
		date := time.Now().UTC().Format("Jan 02 2006 15:04:05 MST")
		name = "Unnamed file (" + date + ")"
	}

	if mimeType == "" {
		mimeType = "binary/octet-stream"
	}

	id := makeFileID()

	path, err := store.UploadFile(ctx, "files", id, mimeType, input, size)
	if err != nil {
		return File{}, err
	}

	file := File{
		ID:       id,
		Kind:     "upload",
		Time:     time.Now(),
		Public:   false,
		Author:   author,
		Name:     name,
		MimeType: mimeType,
		Size:     size,
		URL:      store.URLFromPath(path),
	}

	err = fileCollection.Insert(file)
	if err != nil {
		return File{}, err
	}

	return file, nil
}

// FindID finds a file by ID
func FindID(id string) (File, error) {
	file := File{}

	err := fileCollection.FindId(id).One(&file)
	if err != nil {
		return File{}, err
	}

	return file, nil
}

// FindName finds a file by kind, name and author
func FindName(kind string, name string, author string) (File, error) {
	query := bson.M{"kind": kind, "name": name, "author": author}
	file := File{}

	err := fileCollection.Find(query).One(&file)
	if err != nil {
		return File{}, err
	}

	return file, nil
}

// List lists files according to the standard lookup. By default it's just the author's own files,
// but if `public` is true it will alos include files made public by other authors. If `mimeTypes` contains
// any, it will limit the results to that. If `author` is empty, it will only list public files
func List(author string, public bool, mimeTypes []string) ([]File, error) {
	query := bson.M{}

	if author != "" {
		if public {
			query["$or"] = []bson.M{
				bson.M{"author": author},
				bson.M{"public": true},
			}
		} else {
			query["author"] = author
		}
	} else {
		if !public {
			return nil, errors.New("No author specified, and public is unset")
		}

		query["public"] = true
	}

	if len(mimeTypes) > 0 {
		query["mimeTypes"] = bson.M{"$in": mimeTypes}
	}

	return listFiles(query)
}

func listFiles(query interface{}) ([]File, error) {
	list := make([]File, 0, 32)

	err := fileCollection.Find(query).Sort("-time").All(&list)
	if err != nil {
		return nil, err
	}

	return list, nil
}

// makeFileID makes a random file ID that's 32 characters long
func makeFileID() string {
	result := "F" + strconv.FormatInt(time.Now().UnixNano(), 36)
	offset := 0
	data := make([]byte, 32)

	rand.Read(data)
	for len(result) < 32 {
		result += strconv.FormatUint(binary.LittleEndian.Uint64(data[offset:]), 36)
		offset += 8

		if offset >= 32 {
			rand.Read(data)
			offset = 0
		}
	}

	return result[:32]
}

func init() {
	store.HandleInit(func(db *mgo.Database) {
		fileCollection = db.C("file.headers")

		fileCollection.EnsureIndexKey("author")
		fileCollection.EnsureIndexKey("public")
		fileCollection.EnsureIndexKey("kind", "name", "author")
		fileCollection.EnsureIndexKey("author", "public")
		fileCollection.EnsureIndexKey("kind")
	})
}

var allowdMimeTypes = map[string]bool{
	"":                    false,
	"image/jpeg":          true,
	"image/png":           true,
	"image/gif":           true,
	"text/plain":          true,
	"application/json":    true,
	"application/pdf":     false,
	"binary/octet-stream": false,
	"video/mp4":           false,
	"audio/mp3":           false,
}