1 files changed, 43 insertions, 0 deletions
diff --git a/extension/react-app/src/util/api.ts b/extension/react-app/src/util/api.ts
new file mode 100644
index 00000000..bdec1d20
--- /dev/null
+++ b/extension/react-app/src/util/api.ts
@@ -0,0 +1,43 @@
+import {
+  Configuration,
+  DebugApi,
+  UnittestApi,
+  ChatApi,
+} from "../../../src/client";
+import { useSelector } from "react-redux";
+import { useEffect, useState } from "react";
+import { RootStore } from "../redux/store";
+
+export function useApi() {
+  const apiUrl = useSelector((state: RootStore) => state.config.apiUrl);
+  const vscMachineId = useSelector(
+    (state: RootStore) => state.config.vscMachineId
+  );
+  const [debugApi, setDebugApi] = useState<DebugApi>();
+  const [unittestApi, setUnittestApi] = useState<UnittestApi>();
+  const [chatApi, setChatApi] = useState<ChatApi>();
+
+  useEffect(() => {
+    if (apiUrl && vscMachineId) {
+      let config = new Configuration({
+        basePath: apiUrl,
+        fetchApi: fetch,
+        middleware: [
+          {
+            pre: async (context) => {
+              context.init.headers = {
+                ...context.init.headers,
+                "x-vsc-machine-id": vscMachineId,
+              };
+            },
+          },
+        ],
+      });
+      setDebugApi(new DebugApi(config));
+      setUnittestApi(new UnittestApi(config));
+      setChatApi(new ChatApi(config));
+    }
+  }, [apiUrl, vscMachineId]);
+
+  return { debugApi, unittestApi, chatApi };
+}