Update application core

2025-08-29 06:18:08 +02:00 · 2024-01-25 20:38:47 +03:30 · 2024-01-25 20:38:47 +03:30 · c118242c5b
commit c118242c5b
parent 6655664ed6
5 changed files with 474 additions and 27 deletions
--- a/Application/SetupValidator.cs
+++ b/Application/SetupValidator.cs
@ -0,0 +1,140 @@
+using AutoMapper;
+using Microsoft.AspNetCore.Http;
+using Microsoft.Extensions.DependencyInjection;
+using MTWireGuard.Application.Models;
+using MTWireGuard.Application.Repositories;
+using MTWireGuard.Application.Services;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Net.Sockets;
+using System.Text;
+using System.Threading.Tasks;
+using static System.Runtime.InteropServices.JavaScript.JSType;
+
+namespace MTWireGuard.Application
+{
+    public class SetupValidator(IServiceProvider serviceProvider)
+    {
+        private IMikrotikRepository api;
+
+        public async Task Validate()
+        {
+            var envVariables = ValidateEnvironmentVariables();
+            if (envVariables)
+            {
+                Console.BackgroundColor = ConsoleColor.Black;
+                Console.ForegroundColor = ConsoleColor.Red;
+                Console.WriteLine($"[-] Environment variables are not set!");
+                Console.WriteLine($"[!] Please set \"MT_IP\", \"MT_USER\", \"MT_PASS\", \"MT_PUBLIC_IP\" variables in container environment.");
+                Console.ResetColor();
+                Shutdown();
+            }
+
+            serviceProvider.GetService<DBContext>().Database.EnsureCreated();
+            api = serviceProvider.GetService<IMikrotikRepository>();
+
+            var (apiConnection, apiConnectionMessage) = await ValidateAPIConnection();
+            if (!apiConnection)
+            {
+                Console.BackgroundColor = ConsoleColor.Black;
+                Console.ForegroundColor = ConsoleColor.Red;
+                Console.WriteLine($"[-] Error connecting to the router api!");
+                Console.WriteLine($"[!] {apiConnectionMessage}");
+                Console.ResetColor();
+                Shutdown();
+            }
+
+            var ip = GetIPAddress();
+            if (string.IsNullOrEmpty(ip))
+            {
+                Console.BackgroundColor = ConsoleColor.Black;
+                Console.ForegroundColor = ConsoleColor.Red;
+                Console.WriteLine($"[-] Error getting container IP address!");
+                Console.ResetColor();
+                Shutdown();
+            }
+            var scripts = await api.GetScripts();
+            var schedulers = await api.GetSchedulers();
+            var trafficScript = scripts.Find(x => x.Name == "SendTrafficUsage");
+            var handshakeScript = scripts.Find(x => x.Name == "SendActivityUpdates");
+            var trafficScheduler = schedulers.Find(x => x.Name == "TrafficUsage");
+
+            if (trafficScript == null)
+            {
+                var create = await api.CreateScript(new()
+                {
+                    Name = "SendTrafficUsage",
+                    Policies = ["write", "read", "test"],
+                    DontRequiredPermissions = false,
+                    Source = Helper.PeersTrafficUsageScript($"http://{ip}/api/usage")
+                });
+                var result = create.Code;
+            }
+            if (handshakeScript == null)
+            {
+                var create = await api.CreateScript(new()
+                {
+                    Name = "SendActivityUpdates",
+                    Policies = ["write", "read", "test"],
+                    DontRequiredPermissions = false,
+                    Source = Helper.PeersLastHandshakeScript($"http://{ip}/api/activity")
+                });
+                var result = create.Code;
+            }
+            if (trafficScheduler == null)
+            {
+                var create = await api.CreateScheduler(new()
+                {
+                    Name = "TrafficUsage",
+                    Interval = new TimeSpan(0, 5, 0),
+                    OnEvent = "SendTrafficUsage",
+                    Policies = ["write", "read", "test"]
+                });
+                var result = create.Code;
+            }
+        }
+
+        private static bool ValidateEnvironmentVariables()
+        {
+            string? IP = Environment.GetEnvironmentVariable("MT_IP");
+            string? USER = Environment.GetEnvironmentVariable("MT_USER");
+            string? PASS = Environment.GetEnvironmentVariable("MT_PASS");
+            string? PUBLICIP = Environment.GetEnvironmentVariable("MT_PUBLIC_IP");
+
+            return string.IsNullOrEmpty(IP) || string.IsNullOrEmpty(USER) || string.IsNullOrEmpty(PUBLICIP);
+        }
+
+        private async Task<(bool status, string? message)> ValidateAPIConnection()
+        {
+            try
+            {
+                return (await api.TryConnectAsync(), string.Empty);
+            }
+            catch (Exception ex)
+            {
+                return (false, ex.Message);
+            }
+        }
+
+        private static string GetIPAddress()
+        {
+            try
+            {
+                var name = System.Net.Dns.GetHostName();
+                var port = Environment.GetEnvironmentVariable("ASPNETCORE_HTTP_PORTS");
+                return System.Net.Dns.GetHostEntry(name).AddressList.FirstOrDefault(x => x.AddressFamily == AddressFamily.InterNetwork).ToString() + $":{port}";
+            }
+            catch (Exception ex)
+            {
+                Console.WriteLine(ex.Message);
+                return string.Empty;
+            }
+        }
+
+        private static void Shutdown()
+        {
+            Environment.Exit(0);
+        }
+    }
+}