Skip to content

Commit 88b9153

Browse files
mateofmateof
committed
feat: add playlists
1 parent 191bff6 commit 88b9153

15 files changed

Lines changed: 2194 additions & 29 deletions

‎TelegramDownloader/Controllers/FileController.cs‎

Lines changed: 45 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -240,29 +240,56 @@ public async Task<IActionResult> GetFile(string id, string? idChannel, string? i
240240
var file = _fs.ExistFileIntempFolder($"{idChannel}-{idFile}-{id}");
241241
if ( file == null )
242242
{
243-
// HttpResponseMessage fullResponse = new HttpResponseMessage(HttpStatusCode.OK); // Request.CreateResponse(HttpStatusCode.OK);
244-
TL.Message idM = await _ts.getMessageFile(idChannel, Convert.ToInt32(idFile));
245-
ChatMessages cm = new ChatMessages();
246-
cm.message = idM;
247243
string filePath = System.IO.Path.Combine(FileService.TEMPDIR, "_temp", $"{idChannel}-{idFile}-{id}");
248-
file = new FileStream(filePath, FileMode.Create, FileAccess.ReadWrite);
249-
DownloadModel dm = new DownloadModel();
250-
dm.tis = _tis;
251-
dm.startDate = DateTime.Now;
252-
dm.path = filePath;
253-
if (cm.message is Message msgBase)
244+
245+
// Check if file exists (might be downloading by another request - race condition)
246+
if (System.IO.File.Exists(filePath))
254247
{
255-
if (msgBase.media is MessageMediaDocument mediaDoc &&
256-
mediaDoc.document is TL.Document doc)
248+
// File exists, wait for it to be ready and return it
249+
for (int i = 0; i < 60; i++) // Wait up to 30 seconds
257250
{
258-
dm._size = doc.size;
259-
dm.name = doc.Filename;
251+
await Task.Delay(500);
252+
try
253+
{
254+
file = new FileStream(filePath, FileMode.Open, FileAccess.Read, FileShare.ReadWrite);
255+
break;
256+
}
257+
catch (IOException)
258+
{
259+
// File still being written, wait more
260+
}
261+
}
262+
if (file == null)
263+
{
264+
return StatusCode(503, "File is being processed, please try again");
260265
}
261266
}
262-
dm.channelName = _ts.getChatName(Convert.ToInt64(idChannel));
263-
_tis.addToDownloadList(dm);
264-
await _ts.DownloadFileAndReturn(cm, file, model: dm);
265-
file.Position = 0;
267+
else
268+
{
269+
// File doesn't exist, download it
270+
TL.Message idM = await _ts.getMessageFile(idChannel, Convert.ToInt32(idFile));
271+
ChatMessages cm = new ChatMessages();
272+
cm.message = idM;
273+
274+
file = new FileStream(filePath, FileMode.Create, FileAccess.ReadWrite, FileShare.ReadWrite);
275+
DownloadModel dm = new DownloadModel();
276+
dm.tis = _tis;
277+
dm.startDate = DateTime.Now;
278+
dm.path = filePath;
279+
if (cm.message is Message msgBase)
280+
{
281+
if (msgBase.media is MessageMediaDocument mediaDoc &&
282+
mediaDoc.document is TL.Document doc)
283+
{
284+
dm._size = doc.size;
285+
dm.name = doc.Filename;
286+
}
287+
}
288+
dm.channelName = _ts.getChatName(Convert.ToInt64(idChannel));
289+
_tis.addToDownloadList(dm);
290+
await _ts.DownloadFileAndReturn(cm, file, model: dm);
291+
file.Position = 0;
292+
}
266293
}
267294
var request = HttpContext.Request;
268295
var rangeHeader = request.Headers["Range"].ToString();

‎TelegramDownloader/Data/FileService.cs‎

Lines changed: 121 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2264,5 +2264,126 @@ public virtual Task<int> PreloadFilesToTemp(string channelId, List<Syncfusion.Bl
22642264
{
22652265
throw new NotImplementedException("PreloadFilesToTemp is implemented in FileServiceV2");
22662266
}
2267+
2268+
public async Task DownloadPlaylistToLocal(PlaylistModel playlist, string destinationFolder)
2269+
{
2270+
_logger.LogInformation("Starting playlist download - Playlist: {Name}, Tracks: {Count}, Destination: {Dest}",
2271+
playlist.Name, playlist.TrackCount, destinationFolder);
2272+
2273+
NotificationModel nm = new NotificationModel();
2274+
nm.sendEvent(new Notification($"Starting download of playlist '{playlist.Name}'", "Playlist Download", NotificationTypes.Info));
2275+
2276+
try
2277+
{
2278+
// Create destination folder if it doesn't exist
2279+
Directory.CreateDirectory(destinationFolder);
2280+
2281+
int downloaded = 0;
2282+
int failed = 0;
2283+
2284+
foreach (var track in playlist.Tracks)
2285+
{
2286+
try
2287+
{
2288+
string destPath = System.IO.Path.Combine(destinationFolder, track.FileName);
2289+
2290+
// Check if file already exists
2291+
if (File.Exists(destPath))
2292+
{
2293+
_logger.LogInformation("File already exists, skipping: {FileName}", track.FileName);
2294+
downloaded++;
2295+
continue;
2296+
}
2297+
2298+
// Handle local files - copy instead of download
2299+
if (track.IsLocalFile && !string.IsNullOrEmpty(track.DirectUrl))
2300+
{
2301+
try
2302+
{
2303+
// Extract local path from URL
2304+
var uri = new Uri(track.DirectUrl);
2305+
var localPath = Uri.UnescapeDataString(uri.AbsolutePath);
2306+
// Remove the static file prefix to get the actual file path
2307+
if (localPath.Contains(STATICRELATIVELOCALDIR.Replace("\\", "/")))
2308+
{
2309+
var relativePath = localPath.Substring(localPath.IndexOf(STATICRELATIVELOCALDIR.Replace("\\", "/")) + STATICRELATIVELOCALDIR.Length);
2310+
var sourcePath = System.IO.Path.Combine(Directory.GetCurrentDirectory(), STATICRELATIVELOCALDIR, relativePath.TrimStart('/'));
2311+
if (File.Exists(sourcePath))
2312+
{
2313+
File.Copy(sourcePath, destPath);
2314+
downloaded++;
2315+
_logger.LogInformation("Copied local track {Downloaded}/{Total}: {FileName}",
2316+
downloaded, playlist.TrackCount, track.FileName);
2317+
continue;
2318+
}
2319+
}
2320+
}
2321+
catch (Exception ex)
2322+
{
2323+
_logger.LogWarning(ex, "Could not copy local file: {FileName}", track.FileName);
2324+
}
2325+
// If we couldn't copy, count as skipped (local file already exists somewhere)
2326+
downloaded++;
2327+
continue;
2328+
}
2329+
2330+
// Get file info from MongoDB for Telegram files
2331+
var file = await getItemById(track.ChannelId, track.FileId);
2332+
if (file == null)
2333+
{
2334+
_logger.LogWarning("File not found for track: {FileName}", track.FileName);
2335+
failed++;
2336+
continue;
2337+
}
2338+
2339+
if (file.isSplit)
2340+
{
2341+
// Handle split files
2342+
int i = 1;
2343+
List<string> splitPaths = new List<string>();
2344+
foreach (int messageId in file.ListMessageId)
2345+
{
2346+
string filePathPart = System.IO.Path.Combine(destinationFolder, $"({i})" + track.FileName);
2347+
await downloadFromTelegram(track.ChannelId, messageId, filePathPart, file);
2348+
splitPaths.Add(filePathPart);
2349+
i++;
2350+
}
2351+
await mergeFileStreamAsync(splitPaths, destPath);
2352+
foreach (string filePath in splitPaths)
2353+
{
2354+
File.Delete(filePath);
2355+
}
2356+
}
2357+
else
2358+
{
2359+
// Download single file
2360+
await downloadFromTelegram(track.ChannelId, (int)file.MessageId, destPath, file);
2361+
}
2362+
2363+
downloaded++;
2364+
_logger.LogInformation("Downloaded track {Downloaded}/{Total}: {FileName}",
2365+
downloaded, playlist.TrackCount, track.FileName);
2366+
}
2367+
catch (Exception ex)
2368+
{
2369+
_logger.LogError(ex, "Error downloading track: {FileName}", track.FileName);
2370+
failed++;
2371+
}
2372+
}
2373+
2374+
string message = failed > 0
2375+
? $"Playlist '{playlist.Name}' completed: {downloaded} downloaded, {failed} failed"
2376+
: $"Playlist '{playlist.Name}' downloaded successfully ({downloaded} tracks)";
2377+
2378+
nm.sendEvent(new Notification(message, "Playlist Download",
2379+
failed > 0 ? NotificationTypes.Warn : NotificationTypes.Success));
2380+
}
2381+
catch (Exception ex)
2382+
{
2383+
_logger.LogError(ex, "Error downloading playlist: {Name}", playlist.Name);
2384+
nm.sendEvent(new Notification($"Error downloading playlist: {ex.Message}", "Playlist Download", NotificationTypes.Error));
2385+
throw;
2386+
}
2387+
}
22672388
}
22682389
}

