mirror of
https://github.com/wabbajack-tools/wabbajack.git
synced 2024-08-30 18:42:17 +00:00
107 lines
3.2 KiB
C#
107 lines
3.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using GameFinder.StoreHandlers.EGS;
|
|
using GameFinder.StoreHandlers.GOG;
|
|
using GameFinder.StoreHandlers.Origin;
|
|
using GameFinder.StoreHandlers.Steam;
|
|
using Microsoft.Extensions.Logging;
|
|
using Wabbajack.DTOs;
|
|
using Wabbajack.Paths;
|
|
|
|
namespace Wabbajack.Downloaders.GameFile
|
|
{
|
|
public class GameLocator : IGameLocator
|
|
{
|
|
private readonly Dictionary<Game, AbsolutePath> _locationCache;
|
|
private readonly ILogger<GameLocator> _logger;
|
|
private readonly SteamHandler _steam;
|
|
private readonly OriginHandler? _origin;
|
|
private readonly GOGHandler _gog;
|
|
private readonly EGSHandler _egs;
|
|
|
|
public GameLocator(ILogger<GameLocator> logger)
|
|
{
|
|
_logger = logger;
|
|
_steam = new SteamHandler(logger);
|
|
|
|
if (OperatingSystem.IsWindows())
|
|
{
|
|
_origin = new OriginHandler(true, false, logger);
|
|
_gog = new GOGHandler(logger);
|
|
|
|
_egs = new EGSHandler(logger);
|
|
}
|
|
|
|
_locationCache = new Dictionary<Game, AbsolutePath>();
|
|
|
|
_steam.FindAllGames();
|
|
_origin?.FindAllGames();
|
|
_gog?.FindAllGames();
|
|
_egs?.FindAllGames();
|
|
}
|
|
|
|
public AbsolutePath GameLocation(Game game)
|
|
{
|
|
if (TryFindLocation(game, out var path))
|
|
return path;
|
|
throw new Exception($"Can't find game {game}");
|
|
}
|
|
|
|
public bool IsInstalled(Game game)
|
|
{
|
|
return TryFindLocation(game, out _);
|
|
}
|
|
|
|
public bool TryFindLocation(Game game, out AbsolutePath path)
|
|
{
|
|
lock (_locationCache)
|
|
{
|
|
if (_locationCache.TryGetValue(game, out path))
|
|
return true;
|
|
|
|
if (TryFindLocationInner(game, out path))
|
|
{
|
|
_locationCache.Add(game, path);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private bool TryFindLocationInner(Game game, out AbsolutePath path)
|
|
{
|
|
var metaData = game.MetaData();
|
|
foreach (var steamGame in _steam.Games.Where(steamGame => metaData.SteamIDs.Contains(steamGame.ID)))
|
|
{
|
|
path = steamGame!.Path.ToAbsolutePath();
|
|
return true;
|
|
}
|
|
|
|
foreach (var gogGame in _gog.Games.Where(gogGame => metaData.GOGIDs.Contains(gogGame.GameID)))
|
|
{
|
|
path = gogGame!.Path.ToAbsolutePath();
|
|
return true;
|
|
}
|
|
|
|
foreach (var egsGame in _egs.Games.Where(egsGame => metaData.EpicGameStoreIDs.Contains(egsGame.CatalogItemId)))
|
|
{
|
|
path = egsGame!.Path.ToAbsolutePath();
|
|
return true;
|
|
}
|
|
|
|
if (_origin != null)
|
|
{
|
|
foreach (var originGame in _origin.Games.Where(originGame => metaData.EpicGameStoreIDs.Contains(originGame.Id)))
|
|
{
|
|
path = originGame.Path.ToAbsolutePath();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
path = default;
|
|
return false;
|
|
}
|
|
}
|
|
} |