222 lines
9.0 KiB
JavaScript
222 lines
9.0 KiB
JavaScript
import fs from 'node:fs';
|
||
import path from 'node:path';
|
||
|
||
const repoRoot = process.cwd();
|
||
|
||
function readUtf8(relativePath) {
|
||
const absolute = path.join(repoRoot, relativePath);
|
||
if (!fs.existsSync(absolute)) {
|
||
failures.push(`${relativePath}: 文件不存在，无法执行 SpacetimeDB runtime access 检查`);
|
||
return null;
|
||
}
|
||
return fs.readFileSync(absolute, 'utf8');
|
||
}
|
||
|
||
const forbiddenSnippets = [
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/puzzle.rs',
|
||
snippet: '.puzzle_work_profile()\n .iter()\n .filter(|row| row.owner_user_id == input.owner_user_id)',
|
||
reason: 'puzzle_work_profile 已有 by_puzzle_work_owner_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/puzzle.rs',
|
||
snippet: '.puzzle_work_profile()\n .iter()\n .filter(|row| row.publication_status == PuzzlePublicationStatus::Published)',
|
||
reason: 'puzzle_work_profile 已有 by_puzzle_work_publication_status 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/puzzle.rs',
|
||
snippet: '.puzzle_leaderboard_entry()\n .iter()\n .filter(|row| row.profile_id == profile_id && row.grid_size == grid_size)',
|
||
reason: 'puzzle_leaderboard_entry 已有 by_puzzle_leaderboard_profile_grid 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/match3d/mod.rs',
|
||
snippet: '.match3d_work_profile()\n .iter()\n .filter(|row| {',
|
||
reason: 'match3d_work_profile 已有 owner/status 索引，列表不应整表过滤',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/visual_novel.rs',
|
||
snippet: '.visual_novel_work_profile()\n .iter()\n .filter(|row| {',
|
||
reason: 'visual_novel_work_profile 已有 owner/status 索引，列表不应整表过滤',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/asset_metadata/objects.rs',
|
||
snippet: '.asset_object()\n .iter()\n .find(|row| row.bucket == input.bucket && row.object_key == input.object_key)',
|
||
reason: 'asset_object 已有 by_bucket_object_key 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/asset_metadata/objects.rs',
|
||
snippet: '.asset_object()\n .iter()\n .filter(|row| row.asset_kind == asset_kind)',
|
||
reason: 'asset_object 已有 asset_kind 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/ai/stages.rs',
|
||
snippet: '.ai_task_stage()\n .iter()\n .filter(|row| row.task_id == task_id)',
|
||
reason: 'ai_task_stage 已有 by_ai_task_stage_task_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/ai/stages.rs',
|
||
snippet: '.ai_text_chunk()\n .iter()\n .filter(|row| row.task_id == task_id && row.stage_kind == stage_kind)',
|
||
reason: 'ai_text_chunk 已有 by_ai_text_chunk_task_id / by_ai_text_chunk_task_stage_sequence 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/ai/snapshots.rs',
|
||
snippet: '.ai_task_stage()\n .iter()\n .filter(|stage| stage.task_id == row.task_id)',
|
||
reason: 'ai_task_stage 快照组装应使用 by_ai_task_stage_task_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/ai/snapshots.rs',
|
||
snippet: '.ai_result_reference()\n .iter()\n .filter(|reference| reference.task_id == row.task_id)',
|
||
reason: 'ai_result_reference 快照组装应使用 by_ai_result_reference_task_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/runtime/profile.rs',
|
||
snippet: '.profile_save_archive()\n .iter()\n .filter(|row| row.user_id == validated_input.user_id)',
|
||
reason: 'profile_save_archive 已有 by_profile_save_archive_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/runtime/profile.rs',
|
||
snippet: '.profile_played_world()\n .iter()\n .filter(|row| row.user_id == validated_input.user_id)',
|
||
reason: 'profile_played_world 已有 by_profile_played_world_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/runtime/profile.rs',
|
||
snippet: '.profile_wallet_ledger()\n .iter()\n .filter(|row| row.user_id == validated_input.user_id)',
|
||
reason: 'profile_wallet_ledger 已有 by_profile_wallet_ledger_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/runtime/profile.rs',
|
||
snippet: '.profile_referral_relation()\n .iter()\n .filter(|row| row.inviter_user_id == user_id)',
|
||
reason: 'profile_referral_relation 已有 by_profile_referral_inviter_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/runtime/profile.rs',
|
||
snippet: '.profile_recharge_order()\n .iter()\n .filter(|row| row.user_id == user_id)',
|
||
reason: 'profile_recharge_order 已有 by_profile_recharge_order_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/runtime/profile.rs',
|
||
snippet: '.tracking_daily_stat()\n .iter()\n .filter(|row| {',
|
||
reason: 'tracking_daily_stat 已有 by_tracking_daily_stat_scope_day / event_day 索引，analytics 查询不应整表过滤',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/custom_world/mod.rs',
|
||
snippet: '.custom_world_profile()\n .iter()\n .find(|row| {',
|
||
reason: 'custom_world_profile owner 维度已有 by_custom_world_profile_owner_user_id 索引',
|
||
},
|
||
{
|
||
file: 'server-rs/crates/spacetime-module/src/custom_world/mod.rs',
|
||
snippet: '.custom_world_profile()\n .iter()\n .filter(|profile| {',
|
||
reason: 'custom_world_profile Published 同步已有 by_custom_world_profile_publication_status 索引',
|
||
},
|
||
];
|
||
|
||
const procedureResultFiles = [
|
||
'server-rs/crates/module-puzzle/src/application.rs',
|
||
'server-rs/crates/module-big-fish/src/domain.rs',
|
||
'server-rs/crates/spacetime-module/src/match3d/types.rs',
|
||
'server-rs/crates/spacetime-module/src/square_hole/types.rs',
|
||
'server-rs/crates/spacetime-module/src/visual_novel.rs',
|
||
'server-rs/crates/spacetime-module/src/bark_battle/types.rs',
|
||
];
|
||
|
||
const mapperCompatibilityFiles = [
|
||
'server-rs/crates/spacetime-client/src/mapper.rs',
|
||
'server-rs/crates/spacetime-client/src/lib.rs',
|
||
];
|
||
|
||
const bigFishRuntimeFiles = [
|
||
'server-rs/crates/module-big-fish/src/commands.rs',
|
||
'server-rs/crates/spacetime-module/src/big_fish/runtime.rs',
|
||
'server-rs/crates/spacetime-module/src/big_fish/session.rs',
|
||
];
|
||
|
||
const legacyMapperPatterns = [
|
||
{
|
||
pattern: /\b[A-Za-z0-9_]*JsonRecord\b/u,
|
||
reason: 'spacetime-client mapper 不应保留旧 ProcedureResult JSON 兼容 Record',
|
||
},
|
||
{
|
||
pattern: /\bCompatibleBigFish[A-Za-z0-9_]*\b/u,
|
||
reason: 'spacetime-client mapper 不应保留 BigFish 旧 JSON 兼容结构',
|
||
},
|
||
{
|
||
pattern: /\bmap_[A-Za-z0-9_]*_json\b/u,
|
||
reason: 'spacetime-client mapper 不应再通过 map_*_json 反序列化 procedure payload',
|
||
},
|
||
{
|
||
pattern: /serde_json::from_str::<[A-Za-z0-9_:]*JsonRecord/u,
|
||
reason: 'spacetime-client mapper 不应把 procedure result 再反序列化为 JsonRecord',
|
||
},
|
||
{
|
||
pattern: /\b(?:items|run|work|session|event|feedback)_json:\s*Some\(/u,
|
||
reason: 'mapper 测试与兼容路径不应再构造旧 procedure JSON 字符串字段',
|
||
},
|
||
];
|
||
|
||
const typedProcedurePayloadFieldPattern =
|
||
/\b(?:row|session|work|item|items|run|event|feedback)_json:\s*Option<String>/gu;
|
||
|
||
const failures = [];
|
||
|
||
for (const rule of forbiddenSnippets) {
|
||
const content = readUtf8(rule.file);
|
||
if (content === null) {
|
||
continue;
|
||
}
|
||
if (content.includes(rule.snippet)) {
|
||
failures.push(`${rule.file}: ${rule.reason}`);
|
||
}
|
||
}
|
||
|
||
for (const file of procedureResultFiles) {
|
||
const content = readUtf8(file);
|
||
if (content === null) {
|
||
continue;
|
||
}
|
||
const resultBlocks = content.match(/pub struct [A-Za-z0-9_]*ProcedureResult\s*\{[\s\S]*?\n\}/g) ?? [];
|
||
for (const block of resultBlocks) {
|
||
const jsonFields = block.match(typedProcedurePayloadFieldPattern);
|
||
if (jsonFields?.length) {
|
||
const name = block.match(/pub struct ([A-Za-z0-9_]+)/)?.[1] ?? 'ProcedureResult';
|
||
failures.push(`${file}: ${name} 仍通过 ${jsonFields.join(', ')} 跨层返回 JSON 字符串`);
|
||
}
|
||
}
|
||
}
|
||
|
||
for (const file of mapperCompatibilityFiles) {
|
||
const content = readUtf8(file);
|
||
if (content === null) {
|
||
continue;
|
||
}
|
||
for (const rule of legacyMapperPatterns) {
|
||
if (rule.pattern.test(content)) {
|
||
failures.push(`${file}: ${rule.reason}`);
|
||
}
|
||
}
|
||
}
|
||
|
||
for (const file of bigFishRuntimeFiles) {
|
||
const content = readUtf8(file);
|
||
if (content === null) {
|
||
continue;
|
||
}
|
||
const resultBlocks = content.match(/pub struct [A-Za-z0-9_]*ProcedureResult\s*\{[\s\S]*?\n\}/g) ?? [];
|
||
for (const block of resultBlocks) {
|
||
const jsonFields = block.match(typedProcedurePayloadFieldPattern);
|
||
if (jsonFields?.length) {
|
||
const name = block.match(/pub struct ([A-Za-z0-9_]+)/)?.[1] ?? 'ProcedureResult';
|
||
failures.push(`${file}: ${name} 仍通过 ${jsonFields.join(', ')} 跨层返回 JSON 字符串`);
|
||
}
|
||
}
|
||
}
|
||
|
||
if (failures.length > 0) {
|
||
console.error('SpacetimeDB runtime access 检查失败：');
|
||
for (const failure of failures) {
|
||
console.error(`- ${failure}`);
|
||
}
|
||
process.exit(1);
|
||
}
|
||
|
||
console.log('SpacetimeDB runtime access 检查通过。');
|