echoes/frontend/app/init.tsx

import React, { createContext, useState } from "react";
import { DEFAULT_CONFIG } from "app/env";
import { HttpClient } from "core/http";
import { ThemeModeToggle } from "hooks/themeMode";
import {
  Theme,
  Button,
  Select,
  Flex,
  Container,
  Heading,
  Text,
  Box,
  TextField,
} from "@radix-ui/themes";
import { toast } from "hooks/notification";
import { Echoes } from "hooks/echoes";

interface SetupContextType {
  currentStep: number;
  setCurrentStep: (step: number) => void;
}

const SetupContext = createContext<SetupContextType>({
  currentStep: 1,
  setCurrentStep: () => {},
});

// 步骤组件的通用属性接口
interface StepProps {
  onNext: () => void;
}

const StepContainer: React.FC<{ title: string; children: React.ReactNode }> = ({
  title,
  children,
}) => (
  <Box style={{ width: "90%", maxWidth: "600px", margin: "0 auto" }}>
    <Heading size="5" mb="4" weight="bold">
      {title}
    </Heading>
    <Flex direction="column" gap="4">
      {children}
    </Flex>
  </Box>
);

// 通用的导航按钮组件
const NavigationButtons: React.FC<
  StepProps & { loading?: boolean; disabled?: boolean }
> = ({ onNext, loading = false, disabled = false }) => (
  <Flex justify="end" mt="4">
    <Button
      size="3"
      disabled={loading || disabled}
      onClick={onNext}
      style={{ width: "100%" }}
    >
      {loading ? "处理中..." : "下一步"}
    </Button>
  </Flex>
);

// 修改输入框组件
const InputField: React.FC<{
  label: string;
  name: string;
  defaultValue?: string | number;
  hint?: string;
  required?: boolean;
}> = ({ label, name, defaultValue, hint, required = true }) => (
  <Box mb="4">
    <Text as="label" size="2" weight="medium" className="block mb-2">
      {label} {required && <Text color="red">*</Text>}
    </Text>
    <TextField.Root
      name={name}
      defaultValue={defaultValue?.toString()}
      required={required}
    >
      <TextField.Slot></TextField.Slot>
    </TextField.Root>
    {hint && (
      <Text color="gray" size="1" mt="1">
        {hint}
      </Text>
    )}
  </Box>
);

const Introduction: React.FC<StepProps> = ({ onNext }) => (
  <StepContainer title="安装说明">
    <Text size="3" style={{ lineHeight: 1.6 }}>
      欢迎使用 Echoes
    </Text>
    <NavigationButtons onNext={onNext} />
  </StepContainer>
);

