added new node and simple workflow

- added compilation and upload
- added a node to write arduino code
- added a simple workflow to write and upload code

Author: Juste-Leo2

README.md

@@ -4,27 +4,33 @@
 
 ### Core Features (Roadmap)
 
+<p align="center">
+  <img src="docs/image1.png" alt="workflow screen">
+</p>
+
+
 Here are the initial goals for the project:
 
 - [x] **Self-contained Installer Node:** Automatically downloads and manages `arduino-cli` locally. No manual setup required from the user, ensuring a smooth "all-in-one" experience.
-- [ ] **Dynamic Upload Node:** Visually build logic with nodes, which will then generate, compile, and upload a custom sketch to your connected Arduino board.
+- [x] **Dynamic Upload Node:** Visually build logic with nodes, which will then generate, compile, and upload a custom sketch to your connected Arduino board.
 - [ ] **Real-time Communication:** Implement a standard sketch and corresponding nodes to send live data (e.g., servo angles, LED colors) from ComfyUI to a running Arduino without re-uploading.
-- [ ] **Example Workflows:** Provide simple, functional examples to help users get started quickly.
+- [x] **Example Workflows:** Provide simple, functional examples to help users get started quickly.
 
-### Future Vision
+### other features
 
 Ideas for the long-term development of the project:
 
 - [ ] **Arduino Library Management:** A node to automatically install required libraries for your sketch using `arduino-cli`.
 - [ ] **High-Level Hardware Nodes:** Easy-to-use nodes for common components (servos, sensors, NeoPixel LEDs, etc.) that abstract away the low-level code.
-- [ ] **In-Workflow C++ Editor:** A dedicated node to write or paste raw Arduino (C++) code directly within ComfyUI.
+- [x] **In-Workflow C++ Editor:** A dedicated node to write or paste raw Arduino (C++) code directly within ComfyUI.
 - [ ] **Full Sketch Export:** An option to export the generated C++ code and a list of its dependencies, allowing it to be used outside of ComfyUI.
 
 ### Installation
 
 1.  Navigate to your `ComfyUI/custom_nodes/` directory.
 2.  Clone this repository: `git clone https://github.com/Juste-Leo2/ComfyUI-Arduino.git`
-3.  Restart ComfyUI. The necessary dependencies, including `arduino-cli`, should be installed automatically on the first run.
+3.  install dependencies: `pip install -r requirements.txt`
+4.  Restart ComfyUI. The necessary dependencies, including `arduino-cli`, should be installed automatically on the first run.
 
 ### Contributing
 

__init__.py

@@ -1,15 +1,19 @@
 # __init__.py
 
-from .nodes import ArduinoTargetNode # <-- NOM MIS À JOUR
+from .nodes import ArduinoTargetNode, ArduinoCodeNode, ArduinoCompileUploadNode
 
 # --- Mappings for ComfyUI ---
 
 NODE_CLASS_MAPPINGS = {
-    "ArduinoTarget": ArduinoTargetNode, # <-- NOM MIS À JOUR
+    "ArduinoTarget": ArduinoTargetNode,
+    "ArduinoCode": ArduinoCodeNode,
+    "ArduinoCompileUpload": ArduinoCompileUploadNode,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "ArduinoTarget": "Define Arduino Target", # <-- NOM MIS À JOUR
+    "ArduinoTarget": "Define Arduino Target",
+    "ArduinoCode": "Arduino Code",
+    "ArduinoCompileUpload": "Compile & Upload to Arduino",
 }
 
 print("...Custom Nodes: ComfyUI-Arduino loaded.")

docs/image1.png

@@ -3,6 +3,7 @@
 import os
 from .src.arduino_installer import setup_arduino_cli
 from .src.arduino_board_finder import get_available_boards, get_fqbn_by_name
+from .src.arduino_actions import compile_and_upload_sketch # <-- NOUVEL IMPORT
 import serial.tools.list_ports
 
 def list_physical_ports():