‎TelegramDownloader/Data/IFileService.cs‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,5 +44,6 @@ public interface IFileService
4444
Task refreshChannelFIles(string channelId, bool force = false);
4545
bool isChannelRefreshing(string channelId);
4646
Task<int> PreloadFilesToTemp(string channelId, List<FileManagerDirectoryContent> items);
47+
Task DownloadPlaylistToLocal(PlaylistModel playlist, string destinationFolder);
4748
}
4849
}

‎TelegramDownloader/Data/db/DbService.cs‎

Lines changed: 132 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1181,5 +1181,137 @@ private string CalculateFilterId(string parentId, Dictionary<string, BsonFileMan
11811181

11821182
#endregion
11831183

1184+
#region Playlist Operations
1185+
1186+
private const string PLAYLIST_COLLECTION = "playlists";
1187+
1188+
public async Task<PlaylistModel> CreatePlaylist(PlaylistModel playlist)
1189+
{
1190+
if (string.IsNullOrEmpty(playlist.Id))
1191+
{
1192+
playlist.Id = MongoDB.Bson.ObjectId.GenerateNewId().ToString();
1193+
}
1194+
playlist.DateCreated = DateTime.Now;
1195+
playlist.DateModified = DateTime.Now;
1196+
1197+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1198+
await collection.InsertOneAsync(playlist);
1199+
1200+
_logger.LogInformation("Created playlist {Name} with Id {Id}", playlist.Name, playlist.Id);
1201+
return playlist;
1202+
}
1203+
1204+
public async Task<List<PlaylistModel>> GetAllPlaylists()
1205+
{
1206+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1207+
var playlists = await (await collection.FindAsync(Builders<PlaylistModel>.Filter.Empty)).ToListAsync();
1208+
return playlists.OrderByDescending(p => p.DateModified).ToList();
1209+
}
1210+
1211+
public async Task<PlaylistModel?> GetPlaylistById(string id)
1212+
{
1213+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1214+
return await (await collection.FindAsync(Builders<PlaylistModel>.Filter.Eq(x => x.Id, id))).FirstOrDefaultAsync();
1215+
}
1216+
1217+
public async Task UpdatePlaylist(PlaylistModel playlist)
1218+
{
1219+
playlist.DateModified = DateTime.Now;
1220+
1221+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1222+
var filter = Builders<PlaylistModel>.Filter.Eq(x => x.Id, playlist.Id);
1223+
await collection.ReplaceOneAsync(filter, playlist, new ReplaceOptions { IsUpsert = false });
1224+
1225+
_logger.LogInformation("Updated playlist {Name} with Id {Id}", playlist.Name, playlist.Id);
1226+
}
1227+
1228+
public async Task DeletePlaylist(string id)
1229+
{
1230+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1231+
await collection.DeleteOneAsync(Builders<PlaylistModel>.Filter.Eq(x => x.Id, id));
1232+
1233+
_logger.LogInformation("Deleted playlist with Id {Id}", id);
1234+
}
1235+
1236+
public async Task AddTrackToPlaylist(string playlistId, PlaylistTrackModel track)
1237+
{
1238+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1239+
1240+
// Get current playlist to determine next order
1241+
var playlist = await GetPlaylistById(playlistId);
1242+
if (playlist == null)
1243+
{
1244+
_logger.LogWarning("Playlist {Id} not found when adding track", playlistId);
1245+
return;
1246+
}
1247+
1248+
// Set order to be at the end
1249+
track.Order = playlist.Tracks.Count;
1250+
track.DateAdded = DateTime.Now;
1251+
1252+
var filter = Builders<PlaylistModel>.Filter.Eq(x => x.Id, playlistId);
1253+
var update = Builders<PlaylistModel>.Update
1254+
.Push(x => x.Tracks, track)
1255+
.Set(x => x.DateModified, DateTime.Now);
1256+
1257+
await collection.UpdateOneAsync(filter, update);
1258+
1259+
_logger.LogInformation("Added track {FileName} to playlist {Id}", track.FileName, playlistId);
1260+
}
1261+
1262+
public async Task RemoveTrackFromPlaylist(string playlistId, string fileId)
1263+
{
1264+
var collection = getDatabase(CONFIG_DB_NAME).GetCollection<PlaylistModel>(PLAYLIST_COLLECTION);
1265+
1266+
var filter = Builders<PlaylistModel>.Filter.Eq(x => x.Id, playlistId);
1267+
var update = Builders<PlaylistModel>.Update
1268+
.PullFilter(x => x.Tracks, t => t.FileId == fileId)
1269+
.Set(x => x.DateModified, DateTime.Now);
1270+
1271+
await collection.UpdateOneAsync(filter, update);
1272+
1273+
// Reorder remaining tracks
1274+
var playlist = await GetPlaylistById(playlistId);
1275+
if (playlist != null && playlist.Tracks.Count > 0)
1276+
{
1277+
for (int i = 0; i < playlist.Tracks.Count; i++)
1278+
{
1279+
playlist.Tracks[i].Order = i;
1280+
}
1281+
await UpdatePlaylist(playlist);
1282+
}
1283+
1284+
_logger.LogInformation("Removed track {FileId} from playlist {Id}", fileId, playlistId);
1285+
}
1286+
1287+
public async Task ReorderPlaylistTracks(string playlistId, List<string> orderedFileIds)
1288+
{
1289+
var playlist = await GetPlaylistById(playlistId);
1290+
if (playlist == null)
1291+
{
1292+
_logger.LogWarning("Playlist {Id} not found when reordering tracks", playlistId);
1293+
return;
1294+
}
1295+
1296+
// Reorder tracks based on the provided order
1297+
var reorderedTracks = new List<PlaylistTrackModel>();
1298+
for (int i = 0; i < orderedFileIds.Count; i++)
1299+
{
1300+
var track = playlist.Tracks.FirstOrDefault(t => t.FileId == orderedFileIds[i]);
1301+
if (track != null)
1302+
{
1303+
track.Order = i;
1304+
reorderedTracks.Add(track);
1305+
}
1306+
}
1307+
1308+
playlist.Tracks = reorderedTracks;
1309+
await UpdatePlaylist(playlist);
1310+
1311+
_logger.LogInformation("Reordered {Count} tracks in playlist {Id}", reorderedTracks.Count, playlistId);
1312+
}
1313+
1314+
#endregion
1315+
11841316
}
11851317
}

‎TelegramDownloader/Data/db/IDbService.cs‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,16 @@ public interface IDbService
7474
Task<DatabaseStats> GetDatabaseStats(string dbName);
7575
Task<FilterPathAnalysisResult> AnalyzeFilterPaths(string dbName, string collectionName = "directory");
7676
Task<int> RepairFilterPaths(string dbName, string collectionName = "directory");
77+
78+
// Playlist operations
79+
Task<PlaylistModel> CreatePlaylist(PlaylistModel playlist);
80+
Task<List<PlaylistModel>> GetAllPlaylists();
81+
Task<PlaylistModel?> GetPlaylistById(string id);
82+
Task UpdatePlaylist(PlaylistModel playlist);
83+
Task DeletePlaylist(string id);
84+
Task AddTrackToPlaylist(string playlistId, PlaylistTrackModel track);
85+
Task RemoveTrackFromPlaylist(string playlistId, string fileId);
86+
Task ReorderPlaylistTracks(string playlistId, List<string> orderedFileIds);
7787
}
7888

7989
public class DatabaseStats

0 commit comments

Comments
 (0)