summaryrefslogtreecommitdiff
path: root/pkg/media/library.go
blob: b18fe3a3b232255ebf52a723b4bd0262f9d5e951 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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
}