show >dice after entering blocklisted command

DiceDB · Jan 2, 2025 · 1bdc5e2 · 1bdc5e2
1 parent ee239ef
commit 1bdc5e2
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 6 deletions.
diff --git a/apps/playground-web/components/Shell/hooks/useShell.tsx b/apps/playground-web/components/Shell/hooks/useShell.tsx
@@ -2,7 +2,7 @@
 import { useState, useEffect, useRef, KeyboardEvent, ChangeEvent } from 'react';
 
 // utils
-import { handleCommand } from '@/shared/utils/shellUtils';
+import { handleCommand, handleInvalidCommand } from '@/shared/utils/shellUtils';
 import blocklistedCommands from '@/shared/utils/blocklist';
 
 export const useShell = (
@@ -29,10 +29,7 @@ export const useShell = (
     }
 
     if (blocklistedCommands.includes(commandName)) {
-      setOutput((prevOutput) => [
-        ...prevOutput,
-        `(error) ERR unknown command '${commandName}'`,
-      ]);
+      handleInvalidCommand({ command, setOutput });
     } else {
       handleCommand({ command, setOutput, onCommandExecuted }); // Execute if not blocklisted
     }

diff --git a/apps/playground-web/shared/utils/shellUtils.ts b/apps/playground-web/shared/utils/shellUtils.ts
@@ -1,9 +1,21 @@
 // src/shared/utils/shellUtils.ts
 
 import { executeShellCommandOnServer } from '@/lib/api';
-import { CommandHandler } from '@/types';
+import { CommandHandler, InvalidCommandHandler } from '@/types';
 import { handleResult } from '@/shared/utils/commonUtils';
 
+export const handleInvalidCommand = async ({
+  command, 
+  setOutput
+}: InvalidCommandHandler) => {
+  const newOutput = `dice > ${command}`;
+  setOutput((prevOutput: any) => [
+    ...prevOutput,
+    newOutput,
+    `(error) ERR unknown command '${command}'`,
+  ]);
+}
+
 export const handleCommand = async ({
   command,
   setOutput,

diff --git a/apps/playground-web/types/index.d.ts b/apps/playground-web/types/index.d.ts
@@ -5,3 +5,8 @@ export interface CommandHandler {
   setOutput: React.Dispatch<React.SetStateAction<string[]>>;
   onCommandExecuted: (commandsLeft: number, cleanupTimeLeft: number) => void;
 }
+
+export interface InvalidCommandHandler {
+  command: string;
+  setOutput: React.Dispatch<React.SetStateAction<string[]>>;
+}