123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
import os
|
|
from re import compile as re_compile
|
|
from shutil import rmtree, move as shutil_move
|
|
from typing import Tuple
|
|
|
|
from ui.utils import path_to_dict
|
|
|
|
|
|
class ConfigFiles:
|
|
def __init__(self):
|
|
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"]
|
|
|
|
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 os.path.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 os.path.isfile(path):
|
|
os.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 = os.path.join(path, name)
|
|
try:
|
|
os.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 = os.path.join(path, name)
|
|
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) -> Tuple[str, int]:
|
|
new_folder_path = os.path.dirname(os.path.join(path, name))
|
|
|
|
if path == new_folder_path:
|
|
return (
|
|
f"{path} was not renamed because the name didn't change",
|
|
0,
|
|
)
|
|
|
|
try:
|
|
shutil_move(path, new_folder_path)
|
|
except OSError:
|
|
return f"Could not move {path}", 1
|
|
|
|
return f"The folder {path} was successfully renamed to {new_folder_path}", 0
|
|
|
|
def edit_file(self, path: str, name: str, content: str) -> Tuple[str, int]:
|
|
new_path = os.path.dirname(os.path.join(path, name))
|
|
try:
|
|
with open(path, "r") as f:
|
|
file_content = f.read()
|
|
except FileNotFoundError:
|
|
return f"Could not find {path}", 1
|
|
|
|
if path == new_path and file_content == content:
|
|
return (
|
|
f"{path} was not edited because the content and the name didn't change",
|
|
0,
|
|
)
|
|
elif file_content == content:
|
|
try:
|
|
os.replace(path, new_path)
|
|
return f"{path} was successfully renamed to {new_path}", 0
|
|
except OSError:
|
|
return f"Could not rename {path} into {new_path}", 1
|
|
elif path == new_path:
|
|
new_path = path
|
|
else:
|
|
try:
|
|
os.remove(path)
|
|
except OSError:
|
|
return f"Could not remove {path}", 1
|
|
|
|
with open(new_path, "w") as f:
|
|
f.write(content)
|
|
|
|
return f"The file {path} was successfully edited", 0
|