new-api/web/src/pages/Setup/index.js

import React, { useContext, useEffect, useState, useRef } from 'react';
import { Card, Col, Row, Form, Button, Typography, Space, RadioGroup, Radio, Modal, Banner } from '@douyinfe/semi-ui';
import { API, showError, showNotice, timestamp2string } from '../../helpers';
import { StatusContext } from '../../context/Status';
import { marked } from 'marked';
import { StyleContext } from '../../context/Style/index.js';
import { useTranslation } from 'react-i18next';
import { IconHelpCircle, IconInfoCircle, IconAlertTriangle } from '@douyinfe/semi-icons';

const Setup = () => {
  const { t, i18n } = useTranslation();
  const [statusState] = useContext(StatusContext);
  const [styleState, styleDispatch] = useContext(StyleContext);
  const [loading, setLoading] = useState(false);
  const [selfUseModeInfoVisible, setUsageModeInfoVisible] = useState(false);
  const [setupStatus, setSetupStatus] = useState({
    status: false,
    root_init: false,
    database_type: ''
  });
  const { Text, Title } = Typography;
  const formRef = useRef(null);
  
  const [formData, setFormData] = useState({
    username: '',
    password: '',
    confirmPassword: '',
    usageMode: 'external'
  });

  useEffect(() => {
    fetchSetupStatus();
  }, []);

  const fetchSetupStatus = async () => {
    try {
      const res = await API.get('/api/setup');
      const { success, data } = res.data;
      if (success) {
        setSetupStatus(data);
        
        // If setup is already completed, redirect to home
        if (data.status) {
          window.location.href = '/';
        }
      } else {
        showError(t('获取初始化状态失败'));
      }
    } catch (error) {
      console.error('Failed to fetch setup status:', error);
      showError(t('获取初始化状态失败'));
    }
  };

  const handleUsageModeChange = (val) => {
    setFormData({...formData, usageMode: val});
  };

  const onSubmit = () => {
    if (!formRef.current) {
      console.error("Form reference is null");
      showError(t('表单引用错误，请刷新页面重试'));
      return;
    }
    
    const values = formRef.current.getValues();
    console.log("Form values:", values);
    
    // For root_init=false, validate admin username and password
    if (!setupStatus.root_init) {
      if (!values.username || !values.username.trim()) {
        showError(t('请输入管理员用户名'));
        return;
      }
      
      if (!values.password || values.password.length < 8) {
        showError(t('密码长度至少为8个字符'));
        return;
      }
      
      if (values.password !== values.confirmPassword) {
        showError(t('两次输入的密码不一致'));
        return;
      }
    }
    
    // Prepare submission data
    const formValues = {...values};
    formValues.SelfUseModeEnabled = values.usageMode === 'self';
    formValues.DemoSiteEnabled = values.usageMode === 'demo';
    
    // Remove usageMode as it's not needed by the backend
    delete formValues.usageMode;
    
    console.log("Submitting data to backend:", formValues);
    setLoading(true);
    
    // Submit to backend
    API.post('/api/setup', formValues)
      .then(res => {
        const { success, message } = res.data;
        console.log("API response:", res.data);
        
        if (success) {
          showNotice(t('系统初始化成功，正在跳转...'));
          setTimeout(() => {
            window.location.reload();
          }, 1500);
        } else {
          showError(message || t('初始化失败，请重试'));
        }
      })
      .catch(error => {
        console.error('API error:', error);
        showError(t('系统初始化失败，请重试'));
        setLoading(false);
      })
      .finally(() => {
        // setLoading(false);
      });
  };

  return (
    <>
      <div style={{ maxWidth: '800px', margin: '0 auto', padding: '20px' }}>
        <Card>
          <Title heading={2} style={{ marginBottom: '24px' }}>{t('系统初始化')}</Title>
          
          {setupStatus.database_type === 'sqlite' && (
            <Banner
              type="warning"
              icon={<IconAlertTriangle size="large" />}
              closeIcon={null}
              title={t('数据库警告')}
              description={
                <div>
                  <p>{t('您正在使用 SQLite 数据库。如果您在容器环境中运行，请确保已正确设置数据库文件的持久化映射，否则容器重启后所有数据将丢失！')}</p>
                  <p>{t('建议在生产环境中使用 MySQL 或 PostgreSQL 数据库，或确保 SQLite 数据库文件已映射到宿主机的持久化存储。')}</p>
                </div>
              }
              style={{ marginBottom: '24px' }}
            />
          )}
          
          <Form
            getFormApi={(formApi) => { formRef.current = formApi; console.log("Form API set:", formApi); }}
            initValues={formData}
          >
            {setupStatus.root_init ? (
              <Banner
                type="info"
                icon={<IconInfoCircle />}
                closeIcon={null}
                description={t('管理员账号已经初始化过，请继续设置系统参数')}
                style={{ marginBottom: '24px' }}
              />
            ) : (
              <Form.Section text={t('管理员账号')}>
                <Form.Input
                  field="username"
                  label={t('用户名')}
                  placeholder={t('请输入管理员用户名')}
                  showClear
                  onChange={(value) => setFormData({...formData, username: value})}
                />
                <Form.Input
                  field="password"
                  label={t('密码')}
                  placeholder={t('请输入管理员密码')}
                  type="password"
                  showClear
                  onChange={(value) => setFormData({...formData, password: value})}
                />
                <Form.Input
                  field="confirmPassword"
                  label={t('确认密码')}
                  placeholder={t('请确认管理员密码')}
                  type="password"
                  showClear
                  onChange={(value) => setFormData({...formData, confirmPassword: value})}
                />
              </Form.Section>
            )}
            
            <Form.Section text={
              <div style={{ display: 'flex', alignItems: 'center' }}>
                {t('系统设置')}
              </div>
            }>
              <Form.RadioGroup 
                field="usageMode" 
                label={
                  <div style={{ display: 'flex', alignItems: 'center' }}>
                    {t('使用模式')}
                    <IconHelpCircle 
                      style={{ marginLeft: '4px', color: 'var(--semi-color-primary)', verticalAlign: 'middle', cursor: 'pointer' }} 
                      onClick={(e) => {
                        // e.preventDefault();
                        // e.stopPropagation();
                        setUsageModeInfoVisible(true);
                      }}
                    />
                  </div>
                }
                extraText={t('可在初始化后修改')}
                initValue="external"
                onChange={handleUsageModeChange}
              >
                <Form.Radio value="external">{t('对外运营模式')}</Form.Radio>
                <Form.Radio value="self">{t('自用模式')}</Form.Radio>
                <Form.Radio value="demo">{t('演示站点模式')}</Form.Radio>
              </Form.RadioGroup>
            </Form.Section>
          </Form>

          <div style={{ marginTop: '24px', textAlign: 'right' }}>
            <Button type="primary" onClick={onSubmit} loading={loading}>
              {t('初始化系统')}
            </Button>
          </div>
        </Card>
      </div>

      <Modal
        title={t('使用模式说明')}
        visible={selfUseModeInfoVisible}
        onOk={() => setUsageModeInfoVisible(false)}
        onCancel={() => setUsageModeInfoVisible(false)}
        closeOnEsc={true}
        okText={t('确定')}
        cancelText={null}
      >
        <div style={{ padding: '8px 0' }}>
          <Title heading={6}>{t('对外运营模式')}</Title>
          <p>{t('默认模式，适用于为多个用户提供服务的场景。')}</p>
          <p>{t('此模式下，系统将计算每次调用的用量，您需要对每个模型都设置价格，如果没有设置价格，用户将无法使用该模型。')}</p>
        </div>
        <div style={{ padding: '8px 0' }}>
          <Title heading={6}>{t('自用模式')}</Title>
          <p>{t('适用于个人使用的场景。不需要设置模型价格，您可专注于使用模型。')}</p>
        </div>
        <div style={{ padding: '8px 0' }}>
          <Title heading={6}>{t('演示站点模式')}</Title>
          <p>{t('适用于展示系统功能的场景。')}</p>
        </div>
      </Modal>
    </>
  );
};

export default Setup;