node-red-nodes/hardware/blinkstick/76-blinkstick.js

module.exports = function(RED) {
    "use strict";
    var blinkstick = require("blinkstick");

    var availableModes = ["normal", "inverted", "neopixel"];
    var availableTasks = ["set_color", "blink", "pulse", "morph"];

    Object.size = function(obj) {
        var size = 0;
        for (var key in obj) { if (obj.hasOwnProperty(key)) { size++; } }
        return size;
    };

    //Helper function to convert decimal number to hex with padding
    function decimalToHex(d, padding) {
        var hex = Number(d).toString(16);
        padding = typeof (padding) === "undefined" || padding === null ? padding = 2 : padding;

        while (hex.length < padding) {
            hex = "0" + hex;
        }

        return hex;
    }

    function validateInt(value, defaultValue) {
        return typeof (value) === "undefined" || value === null ? value = defaultValue : parseInt(value);
    }

    function validate(value, defaultValue) {
        return typeof (value) === "undefined" || value === null ? value = defaultValue : value;
    }

    function validateArray(value, defaultValue) {
        return typeof (value) === "undefined" || Array.isArray(value) ? value : defaultValue;
    }

    function validatePayloadObject (obj) {
        var
            task = validate(obj.task),
            delay = validateInt(obj.delay),
            repeats = validateInt(obj.repeats),
            duration = validateInt(obj.duration),
            steps = validateInt(obj.steps),
            repeat = validate(obj.repeat),
            color = validate(obj.color),
            channel = validateInt(obj.channel),
            index = validateInt(obj.index),
            row = validateArray(obj.row);

        if (typeof(task) !== 'undefined' && availableTasks.indexOf(task) === -1) {
            return "Task is invalid";
        }

        if (typeof(color) === 'undefined' && typeof(row) === 'undefined') {
            return "Color parameter is not set";
        }

        return { 'task': task, 'delay': delay, 'repeats': repeats, 'duration': duration, 'steps': steps,
            'repeat': repeat, 'color': color, 'channel': channel, 'index': index, 'row': row };
    }

    function BlinkStick(n) {
        RED.nodes.createNode(this,n);

        this.name = n.name;
        this.serial = n.serial;
        this.mode = n.mode || "normal";
        this.task = n.task || "set_color";
        this.delay = n.delay || 500;
        this.repeats = n.repeats || 1;
        this.duration = n.duration || 1000;
        this.steps = n.steps || 50;
        this.repeat = n.repeat;
        this.channel = 0;
        this.index = 0;
        this.row = [];
        this.closing = false;
        this.color = '';
        this.previousColor = '';

        var p1 = /[0-9]+,[0-9]+,[0-9]+/;
        var node = this;
        //Keeps track for active animations
        var animationComplete = true;

        //Find BlinkStick based on serial number if supplied, otherwise look for first
        //Blinkstick in the USB device list
        var findBlinkStick = function (callback) {
            if (typeof(node.serial) == 'string' && node.serial.replace(/\s+/g,'') !== '') {
                blinkstick.findBySerial(node.serial, function (device) {
                    node.led = device;

                    if (Object.size(node.led) === 0) {
                        node.status({fill:"red",shape:"ring",text:"not found"});
                        node.error("BlinkStick with serial number " + node.serial + " not found");
                    }
                    else {
                        node.status({fill:"green",shape:"dot",text:"connected"});
                        if (node.mode == "normal") {node.led.setMode(0);}
                        else if (node.mode == "inverted") {node.led.setMode(1);}
                        else if (node.mode == "neopixel") {node.led.setMode(2);}
                        if (callback) { callback(); }
                    }
                });
            }
            else {
                node.led = blinkstick.findFirst();

                if (Object.size(node.led) === 0) {
                    node.status({fill:"red",shape:"ring",text:"not found"});
                    node.error("No BlinkStick found");
                }
                else {
                    node.status({fill:"green",shape:"dot",text:"connected"});
                    if (node.mode == "normal") {node.led.setMode(0);}
                    else if (node.mode == "inverted") {node.led.setMode(1);}
                    else if (node.mode == "neopixel") {node.led.setMode(2);}
                    if (callback) { callback(); }
                }
            }
        };

        //Check if repeat is enabled. Works only for pulse and blink tasks
        var canRepeat = function () {
            return node.task == "pulse" || node.task == "blink";
        };

        //Event handler for all animation complete events
        var blinkstickAnimationComplete = function (err) {
            if (typeof(err) !== 'undefined') {
                node.warn(err);

                if (typeof(err.name) === 'undefined' || err.name !== 'ReferenceError') {
                    //USB error occurred when BlinkStick was animating
                    node.led.close(function () {
                        node.led = undefined;
                        findBlinkStick();
                    });
                }
            }

            animationComplete = true;

            //Apply queued color animation
            if (!node.closing && node.color !== '') {
                //Apply new color only if there was no error or queued color is different from the previous color
                if (typeof(err) === 'undefined' || typeof(err) !== 'undefined' && node.color != node.previousColor) {
                    applyColor();
                }
            }
        };

        var applyColor = function () {
            animationComplete = false;

            //Store the value of color to check if it has changed
            node.previousColor = node.color;

            try {
                //Select animation to perform
                if (node.task == "pulse") {
                    node.led.pulse(node.color, {'duration': node.duration, 'steps': node.steps, 'channel': node.channel, 'index': node.index }, blinkstickAnimationComplete);
                }
                else if (node.task == "morph") {
                    node.led.morph(node.color, {'duration': node.duration, 'steps': node.steps, 'channel': node.channel, 'index': node.index }, blinkstickAnimationComplete);
                }
                else if (node.task == "blink") {
                    node.led.blink(node.color,{'repeats': node.repeats, 'delay': node.delay, 'channel': node.channel, 'index': node.index }, blinkstickAnimationComplete);
                }
                else {
                    if (node.row.length > 0) {
                        var dat = [];
                        for (var i = 0; i < node.row.length; i++) {
                            if (typeof node.row[i] === "string") { // if string then assume must be colour names
                                var params = node.led.interpretParameters(node.row[i]); // lookup colour code from name
                                if (params) {
                                    dat.push(params.green);
                                    dat.push(params.red);
                                    dat.push(params.blue);
                                }
                                else { node.warn("invalid colour: "+node.row[i]); }
                            }
                            else { // otherwise lets use numbers 0-255
                                dat.push(node.row[i+1]);
                                dat.push(node.row[i]);
                                dat.push(node.row[i+2]);
                                i += 2;
                            }
                        }
                        if ((dat.length % 3) === 0) { // by now length must be a multiple of 3
                            node.led.setColors(node.channel, dat, blinkstickAnimationComplete);
                        }
                        else {
                            node.warn("Colour array length not / 3");
                        }
                    }
                    else {
                        node.led.setColor(node.color, {'channel': node.channel, 'index': node.index}, blinkstickAnimationComplete);
                    }
                }
            }
            catch (err) {
                if (err.toString().indexOf("setColor") !== -1) {
                    node.led.setColour(node.color, blinkstickAnimationComplete);
                    node.warn("Old version - please upgrade Blinkstick npm");
                }
                else {
                    node.warn("BlinkStick missing ? " + err);
                }
                node.log(err);
                //Reset animation
                animationComplete = true;
                //Clear color
                node.color = '';
                //Look for a BlinkStick
                findBlinkStick();
                return;
            }

            //Clear color value until next one is received, unless repeat option is set to true
            if (!node.repeat || !canRepeat()) {
                node.color = '';
            }
        };

        findBlinkStick();

        this.on("input", function(msg) {
            if (typeof(node.led) !== "undefined") {
                if (typeof(msg.payload) === 'object' ) {
                    // if it's an array then hopefully it's r,g,b,r,g,b or name,name,name
                    if (Array.isArray(msg.payload)) {
                        if (Object.size(node.led) !== 0) {
                            var dat = [];
                            for (var i = 0; i < msg.payload.length; i++) {
                                if (typeof msg.payload[i] === "string") { // if string then assume must be colour names
                                    var params = node.led.interpretParameters(msg.payload[i]); // lookup colour code from name
                                    if (params) {
                                        dat.push(params.green);
                                        dat.push(params.red);
                                        dat.push(params.blue);
                                    }
                                    else { node.warn("invalid colour: "+msg.payload[i]); }
                                }
                                else { // otherwise lets use numbers 0-255
                                    dat.push(msg.payload[i+1]);
                                    dat.push(msg.payload[i]);
                                    dat.push(msg.payload[i+2]);
                                    i += 2;
                                }
                            }
                            if ((dat.length % 3) === 0) { // by now length must be a multiple of 3
                                node.led.setColors(0, dat, function(err) {
                                    if (err) { node.log(err); }
                                });
                            }
                            else {
                                node.warn("Colour array length not / 3");
                            }
                            return;
                        } // else if no blinkstick let it get caught below
                    }
                    // not an array - must be the "normal" object....
                    else {
                        var data = validatePayloadObject(msg.payload);
                        if (typeof(data) === 'object') {
                            node.task = data.task ? data.task : node.task;
                            node.delay = data.delay ? data.delay : node.delay;
                            node.repeats = data.repeats ? data.repeats : node.repeats;
                            node.duration = data.duration ? data.duration : node.duration;
                            node.steps = data.steps ? data.steps : node.steps;
                            node.repeat = data.repeat ? data.repeat : node.repeat;
                            node.color = data.color ? data.color : node.color;
                            node.channel = typeof(data.channel) !== 'undefined' ? data.channel : node.channel;
                            node.index = data.index ? data.index : node.index;
                            node.row = data.row ? data.row : node.row;
                        }
                        else {
                            node.error(data);
                            return;
                        }
                    }
                }
                else if (p1.test(msg.payload)) {
                    //Color value is represented as "red,green,blue" string of bytes
                    var rgb = msg.payload.split(",");

                    //Convert color value back to HEX string for easier implementation
                    node.color = "#" + decimalToHex(parseInt(rgb[0])&255) +
                        decimalToHex(parseInt(rgb[1])&255) + decimalToHex(parseInt(rgb[2])&255);
                }
                else {
                    //Sanitize color value
                    node.color = msg.payload.toLowerCase().replace(/\s+/g,'');
                    if (node.color === "amber") { node.color = "#FFBF00"; }
                    if (node.color === "off") { node.color = "#000000"; }
                }

                if (Object.size(node.led) !== 0) {
                    //Start color animation, otherwise the color is queued until current animation completes
                    if (animationComplete) {
                        applyColor();
                    }
                }
                else {
                    //Attempt to find BlinkStick and start animation if it's found
                    findBlinkStick(function() {
                        if (animationComplete) {
                            applyColor();
                        }
                    });
                }
            }
            else {
                node.status({fill:"red",shape:"ring",text:"not found"});
                node.error("BlinkStick not found",msg);
            }
        });

        this.on("close", function() {
            //Set the flag to finish all animations
            this.closing = true;

            if (Object.size(node.led) !== 0) {
                //Close device and stop animations
                if (typeof this.led.close === "function") { this.led.close(); }
                else { node.warn("Please upgrade blinkstick npm"); }
            }
        });
    }

    RED.nodes.registerType("blinkstick",BlinkStick);

    RED.httpAdmin.get("/blinksticklist", RED.auth.needsPermission("blinkstick.read"), function(req,res) {
        blinkstick.findAllSerials(function(serials) {
            res.json(serials);
        });
    });
};