From f83499763a510f1bc6984f9e33c997babd946a7b Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Fri, 13 Jan 2023 18:40:20 -0600
Subject: [PATCH] feat(gui): get API root from query string if available

---
 gui/src/main.tsx | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/gui/src/main.tsx b/gui/src/main.tsx
index 945f3394..b1118160 100644
--- a/gui/src/main.tsx
+++ b/gui/src/main.tsx
@@ -1,5 +1,5 @@
 /* eslint-disable no-console */
-import { mustExist } from '@apextoaster/js-utils';
+import { doesExist, mustExist } from '@apextoaster/js-utils';
 import { merge } from 'lodash';
 import * as React from 'react';
 import ReactDOM from 'react-dom/client';
@@ -9,15 +9,27 @@ import { createJSONStorage, persist } from 'zustand/middleware';
 
 import { makeClient } from './api/client.js';
 import { OnnxWeb } from './components/OnnxWeb.js';
-import { loadConfig } from './config.js';
+import { Config, loadConfig } from './config.js';
 import { ClientContext, createStateSlices, OnnxState, StateContext } from './state.js';
 
+export function getApiRoot(config: Config): string {
+  const query = new URLSearchParams(window.location.search);
+  const api = query.get('api');
+
+  if (doesExist(api)) {
+    return api;
+  } else {
+    return config.api.root;
+  }
+}
+
 export async function main() {
   // load config from GUI server
   const config = await loadConfig();
 
   // use that to create an API client
-  const client = makeClient(config.api.root);
+  const root = getApiRoot(config);
+  const client = makeClient(root);
 
   // load full params from the API server and merge with the initial client config
   const params = await client.params();