bunkerweb/ui/src/ConfigFiles.py

from os import listdir, mkdir, remove, replace, walk
from os.path import dirname, exists, join, isfile
from re import compile as re_compile
from shutil import rmtree, move as shutil_move
from typing import Tuple

from utils import path_to_dict


class ConfigFiles:
    def __init__(self, logger, db):
        self.__name_regex = re_compile(r"^[a-zA-Z0-9_-]{1,64}$")
        self.__root_dirs = [
            child["name"]
            for child in path_to_dict("/opt/bunkerweb/configs")["children"]
        ]
        self.__file_creation_blacklist = ["http", "stream"]
        self.__logger = logger
        self.__db = db

    def save_configs(self) -> str:
        custom_configs = []
        root_dirs = listdir("/opt/bunkerweb/configs")
        for (root, dirs, files) in walk("/opt/bunkerweb/configs", topdown=True):
            if (
                root != "configs"
                and (dirs and not root.split("/")[-1] in root_dirs)
                or files
            ):
                path_exploded = root.split("/")
                for file in files:
                    with open(join(root, file), "r") as f:
                        custom_configs.append(
                            {
                                "value": f.read(),
                                "exploded": (
                                    f"{path_exploded.pop()}"
                                    if path_exploded[-1] not in root_dirs
                                    else "",
                                    path_exploded[-1],
                                    file.replace(".conf", ""),
                                ),
                            }
                        )

        ret = self.__db.save_custom_configs(custom_configs, "ui")
        if ret:
            self.__logger.error(f"Could not save custom configs: {ret}")
            return "Couldn't save custom configs to database"

        return ""

    def check_name(self, name: str) -> bool:
        return self.__name_regex.match(name)

    def check_path(self, path: str, root_path: str = "/opt/bunkerweb/configs/") -> str:
        root_dir: str = path.split("/")[4]
        if not (
            path.startswith(root_path)
            or root_path == "/opt/bunkerweb/configs/"
            and path.startswith(root_path)
            and root_dir in self.__root_dirs
            and (
                not path.endswith(".conf")
                or root_dir not in self.__file_creation_blacklist
                or len(path.split("/")) > 5
            )
        ):
            return f"{path} is not a valid path"

        if root_path == "/opt/bunkerweb/configs/":
            dirs = path.split("/")[5:]
            nbr_children = len(dirs)
            dirs = "/".join(dirs)
            if len(dirs) > 1:
                for x in range(nbr_children - 1):
                    if not exists(
                        f"{root_path}{root_dir}/{'/'.join(dirs.split('/')[0:-x])}"
                    ):
                        return f"{root_path}{root_dir}/{'/'.join(dirs.split('/')[0:-x])} doesn't exist"

        return ""

    def delete_path(self, path: str) -> Tuple[str, int]:
        try:
            if isfile(path):
                remove(path)
            else:
                rmtree(path)
        except OSError:
            return f"Could not delete {path}", 1

        return f"{path} was successfully deleted", 0

    def create_folder(self, path: str, name: str) -> Tuple[str, int]:
        folder_path = join(path, name)
        try:
            mkdir(folder_path)
        except OSError:
            return f"Could not create {folder_path}", 1

        return f"The folder {folder_path} was successfully created", 0

    def create_file(self, path: str, name: str, content: str) -> Tuple[str, int]:
        file_path = join(path, name)
        mkdir(path, exist_ok=True)

        with open(file_path, "w") as f:
            f.write(content)

        return f"The file {file_path} was successfully created", 0

    def edit_folder(self, path: str, name: str, old_name: str) -> Tuple[str, int]:
        new_folder_path = dirname(join(path, name))
        old_folder_path = dirname(join(path, old_name))

        if old_folder_path == new_folder_path:
            return (
                f"{old_folder_path} was not renamed because the name didn't change",
                0,
            )

        try:
            shutil_move(old_folder_path, new_folder_path)
        except OSError:
            return f"Could not move {old_folder_path}", 1

        return (
            f"The folder {old_folder_path} was successfully renamed to {new_folder_path}",
            0,
        )

    def edit_file(
        self, path: str, name: str, old_name: str, content: str
    ) -> Tuple[str, int]:
        new_path = dirname(join(path, name))
        old_path = dirname(join(path, old_name))

        try:
            with open(old_path, "r") as f:
                file_content = f.read()
        except FileNotFoundError:
            return f"Could not find {old_path}", 1

        if old_path == new_path and file_content == content:
            return (
                f"{old_path} was not edited because the content and the name didn't change",
                0,
            )
        elif file_content == content:
            try:
                replace(path, new_path)
                return f"{old_path} was successfully renamed to {new_path}", 0
            except OSError:
                return f"Could not rename {old_path} into {new_path}", 1
        elif old_path == new_path:
            new_path = old_path
        else:
            try:
                remove(old_path)
            except OSError:
                return f"Could not remove {old_path}", 1

        with open(new_path, "w") as f:
            f.write(content)

        return f"The file {old_path} was successfully edited", 0