AynaLivePlayer/provider/local_helper.go

package provider

import (
	"AynaLivePlayer/config"
	"AynaLivePlayer/player"
	"AynaLivePlayer/util"
	"github.com/dhowden/tag"
	"io/ioutil"
	"os"
	"path/filepath"
)

func getPlaylistNames() []string {
	names := make([]string, 0)
	items, _ := ioutil.ReadDir(config.Provider.LocalDir)
	for _, item := range items {
		if item.IsDir() {
			names = append(names, item.Name())
		}
	}
	return names
}

// readLocalPlaylist read files under a directory
// and return a _LocalPlaylist object.
// This function assume this directory exists
func readLocalPlaylist(playlist *_LocalPlaylist) error {
	p1th := playlist.Name
	playlist.Medias = make([]*player.Media, 0)
	fullPath := filepath.Join(config.Provider.LocalDir, p1th)
	if _, err := os.Stat(fullPath); os.IsNotExist(err) {
		return err
	}
	items, _ := ioutil.ReadDir(fullPath)
	for _, item := range items {
		// if item is a file, read file
		if !item.IsDir() {
			fn := item.Name()
			media := player.Media{
				Meta: Meta{
					Name: LocalAPI.GetName(),
					Id:   filepath.Join(fullPath, fn),
				},
			}
			if readMediaFile(&media) != nil {
				continue
			}
			playlist.Medias = append(playlist.Medias, &media)
		}
	}
	return nil
}

func readMediaFile(media *player.Media) error {
	p := media.Meta.(Meta).Id
	f, err := os.Open(p)
	if err != nil {
		return err
	}
	defer f.Close()
	meta, err := tag.ReadFrom(f)
	if err != nil {
		return err
	}
	media.Title = util.GetOrDefault(meta.Title(), filepath.Base(p))
	media.Artist = util.GetOrDefault(meta.Artist(), "Unknown")
	media.Album = util.GetOrDefault(meta.Album(), "Unknown")
	media.Lyric = meta.Lyrics()
	if meta.Picture() != nil {
		media.Cover.Data = meta.Picture().Data
	}
	return nil
}