Modernize

[go-anidb.git] / mylistmanip.go

diff --git a/mylistmanip.go b/mylistmanip.go
index beca6bb02d5f39d53458dbfe0acf49f85ffdf4c4..271673abeda27335629683f07438c8d309a451d5 100644 (file)
--- a/mylistmanip.go
+++ b/mylistmanip.go
@@ -1,7 +1,7 @@
 package anidb
 
 import (
-       "github.com/Kovensky/go-fscache"
+       "github.com/EliasFleckenstein03/go-fscache"
        "strconv"
        "time"
 )
@@ -67,11 +67,10 @@ func (set *MyListSet) update(uid UID, f *File, lid LID) {
        eg.Add(f.EpisodeNumber)
        mla.EpisodesPerGroup[f.GID] = eg
 
+       newState := MyListStateUnknown
        if set != nil {
                if set.State != nil {
-                       es := mla.EpisodesWithState[*set.State]
-                       es.Add(f.EpisodeNumber)
-                       mla.EpisodesWithState[*set.State] = es
+                       newState = *set.State
                }
 
                if set.Watched != nil && *set.Watched ||
@@ -80,16 +79,36 @@ func (set *MyListSet) update(uid UID, f *File, lid LID) {
                }
        }
 
+       es := mla.EpisodesWithState[newState]
+       es.Add(f.EpisodeNumber)
+       mla.EpisodesWithState[newState] = es
+
        Cache.Set(mla, "mylist-anime", uid, f.AID)
        Cache.Chtime(mla.Cached, "mylist-anime", uid, f.AID)
 
+       e := lid.MyListEntry()
        if set == nil ||
                (set.ViewDate == nil && set.Watched == nil && set.State == nil &&
                        set.Source == nil && set.Storage == nil && set.Other == nil) {
                return
        }
+       if e == nil {
+               now := time.Now()
+               e = &MyListEntry{
+                       LID: lid,
+
+                       AID: f.AID,
+                       EID: f.EID,
+                       FID: f.FID,
+                       GID: f.GID,
+
+                       DateAdded: now,
+
+                       // make sure it's expired
+                       Cached: time.Unix(0, 0),
+               }
+       }
 
-       e := lid.MyListEntry()
        if set.ViewDate != nil {
                e.DateWatched = *set.ViewDate
        } else if set.Watched != nil {