feat: add function worker entry and service count

Author: moonrailgun

src/client/components/layout/DesktopLayout.tsx

@@ -9,6 +9,7 @@ import {
   LuWifi,
   LuTabletSmartphone,
   LuBrainCircuit,
+  LuSquareFunction,
 } from 'react-icons/lu';
 import {
   ResizableHandle,
@@ -32,6 +33,8 @@ import { WorkspacePauseTip } from '../workspace/WorkspacePauseTip';
 import { FreeTierTip } from '../FreeTierTip';
 import { DevContainer } from '../DevContainer';
 import { AIPanel } from '../ai/AIPanel';
+import { compact } from 'lodash-es';
+import { useGlobalConfig } from '@/hooks/useConfig';
 
 const defaultLayout: [number, number, number] = [15, 25, 60];
 
@@ -56,6 +59,7 @@ export const DesktopLayout: React.FC<LayoutProps> = React.memo((props) => {
     }
   );
   const { t } = useTranslation();
+  const { alphaMode, enableFunctionWorker } = useGlobalConfig();
 
   const navbar = (
     <>
@@ -89,7 +93,7 @@ export const DesktopLayout: React.FC<LayoutProps> = React.memo((props) => {
 
       <Nav
         isCollapsed={isCollapsed}
-        links={[
+        links={compact([
           {
             title: t('Website'),
             label: String(serviceCount?.website ?? ''),
@@ -144,7 +148,14 @@ export const DesktopLayout: React.FC<LayoutProps> = React.memo((props) => {
             icon: LuBrainCircuit,
             to: '/aiGateway',
           },
-        ]}
+          alphaMode &&
+            enableFunctionWorker && {
+              title: t('Function'),
+              label: String(serviceCount?.functionWorker ?? ''),
+              icon: LuSquareFunction,
+              to: '/worker',
+            },
+        ])}
       />
 
       <Separator />

src/client/routes/insights/index.tsx

@@ -74,22 +74,29 @@ function PageComponent() {
                     <SelectValue placeholder={t('Please select target')} />
                   </SelectTrigger>
                   <SelectContent>
-                    <SelectGroup>
-                      <SelectLabel>{t('Websites')}</SelectLabel>
-                      {websites.map((item) => (
-                        <SelectItem key={item.id} value={item.id}>
-                          {item.name}
-                        </SelectItem>
-                      ))}
-                    </SelectGroup>
-                    <SelectGroup>
-                      <SelectLabel>{t('Surveys')}</SelectLabel>
-                      {surveys.map((item) => (
-                        <SelectItem key={item.id} value={item.id}>
-                          {item.name}
-                        </SelectItem>
-                      ))}
-                    </SelectGroup>
+                    {websites.length > 0 && (
+                      <SelectGroup>
+                        <SelectLabel>{t('Websites')}</SelectLabel>
+
+                        {websites.map((item) => (
+                          <SelectItem key={item.id} value={item.id}>
+                            {item.name}
+                          </SelectItem>
+                        ))}
+                      </SelectGroup>
+                    )}
+
+                    {surveys.length > 0 && (
+                      <SelectGroup>
+                        <SelectLabel>{t('Surveys')}</SelectLabel>
+                        {surveys.map((item) => (
+                          <SelectItem key={item.id} value={item.id}>
+                            {item.name}
+                          </SelectItem>
+                        ))}
+                      </SelectGroup>
+                    )}
+
                     {warehouseApplicationIds.length > 0 && (
                       <SelectGroup>
                         <SelectLabel>{t('Warehouse')}</SelectLabel>

src/server/model/workspace.ts

@@ -84,6 +84,7 @@ export async function getWorkspaceServiceCount(workspaceId: string) {
     survey,
     feed,
     aiGateway,
+    functionWorker,
   ] = await Promise.all([
     prisma.website.count({
       where: {
@@ -125,6 +126,11 @@ export async function getWorkspaceServiceCount(workspaceId: string) {
         workspaceId,
       },
     }),
+    prisma.functionWorker.count({
+      where: {
+        workspaceId,
+      },
+    }),
   ]);
 
   return {
@@ -136,6 +142,7 @@ export async function getWorkspaceServiceCount(workspaceId: string) {
     survey,
     feed,
     aiGateway,
+    functionWorker,
   };
 }
 

src/server/trpc/routers/workspace.ts

@@ -521,6 +521,7 @@ export const workspaceRouter = router({
         survey: z.number(),
         feed: z.number(),
         aiGateway: z.number(),
+        functionWorker: z.number(),
       })
     )
     .query(async ({ input }) => {
@@ -540,6 +541,7 @@ export const workspaceRouter = router({
         survey,
         feed,
         aiGateway,
+        functionWorker,
       } = serviceCount;
 
       return {
@@ -552,6 +554,7 @@ export const workspaceRouter = router({
         survey,
         feed,
         aiGateway,
+        functionWorker,
       };
     }),
   /**