From 04c250fed5a0c7fb97be175b0d69ef514a5f40d6 Mon Sep 17 00:00:00 2001
From: Hide_D <hided62@gmail.com>
Date: Sun, 28 Dec 2025 12:10:39 +0000
Subject: [PATCH] =?UTF-8?q?feat:=20=EB=8D=B0=EC=9D=B4=ED=84=B0=EB=B2=A0?=
 =?UTF-8?q?=EC=9D=B4=EC=8A=A4=20=EB=B0=8F=20Redis=20=EC=97=B0=EA=B2=B0=20?=
 =?UTF-8?q?=ED=85=8C=EC=8A=A4=ED=8A=B8=20=EC=8A=A4=ED=81=AC=EB=A6=BD?=
 =?UTF-8?q?=ED=8A=B8=20=EC=B6=94=EA=B0=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tools/test-dbms/check-connections.mts | 148 ++++++++++++++++++++++++++
 1 file changed, 148 insertions(+)
 create mode 100644 tools/test-dbms/check-connections.mts

diff --git a/tools/test-dbms/check-connections.mts b/tools/test-dbms/check-connections.mts
new file mode 100644
index 0000000..d700849
--- /dev/null
+++ b/tools/test-dbms/check-connections.mts
@@ -0,0 +1,148 @@
+import fs from 'node:fs/promises';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { createRequire } from 'node:module';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const DEFAULT_ENV_FILE = path.resolve(__dirname, '..', '..', '.env.ci');
+const COMMON_PACKAGE_JSON = path.resolve(
+    __dirname,
+    '..',
+    '..',
+    'packages',
+    'common',
+    'package.json'
+);
+const commonRequire = createRequire(COMMON_PACKAGE_JSON);
+const { PrismaClient } = commonRequire('@prisma/client');
+const { createClient } = commonRequire('redis');
+
+const parseEnvFile = (rawText) => {
+    const env = {};
+    const lines = rawText.split(/\r?\n/);
+    for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed || trimmed.startsWith('#')) {
+            continue;
+        }
+        const index = trimmed.indexOf('=');
+        if (index < 0) {
+            continue;
+        }
+        const key = trimmed.slice(0, index).trim();
+        let value = trimmed.slice(index + 1).trim();
+        if (
+            (value.startsWith('"') && value.endsWith('"')) ||
+            (value.startsWith("'") && value.endsWith("'"))
+        ) {
+            value = value.slice(1, -1);
+        }
+        env[key] = value;
+    }
+    return env;
+};
+
+const loadEnvFile = async (envFile) => {
+    const text = await fs.readFile(envFile, 'utf8');
+    return parseEnvFile(text);
+};
+
+const maskUrlPassword = (url) => url.replace(/:(?:[^@]+)@/, ':***@');
+
+const resolveDatabaseUrl = (env) => {
+    if (process.env.DATABASE_URL) {
+        return process.env.DATABASE_URL;
+    }
+    if (env.DATABASE_URL) {
+        return env.DATABASE_URL;
+    }
+    const host = env.POSTGRES_HOST ?? '127.0.0.1';
+    const port = env.POSTGRES_PORT ?? '15432';
+    const user = env.POSTGRES_USER ?? 'sammo';
+    const password = env.POSTGRES_PASSWORD ?? '';
+    const dbName = env.POSTGRES_DB ?? 'sammo';
+    return `postgresql://${user}:${password}@${host}:${port}/${dbName}?schema=public`;
+};
+
+const resolveRedisUrl = (env) => {
+    if (process.env.REDIS_URL) {
+        return process.env.REDIS_URL;
+    }
+    if (env.REDIS_URL) {
+        return env.REDIS_URL;
+    }
+    const host = env.REDIS_HOST ?? '127.0.0.1';
+    const port = env.REDIS_PORT ?? '16379';
+    const db = env.REDIS_DB ?? '0';
+    const password = env.REDIS_PASSWORD ?? '';
+    return `redis://:${password}@${host}:${port}/${db}`;
+};
+
+const testPostgres = async (databaseUrl) => {
+    const prisma = new PrismaClient({
+        datasources: {
+            db: {
+                url: databaseUrl,
+            },
+        },
+    });
+
+    await prisma.$connect();
+    await prisma.$queryRawUnsafe('SELECT 1');
+    await prisma.$disconnect();
+};
+
+const testRedis = async (redisUrl) => {
+    const client = createClient({ url: redisUrl });
+    await client.connect();
+    await client.ping();
+    await client.quit();
+};
+
+const main = async () => {
+    const envFile = process.env.SAMMO_ENV_FILE ?? DEFAULT_ENV_FILE;
+    let env = {};
+
+    try {
+        env = await loadEnvFile(envFile);
+    } catch (error) {
+        if (error?.code !== 'ENOENT') {
+            throw error;
+        }
+    }
+
+    const databaseUrl = resolveDatabaseUrl(env);
+    const redisUrl = resolveRedisUrl(env);
+
+    console.log(`Using env file: ${envFile}`);
+    console.log(`DATABASE_URL: ${maskUrlPassword(databaseUrl)}`);
+    console.log(`REDIS_URL: ${maskUrlPassword(redisUrl)}`);
+
+    const errors = [];
+
+    try {
+        await testPostgres(databaseUrl);
+        console.log('Postgres connection: OK');
+    } catch (error) {
+        console.error('Postgres connection: FAILED');
+        console.error(error?.message ?? error);
+        errors.push('postgres');
+    }
+
+    try {
+        await testRedis(redisUrl);
+        console.log('Redis connection: OK');
+    } catch (error) {
+        console.error('Redis connection: FAILED');
+        console.error(error?.message ?? error);
+        errors.push('redis');
+    }
+
+    if (errors.length > 0) {
+        process.exitCode = 1;
+    }
+};
+
+await main();