57 lines
976 B
Go
57 lines
976 B
Go
|
package media
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"sort"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Library struct {
|
||
|
|
Videos map[string]*Video
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLibrary() *Library {
|
||
|
|
lib := &Library{
|
||
|
|
Videos: make(map[string]*Video),
|
||
|
|
}
|
||
|
|
return lib
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lib *Library) Import(path string) error {
|
||
|
|
files, err := ioutil.ReadDir(path)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
for _, info := range files {
|
||
|
|
name := info.Name()
|
||
|
|
v, err := ParseVideo(path + "/" + name)
|
||
|
|
if err != nil {
|
||
|
|
// Ignore files that can't be parsed
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
// Set modified date property
|
||
|
|
v.Modified = info.ModTime().Format("2006-01-02")
|
||
|
|
// Default title is filename
|
||
|
|
if v.Title == "" {
|
||
|
|
v.Title = name
|
||
|
|
}
|
||
|
|
// ID is name without extension
|
||
|
|
idx := strings.LastIndex(name, ".")
|
||
|
|
if idx == -1 {
|
||
|
|
idx = len(name)
|
||
|
|
}
|
||
|
|
v.ID = name[:idx]
|
||
|
|
lib.Videos[v.ID] = v
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lib *Library) Playlist() Playlist {
|
||
|
|
pl := make(Playlist, 0)
|
||
|
|
for _, v := range lib.Videos {
|
||
|
|
pl = append(pl, v)
|
||
|
|
}
|
||
|
|
sort.Sort(pl)
|
||
|
|
return pl
|
||
|
|
}
|