Files
Cleanuparr/code/Executable/Controllers/StatusController.cs
Flaminel c675924be7 fix
2025-05-19 12:08:58 +03:00

251 lines
9.0 KiB
C#

using Common.Configuration.Arr;
using Common.Configuration.DownloadClient;
using Infrastructure.Configuration;
using Infrastructure.Verticals.Arr;
using Infrastructure.Verticals.DownloadClient;
using Microsoft.AspNetCore.Mvc;
using System.Diagnostics;
namespace Executable.Controllers;
[ApiController]
[Route("api/[controller]")]
public class StatusController : ControllerBase
{
private readonly ILogger<StatusController> _logger;
private readonly IConfigManager _configManager;
private readonly DownloadServiceFactory _downloadServiceFactory;
private readonly ArrClientFactory _arrClientFactory;
public StatusController(
ILogger<StatusController> logger,
IConfigManager configManager,
DownloadServiceFactory downloadServiceFactory,
ArrClientFactory arrClientFactory)
{
_logger = logger;
_configManager = configManager;
_downloadServiceFactory = downloadServiceFactory;
_arrClientFactory = arrClientFactory;
}
[HttpGet]
public async Task<IActionResult> GetSystemStatus()
{
try
{
var process = Process.GetCurrentProcess();
// Get configuration
var downloadClientConfig = await _configManager.GetConfigurationAsync<DownloadClientConfig>();
var sonarrConfig = await _configManager.GetConfigurationAsync<SonarrConfig>();
var radarrConfig = await _configManager.GetConfigurationAsync<RadarrConfig>();
var lidarrConfig = await _configManager.GetConfigurationAsync<LidarrConfig>();
var status = new
{
Application = new
{
Version = GetType().Assembly.GetName().Version?.ToString() ?? "Unknown",
process.StartTime,
UpTime = DateTime.Now - process.StartTime,
MemoryUsageMB = Math.Round(process.WorkingSet64 / 1024.0 / 1024.0, 2),
ProcessorTime = process.TotalProcessorTime
},
DownloadClient = new
{
// TODO
},
MediaManagers = new
{
Sonarr = new
{
IsEnabled = sonarrConfig.Enabled,
InstanceCount = sonarrConfig.Instances.Count
},
Radarr = new
{
IsEnabled = radarrConfig.Enabled,
InstanceCount = radarrConfig.Instances.Count
},
Lidarr = new
{
IsEnabled = lidarrConfig.Enabled,
InstanceCount = lidarrConfig.Instances.Count
}
}
};
return Ok(status);
}
catch (Exception ex)
{
_logger.LogError(ex, "Error retrieving system status");
return StatusCode(500, "An error occurred while retrieving system status");
}
}
[HttpGet("download-client")]
public async Task<IActionResult> GetDownloadClientStatus()
{
try
{
var downloadClientConfig = await _configManager.GetConfigurationAsync<DownloadClientConfig>();
var result = new Dictionary<string, object>();
// Check for configured clients
if (downloadClientConfig.Clients.Count > 0)
{
var clientsStatus = new List<object>();
foreach (var client in downloadClientConfig.Clients)
{
clientsStatus.Add(new
{
client.Id,
client.Name,
client.Type,
client.Host,
client.Enabled,
IsConnected = client.Enabled, // We can't check connection status without implementing test methods
});
}
result["Clients"] = clientsStatus;
}
return Ok(result);
}
catch (Exception ex)
{
_logger.LogError(ex, "Error retrieving download client status");
return StatusCode(500, "An error occurred while retrieving download client status");
}
}
[HttpGet("media-managers")]
public async Task<IActionResult> GetMediaManagersStatus()
{
try
{
var status = new Dictionary<string, object>();
// Get configurations
var sonarrConfig = await _configManager.GetConfigurationAsync<SonarrConfig>();
var radarrConfig = await _configManager.GetConfigurationAsync<RadarrConfig>();
var lidarrConfig = await _configManager.GetConfigurationAsync<LidarrConfig>();
// Check Sonarr instances
if (sonarrConfig is { Enabled: true, Instances.Count: > 0 })
{
var sonarrStatus = new List<object>();
foreach (var instance in sonarrConfig.Instances)
{
try
{
var sonarrClient = _arrClientFactory.GetClient(Domain.Enums.InstanceType.Sonarr);
await sonarrClient.TestConnectionAsync(instance);
sonarrStatus.Add(new
{
instance.Name,
instance.Url,
IsConnected = true,
Message = "Successfully connected"
});
}
catch (Exception ex)
{
sonarrStatus.Add(new
{
instance.Name,
instance.Url,
IsConnected = false,
Message = $"Connection failed: {ex.Message}"
});
}
}
status["Sonarr"] = sonarrStatus;
}
// Check Radarr instances
if (radarrConfig is { Enabled: true, Instances.Count: > 0 })
{
var radarrStatus = new List<object>();
foreach (var instance in radarrConfig.Instances)
{
try
{
var radarrClient = _arrClientFactory.GetClient(Domain.Enums.InstanceType.Radarr);
await radarrClient.TestConnectionAsync(instance);
radarrStatus.Add(new
{
instance.Name,
instance.Url,
IsConnected = true,
Message = "Successfully connected"
});
}
catch (Exception ex)
{
radarrStatus.Add(new
{
instance.Name,
instance.Url,
IsConnected = false,
Message = $"Connection failed: {ex.Message}"
});
}
}
status["Radarr"] = radarrStatus;
}
// Check Lidarr instances
if (lidarrConfig.Enabled == true && lidarrConfig.Instances?.Count > 0)
{
var lidarrStatus = new List<object>();
foreach (var instance in lidarrConfig.Instances)
{
try
{
var lidarrClient = _arrClientFactory.GetClient(Domain.Enums.InstanceType.Lidarr);
await lidarrClient.TestConnectionAsync(instance);
lidarrStatus.Add(new
{
instance.Name,
instance.Url,
IsConnected = true,
Message = "Successfully connected"
});
}
catch (Exception ex)
{
lidarrStatus.Add(new
{
instance.Name,
instance.Url,
IsConnected = false,
Message = $"Connection failed: {ex.Message}"
});
}
}
status["Lidarr"] = lidarrStatus;
}
return Ok(status);
}
catch (Exception ex)
{
_logger.LogError(ex, "Error retrieving media managers status");
return StatusCode(500, "An error occurred while retrieving media managers status");
}
}
}