@@ -61,4 +62,67 @@ def define_target(self, board_name, port_str):
             return ("ERROR", "ERROR", error)
 
         status_message = f"✅ Target defined: {board_name} ({fqbn}) on {port}"
-        return (port, fqbn, status_message)
+        return (port, fqbn, status_message)
+
+# --------------------------------------------------------------------
+# -- NODE: ARDUINO CODE INPUT ----------------------------------------
+# --------------------------------------------------------------------
+class ArduinoCodeNode:
+    @classmethod
+    def INPUT_TYPES(s):
+        default_code = "void setup() {\n  // put your setup code here, to run once:\n\n}\n\nvoid loop() {\n  // put your main code here, to run repeatedly:\n\n}"
+        return {
+            "required": {
+                "code": ("STRING", {"multiline": True, "default": default_code}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("code",)
+    FUNCTION = "get_code"
+    CATEGORY = "Arduino"
+
+    def get_code(self, code):
+        return (code,)
+
+# --------------------------------------------------------------------
+# -- NODE: COMPILE & UPLOAD SKETCH -----------------------------------
+# --------------------------------------------------------------------
+class ArduinoCompileUploadNode:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "port": ("STRING", {"forceInput": True}),
+                "fqbn": ("STRING", {"forceInput": True}),
+                "code": ("STRING", {"forceInput": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("status",)
+    FUNCTION = "compile_and_upload"
+    CATEGORY = "Arduino"
+
+    def compile_and_upload(self, port, fqbn, code):
+        # Step 1: Node-level validation
+        if SETUP_ERROR is not None:
+            return (f"❌ ERROR: Setup failed: {SETUP_ERROR}",)
+        if port == "ERROR" or fqbn == "ERROR":
+            return ("❌ ERROR: Invalid target. Check 'Define Arduino Target' node.",)
+
+        print(f"--- Arduino: Starting compile & upload for {fqbn} on {port} ---")
+        
+        # Step 2: Call the backend logic function
+        success, message = compile_and_upload_sketch(
+            cli_path=ARDUINO_CLI_PATH,
+            config_path=ARDUINO_CONFIG_PATH,
+            port=port,
+            fqbn=fqbn,
+            code=code
+        )
+        
+        print(f"--- Arduino: Process finished. ---")
+        
+        # Step 3: Return the result from the logic function
+        return (message,)

nodes.py

@@ -0,0 +1,62 @@
+# src/arduino_actions.py
+
+import os
+import shutil
+import tempfile
+from .cli_utils import run_cli_command
+
+def compile_and_upload_sketch(cli_path: str, config_path: str, port: str, fqbn: str, code: str) -> tuple[bool, str]:
+    """
+    Handles the entire process of compiling and uploading an Arduino sketch.
+
+    Args:
+        cli_path: Path to the arduino-cli executable.
+        config_path: Path to the arduino-cli.yaml config file.
+        port: The COM port to upload to (e.g., "COM3").
+        fqbn: The Fully Qualified Board Name (e.g., "arduino:avr:uno").
+        code: A string containing the Arduino C++ code.
+
+    Returns:
+        A tuple containing:
+        - bool: True if successful, False otherwise.
+        - str: A status message detailing the outcome.
+    """
+    # Arduino CLI requires the .ino file to be in a folder with the same name.
+    # We create a temporary directory for this.
+    temp_dir = tempfile.mkdtemp()
+    sketch_name = os.path.basename(temp_dir)
+    sketch_path = os.path.join(temp_dir, f"{sketch_name}.ino")
+
+    try:
+        # 1. Write the code to the temporary .ino file
+        with open(sketch_path, 'w', encoding='utf-8') as f:
+            f.write(code)
+        print(f"   - Temporary sketch created at {sketch_path}")
+
+        # 2. Compile the sketch
+        print(f"   - Compiling for board {fqbn}...")
+        compile_args = ["compile", "--fqbn", fqbn, temp_dir]
+        success, result = run_cli_command(cli_path, config_path, compile_args)
+        if not success:
+            error_msg = f"❌ Compilation failed: {result}"
+            print(f"   - {error_msg}")
+            return False, error_msg
+        print("   - ✅ Compilation successful.")
+
+        # 3. Upload the sketch
+        print(f"   - Uploading to port {port}...")
+        upload_args = ["upload", "-p", port, "--fqbn", fqbn, temp_dir]
+        success, result = run_cli_command(cli_path, config_path, upload_args)
+        if not success:
+            error_msg = f"❌ Upload failed: {result}"
+            print(f"   - {error_msg}")
+            return False, error_msg
+        
+        status_message = f"✅✅✅ Upload to {fqbn} on {port} successful!"
+        print(f"   - {status_message}")
+        return True, status_message
+
+    finally:
+        # 4. Clean up the temporary directory, regardless of outcome
+        print(f"   - Cleaning up temporary sketch directory: {temp_dir}")
+        shutil.rmtree(temp_dir)

src/arduino_actions.py

@@ -0,0 +1 @@
+{"id":"01e2f77f-e4dc-45d5-8489-8923142e53bb","revision":0,"last_node_id":8,"last_link_id":8,"nodes":[{"id":6,"type":"ArduinoCompileUpload","pos":[1288.7777099609375,476.75299072265625],"size":[220.41665649414062,66],"flags":{},"order":2,"mode":0,"inputs":[{"localized_name":"port","name":"port","type":"STRING","link":4},{"localized_name":"fqbn","name":"fqbn","type":"STRING","link":5},{"localized_name":"code","name":"code","type":"STRING","link":6}],"outputs":[{"localized_name":"status","name":"status","type":"STRING","links":[7]}],"properties":{"Node name for S&R":"ArduinoCompileUpload"},"widgets_values":[]},{"id":7,"type":"PreviewAny","pos":[1579.9434814453125,475.4234619140625],"size":[515.86572265625,269.7618713378906],"flags":{},"order":3,"mode":0,"inputs":[{"localized_name":"source","name":"source","type":"*","link":7}],"outputs":[],"properties":{"cnr_id":"comfy-core","ver":"0.3.52","Node name for S&R":"PreviewAny"},"widgets_values":[]},{"id":5,"type":"ArduinoCode","pos":[813.4561157226562,671.795654296875],"size":[400,200],"flags":{},"order":1,"mode":0,"inputs":[{"localized_name":"code","name":"code","type":"STRING","widget":{"name":"code"},"link":null}],"outputs":[{"localized_name":"code","name":"code","type":"STRING","links":[6]}],"properties":{"Node name for S&R":"ArduinoCode"},"widgets_values":["void setup() {\n  // put your setup code here, to run once:\n  pinMode(7, OUTPUT);\n\n}\n\nvoid loop() {\n  // put your main code here, to run repeatedly:\n  digitalWrite(7, HIGH);\n}"]},{"id":1,"type":"ArduinoTarget","pos":[877.5122680664062,371.6294250488281],"size":[270,122],"flags":{},"order":0,"mode":0,"inputs":[{"localized_name":"board_name","name":"board_name","type":"COMBO","widget":{"name":"board_name"},"link":null},{"localized_name":"port_str","name":"port_str","type":"COMBO","widget":{"name":"port_str"},"link":null}],"outputs":[{"localized_name":"port","name":"port","type":"STRING","links":[4]},{"localized_name":"fqbn","name":"fqbn","type":"STRING","links":[5]},{"localized_name":"status","name":"status","type":"STRING","links":[]}],"properties":{"Node name for S&R":"ArduinoTarget"},"widgets_values":["Arduino Uno","COM3 - Silicon Labs CP210x USB to UART Bridge (COM3)"]}],"links":[[4,1,0,6,0,"STRING"],[5,1,1,6,1,"STRING"],[6,5,0,6,2,"STRING"],[7,6,0,7,0,"*"]],"groups":[],"config":{},"extra":{"ds":{"scale":1.3310000000000028,"offset":[-581.8514898039264,-169.2931295445007]}},"version":0.4}