32603d281a
- 在Comment表中添加site_id字段，用于区分不同站点的评论 - 更新前后端API以支持site_id参数传递 - 修改自动迁移脚本，添加site_id字段迁移逻辑 - 更新前端组件配置，支持设置siteId参数
105 lines
3.9 KiB
JavaScript
105 lines
3.9 KiB
JavaScript
const fs = require('fs');
|
|
const path = require('path');
|
|
const { execSync } = require('child_process');
|
|
|
|
function findDatabaseName() {
|
|
const files = ['wrangler.jsonc', 'wrangler.toml'];
|
|
const baseDirs = [process.cwd(), path.resolve(__dirname, '..')];
|
|
for (const base of baseDirs) {
|
|
for (const file of files) {
|
|
const filePath = path.join(base, file);
|
|
if (!fs.existsSync(filePath)) {
|
|
continue;
|
|
}
|
|
const content = fs.readFileSync(filePath, 'utf-8');
|
|
if (file.endsWith('.jsonc')) {
|
|
const jsonMatch = content.match(/"database_name"\s*:\s*["'](.*?)["']/);
|
|
if (jsonMatch) {
|
|
return jsonMatch[1];
|
|
}
|
|
} else {
|
|
const chunks = content.split('[[d1_databases]]');
|
|
for (const chunk of chunks) {
|
|
if (chunk.includes('CWD_DB') || chunk.includes('"CWD_DB"')) {
|
|
const nameMatch = chunk.match(/database_name\s*=\s*["'](.*?)["']/);
|
|
if (nameMatch) {
|
|
return nameMatch[1];
|
|
}
|
|
}
|
|
}
|
|
const match = content.match(/database_name\s*=\s*["'](.*?)["']/);
|
|
if (match) {
|
|
return match[1];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function run() {
|
|
try {
|
|
const dbName = findDatabaseName();
|
|
if (!dbName) {
|
|
return;
|
|
}
|
|
try {
|
|
const checkCmd = `npx wrangler d1 execute ${dbName} --command "SELECT count(*) as count FROM pragma_table_info('Comment') WHERE name='post_url'" --remote --json`;
|
|
const output = execSync(checkCmd, { encoding: 'utf-8', stdio: ['ignore', 'pipe', 'ignore'] });
|
|
|
|
const result = JSON.parse(output);
|
|
const count = result[0]?.results?.[0]?.count;
|
|
|
|
if (count === 0) {
|
|
const sql = `
|
|
ALTER TABLE Comment ADD COLUMN post_url TEXT;
|
|
UPDATE Comment SET post_url = post_slug;
|
|
UPDATE Comment
|
|
SET post_slug = SUBSTR(
|
|
REPLACE(REPLACE(post_slug, 'https://', ''), 'http://', ''),
|
|
INSTR(REPLACE(REPLACE(post_slug, 'https://', ''), 'http://', ''), '/')
|
|
)
|
|
WHERE post_slug LIKE 'http%' AND INSTR(REPLACE(REPLACE(post_slug, 'https://', ''), 'http://', ''), '/') > 0;
|
|
`;
|
|
|
|
const flatSql = sql.replace(/\s+/g, ' ').trim();
|
|
const migrateCmd = `npx wrangler d1 execute ${dbName} --command "${flatSql}" --remote --yes`;
|
|
|
|
execSync(migrateCmd, { stdio: 'inherit' });
|
|
console.log('[Auto-Migrate] Migration applied for Comment.post_url.');
|
|
}
|
|
} catch (e) {
|
|
console.error('[Auto-Migrate] Failed post_url migration:', e.message);
|
|
}
|
|
|
|
try {
|
|
const checkCmd = `npx wrangler d1 execute ${dbName} --command "SELECT count(*) as count FROM pragma_table_info('Comment') WHERE name='site_id'" --remote --json`;
|
|
const output = execSync(checkCmd, { encoding: 'utf-8', stdio: ['ignore', 'pipe', 'ignore'] });
|
|
|
|
const result = JSON.parse(output);
|
|
const count = result[0]?.results?.[0]?.count;
|
|
|
|
if (count === 0) {
|
|
const sql = `
|
|
ALTER TABLE Comment ADD COLUMN site_id TEXT NOT NULL DEFAULT '';
|
|
CREATE INDEX IF NOT EXISTS idx_site_id ON Comment(site_id);
|
|
`;
|
|
|
|
const flatSql = sql.replace(/\s+/g, ' ').trim();
|
|
const migrateCmd = `npx wrangler d1 execute ${dbName} --command "${flatSql}" --remote --yes`;
|
|
|
|
execSync(migrateCmd, { stdio: 'inherit' });
|
|
console.log('[Auto-Migrate] Migration applied for Comment.site_id.');
|
|
}
|
|
} catch (e) {
|
|
console.error('[Auto-Migrate] Failed site_id migration:', e.message);
|
|
}
|
|
|
|
} catch (error) {
|
|
console.error('[Auto-Migrate] Failed:', error.message);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
run();
|