Refactor build info

BrunoQuaresma · BrunoQuaresma · commit 7560d9929400 · 2023-09-11T17:52:51.000Z
diff --git a/site/src/api/queries/buildInfo.ts b/site/src/api/queries/buildInfo.ts
@@ -0,0 +1,24 @@
+import { useQuery } from "@tanstack/react-query";
+import * as API from "api/api";
+
+export const useBuildInfo = () => {
+  return useQuery({
+    queryKey: ["buildInfo"],
+    queryFn: fetchBuildInfo,
+  });
+};
+
+const fetchBuildInfo = async () => {
+  // Build info is injected by the Coder server into the HTML document.
+  const buildInfo = document.querySelector("meta[property=build-info]");
+  if (buildInfo) {
+    const rawContent = buildInfo.getAttribute("content");
+    try {
+      return JSON.parse(rawContent as string);
+    } catch (e) {
+      console.warn("Failed to parse build info from document", e);
+    }
+  }
+
+  return API.getBuildInfo();
+};
diff --git a/site/src/components/Dashboard/DashboardProvider.tsx b/site/src/components/Dashboard/DashboardProvider.tsx
@@ -1,4 +1,5 @@
 import { useMachine } from "@xstate/react";
+import { useBuildInfo } from "api/queries/buildInfo";
 import {
   AppearanceConfig,
   BuildInfoResponse,
@@ -8,7 +9,6 @@ import {
 import { FullScreenLoader } from "components/Loader/FullScreenLoader";
 import { createContext, FC, PropsWithChildren, useContext } from "react";
 import { appearanceMachine } from "xServices/appearance/appearanceXService";
-import { buildInfoMachine } from "xServices/buildInfo/buildInfoXService";
 import { entitlementsMachine } from "xServices/entitlements/entitlementsXService";
 import { experimentsMachine } from "xServices/experiments/experimentsMachine";
 
@@ -31,15 +31,15 @@ export const DashboardProviderContext = createContext<
 >(undefined);
 
 export const DashboardProvider: FC<PropsWithChildren> = ({ children }) => {
-  const [buildInfoState] = useMachine(buildInfoMachine);
+  const buildInfoQuery = useBuildInfo();
   const [entitlementsState] = useMachine(entitlementsMachine);
   const [appearanceState, appearanceSend] = useMachine(appearanceMachine);
   const [experimentsState] = useMachine(experimentsMachine);
-  const { buildInfo } = buildInfoState.context;
   const { entitlements } = entitlementsState.context;
   const { appearance, preview } = appearanceState.context;
   const { experiments } = experimentsState.context;
-  const isLoading = !buildInfo || !entitlements || !appearance || !experiments;
+  const isLoading =
+    !buildInfoQuery.data || !entitlements || !appearance || !experiments;
 
   const setAppearancePreview = (config: AppearanceConfig) => {
     appearanceSend({
@@ -62,7 +62,7 @@ export const DashboardProvider: FC<PropsWithChildren> = ({ children }) => {
   return (
     <DashboardProviderContext.Provider
       value={{
-        buildInfo,
+        buildInfo: buildInfoQuery.data,
         entitlements,
         experiments,
         appearance: {
diff --git a/site/src/xServices/buildInfo/buildInfoXService.ts b/site/src/xServices/buildInfo/buildInfoXService.ts
