Update tools.py

tools.py

@@ -16,10 +16,15 @@ from pydantic import BaseModel
 from pymongo import MongoClient
 from models import *
 from RyuzakiLib import AsyicXSearcher
-from main import validate_api_key
 
 load_dotenv()
 TOOLS_NEW_URL = os.environ["TOOLS_NEW_URL"]
+MONGO_URL = os.environ["MONGO_URL"]
+
+client_mongo = MongoClient(MONGO_URL)
+db = client_mongo["tiktokbot"]
+collection = db["users"]
+
 router = APIRouter()
 
 class XnxxSearch(BaseModel):
@@ -28,6 +33,20 @@ class XnxxSearch(BaseModel):
 class XnxxLinks(BaseModel):
     url: str
 
+def get_all_api_keys():
+    user = collection.find({})
+    api_keys = []
+    for x in user:
+        api_key = x.get("ryuzaki_api_key")
+        if api_key:
+            api_keys.append(api_key)
+    return api_keys
+
+def validate_api_key(api_key: str = Header(...)):
+    USERS_API_KEYS = get_all_api_keys()
+    if api_key not in USERS_API_KEYS:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
 async def tools_search(
     name=None,
     parameter=None,
@@ -82,7 +101,7 @@ async def toanime_endpoint(
     try:
         with open(file_path, "wb") as f:
             f.write(await file.read())
-    except Exception as e:
+    except Exception:
         raise HTTPException(status_code=500, detail="Failed to save file")
     try:
         response = await toanime(file_path)