Device discovery and maybe adding devices ╮(╯▽╰)╭

src/Elektrifikatsiya/Elektrifikatsiya/Pages/AddPlug.razor

@@ -2,13 +2,18 @@
 @using Blazorise.Components;
 @using Elektrifikatsiya.Models;
 @using System.Net
+@using Elektrifikatsiya.Services
+@using Elektrifikatsiya.Services.Implementations
+@using Elektrifikatsiya.Utilities
 @inject IVersionProvider VersionProvider
+@inject IDeviceManagmentService DeviceManagmentService
+@inject IAuthenticationService AuthenticationService 
 
 <Row Height=Height.Is50 Padding="Padding.Is3.OnDesktop.Is0">
 	<Column ColumnSize="ColumnSize.Is12" style="border-bottom-style:solid; border-bottom-color:#1a237e">
 		<ListView  Class="w-100"
-				  TItem="Device"
-				  Data="@plugs"
+				  TItem="IPAddress"
+				  Data="@ipAddresses"
 				  TextField="@((_)=>(""))"
 				  ValueField="@((_)=>(""))"
 				  Mode="ListGroupMode.Static">
@@ -18,17 +23,17 @@
 						<Column ColumnSize="ColumnSize.Is6">
 							<Row>
 								<Column >
-									<Heading Size="HeadingSize.Is6" Margin="Margin.Is1.FromBottom">@context.Item.Name</Heading>
+									<Heading Size="HeadingSize.Is6" Margin="Margin.Is1.FromBottom">@context.Item</Heading>
 								</Column>
 							</Row>
 							<Row Width="Width.Is100">
 								<Column>
-									<Small>IP: @context.Item.IpAddress</Small>
+									<Small>IP: @context.Item</Small>
 								</Column>
 							</Row>
 						</Column>
 						<Column ColumnSize="ColumnSize.Is4.Is2.WithOffset">
-							<Button Position="Position.Absolute.Top.Is50.Start.Is50.Translate.Middle" Color="Color.Dark" Outline>
+							<Button Position="Position.Absolute.Top.Is50.Start.Is50.Translate.Middle" Color="Color.Dark" Clicked="@(async () => await DeviceManagmentService.RegisterDevice(context.Item,(await AuthenticationService.GetUserAsync()).ValueOrDefault))" Outline>
 								<BarIcon IconName="IconName.Add" />
 							</Button>
 						</Column>
@@ -40,5 +45,14 @@
 </Row>
 
 @code {
-	List<Device> plugs = new List<Device>() { new Device("ffff:ffff:ffff:ffff", "Huso", IPAddress.Broadcast, new User("Joe", "qiowruioewjfopa", Role.Admin), "Raum"), new Device("ffff:ffff:ffff:ffff", "Huso", IPAddress.Broadcast, new User("Joe", "qiowruioewjfopa", Role.Admin), "Raum"), new Device("ffff:ffff:ffff:ffff", "Huso", IPAddress.Broadcast, new User("Joe", "qiowruioewjfopa", Role.Admin), "Raum"), };
+	List<IPAddress> ipAddresses = new List<IPAddress>();
+
+	protected override async Task OnInitializedAsync()
+	{
+		MdnsDiscovery.OnDeviceFound += address =>
+		{
+			ipAddresses.Add(address);
+			InvokeAsync(StateHasChanged);
+		};
+	}
 }

src/Elektrifikatsiya/Elektrifikatsiya/Program.cs

@@ -15,9 +15,13 @@ WebApplicationBuilder builder = WebApplication.CreateBuilder(args);
 // Add services to the container.
 builder.Services.AddRazorPages();
 builder.Services.AddServerSideBlazor();
+builder.Services.AddHttpContextAccessor();
 builder.Services.AddHostedService<UpdateService>();
 builder.Services.AddSingleton<IDeviceStatusService, DeviceStatusService>();
 builder.Services.AddTransient<IDeviceManagmentService, DeviceManagmentService>();
+builder.Services.AddScoped<IAuthenticationService, AuthenticationService>();
+builder.Services.AddScoped<IAuthorizationService, AuthorizationService>();
+builder.Services.AddScoped<ICookieService, CookieService>();
 builder.Services.AddDbContext<UserDatabaseContext>(options => options.UseSqlite("Data Source=./UserDatabase.sqlite"));
 builder.Services.AddDbContext<DeviceManagmentDatabaseContext>((options) => options.UseSqlite("Data Source=./DeviceManagement.sqlite"));
 builder.Services.AddBootstrapProviders();
