From c31c3183214107fbbc8aefcd9f5c41bf05e40bdb Mon Sep 17 00:00:00 2001 From: YouROK <8yourok8@mail.ru> Date: Thu, 17 Jun 2021 10:57:28 +0300 Subject: [PATCH] set torrent name if empty title --- server/torr/apihelper.go | 40 ++++++++++++++++++++++++++++------------ 1 file changed, 28 insertions(+), 12 deletions(-) diff --git a/server/torr/apihelper.go b/server/torr/apihelper.go index 5229045..65cb0ea 100644 --- a/server/torr/apihelper.go +++ b/server/torr/apihelper.go @@ -107,21 +107,37 @@ func GetTorrent(hashHex string) *Torrent { func SetTorrent(hashHex, title, poster, data string) *Torrent { hash := metainfo.NewHashFromHex(hashHex) - tor := bts.GetTorrent(hash) - if tor != nil { - tor.Title = title - tor.Poster = poster - tor.Data = data + torr := bts.GetTorrent(hash) + torrDb := GetTorrentDB(hash) + + if title == "" && torr == nil && torrDb != nil { + torr = GetTorrent(hashHex) + torr.GotInfo() + if torr.Torrent != nil && torr.Torrent.Info() != nil { + title = torr.Info().Name + } } - tor = GetTorrentDB(hash) - if tor != nil { - tor.Title = title - tor.Poster = poster - tor.Data = data - AddTorrentDB(tor) + if torr != nil { + if title == "" && torr.Torrent != nil && torr.Torrent.Info() != nil { + title = torr.Info().Name + } + torr.Title = title + torr.Poster = poster + torr.Data = data + } + + if torrDb != nil { + torrDb.Title = title + torrDb.Poster = poster + torrDb.Data = data + AddTorrentDB(torrDb) + } + if torr != nil { + return torr + } else { + return torrDb } - return tor } func RemTorrent(hashHex string) {