chore: flatten repository to completely erase 255MB history
This commit is contained in:
@@ -0,0 +1,87 @@
|
||||
const { Pool } = require('pg');
|
||||
require('dotenv').config();
|
||||
|
||||
// Coolify uses DATABASE_URL
|
||||
// If not set, we can fallback to a local postgres or just let it fail
|
||||
const pool = new Pool({
|
||||
connectionString: process.env.DATABASE_URL,
|
||||
});
|
||||
|
||||
pool.on('error', (err, client) => {
|
||||
console.error('Unexpected error on idle client', err);
|
||||
process.exit(-1);
|
||||
});
|
||||
|
||||
const initDB = async () => {
|
||||
try {
|
||||
const client = await pool.connect();
|
||||
console.log('Connected to the PostgreSQL database.');
|
||||
|
||||
await client.query(`
|
||||
CREATE TABLE IF NOT EXISTS profiles (
|
||||
id TEXT PRIMARY KEY,
|
||||
email TEXT UNIQUE,
|
||||
password_hash TEXT,
|
||||
role TEXT DEFAULT 'user',
|
||||
api_credits INTEGER DEFAULT 10,
|
||||
storage_limit_mb INTEGER DEFAULT 500,
|
||||
is_active BOOLEAN DEFAULT false
|
||||
)
|
||||
`);
|
||||
|
||||
// In Postgres, ALTER TABLE ADD COLUMN IF NOT EXISTS requires PG >= 9.6
|
||||
// So we can catch the error if column exists just like sqlite logic
|
||||
try {
|
||||
await client.query(`ALTER TABLE profiles ADD COLUMN is_active BOOLEAN DEFAULT false`);
|
||||
} catch (err) {
|
||||
// Ignore if it already exists
|
||||
}
|
||||
|
||||
await client.query(`
|
||||
CREATE TABLE IF NOT EXISTS projects (
|
||||
id TEXT PRIMARY KEY,
|
||||
user_id TEXT REFERENCES profiles(id),
|
||||
name TEXT,
|
||||
scene_data TEXT,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`);
|
||||
|
||||
client.release();
|
||||
} catch (err) {
|
||||
console.error('Error initializing database:', err);
|
||||
}
|
||||
};
|
||||
|
||||
initDB();
|
||||
|
||||
// Wrapper functions to simulate the old sqlite API but backed by pg
|
||||
|
||||
const run = async (sql, params = []) => {
|
||||
// convert ? to $1, $2, etc.
|
||||
let paramIndex = 1;
|
||||
const pgSql = sql.replace(/\?/g, () => `$${paramIndex++}`);
|
||||
|
||||
const res = await pool.query(pgSql, params);
|
||||
// `run` in sqlite returns { id: this.lastID, changes: this.changes }
|
||||
// Postgres doesn't return lastID unless RETURNING is used, but for uuids it doesn't matter.
|
||||
return { changes: res.rowCount };
|
||||
};
|
||||
|
||||
const get = async (sql, params = []) => {
|
||||
let paramIndex = 1;
|
||||
const pgSql = sql.replace(/\?/g, () => `$${paramIndex++}`);
|
||||
|
||||
const res = await pool.query(pgSql, params);
|
||||
return res.rows[0];
|
||||
};
|
||||
|
||||
const all = async (sql, params = []) => {
|
||||
let paramIndex = 1;
|
||||
const pgSql = sql.replace(/\?/g, () => `$${paramIndex++}`);
|
||||
|
||||
const res = await pool.query(pgSql, params);
|
||||
return res.rows;
|
||||
};
|
||||
|
||||
module.exports = { db: pool, run, get, all };
|
||||
Reference in New Issue
Block a user