titlecache.go

   1 package anidb
   2
   3 import (
   4         "bytes"
   5         "github.com/Kovensky/go-anidb/titles"
   6         "io"
   7         "net/http"
   8         "os"
   9         "time"
  10 )
  11
  12 var titlesDB = &titles.TitlesDatabase{}
  13
  14 // Loads the database from anime-titles.dat.gz in the cache dir.
  15 func RefreshTitles() error {
  16         if lock, err := Cache.Lock("anime-titles.dat.gz"); err != nil {
  17                 return err
  18         } else {
  19                 defer lock.Unlock()
  20         }
  21
  22         fh, err := Cache.Open("anime-titles.dat.gz")
  23         if err != nil {
  24                 return err
  25         }
  26         defer fh.Close()
  27
  28         titlesDB.LoadDB(fh)
  29         return nil
  30 }
  31
  32 // Returns true if the titles database is up-to-date (newer than 24 hours).
  33 func TitlesUpToDate() (ok bool) {
  34         return time.Now().Sub(titlesDB.UpdateTime) < 24*time.Hour
  35 }
  36
  37 // Downloads a new anime-titles database if the database is outdated.
  38 //
  39 // Saves the database as anime-titles.dat.gz in the cache dir.
  40 func (adb *AniDB) UpdateTitles() error {
  41         // needs the AniDB for the Logger
  42
  43         // too new, no need to update
  44         if TitlesUpToDate() {
  45                 return nil
  46         }
  47
  48         switch lock, err := Cache.Lock("anime-titles.dat.gz"); {
  49         case os.IsNotExist(err):
  50                 // we're creating it now
  51         case err == nil:
  52                 defer lock.Unlock()
  53         default:
  54                 return err
  55         }
  56
  57         c := &http.Client{Transport: &http.Transport{DisableCompression: true}}
  58
  59         adb.Logger.Printf("HTTP>>> %s", titles.DataDumpURL)
  60
  61         resp, err := c.Get(titles.DataDumpURL)
  62         if err != nil {
  63                 adb.Logger.Printf("HTTP<<< %s", resp.Status)
  64                 return err
  65         }
  66         defer resp.Body.Close()
  67
  68         buf := bytes.Buffer{}
  69         adb.Logger.Printf("HTTP--- %s", resp.Status)
  70
  71         _, err = io.Copy(&buf, resp.Body)
  72         if err != nil {
  73                 adb.Logger.Printf("HTTP--- %v", err)
  74                 return err
  75         }
  76
  77         fh, err := Cache.Create("anime-titles.dat.gz")
  78         if err != nil {
  79                 return err
  80         }
  81
  82         _, err = io.Copy(fh, &buf)
  83         if err != nil {
  84                 return err
  85         }
  86
  87         defer func() {
  88                 adb.Logger.Printf("HTTP<<< Titles version %s", titlesDB.UpdateTime)
  89         }()
  90         return RefreshTitles()
  91 }