ENG-2964 GetEnvironment RTKQuery

src/golang/cmd/server/handler/get_server_environment.go → src/golang/cmd/server/handler/v2/get_server_environment.go

@@ -1,10 +1,11 @@
-package handler
+package v2
 
 import (
 	"context"
 	"net/http"
 	"os"
 
+	"github.com/aqueducthq/aqueduct/cmd/server/handler"
 	"github.com/aqueducthq/aqueduct/lib"
 )
 
@@ -16,7 +17,9 @@ type getServerEnvironmentResponse struct {
 	Version      string `json:"version"`
 }
 
-// Route: /api/environment
+// Route: /api/v2/environment
+// This file should map directly to
+// src/ui/common/src/handlers/v2/EnvironmentGet.tsx
 // Method: GET
 // Request:
 //
@@ -25,7 +28,7 @@ type getServerEnvironmentResponse struct {
 //
 // Response: Aqueduct server's environment variables.
 type GetServerEnvironmentHandler struct {
-	GetHandler
+	handler.GetHandler
 }
 
 func (*GetServerEnvironmentHandler) Name() string {

src/golang/cmd/server/routes/routes.go

@@ -24,6 +24,7 @@ const (
 	NodeDagOperatorsRoute          = "/api/v2/workflow/{workflowID}/dag/{dagID}/node/operators"
 	NodeOperatorContentRoute       = "/api/v2/workflow/{workflowID}/dag/{dagID}/node/operator/{nodeID}/content"
 	NodesResultsRoute              = "/api/v2/workflow/{workflowID}/result/{dagResultID}/nodes/results"
+	GetServerEnvironmentRoute      = "/api/v2/environment"
 
 	// V1 routes
 	GetArtifactVersionsRoute = "/api/artifact/versions"
@@ -73,6 +74,5 @@ const (
 	GetWorkflowDagResultRoute    = "/api/workflow/{workflowId}/result/{workflowDagResultId}"
 	GetWorkflowHistoryRoute      = "/api/workflow/{workflowId}/history"
 
-	GetServerVersionRoute     = "/api/version"
-	GetServerEnvironmentRoute = "/api/environment"
+	GetServerVersionRoute = "/api/version"
 )

src/golang/cmd/server/server/handlers.go

@@ -132,6 +132,7 @@ func (s *AqServer) Handlers() map[string]handler.Handler {
 			DAGResultRepo:   s.DAGResultRepo,
 			OperatorRepo:    s.OperatorRepo,
 		},
+		routes.GetServerEnvironmentRoute: &v2.GetServerEnvironmentHandler{},
 
 		// V1 Handlers
 		// (ENG-2715) Remove deprecated ones
@@ -425,7 +426,6 @@ func (s *AqServer) Handlers() map[string]handler.Handler {
 
 			IntegrationRepo: s.IntegrationRepo,
 		},
-		routes.GetServerVersionRoute:     &handler.GetServerVersionHandler{},
-		routes.GetServerEnvironmentRoute: &handler.GetServerEnvironmentHandler{},
+		routes.GetServerVersionRoute: &handler.GetServerVersionHandler{},
 	}
 }

src/ui/common/src/components/integrations/dialogs/kubernetesDialog.tsx

@@ -1,15 +1,14 @@
 import { Checkbox, FormControlLabel } from '@mui/material';
 import Box from '@mui/material/Box';
 import React from 'react';
-import { useEffect, useState } from 'react';
+import { useEffect } from 'react';
 import { useFormContext } from 'react-hook-form';
 import * as Yup from 'yup';
 
 import {
   IntegrationDialogProps,
   KubernetesConfig,
 } from '../../../utils/integrations';
-import { apiAddress } from '../../hooks/useAqueductConsts';
 import { IntegrationTextInputField } from './IntegrationTextInputField';
 
 const Placeholders: KubernetesConfig = {
@@ -18,9 +17,14 @@ const Placeholders: KubernetesConfig = {
   use_same_cluster: 'false',
 };
 
-export const KubernetesDialog: React.FC<IntegrationDialogProps> = ({
+interface KuberentesDialogProps extends IntegrationDialogProps {
+  inK8sCluster: boolean;
+}
+
+export const KubernetesDialog: React.FC<KuberentesDialogProps> = ({
   editMode = false,
   user,
+  inK8sCluster = false,
 }) => {
   const { register, setValue, getValues } = useFormContext();
   const use_same_cluster = getValues('use_same_cluster');
@@ -31,27 +35,6 @@ export const KubernetesDialog: React.FC<IntegrationDialogProps> = ({
     setValue('use_same_cluster', 'false');
   }, []);
 
-  const [inK8sCluster, setInK8sCluster] = useState(false);
-
-  // TODO: https://linear.app/aqueducthq/issue/ENG-2964/move-k8s-use-same-cluster-request-to-rtkquery
-  useEffect(() => {
-    const fetchEnvironment = async () => {
-      const environmentResponse = await fetch(`${apiAddress}/api/environment`, {
-        method: 'GET',
-        headers: {
-          'api-key': user.apiKey,
-        },
-      });
-
-      const responseBody = await environmentResponse.json();
-      setInK8sCluster(responseBody['inK8sCluster']);
-    };
-
-    if (user) {
-      fetchEnvironment().catch(console.error);
-    }
-  }, [user]);
-
   return (
     <Box sx={{ mt: 2 }}>
       {inK8sCluster && (

src/ui/common/src/components/integrations/dialogs/onDemandKubernetesDialog.tsx

@@ -9,6 +9,7 @@ import { useFormContext } from 'react-hook-form';
 import { useDispatch } from 'react-redux';
 import * as Yup from 'yup';
 
+import { useEnvironmentGetQuery } from '../../../handlers/AqueductApi';
 import { handleConnectToNewIntegration } from '../../../reducers/integration';
 import { AppDispatch } from '../../../stores/store';
 import { IntegrationDialogProps } from '../../../utils/integrations';
@@ -42,6 +43,11 @@ export const OnDemandKubernetesDialog: React.FC<IntegrationDialogProps> = ({
   loading,
   onCloseDialog,
 }) => {
+  const {
+    data: environment,
+    error,
+    isLoading,
+  } = useEnvironmentGetQuery({ apiKey: user.apiKey });
   const { register, setValue } = useFormContext();
 
   const [currentStep, setCurrentStep] = useState('INITIAL');
@@ -137,14 +143,19 @@ export const OnDemandKubernetesDialog: React.FC<IntegrationDialogProps> = ({
     );
   };
 
+  interface RegularK8sStepLayoutProps extends IntegrationDialogProps {
+    inK8sCluster?: boolean;
+  }
+
   // We're going to need to share some more info with the dialogs, as they're not all just forms that we can
   // register anymore in the case of this layout.
-  const RegularK8sStepLayout: React.FC<IntegrationDialogProps> = ({
+  const RegularK8sStepLayout: React.FC<RegularK8sStepLayoutProps> = ({
     user,
     editMode,
     onCloseDialog,
     loading,
     disabled,
+    inK8sCluster = false,
   }) => {
     const methods = useFormContext();
     const dispatch: AppDispatch = useDispatch();
@@ -170,6 +181,7 @@ export const OnDemandKubernetesDialog: React.FC<IntegrationDialogProps> = ({
           onCloseDialog={onCloseDialog}
           loading={loading}
           disabled={disabled}
+          inK8sCluster={inK8sCluster}
         />
         <DialogActionButtons
           onCloseDialog={onCloseDialog}
@@ -376,6 +388,7 @@ export const OnDemandKubernetesDialog: React.FC<IntegrationDialogProps> = ({
           loading={loading}
           onCloseDialog={onCloseDialog}
           editMode={editMode}
+          inK8sCluster={environment?.inK8sCluster}
         />
       );
     case 'ONDEMAND_K8S':

src/ui/common/src/components/layouts/menuSidebar.tsx

@@ -15,9 +15,9 @@ import { useDispatch } from 'react-redux';
 import { Link as RouterLink, useLocation } from 'react-router-dom';
 import UserProfile from 'src/utils/auth';
 
+import { useEnvironmentGetQuery } from '../../handlers/AqueductApi';
 import { AppDispatch } from '../../stores/store';
 import { getPathPrefix } from '../../utils/getPathPrefix';
-import { apiAddress } from '../hooks/useAqueductConsts';
 import {
   menuSidebar,
   menuSidebarContent,
@@ -123,22 +123,19 @@ const MenuSidebar: React.FC<{
   const [versionNumber, setVersionNumber] = useState('');
   const location = useLocation();
 
+  const { data } = useEnvironmentGetQuery({ apiKey: user.apiKey } as any, {
+    skip: !user?.apiKey,
+  });
+
   useEffect(() => {
     setCurrentPage(location.pathname);
   }, [dispatch, location.pathname]);
 
   useEffect(() => {
-    async function fetchVersionNumber() {
-      const res = await fetch(`${apiAddress}/api/version`, {
-        method: 'GET',
-        headers: { 'api-key': user.apiKey },
-      });
-      const versionNumberResponse = await res.json();
-      setVersionNumber(versionNumberResponse.version);
+    if (data) {
+      setVersionNumber(data.version);
     }
-
-    fetchVersionNumber();
-  }, [user.apiKey]);
+  }, [user.apiKey, data]);
 
   const pathPrefix = getPathPrefix();
   return (

src/ui/common/src/handlers/AqueductApi.ts

@@ -18,6 +18,10 @@ import {
   DagResultsGetRequest,
   DagResultsGetResponse,
 } from './v2/DagResultsGet';
+import {
+  environmentGetQuery,
+  EnvironmentGetRequest,
+} from './v2/EnvironmentGet';
 import {
   integrationOperatorsGetQuery,
   IntegrationOperatorsGetRequest,
@@ -134,6 +138,13 @@ export const aqueductApi = createApi({
       query: (req) => dagResultsGetQuery(req),
       transformErrorResponse,
     }),
+    environmentGet: builder.query<
+      EnvironmentGetResponse,
+      EnvironmentGetRequest
+    >({
+      query: (req) => environmentGetQuery(req),
+      transformErrorResponse,
+    }),
     integrationOperatorsGet: builder.query<
       IntegrationOperatorsGetResponse,
       IntegrationOperatorsGetRequest
@@ -246,6 +257,7 @@ export const {
   useDagOperatorsGetQuery,
   useDagResultGetQuery,
   useDagResultsGetQuery,
+  useEnvironmentGetQuery,
   useIntegrationOperatorsGetQuery,
   useIntegrationWorkflowsGetQuery,
   useIntegrationsWorkflowsGetQuery,

src/ui/common/src/handlers/v2/EnvironmentGet.ts

@@ -0,0 +1,13 @@
+import { APIKeyParameter } from '../parameters/Header';
+
+export type EnvironmentGetRequest = APIKeyParameter;
+
+export type EnvironmentGetResponse = {
+  inK8sCluster: boolean;
+  version: string;
+};
+
+export const environmentGetQuery = (req: EnvironmentGetRequest) => ({
+  url: 'environment',
+  headers: { 'api-key': req.apiKey },
+});