feat(admin): batch ops, transfer, watermark, usage and qr share pages
Co-Authored-By: Claude Code <noreply@anthropic.com>
This commit is contained in:
浅念
+232
-11
@@ -2,24 +2,50 @@
|
||||
// 为什么不用 API 造数：目标照片需要 published/processing_failed 等终态，
|
||||
// 真实链路要 worker 处理流水线；e2e 只验证管理页展示与操作，直插 DB 行
|
||||
// （uploadedAt 置位即可入管理列表，见 services/api listAlbumPhotos 过滤条件）。
|
||||
// 依赖经 PHOTO_DB_CN 环境变量注入 @photo/db-cn 绝对路径（apps/admin 不依赖
|
||||
// 该包，node 从文件位置解析不到；services/api 的 node_modules 里有现成链接）。
|
||||
// 用法：
|
||||
// node seed-fixture.cjs staff <account> <displayName>
|
||||
// node seed-fixture.cjs photos <albumId> <uploaderAccount>
|
||||
// 幂等：账号存在即跳过；照片按 objectKey upsert（同相册重复种不重复建行）。
|
||||
const db_cn = process.env.PHOTO_DB_CN
|
||||
// 依赖经环境变量注入包绝对路径（apps/admin 不依赖这些包，node 从文件位置
|
||||
// 解析不到；services/api 的 node_modules 里有现成链接）：
|
||||
// PHOTO_DB_CN=@photo/db-cn（必填） PHOTO_STORAGE=@photo/storage（水印模式）
|
||||
// 模式：
|
||||
// staff <account> <displayName> 员工账号（不可登录）
|
||||
// photos <albumId> <uploaderAccount> published/failed 各 1 张
|
||||
// gov-photos <albumId> <uploaderAccount> <count> [bytes] 批量隐藏/用量旅程：
|
||||
// N 张 published/ready
|
||||
// DB 直种（含 thumb/browse
|
||||
// 变体行——cn 公开列表计入；
|
||||
// bytes 可调大以跨阈值）
|
||||
// wm-photos <albumId> <uploaderAccount> <count> 水印旅程：N 张已发布
|
||||
// 照片 + 真实 MinIO 对象
|
||||
// （原图/browse/thumb，键
|
||||
// 对齐 worker 发布后形态）
|
||||
// watermark-run <albumId> 直驱 worker dist 的
|
||||
// WatermarkRegenService 把
|
||||
// 最新任务跑到终态（e2e 无
|
||||
// 常驻 worker，同 services/
|
||||
// api/test/watermark.e2e-
|
||||
// spec.ts 直驱模式）
|
||||
// 幂等：账号存在即跳过；照片按确定性 photoId/objectKey upsert（同相册重复种
|
||||
// 不重复建行）；watermark-run 对已 done 的任务返回 skipped 视为成功。
|
||||
/* eslint-disable @typescript-eslint/no-require-imports */
|
||||
const path = require('node:path')
|
||||
const { execSync } = require('node:child_process')
|
||||
const { existsSync, statSync } = require('node:fs')
|
||||
const { createHash } = require('node:crypto')
|
||||
|
||||
const dbCn = process.env.PHOTO_DB_CN
|
||||
const databaseUrl = process.env.DATABASE_URL_CN
|
||||
if (!db_cn || !databaseUrl) {
|
||||
if (!dbCn || !databaseUrl) {
|
||||
console.error('seed-fixture: PHOTO_DB_CN and DATABASE_URL_CN env vars are required')
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
// CommonJS（node 直跑，无构建步骤）：动态 require 经环境变量注入的包路径
|
||||
// eslint-disable-next-line @typescript-eslint/no-require-imports
|
||||
const { PrismaClient } = require(db_cn)
|
||||
|
||||
const { PrismaClient } = require(dbCn)
|
||||
const prisma = new PrismaClient({ datasources: { db: { url: databaseUrl } } })
|
||||
|
||||
/** worker 包根（sharp / dist 直驱均从这里解析）。 */
|
||||
const workerRoot = path.resolve(__dirname, '..', '..', '..', 'services', 'worker')
|
||||
|
||||
async function seedStaff(account, displayName) {
|
||||
const existing = await prisma.users.findUnique({ where: { account } })
|
||||
if (existing) {
|
||||
@@ -88,12 +114,207 @@ async function seedPhotos(albumId, uploaderAccount) {
|
||||
console.log(`seed-fixture: 2 photos seeded for album ${albumId}`)
|
||||
}
|
||||
|
||||
/** 确定性 photoId（同相册同序号重复种不重复建行；长度/字符集适配 cuid 列）。 */
|
||||
function deterministicPhotoId(scope, index) {
|
||||
return `e2e${createHash('md5').update(`${scope}:${index}`).digest('hex')}`
|
||||
}
|
||||
|
||||
async function findUploader(account) {
|
||||
const uploader = await prisma.users.findUnique({ where: { account } })
|
||||
if (!uploader) {
|
||||
throw new Error(`seed-fixture: uploader "${account}" not found`)
|
||||
}
|
||||
return uploader
|
||||
}
|
||||
|
||||
/**
|
||||
* gov-photos：N 张 published/ready 照片 DB 直种。带 thumb/browse 变体行
|
||||
* （cn 公开列表计入的必要条件——服务端只现签行上的键，不校验对象存在，
|
||||
* 故无需真实对象）；bytes 可调（用量旅程用大值跨 perAlbum 阈值）。
|
||||
*/
|
||||
async function seedGovPhotos(albumId, uploaderAccount, count, bytesEach) {
|
||||
const uploader = await findUploader(uploaderAccount)
|
||||
const total = Math.max(1, Number(count) || 1)
|
||||
const bytes = Math.max(1, Number(bytesEach) || 2048)
|
||||
const now = new Date()
|
||||
for (let i = 0; i < total; i++) {
|
||||
const photoId = deterministicPhotoId(`${albumId}:gov`, i)
|
||||
const objectKey = `e2e-gov/albums/${albumId}/originals/${photoId}.jpg`
|
||||
await prisma.photos.upsert({
|
||||
where: { objectKey },
|
||||
update: { visibility: 'published', processingState: 'ready' },
|
||||
create: {
|
||||
id: photoId,
|
||||
albumId,
|
||||
uploaderId: uploader.id,
|
||||
objectKey,
|
||||
bytes,
|
||||
width: 1200,
|
||||
height: 800,
|
||||
uploadedAt: now,
|
||||
publishedAt: now,
|
||||
processingState: 'ready',
|
||||
visibility: 'published'
|
||||
}
|
||||
})
|
||||
// 变体行（thumb/browse）：公开列表计入 + 现签 URL 可用；对象不落（e2e 不取图）
|
||||
const variants = [
|
||||
{ variant: 'thumb', width: 480, height: 320 },
|
||||
{ variant: 'browse', width: 1200, height: 800 }
|
||||
]
|
||||
for (const v of variants) {
|
||||
await prisma.photo_variants.upsert({
|
||||
where: { photoId_variant: { photoId, variant: v.variant } },
|
||||
update: {},
|
||||
create: {
|
||||
photoId,
|
||||
variant: v.variant,
|
||||
objectKey: `e2e-gov/albums/${albumId}/variants/${photoId}/1/${v.variant}.jpg`,
|
||||
bytes: Math.round(bytes / 4),
|
||||
version: 1,
|
||||
width: v.width,
|
||||
height: v.height
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
console.log(`seed-fixture: ${total} gov photos seeded for album ${albumId} (${bytes} B each)`)
|
||||
}
|
||||
|
||||
/** 测试内生成真实 JPEG（带纹理；与 worker test makeJpeg 同法，水印差异可观测）。 */
|
||||
function makeJpegBuffer(sharp, width, height) {
|
||||
const svg = Buffer.from(
|
||||
`<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${height}">` +
|
||||
Array.from(
|
||||
{ length: 12 },
|
||||
(_, i) =>
|
||||
`<rect x="${(i * 83) % width}" y="${(i * 47) % height}" width="${50 + i * 9}" height="${35 + i * 7}" fill="#${(0x50 + i * 7).toString(16).padStart(2, '0')}6030" fill-opacity="0.85"/>`
|
||||
).join('') +
|
||||
'</svg>'
|
||||
)
|
||||
return sharp(svg).jpeg({ quality: 85 }).toBuffer()
|
||||
}
|
||||
|
||||
/**
|
||||
* wm-photos：N 张已发布照片（原图 + browse/thumb 对象与行，形状对齐 worker
|
||||
* 发布后形态——watermark regen 读原图、公开列表需 thumb/browse 行）。
|
||||
* 键布局走 @photo/storage OBJECT_KEY_ROOT 约定（photo-live/dev）。
|
||||
*/
|
||||
async function seedWmPhotos(albumId, uploaderAccount, count) {
|
||||
const storagePath = process.env.PHOTO_STORAGE
|
||||
if (!storagePath) {
|
||||
throw new Error('seed-fixture: PHOTO_STORAGE env var is required for wm-photos')
|
||||
}
|
||||
|
||||
const { OBJECT_KEY_ROOT, createStorageFromEnv } = require(storagePath)
|
||||
|
||||
const sharp = require(path.join(workerRoot, 'node_modules', 'sharp'))
|
||||
const uploader = await findUploader(uploaderAccount)
|
||||
const storage = createStorageFromEnv()
|
||||
const total = Math.max(1, Number(count) || 1)
|
||||
const now = new Date()
|
||||
for (let i = 0; i < total; i++) {
|
||||
const photoId = deterministicPhotoId(`${albumId}:wm`, i)
|
||||
const objectKey = `${OBJECT_KEY_ROOT}/albums/${albumId}/originals/${photoId}.jpg`
|
||||
const browseKey = `${OBJECT_KEY_ROOT}/albums/${albumId}/variants/${photoId}/1/browse.jpg`
|
||||
const thumbKey = `${OBJECT_KEY_ROOT}/albums/${albumId}/variants/${photoId}/1/thumb.jpg`
|
||||
const original = await makeJpegBuffer(sharp, 900, 600)
|
||||
// browse/thumb 与原图同缓冲即可（e2e 断言 URL 指向，不比对变体字节）
|
||||
await storage.put(objectKey, original, 'image/jpeg')
|
||||
await storage.put(browseKey, original, 'image/jpeg')
|
||||
await storage.put(thumbKey, original, 'image/jpeg')
|
||||
await prisma.photos.upsert({
|
||||
where: { objectKey },
|
||||
update: {},
|
||||
create: {
|
||||
id: photoId,
|
||||
albumId,
|
||||
uploaderId: uploader.id,
|
||||
objectKey,
|
||||
bytes: original.length,
|
||||
width: 900,
|
||||
height: 600,
|
||||
uploadedAt: now,
|
||||
publishedAt: now,
|
||||
processingState: 'ready',
|
||||
visibility: 'published'
|
||||
}
|
||||
})
|
||||
const variants = [
|
||||
{ variant: 'thumb', key: thumbKey, width: 480, height: 320 },
|
||||
{ variant: 'browse', key: browseKey, width: 900, height: 600 }
|
||||
]
|
||||
for (const v of variants) {
|
||||
await prisma.photo_variants.upsert({
|
||||
where: { photoId_variant: { photoId, variant: v.variant } },
|
||||
update: {},
|
||||
create: {
|
||||
photoId,
|
||||
variant: v.variant,
|
||||
objectKey: v.key,
|
||||
bytes: original.length,
|
||||
version: 1,
|
||||
width: v.width,
|
||||
height: v.height
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
console.log(`seed-fixture: ${total} wm photos (with real objects) seeded for album ${albumId}`)
|
||||
}
|
||||
|
||||
/**
|
||||
* watermark-run：把相册最新水印任务直驱到终态。e2e 环境无常驻 worker
|
||||
* （Playwright 只起 api + admin dev），借 worker dist 的服务核心直跑——与
|
||||
* services/api/test/watermark.e2e-spec.ts 同一入口（processJob），API 侧
|
||||
* POST 已落好 config/job 行。dist 过期（缺文件或源更新）则先 tsc 构建。
|
||||
*/
|
||||
async function runWatermark(albumId) {
|
||||
const distService = path.join(workerRoot, 'dist', 'watermark.service.js')
|
||||
const srcService = path.join(workerRoot, 'src', 'watermark.service.ts')
|
||||
const stale =
|
||||
!existsSync(distService) ||
|
||||
(existsSync(srcService) && statSync(srcService).mtimeMs > statSync(distService).mtimeMs)
|
||||
if (stale) {
|
||||
console.log('seed-fixture: building services/worker dist (stale or missing)...')
|
||||
execSync('npx tsc -p tsconfig.build.json', { cwd: workerRoot, stdio: 'inherit' })
|
||||
}
|
||||
|
||||
const { WatermarkRegenService } = require(distService)
|
||||
const storagePath = process.env.PHOTO_STORAGE
|
||||
if (!storagePath) {
|
||||
throw new Error('seed-fixture: PHOTO_STORAGE env var is required for watermark-run')
|
||||
}
|
||||
|
||||
const { createStorageFromEnv } = require(storagePath)
|
||||
const storage = createStorageFromEnv()
|
||||
const job = await prisma.watermark_jobs.findFirst({
|
||||
where: { albumId },
|
||||
orderBy: { createdAt: 'desc' }
|
||||
})
|
||||
if (!job) {
|
||||
throw new Error(`seed-fixture: no watermark job found for album ${albumId}`)
|
||||
}
|
||||
const service = new WatermarkRegenService(prisma, storage)
|
||||
const outcome = await service.processJob(job.id)
|
||||
console.log(`seed-fixture: watermark job ${job.id} → ${outcome}`)
|
||||
if (outcome === 'failed') {
|
||||
throw new Error('seed-fixture: watermark job failed (task-level abort)')
|
||||
}
|
||||
}
|
||||
|
||||
async function main() {
|
||||
const [mode, argA, argB] = process.argv.slice(2)
|
||||
const [mode, argA, argB, argC, argD] = process.argv.slice(2)
|
||||
if (mode === 'staff') {
|
||||
await seedStaff(argA, argB)
|
||||
} else if (mode === 'photos') {
|
||||
await seedPhotos(argA, argB)
|
||||
} else if (mode === 'gov-photos') {
|
||||
await seedGovPhotos(argA, argB, argC, argD)
|
||||
} else if (mode === 'wm-photos') {
|
||||
await seedWmPhotos(argA, argB, argC)
|
||||
} else if (mode === 'watermark-run') {
|
||||
await runWatermark(argA)
|
||||
} else {
|
||||
throw new Error(`seed-fixture: unknown mode "${mode}"`)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user