82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
package media_store
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/pkg/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Store []VideoInfo
|
|
|
|
type VideoInfo struct {
|
|
Hash string
|
|
Name string
|
|
Duration uint
|
|
}
|
|
|
|
func NewStore(path string) (Store, error) {
|
|
mapFile := filepath.Join(path, "map.json")
|
|
log.Infof("read in from: %s", mapFile)
|
|
if _, err := os.Stat(mapFile); err != nil {
|
|
if os.IsNotExist(err) {
|
|
if err := os.WriteFile(mapFile, []byte("[]"), os.ModePerm); err != nil {
|
|
return nil, errors.Wrapf(err, "write file: %s", mapFile)
|
|
}
|
|
}
|
|
}
|
|
|
|
b, err := os.ReadFile(mapFile)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "read file: %s", mapFile)
|
|
}
|
|
|
|
var store Store
|
|
err = json.Unmarshal(b, &store)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "unmarshal json: %s", mapFile)
|
|
}
|
|
|
|
return store, nil
|
|
}
|
|
|
|
func (s Store) Save(path string) error {
|
|
mapFile := filepath.Join(path, "map.json")
|
|
b, err := json.Marshal(s)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "marshal json: %s", mapFile)
|
|
}
|
|
|
|
if err := os.WriteFile(mapFile, b, os.ModePerm); err != nil {
|
|
return errors.Wrapf(err, "write file: %s", mapFile)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s Store) Append(info VideoInfo) Store {
|
|
return append(s, info)
|
|
}
|
|
|
|
func (s Store) Hashes() []string {
|
|
var hashes []string
|
|
for _, m := range s {
|
|
hashes = append(hashes, m.Hash)
|
|
}
|
|
|
|
return hashes
|
|
}
|
|
|
|
// Exists
|
|
func (s Store) HashExists(hash string) bool {
|
|
for _, m := range s {
|
|
if m.Hash == hash {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|