import psycopg2
import sys
import os

sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))

from src.config import DATABASE_HOSTNAME, DATABASE_PORT, COLLECTOR_HOSTNAME, COLLECTOR_PORT, _DATABASE_CREDENTIAL_USER, _DATABASE_CREDENTIAL_PASSWORD, _COLLECTOR_CREDENTIAL_USER, _COLLECTOR_CREDENTIAL_PASSWORD, DATABASE_NAME, COLLECTOR_NAME

def get_production_connection():
    try:
        # Konfigurasi koneksi database produksi
        production_connection = psycopg2.connect(
            dbname=COLLECTOR_NAME,
            user=_COLLECTOR_CREDENTIAL_USER,
            password=_COLLECTOR_CREDENTIAL_PASSWORD,
            host=COLLECTOR_HOSTNAME,
            port=COLLECTOR_PORT,
        )
        return production_connection
    except Exception as e:
        print("Error saat koneksi ke database produksi:", e)
        return None

def get_connection():
    try:
        # Konfigurasi koneksi database

        # connection = psycopg2.connect(
        #     host="localhost",
        #     port=5432,
        #     database="postgres",
        #     user="postgres",
        #     password="ariwa"
        # )
        connection = psycopg2.connect(
            dbname=DATABASE_NAME,
            user=_DATABASE_CREDENTIAL_USER,
            password=_DATABASE_CREDENTIAL_PASSWORD,
            host=DATABASE_HOSTNAME,
            port=DATABASE_PORT,
        )
        connection_wo_db = psycopg2.connect(
            dbname=COLLECTOR_NAME,
            user=_COLLECTOR_CREDENTIAL_USER,
            password=_COLLECTOR_CREDENTIAL_PASSWORD,
            host=COLLECTOR_HOSTNAME,
            port=COLLECTOR_PORT,
        )
        return connection, connection_wo_db
    except Exception as e:
        print("Error saat koneksi ke database:", e)
        return None