Modernize

[go-anidb.git] / titlecache.go

diff --git a/titlecache.go b/titlecache.go
index a69a5e458ea92fc135b93b770f616ac8e641fd11..c87dac38671781e54263c88829bda865219446d3 100644 (file)
--- a/titlecache.go
+++ b/titlecache.go
@@ -2,9 +2,8 @@ package anidb
 
 import (
        "bytes"
-       "github.com/Kovensky/go-anidb/titles"
+       "github.com/EliasFleckenstein03/go-anidb/titles"
        "io"
-       "log"
        "net/http"
        "os"
        "time"
@@ -35,10 +34,17 @@ func TitlesUpToDate() (ok bool) {
        return time.Now().Sub(titlesDB.UpdateTime) < 24*time.Hour
 }
 
+// Returns the number of anime in the titles database
+func AnimeCount() int {
+       return len(titlesDB.AnimeMap)
+}
+
 // Downloads a new anime-titles database if the database is outdated.
 //
 // Saves the database as anime-titles.dat.gz in the cache dir.
-func UpdateTitles() error {
+func (adb *AniDB) UpdateTitles() error {
+       // needs the AniDB for the Logger
+
        // too new, no need to update
        if TitlesUpToDate() {
                return nil
@@ -55,21 +61,21 @@ func UpdateTitles() error {
 
        c := &http.Client{Transport: &http.Transport{DisableCompression: true}}
 
-       log.Printf("HTTP>>> %s", titles.DataDumpURL)
+       adb.Logger.Printf("HTTP>>> %s", titles.DataDumpURL)
 
        resp, err := c.Get(titles.DataDumpURL)
        if err != nil {
-               log.Printf("HTTP<<< %s", resp.Status)
+               adb.Logger.Printf("HTTP<<< %s", resp.Status)
                return err
        }
        defer resp.Body.Close()
 
        buf := bytes.Buffer{}
-       log.Printf("HTTP--- %s", resp.Status)
+       adb.Logger.Printf("HTTP--- %s", resp.Status)
 
        _, err = io.Copy(&buf, resp.Body)
        if err != nil {
-               log.Printf("HTTP--- %v", err)
+               adb.Logger.Printf("HTTP--- %v", err)
                return err
        }
 
@@ -84,7 +90,7 @@ func UpdateTitles() error {
        }
 
        defer func() {
-               log.Printf("HTTP<<< Titles version %s", titlesDB.UpdateTime)
+               adb.Logger.Printf("HTTP<<< Titles version %s", titlesDB.UpdateTime)
        }()
        return RefreshTitles()
 }