fix spaces

r2env/core.py

@@ -90,8 +90,11 @@ class R2Env:
         with open("version.txt", "r") as version:
             return version.read()
 
-    def shell(self):
-        pass
+    def shell(self, cmd=""):
+        line = "export PS1=\"r2env\\$ \";export PATH=\"" + self._r2env_path + "/bin:$PATH\"; $SHELL -f"
+        if cmd.strip() == "":
+            return os.system(line)
+        return os.system(line + " -c '" + cmd + "'")
 
     @staticmethod
     def _load_config():

r2env/repl.py

@@ -28,13 +28,15 @@ def show_help():
     print_console(HELP_MESSAGE)
 
 
-actions_with_argument = ["install", "uninstall"]
+actions_with_argument = ["install", "uninstall", "use"]
+actions_with_arguments = ["shell"]
 actions = {
     "init": R2Env().init,
     "version": R2Env().get_r2_path,
     "config": R2Env().show_config,
     "list": R2Env().list_available_packages,
     "install": R2Env().install,
+    "shell": R2Env().shell,
     "uninstall": R2Env().uninstall,
     "use": R2Env().use,
     "versions": R2Env().list_installed_packages,
@@ -45,9 +47,10 @@ actions = {
 def run_action(action, args, use_meson):
     if action not in actions:
         print_console("[X] Action not found", ERROR)
-        return
-    if action in actions_with_argument:
-        exit_if_not_argument_is_set(args)
+    elif action in actions_with_arguments:
+        actions[action](" ".join(args))
+    elif action in actions_with_argument:
+        exit_if_not_argument_is_set(args, action)
         if action == "install":
             actions[action](args[0], use_meson=use_meson)
         else:
@@ -56,8 +59,13 @@ def run_action(action, args, use_meson):
         actions[action]()
 
 
-def exit_if_not_argument_is_set(args):
+def exit_if_not_argument_is_set(args, action):
     if len(args) < 1:
+        if action == "use":
+            print_console("[x] Package not defined. Please use 'r2env use' with one installed package. ", ERROR)
+            actions["versions"]()
+        else:
+            print_console("[x] Missing package argument. ( as for example: radare2@latest)", ERROR)
         print_console("[x] Missing package argument. ( as for example: radare2@latest)", ERROR)
         sys.exit(-1)