X-Git-Url: https://git.lizzy.rs/?a=blobdiff_plain;f=mylistanimecache.go;h=2ea83254ae0e07a2d2e351e19dd340a2c3f446a0;hb=091a27e88ce7152a2dd783451ffba5b8a465cab5;hp=c760bf33e866d59cff4d1d858e6d93a922e98afe;hpb=2972daf46d4096f1fb7a41d30814b7186b31ac65;p=go-anidb.git

diff --git a/mylistanimecache.go b/mylistanimecache.go
index c760bf3..2ea8325 100644
--- a/mylistanimecache.go
+++ b/mylistanimecache.go
@@ -77,22 +77,15 @@ func (adb *AniDB) MyListAnime(aid AID) <-chan *MyListAnime {
 			ep := <-adb.EpisodeByID(r.EID)
 			list := misc.EpisodeToList(&ep.Episode)
 
-			switch r.MyListState {
-			case MyListStateUnknown:
-				entry.UnknownState = list
-			case MyListStateHDD:
-				entry.OnHDD = list
-			case MyListStateCD:
-				entry.OnCD = list
-			case MyListStateDeleted:
-				entry.Deleted = list
+			entry.EpisodesWithState = MyListStateMap{
+				r.MyListState: list,
 			}
 
 			if !r.DateWatched.IsZero() {
 				entry.WatchedEpisodes = list
 			}
 
-			entry.EpisodesPerGroup = map[GID]misc.EpisodeList{
+			entry.EpisodesPerGroup = GroupEpisodes{
 				r.GID: list,
 			}
 		case 312:
@@ -173,10 +166,12 @@ func (adb *AniDB) parseMylistAnime(reply udpapi.APIReply) *MyListAnime {
 	}
 
 	return &MyListAnime{
-		UnknownState: misc.ParseEpisodeList(parts[2]),
-		OnHDD:        misc.ParseEpisodeList(parts[3]),
-		OnCD:         misc.ParseEpisodeList(parts[4]),
-		Deleted:      misc.ParseEpisodeList(parts[5]),
+		EpisodesWithState: MyListStateMap{
+			MyListStateUnknown: misc.ParseEpisodeList(parts[2]),
+			MyListStateHDD:     misc.ParseEpisodeList(parts[3]),
+			MyListStateCD:      misc.ParseEpisodeList(parts[4]),
+			MyListStateDeleted: misc.ParseEpisodeList(parts[5]),
+		},
 
 		WatchedEpisodes: misc.ParseEpisodeList(parts[6]),