fastled 1.1.6__py2.py3-none-any.whl → 1.1.15__py2.py3-none-any.whl

fastled/__init__.py

@@ -0,0 +1,3 @@
+"""FastLED Wasm Compiler package."""
+
+__version__ = "1.1.15"

fastled/app.py

@@ -1,7 +1,5 @@
 """
 Uses the latest wasm compiler image to compile the FastLED sketch.
-
-
 """
 
 import argparse
@@ -15,13 +13,20 @@ import time
 from dataclasses import dataclass
 from pathlib import Path
 
+from fastled import __version__
 from fastled.build_mode import BuildMode, get_build_mode
-from fastled.compile_server import CompileServer, looks_like_fastled_repo
+from fastled.compile_server import CompileServer
 from fastled.docker_manager import DockerManager
-from fastled.filewatcher import FileChangedNotifier
+from fastled.filewatcher import FileWatcherProcess
+from fastled.keyboard import SpaceBarWatcher
 from fastled.open_browser import open_browser_thread
-from fastled.sketch import get_sketch_files
-from fastled.web_compile import web_compile
+from fastled.sketch import looks_like_fastled_repo, looks_like_sketch_directory
+from fastled.web_compile import (
+    SERVER_PORT,
+    ConnectionResult,
+    find_good_connection,
+    web_compile,
+)
 
 machine = platform.machine().lower()
 IS_ARM: bool = "arm" in machine or "aarch64" in machine
@@ -44,12 +49,15 @@ DOCKER = DockerManager(container_name=CONTAINER_NAME)
 
 def parse_args() -> argparse.Namespace:
     """Parse command-line arguments."""
-    parser = argparse.ArgumentParser(description="FastLED WASM Compiler")
+    parser = argparse.ArgumentParser(description=f"FastLED WASM Compiler {__version__}")
+    parser.add_argument(
+        "--version", action="version", version=f"%(prog)s {__version__}"
+    )
     parser.add_argument(
         "directory",
         type=str,
         nargs="?",
-        default=os.getcwd(),
+        default=None,
         help="Directory containing the FastLED sketch to compile",
     )
     parser.add_argument(
@@ -57,11 +65,6 @@ def parse_args() -> argparse.Namespace:
         action="store_true",
         help="Just compile, skip opening the browser and watching for changes.",
     )
-    parser.add_argument(
-        "--no-auto-clean",
-        action="store_true",
-        help="Big performance gains for compilation, but it's flaky at this time",
-    )
     parser.add_argument(
         "--web",
         "-w",
@@ -71,17 +74,6 @@ def parse_args() -> argparse.Namespace:
         const=DEFAULT_URL,  # Default value when --web is specified without value
         help="Use web compiler. Optional URL can be provided (default: https://fastled.onrender.com)",
     )
