Innovenergy_trunk/csharp/App/VrmGrabber/Database/Db.cs

using System.Diagnostics.CodeAnalysis;
using System.Web;
using CliWrap;
using CliWrap.Buffered;
using InnovEnergy.App.RemoteSupportConsole;
using InnovEnergy.Lib.Utils;
using InnovEnergy.Lib.Victron.VictronVRM;
using SQLite;
using static System.Text.Json.JsonSerializer;
using Installation = InnovEnergy.App.VrmGrabber.DataTypes.Installation;
using VrmInstallation = InnovEnergy.Lib.Victron.VictronVRM.Installation;
using FILE=System.IO.File;



namespace InnovEnergy.App.VrmGrabber.Database;

public class InstallationDetails
{
    public InstallationDetails(String? ip, IReadOnlyList<Detail> details)
    {
        Details = details;
        Ip = ip;
    }

    public IReadOnlyList<Detail>? Details { get; set; }
    public String? Ip { get; set; }
}

public static partial class Db
{
    public static Dictionary<Installation, InstallationDetails> InstallationsAndDetails;


    internal const String DbPath = "./db.sqlite";

    private static SQLiteConnection Connection { get; } = new SQLiteConnection(DbPath);

    public static TableQuery<Installation> Installations => Connection.Table<Installation>();

    public static void Init()
    {
        // used to force static constructor
    }


    static Db()
    {
        // on startup create/migrate tables

        Connection.RunInTransaction(() => { Connection.CreateTable<Installation>(); });
    }

    public static async Task UpdateDetailsAndInstallations()
    {
        while (true)
        {
            await UpdateInstallationsAndDetailsFromVrm(0);
        }
    }

    [UnconditionalSuppressMessage("Trimming", "IL2026:Members annotated with 'RequiresUnreferencedCodeAttribute' require dynamic access otherwise can break functionality when trimming application code", Justification = "<Pending>")]
    private static async Task UpdateInstallationsAndDetailsFromVrm(Int32 _)
    {
        var fileContent = await File.ReadAllTextAsync("./token.json");

        var acc = Deserialize<AccToken>(fileContent);
        var user = VrmAccount.Token(acc!.idUser, acc.token);
        var installations = await user.GetInstallations();

        // var returnDictionary = new Dictionary<VrmInstallation, InstallationDetails>();
        foreach (var installation in installations.Take(20)) //TODO REMOVE TAKE
        {
            Console.WriteLine(installation.Name);
            var details = await GetInstallationDetails(installation);
            var ip = Ip(details);
            var updatedInstallation = new Installation(
                installation.Name,
                ip[0],
                (Int64)installation.IdSite,
                installation.Identifier,
                details.Details.MachineSerial() ?? "Unknown",
                HttpUtility.UrlEncode(installation.Name), 
                ip[1],
                details.Details.Last().Json["timestamp"].ToString(),
                Serialize(details.Details));
            
            if (GetInstallationByIdentifier(installation.Identifier) == null)
            {
                Create(updatedInstallation);
            }
            else
            {
                Update(updatedInstallation);
            }
        }
    }

    private static String?[] Ip(InstallationDetails details)
    {
        var online = "❌";
        var lookup = details.Ip;
        if (lookup == "Unknown")
        {
            var serial = details.Details.MachineSerial() ?? "Unknown";
            if (serial != "Unknown" && FILE.Exists($@"/etc/openvpn/server/Salino/ccd/{serial}"))
                lookup = FILE.ReadAllText($@"/etc/openvpn/server/Salino/ccd/{serial}").Split(' ')[1];
        }
        else
        {
            online = "✅";
        }

        return new[] { lookup, online };
    }

    private static async Task<InstallationDetails> GetInstallationDetails(VrmInstallation i)
    {
        await Task.Delay(1000);
        try
        {
            var details = await i.GetDetails();

            var ip = await VpnInfo.LookUpIp(i.Identifier, details.MachineSerial()) ?? "Unknown";

            if (ip != "Unknown")
                await UpdateInstallationName(i, ip);

            return new InstallationDetails(ip, details);
        }
        catch (Exception e)
        {
            Console.WriteLine(e);
        }

        return new InstallationDetails("Unknown", Array.Empty<Detail>());
    }

    private static async Task UpdateInstallationName(VrmInstallation installation, String? ip)
    {
        var oldNameInFileRequest = await Cli.Wrap("ssh")
            .WithArguments($@"root@{ip}")
            .AppendArgument("-o StrictHostKeyChecking=accept-new")
            .AppendArgument("cat /data/innovenergy/openvpn/installation-name")
            .WithValidation(CommandResultValidation.None).ExecuteBufferedAsync();

        var oldNameInFileWithoutNewLine = oldNameInFileRequest.StandardOutput.TrimEnd();

        if (oldNameInFileRequest.ExitCode == 0 && oldNameInFileWithoutNewLine != installation.Name)
        {
            var overwriteNameCommand = Cli.Wrap("ssh")
                .WithArguments($@"root@{ip}")
                .AppendArgument($"echo '{installation.Name}' > /data/innovenergy/openvpn/installation-name");

            var overwriteNameResponse = await overwriteNameCommand
                .WithValidation(CommandResultValidation.None).ExecuteBufferedAsync();

            if (overwriteNameResponse.ExitCode != 0)
            {
                Console.WriteLine(overwriteNameResponse.StandardError);
                Console.WriteLine("Renaming did not work");
            }
            else
            {
                var rebootAfterRename = await Cli.Wrap("ssh")
                    .WithArguments($@"root@{ip}")
                    .AppendArgument("reboot")
                    .WithValidation(CommandResultValidation.None).ExecuteBufferedAsync();

                if (rebootAfterRename.ExitCode != 0)
                {
                    Console.WriteLine(overwriteNameResponse.StandardError);
                    Console.WriteLine("Rebooting did not work");
                }
                else
                {
                    Console.WriteLine($"Renamed and rebooted {installation.Name}");
                }
            }
        }
    }


    private static Boolean RunTransaction(Func<Boolean> func)
    {
        var savepoint = Connection.SaveTransactionPoint();
        var success = false;

        try
        {
            success = func();
        }
        finally
        {
            if (success)
                Connection.Release(savepoint);
            else
                Connection.RollbackTo(savepoint);
        }

        return success;
    }
}

public class AccToken
{
    public UInt64 idUser { get; init;} 
    public String token { get; init;}
}