From 28a04c64d0c84b7d4531f9a0e8d37524ff9c9cd4 Mon Sep 17 00:00:00 2001
From: pancakes <trypancakes@tutanota.com>
Date: Sat, 29 Mar 2025 13:20:07 +1000
Subject: [PATCH] [backend/razor] Set instance icon on pages if configured

---
 Iceshrimp.Backend/Core/Services/InstanceService.cs |  9 ++++++++-
 Iceshrimp.Backend/Pages/Shared/RootComponent.razor | 12 ++++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/Iceshrimp.Backend/Core/Services/InstanceService.cs b/Iceshrimp.Backend/Core/Services/InstanceService.cs
index 68244380..4e7043eb 100644
--- a/Iceshrimp.Backend/Core/Services/InstanceService.cs
+++ b/Iceshrimp.Backend/Core/Services/InstanceService.cs
@@ -12,7 +12,8 @@ namespace Iceshrimp.Backend.Core.Services;
 public class InstanceService(
 	DatabaseContext db,
 	HttpClient httpClient,
-	ILogger<InstanceService> logger
+	ILogger<InstanceService> logger,
+	MetaService meta
 ) : IScopedService
 {
 	private static readonly AsyncKeyedLocker<string> KeyedLocker = new(o =>
@@ -275,4 +276,10 @@ public class InstanceService(
 
 		return rule;
 	}
+
+	public async Task<string?> GetInstanceImageAsync()
+	{
+		var iconId = await meta.GetAsync(MetaEntity.IconFileId);
+		return await db.DriveFiles.Where(p => p.Id == iconId).Select(p => p.RawAccessUrl).FirstOrDefaultAsync();
+	}
 }
\ No newline at end of file
diff --git a/Iceshrimp.Backend/Pages/Shared/RootComponent.razor b/Iceshrimp.Backend/Pages/Shared/RootComponent.razor
index 917d004c..da8594b4 100644
--- a/Iceshrimp.Backend/Pages/Shared/RootComponent.razor
+++ b/Iceshrimp.Backend/Pages/Shared/RootComponent.razor
@@ -18,6 +18,10 @@
     <VersionedLink rel="stylesheet" href="/Iceshrimp.Backend.styles.css"/>
     <VersionedLink rel="stylesheet" href="/css/default.css"/>
     <VersionedLink rel="icon" type="image/png" href="/_content/Iceshrimp.Assets.Branding/favicon.png"/>
+    @if (_instanceIcon != null)
+    {
+        <link rel="icon" href="@_instanceIcon">
+    }
     <HeadOutlet/>
     <PageTitle>@(_instanceName ?? "Iceshrimp.NET")</PageTitle>
 </head>
@@ -54,18 +58,22 @@
 </html>
 
 @code {
-    [Inject] public required MetaService Meta { get; set; }
+    [Inject] public required InstanceService InstanceSvc { get; set; }
+    [Inject] public required MetaService     Meta        { get; set; }
 
     // This is separate from AuthorizationMiddleware.AuthorizeAttribute, as that middleware is meant for API calls.
     public class RequireAuthorizationAttribute : Attribute;
 
     private string? _instanceName;
     private User?   _user;
+    private string? _instanceIcon;
 
     protected override async Task OnParametersSetAsync()
     {
+        _user = Context.GetUser();
+
         _instanceName = await Meta.GetAsync(MetaEntity.InstanceName);
-        _user         = Context.GetUser();
+        _instanceIcon = await InstanceSvc.GetInstanceImageAsync();
 
         if (Context.GetEndpoint()?.Metadata.GetMetadata<RequireAuthorizationAttribute>() == null)
             return;