import time


def db_set(key, value, db_conn):
    """Set a key-value pair in the database.

    Args:
        key: The key to set.
        value: The value to store.
        db_conn: Database connection.

    Returns:
        Dict with status and message.
    """
    if not db_conn:
        return {"status": "error", "error": "Database not initialized"}

    try:
        cursor = db_conn.cursor()
        cursor.execute(
            """INSERT OR REPLACE INTO kv_store (key, value, timestamp)
                          VALUES (?, ?, ?)""",
            (key, value, time.time()),
        )
        db_conn.commit()
        return {"status": "success", "message": f"Set {key}"}
    except Exception as e:
        return {"status": "error", "error": str(e)}


def db_get(key, db_conn):
    """Get a value from the database.

    Args:
        key: The key to retrieve.
        db_conn: Database connection.

    Returns:
        Dict with status and value.
    """
    if not db_conn:
        return {"status": "error", "error": "Database not initialized"}

    try:
        cursor = db_conn.cursor()
        cursor.execute("SELECT value FROM kv_store WHERE key = ?", (key,))
        result = cursor.fetchone()
        if result:
            return {"status": "success", "value": result[0]}
        else:
            return {"status": "error", "error": "Key not found"}
    except Exception as e:
        return {"status": "error", "error": str(e)}


def db_query(query, db_conn):
    """Execute a database query.

    Args:
        query: SQL query to execute.
        db_conn: Database connection.

    Returns:
        Dict with status and query results.
    """
    if not db_conn:
        return {"status": "error", "error": "Database not initialized"}

    try:
        cursor = db_conn.cursor()
        cursor.execute(query)

        if query.strip().upper().startswith("SELECT"):
            results = cursor.fetchall()
            columns = [desc[0] for desc in cursor.description] if cursor.description else []
            return {"status": "success", "columns": columns, "rows": results}
        else:
            db_conn.commit()
            return {"status": "success", "rows_affected": cursor.rowcount}
    except Exception as e:
        return {"status": "error", "error": str(e)}