mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-18 00:16:37 +08:00
353 lines
14 KiB
JavaScript
353 lines
14 KiB
JavaScript
import React, {useEffect, useRef, useState} from 'react';
|
||
import {useParams, useNavigate} from 'react-router-dom';
|
||
import {API, isMobile, showError, showSuccess, timestamp2string} from '../../helpers';
|
||
import {renderQuota, renderQuotaWithPrompt} from '../../helpers/render';
|
||
import {
|
||
Layout,
|
||
SideSheet,
|
||
Button,
|
||
Space,
|
||
Spin,
|
||
Banner,
|
||
Input,
|
||
DatePicker,
|
||
AutoComplete,
|
||
Typography,
|
||
Checkbox, Select
|
||
} from "@douyinfe/semi-ui";
|
||
import Title from "@douyinfe/semi-ui/lib/es/typography/title";
|
||
import {Divider} from "semantic-ui-react";
|
||
|
||
const EditToken = (props) => {
|
||
const [isEdit, setIsEdit] = useState(false);
|
||
const [loading, setLoading] = useState(isEdit);
|
||
const originInputs = {
|
||
name: '',
|
||
remain_quota: isEdit ? 0 : 500000,
|
||
expired_time: -1,
|
||
unlimited_quota: false,
|
||
model_limits_enabled: false,
|
||
model_limits: [],
|
||
};
|
||
const [inputs, setInputs] = useState(originInputs);
|
||
const {name, remain_quota, expired_time, unlimited_quota, model_limits_enabled, model_limits} = inputs;
|
||
// const [visible, setVisible] = useState(false);
|
||
const [models, setModels] = useState({});
|
||
const navigate = useNavigate();
|
||
const handleInputChange = (name, value) => {
|
||
setInputs((inputs) => ({...inputs, [name]: value}));
|
||
};
|
||
const handleCancel = () => {
|
||
props.handleClose();
|
||
}
|
||
const setExpiredTime = (month, day, hour, minute) => {
|
||
let now = new Date();
|
||
let timestamp = now.getTime() / 1000;
|
||
let seconds = month * 30 * 24 * 60 * 60;
|
||
seconds += day * 24 * 60 * 60;
|
||
seconds += hour * 60 * 60;
|
||
seconds += minute * 60;
|
||
if (seconds !== 0) {
|
||
timestamp += seconds;
|
||
setInputs({...inputs, expired_time: timestamp2string(timestamp)});
|
||
} else {
|
||
setInputs({...inputs, expired_time: -1});
|
||
}
|
||
};
|
||
|
||
const setUnlimitedQuota = () => {
|
||
setInputs({...inputs, unlimited_quota: !unlimited_quota});
|
||
};
|
||
|
||
const loadModels = async () => {
|
||
let res = await API.get(`/api/user/models`);
|
||
const {success, message, data} = res.data;
|
||
if (success) {
|
||
let localModelOptions = data.map((model) => ({
|
||
label: model,
|
||
value: model
|
||
}));
|
||
setModels(localModelOptions);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
}
|
||
|
||
const loadToken = async () => {
|
||
setLoading(true);
|
||
let res = await API.get(`/api/token/${props.editingToken.id}`);
|
||
const {success, message, data} = res.data;
|
||
if (success) {
|
||
if (data.expired_time !== -1) {
|
||
data.expired_time = timestamp2string(data.expired_time);
|
||
}
|
||
if (data.model_limits !== '') {
|
||
data.model_limits = data.model_limits.split(',');
|
||
} else {
|
||
data.model_limits = [];
|
||
}
|
||
setInputs(data);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
setLoading(false);
|
||
};
|
||
useEffect(() => {
|
||
setIsEdit(props.editingToken.id !== undefined);
|
||
}, [props.editingToken.id]);
|
||
|
||
useEffect(() => {
|
||
if (!isEdit) {
|
||
setInputs(originInputs);
|
||
} else {
|
||
loadToken().then(
|
||
() => {
|
||
// console.log(inputs);
|
||
}
|
||
);
|
||
}
|
||
loadModels();
|
||
}, [isEdit]);
|
||
|
||
// 新增 state 变量 tokenCount 来记录用户想要创建的令牌数量，默认为 1
|
||
const [tokenCount, setTokenCount] = useState(1);
|
||
|
||
// 新增处理 tokenCount 变化的函数
|
||
const handleTokenCountChange = (value) => {
|
||
// 确保用户输入的是正整数
|
||
const count = parseInt(value, 10);
|
||
if (!isNaN(count) && count > 0) {
|
||
setTokenCount(count);
|
||
}
|
||
};
|
||
|
||
// 生成一个随机的四位字母数字字符串
|
||
const generateRandomSuffix = () => {
|
||
const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
|
||
let result = '';
|
||
for (let i = 0; i < 6; i++) {
|
||
result += characters.charAt(Math.floor(Math.random() * characters.length));
|
||
}
|
||
return result;
|
||
};
|
||
|
||
const submit = async () => {
|
||
setLoading(true);
|
||
if (isEdit) {
|
||
// 编辑令牌的逻辑保持不变
|
||
let localInputs = {...inputs};
|
||
localInputs.remain_quota = parseInt(localInputs.remain_quota);
|
||
if (localInputs.expired_time !== -1) {
|
||
let time = Date.parse(localInputs.expired_time);
|
||
if (isNaN(time)) {
|
||
showError('过期时间格式错误！');
|
||
setLoading(false);
|
||
return;
|
||
}
|
||
localInputs.expired_time = Math.ceil(time / 1000);
|
||
}
|
||
localInputs.model_limits = localInputs.model_limits.join(',');
|
||
let res = await API.put(`/api/token/`, {...localInputs, id: parseInt(props.editingToken.id)});
|
||
const {success, message} = res.data;
|
||
if (success) {
|
||
showSuccess('令牌更新成功！');
|
||
props.refresh();
|
||
props.handleClose();
|
||
} else {
|
||
showError(message);
|
||
}
|
||
} else {
|
||
// 处理新增多个令牌的情况
|
||
let successCount = 0; // 记录成功创建的令牌数量
|
||
for (let i = 0; i < tokenCount; i++) {
|
||
let localInputs = {...inputs};
|
||
if (i !== 0) {
|
||
// 如果用户想要创建多个令牌，则给每个令牌一个序号后缀
|
||
localInputs.name = `${inputs.name}-${generateRandomSuffix()}`;
|
||
}
|
||
localInputs.remain_quota = parseInt(localInputs.remain_quota);
|
||
|
||
if (localInputs.expired_time !== -1) {
|
||
let time = Date.parse(localInputs.expired_time);
|
||
if (isNaN(time)) {
|
||
showError('过期时间格式错误！');
|
||
setLoading(false);
|
||
break;
|
||
}
|
||
localInputs.expired_time = Math.ceil(time / 1000);
|
||
}
|
||
localInputs.model_limits = localInputs.model_limits.join(',');
|
||
let res = await API.post(`/api/token/`, localInputs);
|
||
const {success, message} = res.data;
|
||
|
||
if (success) {
|
||
successCount++;
|
||
} else {
|
||
showError(message);
|
||
break; // 如果创建失败，终止循环
|
||
}
|
||
}
|
||
|
||
if (successCount > 0) {
|
||
showSuccess(`${successCount}个令牌创建成功，请在列表页面点击复制获取令牌！`);
|
||
props.refresh();
|
||
props.handleClose();
|
||
}
|
||
}
|
||
setLoading(false);
|
||
setInputs(originInputs); // 重置表单
|
||
setTokenCount(1); // 重置数量为默认值
|
||
};
|
||
|
||
|
||
|
||
return (
|
||
<>
|
||
<SideSheet
|
||
placement={isEdit ? 'right' : 'left'}
|
||
title={<Title level={3}>{isEdit ? '更新令牌信息' : '创建新的令牌'}</Title>}
|
||
headerStyle={{borderBottom: '1px solid var(--semi-color-border)'}}
|
||
bodyStyle={{borderBottom: '1px solid var(--semi-color-border)'}}
|
||
visible={props.visiable}
|
||
footer={
|
||
<div style={{display: 'flex', justifyContent: 'flex-end'}}>
|
||
<Space>
|
||
<Button theme='solid' size={'large'} onClick={submit}>提交</Button>
|
||
<Button theme='solid' size={'large'} type={'tertiary'} onClick={handleCancel}>取消</Button>
|
||
</Space>
|
||
</div>
|
||
}
|
||
closeIcon={null}
|
||
onCancel={() => handleCancel()}
|
||
width={isMobile() ? '100%' : 600}
|
||
>
|
||
<Spin spinning={loading}>
|
||
<Input
|
||
style={{marginTop: 20}}
|
||
label='名称'
|
||
name='name'
|
||
placeholder={'请输入名称'}
|
||
onChange={(value) => handleInputChange('name', value)}
|
||
value={name}
|
||
autoComplete='new-password'
|
||
required={!isEdit}
|
||
/>
|
||
<Divider/>
|
||
<DatePicker
|
||
label='过期时间'
|
||
name='expired_time'
|
||
placeholder={'请选择过期时间'}
|
||
onChange={(value) => handleInputChange('expired_time', value)}
|
||
value={expired_time}
|
||
autoComplete='new-password'
|
||
type='dateTime'
|
||
/>
|
||
<div style={{marginTop: 20}}>
|
||
<Space>
|
||
<Button type={'tertiary'} onClick={() => {
|
||
setExpiredTime(0, 0, 0, 0);
|
||
}}>永不过期</Button>
|
||
<Button type={'tertiary'} onClick={() => {
|
||
setExpiredTime(0, 0, 1, 0);
|
||
}}>一小时</Button>
|
||
<Button type={'tertiary'} onClick={() => {
|
||
setExpiredTime(1, 0, 0, 0);
|
||
}}>一个月</Button>
|
||
<Button type={'tertiary'} onClick={() => {
|
||
setExpiredTime(0, 1, 0, 0);
|
||
}}>一天</Button>
|
||
</Space>
|
||
</div>
|
||
|
||
<Divider/>
|
||
<Banner type={'warning'}
|
||
description={'注意，令牌的额度仅用于限制令牌本身的最大额度使用量，实际的使用受到账户的剩余额度限制。'}></Banner>
|
||
<div style={{marginTop: 20}}>
|
||
<Typography.Text>{`额度${renderQuotaWithPrompt(remain_quota)}`}</Typography.Text>
|
||
</div>
|
||
<AutoComplete
|
||
style={{marginTop: 8}}
|
||
name='remain_quota'
|
||
placeholder={'请输入额度'}
|
||
onChange={(value) => handleInputChange('remain_quota', value)}
|
||
value={remain_quota}
|
||
autoComplete='new-password'
|
||
type='number'
|
||
// position={'top'}
|
||
data={[
|
||
{value: 500000, label: '1$'},
|
||
{value: 5000000, label: '10$'},
|
||
{value: 25000000, label: '50$'},
|
||
{value: 50000000, label: '100$'},
|
||
{value: 250000000, label: '500$'},
|
||
{value: 500000000, label: '1000$'},
|
||
]}
|
||
disabled={unlimited_quota}
|
||
/>
|
||
|
||
{!isEdit && (
|
||
<>
|
||
<div style={{marginTop: 20}}>
|
||
<Typography.Text>新建数量</Typography.Text>
|
||
</div>
|
||
<AutoComplete
|
||
style={{ marginTop: 8 }}
|
||
label='数量'
|
||
placeholder={'请选择或输入创建令牌的数量'}
|
||
onChange={(value) => handleTokenCountChange(value)}
|
||
onSelect={(value) => handleTokenCountChange(value)}
|
||
value={tokenCount.toString()}
|
||
autoComplete='off'
|
||
type='number'
|
||
data={[
|
||
{ value: 10, label: '10个' },
|
||
{ value: 20, label: '20个' },
|
||
{ value: 30, label: '30个' },
|
||
{ value: 100, label: '100个' },
|
||
]}
|
||
disabled={unlimited_quota}
|
||
/>
|
||
</>
|
||
)}
|
||
|
||
<div>
|
||
<Button style={{marginTop: 8}} type={'warning'} onClick={() => {
|
||
setUnlimitedQuota();
|
||
}}>{unlimited_quota ? '取消无限额度' : '设为无限额度'}</Button>
|
||
</div>
|
||
<Divider/>
|
||
<div style={{marginTop: 10, display: 'flex'}}>
|
||
<Space>
|
||
<Checkbox
|
||
name='model_limits_enabled'
|
||
checked={model_limits_enabled}
|
||
onChange={(e) => handleInputChange('model_limits_enabled', e.target.checked)}
|
||
>
|
||
</Checkbox>
|
||
<Typography.Text>启用模型限制（非必要，不建议启用）</Typography.Text>
|
||
</Space>
|
||
</div>
|
||
|
||
<Select
|
||
style={{marginTop: 8}}
|
||
placeholder={'请选择该渠道所支持的模型'}
|
||
name='models'
|
||
required
|
||
multiple
|
||
selection
|
||
onChange={value => {
|
||
handleInputChange('model_limits', value);
|
||
}}
|
||
value={inputs.model_limits}
|
||
autoComplete='new-password'
|
||
optionList={models}
|
||
disabled={!model_limits_enabled}
|
||
/>
|
||
</Spin>
|
||
</SideSheet>
|
||
</>
|
||
);
|
||
};
|
||
|
||
export default EditToken;
|