aboutsummaryrefslogtreecommitdiff
path: root/localBackend.go
blob: a09d9a1035523c968516c98535e5e5a708c773db (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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package main

import (
	"fmt"
	"github.com/dhowden/tag"
	"muhq.space/go/wrms/llog"
	"os"
	"path"
	"path/filepath"
	"strings"
	"sync"
)

type LocalBackend struct {
	musicDir string
	lock     sync.RWMutex
	songs    map[string]*Song
}

func NewLocalBackend(musicDir string) *LocalBackend {
	b := LocalBackend{musicDir, sync.RWMutex{}, map[string]*Song{}}

	go b.findSongs()
	return &b
}

func (b *LocalBackend) insert(song *Song) {
	b.lock.Lock()
	defer b.lock.Unlock()
	b.songs[song.Uri] = song
}

func (b *LocalBackend) findSongs() {
	llog.Debug(fmt.Sprintf("Starting song search under: %s", b.musicDir))

	// Insert some Songs
	song := NewSong("No!", "Bukahara", "local", "Bukahara/Phantasma/01-No!.ogg")

	for _, s := range []*Song{&song} {
		b.insert(s)
	}

	err := filepath.Walk(b.musicDir, func(p string, finfo os.FileInfo, err error) error {
		if err != nil {
			llog.Warning(fmt.Sprintf("error %v at path %s\n", err, p))
			return err
		}

		if !finfo.Mode().IsRegular() {
			return nil
		}

		ext := strings.ToLower(path.Ext(p))
		for _, excluded := range []string{".png", ".jpg", ".txt", ".pdf", ".m3u"} {
			if ext == excluded {
				return nil
			}
		}

		f, err := os.Open(p)
		defer f.Close()
		if err != nil {
			llog.Warning(fmt.Sprintf("error %v opening file %s", err, p))
			return err
		}

		m, err := tag.ReadFrom(f)
		if err != nil {
			llog.Warning(fmt.Sprintf("error reading tags from %s: %v", p, err))
			// Contiune walking
			return nil
		}

		s := NewSong(m.Title(), m.Artist(), "local", p)
		b.insert(&s)
		return nil
	})

	if err != nil {
		llog.Error(fmt.Sprintf("error walking the path %q: %v", b.musicDir, err))
	}
}

func (b *LocalBackend) Play(song *Song, player *Player) {
	player.PlayUri("file://" + song.Uri)
}

func (b *LocalBackend) Search(keyword string) []Song {
	pattern := strings.ToLower(keyword)
	results := []Song{}

	b.lock.RLock()
	defer b.lock.RUnlock()

	for _, s := range b.songs {
		if strings.Contains(strings.ToLower(s.Title), pattern) ||
			strings.Contains(strings.ToLower(s.Artist), pattern) {
			results = append(results, *s)
		}
	}

	return results
}