107 lines
3.1 KiB
TypeScript
107 lines
3.1 KiB
TypeScript
import { program as app, Command } from '@/program.ts';
|
||
import { getConfig, writeConfig } from '@/module/get-config.ts';
|
||
import { queryLogin, queryMe, switchOrg, switchMe } from '@/query/index.ts';
|
||
import inquirer from 'inquirer';
|
||
|
||
// 导入 login 命令
|
||
|
||
// 定义login命令，支持 `-u` 和 `-p` 参数来输入用户名和密码
|
||
const loginCommand = new Command('login')
|
||
.description('Login to the application')
|
||
.option('-u, --username <username>', 'Specify username')
|
||
.option('-p, --password <password>', 'Specify password')
|
||
.action(async (options) => {
|
||
const config = getConfig();
|
||
let { username, password } = options;
|
||
|
||
// 如果没有传递参数，则通过交互式输入
|
||
if (!username || !password) {
|
||
const answers = await inquirer.prompt([
|
||
{
|
||
type: 'input',
|
||
name: 'username',
|
||
message: 'Enter your username:',
|
||
when: () => !username, // 当 username 为空时，提示用户输入
|
||
},
|
||
{
|
||
type: 'password',
|
||
name: 'password',
|
||
message: 'Enter your password:',
|
||
mask: '*', // 隐藏输入的字符
|
||
when: () => !password, // 当 password 为空时，提示用户输入
|
||
},
|
||
]);
|
||
username = answers.username || username;
|
||
password = answers.password || password;
|
||
}
|
||
if (config.token) {
|
||
const res = await queryMe();
|
||
if (res.code === 200) {
|
||
const data = res.data;
|
||
if (data.username === username) {
|
||
console.log('Already Login For', data.username);
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
const res = await queryLogin(username, password);
|
||
if (res.code === 200) {
|
||
console.log('登录成功');
|
||
const { token } = res.data;
|
||
writeConfig({ ...config, token });
|
||
} else {
|
||
console.log('登录失败', res.message || '');
|
||
}
|
||
console.log('welcome', username);
|
||
});
|
||
|
||
app.addCommand(loginCommand);
|
||
|
||
const showMe = async () => {
|
||
const me = await queryMe();
|
||
if (me.code === 200) {
|
||
console.log('Me', me.data);
|
||
} else {
|
||
console.log('Me failed', me.message);
|
||
}
|
||
};
|
||
|
||
const switchOrgCommand = new Command('switchOrg').argument('<username>', 'Switch to another organization').action(async (options) => {
|
||
// console.log('options', options);
|
||
const config = getConfig();
|
||
if (!config.token) {
|
||
console.log('Please login first');
|
||
return;
|
||
}
|
||
const res = await switchOrg(options);
|
||
if (res.code === 200) {
|
||
const token = res.data.token;
|
||
writeConfig({ ...config, token });
|
||
console.log('Switch Org Success');
|
||
showMe();
|
||
} else {
|
||
console.log('Switch Org Failed', res.message || '');
|
||
}
|
||
});
|
||
|
||
app.addCommand(switchOrgCommand);
|
||
|
||
const switchMeCommand = new Command('switchMe').action(async () => {
|
||
const config = getConfig();
|
||
if (!config.token) {
|
||
console.log('Please login first');
|
||
return;
|
||
}
|
||
const res = await switchMe();
|
||
if (res.code === 200) {
|
||
const token = res.data.token;
|
||
writeConfig({ ...config, token });
|
||
console.log('Switch Me Success');
|
||
showMe();
|
||
} else {
|
||
console.log('Switch Me Failed', res.message || '');
|
||
}
|
||
});
|
||
app.addCommand(switchMeCommand);
|