implemented proper fetching and matching the source

internal/plugin/interface.go

@@ -1,6 +1,7 @@
 package plugin
 
 import (
+	"errors"
 	"regexp"
 
 	"gitea.elara.ws/Hazel/music-kraken/internal/data"
@@ -8,14 +9,16 @@ import (
 
 type Plugin interface {
 	Name() string
-	GetRegex() *regexp.Regexp
+	Regex() *regexp.Regexp
+	RegexArtist() *regexp.Regexp
+	RegexAlbum() *regexp.Regexp
+	RegexSong() *regexp.Regexp
 
-	Search(query string) []data.MusicObject
+	Search(query string) ([]data.MusicObject, error)
 
-	Fetch(source data.Source) data.MusicObject
-	FetchSong(source data.Source) data.Song
-	FetchAlbum(source data.Source) data.Album
-	FetchArtist(source data.Source) data.Artist
+	FetchArtist(source data.Source) (data.Artist, error)
+	FetchAlbum(source data.Source) (data.Album, error)
+	FetchSong(source data.Source) (data.Song, error)
 }
 
 var namePlugins = map[string]Plugin{}
@@ -25,34 +28,80 @@ func RegisterPlugin(plugin Plugin) {
 	name := plugin.Name()
 
 	nameSourceType[name] = data.SourceType{
-		Name:  name,
-		Regex: plugin.GetRegex(),
+		Name:        name,
+		Regex:       plugin.Regex(),
+		RegexArtist: plugin.RegexArtist(),
+		RegexAlbum:  plugin.RegexAlbum(),
+		RegexSong:   plugin.RegexSong(),
 	}
 
 	namePlugins[name] = plugin
 }
 
-func compileSource(source data.Source) data.Source {
-	if source.Type != nil {
-		return source
+func compileSourceType(source data.Source) (data.Source, error) {
+	if source.SourceType != nil {
+		return source, nil
 	}
 
 	for _, st := range nameSourceType {
 		if m := st.Regex.FindString(source.Url); m != "" {
 			source.Url = m
-			return source
+			return source, nil
 		}
 	}
-	panic("couldn't find source type for " + source.Url)
+
+	return source, errors.New("couldn't find source type for " + source.Url)
 }
 
-func FetchUrl(url string) data.MusicObject {
-	return FetchSource(data.Source{
-		Url:  string,
-		Type: nil,
-	})
+func compileSource(source data.Source) (data.Source, error) {
+	source, err := compileSourceType(source)
+	if err != nil {
+		return source, err
+	}
+
+	// find what object this source corresponds to
+	sourceType := source.SourceType
+
+	if sourceType.RegexSong.MatchString(source.Url) {
+		source.ObjectType = data.SongSource
+		return source, nil
+	}
+
+	if sourceType.RegexAlbum.MatchString(source.Url) {
+		source.ObjectType = data.AlbumSource
+		return source, nil
+	}
+
+	if sourceType.RegexArtist.MatchString(source.Url) {
+		source.ObjectType = data.ArtistSource
+		return source, nil
+	}
+
+	return source, errors.New("couldn't find corresponding object source on " + sourceType.Name + " for " + source.Url)
 }
 
-func FetchSource(source data.Source) data.MusicObject {
+func Fetch(source data.Source) (data.MusicObject, error) {
+	source, err := compileSource(source)
+	if err != nil {
+		return nil, err
+	}
 
+	plugin, ok := namePlugins[source.SourceType.Name]
+	if !ok {
+		return nil, errors.New("didn't find plugin of the name " + source.SourceType.Name)
+	}
+
+	if source.ObjectType == data.SongSource {
+		return plugin.FetchSong(source)
+	}
+
+	if source.ObjectType == data.AlbumSource {
+		return plugin.FetchAlbum(source)
+	}
+
+	if source.ObjectType == data.ArtistSource {
+		return plugin.FetchArtist(source)
+	}
+
+	return nil, nil
 }