const DatabaseConfig: React.FC<StepProps> = ({ onNext }) => {
  const [dbType, setDbType] = useState("postgresql");
  const [loading, setLoading] = useState(false);
  const http = HttpClient.getInstance();

  const validateForm = () => {
    const getRequiredFields = () => {
      switch (dbType) {
        case "sqllite":
          return ["db_prefix", "db_name"];
        case "postgresql":
        case "mysql":
          return [
            "db_host",
            "db_prefix",
            "db_port",
            "db_user",
            "db_password",
            "db_name",
          ];
        default:
          return [];
      }
    };

    const requiredFields = getRequiredFields();
    const emptyFields: string[] = [];

    requiredFields.forEach((field) => {
      const input = document.querySelector(
        `[name="${field}"]`,
      ) as HTMLInputElement;
      if (input && (!input.value || input.value.trim() === "")) {
        emptyFields.push(field);
      }
    });

    if (emptyFields.length > 0) {
      const fieldNames = emptyFields.map((field) => {
        switch (field) {
          case "db_host":
            return "数据库地址";
          case "db_prefix":
            return "数据库前缀";
          case "db_port":
            return "端口";
          case "db_user":
            return "用户名";
          case "db_password":
            return "密码";
          case "db_name":
            return "数据库名";
          default:
            return field;
        }
      });
      toast.error(`请填写以下必填项：${fieldNames.join("、")}`);
      return false;
    }
    return true;
  };

  const handleNext = async () => {
    const validation = validateForm();
    if (validation !== true) {
      return;
    }

    setLoading(true);
    try {
      const formData = {
        db_type: dbType,
        host:
          (
            document.querySelector('[name="db_host"]') as HTMLInputElement
          )?.value?.trim() ?? "",
        db_prefix:
          (
            document.querySelector('[name="db_prefix"]') as HTMLInputElement
          )?.value?.trim() ?? "",
        port: Number(
          (
            document.querySelector('[name="db_port"]') as HTMLInputElement
          )?.value?.trim() ?? 0,
        ),
        user:
          (
            document.querySelector('[name="db_user"]') as HTMLInputElement
          )?.value?.trim() ?? "",
        password:
          (
            document.querySelector('[name="db_password"]') as HTMLInputElement
          )?.value?.trim() ?? "",
        db_name:
          (
            document.querySelector('[name="db_name"]') as HTMLInputElement
          )?.value?.trim() ?? "",
      };

      await http.post("/sql", formData);

      let oldEnv = import.meta.env ?? DEFAULT_CONFIG;
      const viteEnv = Object.entries(oldEnv).reduce(
        (acc, [key, value]) => {
          if (key.startsWith("VITE_")) {
            acc[key] = value;
          }
          return acc;
        },
        {} as Record<string, any>,
      );

      const newEnv = {
        ...viteEnv,
        VITE_INIT_STATUS: "2",
      };

      await http.dev("/env", {
        method: "POST",
        body: JSON.stringify(newEnv),
      });

      Object.assign(import.meta.env, newEnv);

      toast.success("数据库配置成功！");

      setTimeout(() => onNext(), 1000);
    } catch (error: any) {
      console.error(error);
      toast.error(error.message, error.title);
    } finally {
      setLoading(false);
    }
  };

  return (
    <StepContainer title="数据库配置">
      <div>
        <Box mb="6">
          <Text as="label" size="2" weight="medium" mb="2" className="block">
            数据库类型
          </Text>
          <Select.Root value={dbType} onValueChange={setDbType}>
            <Select.Trigger />
            <Select.Content>
              <Select.Group>
                <Select.Item value="postgresql">PostgreSQL</Select.Item>
                <Select.Item value="mysql">MySQL</Select.Item>
                <Select.Item value="sqllite">SQLite</Select.Item>
              </Select.Group>
            </Select.Content>
          </Select.Root>
        </Box>

        {dbType === "postgresql" && (
          <>
            <InputField
              label="数据库地址"
              name="db_host"
              defaultValue="localhost"
              hint="通常使 localhost"
              required
            />
            <InputField
              label="数据库前缀"
              name="db_prefix"
              defaultValue="echoec_"
              hint="通常使用 echoec_"
              required
            />
            <InputField
              label="端口"
              name="db_port"
              defaultValue={5432}
              hint="PostgreSQL 默认端口为 5432"
              required
            />
            <InputField
              label="用户名"
              name="db_user"
              defaultValue="postgres"
              required
            />
            <InputField
              label="密码"
              name="db_password"
              defaultValue="postgres"
              required
            />
            <InputField
              label="数据库名"
              name="db_name"
              defaultValue="echoes"
              required
            />
          </>
        )}
        {dbType === "mysql" && (
          <>
            <InputField
              label="数据库地址"
              name="db_host"
              defaultValue="localhost"
              hint="通常使用 localhost"
              required
            />
            <InputField
              label="数据库前缀"
              name="db_prefix"
              defaultValue="echoec_"
              hint="通常使用 echoec_"
              required
            />
            <InputField
              label="端口"
              name="db_port"
              defaultValue={3306}
              hint="mysql 默认端口为 3306"
              required
            />
            <InputField
              label="用户名"
              name="db_user"
              defaultValue="root"
              required
            />
            <InputField
              label="密码"
              name="db_password"
              defaultValue="mysql"
              required
            />
            <InputField
              label="数据库名"
              name="db_name"
              defaultValue="echoes"
              required
            />
          </>
        )}
        {dbType === "sqllite" && (
          <>
            <InputField
              label="数据库前缀"
              name="db_prefix"
              defaultValue="echoec_"
              hint="通常使用 echoec_"
              required
            />
            <InputField
              label="数据库名"
              name="db_name"
              defaultValue="echoes.db"
              required
            />
          </>
        )}
        <NavigationButtons
          onNext={handleNext}
          loading={loading}
          disabled={loading}
        />
      </div>
    </StepContainer>
  );
};

