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

import { DownloadIcon } from "lucide-react";
import { useTranslation } from "react-i18next";
import { toast } from "sonner";

import { QuickUsage } from "@/components/memshell/quick-usage";
import { Button } from "@/components/ui/button";
import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
import { cn, downloadBytes } from "@/lib/utils";
import type { MemShellResult } from "@/types/memshell";
import CodeViewer from "../code-viewer";
import { BasicInfo } from "./results/basic-info";
import { ResultComponent } from "./results/result-component";

export default function ShellResult({
  packResult,
  packMethod,
  generateResult,
}: Readonly<{
  packResult: string | undefined;
  packMethod: string;
  generateResult?: MemShellResult;
}>) {
  const { t } = useTranslation(["common", "memshell"]);
  const hasHelper =
    generateResult?.injectorHelperBytesBase64Str &&
    generateResult?.injectorHelperBytesBase64Str !== "";

  if (!generateResult) {
    return <QuickUsage />;
  }
  const height = 600;
  return (
    <Tabs defaultValue="packResult">
      <TabsList
        className={cn("grid w-full", hasHelper ? "grid-cols-4" : "grid-cols-3")}
      >
        <TabsTrigger value="packResult">
          {t("common:generateResult")}
        </TabsTrigger>
        <TabsTrigger value="shell">{t("memshell:shellClass")}</TabsTrigger>
        <TabsTrigger value="injector">
          {t("memshell:injectorClass")}
        </TabsTrigger>
        {hasHelper && (
          <TabsTrigger value="injectorHelper">
            {t("memshell:injectorHelperClass")}
          </TabsTrigger>
        )}
      </TabsList>
      <TabsContent value="packResult" className="space-y-2">
        <BasicInfo generateResult={generateResult} />
        <ResultComponent
          packResult={packResult}
          packMethod={packMethod}
          generateResult={generateResult}
        />
      </TabsContent>
      <TabsContent value="shell" className="mt-4">
        <CodeViewer
          showLineNumbers={false}
          header={
            <div className="text-xs truncate">
              {generateResult?.shellClassName}
            </div>
          }
          button={
            <Button
              variant="ghost"
              size="icon"
              type="button"
              className="h-7 w-7 [&_svg]:h-4 [&_svg]:w-4"
              onClick={() => {
                if (!generateResult?.shellBytesBase64Str) {
                  toast.warning(t("memshell:tips.shellBytesEmpty"));
                  return;
                }
                downloadBytes(
                  generateResult?.shellBytesBase64Str,
                  generateResult?.shellClassName,
                );
              }}
            >
              <DownloadIcon className="h-4 w-4" />
            </Button>
          }
          wrapLongLines={true}
          height={height}
          code={generateResult?.shellBytesBase64Str ?? ""}
          language="text"
        />
      </TabsContent>
      <TabsContent value="injector" className="mt-4">
        <CodeViewer
          showLineNumbers={false}
          wrapLongLines={true}
          header={
            <div className="text-xs">{generateResult?.injectorClassName}</div>
          }
          button={
            <Button
              variant="ghost"
              size="icon"
              type="button"
              className="h-7 w-7 [&_svg]:h-4 [&_svg]:w-4"
              onClick={() => {
                if (!generateResult?.injectorBytesBase64Str) {
                  toast.warning(t("memshell:tips.shellBytesEmpty"));
                  return;
                }
                downloadBytes(
                  generateResult?.injectorBytesBase64Str,
                  generateResult?.injectorClassName,
                );
              }}
            >
              <DownloadIcon className="h-4 w-4" />
            </Button>
          }
          height={height}
          code={generateResult?.injectorBytesBase64Str ?? ""}
          language="text"
        />
      </TabsContent>
      {hasHelper && (
        <TabsContent value="injectorHelper" className="mt-4">
          <CodeViewer
            showLineNumbers={false}
            wrapLongLines={true}
            header={
              <div className="text-xs">
                {generateResult?.shellClassName + "$1"}
              </div>
            }
            button={
              <Button
                variant="ghost"
                size="icon"
                type="button"
                className="h-7 w-7 [&_svg]:h-4 [&_svg]:w-4"
                onClick={() => {
                  if (!generateResult?.injectorHelperBytesBase64Str) {
                    toast.warning(t("memshell:tips.shellBytesEmpty"));
                    return;
                  }
                  downloadBytes(
                    generateResult?.injectorHelperBytesBase64Str,
                    generateResult?.shellClassName + "$1",
                  );
                }}
              >
                <DownloadIcon className="h-4 w-4" />
              </Button>
            }
            height={height}
            code={generateResult?.injectorHelperBytesBase64Str ?? ""}
            language="text"
          />
        </TabsContent>
      )}
    </Tabs>
  );
}