@@ -46,12 +50,11 @@ app.UseStaticFiles();
 app.UseRouting();
 app.MapBlazorHub();
 app.MapFallbackToPage("/_Host");
+app.MapControllers();
 
 IServiceScope serviceScope = app.Services.GetRequiredService<IServiceScopeFactory>().CreateScope();
 serviceScope.ServiceProvider.GetRequiredService<UserDatabaseContext>().Database.EnsureCreated();
 
-app.MapControllers();
-
 AsyncServiceScope scope = app.Services.CreateAsyncScope();
 scope.ServiceProvider.GetRequiredService<DeviceManagmentDatabaseContext>().Database.EnsureCreated();
 

src/Elektrifikatsiya/Elektrifikatsiya/Properties/launchSettings.json

@@ -7,7 +7,7 @@
         "ASPNETCORE_ENVIRONMENT": "Development"
       },
       "dotnetRunMessages": true,
-      "applicationUrl": "https://localhost:7297;http://localhost:5255"
+      "applicationUrl": "https://localhost:56656;http://localhost:5255"
     },
     "IIS Express": {
       "commandName": "IISExpress",

src/Elektrifikatsiya/Elektrifikatsiya/Services/IDeviceManagmentService.cs

@@ -8,9 +8,9 @@ namespace Elektrifikatsiya.Services;
 
 public interface IDeviceManagmentService
 {
-    public Task<Result<Device>> Register(IPAddress ip, User user, string? name = null, string room = "default");
+    public Task<Result<Device>> RegisterDevice(IPAddress ip, User user, string? name = null, string room = "default");
 
-    public Task<Result> Unregister(string macAdress);
+    public Task<Result> UnregisterDevice(string macAdress);
 
     public Result<Device> GetDevice(string macAdress);
 
@@ -19,4 +19,6 @@ public interface IDeviceManagmentService
     public Result<List<Device>> GetDevicesInRoom(string room);
 
     public Result<List<Device>> GetDevicesOfUser(int userId);
+
+    public Task<Result> UpdateDevice(Device device);
 }

src/Elektrifikatsiya/Elektrifikatsiya/Services/Implementations/DeviceManagmentService.cs

@@ -11,15 +11,16 @@ namespace Elektrifikatsiya.Services.Implementations;
 public class DeviceManagmentService : IDeviceManagmentService
 {
     private readonly DeviceManagmentDatabaseContext deviceManagmentDatabaseContext;
-	private readonly IDeviceStatusService deviceStatusService;
     private readonly IServiceScopeFactory serviceScopeFactory;
+    private readonly IDeviceStatusService deviceStatusService;
     private readonly HttpClient httpClient;
 
     public DeviceManagmentService(IServiceScopeFactory serviceScopeFactory, IDeviceStatusService deviceStatusService, HttpClient httpClient)
     {
-		this.deviceManagmentDatabaseContext = deviceManagmentDatabaseContext;
+	    this.serviceScopeFactory = serviceScopeFactory;
 	    this.deviceStatusService = deviceStatusService;
         this.httpClient = httpClient;
+
            }
 
     public Result<Device> GetDevice(string macAdress)
@@ -77,21 +78,21 @@ public class DeviceManagmentService : IDeviceManagmentService
         return getDevicesResult.Value.Where(d => d.User.Id == userId).ToList();
     }
 
-	public async Task<Result<Device>> Register(IPAddress ip, User user, string? name = null, string room = "default")
+    public async Task<Result<Device>> RegisterDevice(IPAddress ip, User user, string? name = null, string room = "default")
     {
 	    using IServiceScope scope = serviceScopeFactory.CreateScope();
 	    DeviceManagmentDatabaseContext deviceManagmentDatabaseContext = scope.ServiceProvider.GetRequiredService<DeviceManagmentDatabaseContext>();
 
-		ShellyResponse? shellyResponse = await httpClient.GetFromJsonAsync<ShellyResponse>($"{ip}/shelly");
+		ShellyResponse? shellyResponse = await httpClient.GetFromJsonAsync<ShellyResponse>($"http://{ip}/shelly");
 
         if (shellyResponse is null || shellyResponse.Type != "SHPLG-S")
         {
-			return Result.Fail("Device is not a \"SHPLG-S\" or not reachable!");
+            return Result.Fail("Device is not reachable or not a \"SHPLG-S\"!");
         }
 
         string mac = shellyResponse.Mac;
 
-		if (PhysicalAddress.TryParse(mac, out _))
+        if (!PhysicalAddress.TryParse(mac, out _))
         {
             return Result.Fail("Invalid mac address!");
         }
@@ -99,14 +100,14 @@ public class DeviceManagmentService : IDeviceManagmentService
         Device device = new Device(mac, name ?? mac, ip, user, room);
 
         deviceManagmentDatabaseContext.Add(device);
-		Result saveDatabaseChangesResult = await Result.Try(async Task () => await deviceManagmentDatabaseContext.SaveChangesAsync());
+        Result saveDatabaseChangesResult = await Result.Try(Task () => deviceManagmentDatabaseContext.SaveChangesAsync());
 
         Result trackDeviceResult = deviceStatusService.TrackDevice(device);
 
         return Result.Merge(saveDatabaseChangesResult, trackDeviceResult).ToResult(device);
     }
 
-	public async Task<Result> Unregister(string macAdress)
+    public async Task<Result> UnregisterDevice(string macAdress)
     {
 	    using IServiceScope scope = serviceScopeFactory.CreateScope();
 	    DeviceManagmentDatabaseContext deviceManagmentDatabaseContext = scope.ServiceProvider.GetRequiredService<DeviceManagmentDatabaseContext>();
@@ -127,6 +128,20 @@ public class DeviceManagmentService : IDeviceManagmentService
 
         deviceManagmentDatabaseContext.Remove(getDeviceResult.Value);
 
-		return await Result.Try(async Task () => await deviceManagmentDatabaseContext.SaveChangesAsync());
+        return await Result.Try(Task () => deviceManagmentDatabaseContext.SaveChangesAsync());
+    }
+
+    public async Task<Result> UpdateDevice(Device device)
+    {
+        Result result = deviceStatusService.UpdateDeviceStatus(device);
+
+        if (result.IsFailed)
+        {
+            return result;
+        }
+
+        deviceManagmentDatabaseContext.Update(device);
+
+        return await Result.Try(Task () => deviceManagmentDatabaseContext.SaveChangesAsync());
     }
 }

