feat: add rollback functionality to WorkerRevisionsSection with confirmation dialog and update TRPC router for revision rollback

Author: moonrailgun

src/client/components/worker/WorkerRevisionsSection.tsx

@@ -1,6 +1,6 @@
 import React, { useEffect, useMemo, useState } from 'react';
 import { useTranslation } from '@i18next-toolkit/react';
-import { trpc } from '@/api/trpc';
+import { trpc, defaultErrorHandler } from '@/api/trpc';
 import { Card, CardContent, CardHeader, CardTitle } from '@/components/ui/card';
 import { Button } from '@/components/ui/button';
 import { Badge } from '@/components/ui/badge';
@@ -15,25 +15,45 @@ import {
 import { ScrollArea } from '@/components/ui/scroll-area';
 import { Loading } from '@/components/Loading';
 import { WorkerRevisionDiffViewer } from './WorkerRevisionDiffViewer';
-import { LuArrowLeftRight, LuArrowRight } from 'react-icons/lu';
+import { AlertConfirm } from '@/components/AlertConfirm';
+import {
+  LuArrowLeftRight,
+  LuArrowRight,
+  LuUndo2,
+  LuGitCompareArrows,
+  LuDiff,
+} from 'react-icons/lu';
+import { SimpleTooltip } from '@/components/ui/tooltip';
 import dayjs from 'dayjs';
 
 interface WorkerRevisionsSectionProps {
   workspaceId: string;
   workerId: string;
+  onRollback?: () => void;
 }
 
 export const WorkerRevisionsSection: React.FC<WorkerRevisionsSectionProps> = ({
   workspaceId,
   workerId,
+  onRollback,
 }) => {
   const { t } = useTranslation();
-  const { data: revisions = [], isLoading } = trpc.worker.getRevisions.useQuery(
-    {
-      workspaceId,
-      workerId,
-    }
-  );
+  const {
+    data: revisions = [],
+    isLoading,
+    refetch: refetchRevisions,
+  } = trpc.worker.getRevisions.useQuery({
+    workspaceId,
+    workerId,
+  });
+
+  const rollbackMutation = trpc.worker.rollbackToRevision.useMutation({
+    onError: defaultErrorHandler,
+    onSuccess: () => {
+      refetchRevisions();
+      onRollback?.();
+    },
+  });
   const [baseRevisionId, setBaseRevisionId] = useState<string | null>(null);
   const [targetRevisionId, setTargetRevisionId] = useState<string | null>(null);
 
@@ -186,25 +206,55 @@ export const WorkerRevisionsSection: React.FC<WorkerRevisionsSectionProps> = ({
                                 : ''}
                             </div>
                           </div>
-                          <div className="flex items-center gap-2">
-                            <Button
-                              type="button"
-                              size="sm"
-                              variant="outline"
-                              onClick={() => setBaseRevisionId(revision.id)}
-                              className="h-7 text-xs"
-                            >
-                              {t('Set base')}
-                            </Button>
-                            <Button
-                              type="button"
-                              size="sm"
-                              variant="outline"
-                              onClick={() => setTargetRevisionId(revision.id)}
-                              className="h-7 text-xs"
-                            >
-                              {t('Set compare')}
-                            </Button>
+                          <div className="flex items-center gap-1">
+                            <SimpleTooltip content={t('Set base')}>
+                              <Button
+                                type="button"
+                                size="icon"
+                                variant={isBase ? 'default' : 'ghost'}
+                                onClick={() => setBaseRevisionId(revision.id)}
+                                className="h-7 w-7"
+                                Icon={LuGitCompareArrows}
+                              />
+                            </SimpleTooltip>
+                            <SimpleTooltip content={t('Set compare')}>
+                              <Button
+                                type="button"
+                                size="icon"
+                                variant={isTarget ? 'default' : 'ghost'}
+                                onClick={() => setTargetRevisionId(revision.id)}
+                                className="h-7 w-7"
+                                Icon={LuDiff}
+                              />
+                            </SimpleTooltip>
+                            {!isLatest && (
+                              <AlertConfirm
+                                title={t('Rollback to Revision #{{num}}', {
+                                  num: revision.revision,
+                                })}
+                                description={t(
+                                  'Are you sure you want to rollback the worker code to this revision? This will create a new revision with the rolled-back code.'
+                                )}
+                                onConfirm={async () => {
+                                  await rollbackMutation.mutateAsync({
+                                    workspaceId,
+                                    workerId,
+                                    revisionId: revision.id,
+                                  });
+                                }}
+                              >
+                                <SimpleTooltip content={t('Rollback')}>
+                                  <Button
+                                    type="button"
+                                    size="icon"
+                                    variant="ghost"
+                                    className="h-7 w-7"
+                                    Icon={LuUndo2}
+                                    loading={rollbackMutation.isPending}
+                                  />
+                                </SimpleTooltip>
+                              </AlertConfirm>
+                            )}
                           </div>
                         </div>
                       );

src/client/routes/worker/$workerId/index.tsx

@@ -391,6 +391,7 @@ function PageComponent() {
             <WorkerRevisionsSection
               workspaceId={workspaceId}
               workerId={workerId}
+              onRollback={refetch}
             />
           </TabsContent>
 

src/server/trpc/routers/worker.ts

@@ -621,6 +621,71 @@ export const workerRouter = router({
       return execution;
     }),
 
+  rollbackToRevision: workspaceAdminProcedure
+    .meta(
+      buildWorkerOpenapi({
+        method: 'POST',
+        path: '/{workerId}/rollback',
+        summary: 'Rollback worker to a specific revision',
+      })
+    )
+    .input(
+      z.object({
+        workerId: z.string().cuid2(),
+        revisionId: z.string().cuid2(),
+      })
+    )
+    .output(FunctionWorkerModelSchema)
+    .mutation(async ({ input, ctx }) => {
+      const { workerId, revisionId, workspaceId } = input;
+
+      const worker = await prisma.functionWorker.findUnique({
+        where: {
+          id: workerId,
+          workspaceId,
+        },
+      });
+
+      if (!worker) {
+        throw new Error('Worker not found');
+      }
+
+      const revision = await prisma.functionWorkerRevision.findUnique({
+        where: {
+          id: revisionId,
+          workerId,
+        },
+      });
+
+      if (!revision) {
+        throw new Error('Revision not found');
+      }
+
+      if (revision.code === worker.code) {
+        return worker;
+      }
+
+      const updatedWorker = await workerCronManager.upsert({
+        id: workerId,
+        workspaceId,
+        name: worker.name,
+        description: worker.description ?? undefined,
+        code: revision.code,
+        active: worker.active,
+        enableCron: worker.enableCron,
+        cronExpression: worker.cronExpression,
+      });
+
+      await createAuditLog({
+        workspaceId,
+        relatedId: workerId,
+        relatedType: WorkspaceAuditLogType.FunctionWorker,
+        content: `Rolled back worker: ${worker.name} to revision #${revision.revision} by ${ctx.user?.username}(${ctx.user?.id})`,
+      });
+
+      return updatedWorker;
+    }),
+
   getRevisions: workspaceProcedure
     .meta(
       buildWorkerOpenapi({