fastled 1.2.23__py3-none-any.whl

fastled/parse_args.py

@@ -0,0 +1,195 @@
+import argparse
+import os
+import sys
+from pathlib import Path
+
+from fastled import __version__
+from fastled.project_init import project_init
+from fastled.select_sketch_directory import select_sketch_directory
+from fastled.settings import DEFAULT_URL, IMAGE_NAME
+from fastled.sketch import (
+    find_sketch_directories,
+    looks_like_fastled_repo,
+    looks_like_sketch_directory,
+)
+
+
+def parse_args() -> argparse.Namespace:
+    """Parse command-line arguments."""
+    parser = argparse.ArgumentParser(description=f"FastLED WASM Compiler {__version__}")
+    parser.add_argument("--version", action="version", version=f"{__version__}")
+    parser.add_argument(
+        "directory",
+        type=str,
+        nargs="?",
+        default=None,
+        help="Directory containing the FastLED sketch to compile",
+    )
+    parser.add_argument(
+        "--init",
+        nargs="?",
+        const=True,
+        help="Initialize the FastLED sketch in the current directory. Optional name can be provided",
+    )
+    parser.add_argument(
+        "--just-compile",
+        action="store_true",
+        help="Just compile, skip opening the browser and watching for changes.",
+    )
+    parser.add_argument(
+        "--web",
+        "-w",
+        type=str,
+        nargs="?",
+        # const does not seem to be working as expected
+        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(
+        "-i",
+        "--interactive",
+        action="store_true",
+        help="Run in interactive mode (Not available with --web)",
+    )
+    parser.add_argument(
+        "--profile",
+        action="store_true",
+        help="Enable profiling for web compilation",
+    )
+    parser.add_argument(
+        "--force-compile",
+        action="store_true",
+        help="Skips the test to see if the current directory is a valid FastLED sketch directory",
+    )
+    parser.add_argument(
+        "--no-auto-updates",
+        action="store_true",
+        help="Disable automatic updates of the wasm compiler image when using docker.",
+    )
+    parser.add_argument(
+        "--update",
+        "--upgrade",
+        action="store_true",
+        help="Update the wasm compiler (if necessary) before running",
+    )
+    parser.add_argument(
+        "--localhost",
+        "--local",
+        "-l",
+        action="store_true",
+        help="Use localhost for web compilation from an instance of fastled --server, creating it if necessary",
+    )
+    parser.add_argument(
+        "--build",
+        "-b",
+        action="store_true",
+        help="Build the wasm compiler image from the FastLED repo",
+    )
+    parser.add_argument(
+        "--server",
+        "-s",
+        action="store_true",
+        help="Run the server in the current directory, volume mapping fastled if we are in the repo",
+    )
+    parser.add_argument(
+        "--purge",
+        action="store_true",
+        help="Remove all FastLED containers and images",
+    )
+
+    build_mode = parser.add_mutually_exclusive_group()
+    build_mode.add_argument("--debug", action="store_true", help="Build in debug mode")
+    build_mode.add_argument(
+        "--quick",
+        action="store_true",
+        default=True,
+        help="Build in quick mode (default)",
+    )
+    build_mode.add_argument(
+        "--release", action="store_true", help="Build in release mode"
+    )
+
+    cwd_is_fastled = looks_like_fastled_repo(Path(os.getcwd()))
+
+    args = parser.parse_args()
+
+    if args.purge:
+        from fastled.docker_manager import DockerManager
+
+        docker = DockerManager()
+        docker.purge(IMAGE_NAME)
+        sys.exit(0)
+
+    if args.init:
+        example = args.init if args.init is not True else None
+        try:
+            args.directory = project_init(example, args.directory)
+        except Exception as e:
+            print(f"Failed to initialize project: {e}")
+            sys.exit(1)
+        print("\nInitialized FastLED project in", args.directory)
+        print(f"Use 'fastled {args.directory}' to compile the project.")
+        sys.exit(0)
+
+    if not args.update:
+        if args.no_auto_updates:
+            args.auto_update = False
+        else:
+            args.auto_update = None
+
+        if (
+            not cwd_is_fastled
+            and not args.localhost
+            and not args.web
+            and not args.server
+        ):
+            from fastled.docker_manager import DockerManager
+
+            if DockerManager.is_docker_installed():
+                if not DockerManager.ensure_linux_containers_for_windows():
+                    print(
+                        f"Windows must be in linux containers mode, but is in Windows container mode, Using web compiler at {DEFAULT_URL}."
+                    )
+                    args.web = DEFAULT_URL
+                else:
+                    print(
+                        "Docker is installed. Defaulting to --local mode, use --web to override and use the web compiler instead."
+                    )
+                    args.localhost = True
+            else:
+                print(f"Docker is not installed. Using web compiler at {DEFAULT_URL}.")
+                args.web = DEFAULT_URL
+        if cwd_is_fastled and not args.web and not args.server:
+            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("Searching for sketch directories...")
+                sketch_directories = find_sketch_directories(maybe_sketch_dir)
+                selected_dir = select_sketch_directory(
+                    sketch_directories, cwd_is_fastled
+                )
+                if selected_dir:
+                    print(f"Using sketch directory: {selected_dir}")
+                    args.directory = selected_dir
+                else:
+                    print(
+                        "\nYou either need to specify a sketch directory or run in --server mode."
+                    )
+                    sys.exit(1)
+        elif args.directory is not None and os.path.isfile(args.directory):
+            dir_path = Path(args.directory).parent
+            if looks_like_sketch_directory(dir_path):
+                print(f"Using sketch directory: {dir_path}")
+                args.directory = str(dir_path)
+
+    return args

fastled/paths.py

@@ -0,0 +1,4 @@
+from pathlib import Path
+
+HERE = Path(__file__).resolve().parent
+PROJECT_ROOT = HERE.parent.parent

fastled/project_init.py

@@ -0,0 +1,129 @@
+import _thread
+import threading
+import time
+import zipfile
+from pathlib import Path
+
+import httpx
+
+from fastled.settings import DEFAULT_URL
+from fastled.spinner import Spinner
+
+DEFAULT_EXAMPLE = "wasm"
+
+
+def get_examples(host: str | None = None) -> list[str]:
+    host = host or DEFAULT_URL
+    url_info = f"{host}/info"
+    response = httpx.get(url_info, timeout=4)
+    response.raise_for_status()
+    examples: list[str] = response.json()["examples"]
+    return sorted(examples)
+
+
+def _prompt_for_example() -> str:
+    examples = get_examples()
+    while True:
+        print("Available examples:")
+        for i, example in enumerate(examples):
+            print(f"  [{i+1}]: {example}")
+        answer = input("Enter the example number or name: ").strip()
+        if answer.isdigit():
+            example_num = int(answer) - 1
+            if example_num < 0 or example_num >= len(examples):
+                print("Invalid example number")
+                continue
+            return examples[example_num]
+        elif answer in examples:
+            return answer
+
+
+class DownloadThread(threading.Thread):
+    def __init__(self, url: str, json: str):
+        super().__init__(daemon=True)
+        self.url = url
+        self.json = json
+        self.bytes_downloaded = 0
+        self.content: bytes | None = None
+        self.error: Exception | None = None
+        self.success = False
+
+    def run(self) -> None:
+        timeout = httpx.Timeout(5.0, connect=5.0, read=120.0, write=30.0)
+        try:
+            with httpx.Client(timeout=timeout) as client:
+                with client.stream("POST", self.url, json=self.json) as response:
+                    response.raise_for_status()
+                    content = b""
+                    for chunk in response.iter_bytes():
+                        content += chunk
+                        self.bytes_downloaded += len(chunk)
+                    self.content = content
+                    self.success = True
+        except KeyboardInterrupt:
+            self.error = RuntimeError("Download cancelled")
+            _thread.interrupt_main()
+        except Exception as e:
+            self.error = e
+
+
+def project_init(
+    example: str | None = "PROMPT",  # prompt for example
+    outputdir: Path | None = None,
+    host: str | None = None,
+) -> Path:
+    """
+    Initialize a new FastLED project.
+    """
+    host = host or DEFAULT_URL
+    outputdir = Path(outputdir) if outputdir is not None else Path("fastled")
+    outputdir.mkdir(exist_ok=True, parents=True)
+    if example == "PROMPT" or example is None:
+        try:
+            example = _prompt_for_example()
+        except httpx.HTTPStatusError:
+            print(
+                f"Failed to fetch examples, using default example '{DEFAULT_EXAMPLE}'"
+            )
+            example = DEFAULT_EXAMPLE
+    assert example is not None
+    endpoint_url = f"{host}/project/init"
+    json = example
+    print(f"Initializing project with example '{example}', url={endpoint_url}")
+
+    # Start download thread
+    download_thread = DownloadThread(endpoint_url, json)
+    # spinner = Spinner("Downloading project...")
+    with Spinner(f"Downloading project {example}..."):
+        download_thread.start()
+        while download_thread.is_alive():
+            time.sleep(0.1)
+
+    print()  # New line after progress
+    download_thread.join()
+
+    # Check for errors
+    if not download_thread.success:
+        assert download_thread.error is not None
+        raise download_thread.error
+
+    content = download_thread.content
+    assert content is not None
+    tmpzip = outputdir / "fastled.zip"
+    outputdir.mkdir(exist_ok=True)
+    tmpzip.write_bytes(content)
+    with zipfile.ZipFile(tmpzip, "r") as zip_ref:
+        zip_ref.extractall(outputdir)
+    tmpzip.unlink()
+    out = outputdir / example
+    print(f"Project initialized at {out}")
+    assert out.exists()
+    return out
+
+
+def unit_test() -> None:
+    project_init()
+
+
+if __name__ == "__main__":
+    unit_test()

fastled/select_sketch_directory.py

@@ -0,0 +1,35 @@
+from pathlib import Path
+
+from fastled.string_diff import string_diff_paths
+
+
+def select_sketch_directory(
+    sketch_directories: list[Path], cwd_is_fastled: bool
+) -> str | None:
+    if cwd_is_fastled:
+        exclude = ["src", "dev", "tests"]
+        for ex in exclude:
+            p = Path(ex)
+            if p in sketch_directories:
+                sketch_directories.remove(p)
+
+    if len(sketch_directories) == 1:
+        print(f"\nUsing sketch directory: {sketch_directories[0]}")
+        return str(sketch_directories[0])
+    elif len(sketch_directories) > 1:
+        print("\nMultiple Directories found, choose one:")
+        for i, sketch_dir in enumerate(sketch_directories):
+            print(f"  [{i+1}]: {sketch_dir}")
+        which = input("\nPlease specify a sketch directory: ").strip()
+        try:
+            index = int(which) - 1
+            return str(sketch_directories[index])
+        except (ValueError, IndexError):
+            inputs = [p for p in sketch_directories]
+            top_hits: list[tuple[float, Path]] = string_diff_paths(which, inputs)
+            if len(top_hits) == 1:
+                example = top_hits[0][1]
+                return str(example)
+            else:
+                return select_sketch_directory([p for _, p in top_hits], cwd_is_fastled)
+    return None

fastled/settings.py

@@ -0,0 +1,13 @@
+import os
+import platform
+
+MACHINE = platform.machine().lower()
+IS_ARM: bool = "arm" in MACHINE or "aarch64" in MACHINE
+PLATFORM_TAG: str = "-arm64" if IS_ARM else ""
+CONTAINER_NAME = f"fastled-wasm-compiler{PLATFORM_TAG}"
+DEFAULT_URL = str(os.environ.get("FASTLED_URL", "https://fastled.onrender.com"))
+SERVER_PORT = 9021
+
+IMAGE_NAME = "niteris/fastled-wasm"
+DEFAULT_CONTAINER_NAME = "fastled-wasm-compiler"
+# IMAGE_TAG = "main"