88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"git.milar.in/milarin/anilist"
|
||
|
|
"git.milar.in/nyaanime/model"
|
||
|
|
)
|
||
|
|
|
||
|
|
type AnimePathPatternData struct {
|
||
|
|
Title anilist.MediaTitle
|
||
|
|
Episode int
|
||
|
|
Ext string
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetAnimeEpFilepath(animeEp model.AnimeEpisode, ext string) string {
|
||
|
|
tmplData := AnimePathPatternData{
|
||
|
|
Title: animeEp.Anime.Title,
|
||
|
|
Episode: animeEp.Episode,
|
||
|
|
Ext: ext,
|
||
|
|
}
|
||
|
|
|
||
|
|
b := new(strings.Builder)
|
||
|
|
if err := AnimeEpFilepathPattern.Execute(b, tmplData); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return b.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
func AnimeEpExistsLocally(animeEp model.AnimeEpisode) bool {
|
||
|
|
animeEpPath := filepath.Join(AnimePath, GetAnimeEpFilepath(animeEp, "*"))
|
||
|
|
|
||
|
|
files, err := filepath.Glob(animeEpPath)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return len(files) > 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetLocalAnimeEpProperties(animeEp model.AnimeEpisode) (*FilePriority, bool, error) {
|
||
|
|
animeEpPath := filepath.Join(AnimePath, GetAnimeEpFilepath(animeEp, "*"))
|
||
|
|
files, err := filepath.Glob(animeEpPath)
|
||
|
|
if err != nil {
|
||
|
|
return nil, false, ErrInvalidGlobSyntax.Wrap(err, animeEpPath)
|
||
|
|
}
|
||
|
|
|
||
|
|
var mostPrio *FilePriority
|
||
|
|
|
||
|
|
for _, file := range files {
|
||
|
|
props, err := AnalyzeFile(file)
|
||
|
|
if err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if !HasFileEssentialProperties(props) {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
fp := NewFilePriority(props)
|
||
|
|
|
||
|
|
if mostPrio == nil || fp.Priority > mostPrio.Priority {
|
||
|
|
mostPrio = fp
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return mostPrio, mostPrio != nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO cache?
|
||
|
|
func TorrentFileDownloading(torrent *model.ParsedTorrent) bool {
|
||
|
|
torrentFile := filepath.Join(TorrentPath, fmt.Sprintf("%s.torrent", torrent.Torrent.ID))
|
||
|
|
|
||
|
|
if _, err := os.Stat(torrentFile); err == nil {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
addedTorrentFile := filepath.Join(TorrentPath, fmt.Sprintf("%s.torrent.added", torrent.Torrent.ID))
|
||
|
|
if _, err := os.Stat(addedTorrentFile); err == nil {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|