From 8418b26d702180d84df9adaf728b9ac1c3b4a3c6 Mon Sep 17 00:00:00 2001
From: Roger Far <roger@versluis.ca>
Date: Sun, 11 Jun 2023 12:31:13 -0600
Subject: [PATCH] Add option to set the base href.

---
 client/src/app/auth.service.ts                | 17 +++---
 client/src/app/login/login.component.html     |  2 +-
 client/src/app/navbar/navbar.component.html   |  2 +-
 client/src/app/settings.service.ts            | 19 +++----
 client/src/app/setup/setup.component.html     |  8 +--
 client/src/app/torrent.service.ts             | 30 ++++++-----
 client/src/index.html                         |  3 +-
 .../Models/Internal/AppSettings.cs            |  1 +
 .../Middleware/BaseHrefMiddleware.cs          | 54 +++++++++++++++++++
 server/RdtClient.Web/Program.cs               |  5 ++
 .../Properties/launchSettings.json            |  2 +-
 server/RdtClient.Web/appsettings.json         |  3 +-
 12 files changed, 107 insertions(+), 39 deletions(-)
 create mode 100644 server/RdtClient.Service/Middleware/BaseHrefMiddleware.cs

diff --git a/client/src/app/auth.service.ts b/client/src/app/auth.service.ts
index 56d5a39..47e5b76 100644
--- a/client/src/app/auth.service.ts
+++ b/client/src/app/auth.service.ts
@@ -1,44 +1,45 @@
+import { APP_BASE_HREF } from '@angular/common';
 import { HttpClient } from '@angular/common/http';
