feat: add meta parser from filename
This commit is contained in:
parent
112d9ab28e
commit
9856f52070
50
algo/common/meta.go
Normal file
50
algo/common/meta.go
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type filenameMeta struct {
|
||||||
|
artists []string
|
||||||
|
title string
|
||||||
|
album string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *filenameMeta) GetArtists() []string {
|
||||||
|
return f.artists
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *filenameMeta) GetTitle() string {
|
||||||
|
return f.title
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *filenameMeta) GetAlbum() string {
|
||||||
|
return f.album
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseFilenameMeta(filename string) (meta AudioMeta) {
|
||||||
|
partName := strings.TrimSuffix(filename, path.Ext(filename))
|
||||||
|
items := strings.Split(partName, "-")
|
||||||
|
ret := &filenameMeta{}
|
||||||
|
|
||||||
|
switch len(items) {
|
||||||
|
case 0:
|
||||||
|
// no-op
|
||||||
|
case 1:
|
||||||
|
ret.title = strings.TrimSpace(items[0])
|
||||||
|
default:
|
||||||
|
ret.title = strings.TrimSpace(items[len(items)-1])
|
||||||
|
|
||||||
|
for _, v := range items[:len(items)-1] {
|
||||||
|
artists := strings.FieldsFunc(v, func(r rune) bool {
|
||||||
|
return r == ',' || r == '_'
|
||||||
|
})
|
||||||
|
for _, artist := range artists {
|
||||||
|
ret.artists = append(ret.artists, strings.TrimSpace(artist))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret
|
||||||
|
}
|
38
algo/common/meta_test.go
Normal file
38
algo/common/meta_test.go
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestParseFilenameMeta(t *testing.T) {
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
wantMeta AudioMeta
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "test1",
|
||||||
|
wantMeta: &filenameMeta{title: "test1"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "周杰伦 - 晴天.flac",
|
||||||
|
wantMeta: &filenameMeta{artists: []string{"周杰伦"}, title: "晴天"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Alan Walker _ Iselin Solheim - Sing Me to Sleep.flac",
|
||||||
|
wantMeta: &filenameMeta{artists: []string{"Alan Walker", "Iselin Solheim"}, title: "Sing Me to Sleep"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Christopher,Madcon - Limousine.flac",
|
||||||
|
wantMeta: &filenameMeta{artists: []string{"Christopher", "Madcon"}, title: "Limousine"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if gotMeta := ParseFilenameMeta(tt.name); !reflect.DeepEqual(gotMeta, tt.wantMeta) {
|
||||||
|
t.Errorf("ParseFilenameMeta() = %v, want %v", gotMeta, tt.wantMeta)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user