mirror of
https://github.com/m4rcel-lol/m5rcode.git
synced 2025-12-06 19:13:57 +05:30
271 lines
11 KiB
Python
271 lines
11 KiB
Python
import os
|
|
import cmd
|
|
from colorama import init, Fore, Style
|
|
from pyfiglet import Figlet
|
|
import time
|
|
|
|
# Import Discord RPC library
|
|
from pypresence import Presence, exceptions
|
|
|
|
# Import your existing command modules
|
|
from commands.cmd_new import NewCommand
|
|
from commands.cmd_nano import NanoCommand
|
|
from commands.cmd_run import RunCommand
|
|
from commands.cmd_fastfetch import FastfetchCommand
|
|
from commands.cmd_credits import CreditsCommand
|
|
from commands.cmd_cd import CdCommand
|
|
from commands.cmd_exit import ExitCommand
|
|
from commands.cmd_wdir import WdirCommand # NEW
|
|
from commands.cmd_dir import DirCommand # NEW
|
|
|
|
# Initialize colorama for colored terminal output
|
|
init(autoreset=True)
|
|
|
|
# --- Discord RPC Configuration ---
|
|
CLIENT_ID = '1414669512158220409' # Your Discord Application ID
|
|
|
|
class M5RShell(cmd.Cmd):
|
|
"""
|
|
An interactive shell for m5rcode projects with Discord Rich Presence integration.
|
|
"""
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.base_dir = os.path.join(os.path.expanduser("~"), "m5rcode", "files")
|
|
os.makedirs(self.base_dir, exist_ok=True)
|
|
|
|
self.cwd = self.base_dir
|
|
os.chdir(self.cwd)
|
|
|
|
self.update_prompt()
|
|
|
|
# --- Discord RPC Initialization ---
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
self._connect_rpc()
|
|
|
|
if self.rpc_active:
|
|
self._set_idle_presence()
|
|
|
|
def _connect_rpc(self):
|
|
"""Attempts to connect to Discord RPC."""
|
|
try:
|
|
self.rpc = Presence(CLIENT_ID)
|
|
self.rpc.connect()
|
|
self.rpc_active = True
|
|
except exceptions.DiscordNotFound:
|
|
print(Fore.YELLOW + "[RPC] Discord client not found. Rich Presence will not be active." + Style.RESET_ALL)
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
except Exception as e:
|
|
print(Fore.RED + f"[RPC Error] Could not connect to Discord RPC: {e}" + Style.RESET_ALL)
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
|
|
def _set_idle_presence(self):
|
|
"""Sets the RPC status to 'Using the shell'."""
|
|
if self.rpc_active and self.rpc:
|
|
try:
|
|
self.rpc.update(
|
|
details="Using the shell",
|
|
state="Waiting for commands...",
|
|
small_image="shell",
|
|
small_text="Shell (Idle)"
|
|
)
|
|
except Exception as e:
|
|
print(Fore.RED + f"[RPC Error] Could not update idle presence: {e}" + Style.RESET_ALL)
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
|
|
def _set_editing_presence(self, filename):
|
|
"""Sets the RPC status to 'Editing [filename]'."""
|
|
if self.rpc_active and self.rpc:
|
|
try:
|
|
self.rpc.update(
|
|
details=f"Editing {filename}",
|
|
state="In editor",
|
|
small_image="shell_editing",
|
|
small_text="Editing File"
|
|
)
|
|
except Exception as e:
|
|
print(Fore.RED + f"[RPC Error] Could not update editing presence: {e}" + Style.RESET_ALL)
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
|
|
def _set_running_presence(self, command_name):
|
|
"""Sets the RPC status to 'Running [command_name]'."""
|
|
if self.rpc_active and self.rpc:
|
|
try:
|
|
self.rpc.update(
|
|
details=f"Running {command_name}",
|
|
state="Executing command",
|
|
small_image="shell_running",
|
|
small_text="Command Running"
|
|
)
|
|
except Exception as e:
|
|
print(Fore.RED + f"[RPC Error] Could not update running presence: {e}" + Style.RESET_ALL)
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
|
|
def _clear_presence(self):
|
|
"""Clears the current RPC status."""
|
|
if self.rpc_active and self.rpc:
|
|
try:
|
|
self.rpc.clear()
|
|
except Exception as e:
|
|
print(Fore.RED + f"[RPC Error] Could not clear presence: {e}" + Style.RESET_ALL)
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
|
|
def _close_rpc(self):
|
|
"""Closes the Discord RPC connection."""
|
|
if self.rpc_active and self.rpc:
|
|
try:
|
|
self.rpc.close()
|
|
except Exception as e:
|
|
print(Fore.RED + f"[RPC Error] Could not close RPC: {e}" + Style.RESET_ALL)
|
|
finally:
|
|
self.rpc_active = False
|
|
self.rpc = None
|
|
|
|
def update_prompt(self):
|
|
"""Refresh the prompt to reflect the current working directory."""
|
|
self.prompt = (
|
|
Fore.CYAN
|
|
+ "╭─[" + Fore.MAGENTA + "m5rcode" + Fore.CYAN + "] "
|
|
+ Fore.YELLOW + self.cwd
|
|
+ "\n╰─> "
|
|
+ Style.RESET_ALL
|
|
)
|
|
|
|
def preloop(self):
|
|
"""Called once before the command loop starts."""
|
|
self._print_banner()
|
|
if self.rpc_active:
|
|
self._set_idle_presence()
|
|
|
|
def _print_banner(self):
|
|
"""Prints the initial shell banner."""
|
|
print(Fore.GREEN + "╔" + "═" * 50 + "╗")
|
|
ascii_art = Figlet(font='slant')
|
|
print(Fore.CYAN + ascii_art.renderText("m5rcode"))
|
|
print(Fore.GREEN + "║ Welcome to m5rcode shell! Type help or ? to list commands.")
|
|
print("╚" + "═" * 50 + "╝" + Style.RESET_ALL)
|
|
|
|
def postcmd(self, stop, line):
|
|
"""Called after a command has been executed. Resets RPC to idle state."""
|
|
if self.rpc_active:
|
|
self._set_idle_presence()
|
|
return stop
|
|
|
|
def emptyline(self):
|
|
"""Do nothing on empty input line."""
|
|
pass
|
|
|
|
def default(self, line):
|
|
"""Called when the command is not recognized."""
|
|
print(Fore.RED + Style.BRIGHT + f"Error: Command '{line}' not found." + Style.RESET_ALL)
|
|
print(Fore.YELLOW + "Type 'help' or '?' to see available commands." + Style.RESET_ALL)
|
|
|
|
def do_help(self, arg):
|
|
"""Displays help information for commands."""
|
|
if arg:
|
|
super().do_help(arg)
|
|
else:
|
|
print("\n" + Fore.LIGHTCYAN_EX + "─" * 40)
|
|
print(Fore.LIGHTCYAN_EX + " " * 15 + "M5R COMMANDS")
|
|
print(Fore.LIGHTCYAN_EX + "─" * 40 + Style.RESET_ALL)
|
|
|
|
print(Fore.YELLOW + "\n── File/Project Commands ──" + Style.RESET_ALL)
|
|
self._print_command_help("new", "Create a new .m5r file")
|
|
self._print_command_help("nano", "Edit a file with your editor")
|
|
self._print_command_help("run", "Run a .m5r script (executes only Python blocks)")
|
|
|
|
print(Fore.YELLOW + "\n── Information Commands ──" + Style.RESET_ALL)
|
|
self._print_command_help("fastfetch", "Show language & system info")
|
|
self._print_command_help("credits", "Show project credits")
|
|
|
|
print(Fore.YELLOW + "\n── Navigation & Utility ──" + Style.RESET_ALL)
|
|
self._print_command_help("cd", "Change directory within m5rcode/files")
|
|
self._print_command_help("dir", "List files in the current or given directory") # NEW
|
|
self._print_command_help("wdir", "List directory of a website by URL") # NEW
|
|
self._print_command_help("clear", "Clear the shell output")
|
|
self._print_command_help("exit", "Exit the m5rcode shell")
|
|
self._print_command_help("help", "Display this help message")
|
|
self._print_command_help("?", "Alias for 'help'")
|
|
|
|
print("\n" + Fore.LIGHTCYAN_EX + "─" * 40 + Style.RESET_ALL)
|
|
print(Fore.LIGHTBLACK_EX + "For detailed help on a command, type: " + Fore.CYAN + "help <command>" + Style.RESET_ALL)
|
|
print(Fore.LIGHTCYAN_EX + "─" * 40 + Style.RESET_ALL + "\n")
|
|
|
|
def _print_command_help(self, command, description):
|
|
"""Helper to print a single command's help in a formatted way."""
|
|
print(f" {Fore.GREEN}{command:<12}{Style.RESET_ALL} {Fore.WHITE}→ {description}{Style.RESET_ALL}")
|
|
|
|
def do_clear(self, arg):
|
|
"""clear → Clear the shell output"""
|
|
os.system('cls' if os.name == 'nt' else 'clear')
|
|
self._print_banner()
|
|
self.update_prompt()
|
|
|
|
# ── File/project commands ────────────────────────────────────────────────
|
|
def do_new(self, arg):
|
|
"""new <filename> → create a new .m5r file"""
|
|
if self.rpc_active:
|
|
self._set_running_presence("new file")
|
|
NewCommand(self.cwd, arg.strip()).run()
|
|
|
|
def do_nano(self, arg):
|
|
"""nano <filename> → edit a file with your editor"""
|
|
filename = arg.strip()
|
|
if self.rpc_active:
|
|
self._set_editing_presence(filename)
|
|
NanoCommand(self.cwd, filename).run()
|
|
|
|
def do_run(self, arg):
|
|
"""run <filename> → run a .m5r script (executes only Python blocks)"""
|
|
if self.rpc_active:
|
|
self._set_running_presence(f"script {arg.strip()}")
|
|
RunCommand(self.cwd, arg.strip()).run()
|
|
|
|
def do_fastfetch(self, arg):
|
|
"""fastfetch → show language & system info"""
|
|
if self.rpc_active:
|
|
self._set_running_presence("fastfetch")
|
|
FastfetchCommand().run()
|
|
|
|
def do_credits(self, arg):
|
|
"""credits → show project credits"""
|
|
if self.rpc_active:
|
|
self._set_running_presence("credits")
|
|
CreditsCommand().run()
|
|
|
|
# ── Navigation & exit ───────────────────────────────────────────────────
|
|
def do_cd(self, arg):
|
|
"""cd <directory> → change directory within m5rcode/files"""
|
|
if self.rpc_active:
|
|
self._set_running_presence(f"changing directory to {arg.strip()}")
|
|
CdCommand(self.base_dir, [self], arg).run()
|
|
os.chdir(self.cwd)
|
|
self.update_prompt()
|
|
|
|
def do_dir(self, arg):
|
|
"""dir [path] → list files in a directory"""
|
|
if self.rpc_active:
|
|
self._set_running_presence(f"dir {arg.strip() or self.cwd}")
|
|
DirCommand(self.cwd, arg.strip()).run()
|
|
|
|
def do_wdir(self, arg):
|
|
"""wdir <url> → list directory of a website"""
|
|
if self.rpc_active:
|
|
self._set_running_presence(f"wdir {arg.strip()}")
|
|
WdirCommand(arg).run()
|
|
|
|
def do_exit(self, arg):
|
|
"""exit → exit the m5rcode shell"""
|
|
self._clear_presence()
|
|
self._close_rpc()
|
|
return ExitCommand().run()
|
|
|
|
if __name__ == "__main__":
|
|
M5RShell().cmdloop()
|