Fix startup with no virus scanner

This commit is contained in:
Kieran 2022-03-14 22:30:05 +00:00
parent 1009ff1e92
commit 79b4fe475d
Signed by: Kieran
GPG Key ID: DE71CEB3925BE941

View File

@ -14,11 +14,12 @@ public static class VirusScannerStartup
var avSettings = settings.VirusScanner;
if (avSettings != default)
{
services.AddHostedService<Background.VirusScannerService>();
var loadService = false;
// load ClamAV scanner
if (avSettings.ClamAV != default)
{
loadService = true;
services.AddTransient<IClamClient>((_) =>
new ClamClient(avSettings.ClamAV.Endpoint.Host, avSettings.ClamAV.Endpoint.Port)
{
@ -30,10 +31,16 @@ public static class VirusScannerStartup
// load VirusTotal
if (avSettings.VirusTotal != default)
{
loadService = true;
services.AddTransient((svc) =>
new VirusTotalClient(svc.GetRequiredService<IHttpClientFactory>(), avSettings.VirusTotal));
services.AddTransient<IVirusScanner, VirusTotalScanner>();
}
if (loadService)
{
services.AddHostedService<Background.VirusScannerService>();
}
}
}
}