Upload main.py

main.py

@@ -58,18 +58,20 @@ from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from fastapi.responses import FileResponse
 
+from pymongo import MongoClient
+
 from RyuzakiLib.hackertools.chatgpt import RendyDevChat
 from RyuzakiLib.hackertools.openai import OpenAiToken
 from RyuzakiLib.mental import BadWordsList
 
 import logging
-import database as db
 import functions as code
 
 logging.basicConfig(level=logging.ERROR)
 
 # I DON'T KNOW LIKE THIS HACKER
 load_dotenv()
+MONGO_URL = os.environ["MONGO_URL"]
 REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
 OCR_API_KEY = os.environ["OCR_API_KEY"]
 ONLY_DEVELOPER_API_KEYS = os.environ["ONLY_DEVELOPER_API_KEYS"]
@@ -82,6 +84,10 @@ SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
 DEVELOPER_ID = os.environ["DEVELOPER_ID"]
 
+client_mongo = MongoClient(MONGO_URL)
+db = client_mongo["tiktokbot"]
+collection = db["users"]
+
 description = """
 - Ryuzaki Library: [Library Here](https://github.com/TeamKillerX/RyuzakiLib)
 
@@ -110,6 +116,94 @@ def validate_api_key_only_devs(api_key: str = Header(...)):
     if api_key not in ONLY_DEVELOPER_API_KEYS:
         raise HTTPException(status_code=401, detail="Invalid API key")
 
+RAMDOM_STATUS = [
+    "civilian",
+    "wanted",
+    "undercover",
+    "rogue_agent",
+    "innocent",
+    "fugitive",
+    "covert_operator"
+]
+
+def remove_sibyl_system_banned(user_id):
+    update_doc = {
+        "sibyl_ban": None,
+        "reason_sibyl": None,
+        "is_banned_sibly": None,
+        "date_joined_sib": None,
+        "sibyl_userid": None
+    }
+    return collection.update_one({"user_id": user_id}, {"$unset": update_doc}, upsert=True)
+
+def new_sibyl_system_banned(user_id, name, reason, date_joined):
+    update_doc = {
+        "sibyl_ban": name,
+        "reason_sibyl": reason,
+        "is_banned_sibly": True,
+        "date_joined_sib": date_joined,
+        "sibyl_userid": user_id
+    }
+    return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
+
+def get_sibyl_system_banned(user_id):
+    user = collection.find_one({"user_id": user_id})
+    if user:
+        sibyl_name = user.get("sibyl_ban")
+        reason = user.get("reason_sibyl")
+        is_banned = user.get("is_banned_sibly")
+        date_joined = user.get("date_joined_sib")
+        sibyl_user_id = user.get("sibyl_userid")
+        return sibyl_name, reason, is_banned, date_joined, sibyl_user_id
+    else:
+        return None
+
+def get_all_banned():
+    banned_users = []
+
+    users = collection.find({})
+
+    for user_id in users:
+        reason = user_id.get("reason_sibyl")
+        user_id = user_id.get("sibyl_userid")
+        banned_users.append({"user_id": user_id, "reason": reason})
+    return banned_users
+
+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 new_profile_clone(
+    user_id,
+    first_name,
+    last_name=None,
+    profile_id=None,
+    bio=None
+):
+    update_doc = {
+        "first_name_2": first_name,
+        "last_name_2": last_name,
+        "profile_id_2": profile_id,
+        "bio_2": bio
+    }
+    collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
+
+def get_profile_clone(user_id):
+    user = collection.find_one({"user_id": user_id})
+    if user:
+        first_name = user.get("first_name_2")
+        last_name = user.get("last_name_2")
+        profile_id = user.get("profile_id_2")
+        bio = user.get("bio_2")
+        return first_name, last_name, profile_id, bio
+    else:
+        return None, None, None, None
+
 @app.post("/ryuzaki/profile-clone")
 def profile_clone(
     user_id: int,
@@ -122,11 +216,11 @@ def profile_clone(
     if user_id != int(DEVELOPER_ID):
         return {"status": "false", "message": "Only Developer"}
 
-    first_name, _, profile_id, _ = db.get_profile_clone(user_id)
+    first_name, _, profile_id, _ = get_profile_clone(user_id)
     if first_name and profile_id:
         return {"status": "false", "message": "User is already profile"}
     try:
-        db.new_profile_clone(
+        new_profile_clone(
             user_id=user_id,
             first_name=first_name,
             last_name=last_name,
@@ -151,7 +245,7 @@ def get_profile_clone(
     user_id: int = Query(..., description="User ID in query parameter"),
     api_key: None = Depends(validate_api_key)
 ):
-    first_name, last_name, profile_id, bio = db.get_profile_clone(user_id)
+    first_name, last_name, profile_id, bio = get_profile_clone(user_id)
     if last_name is None:
         last_name_str = ""
     else:
@@ -176,7 +270,7 @@ def get_profile_clone(
 
 @app.get("/ryuzaki/getbanlist")
 def sibyl_get_all_banlist():
-    banned_users = db.get_all_banned()
+    banned_users = get_all_banned()
     return {
         "status": "True",
         "randydev": {
@@ -199,10 +293,10 @@ def sibyl_system_delete(
     api_key: None = Depends(validate_api_key_only_devs)
 ):
     try:
-        _, _, _, _, sibyl_user_id = db.get_sibyl_system_banned(user_id)
+        _, _, _, _, sibyl_user_id = get_sibyl_system_banned(user_id)
 
         if sibyl_user_id:
-            db.remove_sibyl_system_banned(user_id)
+            remove_sibyl_system_banned(user_id)
             return {"status": "true", "message": f"Successfully removed {user_id} from the Sibyl ban list."}
         else:
             return {"status": "false", "message": "Not found user"}
@@ -220,13 +314,13 @@ def sibyl_system_ban(
 
     try:
         date_joined = str(dt.now())
-        sibyl_ban = random.choice(db.RAMDOM_STATUS)
+        sibyl_ban = random.choice(RAMDOM_STATUS)
         _, _, is_banned, _, sibyl_user_id = get_sibyl_system_banned(user_id)
 
         if sibyl_user_id and is_banned:
             return {"status": "false", "message": "User is already banned"}
 
-        db.new_sibyl_system_banned(user_id, sibyl_ban, reason, date_joined)
+        new_sibyl_system_banned(user_id, sibyl_ban, reason, date_joined)
 
         return {
             "status": "true",
@@ -247,7 +341,7 @@ def sibyl_system(
     user_id: int = Query(..., description="User ID in query parameter"),
     api_key: None = Depends(validate_api_key)
 ):
-    sibyl_name, reason, is_banned, date_joined, sibyl_user_id = db.get_sibyl_system_banned(user_id)
+    sibyl_name, reason, is_banned, date_joined, sibyl_user_id = get_sibyl_system_banned(user_id)
     if sibyl_name and reason and is_banned and date_joined and sibyl_user_id:
         return {
             "status": "true",