import sqlite3
import json

class create_db:
    def __init__(self, token, chunk_json1, filename, full_content):
        conn = sqlite3.connect('my_database.db')
        cursor = conn.cursor()

        cursor.execute("""
            CREATE TABLE IF NOT EXISTS token_data (
                token_id TEXT PRIMARY KEY,
                chunk_data TEXT,
                filename TEXT,
                full_content TEXT
            )
        """)

        chunk_json = json.dumps(chunk_json1)

        try:
            cursor.execute(
                "INSERT INTO token_data (token_id, chunk_data, filename, full_content) VALUES (?, ?, ?, ?)",
                (token, chunk_json, filename, full_content)
            )
            conn.commit()
            print({"message": f"✅ {filename} uploaded and stored successfully"})
        except sqlite3.IntegrityError:
            print({"error": f"❌ Token already exists for: {filename}"})

        conn.close()

    @staticmethod
    def get_all_filenames():
        conn = sqlite3.connect('my_database.db')
        cursor = conn.cursor()
        cursor.execute("SELECT filename FROM token_data")
        rows = cursor.fetchall()
        conn.close()
        return {"pdfs": [{"filename": row[0]} for row in rows]}