feat: add reusable private and shared account icons

This commit is contained in:
陈煜 committed 2026-10-01 18:55:15 +08:00
1 parent d2b2681698
commit a4e9d56b8a
44 files changed
+1443 -20

No files matched your search

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.7 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.8 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.7 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.9 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.7 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 2.0 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.5 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.9 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 2.7 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.8 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.2 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.5 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 2.3 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.6 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.9 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.2 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 2.4 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 926 B

Binary file not shown.

After

Width:  |  Height:  |  Size: 806 B

Binary file not shown.

After

Width:  |  Height:  |  Size: 39 KiB

+122
View File
@@ -0,0 +1,122 @@
[
{
"name": "中国工商银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中国工商银行@3x.png",
"file": "01.png",
"sha256": "6a49f1e01332575c2fd5b8f3892d2969f6a254733a7731badae6b6c51ded6108"
},
{
"name": "中国农业银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中国农业银行@3x.png",
"file": "02.png",
"sha256": "cc023287b5a6d0187f9e8198a1791507b0655d593ee1e619e5b48f5cb351484d"
},
{
"name": "中国建设银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中国建设银行@3x.png",
"file": "03.png",
"sha256": "c1a1711a94ebb7e49d7494cb1d502a40f4bec8d30dc1e9f1cbd215c65fe233db"
},
{
"name": "中国银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中国银行@3x.png",
"file": "04.png",
"sha256": "718978d8d444321d356810841919aa81da590e5faf23fe1a0cad6e471dbcb024"
},
{
"name": "交通银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/交通银行@3x.png",
"file": "05.png",
"sha256": "75b5e40cae526194715688b6b8c63684093d4cccbcd498ebc4fe3a730a616e6c"
},
{
"name": "招商银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/招商银行@3x.png",
"file": "06.png",
"sha256": "abf071f561028bac84e7bd07f53314623c6983810d328b6f7990e6ca11736017"
},
{
"name": "中信银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中信银行@3x.png",
"file": "07.png",
"sha256": "51c873fcdabfab733700b48e772f3e03f2cb0918fe7a36a90a064297c018c530"
},
{
"name": "中国光大银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中国光大银行@3x.png",
"file": "08.png",
"sha256": "791d54ef3f49fdd5466c09f2cec955585f3af22bd2593f95ac259548c22fcfbc"
},
{
"name": "中国民生银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/中国民生银行@3x.png",
"file": "09.png",
"sha256": "a466c4469559b966d7096449aa155227b1e0b8965a883f544067d8361b7c0a9d"
},
{
"name": "兴业银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/兴业银行@3x.png",
"file": "10.png",
"sha256": "05b860edad339b114bb38ac954da2c49f41f3f3e8ddc6d8a5c939dc2fc747e07"
},
{
"name": "平安银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/平安银行@3x.png",
"file": "11.png",
"sha256": "7e057cd3f23fd5df3fca54e1dc0b253f9a61005f52e7f933087820f63aade94e"
},
{
"name": "上海浦东发展银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/上海浦东发展银行@3x.png",
"file": "12.png",
"sha256": "b205b7adaa577df7f367d4644717a0e3d6fbdba45a77590c2a0a5e9f2fe3c91f"
},
{
"name": "广发银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/广发银行@3x.png",
"file": "13.png",
"sha256": "1a3c1f7f2c738eabbb330bb3eb1a5888e69f731fd9f7f62729366172ee773919"
},
{
"name": "华夏银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/华夏银行@3x.png",
"file": "14.png",
"sha256": "1547c7d7a2436e80c39d197be26ffb77dc35728f69260608b0dc3bff2743be6f"
},
{
"name": "北京银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/北京银行@3x.png",
"file": "15.png",
"sha256": "d59af4b650391832e3a2a4f3d852b6dde6c791a5842a1ce570f9c2990157c6dd"
},
{
"name": "上海银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/上海银行@3x.png",
"file": "16.png",
"sha256": "0ea8f0650db5ca04c7869cc46a2043606cf0038d701f22f07c40cd3e6b70519e"
},
{
"name": "浙商银行",
"source": "https://raw.githubusercontent.com/cellier/bank-icon-cn/master/png/72/浙商银行@3x.png",
"file": "17.png",
"sha256": "8237233ee5ee7487ace0f84dce2832037ec13e2f004dcf6ed65652d783961656"
},
{
"name": "支付宝",
"source": "https://i.alipayobjects.com/common/favicon/favicon.ico",
"file": "18.png",
"sha256": "b662de58b15b34d1bf4d2a8bc546f7062a3faaa22acd2ecbbbad70f1e7f37a39"
},
{
"name": "微信",
"source": "https://res.wx.qq.com/a/wx_fed/assets/res/NTI4MWU5.ico",
"file": "19.png",
"sha256": "a62d7d84bd02b1718106d294d1f2c8387f9967239696c1e8b446201b63f34dc7"
},
{
"name": "京东金融",
"source": "https://jr.jd.com/logo.png",
"file": "20.png",
"sha256": "8dc9703f571e605df552dc7eb14ae074d9ddab6b27a9140b610e87b1c2a5f693"
}
]
+7 -5
View File
@@ -9,7 +9,8 @@
"db:generate": "prisma generate",
"db:migrate": "node scripts/database.cjs deploy",
"db:status": "node scripts/database.cjs status",
"test:integration": "tsx --test --test-concurrency=1 test/integration.test.ts test/privacy.test.ts"
"test:integration": "tsx --test --test-concurrency=1 test/integration.test.ts test/privacy.test.ts test/icons.test.ts",
"icons:seed": "node scripts/seed-icons.cjs"
},
"dependencies": {
"@nestjs/common": "^11.0.0",
@@ -26,19 +27,20 @@
"multer": "^2.4.0",
"reflect-metadata": "^0.2.2",
"rxjs": "^7.8.2",
"sharp": "^0.35.5",
"yauzl": "^3.4.0",
"zod": "^4.1.0"
},
"devDependencies": {
"@types/archiver": "^8.0.0",
"@types/cookie-parser": "^1.4.9",
"@types/express": "^5.0.0",
"@types/multer": "^2.3.0",
"@types/node": "^24.0.0",
"@types/yauzl": "^3.4.0",
"mysql2": "^3.15.0",
"prisma": "6.19.0",
"tsx": "^4.20.0",
"typescript": "^5.9.0",
"@types/archiver": "^8.0.0",
"@types/yauzl": "^3.4.0",
"@types/multer": "^2.3.0"
"typescript": "^5.9.0"
}
}
@@ -0,0 +1,16 @@
CREATE TABLE `Icon` (
`id` CHAR(36) NOT NULL,
`ownerId` CHAR(36) NULL,
`name` VARCHAR(100) NOT NULL,
`shared` BOOLEAN NOT NULL DEFAULT false,
`hash` CHAR(64) NOT NULL,
`data` MEDIUMBLOB NOT NULL,
`source` VARCHAR(500) NULL,
`createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
PRIMARY KEY (`id`),
UNIQUE INDEX `Icon_ownerId_hash_shared_key` (`ownerId`, `hash`, `shared`),
INDEX `Icon_shared_name_idx` (`shared`, `name`),
CONSTRAINT `Icon_ownerId_fkey` FOREIGN KEY (`ownerId`) REFERENCES `User` (`id`) ON DELETE SET NULL ON UPDATE CASCADE
) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
ALTER TABLE `Position` ADD COLUMN `iconId` CHAR(36) NULL;
ALTER TABLE `Position` ADD CONSTRAINT `Position_iconId_fkey` FOREIGN KEY (`iconId`) REFERENCES `Icon` (`id`) ON DELETE SET NULL ON UPDATE CASCADE;
+18
View File
@@ -17,6 +17,7 @@ model User {
positions Position[]
rates ExchangeRate[]
sessions Session[]
icons Icon[]
}
model Session {
id String @id @db.Char(64)
@@ -44,6 +45,8 @@ model Position {
hidden Boolean @default(false)
createdAt DateTime @default(now())
updatedAt DateTime @updatedAt
iconId String? @db.Char(36)
icon Icon? @relation(fields:[iconId],references:[id],onDelete:SetNull)
revisions Revision[]
outgoing PositionLink[] @relation("Source")
incoming PositionLink[] @relation("Target")
@@ -82,3 +85,18 @@ model ExchangeRate {
source String @db.VarChar(30)
@@unique([userId,currency,baseCurrency,date])
}
model Icon {
id String @id @default(uuid()) @db.Char(36)
ownerId String? @db.Char(36)
owner User? @relation(fields:[ownerId],references:[id],onDelete:SetNull)
name String @db.VarChar(100)
shared Boolean @default(false)
hash String @db.Char(64)
data Bytes @db.MediumBlob
source String? @db.VarChar(500)
createdAt DateTime @default(now())
positions Position[]
@@unique([ownerId,hash,shared])
@@index([shared,name])
}
+14
View File
@@ -0,0 +1,14 @@
"""Optional catalog maintenance: convert downloaded ICO files using Pillow.
Normal installs use the committed PNG assets and do not need Python.
Run after download-icons.cjs, before icons:seed.
"""
from pathlib import Path
from PIL import Image
folder = Path(__file__).resolve().parent.parent / 'assets' / 'icons'
for source in folder.glob('*.ico'):
with Image.open(source) as image:
image.convert('RGBA').save(source.with_suffix('.png'))
source.unlink()
print('Catalog ICO sources converted to PNG.')
+1 -1
View File
@@ -3,7 +3,7 @@ const tsc = require.resolve('typescript/bin/tsc');
if (spawnSync(process.execPath, [tsc], { stdio: 'inherit' }).status !== 0) process.exit(1);
const children = [
spawn(process.execPath, [tsc, '--watch', '--preserveWatchOutput'], { stdio: 'inherit' }),
spawn(process.execPath, ['--watch', 'dist/main.js'], { stdio: 'inherit' }),
spawn(process.execPath, ['--watch', '--watch-path=dist', 'dist/main.js'], { stdio: 'inherit' }),
];
function stop() {
for (const p of children) p.kill();
+70
View File
@@ -0,0 +1,70 @@
// Explicit, fixed public sources only. No user account information is sent.
const fs = require('node:fs/promises');
const path = require('node:path');
const sharp = require('sharp');
const { createHash } = require('node:crypto');
const dir = path.join(__dirname, '../assets/icons');
async function main() {
await fs.mkdir(dir, { recursive: true });
const response = await fetch('https://api.github.com/repos/cellier/bank-icon-cn/contents/png/72');
if (!response.ok) throw Error();
const banks = await response.json();
const names = [
'中国工商银行',
'中国农业银行',
'中国建设银行',
'中国银行',
'交通银行',
'招商银行',
'中信银行',
'中国光大银行',
'中国民生银行',
'兴业银行',
'平安银行',
'上海浦东发展银行',
'广发银行',
'华夏银行',
'北京银行',
'上海银行',
'浙商银行',
];
const sources = names.map((name) => ({
name,
source: banks.find((b) => b.name === name + '@3x.png')?.download_url,
}));
sources.push(
{ name: '支付宝', source: 'https://i.alipayobjects.com/common/favicon/favicon.ico' },
{ name: '微信', source: 'https://res.wx.qq.com/a/wx_fed/assets/res/NTI4MWU5.ico' },
{ name: '京东金融', source: 'https://jr.jd.com/logo.png' },
);
const manifest = [];
for (const item of sources) {
if (!item.source) throw Error('Missing catalog source');
const res = await fetch(item.source, { signal: AbortSignal.timeout(20000) });
if (!res.ok) throw Error('Public icon download failed');
const raw = Buffer.from(await res.arrayBuffer());
const file =
String(manifest.length + 1).padStart(2, '0') +
(item.source.endsWith('.ico') ? '.ico' : '.png');
if (file.endsWith('.ico')) await fs.writeFile(path.join(dir, file), raw);
else
await fs.writeFile(
path.join(dir, file),
await sharp(raw)
.resize(256, 256, { fit: 'inside', withoutEnlargement: true })
.png()
.toBuffer(),
);
manifest.push({
...item,
file: file.replace('.ico', '.png'),
sha256: createHash('sha256').update(raw).digest('hex'),
});
}
await fs.writeFile(path.join(dir, 'sources.json'), JSON.stringify(manifest, null, 2) + '\n');
console.log(`Downloaded ${manifest.length} public icons; convert ICO sources before seeding.`);
}
main().catch(() => {
console.error('Icon source download failed; existing database icons were not changed.');
process.exitCode = 1;
});
+41
View File
@@ -0,0 +1,41 @@
require('dotenv/config');
const { PrismaClient } = require('@prisma/client');
const { readFile } = require('node:fs/promises');
const { join } = require('node:path');
const { createHash } = require('node:crypto');
const sharp = require('sharp');
const db = new PrismaClient();
async function main() {
const dir = join(__dirname, '../assets/icons');
const sources = JSON.parse(await readFile(join(dir, 'sources.json'), 'utf8'));
// Deterministic IDs make repeated runs safe, without overwriting existing records.
for (const item of sources) {
const data = await sharp(await readFile(join(dir, item.file)))
.resize(256, 256, { fit: 'inside', withoutEnlargement: true })
.png()
.toBuffer();
const hex = createHash('sha256')
.update('worthpath-builtin:' + item.name)
.digest('hex');
const id = `${hex.slice(0, 8)}-${hex.slice(8, 12)}-4${hex.slice(13, 16)}-a${hex.slice(17, 20)}-${hex.slice(20, 32)}`;
await db.icon.upsert({
where: { id },
create: {
id,
name: item.name,
shared: true,
data,
source: item.source,
hash: createHash('sha256').update(data).digest('hex'),
},
update: {},
});
}
console.log(`Shared icon catalog ready: ${sources.length} icons.`);
}
main()
.catch(() => {
console.error('Icon seeding failed; check local database configuration.');
process.exitCode = 1;
})
.finally(() => db.$disconnect());
+62 -1
View File
@@ -28,6 +28,7 @@ import { UserRequest } from './auth';
import { positionInput, positionMeta, currency, revisionInput, rateInput } from './validation';
import { createHash } from 'node:crypto';
import { toBusinessDate } from './validation';
import { iconName, validateStoredIcon } from './icons';
import { day, businessTime } from './calculation';
const timestamp = z.iso
.datetime()
@@ -67,6 +68,19 @@ const backupSchema = z
.object({ showSidebar: z.boolean(), idleMinutes: z.number().int().min(0).max(1440) })
.strict()
.optional(),
icons: z
.array(
z
.object({
id: z.string().uuid(),
name: iconName,
shared: z.boolean(),
image: z.string().max(3 * 1024 * 1024),
hash: z.string().regex(/^[a-f0-9]{64}$/),
})
.strict(),
)
.optional(),
positions: z.array(record),
links: z.array(z.object({ sourceId: z.string().uuid(), targetId: z.string().uuid() }).strict()),
rates: z.array(rateInput.safeExtend({ source: z.enum(['manual', 'frankfurter']) })),
@@ -79,6 +93,12 @@ export function validateBackup(raw: unknown) {
if (ids.size !== b.positions.length) throw new BadRequestException('重复项目 ID');
const origins = b.positions.map((p) => p.importedFromId || p.id);
if (new Set(origins).size !== origins.length) throw new BadRequestException('备份内包含重复项目');
const iconIds = new Set((b.icons || []).map((i) => i.id));
if (
iconIds.size !== (b.icons || []).length ||
b.positions.some((p) => p.iconId && !iconIds.has(p.iconId))
)
throw new BadRequestException('图标关联无效');
const revisionIds = new Set<string>();
for (const p of b.positions) {
positionInput.parse({
@@ -180,6 +200,7 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
]);
const positions = ps.map((p) => ({
id: p.id,
iconId: p.iconId,
importedFromId: p.importedFromId,
name: p.name,
kind: p.kind,
@@ -202,6 +223,17 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
updatedAt: r.updatedAt.toISOString(),
})),
}));
const icons = await client.icon.findMany({
where: {
OR: [
{ ownerId: userId },
{
id: { in: ps.flatMap((p) => (p.iconId ? [p.iconId] : [])) },
OR: [{ shared: true }, { ownerId: userId }],
},
],
},
});
return backupSchema.parse({
format: 'worthpath',
version: 2,
@@ -215,6 +247,13 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
...rates.flatMap((r) => [r.currency, r.baseCurrency]),
]),
],
icons: icons.map((i) => ({
id: i.id,
name: i.name,
shared: i.shared,
hash: i.hash,
image: Buffer.from(i.data).toString('base64'),
})),
positions,
links: ps.flatMap((p) =>
p.outgoing.map((l) => ({ sourceId: l.sourceId, targetId: l.targetId })),
@@ -314,6 +353,7 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
(a.currency + a.baseCurrency + a.date).localeCompare(b.currency + b.baseCurrency + b.date),
);
data.currencies.sort();
data.icons?.sort((a, b) => a.id.localeCompare(b.id));
return createHash('sha256').update(JSON.stringify(data)).digest('hex');
}
@Get('clear-status') async clearStatus(@Req() r: UserRequest) {
@@ -332,6 +372,7 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
if (this.fingerprint(await this.data(r.userId, tx)) !== s.backupDigest)
throw new ConflictException('数据已变化,请重新下载备份');
await tx.position.deleteMany({ where: { userId: r.userId } });
await tx.icon.deleteMany({ where: { ownerId: r.userId, shared: false } });
await tx.exchangeRate.deleteMany({ where: { userId: r.userId } });
await tx.session.updateMany({
where: { userId: r.userId },
@@ -345,15 +386,17 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
@Post('preview') async preview(@Req() r: UserRequest, @Body() raw: unknown) {
const b = validateBackup(raw),
existing = await this.data(r.userId);
for (const i of b.icons || []) await validateStoredIcon(i.image, i.hash);
this.conflicts(b, existing);
return {
positions: b.positions.length,
revisions: b.positions.reduce((n, p) => n + p.revisions.length, 0),
rates: b.rates.length,
icons: (b.icons || []).length,
baseCurrency: b.baseCurrency,
currentBaseCurrency: existing.baseCurrency,
message:
'只追加新项目并重建关联,相同汇率保留。不会覆盖已有项目。已有本位币保留,空账户恢复备份本位币和个人设置。确认后以事务导入。',
'只追加新项目并重建关联,相同汇率保留。不会覆盖已有项目。已有本位币保留,空账户恢复备份本位币和个人设置。确认后以事务导入,备份中的图标恢复为私有,不会自动发布到共享库。',
};
}
private conflicts(b: Backup, existing: Backup) {
@@ -379,6 +422,8 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
.strict()
.parse(raw),
b = validateBackup(backup);
const iconData = new Map<string, Buffer>();
for (const i of b.icons || []) iconData.set(i.id, await validateStoredIcon(i.image, i.hash));
return this.db.$transaction(
async (tx) => {
const ps = await tx.position.findMany({
@@ -398,12 +443,28 @@ export class BackupController implements OnModuleDestroy, OnModuleInit {
rates: rs.map((v) => ({ ...v, rate: v.rate.toString(), date: day(v.date) })),
} as unknown as Backup;
this.conflicts(b, existing);
const iconMapping = new Map<string, string>();
for (const i of b.icons || []) {
const row = await tx.icon.upsert({
where: { ownerId_hash_shared: { ownerId: r.userId, hash: i.hash, shared: false } },
create: {
ownerId: r.userId,
name: i.name,
hash: i.hash,
shared: false,
data: new Uint8Array(iconData.get(i.id)!),
},
update: {},
});
iconMapping.set(i.id, row.id);
}
const mapping = new Map<string, string>();
for (const p of b.positions) {
const row = await tx.position.create({
data: {
userId: r.userId,
importedFromId: p.importedFromId || p.id,
iconId: p.iconId ? iconMapping.get(p.iconId) : null,
name: p.name,
kind: p.kind,
side: p.side,
+134
View File
@@ -0,0 +1,134 @@
import {
Controller,
Injectable,
Get,
Post,
Query,
Req,
Res,
Param,
Body,
UploadedFile,
UseInterceptors,
BadRequestException,
NotFoundException,
} from '@nestjs/common';
import { FileInterceptor } from '@nestjs/platform-express';
import { memoryStorage } from 'multer';
import { Response } from 'express';
import sharp from 'sharp';
import { createHash } from 'node:crypto';
import { z } from 'zod';
import { Database } from './database';
import { UserRequest } from './auth';
export const iconName = z.string().trim().min(1).max(100);
export const iconHash = (data: Buffer) => createHash('sha256').update(data).digest('hex');
export async function normalizeIcon(data: Buffer) {
if (!data.length || data.length > 2 * 1024 * 1024)
throw new BadRequestException('图标不能超过 2 MB');
try {
const image = sharp(data, { limitInputPixels: 16000000, animated: false });
const meta = await image.metadata();
if (!['png', 'jpeg', 'webp'].includes(meta.format || '') || (meta.pages || 1) > 1)
throw Error();
return await image
.rotate()
.resize(256, 256, { fit: 'inside', withoutEnlargement: true })
.png()
.toBuffer();
} catch {
throw new BadRequestException('请选择有效的静态 PNG、JPG 或 WebP 图片');
}
}
export async function validateStoredIcon(image: string, hash: string) {
const data = Buffer.from(image, 'base64');
if (data.toString('base64') !== image || iconHash(data) !== hash)
throw new BadRequestException('图标内容或校验值无效');
await normalizeIcon(data);
const meta = await sharp(data).metadata();
if (meta.format !== 'png' || !meta.width || !meta.height || meta.width > 256 || meta.height > 256)
throw new BadRequestException('备份图标必须是规范的 PNG');
return data;
}
@Injectable()
export class IconsService {
constructor(private db: Database) {}
async requireVisible(userId: string, id?: string | null) {
if (
id &&
!(await this.db.icon.findFirst({
where: { id, OR: [{ shared: true }, { ownerId: userId }] },
select: { id: true },
}))
)
throw new BadRequestException('图标不存在或无权使用');
}
}
@Controller('api/icons')
export class IconsController {
constructor(private db: Database) {}
@Get() async list(@Req() r: UserRequest, @Query('q') q = '', @Query('page') page = '1') {
const query = z.string().trim().max(100).parse(q);
const index = z.coerce.number().int().min(1).max(100000).parse(page);
const where = { OR: [{ shared: true }, { ownerId: r.userId }], name: { contains: query } };
const [items, total] = await this.db.$transaction([
this.db.icon.findMany({
where,
select: { id: true, name: true, shared: true, source: true },
orderBy: [{ name: 'asc' }, { id: 'asc' }],
skip: (index - 1) * 60,
take: 60,
}),
this.db.icon.count({ where }),
]);
return { items, total, page: index };
}
@Get(':id/image') async image(
@Req() r: UserRequest,
@Param('id') id: string,
@Res() res: Response,
) {
const icon = await this.db.icon.findFirst({
where: { id, OR: [{ shared: true }, { ownerId: r.userId }] },
});
if (!icon) throw new NotFoundException('图标不存在');
res.setHeader('Content-Type', 'image/png');
res.setHeader('X-Content-Type-Options', 'nosniff');
res.send(Buffer.from(icon.data));
}
@Post('upload')
@UseInterceptors(
FileInterceptor('file', {
storage: memoryStorage(),
limits: { fileSize: 2 * 1024 * 1024, files: 1, fields: 3 },
}),
)
async upload(
@Req() r: UserRequest,
@Body() raw: unknown,
@UploadedFile() file?: Express.Multer.File,
) {
const v = z
.object({
name: iconName,
shared: z.enum(['true', 'false']).default('false'),
confirmed: z.literal('true').optional(),
})
.strict()
.parse(raw);
const shared = v.shared === 'true';
if (shared && (!/\p{Script=Han}/u.test(v.name) || v.confirmed !== 'true'))
throw new BadRequestException('共享图标须填写中文名称并明确确认公开给所有用户');
if (!file) throw new BadRequestException('请选择图标文件');
const data = await normalizeIcon(file.buffer),
hash = iconHash(data);
const icon = await this.db.icon.upsert({
where: { ownerId_hash_shared: { ownerId: r.userId, hash, shared } },
create: { name: v.name, ownerId: r.userId, shared, hash, data },
update: {},
select: { id: true, name: true, shared: true, source: true },
});
return icon;
}
}
+15 -2
View File
@@ -8,6 +8,7 @@ import { json } from 'express';
import { AuthController, AuthGuard, AuthService } from './auth';
import { PortfolioController } from './portfolio';
import { BackupController } from './backup';
import { IconsController, IconsService } from './icons';
import { Database } from './database';
import { RatesService, SettingsController } from './rates';
import { ZodError } from 'zod';
@@ -41,8 +42,20 @@ class SafeErrors implements ExceptionFilter {
}
}
@Module({
providers: [Database, AuthService, RatesService, { provide: APP_GUARD, useClass: AuthGuard }],
controllers: [AuthController, PortfolioController, SettingsController, BackupController],
providers: [
Database,
AuthService,
RatesService,
IconsService,
{ provide: APP_GUARD, useClass: AuthGuard },
],
controllers: [
IconsController,
AuthController,
PortfolioController,
SettingsController,
BackupController,
],
})
class AppModule {}
async function bootstrap() {
+4
View File
@@ -17,12 +17,14 @@ import { positionInput, positionMeta, revisionInput, today, toBusinessDate } fro
import { history, overview } from './calculation';
import { z } from 'zod';
import { Prisma } from '@prisma/client';
import { IconsService } from './icons';
import { RatesService } from './rates';
@Controller('api')
export class PortfolioController {
constructor(
private db: Database,
private fx: RatesService,
private icons: IconsService,
) {}
private async own(userId: string, id: string, revealed = false) {
const p = await this.db.position.findFirst({
@@ -58,6 +60,7 @@ export class PortfolioController {
@Post('positions') async create(@Req() r: UserRequest, @Body() b: unknown) {
const v = positionInput.parse(b),
{ amount, date, ...meta } = v;
await this.icons.requireVisible(r.userId, meta.iconId);
const created = await this.db.position.create({
data: {
...meta,
@@ -89,6 +92,7 @@ export class PortfolioController {
p.side !== 'liability'
)
throw new BadRequestException('信用卡和贷款账户必须为负债');
await this.icons.requireVisible(r.userId, v.iconId);
await this.db.position.update({ where: { id: p.id }, data: v });
return { ok: true };
}
+1
View File
@@ -62,6 +62,7 @@ export const revisionInput = z
.strict();
export const positionMeta = z
.object({
iconId: z.string().uuid().nullable().optional(),
name: z.string().trim().min(1).max(100),
category: z.string().trim().min(1).max(40),
notes,
+15 -5
View File
@@ -15,6 +15,7 @@ const files = [
'history.json',
'links.json',
'rates.json',
'icons.json',
] as const;
const sha = (s: Buffer | string) => createHash('sha256').update(s).digest('hex');
export function packBackup(b: Backup) {
@@ -30,6 +31,7 @@ export function packBackup(b: Backup) {
),
'links.json': b.links,
'rates.json': b.rates,
'icons.json': b.icons || [],
};
const contents = Object.fromEntries(
files.map((name) => [name, JSON.stringify(data[name], null, 2)]),
@@ -37,7 +39,7 @@ export function packBackup(b: Backup) {
contents['manifest.json'] = JSON.stringify(
{
format: 'worthpath',
version: 3,
version: 4,
exportedAt: b.exportedAt,
files: files.map((name) => ({ name, sha256: sha(contents[name]) })),
},
@@ -106,23 +108,30 @@ export async function readBackupZip(input: string | Buffer): Promise<unknown> {
});
zip.readEntry();
});
if (contents.size !== files.length + 1) throw Error();
const parse = (name: string) =>
JSON.parse(new TextDecoder('utf-8', { fatal: true }).decode(contents.get(name)!));
const manifest = z
.object({
format: z.literal('worthpath'),
version: z.literal(3),
version: z.union([z.literal(3), z.literal(4)]),
exportedAt: z.iso.datetime(),
files: z
.array(
z.object({ name: z.enum(files), sha256: z.string().regex(/^[a-f0-9]{64}$/) }).strict(),
)
.length(files.length),
.min(files.length - 1)
.max(files.length),
})
.strict()
.parse(parse('manifest.json'));
if (new Set(manifest.files.map((f) => f.name)).size !== files.length) throw Error();
const expected = manifest.version === 3 ? files.filter((f) => f !== 'icons.json') : [...files];
if (
contents.size !== expected.length + 1 ||
new Set(manifest.files.map((f) => f.name)).size !== expected.length ||
expected.some((name) => !manifest.files.some((f) => f.name === name))
)
throw Error();
for (const f of manifest.files) if (sha(contents.get(f.name)!) !== f.sha256) throw Error();
const settings = z
.object({
@@ -160,6 +169,7 @@ export async function readBackupZip(input: string | Buffer): Promise<unknown> {
positions: positions.map((p) => ({ ...p, revisions: grouped.get(p.id) || [] })),
links: parse('links.json'),
rates: parse('rates.json'),
...(manifest.version === 4 ? { icons: parse('icons.json') } : {}),
};
} catch {
throw new BadRequestException(
+199
View File
@@ -0,0 +1,199 @@
import 'dotenv/config';
import { test } from 'node:test';
import assert from 'node:assert/strict';
import { randomBytes, randomUUID } from 'node:crypto';
import { PrismaClient } from '@prisma/client';
import sharp from 'sharp';
import { readBackupZip } from '../src/zip';
const base = process.env.TEST_API_URL || 'http://127.0.0.1:3100/api';
test('private and shared icons, account reuse and complete ZIP restoration preserve isolation', async () => {
const db = new PrismaClient(),
names: string[] = [],
publicIds: string[] = [];
async function call(path: string, cookie = '', method = 'GET', data?: unknown) {
const res = await fetch(base + path, {
method,
headers: {
Cookie: cookie,
Origin: process.env.WEB_ORIGIN === '*' ? 'http://localhost:5173' : process.env.WEB_ORIGIN!,
...(data ? { 'Content-Type': 'application/json' } : {}),
},
body: data ? JSON.stringify(data) : undefined,
});
return {
status: res.status,
data: res.headers.get('content-type')?.includes('application/zip')
? ((await readBackupZip(Buffer.from(await res.arrayBuffer()))) as any)
: res.headers.get('content-type')?.includes('application/json')
? await res.json()
: Buffer.from(await res.arrayBuffer()),
cookie: res.headers.get('set-cookie')?.split(';')[0] || '',
};
}
async function account() {
const username = 'wp_icons_' + randomUUID();
names.push(username);
const r = await call('/auth/register', '', 'POST', {
username,
password: randomBytes(18).toString('hex'),
});
assert.equal(r.status, 201);
return { ...r, id: (await db.user.findUniqueOrThrow({ where: { username } })).id };
}
const image = await sharp({
create: { width: 400, height: 400, channels: 4, background: '#097c71' },
})
.png()
.toBuffer();
async function upload(
cookie: string,
name: string,
shared = false,
confirmed = false,
content = image,
) {
const f = new FormData();
f.append('file', new Blob([new Uint8Array(content)], { type: 'image/png' }), 'icon.png');
f.append('name', name);
f.append('shared', String(shared));
if (confirmed) f.append('confirmed', 'true');
const r = await fetch(base + '/icons/upload', {
method: 'POST',
headers: { Cookie: cookie, Origin: 'http://localhost:5173' },
body: f,
});
return { status: r.status, data: await r.json() };
}
try {
const a = await account(),
b = await account();
assert.equal((await call('/icons')).status, 401);
const own = await upload(a.cookie, '我的银行');
assert.equal(own.status, 201);
assert.equal((await upload(a.cookie, '同一图片')).data.id, own.data.id);
assert.equal((await call('/icons/' + own.data.id + '/image', b.cookie)).status, 404);
assert.equal(
(await call('/icons?q=' + encodeURIComponent('我的银行'), b.cookie)).data.total,
0,
);
const privateImage = await call('/icons/' + own.data.id + '/image', a.cookie);
assert.equal(privateImage.status, 200);
assert.equal((await sharp(privateImage.data).metadata()).width, 256);
assert.equal((await upload(a.cookie, 'English', true, true)).status, 400);
assert.equal((await upload(a.cookie, '共享银行', true)).status, 400);
assert.equal(
(await upload(a.cookie, '坏图标', false, false, Buffer.from('<svg onload="alert(1)"/>')))
.status,
400,
);
const shared = await upload(a.cookie, '共享测试银行', true, true);
assert.equal(shared.status, 201);
publicIds.push(shared.data.id);
assert.equal((await call('/icons/' + shared.data.id + '/image', b.cookie)).status, 200);
assert.equal(
(await call('/icons?q=' + encodeURIComponent('共享测试银行'), b.cookie)).data.items.some(
(i: any) => i.id === shared.data.id,
),
true,
);
const meta = {
kind: 'account',
side: 'asset',
category: 'bank',
name: '图标关联验收',
currency: 'CNY',
amount: '10',
date: '2026-09-01T10:35',
};
assert.equal(
(await call('/positions', b.cookie, 'POST', { ...meta, iconId: own.data.id })).status,
400,
);
const p = await call('/positions', a.cookie, 'POST', { ...meta, iconId: own.data.id });
assert.equal(p.status, 201);
assert.equal(
(await call('/positions', a.cookie, 'POST', { ...meta, iconId: own.data.id })).status,
201,
);
const other = await call('/positions', b.cookie, 'POST', { ...meta, iconId: shared.data.id });
assert.equal(other.status, 201);
assert.equal(
(
await call('/positions/' + other.data.id, b.cookie, 'PATCH', {
name: meta.name,
category: 'bank',
iconId: own.data.id,
})
).status,
400,
);
assert.equal(
(await db.position.findUniqueOrThrow({ where: { id: other.data.id } })).iconId,
shared.data.id,
);
assert.equal(
(
await call('/positions/' + p.data.id, a.cookie, 'PATCH', {
name: meta.name,
category: 'bank',
})
).status,
200,
);
assert.equal(
(await db.position.findUniqueOrThrow({ where: { id: p.data.id } })).iconId,
own.data.id,
);
const backup = await call('/backup', a.cookie);
assert.equal(backup.status, 200);
assert.equal(
backup.data.icons.some((i: any) => i.id === own.data.id),
true,
);
assert.doesNotMatch(JSON.stringify(backup.data), /passwordHash|sessionId|ownerId|token/i);
const broken = structuredClone(backup.data);
broken.icons[0].image = 'invalid';
assert.equal((await call('/backup/preview', b.cookie, 'POST', broken)).status, 400);
const before = await db.icon.count();
assert.equal(
(await call('/backup/import', b.cookie, 'POST', { confirmed: true, backup: broken })).status,
400,
);
assert.equal(await db.icon.count(), before);
assert.equal(
(await call('/backup/import', b.cookie, 'POST', { confirmed: true, backup: backup.data }))
.status,
201,
);
const imported = await db.position.findMany({
where: { userId: b.id, importedFromId: { not: null } },
include: { icon: true },
});
assert.equal(imported.length, 2);
assert.equal(imported[0].iconId, imported[1].iconId);
assert.equal(imported[0].icon?.ownerId, b.id);
assert.equal(imported[0].icon?.shared, false);
// Backup receipt remains valid; only temporary account A is cleared.
assert.equal(
(await call('/backup/clear', a.cookie, 'POST', { confirmation: '确定清空' })).status,
201,
);
assert.equal(await db.icon.count({ where: { id: own.data.id } }), 0);
assert.equal((await call('/icons/' + shared.data.id + '/image', b.cookie)).status, 200);
assert.equal(
(await db.position.findUniqueOrThrow({ where: { id: other.data.id } })).iconId,
shared.data.id,
);
assert.equal((await db.icon.count({ where: { ownerId: b.id, shared: false } })) > 0, true);
} finally {
const users = await db.user.findMany({
where: { username: { in: names } },
select: { id: true },
});
await db.icon.deleteMany({
where: { OR: [{ ownerId: { in: users.map((u) => u.id) } }, { id: { in: publicIds } }] },
});
await db.user.deleteMany({ where: { username: { in: names } } });
await db.$disconnect();
}
});
+14
View File
@@ -12,6 +12,7 @@ const empty = () =>
baseCurrency: 'CNY',
preferences: { showSidebar: false, idleMinutes: 9 },
currencies: ['CNY'],
icons: [],
positions: [],
rates: [],
links: [],
@@ -37,6 +38,7 @@ test('ZIP contains separate JSON files and restores settings without authenticat
'currencies.json',
'debts.json',
'history.json',
'icons.json',
'links.json',
'manifest.json',
'rates.json',
@@ -93,3 +95,15 @@ test('backup accepts over 1000 positions, 10000 revisions per position and 20000
21001,
);
});
test('legacy v3 ZIP remains readable without icons', async () => {
const contents = packBackup(empty());
delete contents['icons.json'];
const manifest = JSON.parse(contents['manifest.json']);
manifest.version = 3;
manifest.files = manifest.files.filter((f: { name: string }) => f.name !== 'icons.json');
contents['manifest.json'] = JSON.stringify(manifest);
const restored = validateBackup(await readBackupZip(await archive(contents)));
assert.equal(restored.icons, undefined);
assert.deepEqual(restored.positions, []);
});