cli: Enable starting REPL.

Author: laurensvalk

pybricksdev/cli/__init__.py

@@ -97,7 +97,7 @@ def add_parser(self, subparsers: argparse._SubParsersAction):
         parser.add_argument(
             "file",
             metavar="<file>",
-            help="path to a MicroPython script or `-` for stdin",
+            help="path to a MicroPython script, `-` for stdin, or `repl` for interactive prompt",
             type=str,
         )
         parser.add_argument(
@@ -176,15 +176,21 @@ def is_pybricks_usb(dev):
         # Connect to the address and run the script
         await hub.connect()
         try:
-            if args.file == "-":
-                with NamedTemporaryFile(suffix=".py", delete=False) as temp:
-                    temp.write(sys.stdin.buffer.read())
-                args.file = temp.name
-
-            if args.start:
-                await hub.run(args.file, args.wait)
+            # Handle builtin programs.
+            if args.file == "repl":
+                await hub.run(128)
             else:
-                await hub.download(args.file)
+                # If using stdin, save to temporary file first.
+                if args.file == "-":
+                    with NamedTemporaryFile(suffix=".py", delete=False) as temp:
+                        temp.write(sys.stdin.buffer.read())
+                    args.file = temp.name
+
+                # Download program and optionally start it.
+                if args.start:
+                    await hub.run(args.file, args.wait)
+                else:
+                    await hub.download(args.file)
         finally:
             await hub.disconnect()