Implemented factory pattern for server abstraction creation. Implemented raw fs and docker provisioner. Implemented docker event service with observer pattern
This commit is contained in:
@@ -0,0 +1,246 @@
|
||||
using System.Reactive.Subjects;
|
||||
using Docker.DotNet;
|
||||
using Docker.DotNet.Models;
|
||||
using MoonlightServers.Daemon.Mappers;
|
||||
using MoonlightServers.Daemon.ServerSys.Abstractions;
|
||||
using MoonlightServers.Daemon.Services;
|
||||
|
||||
namespace MoonlightServers.Daemon.ServerSys.Implementations;
|
||||
|
||||
public class DockerProvisioner : IProvisioner
|
||||
{
|
||||
public IAsyncObservable<object> OnExited { get; set; }
|
||||
|
||||
private readonly DockerClient DockerClient;
|
||||
private readonly ILogger<DockerProvisioner> Logger;
|
||||
private readonly DockerEventService EventService;
|
||||
private readonly ServerMeta Meta;
|
||||
private readonly IConsole Console;
|
||||
private readonly DockerImageService ImageService;
|
||||
private readonly ServerConfigurationMapper Mapper;
|
||||
private readonly IFileSystem FileSystem;
|
||||
|
||||
private AsyncSubject<object> OnExitedSubject = new();
|
||||
|
||||
private string? ContainerId;
|
||||
private string ContainerName;
|
||||
private IAsyncDisposable? ContainerEventSubscription;
|
||||
|
||||
public DockerProvisioner(
|
||||
DockerClient dockerClient,
|
||||
ILogger<DockerProvisioner> logger,
|
||||
DockerEventService eventService,
|
||||
ServerMeta meta,
|
||||
IConsole console,
|
||||
DockerImageService imageService,
|
||||
ServerConfigurationMapper mapper,
|
||||
IFileSystem fileSystem
|
||||
)
|
||||
{
|
||||
DockerClient = dockerClient;
|
||||
Logger = logger;
|
||||
EventService = eventService;
|
||||
Meta = meta;
|
||||
Console = console;
|
||||
ImageService = imageService;
|
||||
Mapper = mapper;
|
||||
FileSystem = fileSystem;
|
||||
}
|
||||
|
||||
public async Task Initialize()
|
||||
{
|
||||
ContainerName = $"moonlight-runtime-{Meta.Configuration.Id}";
|
||||
|
||||
ContainerEventSubscription = await EventService
|
||||
.OnContainerEvent
|
||||
.SubscribeAsync(HandleContainerEvent);
|
||||
|
||||
// Check for any already existing runtime container
|
||||
// TODO: Implement a way for restoring the state
|
||||
// Needs to be able to be consumed by the restorer
|
||||
}
|
||||
|
||||
private ValueTask HandleContainerEvent(Message message)
|
||||
{
|
||||
// Only handle events for our own container
|
||||
if (message.ID != ContainerId)
|
||||
return ValueTask.CompletedTask;
|
||||
|
||||
// Only handle die events
|
||||
if (message.Action != "die")
|
||||
return ValueTask.CompletedTask;
|
||||
|
||||
OnExitedSubject.OnNext(message);
|
||||
|
||||
return ValueTask.CompletedTask;
|
||||
}
|
||||
|
||||
public Task Sync()
|
||||
{
|
||||
return Task.CompletedTask; // TODO: Implement
|
||||
}
|
||||
|
||||
public async Task Provision()
|
||||
{
|
||||
// Plan of action:
|
||||
// 1. Ensure no other container with that name exist
|
||||
// 2. Ensure the docker image has been downloaded
|
||||
// 3. Create the container from the configuration in the meta
|
||||
|
||||
// 1. Ensure no other container with that name exist
|
||||
|
||||
try
|
||||
{
|
||||
Logger.LogDebug("Searching for orphan container");
|
||||
|
||||
var possibleContainer = await DockerClient.Containers.InspectContainerAsync(ContainerName);
|
||||
|
||||
Logger.LogDebug("Orphan container found. Removing it");
|
||||
await Console.WriteToMoonlight("Found orphan container. Removing it");
|
||||
|
||||
await EnsureContainerOffline(possibleContainer);
|
||||
|
||||
Logger.LogInformation("Removing orphan container");
|
||||
await DockerClient.Containers.RemoveContainerAsync(ContainerName, new());
|
||||
}
|
||||
catch (DockerContainerNotFoundException)
|
||||
{
|
||||
// Ignored
|
||||
}
|
||||
|
||||
// 2. Ensure the docker image has been downloaded
|
||||
await Console.WriteToMoonlight("Downloading docker image");
|
||||
|
||||
await ImageService.Download(Meta.Configuration.DockerImage, async message =>
|
||||
{
|
||||
try
|
||||
{
|
||||
await Console.WriteToMoonlight(message);
|
||||
}
|
||||
catch (Exception)
|
||||
{
|
||||
// Ignored. Not handling it here could cause an application wide crash afaik
|
||||
}
|
||||
});
|
||||
|
||||
// 3. Create the container from the configuration in the meta
|
||||
var hostFsPath = FileSystem.GetExternalPath();
|
||||
|
||||
var parameters = Mapper.ToRuntimeParameters(
|
||||
Meta.Configuration,
|
||||
hostFsPath,
|
||||
ContainerName
|
||||
);
|
||||
|
||||
var createdContainer = await DockerClient.Containers.CreateContainerAsync(parameters);
|
||||
|
||||
ContainerId = createdContainer.ID;
|
||||
|
||||
Logger.LogInformation("Created container");
|
||||
await Console.WriteToMoonlight("Created container");
|
||||
}
|
||||
|
||||
public async Task Start()
|
||||
{
|
||||
if(string.IsNullOrEmpty(ContainerId))
|
||||
throw new ArgumentNullException(nameof(ContainerId), "Container id of runtime is unknown");
|
||||
|
||||
await Console.WriteToMoonlight("Starting container");
|
||||
await DockerClient.Containers.StartContainerAsync(ContainerId, new());
|
||||
}
|
||||
|
||||
public async Task Stop()
|
||||
{
|
||||
if (Meta.Configuration.StopCommand.StartsWith('^'))
|
||||
{
|
||||
await DockerClient.Containers.KillContainerAsync(ContainerId, new()
|
||||
{
|
||||
Signal = Meta.Configuration.StopCommand.Substring(1)
|
||||
});
|
||||
}
|
||||
else
|
||||
await Console.WriteToInput(Meta.Configuration.StopCommand);
|
||||
}
|
||||
|
||||
public async Task Kill()
|
||||
{
|
||||
await EnsureContainerOffline();
|
||||
}
|
||||
|
||||
public async Task Deprovision()
|
||||
{
|
||||
// Plan of action:
|
||||
// 1. Search for the container by id or name
|
||||
// 2. Ensure container is offline
|
||||
// 3. Remove the container
|
||||
|
||||
// 1. Search for the container by id or name
|
||||
ContainerInspectResponse? container = null;
|
||||
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(ContainerId))
|
||||
container = await DockerClient.Containers.InspectContainerAsync(ContainerName);
|
||||
else
|
||||
container = await DockerClient.Containers.InspectContainerAsync(ContainerId);
|
||||
}
|
||||
catch (DockerContainerNotFoundException)
|
||||
{
|
||||
// Ignored
|
||||
|
||||
Logger.LogDebug("Runtime container could not be found. Reporting deprovision success");
|
||||
}
|
||||
|
||||
// No container found? We are done here then
|
||||
if (container == null)
|
||||
return;
|
||||
|
||||
// 2. Ensure container is offline
|
||||
await EnsureContainerOffline(container);
|
||||
|
||||
// 3. Remove the container
|
||||
Logger.LogInformation("Removing container");
|
||||
await Console.WriteToMoonlight("Removing container");
|
||||
|
||||
await DockerClient.Containers.RemoveContainerAsync(container.ID, new());
|
||||
}
|
||||
|
||||
private async Task EnsureContainerOffline(ContainerInspectResponse? container = null)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(ContainerId))
|
||||
container = await DockerClient.Containers.InspectContainerAsync(ContainerName);
|
||||
else
|
||||
container = await DockerClient.Containers.InspectContainerAsync(ContainerId);
|
||||
}
|
||||
catch (DockerContainerNotFoundException)
|
||||
{
|
||||
// Ignored
|
||||
}
|
||||
|
||||
// No container found? We are done here then
|
||||
if (container == null)
|
||||
return;
|
||||
|
||||
// Check if container is running
|
||||
if (!container.State.Running)
|
||||
return;
|
||||
|
||||
await Console.WriteToMoonlight("Killing container");
|
||||
await DockerClient.Containers.KillContainerAsync(ContainerId, new());
|
||||
}
|
||||
|
||||
public Task<ServerCrash?> SearchForCrash()
|
||||
{
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
|
||||
public async ValueTask DisposeAsync()
|
||||
{
|
||||
OnExitedSubject.Dispose();
|
||||
|
||||
if (ContainerEventSubscription != null)
|
||||
await ContainerEventSubscription.DisposeAsync();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user