feat: support fumadocs

2026-09-24 07:51:52 +08:00 · 2025-12-05 11:07:24 +08:00
parent 38ad19c7ea
commit 69910ec187
140 changed files with 4788 additions and 1438 deletions
@@ -0,0 +1,166 @@
+import { useQuery } from "@tanstack/react-query";
+import { HomeLayout } from "fumadocs-ui/layouts/home";
+import { LoaderCircle, WandSparklesIcon } from "lucide-react";
+import { useState, useTransition } from "react";
+import { useForm } from "react-hook-form";
+import { useTranslation } from "react-i18next";
+import { toast } from "sonner";
+import MainConfigCard from "@/components/memshell/main-config-card";
+import PackageConfigCard from "@/components/memshell/package-config-card";
+import ShellResult from "@/components/memshell/shell-result";
+import { Button } from "@/components/ui/button";
+import { Form } from "@/components/ui/form";
+import { env } from "@/config";
+import { siteConfig } from "@/lib/config";
+import {
+  type APIErrorResponse,
+  type MainConfig,
+  type MemShellGenerateResponse,
+  type MemShellResult,
+  type PackerConfig,
+  type ServerConfig,
+  ShellToolType,
+} from "@/types/memshell";
+import {
+  type MemShellFormSchema,
+  memShellFormSchema,
+  useYupValidationResolver,
+} from "@/types/schema";
+import { transformToPostData } from "@/utils/transformer";
+import { baseOptions } from "../lib/layout.shared";
+
+export default function MemShellPage() {
+  const { data: serverConfig } = useQuery<ServerConfig>({
+    queryKey: ["serverConfig"],
+    queryFn: async () => {
+      const response = await fetch(`${env.API_URL}/api/config/servers`);
+      return await response.json();
+    },
+  });
+
+  const { data: mainConfig } = useQuery<MainConfig>({
+    queryKey: ["mainConfig"],
+    queryFn: async () => {
+      const response = await fetch(`${env.API_URL}/api/config`);
+      return await response.json();
+    },
+  });
+
+  const { data: packerConfig } = useQuery<PackerConfig>({
+    queryKey: ["packerConfig"],
+    queryFn: async () => {
+      const response = await fetch(`${env.API_URL}/api/config/packers`);
+      return await response.json();
+    },
+  });
+
+  const { t } = useTranslation(["common", "memshell"]);
+  const form = useForm({
+    resolver: useYupValidationResolver(memShellFormSchema, t),
+    defaultValues: {
+      server: "Tomcat",
+      serverVersion: "unknown",
+      targetJdkVersion: "50",
+      debug: false,
+      byPassJavaModule: false,
+      shellClassName: "",
+      shellTool: ShellToolType.Godzilla,
+      shellType: "Listener",
+      urlPattern: "/*",
+      godzillaPass: "",
+      godzillaKey: "",
+      commandParamName: "",
+      behinderPass: "",
+      antSwordPass: "",
+      headerName: "User-Agent",
+      headerValue: "",
+      injectorClassName: "",
+      packingMethod: "",
+      shrink: true,
+      staticInitialize: true,
+      shellClassBase64: "",
+    },
+  });
+
+  const [packResult, setPackResult] = useState<string | undefined>();
+  const [allPackResults, setAllPackResults] = useState<
+    Map<string, string> | undefined
+  >();
+  const [generateResult, setGenerateResult] = useState<MemShellResult>();
+  const [packMethod, setPackMethod] = useState<string>("");
+  const [isActionPending, startTransition] = useTransition();
+
+  const onSubmit = async (data: MemShellFormSchema) => {
+    startTransition(async () => {
+      try {
+        const postData = transformToPostData(data);
+        const response = await fetch(`${env.API_URL}/api/memshell/generate`, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+          },
+          body: JSON.stringify(postData),
+        });
+
+        if (!response.ok) {
+          const json: APIErrorResponse = await response.json();
+          toast.error(t("toast.generateError", { error: json.error }));
+          return;
+        }
+
+        const result = (await response.json()) as MemShellGenerateResponse;
+        setGenerateResult(result.memShellResult);
+        setPackResult(result.packResult);
+        setAllPackResults(result.allPackResults);
+        setPackMethod(data.packingMethod);
+        toast.success(t("toast.generateSuccess"));
+      } catch (error) {
+        toast.error(
+          t("toast.generateError", { error: (error as Error).message }),
+        );
+      }
+    });
+  };
+
+  return (
+    <HomeLayout {...baseOptions()} links={siteConfig.navLinks}>
+      <div className="container mx-auto max-w-7xl p-4">
+        <Form {...form}>
+          <form
+            onSubmit={form.handleSubmit(onSubmit)}
+            className="flex flex-col xl:flex-row gap-6"
+          >
+            <div className="w-full xl:w-1/2 flex flex-col gap-2">
+              <MainConfigCard
+                servers={serverConfig}
+                mainConfig={mainConfig}
+                form={form}
+              />
+              <PackageConfigCard packerConfig={packerConfig} form={form} />
+              <Button
+                className="w-full"
+                type="submit"
+                disabled={isActionPending}
+              >
+                {isActionPending ? (
+                  <LoaderCircle className="animate-spin" />
+                ) : (
+                  <WandSparklesIcon />
+                )}
+                {t("memshell:buttons.generate")}
+              </Button>
+            </div>
+            <div className="w-full xl:w-1/2 flex flex-col gap-4">
+              <ShellResult
+                packMethod={packMethod}
+                generateResult={generateResult}
+                packResult={packResult}
+                allPackResults={allPackResults}
+              />
+            </div>
+          </form>
+        </Form>
+      </div>
+    </HomeLayout>
+  );
+}