340 lines
9.9 KiB
JavaScript
340 lines
9.9 KiB
JavaScript
/**
|
||
* Database migrations — add columns to existing tables
|
||
*/
|
||
|
||
import { randomUUID } from 'node:crypto'
|
||
|
||
/**
|
||
* Перенос apiType/apiBaseUrl с аккаунтов на хостера (idempotent).
|
||
* Вызывается из миграции и после импорта бэкапа / старого migrate API.
|
||
* @param {import('sql.js').Database} db
|
||
*/
|
||
function selectAllObjects(db, sql, params = []) {
|
||
const prepared = db.prepare(sql)
|
||
if (typeof prepared.all === 'function') {
|
||
return prepared.all(...params)
|
||
}
|
||
const stmt = prepared
|
||
stmt.bind(params)
|
||
const rows = []
|
||
while (stmt.step()) {
|
||
rows.push(stmt.getAsObject())
|
||
}
|
||
stmt.free()
|
||
return rows
|
||
}
|
||
|
||
export function consolidateProviderApiFromAccounts(db) {
|
||
const provRows = selectAllObjects(db, 'SELECT id, apiType, apiBaseUrl FROM providers')
|
||
for (const prov of provRows) {
|
||
const pid = prov.id
|
||
if (String(prov.apiType || '').trim() || String(prov.apiBaseUrl || '').trim()) {
|
||
continue
|
||
}
|
||
const accRows = selectAllObjects(
|
||
db,
|
||
`SELECT apiBaseUrl FROM provider_accounts
|
||
WHERE providerId = ?
|
||
AND lower(trim(COALESCE(apiType, ''))) = 'billmanager'
|
||
AND length(trim(COALESCE(apiBaseUrl, ''))) > 0
|
||
ORDER BY id`,
|
||
[pid],
|
||
)
|
||
if (!accRows.length) continue
|
||
const urls = [...new Set(accRows.map((r) => String(r.apiBaseUrl || '').trim()).filter(Boolean))]
|
||
if (urls.length > 1) {
|
||
console.warn(
|
||
`[vps-tracker] У хостера ${pid} у нескольких аккаунтов разный URL BILLmanager — в настройках хостера взят первый.`,
|
||
)
|
||
}
|
||
const apiBaseUrl = urls[0]
|
||
db.run(`UPDATE providers SET apiType = ?, apiBaseUrl = ? WHERE id = ?`, 'billmanager', apiBaseUrl, pid)
|
||
db.run(
|
||
`UPDATE provider_accounts SET apiType = '', apiBaseUrl = ''
|
||
WHERE providerId = ? AND lower(trim(COALESCE(apiType, ''))) = 'billmanager'
|
||
AND length(trim(COALESCE(apiBaseUrl, ''))) > 0`,
|
||
pid,
|
||
)
|
||
}
|
||
}
|
||
|
||
export const MIGRATIONS = [
|
||
{
|
||
name: 'provider_accounts_api',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN apiType TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN apiBaseUrl TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN apiCredentials TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'settings_sync',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN syncEnabled INTEGER')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN syncIntervalMinutes INTEGER')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'vps_paidUntil',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE vps ADD COLUMN paidUntil TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'provider_accounts_balance_api',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN balance_api REAL')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN balance_currency TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN balance_updated_at TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN enoughmoneyto TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'vps_userOverrides',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE vps ADD COLUMN userOverrides TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'active_tariffs_location_cpu',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE active_tariffs ADD COLUMN location TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE active_tariffs ADD COLUMN cpuModel TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'settings_customFields',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN customFields TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'settings_syncTariffsInterval',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN syncTariffsIntervalMinutes INTEGER')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'active_tariffs_country_datacenter',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE active_tariffs ADD COLUMN country TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE active_tariffs ADD COLUMN datacenterKey TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE active_tariffs ADD COLUMN datacenterName TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'settings_telegram',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN telegramBotToken TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN telegramChatId TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN notifyPaymentExpiryEnabled INTEGER')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN notifyNewTariffsEnabled INTEGER')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'settings_telegram_thread',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN telegramMessageThreadId TEXT')
|
||
} catch (e) {
|
||
if (!e.message?.includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'server_projects',
|
||
run(db) {
|
||
db.run(
|
||
`CREATE TABLE IF NOT EXISTS server_projects (
|
||
id TEXT PRIMARY KEY,
|
||
name TEXT NOT NULL,
|
||
color TEXT,
|
||
sortOrder INTEGER DEFAULT 0,
|
||
notes TEXT,
|
||
createdAt TEXT
|
||
)`,
|
||
)
|
||
try {
|
||
db.run('ALTER TABLE vps ADD COLUMN projectId TEXT')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
|
||
const distinctStmt = db.prepare(
|
||
`SELECT DISTINCT trim(project) AS n FROM vps WHERE length(trim(COALESCE(project, ''))) > 0`,
|
||
)
|
||
const seenLower = new Set()
|
||
const findStmt = db.prepare(
|
||
'SELECT id FROM server_projects WHERE LOWER(name) = LOWER(?) LIMIT 1',
|
||
)
|
||
while (distinctStmt.step()) {
|
||
const row = distinctStmt.getAsObject()
|
||
const t = String(row.n ?? '').trim()
|
||
if (!t) continue
|
||
const lk = t.toLowerCase()
|
||
if (seenLower.has(lk)) continue
|
||
seenLower.add(lk)
|
||
|
||
findStmt.bind([t])
|
||
const exists = Boolean(findStmt.step())
|
||
findStmt.reset()
|
||
if (!exists) {
|
||
const id = `proj-${randomUUID()}`
|
||
const now = new Date().toISOString()
|
||
db.run(
|
||
`INSERT INTO server_projects (id, name, color, sortOrder, notes, createdAt) VALUES (?, ?, NULL, 0, NULL, ?)`,
|
||
[id, t, now],
|
||
)
|
||
}
|
||
}
|
||
distinctStmt.free()
|
||
findStmt.free()
|
||
|
||
db.run(`UPDATE vps SET projectId = (
|
||
SELECT sp.id FROM server_projects sp
|
||
WHERE LOWER(sp.name) = LOWER(trim(COALESCE(vps.project, '')))
|
||
LIMIT 1
|
||
) WHERE length(trim(COALESCE(vps.project, ''))) > 0`)
|
||
},
|
||
},
|
||
{
|
||
name: 'sync_log_summary',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE sync_log ADD COLUMN summary TEXT')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'settings_notify_balance_digest',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN notifyLowBalanceEnabled INTEGER')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE settings ADD COLUMN notifySyncDigestEnabled INTEGER')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'provider_accounts_balance_alert_below',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE provider_accounts ADD COLUMN balance_alert_below REAL')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
},
|
||
},
|
||
{
|
||
name: 'providers_api_integration',
|
||
run(db) {
|
||
try {
|
||
db.exec('ALTER TABLE providers ADD COLUMN apiType TEXT')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
try {
|
||
db.exec('ALTER TABLE providers ADD COLUMN apiBaseUrl TEXT')
|
||
} catch (e) {
|
||
if (!String(e.message || e).includes('duplicate column')) throw e
|
||
}
|
||
consolidateProviderApiFromAccounts(db)
|
||
},
|
||
},
|
||
]
|