scadalink-design/src/ScadaLink.CLI/Commands/SiteCommands.cs

using System.CommandLine;
using System.CommandLine.Parsing;
using ScadaLink.Commons.Messages.Management;

namespace ScadaLink.CLI.Commands;

public static class SiteCommands
{
    public static Command Build(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var command = new Command("site") { Description = "Manage sites" };

        command.Add(BuildList(contactPointsOption, formatOption, usernameOption, passwordOption));
        command.Add(BuildGet(contactPointsOption, formatOption, usernameOption, passwordOption));
        command.Add(BuildCreate(contactPointsOption, formatOption, usernameOption, passwordOption));
        command.Add(BuildUpdate(contactPointsOption, formatOption, usernameOption, passwordOption));
        command.Add(BuildDelete(contactPointsOption, formatOption, usernameOption, passwordOption));
        command.Add(BuildDeployArtifacts(contactPointsOption, formatOption, usernameOption, passwordOption));
        command.Add(BuildArea(contactPointsOption, formatOption, usernameOption, passwordOption));

        return command;
    }

    private static Command BuildGet(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var idOption = new Option<int>("--id") { Description = "Site ID", Required = true };
        var cmd = new Command("get") { Description = "Get a site by ID" };
        cmd.Add(idOption);
        cmd.SetAction(async (ParseResult result) =>
        {
            var id = result.GetValue(idOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new GetSiteCommand(id));
        });
        return cmd;
    }

    private static Command BuildList(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var cmd = new Command("list") { Description = "List all sites" };
        cmd.SetAction(async (ParseResult result) =>
        {
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new ListSitesCommand());
        });
        return cmd;
    }

    private static Command BuildCreate(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var nameOption = new Option<string>("--name") { Description = "Site name", Required = true };
        var identifierOption = new Option<string>("--identifier") { Description = "Site identifier", Required = true };
        var descOption = new Option<string?>("--description") { Description = "Site description" };
        var nodeAOption = new Option<string?>("--node-a-address") { Description = "Akka address for Node A" };
        var nodeBOption = new Option<string?>("--node-b-address") { Description = "Akka address for Node B" };

        var cmd = new Command("create") { Description = "Create a new site" };
        cmd.Add(nameOption);
        cmd.Add(identifierOption);
        cmd.Add(descOption);
        cmd.Add(nodeAOption);
        cmd.Add(nodeBOption);
        cmd.SetAction(async (ParseResult result) =>
        {
            var name = result.GetValue(nameOption)!;
            var identifier = result.GetValue(identifierOption)!;
            var desc = result.GetValue(descOption);
            var nodeA = result.GetValue(nodeAOption);
            var nodeB = result.GetValue(nodeBOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption,
                new CreateSiteCommand(name, identifier, desc, nodeA, nodeB));
        });
        return cmd;
    }

    private static Command BuildUpdate(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var idOption = new Option<int>("--id") { Description = "Site ID", Required = true };
        var nameOption = new Option<string>("--name") { Description = "Site name", Required = true };
        var descOption = new Option<string?>("--description") { Description = "Site description" };
        var nodeAOption = new Option<string?>("--node-a-address") { Description = "Akka address for Node A" };
        var nodeBOption = new Option<string?>("--node-b-address") { Description = "Akka address for Node B" };

        var cmd = new Command("update") { Description = "Update an existing site" };
        cmd.Add(idOption);
        cmd.Add(nameOption);
        cmd.Add(descOption);
        cmd.Add(nodeAOption);
        cmd.Add(nodeBOption);
        cmd.SetAction(async (ParseResult result) =>
        {
            var id = result.GetValue(idOption);
            var name = result.GetValue(nameOption)!;
            var desc = result.GetValue(descOption);
            var nodeA = result.GetValue(nodeAOption);
            var nodeB = result.GetValue(nodeBOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption,
                new UpdateSiteCommand(id, name, desc, nodeA, nodeB));
        });
        return cmd;
    }

    private static Command BuildDelete(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var idOption = new Option<int>("--id") { Description = "Site ID", Required = true };
        var cmd = new Command("delete") { Description = "Delete a site" };
        cmd.Add(idOption);
        cmd.SetAction(async (ParseResult result) =>
        {
            var id = result.GetValue(idOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new DeleteSiteCommand(id));
        });
        return cmd;
    }

    private static Command BuildArea(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var group = new Command("area") { Description = "Manage areas" };

        var siteIdOption = new Option<int>("--site-id") { Description = "Site ID", Required = true };
        var listCmd = new Command("list") { Description = "List areas for a site" };
        listCmd.Add(siteIdOption);
        listCmd.SetAction(async (ParseResult result) =>
        {
            var siteId = result.GetValue(siteIdOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new ListAreasCommand(siteId));
        });
        group.Add(listCmd);

        var createSiteIdOption = new Option<int>("--site-id") { Description = "Site ID", Required = true };
        var nameOption = new Option<string>("--name") { Description = "Area name", Required = true };
        var parentOption = new Option<int?>("--parent-id") { Description = "Parent area ID" };
        var createCmd = new Command("create") { Description = "Create an area" };
        createCmd.Add(createSiteIdOption);
        createCmd.Add(nameOption);
        createCmd.Add(parentOption);
        createCmd.SetAction(async (ParseResult result) =>
        {
            var siteId = result.GetValue(createSiteIdOption);
            var name = result.GetValue(nameOption)!;
            var parentId = result.GetValue(parentOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption,
                new CreateAreaCommand(siteId, name, parentId));
        });
        group.Add(createCmd);

        var updateIdOption = new Option<int>("--id") { Description = "Area ID", Required = true };
        var updateNameOption = new Option<string>("--name") { Description = "New area name", Required = true };
        var updateCmd = new Command("update") { Description = "Update an area" };
        updateCmd.Add(updateIdOption);
        updateCmd.Add(updateNameOption);
        updateCmd.SetAction(async (ParseResult result) =>
        {
            var id = result.GetValue(updateIdOption);
            var name = result.GetValue(updateNameOption)!;
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new UpdateAreaCommand(id, name));
        });
        group.Add(updateCmd);

        var deleteIdOption = new Option<int>("--id") { Description = "Area ID", Required = true };
        var deleteCmd = new Command("delete") { Description = "Delete an area" };
        deleteCmd.Add(deleteIdOption);
        deleteCmd.SetAction(async (ParseResult result) =>
        {
            var id = result.GetValue(deleteIdOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new DeleteAreaCommand(id));
        });
        group.Add(deleteCmd);

        return group;
    }

    private static Command BuildDeployArtifacts(Option<string> contactPointsOption, Option<string> formatOption, Option<string> usernameOption, Option<string> passwordOption)
    {
        var siteIdOption = new Option<int?>("--site-id") { Description = "Target site ID (all sites if omitted)" };
        var cmd = new Command("deploy-artifacts") { Description = "Deploy artifacts to site(s)" };
        cmd.Add(siteIdOption);
        cmd.SetAction(async (ParseResult result) =>
        {
            var siteId = result.GetValue(siteIdOption);
            return await CommandHelpers.ExecuteCommandAsync(
                result, contactPointsOption, formatOption, usernameOption, passwordOption, new MgmtDeployArtifactsCommand(siteId));
        });
        return cmd;
    }
}