-import { Injectable } from '@angular/core';
+import { Inject, Injectable } from '@angular/core';
 import { Observable } from 'rxjs/internal/Observable';
 
 @Injectable({
   providedIn: 'root',
 })
 export class AuthService {
-  constructor(private http: HttpClient) {}
+  constructor(private http: HttpClient, @Inject(APP_BASE_HREF) private baseHref: string) {}
 
   public isLoggedIn(): Observable<boolean> {
-    return this.http.get<boolean>(`/Api/Authentication/IsLoggedIn`);
+    return this.http.get<boolean>(`${this.baseHref}Api/Authentication/IsLoggedIn`);
   }
 
   public create(userName: string, password: string): Observable<void> {
-    return this.http.post<void>(`/Api/Authentication/Create`, {
+    return this.http.post<void>(`${this.baseHref}Api/Authentication/Create`, {
       userName,
       password,
     });
   }
 
   public setupProvider(provider: string, token: string): Observable<void> {
-    return this.http.post<void>(`/Api/Authentication/SetupProvider`, {
+    return this.http.post<void>(`${this.baseHref}Api/Authentication/SetupProvider`, {
       provider,
       token,
     });
   }
 
   public login(userName: string, password: string): Observable<void> {
-    return this.http.post<void>(`/Api/Authentication/Login`, {
+    return this.http.post<void>(`${this.baseHref}Api/Authentication/Login`, {
       userName,
       password,
     });
   }
 
   public logout() {
-    return this.http.post<void>(`/Api/Authentication/Logout`, {});
+    return this.http.post<void>(`${this.baseHref}Api/Authentication/Logout`, {});
   }
 
   public update(userName: string, password: string): Observable<void> {
-    return this.http.post<void>(`/Api/Authentication/Update`, {
+    return this.http.post<void>(`${this.baseHref}Api/Authentication/Update`, {
       userName,
       password,
     });
diff --git a/client/src/app/login/login.component.html b/client/src/app/login/login.component.html
index 465b3a4..87fcb3f 100644
--- a/client/src/app/login/login.component.html
+++ b/client/src/app/login/login.component.html
@@ -3,7 +3,7 @@
     <div class="container">
       <div class="columns is-centered">
         <div class="column is-5-tablet is-5-desktop is-5-widescreen">
-          <img src="../../assets/logo.png" />
+          <img src="assets/logo.png" />
           <div class="box">
             <form (ngSubmit)="login()">
               <div class="field">
diff --git a/client/src/app/navbar/navbar.component.html b/client/src/app/navbar/navbar.component.html
index 69f3e49..b944756 100644
--- a/client/src/app/navbar/navbar.component.html
+++ b/client/src/app/navbar/navbar.component.html
@@ -1,7 +1,7 @@
 <nav class="navbar is-dark is-fixed-top" role="navigation" aria-label="main navigation">
   <div class="navbar-brand">
     <a class="navbar-item" routerLink="/">
-      <img src="../../assets/logo.png" />
+      <img src="assets/logo.png" />
     </a>
 
     <a
diff --git a/client/src/app/settings.service.ts b/client/src/app/settings.service.ts
index 8ba08c0..4f304b9 100644
--- a/client/src/app/settings.service.ts
+++ b/client/src/app/settings.service.ts
@@ -1,41 +1,42 @@
 import { HttpClient } from '@angular/common/http';
-import { Injectable } from '@angular/core';
+import { Inject, Injectable } from '@angular/core';
 import { Observable } from 'rxjs';
 import { Profile } from './models/profile.model';
 import { Setting } from './models/setting.model';
+import { APP_BASE_HREF } from '@angular/common';
 
 @Injectable({
   providedIn: 'root',
 })
 export class SettingsService {
-  constructor(private http: HttpClient) {}
+  constructor(private http: HttpClient, @Inject(APP_BASE_HREF) private baseHref: string) {}
 
   public get(): Observable<Setting[]> {
-    return this.http.get<Setting[]>(`/Api/Settings`);
+    return this.http.get<Setting[]>(`${this.baseHref}Api/Settings`);
   }
 
   public update(settings: Setting[]): Observable<void> {
-    return this.http.put<void>(`/Api/Settings`, settings);
+    return this.http.put<void>(`${this.baseHref}Api/Settings`, settings);
   }
 
   public getProfile(): Observable<Profile> {
-    return this.http.get<Profile>(`/Api/Settings/Profile`);
+    return this.http.get<Profile>(`${this.baseHref}Api/Settings/Profile`);
   }
 
   public testPath(path: string): Observable<void> {
-    return this.http.post<void>(`/Api/Settings/TestPath`, { path });
+    return this.http.post<void>(`${this.baseHref}Api/Settings/TestPath`, { path });
   }
 
   public testDownloadSpeed(): Observable<number> {
-    return this.http.get<number>(`/Api/Settings/TestDownloadSpeed`);
+    return this.http.get<number>(`${this.baseHref}Api/Settings/TestDownloadSpeed`);
   }
 
   public testWriteSpeed(): Observable<number> {
-    return this.http.get<number>(`/Api/Settings/TestWriteSpeed`);
+    return this.http.get<number>(`${this.baseHref}Api/Settings/TestWriteSpeed`);
   }
 
   public testAria2cConnection(url: string, secret: string): Observable<{ version: string }> {
-    return this.http.post<{ version: string }>(`/Api/Settings/TestAria2cConnection`, {
+    return this.http.post<{ version: string }>(`${this.baseHref}Api/Settings/TestAria2cConnection`, {
       url,
       secret,
     });
diff --git a/client/src/app/setup/setup.component.html b/client/src/app/setup/setup.component.html
index aa7a9fd..29ebff0 100644
--- a/client/src/app/setup/setup.component.html
+++ b/client/src/app/setup/setup.component.html
@@ -3,7 +3,7 @@
     <div class="container">
       <div class="columns is-centered">
         <div class="column is-5-tablet is-5-desktop is-5-widescreen">
-          <img src="../../assets/logo.png" />
+          <img src="assets/logo.png" />
           <div class="box" *ngIf="step === 1">
             <div class="notification is-primary">
               Welcome to Real-Debrid Client. Please create your account by entering a username and password.
@@ -52,7 +52,7 @@
                 >Use this link to sign up to AllDebrid.</a
               >
               <br />
-			  <a href="https://www.premiumize.me/" target="_blank" rel="noopener"
+              <a href="https://www.premiumize.me/" target="_blank" rel="noopener"
                 >Use this link to sign up to Premiumize.</a
               >
               <br />
@@ -64,7 +64,7 @@
                 <select [(ngModel)]="provider">
                   <option [value]="0">Real-Debrid</option>
                   <option [value]="1">AllDebrid</option>
-				  <option [value]="2">Premiumize</option>
+                  <option [value]="2">Premiumize</option>
                 </select>
               </div>
             </div>
@@ -85,7 +85,7 @@
                   >https://alldebrid.com/apikeys/</a
                 >.
               </p>
-			  <p class="help" *ngIf="provider === 'Premiumize'">
+              <p class="help" *ngIf="provider === 'Premiumize'">
                 You can find your API key here:
                 <a href="https://www.premiumize.me/account" target="_blank" rel="noopener"
                   >https://www.premiumize.me/account</a
diff --git a/client/src/app/torrent.service.ts b/client/src/app/torrent.service.ts
index 22e607d..b228596 100644
--- a/client/src/app/torrent.service.ts
+++ b/client/src/app/torrent.service.ts
@@ -1,8 +1,9 @@
 import { HttpClient } from '@angular/common/http';
-import { Injectable } from '@angular/core';
+import { Inject, Injectable } from '@angular/core';
 import * as signalR from '@microsoft/signalr';
 import { Observable, Subject } from 'rxjs';
 import { Torrent, TorrentFileAvailability } from './models/torrent.model';
+import { APP_BASE_HREF } from '@angular/common';
 
 @Injectable({
   providedIn: 'root',
@@ -12,7 +13,7 @@ export class TorrentService {
 
   private connection: signalR.HubConnection;
 
-  constructor(private http: HttpClient) {
+  constructor(private http: HttpClient, @Inject(APP_BASE_HREF) private baseHref: string) {
     this.connect();
   }
 
@@ -21,7 +22,10 @@ export class TorrentService {
       return;
     }
 
-    this.connection = new signalR.HubConnectionBuilder().withUrl('/hub').withAutomaticReconnect().build();
+    this.connection = new signalR.HubConnectionBuilder()
+      .withUrl(`${this.baseHref}hub`)
+      .withAutomaticReconnect()
+      .build();
     this.connection.start().catch((err) => console.error(err));
 
     this.connection.on('update', (torrents: Torrent[]) => {
@@ -30,15 +34,15 @@ export class TorrentService {
   }
 
   public getList(): Observable<Torrent[]> {
-    return this.http.get<Torrent[]>(`/Api/Torrents`);
+    return this.http.get<Torrent[]>(`${this.baseHref}Api/Torrents`);
   }
 
   public get(torrentId: string): Observable<Torrent> {
-    return this.http.get<Torrent>(`/Api/Torrents/Get/${torrentId}`);
+    return this.http.get<Torrent>(`${this.baseHref}Api/Torrents/Get/${torrentId}`);
   }
 
   public uploadMagnet(magnetLink: string, torrent: Torrent): Observable<void> {
-    return this.http.post<void>(`/Api/Torrents/UploadMagnet`, {
+    return this.http.post<void>(`${this.baseHref}Api/Torrents/UploadMagnet`, {
       magnetLink,
       torrent,
     });
@@ -48,11 +52,11 @@ export class TorrentService {
     const formData: FormData = new FormData();
     formData.append('file', file);
     formData.append('formData', JSON.stringify({ torrent }));
-    return this.http.post<void>(`/Api/Torrents/UploadFile`, formData);
+    return this.http.post<void>(`${this.baseHref}Api/Torrents/UploadFile`, formData);
   }
 
   public checkFilesMagnet(magnetLink: string): Observable<TorrentFileAvailability[]> {
-    return this.http.post<TorrentFileAvailability[]>(`/Api/Torrents/CheckFilesMagnet`, {
+    return this.http.post<TorrentFileAvailability[]>(`${this.baseHref}Api/Torrents/CheckFilesMagnet`, {
       magnetLink,
     });
   }
@@ -60,7 +64,7 @@ export class TorrentService {
   public checkFiles(file: File): Observable<TorrentFileAvailability[]> {
     const formData: FormData = new FormData();
     formData.append('file', file);
-    return this.http.post<TorrentFileAvailability[]>(`/Api/Torrents/CheckFiles`, formData);
+    return this.http.post<TorrentFileAvailability[]>(`${this.baseHref}Api/Torrents/CheckFiles`, formData);
   }
 
   public delete(
@@ -69,7 +73,7 @@ export class TorrentService {
     deleteRdTorrent: boolean,
     deleteLocalFiles: boolean
   ): Observable<void> {
-    return this.http.post<void>(`/Api/Torrents/Delete/${torrentId}`, {
+    return this.http.post<void>(`${this.baseHref}Api/Torrents/Delete/${torrentId}`, {
       deleteData,
       deleteRdTorrent,
       deleteLocalFiles,
@@ -77,14 +81,14 @@ export class TorrentService {
   }
 
   public retry(torrentId: string): Observable<void> {
-    return this.http.post<void>(`/Api/Torrents/Retry/${torrentId}`, {});
+    return this.http.post<void>(`${this.baseHref}Api/Torrents/Retry/${torrentId}`, {});
   }
 
   public retryDownload(downloadId: string): Observable<void> {
-    return this.http.post<void>(`/Api/Torrents/RetryDownload/${downloadId}`, {});
+    return this.http.post<void>(`${this.baseHref}Api/Torrents/RetryDownload/${downloadId}`, {});
   }
 
   public update(torrent: Torrent): Observable<void> {
-    return this.http.put<void>(`/Api/Torrents/Update`, torrent);
+    return this.http.put<void>(`${this.baseHref}Api/Torrents/Update`, torrent);
   }
 }
diff --git a/client/src/index.html b/client/src/index.html
index d29bf7b..be11a8e 100644
--- a/client/src/index.html
+++ b/client/src/index.html
@@ -6,7 +6,8 @@
     <base href="/" />
     <script>
       (function () {
-        window["_app_base"] = "/" + window.location.pathname.split("/")[1];
+        var pathSegment = window.location.pathname.split("/")[1];
+        window["_app_base"] = pathSegment ? "/" + pathSegment + "/" : "/";
         console.log("setting base href to " + window["_app_base"]);
       })();
     </script>
diff --git a/server/RdtClient.Data/Models/Internal/AppSettings.cs b/server/RdtClient.Data/Models/Internal/AppSettings.cs
index 190ce2b..5212ed1 100644
--- a/server/RdtClient.Data/Models/Internal/AppSettings.cs
+++ b/server/RdtClient.Data/Models/Internal/AppSettings.cs
@@ -6,6 +6,7 @@ public class AppSettings
     public AppSettingsDatabase? Database { get; set; }
         
     public Int32 Port { get; set; }
+    public String? BasePath { get; set; }
 }
 
 public class AppSettingsLogging
diff --git a/server/RdtClient.Service/Middleware/BaseHrefMiddleware.cs b/server/RdtClient.Service/Middleware/BaseHrefMiddleware.cs
new file mode 100644
index 0000000..0b642ef
--- /dev/null
+++ b/server/RdtClient.Service/Middleware/BaseHrefMiddleware.cs
@@ -0,0 +1,54 @@
+using System.Text.RegularExpressions;
+using Microsoft.AspNetCore.Http;
+using RdtClient.Data.Models.Internal;
+
+namespace RdtClient.Service.Middleware;
+
+public class BaseHrefMiddleware
+{
+    private readonly RequestDelegate _next;
+
+    public BaseHrefMiddleware(RequestDelegate next)
+    {
+        _next = next;
+    }
+
+    public async Task InvokeAsync(HttpContext context, AppSettings appSettings)
+    {
+        var originalBody = context.Response.Body;
+
+        try
+        {
+            using var newBody = new MemoryStream();
+
+            context.Response.Body = newBody;
+
+            await _next(context);
+
+            context.Response.Body = originalBody;
+            newBody.Seek(0, SeekOrigin.Begin);
+            var responseBody = await new StreamReader(newBody).ReadToEndAsync();
+
+            // ReSharper disable once ConditionalAccessQualifierIsNonNullableAccordingToAPIContract
+            if (context.Response.ContentType?.Contains("text/html") == true)
+            {
+                var basePath = $"/{appSettings.BasePath!.TrimStart('/').TrimEnd('/')}/";
+                
+                responseBody = Regex.Replace(responseBody, @"<base href=""/""", @$"<base href=""{basePath}""");
+                
+                responseBody = Regex.Replace(responseBody, "(<script.*?src=\")(.*?)(\".*?</script>)", $"$1{basePath}$2$3");
+                responseBody = Regex.Replace(responseBody, "(<link.*?href=\")(.*?)(\".*?>)", $"$1{basePath}$2$3");
+
+                await context.Response.WriteAsync(responseBody);
+            }
+            else
+            {
+                await context.Response.BodyWriter.WriteAsync(newBody.ToArray());
+            }
+        }
+        finally
+        {
+            context.Response.Body = originalBody;
+        }
+    }
+}
\ No newline at end of file
diff --git a/server/RdtClient.Web/Program.cs b/server/RdtClient.Web/Program.cs
index 42adfea..ecf788e 100644
--- a/server/RdtClient.Web/Program.cs
+++ b/server/RdtClient.Web/Program.cs
@@ -164,6 +164,11 @@ try
         }
     });
 
+    if (!String.IsNullOrWhiteSpace(appSettings.BasePath))
+    {
+        app.UseMiddleware<BaseHrefMiddleware>();
+    }
+
     app.UseRouting();
 
     app.UseAuthentication();
diff --git a/server/RdtClient.Web/Properties/launchSettings.json b/server/RdtClient.Web/Properties/launchSettings.json
index 21e8131..50e4bf7 100644
--- a/server/RdtClient.Web/Properties/launchSettings.json
+++ b/server/RdtClient.Web/Properties/launchSettings.json
@@ -17,7 +17,7 @@
     },
     "RdtClient.Web": {
       "commandName": "Project",
-      "launchUrl": "weatherforecast",
+      "launchUrl": "",
       "environmentVariables": {
         "ASPNETCORE_ENVIRONMENT": "Development"
       },
diff --git a/server/RdtClient.Web/appsettings.json b/server/RdtClient.Web/appsettings.json
index f8088bd..95e2b20 100644
--- a/server/RdtClient.Web/appsettings.json
+++ b/server/RdtClient.Web/appsettings.json
@@ -9,5 +9,6 @@
   "Database": {
     "Path": "/data/db/rdtclient.db"
   },
-  "Port": "6500"
+  "Port": "6500",
+  "BasePath": null
 }
\ No newline at end of file