mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
140 lines
4.8 KiB
Python
140 lines
4.8 KiB
Python
import sys
|
|
import cmd
|
|
import time
|
|
import threading
|
|
import logging
|
|
import getpass
|
|
from app.classes.shared.console import Console
|
|
from app.classes.shared.import3 import Import3
|
|
|
|
from app.classes.shared.helpers import Helpers
|
|
from app.classes.shared.tasks import TasksManager
|
|
from app.classes.shared.migration import MigrationManager
|
|
from app.classes.shared.main_controller import Controller
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class MainPrompt(cmd.Cmd):
|
|
def __init__(
|
|
self, helper, tasks_manager, migration_manager, main_controller, import3
|
|
):
|
|
super().__init__()
|
|
self.helper: Helpers = helper
|
|
self.tasks_manager: TasksManager = tasks_manager
|
|
self.migration_manager: MigrationManager = migration_manager
|
|
self.controller: Controller = main_controller
|
|
self.import3: Import3 = import3
|
|
|
|
# overrides the default Prompt
|
|
self.prompt = ""
|
|
|
|
def emptyline(self):
|
|
pass
|
|
|
|
def do_exit(self, _line):
|
|
self.tasks_manager._main_graceful_exit()
|
|
self.universal_exit()
|
|
|
|
def do_migrations(self, line):
|
|
if line == "up":
|
|
self.migration_manager.up()
|
|
elif line == "down":
|
|
self.migration_manager.down()
|
|
elif line == "done":
|
|
Console.info(self.migration_manager.done)
|
|
elif line == "todo":
|
|
Console.info(self.migration_manager.todo)
|
|
elif line == "diff":
|
|
Console.info(self.migration_manager.diff)
|
|
elif line == "info":
|
|
Console.info(f"Done: {self.migration_manager.done}")
|
|
Console.info(f"FS: {self.migration_manager.todo}")
|
|
Console.info(f"Todo: {self.migration_manager.diff}")
|
|
elif line.startswith("add "):
|
|
migration_name = line[len("add ") :]
|
|
self.migration_manager.create(migration_name, False)
|
|
else:
|
|
Console.info("Unknown migration command")
|
|
|
|
def do_set_passwd(self, line):
|
|
|
|
try:
|
|
username = str(line).lower()
|
|
# If no user is found it returns None
|
|
user_id = self.controller.users.get_id_by_name(username)
|
|
if not username:
|
|
Console.error("You must enter a username. Ex: `set_passwd admin'")
|
|
return False
|
|
if not user_id:
|
|
Console.error(
|
|
f"No user found by the name of {username} this is case sensitive"
|
|
)
|
|
return False
|
|
except:
|
|
Console.error(f"User: {line} Not Found")
|
|
return False
|
|
# get new password from user
|
|
new_pass = getpass.getpass(prompt=f"NEW password for: {username} > ")
|
|
# check to make sure it fits our requirements.
|
|
if len(new_pass) > 512:
|
|
Console.warning("Passwords must be greater than 6char long and under 512")
|
|
return False
|
|
if len(new_pass) < 6:
|
|
Console.warning("Passwords must be greater than 6char long and under 512")
|
|
return False
|
|
# grab repeated password input
|
|
new_pass_conf = getpass.getpass(prompt="Re-enter your password: > ")
|
|
|
|
# check to make sure they match
|
|
if new_pass != new_pass_conf:
|
|
Console.error("Passwords do not match. Please try again.")
|
|
return False
|
|
|
|
self.controller.users.update_user(user_id, {"password": new_pass})
|
|
|
|
@staticmethod
|
|
def do_threads(_line):
|
|
for thread in threading.enumerate():
|
|
if sys.version_info >= (3, 8):
|
|
print(
|
|
f"Name: {thread.name}\tIdentifier: "
|
|
f"{thread.ident}\tTID/PID: {thread.native_id}"
|
|
)
|
|
else:
|
|
print(f"Name: {thread.name}\tIdentifier: {thread.ident}")
|
|
|
|
def print_prompt(self):
|
|
self.stdout.write(self.prompt)
|
|
self.stdout.flush()
|
|
|
|
def do_import3(self, _line):
|
|
self.import3.start_import()
|
|
|
|
def universal_exit(self):
|
|
logger.info("Stopping all server daemons / threads")
|
|
Console.info(
|
|
"Stopping all server daemons / threads - This may take a few seconds"
|
|
)
|
|
self.helper.websocket_helper.disconnect_all()
|
|
Console.info("Waiting for main thread to stop")
|
|
while True:
|
|
if self.tasks_manager.get_main_thread_run_status():
|
|
sys.exit(0)
|
|
time.sleep(1)
|
|
|
|
def help_exit(self):
|
|
Console.help("Stops the server if running, Exits the program")
|
|
|
|
def help_migrations(self):
|
|
Console.help("Only for advanced users. Use with caution")
|
|
|
|
def help_import3(self):
|
|
Console.help("Import users and servers from Crafty 3")
|
|
|
|
def help_set_passwd(self):
|
|
Console.help("Set a user's password. Example: set_passwd admin")
|
|
|
|
def help_threads(self):
|
|
Console.help("Get all of the Python threads used by Crafty")
|