main.py

# pegasusOS - modified by Tanishq Agarwal @047pegasus
# smolOS by Krzysztof Krystian Jankowski
# Homepage: http://github.com/047pegasus/pegasusOS

import machine
import uos
import gc
import utime

class pegasusOS:
    def __init__(self):
        self.name="pegasusOS"
        self.version = "0.8.1a-xiao"
        self.board = "Seeed XIAO RP2040"
        self.cpu_speed_range = {"slow":20,"turbo":133} # Mhz
        self.system_led = machine.Pin(25,machine.Pin.OUT)

        self.prompt = "\nsmol $: "
        self.turbo = False
        self.thread_running = False
        self.protected_files = { "boot.py","main.py","ws2812.py" }
        self.user_commands = {
            "help": self.help,
            "list": self.ls,
            "show": self.cat,
            "remove": self.rm,
            "clear": self.cls,
            "stats": self.stats,
            "turbo": self.toggle_turbo,
            "edit": self.ed,
            "info": self.info,
            "py": self.py,
            "led": self.led,
            "exe": self.exe
        }
        self.user_commands_manual = {
            "ls": "list files",
            "show <filename>": "print filename content",
            "info <filename>": "information about a file",
            "remove <filename>": "remove a file (be careful!)",
            "edit <filename>": "text editor, filename is optional",
            "clear": "clears the screen",
            "turbo": "toggles turbo mode (100% vs 50% CPU speed)",
            "stats": "system statistics",
            "py <filename>": "runs user program",
            "led <command>": "manipulating on-board LED. Commands: `on`, `off`.",
            "exe <code>": "Running exec(code)."
        }
        self.ed_commands_manual = {
            "help": "this help",
            "n": "next page",
            "b ": "back one page",
            "10 <line of text>": "replacing 10-th line with a line of text",
            "a": "append new line at the end of a file",
            "write or save": "write changes to a file (not implemented yet)",
            "quit": "quit"
        }

        self.boot()

    def boot(self):
        machine.freq(self.cpu_speed_range["turbo"] * 1000000)
        self.cls()
        self.welcome()
        self.led("boot")
        self.py("grid")
        while True:
            user_input = input(self.prompt)
            parts = user_input.split()
            if len(parts) > 0:
                command = parts[0]
                if command in self.user_commands:
                    if len(parts) > 1:
                        arguments = parts[1:]
                        self.user_commands[command](*arguments)
                    else:
                        self.user_commands[command]()
                else:
                    self.unknown_function()

    def banner(self):
        print("\033[1;33;44m")
        print(
                '''                         ____   _____ 
                                           / __ \ / ____|
      _ __   ___  __ _  __ _ ___ _   _ ___| |  | | (___  
     | '_ \ / _ \/ _` |/ _` / __| | | / __| |  | |\___ \ 
     | |_) |  __/ (_| | (_| \__ \ |_| \__ \ |__| |____) |
     | .__/ \___|\__, |\__,_|___/\__,_|___/\____/|_____/ 
     | |          __/ |                                  
     |_|         |___/                                   

            '''
        )
            
        print("--------------\033[1;5;7mXIAO-RP2040-EDiTiON\033[0;1;33;44m--------------\n\033[0m")

    def welcome(self):
        self.banner()
        self.stats()
        self.print_msg("Type 'help' for the pegasus manual.")

    def man(self,manual):
        for cmd,desc in manual.items():
            print("\t\033[7m"+cmd+"\033[0m -",desc)
        utime.sleep(0.5)

    def help(self):
        print(self.name+ " version "+self.version+" user commands:\n")
        self.man(self.user_commands_manual)
        print("\n\033[0;32mSystem modified by 047pegasus.")
        print("Source code available at \033[4mgithub.com/047pegasus/pegasusOS\033[0m")

    def print_err(self, error):
        print("\n\033[1;37;41m\t<!>",error,"<!>\t\033[0m")
        utime.sleep(1)

    def print_msg(self, message):
        print("\n\033[1;34;47m\t->",message,"\t\033[0m")
        utime.sleep(0.5)

    def unknown_function(self):
        self.print_err("unknown function. Try 'help'.")

    def toggle_turbo(self):
        freq = self.cpu_speed_range["turbo"]
        if self.turbo:
             freq = self.cpu_speed_range["slow"]
        machine.freq(freq * 1000000)
        self.turbo = not self.turbo
        self.print_msg("CPU speed set to "+str(freq)+" Mhz")

    def stats(self):
        print("\t\033[0mBoard:\033[1m",self.board)
        print("\t\033[0mMicroPython:\033[1m",uos.uname().release)
        print("\t\033[0m"+self.name + ":\033[1m",self.version,"(size:",uos.stat("main.py")[6],"bytes)")
        print("\t\033[0mFirmware:\033[1m",uos.uname().version)
        turbo_msg = "\033[0mIn power-saving, \033[1mslow mode\033[0m. Use `turbo` to boost speed."
        if self.turbo:
            turbo_msg = "\033[0mIn \033[1mturbo mode\033[0m. Use `turbo` again for slow mode."
        print("\t\033[0mCPU Speed:\033[1m",machine.freq()*0.000001,"MHz",turbo_msg)
        print("\t\033[0mFree memory:\033[1m",gc.mem_free(),"bytes")
        print("\t\033[0mUsed space:\033[1m",uos.stat("/")[0],"bytes")
        print("\t\033[0mFree space:\033[1m",uos.statvfs("/")[0] * uos.statvfs("/")[3],"bytes")

    def cls(self):
         print("\033[2J")

    def ls(self):
        for file in uos.listdir():
            self.info(file)

    def info(self,filename=""):
        if filename == "":
            self.print_err("No file")
            return
        additional = ""
        file_size = uos.stat(filename)[6]
        if filename in self.protected_files: additional = "protected system file"
        print("\t\033[4m"+filename+"\033[0m\t", file_size, "bytes", "\t"+additional)


    def cat(self,filename=""):
        if filename == "":
            self.print_err("Failed to open the file.")
            return
        with open(filename,'r') as file:
            content = file.read()
            print(content)

    def rm(self,filename=""):
        if filename == "":
            self.print_err("Failed to remove the file.")
            return
        if filename in self.protected_files:
            self.print_err("Can not remove system file!")
        else:
            uos.remove(filename)
            self.print_msg("File '{}' removed successfully.".format(filename))

    def py(self,filename=""):
        if filename == "":
            self.print_err("Specify a file name to run (without .py).")
            return
        exec(open(filename+".py").read())

    def exe(self,command):
        exec(command)

    def led(self,cmd="on"):
        if cmd in ("on",""):
            self.system_led.value(0)
            return
        if cmd=="off":
            self.system_led.value(1)
            return
        if cmd=="boot":
            for _ in range(4):
                self.system_led.value(0)
                utime.sleep(0.1)
                self.system_led.value(1)
                utime.sleep(0.05)
            self.system_led.value(1)
            return

    # txtEDitor
    # Minimum viable text editor
    def ed(self, filename=""):
        self.page_size = 10
        self.file_edited = False
        print("Welcome to \033[7mtxtEDitor\033[0m\nMinimum viable text editor for pegasus operating system")
        try:
            with open(filename,'r+') as file:
                if filename in self.protected_files:
                    self.print_err("Protected file. View only.")
                self.print_msg("Loaded existing "+filename+" file.")
                lines = file.readlines()
                line_count = len(lines)
                start_index = 0

                while True:
                    if start_index < line_count:
                        end_index = min(start_index + self.page_size,line_count)
                        print_lines = lines[start_index:end_index]

                        print("\033[7m    File:",filename,"Lines:",line_count," // `h` help, `b` back,`n` next page\t\033[0m")

                        for line_num,line in enumerate(print_lines,start=start_index + 1):
                            print("{}: {}".format(f"{line_num:03}",line.strip()))

                    user_ed_input = input("\ned $: ")

                    if user_ed_input =="quit":
                        if self.file_edited:
                            self.print_msg("file was edited, `save` it first or write `quit!`")
                        else:
                            self.print_msg("txtEDitor closed")
                            break

                    if user_ed_input == "quit!":
                        self.print_msg("txtEDitor closed")
                        break

                    if user_ed_input == "help":
                        self.man(self.ed_commands_manual)

                    if user_ed_input == "a":
                        line_count += 1
                        lines.append("")

                    if user_ed_input == "n":
                        if start_index+self.page_size < line_count:
                            start_index += self.page_size
                        else:
                            self.print_msg("There is no next page. This is the last page.")

                    if user_ed_input == "b":
                        if start_index-self.page_size >= 0:
                            start_index -= self.page_size
                        else:
                            self.print_msg("Can not go back, it is a first page already.")

                    if user_ed_input in ("save","write"):
                        if filename in self.protected_files:
                            self.print_err("Protected file")
                        else:
                            self.print_err("Saving not implemented yet")

                    parts = user_ed_input.split(" ",1)
                    if len(parts) == 2:
                        if filename in self.protected_files:
                            self.print_err("Protected file")
                        else:
                            line_number = int(parts[0])
                            new_content = parts[1]
                            self.file_edited = True

                            if line_number > 0 and line_number < line_count:
                                lines[line_number - 1] = new_content + "\n"
                            else:
                                self.print_err("Invalid line number")


        except OSError:
            if filename == "":
                self.print_err("Provide an existing file name after the `ed` command.")
            else:
                self.print_err("Failed to open the file.")

pegasus = pegasusOS()