Refactor caching code

This commit is contained in:
Jeddunk 2021-01-13 20:29:12 +01:00
parent cf4c5ea551
commit 3dfc89dfed

View File

@ -27,35 +27,57 @@ namespace GoldbergGUI.Core.Services
public Task<List<SteamApp>> GetListOfDlc(SteamApp steamApp, bool useSteamDb); public Task<List<SteamApp>> GetListOfDlc(SteamApp steamApp, bool useSteamDb);
} }
class SteamCache
{
public string Filename { get; }
public string SteamUri { get; }
public Type ApiVersion { get; }
public AppType SteamAppType { get; }
public HashSet<SteamApp> Cache { get; set; } = new HashSet<SteamApp>();
public SteamCache(string filename, string uri, Type apiVersion, AppType steamAppType)
{
Filename = filename;
SteamUri = uri;
ApiVersion = apiVersion;
SteamAppType = steamAppType;
}
}
// ReSharper disable once UnusedType.Global // ReSharper disable once UnusedType.Global
// ReSharper disable once ClassNeverInstantiated.Global // ReSharper disable once ClassNeverInstantiated.Global
public class SteamService : ISteamService public class SteamService : ISteamService
{ {
// ReSharper disable StringLiteralTypo // ReSharper disable StringLiteralTypo
private readonly List<(string filename, string uri, Type jsonType, AppType type)> _cacheSetup = private readonly Dictionary<AppType, SteamCache> _caches =
new List<(string, string, Type, AppType)> new Dictionary<AppType, SteamCache>
{ {
( {
"steamapps_games.json", AppType.Game,
"https://api.steampowered.com/IStoreService/GetAppList/v1/" + new SteamCache(
"?max_results=50000" + "steamapps_games.json",
"&include_games=1" + "https://api.steampowered.com/IStoreService/GetAppList/v1/" +
"&key=" + Secrets.SteamWebApiKey(), "?max_results=50000" +
typeof(SteamAppsV1), "&include_games=1" +
AppType.Game "&key=" + Secrets.SteamWebApiKey(),
), typeof(SteamAppsV1),
( AppType.Game
"steamapps_dlc.json", )
"https://api.steampowered.com/IStoreService/GetAppList/v1/" + },
"?max_results=50000" + {
"&include_games=0" + AppType.DLC,
"&include_dlc=1" + new SteamCache(
"&key=" + Secrets.SteamWebApiKey(), "steamapps_dlc.json",
typeof(SteamAppsV1), "https://api.steampowered.com/IStoreService/GetAppList/v1/" +
AppType.DLC "?max_results=50000" +
) "&include_games=0" +
"&include_dlc=1" +
"&key=" + Secrets.SteamWebApiKey(),
typeof(SteamAppsV1),
AppType.DLC
)
}
}; };
// ReSharper enable StringLiteralTypo
private static readonly Secrets Secrets = new Secrets(); private static readonly Secrets Secrets = new Secrets();
@ -63,60 +85,51 @@ namespace GoldbergGUI.Core.Services
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) " + "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) " +
"Chrome/87.0.4280.88 Safari/537.36"; "Chrome/87.0.4280.88 Safari/537.36";
//private HashSet<SteamApp> _cacheAll = new HashSet<SteamApp>();
//private HashSet<SteamApp> _cacheGame = new HashSet<SteamApp>();
//private HashSet<SteamApp> _cacheDlc = new HashSet<SteamApp>();
private Dictionary<string, HashSet<SteamApp>> _caches = new Dictionary<string, HashSet<SteamApp>>();
private IMvxLog _log; private IMvxLog _log;
public async Task Initialize(IMvxLog log) public async Task Initialize(IMvxLog log)
{ {
//var (path, uri, jsonType, appType) = _caches[0]; //var (path, uri, jsonType, appType) = _caches[0];
static SteamApps SteamApps(Type type, string s) static SteamApps DeserializeSteamApps(Type type, string cacheString)
{ {
SteamApps steamApps1 = null;
if (type == typeof(SteamAppsV1)) if (type == typeof(SteamAppsV1))
steamApps1 = JsonSerializer.Deserialize<SteamAppsV1>(s); return JsonSerializer.Deserialize<SteamAppsV1>(cacheString);
else if (type == typeof(SteamAppsV2)) else if (type == typeof(SteamAppsV2))
steamApps1 = JsonSerializer.Deserialize<SteamAppsV2>(s); return JsonSerializer.Deserialize<SteamAppsV2>(cacheString);
return steamApps1; return null;
} }
foreach (var (path, uri, jsonType, appType) in _cacheSetup) foreach (var (_, c) in _caches)
{ {
_log = log; _log = log;
_log.Info("Updating cache..."); _log.Info("Updating cache...");
var updateNeeded = DateTime.Now.Subtract(File.GetLastWriteTimeUtc(path)).TotalDays >= 1; var updateNeeded =
var cacheString = await GetCache(updateNeeded, uri, path).ConfigureAwait(false); DateTime.Now.Subtract(File.GetLastWriteTimeUtc(c.Filename)).TotalDays >= 1;
SteamApps steamApps = null; SteamApps steamApps;
try try
{ {
steamApps = SteamApps(jsonType, cacheString); var temp = await GetCache(updateNeeded, c.SteamUri, c.Filename)
.ConfigureAwait(false);
steamApps = DeserializeSteamApps(c.ApiVersion, temp);
} }
catch (JsonException) catch (JsonException)
{ {
_log.Error("Local cache broken, forcing update..."); _log.Error("Local cache broken, forcing update...");
cacheString = await GetCache(true, uri, path).ConfigureAwait(false); var temp = await GetCache(true, c.SteamUri, c.Filename).ConfigureAwait(false);
steamApps = SteamApps(jsonType, cacheString); steamApps = DeserializeSteamApps(c.ApiVersion, temp);
} }
try try
{ {
var cache = new HashSet<SteamApp>(steamApps.AppList.Apps); var cacheRaw = new HashSet<SteamApp>(steamApps.AppList.Apps);
var cache = new HashSet<SteamApp>();
if (appType != null) foreach (var steamApp in cacheRaw)
{ {
var cacheEdited = new HashSet<SteamApp>(); steamApp.type = c.SteamAppType;
foreach (var steamApp in cache) cache.Add(steamApp);
{
steamApp.type = appType;
cacheEdited.Add(steamApp);
}
cache = cacheEdited;
} }
_caches.Add(path, cache); c.Cache = cache;
_log.Info("Loaded cache into memory!"); _log.Info("Loaded cache into memory!");
} }
@ -155,7 +168,7 @@ namespace GoldbergGUI.Core.Services
public IEnumerable<SteamApp> GetListOfAppsByName(string name) public IEnumerable<SteamApp> GetListOfAppsByName(string name)
{ {
var listOfAppsByName = _caches["steamapps_games.json"].Search(x => x.Name) var listOfAppsByName = _caches[AppType.Game].Cache.Search(x => x.Name)
.SetCulture(StringComparison.OrdinalIgnoreCase) .SetCulture(StringComparison.OrdinalIgnoreCase)
.ContainingAll(name.Split(' ')); .ContainingAll(name.Split(' '));
return listOfAppsByName; return listOfAppsByName;
@ -165,7 +178,7 @@ namespace GoldbergGUI.Core.Services
{ {
_log.Info($"Trying to get app {name}"); _log.Info($"Trying to get app {name}");
var comparableName = Regex.Replace(name, Misc.SpecialCharsRegex, "").ToLower(); var comparableName = Regex.Replace(name, Misc.SpecialCharsRegex, "").ToLower();
var app = _caches["steamapps_games.json"].FirstOrDefault(x => x.CompareName(comparableName)); var app = _caches[AppType.Game].Cache.FirstOrDefault(x => x.CompareName(comparableName));
if (app != null) _log.Info($"Successfully got app {app}"); if (app != null) _log.Info($"Successfully got app {app}");
return app; return app;
} }
@ -173,7 +186,7 @@ namespace GoldbergGUI.Core.Services
public SteamApp GetAppById(int appid) public SteamApp GetAppById(int appid)
{ {
_log.Info($"Trying to get app with ID {appid}"); _log.Info($"Trying to get app with ID {appid}");
var app = _caches["steamapps_games.json"].FirstOrDefault(x => x.AppId.Equals(appid)); var app = _caches[AppType.Game].Cache.FirstOrDefault(x => x.AppId.Equals(appid));
if (app != null) _log.Info($"Successfully got app {app}"); if (app != null) _log.Info($"Successfully got app {app}");
return app; return app;
} }
@ -190,8 +203,8 @@ namespace GoldbergGUI.Core.Services
{ {
steamAppDetails.DLC.ForEach(x => steamAppDetails.DLC.ForEach(x =>
{ {
var result = _caches["steamapps_dlc.json"].FirstOrDefault(y => y.AppId.Equals(x)) ?? var result = _caches[AppType.DLC].Cache.FirstOrDefault(y => y.AppId.Equals(x))
new SteamApp {AppId = x, Name = $"Unknown DLC {x}"}; ?? new SteamApp {AppId = x, Name = $"Unknown DLC {x}"};
dlcList.Add(result); dlcList.Add(result);
}); });
@ -202,17 +215,12 @@ namespace GoldbergGUI.Core.Services
// Get Cloudflare cookie // Get Cloudflare cookie
// Scrape and parse HTML page // Scrape and parse HTML page
// Add missing to DLC list // Add missing to DLC list
// ReSharper disable once InvertIf
if (useSteamDb) if (useSteamDb)
{ {
var steamDbUri = new Uri($"https://steamdb.info/app/{steamApp.AppId}/dlc/"); var steamDbUri = new Uri($"https://steamdb.info/app/{steamApp.AppId}/dlc/");
/* var handler = new ClearanceHandler();
var client = new HttpClient(handler);
var content = client.GetStringAsync(steamDbUri).Result;
_log.Debug(content); */
var client = new HttpClient(); var client = new HttpClient();
client.DefaultRequestHeaders.UserAgent.ParseAdd(UserAgent); client.DefaultRequestHeaders.UserAgent.ParseAdd(UserAgent);
@ -228,7 +236,6 @@ namespace GoldbergGUI.Core.Services
var parser = new HtmlParser(); var parser = new HtmlParser();
var doc = parser.ParseDocument(responseBody); var doc = parser.ParseDocument(responseBody);
// Console.WriteLine(doc.DocumentElement.OuterHtml);
var query1 = doc.QuerySelector("#dlc"); var query1 = doc.QuerySelector("#dlc");
if (query1 != null) if (query1 != null)
@ -237,10 +244,10 @@ namespace GoldbergGUI.Core.Services
foreach (var element in query2) foreach (var element in query2)
{ {
var dlcId = element.GetAttribute("data-appid"); var dlcId = element.GetAttribute("data-appid");
var dlcName = $"Unknown DLC {dlcId}";
var query3 = element.QuerySelectorAll("td"); var query3 = element.QuerySelectorAll("td");
if (query3 != null) dlcName = query3[1].Text().Replace("\n", "").Trim(); var dlcName = query3 != null
? query3[1].Text().Replace("\n", "").Trim()
: $"Unknown DLC {dlcId}";
var dlcApp = new SteamApp {AppId = Convert.ToInt32(dlcId), Name = dlcName}; var dlcApp = new SteamApp {AppId = Convert.ToInt32(dlcId), Name = dlcName};
var i = dlcList.FindIndex(x => x.AppId.Equals(dlcApp.AppId)); var i = dlcList.FindIndex(x => x.AppId.Equals(dlcApp.AppId));
if (i > -1) if (i > -1)