package anidb
import (
- "encoding/gob"
"fmt"
"github.com/Kovensky/go-anidb/misc"
"github.com/Kovensky/go-anidb/udp"
+ "github.com/Kovensky/go-fscache"
"image"
"log"
"regexp"
"time"
)
-func init() {
- gob.RegisterName("*github.com/Kovensky/go-anidb.File", &File{})
- gob.RegisterName("*github.com/Kovensky/go-anidb.fidCache", &fidCache{})
- gob.RegisterName("github.com/Kovensky/go-anidb.FID", FID(0))
-}
+var _ cacheable = &File{}
-func (f *File) Touch() {
- f.Cached = time.Now()
+func (f *File) setCachedTS(ts time.Time) {
+ f.Cached = ts
}
func (f *File) IsStale() bool {
return time.Now().Sub(f.Cached) > FileCacheDuration
}
-type FID int
-
-// make FID Cacheable
+func cacheFile(f *File) {
+ CacheSet(f.FID, "fid", "by-ed2k", f.Ed2kHash, f.Filesize)
+ CacheSet(f, "fid", f.FID)
+}
-func (e FID) Touch() {}
-func (e FID) IsStale() bool { return false }
+type FID int
func (fid FID) File() *File {
var f File
- if cache.Get(&f, "fid", fid) == nil {
+ if CacheGet(&f, "fid", fid) == nil {
return &f
}
return nil
}
-type fidCache struct {
- FID
- Time time.Time
-}
-
-func (c *fidCache) Touch() {
- c.Time = time.Now()
-}
-
-func (c *fidCache) IsStale() bool {
- return time.Now().Sub(c.Time) > FileCacheDuration
-}
-
// Prefetches the Anime, Episode and Group that this
// file is linked to using the given AniDB instance.
//
// Retrieves a File by its FID. Uses the UDP API.
func (adb *AniDB) FileByID(fid FID) <-chan *File {
- keys := []cacheKey{"fid", fid}
+ key := []fscache.CacheKey{"fid", fid}
ch := make(chan *File, 1)
return ch
}
- ic := make(chan Cacheable, 1)
+ ic := make(chan notification, 1)
go func() { ch <- (<-ic).(*File); close(ch) }()
- if intentMap.Intent(ic, keys...) {
+ if intentMap.Intent(ic, key...) {
return ch
}
- if !cache.CheckValid(keys...) {
- intentMap.NotifyClose((*File)(nil), keys...)
+ if !Cache.IsValid(InvalidKeyCacheDuration, key...) {
+ intentMap.NotifyClose((*File)(nil), key...)
return ch
}
f := fid.File()
if !f.IsStale() {
- intentMap.NotifyClose(f, keys...)
+ intentMap.NotifyClose(f, key...)
return ch
}
})
if reply.Error() == nil {
- f = adb.parseFileResponse(reply, false)
+ adb.parseFileResponse(&f, reply, false)
- cache.Set(&fidCache{FID: f.FID}, "fid", "by-ed2k", f.Ed2kHash, f.Filesize)
- cache.Set(f, keys...)
+ cacheFile(f)
} else if reply.Code() == 320 {
- cache.MarkInvalid(keys...)
+ Cache.SetInvalid(key...)
}
- intentMap.NotifyClose(f, keys...)
+ intentMap.NotifyClose(f, key...)
}()
return ch
}
// Retrieves a File by its Ed2kHash + Filesize combination. Uses the UDP API.
func (adb *AniDB) FileByEd2kSize(ed2k string, size int64) <-chan *File {
- keys := []cacheKey{"fid", "by-ed2k", ed2k, size}
+ key := []fscache.CacheKey{"fid", "by-ed2k", ed2k, size}
ch := make(chan *File, 1)
// AniDB always uses lower case hashes
ed2k = strings.ToLower(ed2k)
- ic := make(chan Cacheable, 1)
+ ic := make(chan notification, 1)
go func() {
fid := (<-ic).(FID)
if fid > 0 {
}
close(ch)
}()
- if intentMap.Intent(ic, keys...) {
+ if intentMap.Intent(ic, key...) {
return ch
}
- if !cache.CheckValid(keys...) {
- intentMap.NotifyClose(FID(0), keys...)
+ if !Cache.IsValid(InvalidKeyCacheDuration, key...) {
+ intentMap.NotifyClose(FID(0), key...)
return ch
}
fid := FID(0)
- var ec fidCache
- if cache.Get(&ec, keys...) == nil && !ec.IsStale() {
- intentMap.NotifyClose(ec.FID, keys...)
+ switch ts, err := Cache.Get(&fid, key...); {
+ case err != nil && time.Now().Sub(ts) < FileCacheDuration:
+ intentMap.NotifyClose(fid, key...)
return ch
}
- fid = ec.FID
go func() {
reply := <-adb.udp.SendRecv("FILE",
var f *File
if reply.Error() == nil {
- f = adb.parseFileResponse(reply, false)
+ adb.parseFileResponse(&f, reply, false)
fid = f.FID
- cache.Set(&fidCache{FID: fid}, keys...)
- cache.Set(f, "fid", fid)
+ cacheFile(f)
} else if reply.Code() == 320 { // file not found
- cache.MarkInvalid(keys...)
+ Cache.SetInvalid(key...)
} else if reply.Code() == 322 { // multiple files found
panic("Don't know what to do with " + strings.Join(reply.Lines(), "\n"))
}
- intentMap.NotifyClose(fid, keys...)
+ intentMap.NotifyClose(fid, key...)
}()
return ch
}
-var fileFmask = "77da7fe8"
+var fileFmask = "7fda7fe8"
var fileAmask = "00008000"
const (
- fileStateCRCOK = 1 << iota
- fileStateCRCERR
- fileStateV2
- fileStateV3
- fileStateV4
- fileStateV5
- fileStateUncensored
- fileStateCensored
+ stateCRCOK = 1 << iota
+ stateCRCERR
+ stateV2
+ stateV3
+ stateV4
+ stateV5
+ stateUncensored
+ stateCensored
)
func sanitizeCodec(codec string) string {
return codec
}
-func (adb *AniDB) parseFileResponse(reply udpapi.APIReply, calledFromFIDsByGID bool) *File {
+func (adb *AniDB) parseFileResponse(f **File, reply udpapi.APIReply, calledFromFIDsByGID bool) bool {
if reply.Error() != nil {
- return nil
+ return false
}
if reply.Truncated() {
panic("Truncated")
}
+ uidChan := make(chan UID, 1)
+ if adb.udp.credentials != nil {
+ go func() { uidChan <- <-adb.GetUserUID(decrypt(adb.udp.credentials.username)) }()
+ } else {
+ uidChan <- 0
+ close(uidChan)
+ }
+
parts := strings.Split(reply.Lines()[1], "|")
ints := make([]int64, len(parts))
for i, p := range parts {
partial := false
- rels := strings.Split(parts[4], "'")
- relList := make([]EID, 0, len(parts[4]))
- related := make(RelatedEpisodes, len(parts[4]))
+ rels := strings.Split(parts[5], "'")
+ relList := make([]EID, 0, len(parts[5]))
+ related := make(RelatedEpisodes, len(parts[5]))
for _, rel := range rels {
r := strings.Split(rel, ",")
if len(r) < 2 {
}
}
- epno := misc.ParseEpisodeList(parts[23])
+ epno := misc.ParseEpisodeList(parts[24])
fid := FID(ints[0])
aid := AID(ints[1])
eid := EID(ints[2])
gid := GID(ints[3])
+ lid := LID(ints[4])
if !epno[0].Start.ContainsEpisodes(epno[0].End) || len(epno) > 1 || len(relList) > 0 {
// epno is broken -- we need to sanitize it
fids = fids[1:]
// Only entry was API error
if len(fids) == 0 {
- return nil
+ return false
}
}
sort.Sort(sort.IntSlice(fids))
}
version := FileVersion(1)
- switch i := ints[6]; {
- case i&fileStateV5 != 0:
+ switch i := ints[7]; {
+ case i&stateV5 != 0:
version = 5
- case i&fileStateV4 != 0:
+ case i&stateV4 != 0:
version = 4
- case i&fileStateV3 != 0:
+ case i&stateV3 != 0:
version = 3
- case i&fileStateV2 != 0:
+ case i&stateV2 != 0:
version = 2
}
- // codecs (parts[13]), bitrates (ints[14]), langs (parts[19])
- codecs := strings.Split(parts[13], "'")
- bitrates := strings.Split(parts[14], "'")
- alangs := strings.Split(parts[19], "'")
+ codecs := strings.Split(parts[14], "'")
+ bitrates := strings.Split(parts[15], "'")
+ alangs := strings.Split(parts[20], "'")
streams := make([]AudioStream, len(codecs))
for i := range streams {
br, _ := strconv.ParseInt(bitrates[i], 10, 32)
}
}
- sl := strings.Split(parts[20], "'")
+ sl := strings.Split(parts[21], "'")
slangs := make([]Language, len(sl))
for i := range sl {
slangs[i] = Language(sl[i])
}
- depth := int(ints[11])
+ depth := int(ints[12])
if depth == 0 {
depth = 8
}
- res := strings.Split(parts[17], "x")
+ res := strings.Split(parts[18], "x")
width, _ := strconv.ParseInt(res[0], 10, 32)
height, _ := strconv.ParseInt(res[1], 10, 32)
video := VideoInfo{
- Bitrate: int(ints[16]),
- Codec: sanitizeCodec(parts[15]),
+ Bitrate: int(ints[17]),
+ Codec: sanitizeCodec(parts[16]),
ColorDepth: depth,
Resolution: image.Rect(0, 0, int(width), int(height)),
}
- return &File{
+ lidMap := LIDMap{}
+ if *f != nil {
+ lidMap = (*f).LID
+ }
+
+ uid := <-uidChan
+ if uid != 0 {
+ lidMap[uid] = lid
+ }
+
+ *f = &File{
FID: fid,
AID: aid,
EID: eid,
GID: gid,
+ LID: lidMap,
EpisodeNumber: epno,
RelatedEpisodes: related,
- Deprecated: ints[5] != 0,
+ Deprecated: ints[6] != 0,
- CRCMatch: ints[6]&fileStateCRCOK != 0,
- BadCRC: ints[6]&fileStateCRCERR != 0,
+ CRCMatch: ints[7]&stateCRCOK != 0,
+ BadCRC: ints[7]&stateCRCERR != 0,
Version: version,
- Uncensored: ints[6]&fileStateUncensored != 0,
- Censored: ints[6]&fileStateCensored != 0,
+ Uncensored: ints[7]&stateUncensored != 0,
+ Censored: ints[7]&stateCensored != 0,
Incomplete: video.Resolution.Empty(),
- Filesize: ints[7],
- Ed2kHash: parts[8],
- SHA1Hash: parts[9],
- CRC32: parts[10],
+ Filesize: ints[8],
+ Ed2kHash: parts[9],
+ SHA1Hash: parts[10],
+ CRC32: parts[11],
- Source: FileSource(parts[12]),
+ Source: FileSource(parts[13]),
AudioStreams: streams,
SubtitleLanguages: slangs,
VideoInfo: video,
- FileExtension: parts[18],
+ FileExtension: parts[19],
- Length: time.Duration(ints[21]) * time.Second,
- AirDate: time.Unix(ints[22], 0),
+ Length: time.Duration(ints[22]) * time.Second,
+ AirDate: time.Unix(ints[23], 0),
}
+ return true
}