danvics/rdt-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add Getting DL link, tracking via hash

Browse source
This commit is contained in:
Sam Heinz 2024-08-27 19:11:24 +10:00
parent fc9ea4c29a
commit 395f83847e
1 changed files with 13 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

64
server/RdtClient.Service/Services/TorrentClients/TorBoxTorrentClient.cs
View file

@ -127,14 +127,14 @@ public class TorBoxTorrentClient(ILogger<TorBoxTorrentClient> logger, IHttpClien
{ {
var result = await GetClient().Torrents.AddMagnetAsync(magnetLink); var result = await GetClient().Torrents.AddMagnetAsync(magnetLink);
return result.Data?.Torrent_ID.ToString()!; return result.Data?.Hash?.ToString()!;
} }
public async Task<String> AddFile(Byte[] bytes) public async Task<String> AddFile(Byte[] bytes)
{ {
var result = await GetClient().Torrents.AddFileAsync(bytes); var result = await GetClient().Torrents.AddFileAsync(bytes);
return result.Data?.Torrent_ID.ToString()!; return result.Data?.Hash?.ToString()!;
} }
public Task<IList<TorrentClientAvailableFile>> GetAvailableFiles(String hash) public Task<IList<TorrentClientAvailableFile>> GetAvailableFiles(String hash)
@ -157,12 +157,12 @@ public class TorBoxTorrentClient(ILogger<TorBoxTorrentClient> logger, IHttpClien
{ {
var result = await GetClient().Unrestrict.LinkAsync(link); var result = await GetClient().Unrestrict.LinkAsync(link);
if (result.Download == null) if (result.Error != null)
{ {
throw new($"Unrestrict returned an invalid download"); throw new($"Unrestrict returned an invalid download");
} }
return result.Download; return result.Data!;
} }
public async Task<Data.Models.Data.Torrent> UpdateData(Data.Models.Data.Torrent torrent, TorrentClientTorrent? torrentClientTorrent) public async Task<Data.Models.Data.Torrent> UpdateData(Data.Models.Data.Torrent torrent, TorrentClientTorrent? torrentClientTorrent)
@ -250,63 +250,25 @@ public class TorBoxTorrentClient(ILogger<TorBoxTorrentClient> logger, IHttpClien
public async Task<IList<String>?> GetDownloadLinks(Data.Models.Data.Torrent torrent) public async Task<IList<String>?> GetDownloadLinks(Data.Models.Data.Torrent torrent)
{ {
if (torrent.RdId == null) if (torrent.Hash == null)
{ {
return null; return null;
} }
var rdTorrent = await GetInfo(torrent.RdId); var rdTorrent = await GetInfo(torrent.Hash);
if (rdTorrent.Links == null) var files = new List<String>();
if (rdTorrent.Files?.Count != 0)
{ {
return null; foreach (var file in rdTorrent.Files!)
}
var downloadLinks = rdTorrent.Links.Where(m => !String.IsNullOrWhiteSpace(m)).ToList();
Log($"Found {downloadLinks.Count} links", torrent);
foreach (var link in downloadLinks)
{
Log($"{link}", torrent);
}
Log($"Torrent has {torrent.Files.Count(m => m.Selected)} selected files out of {torrent.Files.Count} files, found {downloadLinks.Count} links, torrent ended: {torrent.RdEnded}", torrent);
// Check if all the links are set that have been selected
if (torrent.Files.Count(m => m.Selected) == downloadLinks.Count)
{
Log($"Matched {torrent.Files.Count(m => m.Selected)} selected files expected files to {downloadLinks.Count} found files", torrent);
return downloadLinks;
}
// Check if all all the links are set for manual selection
if (torrent.ManualFiles.Count == downloadLinks.Count)
{
Log($"Matched {torrent.ManualFiles.Count} manual files expected files to {downloadLinks.Count} found files", torrent);
return downloadLinks;
}
// If there is only 1 link, delay for 1 minute to see if more links pop up.
if (downloadLinks.Count == 1 && torrent.RdEnded.HasValue)
{
var expired = DateTime.UtcNow - torrent.RdEnded.Value.ToUniversalTime();
Log($"Waiting to see if more links appear, checked for {expired.TotalSeconds} seconds", torrent);
if (expired.TotalSeconds > 60.0)
{ {
Log($"Waited long enough", torrent); var newFile = new List<String> { rdTorrent.Id, file.Id.ToString() };
files.Add(newFile.ToString()!);
return downloadLinks;
} }
} }
Log($"Did not find any suiteable download links", torrent); return files;
return null;
} }
private DateTimeOffset? ChangeTimeZone(DateTimeOffset? dateTimeOffset) private DateTimeOffset? ChangeTimeZone(DateTimeOffset? dateTimeOffset)