import os
import duckdb
import uvicorn
from fastapi import FastAPI
from huggingface_hub import hf_hub_download

app = FastAPI()

# 🚀 Global Path taaki baar-baar download na ho
DB_PATH = None

def get_db():
    global DB_PATH
    if DB_PATH is None or not os.path.exists(DB_PATH):
        token = os.getenv("HF_TOKEN")
        DB_PATH = hf_hub_download(
            repo_id="Watchhrr/Telegram",
            filename="Midnight_Master_v2.db",
            repo_type="dataset",
            token=token
        )
    return DB_PATH

@app.get("/search")
def search(q: str = None):
    if not q:
        return {"status": "error", "message": "Query 'q' missing"}
    
    try:
        path = get_db()
        # DuckDB SQLite se 10x fast hai aur RAM crash nahi karta
        con = duckdb.connect(database=':memory:')
        con.execute("INSTALL sqlite;")
        con.execute("LOAD sqlite;")
        
        # 🎯 Direct Index Search
        # Midnight_v2 format: ID;PHONE;USER;NAME
        query = f"SELECT user_id FROM sqlite_scan('{path}', 'telegram_users') WHERE user_id LIKE '{q};%' LIMIT 1"
        res = con.execute(query).fetchone()
        
        results = []
        if res:
            p = res[0].split(';')
            results.append({
                "id": p[0],
                "t": p[1] if len(p)>1 else "N/A",
                "u": p[2] if len(p)>2 else "N/A",
                "n": p[3] if len(p)>3 else "N/A"
            })
        
        return {"status": "success", "results": results}
        
    except Exception as e:
        return {"status": "error", "details": str(e)}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)