feat: improve account deletion and Agent authorization review

This commit is contained in:
陈煜 committed 2026-10-04 11:06:15 +08:00
1 parent 5ad64460f7
commit 91365ee315
21 files changed
+2276 -713

No files matched your search

+5 -4
View File
@@ -15,7 +15,7 @@ import { Response } from 'express';
import { z } from 'zod';
import { Database } from '../database';
import { AuthService, UserRequest } from '../auth';
import { AgentOAuth, urls, scopeInput } from './oauth';
import { AgentOAuth, urls, scopeInput, oauthDays } from './oauth';
import { AgentOperations } from './operations';
import { protocolTools, tokenDays } from './information';
@Controller('api/agent')
@@ -34,6 +34,7 @@ export class AgentManagementController {
name: true,
scopes: true,
expiresAt: true,
refreshExpiresAt: true,
createdAt: true,
revokedAt: true,
clientId: true,
@@ -110,11 +111,11 @@ export class AgentManagementController {
@Param('id') id: string,
@Body() raw: unknown,
) {
const { approve, scopes } = z
.object({ approve: z.boolean(), scopes: scopeInput.optional() })
const { approve, scopes, days } = z
.object({ approve: z.boolean(), scopes: scopeInput.optional(), days: oauthDays.default(30) })
.strict()
.parse(raw);
return this.oauth.consent(r.userId, z.string().uuid().parse(id), approve, scopes);
return this.oauth.consent(r.userId, z.string().uuid().parse(id), approve, scopes, days);
}
@Get('operations/:id') preview(@Req() r: UserRequest, @Param('id') id: string) {
return this.operations.preview(r.userId, z.string().uuid().parse(id));
+57 -9
View File
@@ -33,6 +33,14 @@ export const scopeInput = z
(!v.includes('hidden_write') ||
(v.includes('hidden_read') && (v.includes('draft') || v.includes('write')))),
);
export const oauthDays = z.union([
z.literal(1),
z.literal(3),
z.literal(7),
z.literal(30),
z.literal(365),
z.literal(null),
]);
export const digest = (s: string) => createHash('sha256').update(s).digest('hex');
const secret = () => randomBytes(32).toString('base64url');
export function urls() {
@@ -63,6 +71,9 @@ export function urls() {
}
export function webLink(key: string, id: string) {
const u = new URL(urls().web);
u.pathname =
u.pathname.replace(/\/$/, '') +
(key === 'agent_authorization' ? '/agent/authorize' : '/agent/operation');
u.searchParams.set(key, id);
return u.toString();
}
@@ -156,11 +167,18 @@ export class AgentOAuth implements OAuthServerProvider {
resource: p.resource,
};
}
async consent(userId: string, id: string, approved: boolean, selected?: string[]) {
async consent(
userId: string,
id: string,
approved: boolean,
selected?: string[],
days: number | null = 30,
) {
return this.db.atomic(async () => {
await this.pending(id);
const row = await this.db.agentAuthorization.findUniqueOrThrow({ where: { id } });
const parameters = row.parameters as any;
const authorizationDays = oauthDays.parse(days);
const allowed = selected || ['read'];
scopeInput.parse(allowed);
if (allowed.some((scope: string) => !parameters.scopes.includes(scope)))
@@ -172,7 +190,7 @@ export class AgentOAuth implements OAuthServerProvider {
userId,
status: approved ? 'approved' : 'denied',
codeDigest: approved ? digest(code) : null,
parameters: { ...parameters, scopes: allowed },
parameters: { ...parameters, scopes: allowed, authorizationDays },
},
});
if (!changed.count) throw new BadRequestException('授权请求已处理');
@@ -202,18 +220,23 @@ export class AgentOAuth implements OAuthServerProvider {
selected: string[],
days: number | null,
clientId?: string,
authorizationDays: number | null = 30,
) {
if (clientId && days === null) throw new BadRequestException('OAuth 连接必须有期限');
const access = secret(),
refresh = clientId ? secret() : undefined,
sessionId = digest(secret());
const expiresAt = days === null ? null : new Date(Date.now() + days * 86400000),
refreshExpiresAt = clientId ? new Date(Date.now() + 30 * 86400000) : null;
const expiresAt = days === null ? null : new Date(Date.now() + days * 86400000);
const lifetime = oauthDays.parse(authorizationDays);
const refreshExpiresAt =
clientId && lifetime !== null ? new Date(Date.now() + lifetime * 86400000) : null;
await this.db.session.create({
data: {
id: sessionId,
userId,
expiresAt: refreshExpiresAt || expiresAt || new Date(Date.now() + 86400000),
expiresAt: clientId
? refreshExpiresAt || new Date(Date.now() + 30 * 86400000)
: expiresAt || new Date(Date.now() + 86400000),
},
});
const grant = await this.db.agentGrant.create({
@@ -274,6 +297,7 @@ export class AgentOAuth implements OAuthServerProvider {
p.scopes,
1 / 24,
client.client_id,
p.authorizationDays === undefined ? 30 : p.authorizationDays,
)
).tokens;
});
@@ -291,8 +315,7 @@ export class AgentOAuth implements OAuthServerProvider {
!row ||
row.clientId !== client.client_id ||
row.revokedAt ||
!row.refreshExpiresAt ||
row.refreshExpiresAt <= new Date()
(row.refreshExpiresAt && row.refreshExpiresAt <= new Date())
)
throw new InvalidGrantError('Invalid refresh token');
const current = row.scopes as string[];
@@ -303,21 +326,36 @@ export class AgentOAuth implements OAuthServerProvider {
throw new InvalidScopeError('Scope escalation rejected');
const access = secret(),
refresh = secret();
const accessExpiresAt = new Date(
Math.min(Date.now() + 3600000, row.refreshExpiresAt ? +row.refreshExpiresAt : Infinity),
);
const changed = await this.db.agentGrant.updateMany({
where: { id: row.id, refreshDigest: digest(token), revokedAt: null },
data: {
accessDigest: digest(access),
refreshDigest: digest(refresh),
expiresAt: new Date(Date.now() + 3600000),
expiresAt: accessExpiresAt,
scopes: selected || current,
},
});
if (!changed.count) throw new InvalidGrantError('Refresh token already used');
// Permanent OAuth keeps a bounded business session, renewed only after a valid refresh.
const sessionExpiresAt = row.refreshExpiresAt || new Date(Date.now() + 30 * 86400000);
await this.db.session.upsert({
where: { id: row.sessionId },
create: { id: row.sessionId, userId: row.userId, expiresAt: sessionExpiresAt },
update: {
expiresAt: sessionExpiresAt,
revealUntil: null,
backupDigest: null,
backupExpiresAt: null,
},
});
return {
access_token: access,
refresh_token: refresh,
token_type: 'Bearer',
expires_in: 3600,
expires_in: Math.max(0, Math.floor((+accessExpiresAt - Date.now()) / 1000)),
scope: (selected || current).join(' '),
};
});
@@ -328,6 +366,7 @@ export class AgentOAuth implements OAuthServerProvider {
if (
!row ||
row.revokedAt ||
(row.clientId && row.refreshExpiresAt && row.refreshExpiresAt <= new Date()) ||
(row.expiresAt ? row.expiresAt <= new Date() : !!row.clientId) ||
row.resource !== urls().resource.toString()
)
@@ -358,6 +397,15 @@ export class AgentOAuth implements OAuthServerProvider {
id,
...(userId ? { userId } : {}),
revokedAt: null,
AND: [
{
OR: [
{ clientId: null },
{ refreshExpiresAt: null },
{ refreshExpiresAt: { gt: new Date() } },
],
},
],
OR: [{ expiresAt: { gt: new Date() } }, { expiresAt: null, clientId: null }],
},
});
+85 -1
View File
@@ -219,10 +219,94 @@ export class AgentOperations {
if (!row) throw new NotFoundException('操作不存在');
const t = this.get(row.tool),
grant = await this.oauth.grant(row.grantId, userId);
let impact: unknown = { parameters: row.parameters, message: t.description };
const selected = grant.scopes as string[];
const ids = new Set<string>();
const collect = (value: unknown) => {
if (typeof value === 'string' && /^[a-f0-9-]{36}$/i.test(value)) ids.add(value);
else if (Array.isArray(value)) value.forEach(collect);
else if (value && typeof value === 'object') Object.values(value).forEach(collect);
};
collect(row.parameters);
const visible = { userId, ...(selected.includes('hidden_read') ? {} : { hidden: false }) };
const positions = await this.db.position.findMany({
where: { ...visible, id: { in: [...ids] } },
select: { id: true, name: true, currency: true, side: true, kind: true },
});
const schedules = await this.db.schedule.findMany({
where: { userId, id: { in: [...ids] } },
select: { id: true, name: true, sourceId: true, targetId: true },
});
const visibleScheduleIds = new Set(
(
await this.db.position.findMany({
where: {
...visible,
id: { in: schedules.flatMap((s) => [s.sourceId, ...(s.targetId ? [s.targetId] : [])]) },
},
select: { id: true },
})
).map((p) => p.id),
);
const references = Object.fromEntries([
...positions.map((p) => [p.id, p]),
...schedules
.filter(
(s) =>
visibleScheduleIds.has(s.sourceId) &&
(!s.targetId || visibleScheduleIds.has(s.targetId)),
)
.map((s) => [s.id, { name: s.name, kind: 'schedule' }]),
]);
const p = row.parameters as Record<string, any>;
let current: Record<string, unknown> | undefined;
if (['movement_update', 'movement_delete'].includes(row.tool)) {
const movement = await this.db.transfer.findFirst({
where: { id: p.id, userId, source: visible, target: visible },
include: { source: true, target: true },
});
if (movement) {
references[movement.sourceId] = {
name: movement.source.name,
currency: movement.source.currency,
};
references[movement.targetId] = {
name: movement.target.name,
currency: movement.target.currency,
};
current = {
operation: movement.operation,
sourceId: movement.sourceId,
targetId: movement.targetId,
amount: movement.amount.toString(),
received: movement.received.toString(),
fee: movement.fee.toString(),
date: new Date(+movement.effectiveDate + 8 * 3600000).toISOString().slice(0, 16),
notes: movement.notes,
};
}
} else if (['history_update', 'history_delete', 'balance_record'].includes(row.tool)) {
const revision = await this.db.revision.findFirst({
where: {
positionId: p.id,
position: visible,
...(p.revisionId ? { id: p.revisionId } : {}),
},
orderBy: [{ effectiveDate: 'desc' }, { sequence: 'desc' }],
});
if (revision)
current = {
id: p.id,
amount: revision.amount.toString(),
date: new Date(+revision.effectiveDate + 8 * 3600000).toISOString().slice(0, 16),
notes: revision.notes,
};
}
const impact = { parameters: row.parameters, references, current };
return {
...this.view(row),
impact,
connectionName: grant.name,
destructive: !!t.destructive,
description: t.description,
};
}
+1
View File
@@ -209,6 +209,7 @@ export class AgentTransport {
name: g.name,
scopes: g.scopes,
expiresAt: g.expiresAt,
authorizationExpiresAt: g.clientId ? g.refreshExpiresAt : g.expiresAt,
resource: g.resource,
permission: (g.scopes as string[]).includes('write')
? 'write'
+67
View File
@@ -158,6 +158,63 @@ export class PortfolioBusinessService {
});
return { ok: true };
}
async deletion(r: UserRequest, id: string) {
const p = await this.own(r.userId, id, r.revealed);
if (p.kind !== 'account') throw new BadRequestException('此功能仅用于删除账户');
const [historyCount, movementCount, scheduleCount, linkCount] = await Promise.all([
this.db.revision.count({ where: { positionId: id } }),
this.db.transfer.count({
where: { userId: r.userId, OR: [{ sourceId: id }, { targetId: id }] },
}),
this.db.schedule.count({
where: { userId: r.userId, OR: [{ sourceId: id }, { targetId: id }] },
}),
this.db.positionLink.count({ where: { OR: [{ sourceId: id }, { targetId: id }] } }),
]);
return {
name: p.name,
historyCount,
movementCount,
scheduleCount,
linkCount,
canDelete: movementCount === 0 && scheduleCount === 0,
};
}
async remove(r: UserRequest, id: string, raw: unknown) {
const input = z
.object({ confirmation: z.string().min(1).max(100) })
.strict()
.parse(raw);
const result = await this.db.serial(async (tx) => {
await tx.$queryRaw(
Prisma.sql`SELECT id FROM Position WHERE id = ${id} AND userId = ${r.userId} FOR UPDATE`,
);
const p = await tx.position.findFirst({
where: { id, userId: r.userId, ...(r.revealed ? {} : { hidden: false }) },
});
if (!p) throw new NotFoundException('账户不存在');
if (p.kind !== 'account') throw new BadRequestException('此功能仅用于删除账户');
if (input.confirmation !== p.name)
throw new BadRequestException('请输入完整账户名称确认删除');
// Paired movements must be removed through replay, never via cascade.
if (
await tx.transfer.count({
where: { userId: r.userId, OR: [{ sourceId: id }, { targetId: id }] },
})
)
throw new ConflictException('账户存在资金往来,请先撤销相关记录,或选择归档保留历史');
if (
await tx.schedule.count({
where: { userId: r.userId, OR: [{ sourceId: id }, { targetId: id }] },
})
)
throw new ConflictException('账户被定时计划使用,请先删除相关计划,或选择归档');
await tx.position.delete({ where: { id } });
return { ok: true };
});
this.fx.invalidate(r.userId);
return result;
}
async revise(r: UserRequest, id: string, b: unknown) {
const v = revisionInput.parse(b);
if (pairedReasons.includes(v.reason)) throw new BadRequestException('请使用账户转账接口');
@@ -351,6 +408,16 @@ export class PortfolioController {
) {
return this.service.edit(r, id, b);
}
@Get('positions/:id/deletion') async deletion(@Req() r: UserRequest, @Param('id') id: string) {
return this.service.deletion(r, id);
}
@Delete('positions/:id') async remove(
@Req() r: UserRequest,
@Param('id') id: string,
@Body() b: unknown,
) {
return this.service.remove(r, id, b);
}
@Post('positions/:id/revisions') async revise(
@Req() r: UserRequest,
@Param('id') id: string,