Migration to a ServiceWorker for most tasks.
Implementation of basic PatchComplianceTask. First real iteration, basic/raw asf.
This commit is contained in:
317
OversightService/Services/ApiClient.cs
Normal file
317
OversightService/Services/ApiClient.cs
Normal file
@@ -0,0 +1,317 @@
|
||||
using System;
|
||||
using System.IO;
|
||||
using System.Net.Http;
|
||||
using System.Security.Cryptography.X509Certificates;
|
||||
using System.Net.Security;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
using Newtonsoft.Json;
|
||||
|
||||
namespace OversightService.Services
|
||||
{
|
||||
public class ApiClient
|
||||
{
|
||||
/// <summary>
|
||||
/// Constructs the full URL for API requests.
|
||||
/// </summary>
|
||||
private static readonly HttpClient httpClient = CreateHttpClient();
|
||||
private readonly AppConfig _config;
|
||||
private static string jwtToken = null; // 🔹 Store the JWT token globally
|
||||
|
||||
public ApiClient(AppConfig config)
|
||||
{
|
||||
_config = config ?? throw new ArgumentNullException(nameof(config));
|
||||
}
|
||||
|
||||
private void Log(string message)
|
||||
{
|
||||
try
|
||||
{
|
||||
var logDir = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "PSG-Oversight");
|
||||
Directory.CreateDirectory(logDir);
|
||||
var logPath = Path.Combine(logDir, "api_client.log");
|
||||
File.AppendAllText(logPath, $"[{DateTime.Now}] {message}\n");
|
||||
}
|
||||
catch { }
|
||||
}
|
||||
|
||||
|
||||
private string BuildUrl(string route)
|
||||
{
|
||||
return $"{_config.ServerUrl.TrimEnd('/')}/{route.TrimStart('/')}";
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Creates an HttpClient that bypasses SSL validation (for testing purposes only).
|
||||
/// </summary>
|
||||
public static HttpClient CreateHttpClient()
|
||||
{
|
||||
HttpClientHandler handler = new HttpClientHandler
|
||||
{
|
||||
ServerCertificateCustomValidationCallback = (httpRequestMessage, cert, certChain, sslPolicyErrors) =>
|
||||
{
|
||||
return true; // 🔥 Ignores all SSL certificate warnings (TEMPORARY!)
|
||||
}
|
||||
};
|
||||
|
||||
return new HttpClient(handler);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sets the JWT token to be used for subsequent requests.
|
||||
/// </summary>
|
||||
public static void SetJwtToken(string token)
|
||||
{
|
||||
jwtToken = token;
|
||||
httpClient.DefaultRequestHeaders.Authorization =
|
||||
new System.Net.Http.Headers.AuthenticationHeaderValue("Bearer", jwtToken);
|
||||
}
|
||||
/// <summary>
|
||||
/// Gets the JWT token to be used for subsequent requests.
|
||||
/// </summary>
|
||||
public static string GetJwtToken()
|
||||
{
|
||||
return jwtToken;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Authenticates with the API using credentials from config.json.
|
||||
/// </summary>
|
||||
public async Task<LoginResponse> AuthenticateAsync(string username, string password)
|
||||
{
|
||||
try
|
||||
{
|
||||
string encryptedUsername = EncryptionHelper.EncryptData(username);
|
||||
string encryptedPassword = EncryptionHelper.EncryptData(password);
|
||||
|
||||
var payload = new
|
||||
{
|
||||
username = encryptedUsername,
|
||||
password = encryptedPassword
|
||||
};
|
||||
|
||||
string jsonContent = JsonConvert.SerializeObject(payload);
|
||||
var content = new StringContent(jsonContent, Encoding.UTF8, "application/json");
|
||||
|
||||
HttpResponseMessage response = await httpClient.PostAsync(BuildUrl("/api/auth/login"), content);
|
||||
string rawResponse = await response.Content.ReadAsStringAsync();
|
||||
|
||||
if (!response.IsSuccessStatusCode)
|
||||
{
|
||||
Log($"❌ Authentication failed: {response.StatusCode} - {rawResponse}");
|
||||
return null;
|
||||
}
|
||||
|
||||
var loginResponse = JsonConvert.DeserializeObject<LoginResponse>(rawResponse);
|
||||
|
||||
if (!string.IsNullOrEmpty(loginResponse?.Token))
|
||||
{
|
||||
SetJwtToken(loginResponse.Token);
|
||||
}
|
||||
|
||||
return loginResponse;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Log($"❌ Exception during login: {ex.Message}");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Checks connectivity to the server using the stored JWT token.
|
||||
/// </summary>
|
||||
public async Task<bool> CheckConnectivity()
|
||||
{
|
||||
string testUrl = "https://localhost:8443/api/system/ping-status"; // Your HTTPS URL with port 8443
|
||||
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(jwtToken))
|
||||
{
|
||||
throw new Exception("No JWT token stored. Please authenticate first.");
|
||||
}
|
||||
|
||||
var response = await SendWithAutoReauthAsync(() =>
|
||||
httpClient.GetAsync(testUrl)
|
||||
);
|
||||
|
||||
|
||||
if (response.IsSuccessStatusCode)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
Log($"❌ Server Unreachable: {response.StatusCode}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Log($"❌ Connection Error: {ex.Message}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sends the collected data to the server.
|
||||
/// </summary>
|
||||
public async Task<string> StoreSystemInfoAsync(string encryptedData)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(jwtToken))
|
||||
return "❌ Error: Not authenticated. Please log in first.";
|
||||
|
||||
var payload = new { data = encryptedData };
|
||||
string jsonContent = JsonConvert.SerializeObject(payload);
|
||||
var content = new StringContent(jsonContent, Encoding.UTF8, "application/json");
|
||||
|
||||
var response = await SendWithAutoReauthAsync(() =>
|
||||
httpClient.PostAsync(BuildUrl("/api/system-info"), content)
|
||||
);
|
||||
|
||||
if (response.IsSuccessStatusCode)
|
||||
return await response.Content.ReadAsStringAsync();
|
||||
else
|
||||
return $"❌ Error: {response.StatusCode} - {await response.Content.ReadAsStringAsync()}";
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
return $"❌ Exception: {ex.Message}";
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Refreshes the current JWT token without requiring username/password.
|
||||
/// </summary>
|
||||
public async Task<bool> RefreshTokenAsync()
|
||||
{
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(jwtToken))
|
||||
{
|
||||
Log("⚠️ No token to refresh.");
|
||||
return false;
|
||||
}
|
||||
|
||||
HttpResponseMessage response = await httpClient.PostAsync(BuildUrl("/api/auth/refresh"), null);
|
||||
string rawResponse = await response.Content.ReadAsStringAsync();
|
||||
|
||||
if (!response.IsSuccessStatusCode)
|
||||
{
|
||||
Log($"❌ Token refresh failed: {response.StatusCode} - {rawResponse}");
|
||||
return false;
|
||||
}
|
||||
|
||||
// Parse the response to get the new token if provided in body
|
||||
// Note: Your server sets it via cookie, but may also return it in response
|
||||
try
|
||||
{
|
||||
var refreshResponse = JsonConvert.DeserializeObject<Dictionary<string, string>>(rawResponse);
|
||||
if (refreshResponse?.ContainsKey("token") == true)
|
||||
{
|
||||
SetJwtToken(refreshResponse["token"]);
|
||||
}
|
||||
}
|
||||
catch
|
||||
{
|
||||
// Token might only be in cookie, which is fine
|
||||
}
|
||||
|
||||
Log("✅ Token refreshed successfully.");
|
||||
return true;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Log($"❌ Exception during token refresh: {ex.Message}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sends the collected data to the server with automatic token refresh and re-auth.
|
||||
/// </summary>
|
||||
private async Task<HttpResponseMessage> SendWithAutoReauthAsync(Func<Task<HttpResponseMessage>> requestFunc)
|
||||
{
|
||||
var response = await requestFunc();
|
||||
|
||||
if (response.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
||||
{
|
||||
Log("⚠️ Token expired or invalid. Attempting token refresh...");
|
||||
|
||||
// Try refreshing the token first (faster than full re-auth)
|
||||
bool refreshed = await RefreshTokenAsync();
|
||||
|
||||
if (refreshed)
|
||||
{
|
||||
// Retry original request after refresh
|
||||
response = await requestFunc();
|
||||
|
||||
if (response.IsSuccessStatusCode)
|
||||
{
|
||||
return response;
|
||||
}
|
||||
}
|
||||
|
||||
// If refresh failed or didn't work, fall back to full re-authentication
|
||||
Log("⚠️ Token refresh failed. Attempting full re-auth...");
|
||||
var loginResponse = await AuthenticateAsync(_config.Auth.Username, _config.Auth.Password);
|
||||
if (loginResponse == null || string.IsNullOrEmpty(loginResponse.Token))
|
||||
{
|
||||
Log("❌ Re-authentication failed.");
|
||||
return response; // Return original 401 response
|
||||
}
|
||||
|
||||
// Retry original request after reauth
|
||||
response = await requestFunc();
|
||||
}
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sends Windows update history (patch compliance data) to the backend API
|
||||
/// </summary>
|
||||
public async Task<bool> PostPatchComplianceAsync(object patchData)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(jwtToken))
|
||||
{
|
||||
Log("❌ Not authenticated. Cannot send patch compliance data.");
|
||||
return false;
|
||||
}
|
||||
|
||||
var jsonContent = JsonConvert.SerializeObject(patchData);
|
||||
var content = new StringContent(jsonContent, Encoding.UTF8, "application/json");
|
||||
|
||||
var response = await SendWithAutoReauthAsync(() =>
|
||||
httpClient.PostAsync(BuildUrl("/api/patch-compliance"), content)
|
||||
);
|
||||
|
||||
if (response.IsSuccessStatusCode)
|
||||
{
|
||||
Log("✅ Patch compliance data sent successfully.");
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
Log($"❌ Failed to send patch compliance data: {response.StatusCode}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Log($"❌ Exception sending patch compliance data: {ex.Message}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
9
OversightService/Services/LoginResponse.cs
Normal file
9
OversightService/Services/LoginResponse.cs
Normal file
@@ -0,0 +1,9 @@
|
||||
namespace OversightService.Services
|
||||
{
|
||||
public class LoginResponse
|
||||
{
|
||||
public string Token { get; set; }
|
||||
public string Username { get; set; }
|
||||
public int UserId { get; set; }
|
||||
}
|
||||
}
|
||||
97
OversightService/Services/OsqueryService.cs
Normal file
97
OversightService/Services/OsqueryService.cs
Normal file
@@ -0,0 +1,97 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Diagnostics;
|
||||
using System.IO;
|
||||
|
||||
using Newtonsoft.Json.Linq;
|
||||
|
||||
namespace OversightService.Services
|
||||
{
|
||||
public static class OsqueryService
|
||||
{
|
||||
private static string GetLogPath(string filename)
|
||||
{
|
||||
string logDir = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "PSG-Oversight");
|
||||
Directory.CreateDirectory(logDir);
|
||||
return Path.Combine(logDir, filename);
|
||||
}
|
||||
|
||||
private static string GetOsqueryPath()
|
||||
{
|
||||
var baseDir = AppContext.BaseDirectory;
|
||||
var path = Path.Combine(baseDir, "Assets", "osqueryi.exe");
|
||||
|
||||
if (!File.Exists(path))
|
||||
{
|
||||
try
|
||||
{
|
||||
File.AppendAllText(GetLogPath("osquery_error.log"), $"[{DateTime.Now}] ❌ osqueryi.exe not found at {path}\n");
|
||||
}
|
||||
catch { /* Silently fail if we can't write logs */ }
|
||||
throw new FileNotFoundException("osqueryi.exe not found", path);
|
||||
}
|
||||
|
||||
return path;
|
||||
}
|
||||
|
||||
private static string RunQuery(string sql)
|
||||
{
|
||||
var processInfo = new ProcessStartInfo
|
||||
{
|
||||
FileName = GetOsqueryPath(),
|
||||
Arguments = $"--json \"{sql}\"",
|
||||
RedirectStandardOutput = true,
|
||||
UseShellExecute = false,
|
||||
CreateNoWindow = true
|
||||
};
|
||||
|
||||
using var process = Process.Start(processInfo); // ✅ FIXED — actually start the process
|
||||
if (process == null)
|
||||
throw new Exception("Failed to start osquery process");
|
||||
|
||||
string output = process.StandardOutput.ReadToEnd();
|
||||
process.WaitForExit();
|
||||
|
||||
// Optional debug logging
|
||||
try
|
||||
{
|
||||
File.AppendAllText(GetLogPath("osquery_debug.log"),
|
||||
$"[{DateTime.Now}] Ran query: {sql}\nOutput length: {output.Length}\n");
|
||||
}
|
||||
catch { /* Silently fail if we can't write logs */ }
|
||||
|
||||
return output;
|
||||
}
|
||||
|
||||
public static List<Dictionary<string, string>> Query(string sql)
|
||||
{
|
||||
string json = RunQuery(sql);
|
||||
|
||||
try
|
||||
{
|
||||
var jArray = JArray.Parse(json);
|
||||
var results = new List<Dictionary<string, string>>();
|
||||
|
||||
foreach (var obj in jArray)
|
||||
{
|
||||
var dict = new Dictionary<string, string>();
|
||||
foreach (var prop in (JObject)obj)
|
||||
dict[prop.Key] = prop.Value?.ToString() ?? "";
|
||||
results.Add(dict);
|
||||
}
|
||||
|
||||
return results;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
try
|
||||
{
|
||||
File.AppendAllText(GetLogPath("osquery_error.log"),
|
||||
$"[{DateTime.Now}] ⚠️ JSON parse failed for query '{sql}': {ex.Message}\n");
|
||||
}
|
||||
catch { /* Silently fail if we can't write logs */ }
|
||||
return new List<Dictionary<string, string>>();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
152
OversightService/Services/OsqueryTaskScheduler.cs
Normal file
152
OversightService/Services/OsqueryTaskScheduler.cs
Normal file
@@ -0,0 +1,152 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using System.IO;
|
||||
using Newtonsoft.Json;
|
||||
|
||||
namespace OversightService.Services
|
||||
{
|
||||
/// <summary>
|
||||
/// Manages scheduled osquery-based tasks
|
||||
/// Designed to be portable to Windows Service (OversightService) later
|
||||
/// </summary>
|
||||
public class OsqueryTaskScheduler
|
||||
{
|
||||
private readonly List<IScheduledTask> _tasks = new();
|
||||
private readonly Timer _timer;
|
||||
private readonly string _stateFilePath;
|
||||
private bool _isRunning;
|
||||
|
||||
public OsqueryTaskScheduler()
|
||||
{
|
||||
// Store task state in AppData to persist last run times
|
||||
var appDataDir = Path.Combine(
|
||||
Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
|
||||
"PSG-Oversight"
|
||||
);
|
||||
Directory.CreateDirectory(appDataDir);
|
||||
_stateFilePath = Path.Combine(appDataDir, "task_scheduler_state.json");
|
||||
|
||||
// Check every minute for tasks that need to run
|
||||
_timer = new Timer(CheckAndRunTasks, null, TimeSpan.Zero, TimeSpan.FromMinutes(1));
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Register a task with the scheduler
|
||||
/// </summary>
|
||||
public void RegisterTask(IScheduledTask task)
|
||||
{
|
||||
_tasks.Add(task);
|
||||
LoadTaskState(task);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Start the scheduler
|
||||
/// </summary>
|
||||
public void Start()
|
||||
{
|
||||
_isRunning = true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Stop the scheduler
|
||||
/// </summary>
|
||||
public void Stop()
|
||||
{
|
||||
_isRunning = false;
|
||||
_timer?.Change(Timeout.Infinite, Timeout.Infinite);
|
||||
}
|
||||
|
||||
private async void CheckAndRunTasks(object? state)
|
||||
{
|
||||
if (!_isRunning) return;
|
||||
|
||||
foreach (var task in _tasks.Where(t => t.ShouldRun()))
|
||||
{
|
||||
try
|
||||
{
|
||||
await task.ExecuteAsync();
|
||||
task.LastRun = DateTime.Now;
|
||||
SaveTaskState(task);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
LogError($"Task '{task.TaskName}' failed: {ex.Message}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void LoadTaskState(IScheduledTask task)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (!File.Exists(_stateFilePath)) return;
|
||||
|
||||
var json = File.ReadAllText(_stateFilePath);
|
||||
var states = JsonConvert.DeserializeObject<Dictionary<string, DateTime>>(json);
|
||||
|
||||
if (states != null && states.TryGetValue(task.TaskName, out var lastRun))
|
||||
{
|
||||
task.LastRun = lastRun;
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
LogError($"Failed to load task state: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
private void SaveTaskState(IScheduledTask task)
|
||||
{
|
||||
try
|
||||
{
|
||||
Dictionary<string, DateTime> states;
|
||||
|
||||
if (File.Exists(_stateFilePath))
|
||||
{
|
||||
var json = File.ReadAllText(_stateFilePath);
|
||||
states = JsonConvert.DeserializeObject<Dictionary<string, DateTime>>(json)
|
||||
?? new Dictionary<string, DateTime>();
|
||||
}
|
||||
else
|
||||
{
|
||||
states = new Dictionary<string, DateTime>();
|
||||
}
|
||||
|
||||
states[task.TaskName] = task.LastRun ?? DateTime.Now;
|
||||
|
||||
var updatedJson = JsonConvert.SerializeObject(states, Formatting.Indented);
|
||||
File.WriteAllText(_stateFilePath, updatedJson);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
LogError($"Failed to save task state: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
private void LogError(string message)
|
||||
{
|
||||
try
|
||||
{
|
||||
var logDir = Path.Combine(
|
||||
Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
|
||||
"PSG-Oversight"
|
||||
);
|
||||
var logPath = Path.Combine(logDir, "scheduler_error.log");
|
||||
File.AppendAllText(logPath, $"[{DateTime.Now}] {message}\n");
|
||||
}
|
||||
catch
|
||||
{
|
||||
// Silently fail if we can't write logs
|
||||
}
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
Stop();
|
||||
_timer?.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
||||
132
OversightService/Services/PatchComplianceTask.cs
Normal file
132
OversightService/Services/PatchComplianceTask.cs
Normal file
@@ -0,0 +1,132 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace OversightService.Services
|
||||
{
|
||||
/// <summary>
|
||||
/// Scheduled task that collects Windows update history using osquery
|
||||
/// and sends it to the backend API
|
||||
/// </summary>
|
||||
public class PatchComplianceTask : ScheduledTask
|
||||
{
|
||||
private readonly AppConfig _config;
|
||||
private readonly ApiClient _apiClient;
|
||||
|
||||
public override string TaskName => "PatchCompliance";
|
||||
|
||||
public override TimeSpan Interval =>
|
||||
TimeSpan.FromHours(_config.PatchCompliance.CheckIntervalHours);
|
||||
|
||||
public PatchComplianceTask(AppConfig config, ApiClient apiClient)
|
||||
{
|
||||
_config = config;
|
||||
_apiClient = apiClient;
|
||||
|
||||
// Initialize LastRun from config if available
|
||||
LastRun = _config.PatchCompliance.LastCheckTime;
|
||||
}
|
||||
|
||||
public override bool ShouldRun()
|
||||
{
|
||||
// Don't run if disabled in config
|
||||
if (!_config.PatchCompliance.Enabled)
|
||||
return false;
|
||||
|
||||
return base.ShouldRun();
|
||||
}
|
||||
|
||||
public override async Task ExecuteAsync()
|
||||
{
|
||||
try
|
||||
{
|
||||
// Query windows_update_history from osquery
|
||||
var updates = OsqueryService.Query(@"
|
||||
SELECT
|
||||
date,
|
||||
title,
|
||||
update_id
|
||||
FROM windows_update_history
|
||||
ORDER BY date DESC;
|
||||
");
|
||||
|
||||
if (!updates.Any())
|
||||
{
|
||||
LogInfo("No Windows update history found.");
|
||||
return;
|
||||
}
|
||||
|
||||
// Transform to API-friendly format
|
||||
var patchData = updates.Select(u => new
|
||||
{
|
||||
date = u.GetValueOrDefault("date"),
|
||||
title = u.GetValueOrDefault("title"),
|
||||
updateId = u.GetValueOrDefault("update_id")
|
||||
}).ToList();
|
||||
|
||||
LogInfo($"Collected {patchData.Count} Windows updates.");
|
||||
|
||||
// Send to API
|
||||
await _apiClient.PostPatchComplianceAsync(patchData);
|
||||
|
||||
// Update config with last check time
|
||||
_config.PatchCompliance.LastCheckTime = DateTime.Now;
|
||||
SaveConfig();
|
||||
|
||||
LogInfo("Patch compliance data sent successfully.");
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
LogError($"PatchComplianceTask failed: {ex.Message}");
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
private void SaveConfig()
|
||||
{
|
||||
try
|
||||
{
|
||||
var configPath = System.IO.Path.Combine(AppContext.BaseDirectory, "config.json");
|
||||
var json = Newtonsoft.Json.JsonConvert.SerializeObject(_config, Newtonsoft.Json.Formatting.Indented);
|
||||
System.IO.File.WriteAllText(configPath, json);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
LogError($"Failed to save config: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
private void LogInfo(string message)
|
||||
{
|
||||
if (!_config.EnableLogging) return;
|
||||
|
||||
try
|
||||
{
|
||||
var logDir = System.IO.Path.Combine(
|
||||
Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
|
||||
"PSG-Oversight"
|
||||
);
|
||||
System.IO.Directory.CreateDirectory(logDir);
|
||||
var logPath = System.IO.Path.Combine(logDir, "patch_compliance.log");
|
||||
System.IO.File.AppendAllText(logPath, $"[{DateTime.Now}] INFO: {message}\n");
|
||||
}
|
||||
catch { }
|
||||
}
|
||||
|
||||
private void LogError(string message)
|
||||
{
|
||||
try
|
||||
{
|
||||
var logDir = System.IO.Path.Combine(
|
||||
Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
|
||||
"PSG-Oversight"
|
||||
);
|
||||
System.IO.Directory.CreateDirectory(logDir);
|
||||
var logPath = System.IO.Path.Combine(logDir, "patch_compliance.log");
|
||||
System.IO.File.AppendAllText(logPath, $"[{DateTime.Now}] ERROR: {message}\n");
|
||||
}
|
||||
catch { }
|
||||
}
|
||||
}
|
||||
}
|
||||
35
OversightService/Services/ScheduledTask.cs
Normal file
35
OversightService/Services/ScheduledTask.cs
Normal file
@@ -0,0 +1,35 @@
|
||||
using System;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace OversightService.Services
|
||||
{
|
||||
/// <summary>
|
||||
/// Base interface for scheduled tasks
|
||||
/// </summary>
|
||||
public interface IScheduledTask
|
||||
{
|
||||
string TaskName { get; }
|
||||
TimeSpan Interval { get; }
|
||||
DateTime? LastRun { get; set; }
|
||||
Task ExecuteAsync();
|
||||
bool ShouldRun();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Base class for scheduled tasks with common logic
|
||||
/// </summary>
|
||||
public abstract class ScheduledTask : IScheduledTask
|
||||
{
|
||||
public abstract string TaskName { get; }
|
||||
public abstract TimeSpan Interval { get; }
|
||||
public DateTime? LastRun { get; set; }
|
||||
|
||||
public abstract Task ExecuteAsync();
|
||||
|
||||
public virtual bool ShouldRun()
|
||||
{
|
||||
if (LastRun == null) return true;
|
||||
return DateTime.Now - LastRun >= Interval;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user