contented/Metadata.go

94 lines
1.7 KiB
Go

package contented
import (
"encoding/json"
"errors"
"log"
"net/http"
"os"
"time"
"github.com/speps/go-hashids"
bolt "go.etcd.io/bbolt"
)
const (
hashIdSalt = "contented"
hashIdMinLength = 2
)
type Metadata struct {
FileHash string
FileSize int64
UploadTime time.Time
UploadIP string
Filename string
MimeType string
}
func (this *Server) Metadata(fileID string) (*Metadata, error) {
var m Metadata
err := this.db.View(func(tx *bolt.Tx) error {
content := tx.Bucket(this.metadataBucket).Get([]byte(fileID))
if len(content) == 0 {
return os.ErrNotExist
}
return json.Unmarshal(content, &m)
})
if err != nil {
return nil, err
}
return &m, nil
}
func idToString(v uint64) string {
hd := hashids.NewData()
hd.Salt = hashIdSalt
hd.MinLength = hashIdMinLength
h := hashids.NewWithData(hd)
s, _ := h.EncodeInt64([]int64{int64(v)})
return s
}
func (this *Server) AddMetadata(m Metadata) (string, error) {
jb, err := json.Marshal(m)
if err != nil {
return "", err
}
var shortRef string = ""
err = this.db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket(this.metadataBucket)
seq, _ := b.NextSequence() // cannot fail
shortRef = idToString(seq)
return tx.Bucket(this.metadataBucket).Put([]byte(shortRef), jb)
})
if err != nil {
return "", err
}
if shortRef == "" {
return "", errors.New("Invalid URL generated")
}
return shortRef, nil
}
func (this *Server) handleInformation(w http.ResponseWriter, fileID string) {
m, err := this.Metadata(fileID)
if err != nil {
if os.IsNotExist(err) {
http.Error(w, "Not found", 404)
return
}
log.Println(err.Error())
http.Error(w, "Internal error", 500)
return
}
this.serveJsonObject(w, m)
}