Restore playtime resync after download complete

It will work only during repairing.
hawkeye116477 · Oct 23, 2024 · 5ebc0d4 · 5ebc0d4
1 parent b4b3f0b
commit 5ebc0d4
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/LegendaryDownloadManager.xaml.cs b/src/LegendaryDownloadManager.xaml.cs
@@ -20,6 +20,7 @@
 using System.Windows.Input;
 using Playnite.SDK.Plugins;
 using System.Collections.Specialized;
+using LegendaryLibraryNS.Services;
 
 namespace LegendaryLibraryNS
 {
@@ -543,6 +544,29 @@ public async Task Install(DownloadManagerData.Download taskData)
                                             game.Version = installedGameInfo.Version;
                                             game.InstallSize = (ulong?)installedGameInfo.Install_size;
                                             game.IsInstalled = true;
+                                            var playtimeSyncEnabled = false;
+                                            if (downloadProperties.downloadAction == DownloadAction.Repair)
+                                            {
+                                                if (playniteAPI.ApplicationSettings.PlaytimeImportMode != PlaytimeImportMode.Never)
+                                                {
+                                                    playtimeSyncEnabled = LegendaryLibrary.GetSettings().SyncPlaytime;
+                                                    var gameSettings = LegendaryGameSettingsView.LoadGameSettings(game.GameId);
+                                                    if (gameSettings?.AutoSyncPlaytime != null)
+                                                    {
+                                                        playtimeSyncEnabled = (bool)gameSettings.AutoSyncPlaytime;
+                                                    }
+                                                }
+                                                if (playtimeSyncEnabled)
+                                                {
+                                                    var accountApi = new EpicAccountClient(playniteAPI, LegendaryLauncher.TokensPath);
+                                                    var playtimeItems = await accountApi.GetPlaytimeItems();
+                                                    var playtimeItem = playtimeItems?.FirstOrDefault(x => x.artifactId == gameID);
+                                                    if (playtimeItem != null)
+                                                    {
+                                                        game.Playtime = playtimeItem.totalTime;
+                                                    }
+                                                }
+                                            }
                                             // Some games need specific key in registry, otherwise they can't launch
                                             try
                                             {