command.py

from fake_filesystem import FakeFileSystem

class CommandProcessor:
    def __init__(self):
        self.fs = FakeFileSystem()
        self.commands = {
            'help': self._help,
            'dir': self._dir,
            'cd': self._cd,
            'cls': self._cls,
            'echo': self._echo,
            'exit': self._exit,
            'type': self._type,
            'ver': self._ver
        }
    
    def get_matching_files(self, partial):
        """Restituisce i file/cartelle che corrispondono al percorso parziale"""
        current_dir = self.fs.get_current_directory()
        contents = self.fs.get_current_directory_contents()
        
        if not contents:
            return []
        
        # Se il percorso è assoluto
        if partial.startswith('C:\\'):
            base_path = 'C:\\'
            rel_path = partial[3:]
        else:
            base_path = current_dir
            rel_path = partial
        
        # Trova la directory base
        path_parts = [p for p in rel_path.split('\\') if p]
        current = self.fs.file_system['C:\\']
        
        for part in path_parts[:-1]:
            if part in current['contents'] and current['contents'][part]['type'] == 'dir':
                current = current['contents'][part]
            else:
                return []
        
        # Trova i match per l'ultima parte
        last_part = path_parts[-1] if path_parts else ''
        matches = [name for name in current['contents'].keys() if name.startswith(last_part)]
        
        return matches
    
    def process_command(self, command_str):
        if not command_str.strip():
            return
        
        parts = command_str.split()
        command = parts[0].lower()
        args = parts[1:] if len(parts) > 1 else []
        
        if command in self.commands:
            return self.commands[command](args)
        else:
            return f"'{command}' is not recognized as an internal or external command,\noperable program or batch file."

    def _help(self, args):
        help_text = """For more information on a specific command, type HELP command-name
ASSOC          Displays or modifies file extension associations.
ATTRIB         Displays or changes file attributes.
CD             Displays the name of or changes the current directory.
CLS            Clears the screen.
COPY           Copies one or more files to another location.
DIR            Displays a list of files and subdirectories in a directory.
ECHO           Displays messages, or turns command echoing on or off.
EXIT           Quits the CMD.EXE program (command interpreter).
TYPE           Displays the contents of a text file.
VER            Displays the version of Windows."""
        return help_text

    def _dir(self, args):
        contents = self.fs.get_current_directory_contents()
        if contents is None:
            return "Directory not found."
        
        output = []
        output.append(f" Directory of {self.fs.get_current_directory()}\n")
        
        dirs = []
        files = []
        
        for name, info in contents.items():
            if info['type'] == 'dir':
                dirs.append(name)
            else:
                files.append((name, info['size']))
        
        # Sort directories and files
        dirs.sort()
        files.sort()
        
        for d in dirs:
            output.append(f"{'<DIR>':14} {d}")
        
        for f, size in files:
            output.append(f"{size:14} {f}")
        
        output.append(f"\n{'{} File(s)'.format(len(files)):20} {'{} bytes'.format(sum(f[1] for f in files))}")
        output.append(f"{'{} Dir(s)'.format(len(dirs)):20} {'0 bytes free'.format()}")
        
        return '\n'.join(output)

    def _cd(self, args):
        if not args:
            return self.fs.get_current_directory()
        
        new_dir = args[0]
        success = self.fs.change_directory(new_dir)
        if not success:
            return "The system cannot find the path specified."

    def _cls(self, args):
        return "\n" * 100  # Simulate clearing the screen

    def _echo(self, args):
        return ' '.join(args)

    def _exit(self, args):
        return "EXIT"

    def _type(self, args):
        if not args:
            return "Invalid syntax."
        
        filename = args[0]
        if not filename.startswith('C:\\'):
            filename = self.fs.get_current_directory() + filename
        
        # Find the file
        path_parts = filename.split('\\')
        path_parts = [part for part in path_parts if part]  # Remove empty parts
        
        current = self.fs.file_system['C:\\']
        for part in path_parts[1:]:  # Skip 'C:'
            if part in current['contents']:
                current = current['contents'][part]
            else:
                return "File not found."
        
        if current['type'] != 'file':
            return "Access is denied."
        
        # Return some fake content based on filename
        if filename.endswith('.txt'):
            return f"This is the content of {filename}"
        elif filename.endswith('.bat'):
            return f"@echo off\nrem This is a batch file\npause"
        else:
            return "This is a binary file and cannot be displayed."

    def _ver(self, args):
        return "Microsoft Windows [Version 10.0.19041.1237]"