642 lines
22 KiB
JavaScript
642 lines
22 KiB
JavaScript
import React, {useEffect, useState} from 'react';
|
||
import {Input, Label, Message, Popup} from 'semantic-ui-react';
|
||
import {Link} from 'react-router-dom';
|
||
import {API, setPromptShown, shouldShowPrompt, showError, showInfo, showSuccess, timestamp2string} from '../helpers';
|
||
|
||
import {CHANNEL_OPTIONS, ITEMS_PER_PAGE} from '../constants';
|
||
import {renderGroup, renderNumber, renderQuota, renderQuotaWithPrompt} from '../helpers/render';
|
||
import {
|
||
Avatar,
|
||
Tag,
|
||
Table,
|
||
Button,
|
||
Popover,
|
||
Form,
|
||
Modal,
|
||
Popconfirm,
|
||
Space,
|
||
Tooltip,
|
||
Switch,
|
||
Typography, InputNumber
|
||
} from "@douyinfe/semi-ui";
|
||
import EditChannel from "../pages/Channel/EditChannel";
|
||
|
||
function renderTimestamp(timestamp) {
|
||
return (
|
||
<>
|
||
{timestamp2string(timestamp)}
|
||
</>
|
||
);
|
||
}
|
||
|
||
let type2label = undefined;
|
||
|
||
function renderType(type) {
|
||
if (!type2label) {
|
||
type2label = new Map;
|
||
for (let i = 0; i < CHANNEL_OPTIONS.length; i++) {
|
||
type2label[CHANNEL_OPTIONS[i].value] = CHANNEL_OPTIONS[i];
|
||
}
|
||
type2label[0] = {value: 0, text: '未知类型', color: 'grey'};
|
||
}
|
||
return <Tag size='large' color={type2label[type]?.color}>{type2label[type]?.text}</Tag>;
|
||
}
|
||
|
||
function renderBalance(type, balance) {
|
||
switch (type) {
|
||
case 1: // OpenAI
|
||
return <span>${balance.toFixed(2)}</span>;
|
||
case 4: // CloseAI
|
||
return <span>¥{balance.toFixed(2)}</span>;
|
||
case 8: // 自定义
|
||
return <span>${balance.toFixed(2)}</span>;
|
||
case 5: // OpenAI-SB
|
||
return <span>¥{(balance / 10000).toFixed(2)}</span>;
|
||
case 10: // AI Proxy
|
||
return <span>{renderNumber(balance)}</span>;
|
||
case 12: // API2GPT
|
||
return <span>¥{balance.toFixed(2)}</span>;
|
||
case 13: // AIGC2D
|
||
return <span>{renderNumber(balance)}</span>;
|
||
default:
|
||
return <span>不支持</span>;
|
||
}
|
||
}
|
||
|
||
const ChannelsTable = () => {
|
||
const columns = [
|
||
{
|
||
title: 'ID',
|
||
dataIndex: 'id',
|
||
},
|
||
{
|
||
title: '名称',
|
||
dataIndex: 'name',
|
||
},
|
||
{
|
||
title: '分组',
|
||
dataIndex: 'group',
|
||
render: (text, record, index) => {
|
||
return (
|
||
<div>
|
||
<Space spacing={2}>
|
||
{
|
||
text.split(',').map((item, index) => {
|
||
return (renderGroup(item))
|
||
})
|
||
}
|
||
</Space>
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
title: '类型',
|
||
dataIndex: 'type',
|
||
render: (text, record, index) => {
|
||
return (
|
||
<div>
|
||
{renderType(text)}
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
title: '状态',
|
||
dataIndex: 'status',
|
||
render: (text, record, index) => {
|
||
return (
|
||
<div>
|
||
{renderStatus(text)}
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
title: '响应时间',
|
||
dataIndex: 'response_time',
|
||
render: (text, record, index) => {
|
||
return (
|
||
<div>
|
||
{renderResponseTime(text)}
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
title: '已用/剩余',
|
||
dataIndex: 'expired_time',
|
||
render: (text, record, index) => {
|
||
return (
|
||
<div>
|
||
<Space spacing={1}>
|
||
<Tooltip content={'已用额度'}>
|
||
<Tag color='white' type='ghost' size='large'>{renderQuota(record.used_quota)}</Tag>
|
||
</Tooltip>
|
||
<Tooltip content={'剩余额度，点击更新'}>
|
||
<Tag color='white' type='ghost' size='large' onClick={() => {updateChannelBalance(record)}}>${record.balance}</Tag>
|
||
</Tooltip>
|
||
</Space>
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
title: '优先级',
|
||
dataIndex: 'priority',
|
||
render: (text, record, index) => {
|
||
return (
|
||
<div>
|
||
<InputNumber
|
||
style={{width: 70}}
|
||
name='name'
|
||
onChange={value => {
|
||
manageChannel(record.id, 'priority', record, value);
|
||
}}
|
||
defaultValue={record.priority}
|
||
min={0}
|
||
/>
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
title: '',
|
||
dataIndex: 'operate',
|
||
render: (text, record, index) => (
|
||
<div>
|
||
<Button theme='light' type='primary' style={{marginRight: 1}} onClick={()=>testChannel(record)}>测试</Button>
|
||
<Popconfirm
|
||
title="确定是否要删除此渠道？"
|
||
content="此修改将不可逆"
|
||
okType={'danger'}
|
||
position={'left'}
|
||
onConfirm={() => {
|
||
manageChannel(record.id, 'delete', record).then(
|
||
() => {
|
||
removeRecord(record.id);
|
||
}
|
||
)
|
||
}}
|
||
>
|
||
<Button theme='light' type='danger' style={{marginRight: 1}}>删除</Button>
|
||
</Popconfirm>
|
||
{
|
||
record.status === 1 ?
|
||
<Button theme='light' type='warning' style={{marginRight: 1}} onClick={
|
||
async () => {
|
||
manageChannel(
|
||
record.id,
|
||
'disable',
|
||
record
|
||
)
|
||
}
|
||
}>禁用</Button> :
|
||
<Button theme='light' type='secondary' style={{marginRight: 1}} onClick={
|
||
async () => {
|
||
manageChannel(
|
||
record.id,
|
||
'enable',
|
||
record
|
||
);
|
||
}
|
||
}>启用</Button>
|
||
}
|
||
<Button theme='light' type='tertiary' style={{marginRight: 1}} onClick={
|
||
() => {
|
||
setEditingChannel(record);
|
||
setShowEdit(true);
|
||
}
|
||
}>编辑</Button>
|
||
</div>
|
||
),
|
||
},
|
||
];
|
||
|
||
const [channels, setChannels] = useState([]);
|
||
const [loading, setLoading] = useState(true);
|
||
const [activePage, setActivePage] = useState(1);
|
||
const [idSort, setIdSort] = useState(false);
|
||
const [searchKeyword, setSearchKeyword] = useState('');
|
||
const [searchGroup, setSearchGroup] = useState('');
|
||
const [searching, setSearching] = useState(false);
|
||
const [updatingBalance, setUpdatingBalance] = useState(false);
|
||
const [pageSize, setPageSize] = useState(ITEMS_PER_PAGE);
|
||
const [showPrompt, setShowPrompt] = useState(shouldShowPrompt("channel-test"));
|
||
const [channelCount, setChannelCount] = useState(pageSize);
|
||
const [groupOptions, setGroupOptions] = useState([]);
|
||
const [showEdit, setShowEdit] = useState(false);
|
||
const [editingChannel, setEditingChannel] = useState({
|
||
id: undefined,
|
||
});
|
||
|
||
const removeRecord = id => {
|
||
let newDataSource = [...channels];
|
||
if (id != null) {
|
||
let idx = newDataSource.findIndex(data => data.id === id);
|
||
|
||
if (idx > -1) {
|
||
newDataSource.splice(idx, 1);
|
||
setChannels(newDataSource);
|
||
}
|
||
}
|
||
};
|
||
|
||
const setChannelFormat = (channels) => {
|
||
for (let i = 0; i < channels.length; i++) {
|
||
channels[i].key = '' + channels[i].id;
|
||
}
|
||
// data.key = '' + data.id
|
||
setChannels(channels);
|
||
if (channels.length >= pageSize) {
|
||
setChannelCount(channels.length + pageSize);
|
||
} else {
|
||
setChannelCount(channels.length);
|
||
}
|
||
}
|
||
|
||
const loadChannels = async (startIdx, pageSize, idSort) => {
|
||
setLoading(true);
|
||
const res = await API.get(`/api/channel/?p=${startIdx}&page_size=${pageSize}&id_sort=${idSort}`);
|
||
const {success, message, data} = res.data;
|
||
if (success) {
|
||
if (startIdx === 0) {
|
||
setChannelFormat(data);
|
||
} else {
|
||
let newChannels = [...channels];
|
||
newChannels.splice(startIdx * pageSize, data.length, ...data);
|
||
setChannelFormat(newChannels);
|
||
}
|
||
} else {
|
||
showError(message);
|
||
}
|
||
setLoading(false);
|
||
};
|
||
|
||
const refresh = async () => {
|
||
await loadChannels(activePage - 1, pageSize, idSort);
|
||
};
|
||
|
||
useEffect(() => {
|
||
// console.log('default effect')
|
||
const localIdSort = localStorage.getItem('id-sort') === 'true';
|
||
setIdSort(localIdSort)
|
||
loadChannels(0, pageSize, localIdSort)
|
||
.then()
|
||
.catch((reason) => {
|
||
showError(reason);
|
||
});
|
||
fetchGroups().then();
|
||
}, []);
|
||
|
||
// useEffect(() => {
|
||
// console.log('search effect')
|
||
// searchChannels()
|
||
// }, [searchGroup]);
|
||
|
||
// useEffect(() => {
|
||
// localStorage.setItem('id-sort', idSort + '');
|
||
// refresh()
|
||
// }, [idSort]);
|
||
|
||
const manageChannel = async (id, action, record, value) => {
|
||
let data = {id};
|
||
let res;
|
||
switch (action) {
|
||
case 'delete':
|
||
res = await API.delete(`/api/channel/${id}/`);
|
||
break;
|
||
case 'enable':
|
||
data.status = 1;
|
||
res = await API.put('/api/channel/', data);
|
||
break;
|
||
case 'disable':
|
||
data.status = 2;
|
||
res = await API.put('/api/channel/', data);
|
||
break;
|
||
case 'priority':
|
||
if (value === '') {
|
||
return;
|
||
}
|
||
data.priority = parseInt(value);
|
||
res = await API.put('/api/channel/', data);
|
||
break;
|
||
case 'weight':
|
||
if (value === '') {
|
||
return;
|
||
}
|
||
data.weight = parseInt(value);
|
||
if (data.weight < 0) {
|
||
data.weight = 0;
|
||
}
|
||
res = await API.put('/api/channel/', data);
|
||
break;
|
||
}
|
||
const {success, message} = res.data;
|
||
if (success) {
|
||
showSuccess('操作成功完成！');
|
||
let channel = res.data.data;
|
||
let newChannels = [...channels];
|
||
if (action === 'delete') {
|
||
|
||
} else {
|
||
record.status = channel.status;
|
||
}
|
||
setChannels(newChannels);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
};
|
||
|
||
const renderStatus = (status) => {
|
||
switch (status) {
|
||
case 1:
|
||
return <Tag size='large' color='green'>已启用</Tag>;
|
||
case 2:
|
||
return (
|
||
<Popup
|
||
trigger={<Tag size='large' color='red'>
|
||
已禁用
|
||
</Tag>}
|
||
content='本渠道被手动禁用'
|
||
basic
|
||
/>
|
||
);
|
||
case 3:
|
||
return (
|
||
<Popup
|
||
trigger={<Tag size='large' color='yellow'>
|
||
已禁用
|
||
</Tag>}
|
||
content='本渠道被程序自动禁用'
|
||
basic
|
||
/>
|
||
);
|
||
default:
|
||
return (
|
||
<Tag size='large' color='grey'>
|
||
未知状态
|
||
</Tag>
|
||
);
|
||
}
|
||
};
|
||
|
||
const renderResponseTime = (responseTime) => {
|
||
let time = responseTime / 1000;
|
||
time = time.toFixed(2) + ' 秒';
|
||
if (responseTime === 0) {
|
||
return <Tag size='large' color='grey'>未测试</Tag>;
|
||
} else if (responseTime <= 1000) {
|
||
return <Tag size='large' color='green'>{time}</Tag>;
|
||
} else if (responseTime <= 3000) {
|
||
return <Tag size='large' color='lime'>{time}</Tag>;
|
||
} else if (responseTime <= 5000) {
|
||
return <Tag size='large' color='yellow'>{time}</Tag>;
|
||
} else {
|
||
return <Tag size='large' color='red'>{time}</Tag>;
|
||
}
|
||
};
|
||
|
||
const searchChannels = async (searchKeyword, searchGroup) => {
|
||
if (searchKeyword === '' && searchGroup === '') {
|
||
// if keyword is blank, load files instead.
|
||
await loadChannels(0, pageSize, idSort);
|
||
setActivePage(1);
|
||
return;
|
||
}
|
||
setSearching(true);
|
||
const res = await API.get(`/api/channel/search?keyword=${searchKeyword}&group=${searchGroup}`);
|
||
const {success, message, data} = res.data;
|
||
if (success) {
|
||
setChannels(data);
|
||
setActivePage(1);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
setSearching(false);
|
||
};
|
||
|
||
const testChannel = async (record) => {
|
||
const res = await API.get(`/api/channel/test/${record.id}/`);
|
||
const {success, message, time} = res.data;
|
||
if (success) {
|
||
let newChannels = [...channels];
|
||
record.response_time = time * 1000;
|
||
record.test_time = Date.now() / 1000;
|
||
setChannels(newChannels);
|
||
showInfo(`通道 ${record.name} 测试成功，耗时 ${time.toFixed(2)} 秒。`);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
};
|
||
|
||
const testAllChannels = async () => {
|
||
const res = await API.get(`/api/channel/test`);
|
||
const {success, message} = res.data;
|
||
if (success) {
|
||
showInfo('已成功开始测试所有已启用通道，请刷新页面查看结果。');
|
||
} else {
|
||
showError(message);
|
||
}
|
||
};
|
||
|
||
const deleteAllDisabledChannels = async () => {
|
||
const res = await API.delete(`/api/channel/disabled`);
|
||
const {success, message, data} = res.data;
|
||
if (success) {
|
||
showSuccess(`已删除所有禁用渠道，共计 ${data} 个`);
|
||
await refresh();
|
||
} else {
|
||
showError(message);
|
||
}
|
||
};
|
||
|
||
const updateChannelBalance = async (record) => {
|
||
const res = await API.get(`/api/channel/update_balance/${record.id}/`);
|
||
const {success, message, balance} = res.data;
|
||
if (success) {
|
||
record.balance = balance;
|
||
record.balance_updated_time = Date.now() / 1000;
|
||
showInfo(`通道 ${record.name} 余额更新成功！`);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
};
|
||
|
||
const updateAllChannelsBalance = async () => {
|
||
setUpdatingBalance(true);
|
||
const res = await API.get(`/api/channel/update_balance`);
|
||
const {success, message} = res.data;
|
||
if (success) {
|
||
showInfo('已更新完毕所有已启用通道余额！');
|
||
} else {
|
||
showError(message);
|
||
}
|
||
setUpdatingBalance(false);
|
||
};
|
||
|
||
const sortChannel = (key) => {
|
||
if (channels.length === 0) return;
|
||
setLoading(true);
|
||
let sortedChannels = [...channels];
|
||
if (typeof sortedChannels[0][key] === 'string') {
|
||
sortedChannels.sort((a, b) => {
|
||
return ('' + a[key]).localeCompare(b[key]);
|
||
});
|
||
} else {
|
||
sortedChannels.sort((a, b) => {
|
||
if (a[key] === b[key]) return 0;
|
||
if (a[key] > b[key]) return -1;
|
||
if (a[key] < b[key]) return 1;
|
||
});
|
||
}
|
||
if (sortedChannels[0].id === channels[0].id) {
|
||
sortedChannels.reverse();
|
||
}
|
||
setChannels(sortedChannels);
|
||
setLoading(false);
|
||
};
|
||
|
||
let pageData = channels.slice((activePage - 1) * pageSize, activePage * pageSize);
|
||
|
||
const handlePageChange = page => {
|
||
setActivePage(page);
|
||
if (page === Math.ceil(channels.length / pageSize) + 1) {
|
||
// In this case we have to load more data and then append them.
|
||
loadChannels(page - 1, pageSize, idSort).then(r => {
|
||
});
|
||
}
|
||
};
|
||
|
||
const handlePageSizeChange = async(size) => {
|
||
setPageSize(size)
|
||
setActivePage(1)
|
||
loadChannels(0, size, idSort)
|
||
.then()
|
||
.catch((reason) => {
|
||
showError(reason);
|
||
})
|
||
};
|
||
|
||
const fetchGroups = async () => {
|
||
try {
|
||
let res = await API.get(`/api/group/`);
|
||
// add 'all' option
|
||
// res.data.data.unshift('all');
|
||
setGroupOptions(res.data.data.map((group) => ({
|
||
label: group,
|
||
value: group,
|
||
})));
|
||
} catch (error) {
|
||
showError(error.message);
|
||
}
|
||
};
|
||
|
||
const closeEdit = () => {
|
||
setShowEdit(false);
|
||
}
|
||
|
||
const handleRow = (record, index) => {
|
||
if (record.status !== 1) {
|
||
return {
|
||
style: {
|
||
background: 'var(--semi-color-disabled-border)',
|
||
},
|
||
};
|
||
} else {
|
||
return {};
|
||
}
|
||
};
|
||
|
||
return (
|
||
<>
|
||
<EditChannel refresh={refresh} visible={showEdit} handleClose={closeEdit} editingChannel={editingChannel}/>
|
||
<Form onSubmit={() => {searchChannels(searchKeyword, searchGroup)}} labelPosition='left'>
|
||
|
||
<div style={{display: 'flex'}}>
|
||
<Space>
|
||
<Form.Input
|
||
field='search'
|
||
label='关键词'
|
||
placeholder='ID，名称和密钥 ...'
|
||
value={searchKeyword}
|
||
loading={searching}
|
||
onChange={(v)=>{
|
||
setSearchKeyword(v.trim())
|
||
}}
|
||
/>
|
||
<Form.Select field="group" label='分组' optionList={groupOptions} onChange={(v) => {
|
||
setSearchGroup(v)
|
||
searchChannels(searchKeyword, v)
|
||
}}/>
|
||
</Space>
|
||
</div>
|
||
</Form>
|
||
<div style={{marginTop: 10, display: 'flex'}}>
|
||
<Space>
|
||
<Typography.Text strong>使用ID排序</Typography.Text>
|
||
<Switch checked={idSort} label='使用ID排序' uncheckedText="关" aria-label="是否用ID排序" onChange={(v) => {
|
||
localStorage.setItem('id-sort', v + '')
|
||
setIdSort(v)
|
||
loadChannels(0, pageSize, v)
|
||
.then()
|
||
.catch((reason) => {
|
||
showError(reason);
|
||
})
|
||
}}></Switch>
|
||
</Space>
|
||
</div>
|
||
|
||
<Table columns={columns} dataSource={pageData} pagination={{
|
||
currentPage: activePage,
|
||
pageSize: pageSize,
|
||
total: channelCount,
|
||
pageSizeOpts: [10, 20, 50, 100],
|
||
showSizeChanger: true,
|
||
onPageSizeChange: (size) => {
|
||
handlePageSizeChange(size).then()
|
||
},
|
||
onPageChange: handlePageChange,
|
||
}} loading={loading} onRow={handleRow}/>
|
||
<div style={{display: 'flex'}}>
|
||
<Space>
|
||
<Button theme='light' type='primary' style={{marginRight: 8}} onClick={
|
||
() => {
|
||
setEditingChannel({
|
||
id: undefined,
|
||
});
|
||
setShowEdit(true)
|
||
}
|
||
}>添加渠道</Button>
|
||
<Popconfirm
|
||
title="确定？"
|
||
okType={'warning'}
|
||
onConfirm={testAllChannels}
|
||
>
|
||
<Button theme='light' type='warning' style={{marginRight: 8}}>测试所有已启用通道</Button>
|
||
</Popconfirm>
|
||
<Popconfirm
|
||
title="确定？"
|
||
okType={'secondary'}
|
||
onConfirm={updateAllChannelsBalance}
|
||
>
|
||
<Button theme='light' type='secondary' style={{marginRight: 8}}>更新所有已启用通道余额</Button>
|
||
</Popconfirm>
|
||
<Popconfirm
|
||
title="确定是否要删除禁用通道？"
|
||
content="此修改将不可逆"
|
||
okType={'danger'}
|
||
onConfirm={deleteAllDisabledChannels}
|
||
>
|
||
<Button theme='light' type='danger' style={{marginRight: 8}}>删除禁用通道</Button>
|
||
</Popconfirm>
|
||
|
||
<Button theme='light' type='primary' style={{marginRight: 8}} onClick={refresh}>刷新</Button>
|
||
</Space>
|
||
</div>
|
||
</>
|
||
);
|
||
};
|
||
|
||
export default ChannelsTable;
|