files.py

from pymongo import MongoClient
import time
from uuid import uuid4

from requests import delete

"""

Meower Files Module

This module provides filesystem functionality and a primitive JSON-file based database interface.
This file should be modified/refactored to interact with a JSON-friendly database server instead of filesystem directories and files.

"""

class Files:
    def __init__(self, logger, errorhandler):
        self.log = logger
        self.errorhandler = errorhandler

        mongo_ip = "mongodb://localhost:27017"
        self.log("Connecting to database '{0}'\n(If it seems like the server is stuck or the server randomly crashes, it probably means it couldn't connect to the database)".format(mongo_ip))
        self.db = MongoClient(mongo_ip)["meowerserver"]

        # Check connection status
        if self.db.client.get_database("meowerserver") == None:
            self.log("Failed to connect to MongoDB database!")
        else:
            self.log("Connected to database")

        # Create database collections
        for item in ["config", "usersv0", "usersv1", "netlog", "posts", "chats", "reports"]:
            if not item in self.db.list_collection_names():
                self.log("Creating collection {0}".format(item))
                self.db.create_collection(name=item)
        
        # Create collection indexes
        self.db["netlog"].create_index("users")
        self.db["usersv0"].create_index("lower_username")
        self.db["posts"].create_index("u")
        self.db["posts"].create_index("post_origin")
        self.db["posts"].create_index("type")
        self.db["posts"].create_index("p")
        self.db["chats"].create_index("members")
        
        # Create reserved accounts
        for username in ["Server", "Deleted", "Meower", "Admin", "username"]:
            self.create_item("usersv0", username, {
                "lower_username": username.lower(),
                "created": int(time.time()),
                "uuid": str(uuid4()),
                "unread_inbox": False,
                "theme": "",
                "mode": None,
                "sfx": None,
                "debug": None,
                "bgm": None,
                "bgm_song": None,
                "layout": None,
                "pfp_data": None,
                "quote": None,
                "email": None,
                "pswd": None,
                "tokens": [],
                "lvl": None,
                "banned": False,
                "last_ip": None
            })

        # Create IP banlist file
        self.create_item("config", "IPBanlist", {
            "wildcard": [],
            "users": {}
        })
        
        # Create Version support file
        self.create_item("config", "supported_versions", {
            "index": [
                "scratch-beta-5-r7",
            ]
        })
        
        # Create Trust Keys file
        self.create_item("config", "trust_keys", {
            "index": [
                "meower",
            ]
        })

        # Create Filter file
        self.create_item("config", "filter", {
            "whitelist": [], 
            "blacklist": []
        })

        # Create status file
        self.create_item("config", "status", {
            "repair_mode": False,
            "is_deprecated": False
        })

        self.log("Files initialized!")

    def does_item_exist(self, collection, id):
        if collection in self.db.list_collection_names():
            if self.db[collection].find_one({"_id": id}) != None:
                return True
            else:
                return False
        else:
            return False
             
    def create_item(self, collection, id, data):
        if collection in self.db.list_collection_names():
            if not self.does_item_exist(collection, id):
                data["_id"] = id
                self.db[collection].insert_one(data)
                return True
            else:
                self.log("{0} already exists in {1}".format(id, collection))
                return False
        else:
            self.log("{0} collection doesn't exist".format(collection))
            return False

    def update_item(self, collection, id, data):
        if collection in self.db.list_collection_names():
            if self.does_item_exist(collection, id):
                self.db[collection].update_one({"_id": id}, {"$set": data})
                return True
            else:
                return False
        else:
            return False

    def write_item(self, collection, id, data):
        if collection in self.db.list_collection_names():
            if self.does_item_exist(collection, id):
                data["_id"] = id
                self.db[collection].find_one_and_replace({"_id": id}, data)
                return True
            else:
                return False
        else:
            return False

    def load_item(self, collection, id):
        if collection in self.db.list_collection_names():
            if self.does_item_exist(collection, id):
                return True, self.db[collection].find_one({"_id": id})
            else:
                return False, None
        else:
            return False, None

    def find_items(self, collection, query):
        if collection in self.db.list_collection_names():
            payload = []
            for item in self.db[collection].find(query):
                payload.append(item["_id"])
            return payload
        else:
            return []

    def count_items(self, collection, query):
        if collection in self.db.list_collection_names():
            return self.db[collection].count_documents(query)
        else:
            return 0

    def delete_item(self, collection, id):
        if collection in self.db.list_collection_names():
            if self.does_item_exist(collection, id):
                self.db[collection].delete_one({"_id": id})
                return True
            else:
                return False
        else:
            return False