src/Elektrifikatsiya/Elektrifikatsiya/Services/Implementations/DeviceStatusService.cs

@@ -6,10 +6,15 @@ namespace Elektrifikatsiya.Services.Implementations;
 
 public class DeviceStatusService : IDeviceStatusService
 {
+	private readonly ILogger<DeviceStatusService> logger;
 	public event EventHandler<DeviceStatusChagedEventArgs>? OnDeviceStatusChanged;
 
     private readonly Dictionary<string, Device> devices = new();
     
+    public DeviceStatusService(ILogger<DeviceStatusService> logger)
+    {
+	    this.logger = logger;
+    }
     public Result<List<Device>> GetDevices()
     {
         return devices.Values.ToList();
@@ -41,6 +46,8 @@ public class DeviceStatusService : IDeviceStatusService
             return Result.Fail("Device already tracked!");
         }
 
+        logger.LogInformation("Tracking new Device {device}", device.IpAddress);
+
         devices[device.MacAddress] = device;
 
         return Result.Ok();

src/Elektrifikatsiya/Elektrifikatsiya/Utilities/MdnsDiscovery.cs

@@ -0,0 +1,20 @@
+using System.Net;
+using FluentResults;
+using Tmds.MDns;
+
+namespace Elektrifikatsiya.Utilities;
+
+public static class MdnsDiscovery
+{
+	public static event Action<IPAddress> OnDeviceFound;
+
+	static ServiceBrowser serviceBrowser = new ServiceBrowser();
+
+	static MdnsDiscovery()
+	{
+		serviceBrowser.StartBrowse("_http._tcp");
+        serviceBrowser.ServiceAdded += (_, eventArgs) => OnDeviceFound?.Invoke(eventArgs.Announcement.Addresses.First());
+        serviceBrowser.ServiceChanged += (_, eventArgs) => OnDeviceFound?.Invoke(eventArgs.Announcement.Addresses.First());
+        serviceBrowser.ServiceRemoved += (_, eventArgs) => OnDeviceFound?.Invoke(eventArgs.Announcement.Addresses.First());
+    }
+}