66 lines
2.4 KiB
JavaScript
66 lines
2.4 KiB
JavaScript
/**
|
||
* Migration 006 — Settings 表 + products.generated_aff_url
|
||
*
|
||
* - settings: 键值对存储系统配置（TG_BOT_TOKEN, TG_DEFAULT_CHANNEL_ID 等）
|
||
* - products.generated_aff_url: 缓存自动生成的 aff 链接，方便查询
|
||
*/
|
||
const Database = require('better-sqlite3');
|
||
const path = require('path');
|
||
require('dotenv').config({ path: path.join(__dirname, '..', '.env') });
|
||
|
||
const dbPath = path.resolve(__dirname, '..', process.env.DB_PATH || 'db/monitor.sqlite');
|
||
const db = new Database(dbPath);
|
||
db.pragma('journal_mode = WAL');
|
||
|
||
// ── 创建 settings 表 ──
|
||
db.exec(`
|
||
CREATE TABLE IF NOT EXISTS settings (
|
||
key TEXT PRIMARY KEY,
|
||
value TEXT,
|
||
updated_at TEXT DEFAULT (datetime('now'))
|
||
);
|
||
`);
|
||
console.log('+ settings table');
|
||
|
||
// ── products 加 generated_aff_url 列 ──
|
||
function ensureColumn(table, column, sql) {
|
||
const cols = db.prepare(`PRAGMA table_info(${table})`).all().map(c => c.name);
|
||
if (!cols.includes(column)) {
|
||
db.exec(`ALTER TABLE ${table} ADD COLUMN ${sql}`);
|
||
console.log(`+ ${table}.${column}`);
|
||
} else {
|
||
console.log(` ${table}.${column} (already exists)`);
|
||
}
|
||
}
|
||
|
||
ensureColumn('products', 'generated_aff_url', 'generated_aff_url TEXT');
|
||
|
||
// ── 回填 generated_aff_url（兼容老库：aff_code/aff_param 可能还不存在） ──
|
||
const cols = db.prepare(`PRAGMA table_info(products)`).all().map(c => c.name);
|
||
const hasAffCode = cols.includes('aff_code');
|
||
const hasAffParam = cols.includes('aff_param');
|
||
|
||
if (hasAffCode) {
|
||
const { buildAffUrl } = require('../src/utils/affHelper');
|
||
const selectSql = hasAffParam
|
||
? "SELECT id, buy_url, aff_code, aff_param FROM products WHERE aff_code IS NOT NULL AND aff_code != ''"
|
||
: "SELECT id, buy_url, aff_code, NULL as aff_param FROM products WHERE aff_code IS NOT NULL AND aff_code != ''";
|
||
const products = db.prepare(selectSql).all();
|
||
const updateStmt = db.prepare('UPDATE products SET generated_aff_url = ? WHERE id = ?');
|
||
|
||
db.transaction(() => {
|
||
for (const p of products) {
|
||
const baseUrl = p.buy_url;
|
||
if (!baseUrl) continue;
|
||
const affUrl = buildAffUrl(baseUrl, p.aff_code, p.aff_param || 'aff');
|
||
updateStmt.run(affUrl, p.id);
|
||
console.log(` product #${p.id}: generated_aff_url=${affUrl}`);
|
||
}
|
||
})();
|
||
} else {
|
||
console.log(' skip backfill: products.aff_code not found yet');
|
||
}
|
||
|
||
console.log('✅ migration-006 done:', dbPath);
|
||
db.close();
|