interface InstallReplyData {
  token: string;
  username: string;
  password: string;
}

const AdminConfig: React.FC<StepProps> = ({ onNext }) => {
  const [loading, setLoading] = useState(false);
  const http = HttpClient.getInstance();

  const handleNext = async () => {
    setLoading(true);
    try {
      const formData = {
        username: (
          document.querySelector('[name="admin_username"]') as HTMLInputElement
        )?.value,
        password: (
          document.querySelector('[name="admin_password"]') as HTMLInputElement
        )?.value,
        email: (
          document.querySelector('[name="admin_email"]') as HTMLInputElement
        )?.value,
      };

      const response = (await http.post(
        "/administrator",
        formData,
      )) as InstallReplyData;
      const data = response;

      localStorage.setItem("token", data.token);

      let oldEnv = import.meta.env ?? DEFAULT_CONFIG;
      const viteEnv = Object.entries(oldEnv).reduce(
        (acc, [key, value]) => {
          if (key.startsWith("VITE_")) {
            acc[key] = value;
          }
          return acc;
        },
        {} as Record<string, any>,
      );

      const newEnv = {
        ...viteEnv,
        VITE_INIT_STATUS: "3",
        VITE_API_USERNAME: data.username,
        VITE_API_PASSWORD: data.password,
      };

      await http.dev("/env", {
        method: "POST",
        body: JSON.stringify(newEnv),
      });

      Object.assign(import.meta.env, newEnv);

      toast.success("管理员账号创建成功！");
      onNext();
    } catch (error: any) {
      console.error(error);
      toast.error(error.message, error.title);
    } finally {
      setLoading(false);
    }
  };

  return (
    <StepContainer title="创建管理员账号">
      <div className="space-y-6">
        <InputField label="用户名" name="admin_username" />
        <InputField label="密码" name="admin_password" />
        <InputField label="邮箱" name="admin_email" />
        <NavigationButtons onNext={handleNext} loading={loading} />
      </div>
    </StepContainer>
  );
};

const SetupComplete: React.FC = () => (
  <StepContainer title="安装完成">
    <Flex direction="column" align="center" gap="4">
      <Text size="5" weight="medium">
        恭喜！安装已完成
      </Text>
      <Text size="3">系统正在重启中，请稍候...</Text>
      <Box mt="4">
        <Flex justify="center">
          <Box className="animate-spin rounded-full h-8 w-8 border-b-2 border-current"></Box>
        </Flex>
      </Box>
    </Flex>
  </StepContainer>
);

export default function SetupPage() {
  const [currentStep, setCurrentStep] = useState(() => {
    return Number(import.meta.env.VITE_INIT_STATUS ?? 0) + 1;
  });

  return (
    <Theme
      grayColor="gray"
      accentColor="gray"
      radius="medium"
      panelBackground="solid"
      appearance="inherit"
    >
      <Box className="min-h-screen w-full">
        <Box position="fixed" top="2" right="4">
          <ThemeModeToggle />
        </Box>

        <Flex justify="center" pt="2">
          <Box className="w-20 h-20">
            <Echoes />
          </Box>
        </Flex>

        <Flex direction="column" className="min-h-screen w-full pb-4">
          <Container className="w-full">
            <SetupContext.Provider value={{ currentStep, setCurrentStep }}>
              {currentStep === 1 && (
                <Introduction onNext={() => setCurrentStep(currentStep + 1)} />
              )}
              {currentStep === 2 && (
                <DatabaseConfig
                  onNext={() => setCurrentStep(currentStep + 1)}
                />
              )}
              {currentStep === 3 && (
                <AdminConfig onNext={() => setCurrentStep(currentStep + 1)} />
              )}
              {currentStep === 4 && <SetupComplete />}
            </SetupContext.Provider>
          </Container>
        </Flex>
      </Box>
    </Theme>
  );
}