feat: implement ArduinoUnoQ class for enhanced LED and matrix control

Author: dido18

scratch-arduino-extensions/packages/scratch-vm/src/extensions/ArduinoUnoQ.js

@@ -0,0 +1,104 @@
+const io = require("./socket.io.min.js");
+
+class ArduinoUnoQ {
+  constructor(host, port) {
+    this.serverURL = `wss://${host}:${port}`;
+
+    this.io = io(this.serverURL, {
+      path: "/socket.io",
+      transports: ["polling", "websocket"],
+      autoConnect: true,
+    });
+    this.isConnected = false;
+
+    this._setupConnectionHandlers();
+  }
+
+  _setupConnectionHandlers() {
+    this.io.on("connect", () => {
+      this.isConnected = true;
+      console.log(`Connected to Arduino UNO Q at ${this.serverURL}`);
+    });
+
+    this.io.on("disconnect", (reason) => {
+      this.isConnected = false;
+      console.log(`Disconnected from Arduino UNO Q: ${reason}`);
+    });
+
+    this.io.on("connect_error", (error) => {
+      console.error(`Connection error:`, error.message);
+    });
+
+    this.io.on("reconnect", (attemptNumber) => {
+      console.log(`Reconnected to Arduino UNO Q after ${attemptNumber} attempts`);
+    });
+  }
+
+  connect() {
+    if (!this.io.connected) {
+      console.log("Attempting to connect to Arduino UNO Q...");
+      this.io.connect();
+    }
+  }
+
+  disconnect() {
+    if (this.io.connected) {
+      console.log("Disconnecting from Arduino UNO Q...");
+      this.io.disconnect();
+    }
+  }
+
+  // ===== LED CONTROL METHODS =====
+  /**
+   * Set RGB LED color
+   * @param {string} led - LED identifier ("LED3" or "LED4")
+   * @param {number} r - Red value (0-255)
+   * @param {number} g - Green value (0-255)
+   * @param {number} b - Blue value (0-255)
+   */
+  setLedRGB(led, r, g, b) {
+    this.io.emit("set_led_rgb", {
+      led: led,
+      r: Math.max(0, Math.min(255, r)),
+      g: Math.max(0, Math.min(255, g)),
+      b: Math.max(0, Math.min(255, b)),
+    });
+    console.log(`Setting ${led} to RGB(${r}, ${g}, ${b})`);
+  }
+
+  /**
+   * Turn off LED
+   * @param {string} led - LED identifier ("LED3" or "LED4")
+   */
+  turnOffLed(led) {
+    this.setLedRGB(led, 0, 0, 0);
+  }
+
+  // ===== MATRIX CONTROL METHODS =====
+
+  /**
+   * Draw frame on LED matrix
+   * @param {string} frame - 25-character string representing 5x5 matrix (0s and 1s)
+   */
+  matrixDraw(frame) {
+    if (typeof frame !== "string" || frame.length !== 25) {
+      console.error("Invalid frame format. Expected 25-character string of 0s and 1s");
+      return;
+    }
+    // Validate frame contains only 0s and 1s
+    if (!/^[01]+$/.test(frame)) {
+      console.error("Frame must contain only 0s and 1s");
+      return;
+    }
+
+    this.io.emit("matrix_draw", { frame: frame });
+    console.log(`Drawing matrix frame: ${frame}`);
+  }
+
+  matrixClear() {
+    const clearFrame = "0".repeat(25);
+    this.matrixDraw(clearFrame);
+  }
+}
+
+module.exports = ArduinoUnoQ;

scratch-arduino-extensions/packages/scratch-vm/src/extensions/arduino_basics/index.js

@@ -1,35 +1,18 @@
-// const formatMessage = require('../../../../../../scratch-editor/node_modules/format-message');
 const BlockType = require("../../../../../../scratch-editor/packages/scratch-vm/src/extension-support/block-type");
 const ArgumentType = require(
   "../../../../../../scratch-editor/packages/scratch-vm/src/extension-support/argument-type",
 );
-const io = require("../socket.io.min.js");
+const ArduinoUnoQ = require("../ArduinoUnoQ");
 
-/**
- * Url of icon to be displayed at the left edge of each extension block.
- * @type {string}
- */
-// eslint-disable-next-line max-len
+// TODO: add icons
 const iconURI = "";
-
-/**
- * Url of icon to be displayed in the toolbox menu for the extension category.
- * @type {string}
- */
-// eslint-disable-next-line max-len
 const menuIconURI = "";
 
-const wsServerURL = `${window.location.protocol}//${window.location.hostname}:7000`;
-
 class ArduinoBasics {
   constructor(runtime) {
     this.runtime = runtime;
-
-    this.io = io(wsServerURL, {
-      path: "/socket.io",
-      transports: ["polling", "websocket"],
-      autoConnect: true,
-    });
+    this.unoq = new ArduinoUnoQ("192.168.1.39", 7000);
+    this.unoq.connect();
   }
 }
 
@@ -88,27 +71,23 @@ ArduinoBasics.prototype.getInfo = function() {
 };
 
 ArduinoBasics.prototype.matrixDraw = function(args) {
-  console.log(`Drawing frame on matrix: ${args}`);
-  this.io.emit("matrix_draw", { frame: args.FRAME });
+  this.unoq.matrixDraw(args.FRAME);
 };
 
 ArduinoBasics.prototype.matrixClear = function() {
-  console.log("Clearing matrix");
-  this.io.emit("matrix_draw", { frame: "0000000000000000000000000" });
+  this.unoq.matrixClear();
 };
 
 ArduinoBasics.prototype.setLed3 = function(args) {
   const hexColor = args.HEX;
   const rgb = this.hexToRgb(hexColor);
-  console.log(`Setting led 3 to: r:${rgb.r}, g:${rgb.g}, b:${rgb.b} (HEX: ${hexColor})`);
-  this.io.emit("set_led_rgb", { led: "LED3", r: rgb.r, g: rgb.g, b: rgb.b });
+  this.unoq.setLedRGB("LED3", rgb.r, rgb.g, rgb.b);
 };
 
 ArduinoBasics.prototype.setLed4 = function(args) {
   const hexColor = args.HEX;
   const rgb = this.hexToRgb(hexColor);
-  console.log(`Setting led 4 to: r:${rgb.r}, g:${rgb.g}, b:${rgb.b} (HEX: ${hexColor})`);
-  this.io.emit("set_led_rgb", { led: "LED4", r: rgb.r, g: rgb.g, b: rgb.b });
+  this.unoq.setLedRGB("LED4", rgb.r, rgb.g, rgb.b);
 };
 
 ArduinoBasics.prototype.hexToRgb = function(hex) {