108 lines
3.5 KiB
JavaScript
108 lines
3.5 KiB
JavaScript
|
// e2e 数据种子夹具（Node 直跑，Playwright 用例内按需调用）。
|
|||
|
|
// 为什么不用 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
|
|||
|
|
const databaseUrl = process.env.DATABASE_URL_CN
|
|||
|
|
if (!db_cn || !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 prisma = new PrismaClient({ datasources: { db: { url: databaseUrl } } })
|
|||
|
|
|
|||
|
|
async function seedStaff(account, displayName) {
|
|||
|
|
const existing = await prisma.users.findUnique({ where: { account } })
|
|||
|
|
if (existing) {
|
|||
|
|
console.log(`seed-fixture: staff "${account}" exists, skipping`)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// passwordHash 占位（不可登录）：e2e 只经账号查询邀请该用户，从不登录
|
|||
|
|
await prisma.users.create({
|
|||
|
|
data: {
|
|||
|
|
account,
|
|||
|
|
passwordHash: 'seeded-no-login',
|
|||
|
|
displayName: displayName || account,
|
|||
|
|
role: 'staff',
|
|||
|
|
credentialStatus: 'active'
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
console.log(`seed-fixture: staff "${account}" created`)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function seedPhotos(albumId, uploaderAccount) {
|
|||
|
|
const uploader = await prisma.users.findUnique({ where: { account: uploaderAccount } })
|
|||
|
|
if (!uploader) {
|
|||
|
|
throw new Error(`seed-fixture: uploader "${uploaderAccount}" not found`)
|
|||
|
|
}
|
|||
|
|
const now = new Date()
|
|||
|
|
const rows = [
|
|||
|
|
{
|
|||
|
|
suffix: 'published',
|
|||
|
|
visibility: 'published',
|
|||
|
|
processingState: 'ready',
|
|||
|
|
width: 1200,
|
|||
|
|
height: 800,
|
|||
|
|
publishedAt: now,
|
|||
|
|
errorCode: null
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
suffix: 'failed',
|
|||
|
|
visibility: 'unpublished',
|
|||
|
|
processingState: 'processing_failed',
|
|||
|
|
width: null,
|
|||
|
|
height: null,
|
|||
|
|
publishedAt: null,
|
|||
|
|
errorCode: 'DECODE_FAILED'
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
for (const row of rows) {
|
|||
|
|
const objectKey = `e2e-seed/albums/${albumId}/${row.suffix}.jpg`
|
|||
|
|
await prisma.photos.upsert({
|
|||
|
|
where: { objectKey },
|
|||
|
|
update: {},
|
|||
|
|
create: {
|
|||
|
|
albumId,
|
|||
|
|
uploaderId: uploader.id,
|
|||
|
|
objectKey,
|
|||
|
|
bytes: 2048,
|
|||
|
|
width: row.width,
|
|||
|
|
height: row.height,
|
|||
|
|
uploadedAt: now,
|
|||
|
|
publishedAt: row.publishedAt,
|
|||
|
|
processingState: row.processingState,
|
|||
|
|
errorCode: row.errorCode,
|
|||
|
|
visibility: row.visibility
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
console.log(`seed-fixture: 2 photos seeded for album ${albumId}`)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function main() {
|
|||
|
|
const [mode, argA, argB] = process.argv.slice(2)
|
|||
|
|
if (mode === 'staff') {
|
|||
|
|
await seedStaff(argA, argB)
|
|||
|
|
} else if (mode === 'photos') {
|
|||
|
|
await seedPhotos(argA, argB)
|
|||
|
|
} else {
|
|||
|
|
throw new Error(`seed-fixture: unknown mode "${mode}"`)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
main()
|
|||
|
|
.catch((error) => {
|
|||
|
|
console.error(error)
|
|||
|
|
process.exitCode = 1
|
|||
|
|
})
|
|||
|
|
.finally(() => prisma.$disconnect())
|