Create a middleware for admin api using ServiceWorker (#111)

Signed-off-by: Nik Nasr <[email protected]>
restatedev · Nov 18, 2024 · 99645e3 · 99645e3
1 parent 546d2ad
commit 99645e3
Show file tree

Hide file tree

Showing 24 changed files with 3,227 additions and 89 deletions.
diff --git a/apps/web-ui/app/entry.client.tsx b/apps/web-ui/app/entry.client.tsx
@@ -1,12 +1,15 @@
 import { RemixBrowser } from '@remix-run/react';
 import { startTransition, StrictMode } from 'react';
 import { hydrateRoot } from 'react-dom/client';
+import { register } from '@restate/data-access/middleware-service-worker';
 
-startTransition(() => {
-  hydrateRoot(
-    document,
-    <StrictMode>
-      <RemixBrowser />
-    </StrictMode>
-  );
+register().then(() => {
+  startTransition(() => {
+    hydrateRoot(
+      document,
+      <StrictMode>
+        <RemixBrowser />
+      </StrictMode>
+    );
+  });
 });
diff --git a/apps/web-ui/vite.config.ts b/apps/web-ui/vite.config.ts
@@ -12,6 +12,7 @@ export default defineConfig(({ mode }) => {
   const ADMIN_BASE_URL = env['ADMIN_BASE_URL'] || '';
   const SERVER_HEADERS = {
     'Set-Cookie': `adminBaseUrl=${ADMIN_BASE_URL}; SameSite=Strict; Path=/`,
+    'Service-Worker-Allowed': '/',
   };
 
   return {
@@ -51,9 +52,9 @@ export default defineConfig(({ mode }) => {
     ],
 
     // Uncomment this if you are using workers.
-    // worker: {
-    //  plugins: [ nxViteTsPaths() ],
-    // },
+    worker: {
+      plugins: () => [nxViteTsPaths()],
+    },
     build: {
       commonjsOptions: {
         transformMixedEsModules: true,

diff --git a/libs/data-access/admin-api/project.json b/libs/data-access/admin-api/project.json
@@ -10,9 +10,12 @@
       "executor": "nx:run-commands",
       "options": {
         "commands": [
-          "redocly lint ./libs/data-access/admin-api/src/lib/api/spec.json || true",
-          "openapi-typescript ./libs/data-access/admin-api/src/lib/api/spec.json -o ./libs/data-access/admin-api/src/lib/api/index.d.ts"
-        ]
+          "./libs/data-access/admin-api/src/merge.mjs",
+          "redocly lint ./libs/data-access/admin-api/src/lib/api/output.json || true",
+          "openapi-typescript ./libs/data-access/admin-api/src/lib/api/output.json -o ./libs/data-access/admin-api/src/lib/api/index.d.ts",
+          "nx format"
+        ],
+        "parallel": false
       }
     }
   }

diff --git a/libs/data-access/admin-api/src/lib/api/hooks.ts b/libs/data-access/admin-api/src/lib/api/hooks.ts
@@ -227,3 +227,14 @@ export function useServiceOpenApi(
 
   return { ...results, queryKey: queryOptions.queryKey };
 }
+
+export function useListInvocations(
+  options?: HookQueryOptions<'/query/invocations', 'get'>
+) {
+  const baseUrl = useAdminBaseUrl();
+
+  return useQuery({
+    ...adminApi('query', '/query/invocations', 'get', { baseUrl }),
+    ...options,
+  });
+}
diff --git a/libs/data-access/admin-api/src/lib/api/index.d.ts b/libs/data-access/admin-api/src/lib/api/index.d.ts
@@ -301,6 +301,46 @@ export interface paths {
     patch?: never;
     trace?: never;
   };
+  '/query': {
+    parameters: {
+      query?: never;
+      header?: never;
+      path?: never;
+      cookie?: never;
+    };
+    get?: never;
+    put?: never;
+    /**
+     * Query journal
+     * @description Query journal
+     */
+    post: operations['query'];
+    delete?: never;
+    options?: never;
+    head?: never;
+    patch?: never;
+    trace?: never;
+  };
+  '/query/invocations': {
+    parameters: {
+      query?: never;
+      header?: never;
+      path?: never;
+      cookie?: never;
+    };
+    /**
+     * List invocations
+     * @description List invocations
+     */
+    get: operations['list_invocations'];
+    put?: never;
+    post?: never;
+    delete?: never;
+    options?: never;
+    head?: never;
+    patch?: never;
+    trace?: never;
+  };
 }
 export type webhooks = Record<string, never>;
 export interface components {
@@ -503,15 +543,14 @@ export interface components {
     ListDeploymentsResponse: {
       deployments: components['schemas']['DeploymentResponse'][];
     };
-    DeploymentResponse: {
-      id: components['schemas']['String'];
-      /**
-       * Services
-       * @description List of services exposed by this deployment.
-       */
-      services: components['schemas']['ServiceNameRevPair'][];
-    } & (
+    DeploymentResponse:
       | {
+          id: components['schemas']['String'];
+          /**
+           * Services
+           * @description List of services exposed by this deployment.
+           */
+          services: components['schemas']['ServiceNameRevPair'][];
           uri: string;
           protocol_type: components['schemas']['ProtocolType'];
           http_version: string;
@@ -525,6 +564,12 @@ export interface components {
           max_protocol_version: number;
         }
       | {
+          id: components['schemas']['String'];
+          /**
+           * Services
+           * @description List of services exposed by this deployment.
+           */
+          services: components['schemas']['ServiceNameRevPair'][];
           arn: components['schemas']['LambdaARN'];
           assume_role_arn?: string | null;
           additional_headers?: {
@@ -535,8 +580,7 @@ export interface components {
           min_protocol_version: number;
           /** Format: int32 */
           max_protocol_version: number;
-        }
-    );
+        };
     ServiceNameRevPair: {
       name: string;
       /** Format: uint32 */
@@ -648,14 +692,7 @@ export interface components {
       id: components['schemas']['String'];
       services: components['schemas']['ServiceMetadata'][];
     };
-    DetailedDeploymentResponse: {
-      id: components['schemas']['String'];
-      /**
-       * Services
-       * @description List of services exposed by this deployment.
-       */
-      services: components['schemas']['ServiceMetadata'][];
-    } & (
+    DetailedDeploymentResponse:
       | {
           uri: string;
           protocol_type: components['schemas']['ProtocolType'];
@@ -668,6 +705,12 @@ export interface components {
           min_protocol_version: number;
           /** Format: int32 */
           max_protocol_version: number;
+          id: components['schemas']['String'];
+          /**
+           * Services
+           * @description List of services exposed by this deployment.
+           */
+          services: components['schemas']['ServiceMetadata'][];
         }
       | {
           arn: components['schemas']['LambdaARN'];
@@ -680,8 +723,13 @@ export interface components {
           min_protocol_version: number;
           /** Format: int32 */
           max_protocol_version: number;
-        }
-    );
+          id?: components['schemas']['String'];
+          /**
+           * Services
+           * @description List of services exposed by this deployment.
+           */
+          services: components['schemas']['ServiceMetadata'][];
+        };
   };
   responses: never;
   parameters: never;
@@ -1793,4 +1841,144 @@ export interface operations {
       };
     };
   };
+  query: {
+    parameters: {
+      query?: never;
+      header?: never;
+      path?: never;
+      cookie?: never;
+    };
+    requestBody: {
+      content: {
+        'application/json': {
+          query: string;
+        };
+      };
+    };
+    responses: {
+      200: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': Record<string, never>;
+        };
+      };
+      400: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      403: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      404: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      409: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      500: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      503: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+    };
+  };
+  list_invocations: {
+    parameters: {
+      query?: never;
+      header?: never;
+      path?: never;
+      cookie?: never;
+    };
+    requestBody?: never;
+    responses: {
+      200: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': Record<string, never>;
+        };
+      };
+      400: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      403: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      404: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      409: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      500: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+      503: {
+        headers: {
+          [name: string]: unknown;
+        };
+        content: {
+          'application/json': components['schemas']['ErrorDescriptionResponse'];
+        };
+      };
+    };
+  };
 }