283 lines
8.5 KiB
JavaScript
283 lines
8.5 KiB
JavaScript
import { createRequire } from 'module';
|
||
import fs from 'fs';
|
||
import path from 'path';
|
||
import { fileURLToPath } from 'url';
|
||
|
||
const require = createRequire(import.meta.url);
|
||
const __filename = fileURLToPath(import.meta.url);
|
||
const __dirname = path.dirname(__filename);
|
||
|
||
let runtimeConfig = {};
|
||
try {
|
||
runtimeConfig = require('../pocket-base/bai_api_pb_hooks/bai_api_shared/config/runtime.js');
|
||
} catch (_error) {
|
||
runtimeConfig = {};
|
||
}
|
||
|
||
function readEnvFile(filePath) {
|
||
if (!fs.existsSync(filePath)) return {};
|
||
|
||
const content = fs.readFileSync(filePath, 'utf8');
|
||
const result = {};
|
||
|
||
for (const rawLine of content.split(/\r?\n/)) {
|
||
const line = rawLine.trim();
|
||
if (!line || line.startsWith('#')) continue;
|
||
const index = line.indexOf('=');
|
||
if (index === -1) continue;
|
||
const key = line.slice(0, index).trim();
|
||
const value = line.slice(index + 1).trim();
|
||
result[key] = value;
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
const backendEnv = readEnvFile(path.resolve(__dirname, '..', 'back-end', '.env'));
|
||
|
||
const PB_URL = (process.env.PB_URL || backendEnv.POCKETBASE_API_URL || 'https://bai-api.blv-oa.com/pb').replace(/\/+$/, '');
|
||
const AUTH_TOKEN = process.env.POCKETBASE_AUTH_TOKEN || runtimeConfig.POCKETBASE_AUTH_TOKEN || '';
|
||
const DRY_RUN = String(process.env.DRY_RUN || '').trim() === '1';
|
||
|
||
if (!AUTH_TOKEN) {
|
||
console.error('❌ 缺少认证信息，请提供 POCKETBASE_AUTH_TOKEN。');
|
||
process.exit(1);
|
||
}
|
||
|
||
function getUrlCandidates(url) {
|
||
const trimmed = String(url || '').replace(/\/+$/, '');
|
||
const candidates = [];
|
||
if (trimmed) {
|
||
candidates.push(trimmed);
|
||
}
|
||
|
||
const withoutPb = trimmed.replace(/\/pb$/i, '');
|
||
if (withoutPb && candidates.indexOf(withoutPb) === -1) {
|
||
candidates.push(withoutPb);
|
||
}
|
||
|
||
return candidates;
|
||
}
|
||
|
||
function normalizeText(value) {
|
||
return String(value || '').trim();
|
||
}
|
||
|
||
function toParameterArray(value) {
|
||
if (value === null || typeof value === 'undefined' || value === '') {
|
||
return [];
|
||
}
|
||
|
||
let source = value;
|
||
if (typeof source === 'string') {
|
||
try {
|
||
source = JSON.parse(source);
|
||
} catch (_error) {
|
||
const raw = normalizeText(source);
|
||
if (raw.indexOf('map[') === 0 && raw.endsWith(']')) {
|
||
const mapped = {};
|
||
const body = raw.slice(4, -1).trim();
|
||
const pairs = body ? body.split(/\s+/) : [];
|
||
for (let i = 0; i < pairs.length; i += 1) {
|
||
const pair = pairs[i];
|
||
const separatorIndex = pair.indexOf(':');
|
||
if (separatorIndex <= 0) continue;
|
||
const key = normalizeText(pair.slice(0, separatorIndex));
|
||
if (!key) continue;
|
||
mapped[key] = pair.slice(separatorIndex + 1);
|
||
}
|
||
source = mapped;
|
||
} else {
|
||
return [];
|
||
}
|
||
}
|
||
}
|
||
|
||
const rows = [];
|
||
const indexByName = {};
|
||
|
||
function upsert(nameValue, rawValue) {
|
||
const name = normalizeText(nameValue);
|
||
if (!name) return;
|
||
|
||
const valueText = rawValue === null || typeof rawValue === 'undefined' ? '' : String(rawValue);
|
||
const existingIndex = indexByName[name];
|
||
if (typeof existingIndex === 'number') {
|
||
rows[existingIndex].value = valueText;
|
||
return;
|
||
}
|
||
|
||
indexByName[name] = rows.length;
|
||
rows.push({ name, value: valueText });
|
||
}
|
||
|
||
if (Array.isArray(source)) {
|
||
for (let i = 0; i < source.length; i += 1) {
|
||
const item = source[i] && typeof source[i] === 'object' ? source[i] : null;
|
||
if (!item) continue;
|
||
upsert(item.name || item.key, item.value);
|
||
}
|
||
return rows;
|
||
}
|
||
|
||
if (typeof source !== 'object') {
|
||
return [];
|
||
}
|
||
|
||
try {
|
||
source = JSON.parse(JSON.stringify(source));
|
||
} catch (_error) {}
|
||
|
||
const keys = Object.keys(source || {});
|
||
for (let i = 0; i < keys.length; i += 1) {
|
||
upsert(keys[i], source[keys[i]]);
|
||
}
|
||
|
||
return rows;
|
||
}
|
||
|
||
function isSameArray(left, right) {
|
||
return JSON.stringify(left || []) === JSON.stringify(right || []);
|
||
}
|
||
|
||
async function requestPbJson(baseUrl, method, apiPath, token, body) {
|
||
const res = await fetch(baseUrl + apiPath, {
|
||
method: method,
|
||
headers: Object.assign(
|
||
{
|
||
Authorization: `Bearer ${token}`,
|
||
},
|
||
body ? { 'Content-Type': 'application/json' } : {},
|
||
),
|
||
body: body ? JSON.stringify(body) : undefined,
|
||
});
|
||
|
||
const text = await res.text();
|
||
let json = {};
|
||
try {
|
||
json = text ? JSON.parse(text) : {};
|
||
} catch (_error) {
|
||
json = { raw: text };
|
||
}
|
||
|
||
if (!res.ok) {
|
||
const err = new Error(json.message || `HTTP ${res.status}`);
|
||
err.status = res.status;
|
||
err.response = json;
|
||
throw err;
|
||
}
|
||
|
||
return json;
|
||
}
|
||
|
||
async function migrate() {
|
||
try {
|
||
const urlCandidates = getUrlCandidates(PB_URL);
|
||
let activeUrl = '';
|
||
let lastError = null;
|
||
|
||
for (let i = 0; i < urlCandidates.length; i += 1) {
|
||
const candidate = urlCandidates[i];
|
||
try {
|
||
await requestPbJson(
|
||
candidate,
|
||
'GET',
|
||
'/api/collections/tbl_product_list/records?page=1&perPage=1',
|
||
AUTH_TOKEN,
|
||
);
|
||
activeUrl = candidate;
|
||
console.log(`✅ 已加载 POCKETBASE_AUTH_TOKEN（${candidate}）。`);
|
||
break;
|
||
} catch (error) {
|
||
lastError = error;
|
||
console.log(`⚠️ 候选地址探测失败（${candidate}）:`, {
|
||
status: error && error.status,
|
||
message: error && error.message,
|
||
});
|
||
}
|
||
}
|
||
|
||
if (!activeUrl) {
|
||
throw lastError || new Error('无法连接 PocketBase 或认证失败');
|
||
}
|
||
|
||
console.log(`🔄 开始迁移 tbl_product_list.prod_list_parameters，PB: ${activeUrl}`);
|
||
if (DRY_RUN) {
|
||
console.log('🧪 当前为 DRY_RUN=1，仅预览不会写入数据库。');
|
||
}
|
||
|
||
let page = 1;
|
||
const perPage = 200;
|
||
let total = 0;
|
||
let changed = 0;
|
||
let skipped = 0;
|
||
let failed = 0;
|
||
|
||
while (true) {
|
||
const list = await requestPbJson(
|
||
activeUrl,
|
||
'GET',
|
||
`/api/collections/tbl_product_list/records?page=${page}&perPage=${perPage}`,
|
||
AUTH_TOKEN,
|
||
);
|
||
|
||
const items = Array.isArray(list.items) ? list.items : [];
|
||
if (!items.length) {
|
||
break;
|
||
}
|
||
|
||
for (let i = 0; i < items.length; i += 1) {
|
||
const item = items[i];
|
||
total += 1;
|
||
|
||
try {
|
||
const current = item.prod_list_parameters;
|
||
const normalized = toParameterArray(current);
|
||
|
||
if (isSameArray(current, normalized)) {
|
||
skipped += 1;
|
||
continue;
|
||
}
|
||
|
||
if (!DRY_RUN) {
|
||
await requestPbJson(
|
||
activeUrl,
|
||
'PATCH',
|
||
`/api/collections/tbl_product_list/records/${item.id}`,
|
||
AUTH_TOKEN,
|
||
{ prod_list_parameters: normalized },
|
||
);
|
||
}
|
||
|
||
changed += 1;
|
||
console.log(`✅ 已迁移: ${item.prod_list_id || item.id}`);
|
||
} catch (error) {
|
||
failed += 1;
|
||
console.error(`❌ 迁移失败: ${item.prod_list_id || item.id}`, error.message || error);
|
||
}
|
||
}
|
||
|
||
const totalPages = Number(list.totalPages || 1);
|
||
if (page >= totalPages) {
|
||
break;
|
||
}
|
||
page += 1;
|
||
}
|
||
|
||
console.log('\n🎯 迁移完成');
|
||
console.log(`- 总记录: ${total}`);
|
||
console.log(`- 已变更: ${changed}`);
|
||
console.log(`- 跳过不变: ${skipped}`);
|
||
console.log(`- 失败: ${failed}`);
|
||
} catch (error) {
|
||
console.error('❌ 迁移任务执行失败:', {
|
||
status: error && error.status,
|
||
message: error && error.message,
|
||
response: error && error.response,
|
||
});
|
||
process.exitCode = 1;
|
||
}
|
||
}
|
||
|
||
migrate();
|