node build fixed
This commit is contained in:
ra_ma
149
seanime-2.9.10/internal/torrents/seadex/provider.go
Normal file
149
seanime-2.9.10/internal/torrents/seadex/provider.go
Normal file
@@ -0,0 +1,149 @@
|
||||
package seadex
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/5rahim/habari"
|
||||
"github.com/rs/zerolog"
|
||||
"net/http"
|
||||
"seanime/internal/torrents/nyaa"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
hibiketorrent "seanime/internal/extension/hibike/torrent"
|
||||
)
|
||||
|
||||
const (
|
||||
ProviderName = "seadex"
|
||||
)
|
||||
|
||||
type Provider struct {
|
||||
logger *zerolog.Logger
|
||||
seadex *SeaDex
|
||||
}
|
||||
|
||||
func NewProvider(logger *zerolog.Logger) hibiketorrent.AnimeProvider {
|
||||
return &Provider{
|
||||
logger: logger,
|
||||
seadex: New(logger),
|
||||
}
|
||||
}
|
||||
|
||||
func (n *Provider) GetSettings() hibiketorrent.AnimeProviderSettings {
|
||||
return hibiketorrent.AnimeProviderSettings{
|
||||
Type: hibiketorrent.AnimeProviderTypeSpecial,
|
||||
CanSmartSearch: true, // Setting to true to allow previews
|
||||
SupportsAdult: false,
|
||||
}
|
||||
}
|
||||
|
||||
func (n *Provider) GetType() hibiketorrent.AnimeProviderType {
|
||||
return hibiketorrent.AnimeProviderTypeSpecial
|
||||
}
|
||||
|
||||
func (n *Provider) GetLatest() (ret []*hibiketorrent.AnimeTorrent, err error) {
|
||||
return
|
||||
}
|
||||
|
||||
func (n *Provider) Search(opts hibiketorrent.AnimeSearchOptions) (ret []*hibiketorrent.AnimeTorrent, err error) {
|
||||
return n.findTorrents(&opts.Media)
|
||||
}
|
||||
|
||||
func (n *Provider) SmartSearch(opts hibiketorrent.AnimeSmartSearchOptions) (ret []*hibiketorrent.AnimeTorrent, err error) {
|
||||
return n.findTorrents(&opts.Media)
|
||||
}
|
||||
|
||||
func (n *Provider) findTorrents(media *hibiketorrent.Media) (ret []*hibiketorrent.AnimeTorrent, err error) {
|
||||
seadexTorrents, err := n.seadex.FetchTorrents(media.ID, media.RomajiTitle)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
wg := sync.WaitGroup{}
|
||||
mu := sync.Mutex{}
|
||||
wg.Add(len(seadexTorrents))
|
||||
|
||||
for _, t := range seadexTorrents {
|
||||
go func(t *Torrent) {
|
||||
defer wg.Done()
|
||||
mu.Lock()
|
||||
ret = append(ret, t.toAnimeTorrent(ProviderName))
|
||||
mu.Unlock()
|
||||
}(t)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------------------------------------------------------//
|
||||
|
||||
func (n *Provider) GetTorrentInfoHash(torrent *hibiketorrent.AnimeTorrent) (string, error) {
|
||||
return torrent.MagnetLink, nil
|
||||
}
|
||||
|
||||
func (n *Provider) GetTorrentMagnetLink(torrent *hibiketorrent.AnimeTorrent) (string, error) {
|
||||
return nyaa.TorrentMagnet(torrent.Link)
|
||||
}
|
||||
|
||||
func (t *Torrent) toAnimeTorrent(providerName string) *hibiketorrent.AnimeTorrent {
|
||||
metadata := habari.Parse(t.Name)
|
||||
|
||||
ret := &hibiketorrent.AnimeTorrent{
|
||||
Name: t.Name,
|
||||
Date: t.Date,
|
||||
Size: 0, // Should be scraped
|
||||
FormattedSize: "", // Should be scraped
|
||||
Seeders: 0, // Should be scraped
|
||||
Leechers: 0, // Should be scraped
|
||||
DownloadCount: 0, // Should be scraped
|
||||
Link: t.Link,
|
||||
DownloadUrl: "", // Should be scraped
|
||||
InfoHash: t.InfoHash,
|
||||
MagnetLink: "", // Should be scraped
|
||||
Resolution: "", // Should be parsed
|
||||
IsBatch: true, // Should be parsed
|
||||
EpisodeNumber: -1, // Should be parsed
|
||||
ReleaseGroup: "", // Should be parsed
|
||||
Provider: providerName,
|
||||
IsBestRelease: true,
|
||||
Confirmed: true,
|
||||
}
|
||||
|
||||
var seeders, leechers, downloads int
|
||||
var title, downloadUrl, formattedSize string
|
||||
|
||||
// Try scraping from Nyaa
|
||||
// Since nyaa tends to be blocked, try for a few seconds only
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
if t.Link != "" {
|
||||
downloadUrl = t.Link
|
||||
|
||||
client := http.DefaultClient
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodGet, ret.Link, nil)
|
||||
if err == nil {
|
||||
resp, err := client.Do(req)
|
||||
if err == nil {
|
||||
defer resp.Body.Close()
|
||||
|
||||
title, seeders, leechers, downloads, formattedSize, _, _, err = nyaa.TorrentInfo(ret.Link)
|
||||
if err == nil && title != "" {
|
||||
ret.Name = title // Override title
|
||||
ret.Seeders = seeders
|
||||
ret.Leechers = leechers
|
||||
ret.DownloadCount = downloads
|
||||
ret.DownloadUrl = downloadUrl
|
||||
ret.Size = 1
|
||||
ret.FormattedSize = formattedSize
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ret.Resolution = metadata.VideoResolution
|
||||
ret.ReleaseGroup = metadata.ReleaseGroup
|
||||
|
||||
return ret
|
||||
}
|
||||
109
seanime-2.9.10/internal/torrents/seadex/seadex.go
Normal file
109
seanime-2.9.10/internal/torrents/seadex/seadex.go
Normal file
@@ -0,0 +1,109 @@
|
||||
package seadex
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"seanime/internal/extension"
|
||||
"seanime/internal/util"
|
||||
"strings"
|
||||
|
||||
"github.com/goccy/go-json"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
type (
|
||||
SeaDex struct {
|
||||
logger *zerolog.Logger
|
||||
uri string
|
||||
}
|
||||
|
||||
Torrent struct {
|
||||
Name string `json:"name"`
|
||||
Date string `json:"date"`
|
||||
Size int64 `json:"size"`
|
||||
Link string `json:"link"`
|
||||
InfoHash string `json:"infoHash"`
|
||||
ReleaseGroup string `json:"releaseGroup,omitempty"`
|
||||
}
|
||||
)
|
||||
|
||||
func New(logger *zerolog.Logger) *SeaDex {
|
||||
return &SeaDex{
|
||||
logger: logger,
|
||||
uri: util.Decode("aHR0cHM6Ly9yZWxlYXNlcy5tb2UvYXBpL2NvbGxlY3Rpb25zL2VudHJpZXMvcmVjb3Jkcw=="),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SeaDex) SetSavedUserConfig(savedConfig *extension.SavedUserConfig) {
|
||||
url, _ := savedConfig.Values["apiUrl"]
|
||||
if url != "" {
|
||||
s.uri = url
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SeaDex) FetchTorrents(mediaId int, title string) (ret []*Torrent, err error) {
|
||||
|
||||
ret = make([]*Torrent, 0)
|
||||
|
||||
records, err := s.fetchRecords(mediaId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(records) == 0 {
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
if len(records[0].Expand.Trs) == 0 {
|
||||
return ret, nil
|
||||
}
|
||||
for _, tr := range records[0].Expand.Trs {
|
||||
if tr.InfoHash == "" || tr.InfoHash == "<redacted>" || tr.Tracker != "Nyaa" || !strings.Contains(tr.URL, "nyaa.si") {
|
||||
continue
|
||||
}
|
||||
ret = append(ret, &Torrent{
|
||||
Name: fmt.Sprintf("[%s] %s%s", tr.ReleaseGroup, title, map[bool]string{true: " [Dual-Audio]", false: ""}[tr.DualAudio]),
|
||||
Date: tr.Created,
|
||||
Size: int64(s.getTorrentSize(tr.Files)),
|
||||
Link: tr.URL,
|
||||
InfoHash: tr.InfoHash,
|
||||
ReleaseGroup: tr.ReleaseGroup,
|
||||
})
|
||||
}
|
||||
|
||||
return ret, nil
|
||||
|
||||
}
|
||||
|
||||
func (s *SeaDex) fetchRecords(mediaId int) (ret []*RecordItem, err error) {
|
||||
|
||||
uri := fmt.Sprintf("%s?page=1&perPage=1&filter=alID%%3D%%22%d%%22&skipTotal=1&expand=trs", s.uri, mediaId)
|
||||
|
||||
resp, err := http.Get(uri)
|
||||
if err != nil {
|
||||
s.logger.Error().Err(err).Msgf("seadex: error getting media records: %v", mediaId)
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var res RecordsResponse
|
||||
if err = json.NewDecoder(resp.Body).Decode(&res); err != nil {
|
||||
s.logger.Error().Err(err).Msgf("seadex: error decoding response: %v", mediaId)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return res.Items, nil
|
||||
}
|
||||
|
||||
func (s *SeaDex) getTorrentSize(fls []*TrFile) int {
|
||||
if fls == nil || len(fls) == 0 {
|
||||
return 0
|
||||
}
|
||||
|
||||
var size int
|
||||
for _, f := range fls {
|
||||
size += f.Length
|
||||
}
|
||||
|
||||
return size
|
||||
}
|
||||
48
seanime-2.9.10/internal/torrents/seadex/seadex_test.go
Normal file
48
seanime-2.9.10/internal/torrents/seadex/seadex_test.go
Normal file
@@ -0,0 +1,48 @@
|
||||
package seadex
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/davecgh/go-spew/spew"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"seanime/internal/api/anilist"
|
||||
"seanime/internal/test_utils"
|
||||
"seanime/internal/util"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestSeaDex(t *testing.T) {
|
||||
test_utils.InitTestProvider(t, test_utils.Anilist())
|
||||
|
||||
anilistClient := anilist.TestGetMockAnilistClient()
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
mediaId int
|
||||
}{
|
||||
{
|
||||
name: "86 - Eighty Six Part 2",
|
||||
mediaId: 131586,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
|
||||
mediaF, err := anilistClient.BaseAnimeByID(context.Background(), &tt.mediaId)
|
||||
if assert.NoErrorf(t, err, "error getting media: %v", tt.mediaId) {
|
||||
|
||||
media := mediaF.GetMedia()
|
||||
|
||||
torrents, err := New(util.NewLogger()).FetchTorrents(tt.mediaId, media.GetRomajiTitleSafe())
|
||||
if assert.NoErrorf(t, err, "error fetching records: %v", tt.mediaId) {
|
||||
|
||||
spew.Dump(torrents)
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
})
|
||||
}
|
||||
|
||||
}
|
||||
42
seanime-2.9.10/internal/torrents/seadex/types.go
Normal file
42
seanime-2.9.10/internal/torrents/seadex/types.go
Normal file
@@ -0,0 +1,42 @@
|
||||
package seadex
|
||||
|
||||
type (
|
||||
RecordsResponse struct {
|
||||
Items []*RecordItem `json:"items"`
|
||||
}
|
||||
|
||||
RecordItem struct {
|
||||
AlID int `json:"alID"`
|
||||
CollectionID string `json:"collectionId"`
|
||||
CollectionName string `json:"collectionName"`
|
||||
Comparison string `json:"comparison"`
|
||||
Created string `json:"created"`
|
||||
Expand struct {
|
||||
Trs []*Tr `json:"trs"`
|
||||
} `json:"expand"`
|
||||
Trs []string `json:"trs"`
|
||||
Updated string `json:"updated"`
|
||||
ID string `json:"id"`
|
||||
Incomplete bool `json:"incomplete"`
|
||||
Notes string `json:"notes"`
|
||||
TheoreticalBest string `json:"theoreticalBest"`
|
||||
}
|
||||
|
||||
Tr struct {
|
||||
Created string `json:"created"`
|
||||
CollectionID string `json:"collectionId"`
|
||||
CollectionName string `json:"collectionName"`
|
||||
DualAudio bool `json:"dualAudio"`
|
||||
Files []*TrFile `json:"files"`
|
||||
ID string `json:"id"`
|
||||
InfoHash string `json:"infoHash"`
|
||||
IsBest bool `json:"isBest"`
|
||||
ReleaseGroup string `json:"releaseGroup"`
|
||||
Tracker string `json:"tracker"`
|
||||
URL string `json:"url"`
|
||||
}
|
||||
TrFile struct {
|
||||
Length int `json:"length"`
|
||||
Name string `json:"name"`
|
||||
}
|
||||
)
|
||||
Reference in New Issue
Block a user