27 lines
863 B
TypeScript
27 lines
863 B
TypeScript
import { createUserWithPersonalLedger } from "./accounts.js";
|
||
import { initializeDatabase, pool } from "./db.js";
|
||
|
||
const name = process.argv[2]?.trim() ?? "";
|
||
const password = process.argv[3] ?? "";
|
||
if (!name || !password) {
|
||
throw new Error("用法：npm run user:add -- <用户名> <密码>");
|
||
}
|
||
|
||
await initializeDatabase();
|
||
const client = await pool.connect();
|
||
try {
|
||
await client.query("BEGIN");
|
||
const user = await createUserWithPersonalLedger(client, name, password);
|
||
await client.query("COMMIT");
|
||
process.stdout.write(`已添加用户 ${user.name}，并创建个人账本。\n`);
|
||
} catch (error) {
|
||
await client.query("ROLLBACK");
|
||
if (typeof error === "object" && error !== null && "code" in error && error.code === "23505") {
|
||
throw new Error("该用户名已存在");
|
||
}
|
||
throw error;
|
||
} finally {
|
||
client.release();
|
||
await pool.end();
|
||
}
|