mylistadd: Only add if the file isn't on mylist yet

[go-anidb.git] / anidb.go

diff --git a/anidb.go b/anidb.go
index b553108674a4549e209e00c037be3aff17b56036..7029a46f04d6b470407a5f170552297cd8eabfee 100644 (file)
--- a/anidb.go
+++ b/anidb.go
@@ -2,6 +2,8 @@
 package anidb
 
 import (
+       "io/ioutil"
+       "log"
        "time"
 )
 
@@ -19,21 +21,30 @@ import (
 // by a previous Auth() call.
 type AniDB struct {
        Timeout time.Duration // Timeout for the various calls (default: 45s)
+       Logger  *log.Logger   // Logger where HTTP/UDP traffic is logged
 
        udp *udpWrap
 }
 
 // Initialises a new AniDB.
 func NewAniDB() *AniDB {
-       return &AniDB{
+       ret := &AniDB{
                Timeout: 45 * time.Second,
-               udp:     newUDPWrap(),
+               Logger:  log.New(ioutil.Discard, "", log.LstdFlags),
        }
+       ret.udp = newUDPWrap(ret)
+       return ret
 }
 
 func (adb *AniDB) User() *User {
        if adb != nil && adb.udp != nil {
-               return adb.udp.user
+               if adb.udp.user != nil {
+                       return adb.udp.user
+               } else if adb.udp.credentials != nil {
+                       // see if we can get it from the cache
+                       adb.udp.user = UserByName(decrypt(adb.udp.credentials.username))
+                       return adb.udp.user
+               }
        }
        return nil
 }