-    parser.add_argument(
-        "--reuse",
-        action="store_true",
-        help="Reuse the existing container if it exists. (Not available with --web)",
-    )
-    parser.add_argument(
-        "--exclude",
-        type=str,
-        nargs="+",
-        help="Additional patterns to exclude from file watching (Not available with --web)",
-    )
     parser.add_argument(
         "-i",
         "--interactive",
@@ -104,6 +96,12 @@ def parse_args() -> argparse.Namespace:
     build_mode.add_argument(
         "--release", action="store_true", help="Build in release mode"
     )
+    build_mode.add_argument(
+        "--localhost",
+        "--local",
+        action="store_true",
+        help="Use localhost for web compilation from an instance of fastled --server, creating it if necessary",
+    )
     build_mode.add_argument(
         "--server",
         action="store_true",
@@ -116,14 +114,30 @@ def parse_args() -> argparse.Namespace:
         help="Skips the test to see if the current directory is a valid FastLED sketch directory",
     )
 
-    args = parser.parse_args()
+    cwd_is_fastled = looks_like_fastled_repo(Path(os.getcwd()))
 
-    # Handle --web implications
-    if args.web:
-        if args.reuse:
-            parser.error("--reuse cannot be used with --web")
-        if args.exclude:
-            parser.error("--exclude cannot be used with --web")
+    args = parser.parse_args()
+    if not cwd_is_fastled and not args.localhost and not args.web and not args.server:
+        print(f"Using web compiler at {DEFAULT_URL}")
+        args.web = DEFAULT_URL
+    if cwd_is_fastled and not args.web:
+        print("Forcing --local mode because we are in the FastLED repo")
+        args.localhost = True
+    if args.localhost:
+        args.web = "localhost"
+    if args.interactive and not args.server:
+        print("--interactive forces --server mode")
+        args.server = True
+    if args.directory is None and not args.server:
+        # does current directory look like a sketch?
+        maybe_sketch_dir = Path(os.getcwd())
+        if looks_like_sketch_directory(maybe_sketch_dir):
+            args.directory = str(maybe_sketch_dir)
+        else:
+            print(
+                "\nYou either need to specify a sketch directory or run in --server mode."
+            )
+            sys.exit(1)
 
     return args
 
@@ -188,16 +202,33 @@ def run_web_compiler(
 
 
 def _try_start_server_or_get_url(args: argparse.Namespace) -> str | CompileServer:
-    if args.web:
+    is_local_host = "localhost" in args.web or "127.0.0.1" in args.web or args.localhost
+
+    # test to see if there is already a local host server
+    local_host_needs_server = False
+    if is_local_host:
+        addr = "localhost" if args.localhost else args.web
+        urls = [addr]
+        if ":" not in addr:
+            urls.append(f"{addr}:{SERVER_PORT}")
+
+        result: ConnectionResult | None = find_good_connection(urls)
+        if result is not None:
+            print(f"Found local server at {result.host}")
+            return result.host
+        else:
+            local_host_needs_server = True
+
+    if not local_host_needs_server and args.web:
         if isinstance(args.web, str):
             return args.web
         if isinstance(args.web, bool):
             return DEFAULT_URL
         return args.web
     else:
-        disable_auto_clean = args.no_auto_clean
         try:
-            compile_server = CompileServer(disable_auto_clean=disable_auto_clean)
+            print("No local server found, starting one...")
+            compile_server = CompileServer()
             print("Waiting for the local compiler to start...")
             if not compile_server.wait_for_startup():
                 print("Failed to start local compiler.")
@@ -210,42 +241,11 @@ def _try_start_server_or_get_url(args: argparse.Namespace) -> str | CompileServe
             return DEFAULT_URL
 
 
-def _lots_and_lots_of_files(directory: Path) -> bool:
-    return len(get_sketch_files(directory)) > 100
-
-
-def _looks_like_sketch_directory(directory: Path) -> bool:
-    if looks_like_fastled_repo(directory):
-        print("Directory looks like the FastLED repo")
-        return False
-
-    if _lots_and_lots_of_files(directory):
-        print("Too many files in the directory, bailing out")
-        return False
-
-    # walk the path and if there are over 30 files, return False
-    # at the root of the directory there should either be an ino file or a src directory
-    # or some cpp files
-    # if there is a platformio.ini file, return True
-    ino_file_at_root = list(directory.glob("*.ino"))
-    if ino_file_at_root:
-        return True
-    cpp_file_at_root = list(directory.glob("*.cpp"))
-    if cpp_file_at_root:
-        return True
-    platformini_file = list(directory.glob("platformio.ini"))
-    if platformini_file:
-        return True
-    return False
-
-
 def run_client(args: argparse.Namespace) -> int:
     compile_server: CompileServer | None = None
-    open_web_browser = not args.just_compile
+    open_web_browser = not args.just_compile and not args.interactive
     profile = args.profile
-    if not args.force_compile and not _looks_like_sketch_directory(
-        Path(args.directory)
-    ):
+    if not args.force_compile and not looks_like_sketch_directory(Path(args.directory)):
         print(
             "Error: Not a valid FastLED sketch directory, if you are sure it is, use --force-compile"
         )
@@ -324,40 +324,76 @@ def run_client(args: argparse.Namespace) -> int:
             compile_server.stop()
         return 1
 
-    # Watch mode
     print("\nWatching for changes. Press Ctrl+C to stop...")
-    watcher = FileChangedNotifier(args.directory, excluded_patterns=["fastled_js"])
-    watcher.start()
+    sketch_filewatcher = FileWatcherProcess(
+        args.directory, excluded_patterns=["fastled_js"]
+    )
+
+    source_code_watcher: FileWatcherProcess | None = None
+    if compile_server and compile_server.using_fastled_src_dir_volume():
+        assert compile_server.fastled_src_dir is not None
+        source_code_watcher = FileWatcherProcess(
+            compile_server.fastled_src_dir, excluded_patterns=[]
+        )
+
+    def trigger_rebuild_if_sketch_changed(
+        last_compiled_result: CompiledResult,
+    ) -> CompiledResult:
+        changed_files = sketch_filewatcher.get_all_changes()
+        if changed_files:
+            print(f"\nChanges detected in {changed_files}")
+            last_hash_value = last_compiled_result.hash_value
+            out = compile_function(last_hash_value=last_hash_value)
+            if not out.success:
+                print("\nRecompilation failed.")
+            else:
+                print("\nRecompilation successful.")
+            return out
+        return last_compiled_result
 
     try:
         while True:
-            try:
-                changed_files = watcher.get_all_changes()
-            except KeyboardInterrupt:
-                print("\nExiting from watcher...")
-                raise
-            except Exception as e:
-                print(f"Error getting changes: {e}")
-                changed_files = []
-            if changed_files:
-                print(f"\nChanges detected in {changed_files}")
-                last_hash_value = last_compiled_result.hash_value
-                result = compile_function(last_hash_value=last_hash_value)
-                if not result.success:
-                    print("\nRecompilation failed.")
-                else:
-                    print("\nRecompilation successful.")
-            time.sleep(0.3)
+            last_compiled_result = trigger_rebuild_if_sketch_changed(
+                last_compiled_result
+            )
+            if compile_server and not compile_server.proceess_running():
+                print("Server process is not running. Exiting...")
+                return 1
+            if source_code_watcher is not None:
+                changed_files = source_code_watcher.get_all_changes()
+                # de-duplicate changes
+                changed_files = sorted(list(set(changed_files)))
+                if changed_files:
+                    print(f"\nChanges detected in FastLED source code: {changed_files}")
+                    print("Press space bar to trigger compile.")
+
+                    space_key_watcher = SpaceBarWatcher()
+                    try:
+                        while True:
+                            if space_key_watcher.space_bar_pressed():
+                                print("Space bar pressed, triggering recompile...")
+                                last_compiled_result = compile_function(
+                                    last_hash_value=None
+                                )
+                                print("Finished recompile.")
+                                break
+                            elif len(sketch_filewatcher.get_all_changes()) > 0:
+                                last_compiled_result = compile_function(
+                                    last_hash_value=None
+                                )
+                                break
+                            time.sleep(0.1)
+                    finally:
+                        space_key_watcher.stop()
+
     except KeyboardInterrupt:
-        watcher.stop()
         print("\nStopping watch mode...")
         return 0
     except Exception as e:
-        watcher.stop()
         print(f"Error: {e}")
         return 1
     finally:
-        watcher.stop()
+        sketch_filewatcher.stop()
         if compile_server:
             compile_server.stop()
         if browser_proc:
@@ -366,11 +402,9 @@ def run_client(args: argparse.Namespace) -> int:
 
 def run_server(args: argparse.Namespace) -> int:
     interactive = args.interactive
-    compile_server = CompileServer(
-        disable_auto_clean=args.no_auto_clean, interactive=interactive
-    )
-    print(f"Server started at {compile_server.url()}")
-    compile_server.start()
+    compile_server = CompileServer(interactive=interactive)
+    if not interactive:
+        print(f"Server started at {compile_server.url()}")
     compile_server.wait_for_startup()
     try:
         while True:
@@ -388,13 +422,7 @@ def run_server(args: argparse.Namespace) -> int:
 
 def main() -> int:
     args = parse_args()
-    target_dir = Path(args.directory)
-    cwd_is_target_dir = target_dir == Path(os.getcwd())
-    force_server = cwd_is_target_dir and looks_like_fastled_repo(target_dir)
-    auto_server = (args.server or args.interactive or cwd_is_target_dir) and (
-        not args.web and not args.just_compile
-    )
-    if auto_server or force_server:
+    if args.server:
         print("Running in server only mode.")
         return run_server(args)
     else:
@@ -404,9 +432,9 @@ def main() -> int:
 
 if __name__ == "__main__":
     try:
-        sys.argv.append("examples/wasm")
-        sys.argv.append("-w")
-        sys.argv.append("localhost")
+        # os.chdir("../fastled")
+        sys.argv.append("examples/SdCard")
+        sys.argv.append("--localhost")
         sys.exit(main())
     except KeyboardInterrupt:
         print("\nExiting from main...")