Merge pull request #314 from doronz88/refactor/shell

Refactor/shell

src/rpcclient/rpcclient/fs.py

@@ -328,7 +328,7 @@ def open(self, file: str, mode: str, access: int = 0o777) -> File:
             'r+' - read and write. exception if file doesn't exist
             'rw' - read and write. create if it doesn't exist. also truncate.
             'w' - write only. create if it doesn't exist. also truncate.
-            'w+' - read and write. create if doesn't exist.
+            'w+' - read and write. create if it doesn't exist.
         :param access: access mode as octal value
         :return: a context manager file object
         """

src/rpcclient/rpcclient/xonshrc.py

@@ -3,6 +3,7 @@
 import os
 import plistlib
 import posixpath
+import shutil
 import sys
 import tempfile
 import time
@@ -204,7 +205,7 @@ def __init__(self):
         * VolDown: ControlShiftDown
         ''')
 
-    def _register_arg_parse_alias(self, name: str, handler: Callable):
+    def _register_arg_parse_alias(self, name: str, handler: Union[Callable, str]):
         handler = ArgParserAlias(func=handler, has_args=True, prog=name)
         self._commands[name] = handler
         if XSH.aliases.get(name):
@@ -228,6 +229,13 @@ def _rpc_connect(self):
         """
         self.client = create_client(XSH.ctx['_create_socket_cb'])
 
+        # clear all host commands except for some useful ones
+        XSH.env['PATH'].clear()
+        for cmd in ['wc', 'grep', 'egrep', 'sed', 'awk', 'print', 'yes', 'cat']:
+            executable = shutil.which(cmd)
+            if executable is not None:
+                self._register_arg_parse_alias(cmd, executable)
+
         # -- rpc
         self._register_arg_parse_alias('rpc-disconnect', self._rpc_disconnect)
         self._register_arg_parse_alias('rpc-which', self._rpc_which)