refactor: local server import/export

Signed-off-by: Kristóf Marussy <kristof@marussy.com>

3 files changed, 38 insertions, 4 deletions

diff --git a/src/api/apiBase.ts b/src/api/apiBase.ts
index 701919785..974d513a1 100644
--- a/src/api/apiBase.ts
+++ b/src/api/apiBase.ts
@@ -34,6 +34,23 @@ export default function apiBase(withVersion = true) {
   return fixUrl(withVersion ? `${url}/${API_VERSION}` : url);
 };
 
+export function needsToken(): boolean {
+  return (window as any).ferdium.stores.settings.all.app.server === LOCAL_SERVER;
+}
+
+export function localServerToken(): string | undefined {
+  return needsToken()
+    ? (window as any).ferdium.stores.requests.localServerToken
+    : undefined;
+}
+
+export function importExportURL() {
+  const base = apiBase(false);
+  return needsToken()
+    ? `${base}/token/${localServerToken()}`
+    : base;
+}
+
 export function serverBase() {
   const serverType = (window as any).ferdium.stores.settings.all.app.server;
   const noServer = 'You are using Ferdium without a server';
diff --git a/src/api/server/ServerApi.ts b/src/api/server/ServerApi.ts
index 77a759b3e..8b551ade2 100644
--- a/src/api/server/ServerApi.ts
+++ b/src/api/server/ServerApi.ts
@@ -25,7 +25,7 @@ import { SERVER_NOT_LOADED } from '../../config';
 import { userDataRecipesPath, userDataPath } from '../../environment-remote';
 import { asarRecipesPath } from '../../helpers/asar-helpers';
 import apiBase from '../apiBase';
-import { prepareAuthRequest, sendAuthRequest } from '../utils/auth';
+import { prepareAuthRequest, prepareLocalToken, sendAuthRequest } from '../utils/auth';
 
 import {
   getRecipeDirectory,
@@ -246,6 +246,8 @@ export default class ServerApi {
 
     delete requestData.headers['Content-Type'];
 
+    await prepareLocalToken(requestData);
+
     const request = await window.fetch(
       `${apiBase()}/service/${serviceId}`,
       // @ts-expect-error Argument of type '{ method: string; } & { mode: string; headers: any; }' is not assignable to parameter of type 'RequestInit | undefined'.
diff --git a/src/api/utils/auth.ts b/src/api/utils/auth.ts
index a7a73309d..282d00459 100644
--- a/src/api/utils/auth.ts
+++ b/src/api/utils/auth.ts
@@ -1,4 +1,6 @@
 import localStorage from 'mobx-localstorage';
+import { when } from 'mobx';
+import { localServerToken, needsToken } from '../apiBase';
 import { ferdiumLocale, ferdiumVersion } from '../../environment-remote';
 
 export const prepareAuthRequest = (
@@ -29,10 +31,23 @@ export const prepareAuthRequest = (
   return request;
 };
 
-export const sendAuthRequest = (
+export const prepareLocalToken = async (
+  requestData: { method: string; headers?: any; body?: any },
+) => {
+  await when(() => !needsToken() || !!localServerToken(), { timeout: 2000 });
+  const token = localServerToken();
+  if (token) {
+    requestData.headers['X-Ferdium-Local-Token'] = token;
+  }
+}
+
+export const sendAuthRequest = async (
   url: RequestInfo,
   options?: { method: string; headers?: any; body?: any },
   auth?: boolean,
-) =>
+) => {
+  const request = prepareAuthRequest(options, auth);
+  await prepareLocalToken(request);
   // @ts-expect-error Argument of type '{ method: string; } & { mode: string; headers: any; }' is not assignable to parameter of type 'RequestInit | undefined'.
-  window.fetch(url, prepareAuthRequest(options, auth));
+  return window.fetch(url, request);
+};