feat: support auto decompiling with cfr

web/app/components/memshell/shell-result.tsx

@@ -4,12 +4,13 @@ import { DownloadIcon } from "lucide-react";
 import { useTranslation } from "react-i18next";
 import { toast } from "sonner";
 
+import DecompiledCodeViewer from "@/components/memshell/decompiled-code-viewer";
 import { QuickUsage } from "@/components/memshell/quick-usage";
 import { Button } from "@/components/ui/button";
 import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
+import { useDecompiledSources } from "@/lib/decompile/use-decompiled-sources";
 import { downloadBytes } from "@/lib/utils";
 
-import CodeViewer from "../code-viewer";
 import { BasicInfo } from "./results/basic-info";
 import { ResultComponent } from "./results/result-component";
 
@@ -23,10 +24,24 @@ export default function ShellResult({
   generateResult?: MemShellResult;
 }>) {
   const { t } = useTranslation(["common", "memshell"]);
+  const { sources, isDecompiling, error } = useDecompiledSources(generateResult);
+
   if (!generateResult) {
     return <QuickUsage />;
   }
+
+  const shellClassName = generateResult.shellClassName;
+  const shellBytesBase64 = generateResult.shellBytesBase64Str;
+  const injectorClassName = generateResult.injectorClassName;
+  const injectorBytesBase64 = generateResult.injectorBytesBase64Str;
+
   const height = 800;
+  const sourcePlaceholder = isDecompiling
+    ? `// ${t("common:decompiling")}`
+    : error
+      ? `// ${t("common:decompileFailed", { error })}`
+      : "";
+
   return (
     <Tabs defaultValue="packResult">
       <TabsList className="grid w-full grid-cols-3">
@@ -42,61 +57,84 @@ export default function ShellResult({
           generateResult={generateResult}
         />
       </TabsContent>
-      <TabsContent value="shell" className="mt-4">
-        <CodeViewer
-          showLineNumbers={false}
-          header={<div className="truncate text-xs">{generateResult?.shellClassName}</div>}
+      <TabsContent value="shell" className="mt-4" keepMounted>
+        <DecompiledCodeViewer
+          copyLabel={t("common:copy")}
+          copyOptions={[
+            {
+              label: t("memshell:copySource"),
+              value: sources.shell?.source ?? "",
+              disabled: !sources.shell,
+            },
+            {
+              label: t("memshell:copyBase64"),
+              value: shellBytesBase64 ?? "",
+              disabled: !shellBytesBase64,
+            },
+          ]}
+          header={<div className="truncate text-xs">{shellClassName}</div>}
           button={
             <Button
               variant="ghost"
               size="icon"
               type="button"
               className="h-7 w-7 [&_svg]:h-4 [&_svg]:w-4"
+              aria-label={t("common:download")}
+              title={t("common:download")}
               onClick={() => {
-                if (!generateResult?.shellBytesBase64Str) {
+                if (!shellBytesBase64) {
                   toast.warning(t("memshell:tips.shellBytesEmpty"));
                   return;
                 }
-                downloadBytes(generateResult?.shellBytesBase64Str, generateResult?.shellClassName);
+                downloadBytes(shellBytesBase64, shellClassName);
               }}
             >
               <DownloadIcon className="h-4 w-4" />
             </Button>
           }
-          wrapLongLines={true}
           height={height}
-          code={generateResult?.shellBytesBase64Str ?? ""}
-          language="text"
+          lines={sources.shell?.lines ?? null}
+          placeholder={sourcePlaceholder}
         />
       </TabsContent>
-      <TabsContent value="injector" className="mt-4">
-        <CodeViewer
-          showLineNumbers={false}
-          wrapLongLines={true}
-          header={<div className="text-xs">{generateResult?.injectorClassName}</div>}
+      <TabsContent value="injector" className="mt-4" keepMounted>
+        <DecompiledCodeViewer
+          copyLabel={t("common:copy")}
+          copyOptions={[
+            {
+              label: t("memshell:copySource"),
+              value: sources.injector?.source ?? "",
+              disabled: !sources.injector,
+            },
+            {
+              label: t("memshell:copyBase64"),
+              value: injectorBytesBase64 ?? "",
+              disabled: !injectorBytesBase64,
+            },
+          ]}
+          header={<div className="truncate text-xs">{injectorClassName}</div>}
           button={
             <Button
               variant="ghost"
               size="icon"
               type="button"
               className="h-7 w-7 [&_svg]:h-4 [&_svg]:w-4"
+              aria-label={t("common:download")}
+              title={t("common:download")}
               onClick={() => {
-                if (!generateResult?.injectorBytesBase64Str) {
+                if (!injectorBytesBase64) {
                   toast.warning(t("memshell:tips.shellBytesEmpty"));
                   return;
                 }
-                downloadBytes(
-                  generateResult?.injectorBytesBase64Str,
-                  generateResult?.injectorClassName,
-                );
+                downloadBytes(injectorBytesBase64, injectorClassName);
               }}
             >
               <DownloadIcon className="h-4 w-4" />
             </Button>
           }
           height={height}
-          code={generateResult?.injectorBytesBase64Str ?? ""}
-          language="text"
+          lines={sources.injector?.lines ?? null}
+          placeholder={sourcePlaceholder}
         />
       </TabsContent>
     </Tabs>