@willieee802/zigbee-herdsman 0.48.3 → 0.49.1

package/src/zspec/zcl/utils.ts

@@ -0,0 +1,504 @@
+import {Clusters} from "./definition/cluster";
+import {ZCL_TYPE_INVALID_BY_TYPE} from "./definition/datatypes";
+import {DataType, DataTypeClass} from "./definition/enums";
+import {Foundation, type FoundationCommandName, type FoundationDefinition} from "./definition/foundation";
+import type {Attribute, Cluster, ClusterDefinition, ClusterName, Command, CustomClusters, Parameter} from "./definition/tstype";
+
+const DATA_TYPE_CLASS_DISCRETE = [
+    DataType.DATA8,
+    DataType.DATA16,
+    DataType.DATA24,
+    DataType.DATA32,
+    DataType.DATA40,
+    DataType.DATA48,
+    DataType.DATA56,
+    DataType.DATA64,
+    DataType.BOOLEAN,
+    DataType.BITMAP8,
+    DataType.BITMAP16,
+    DataType.BITMAP24,
+    DataType.BITMAP32,
+    DataType.BITMAP40,
+    DataType.BITMAP48,
+    DataType.BITMAP56,
+    DataType.BITMAP64,
+    DataType.ENUM8,
+    DataType.ENUM16,
+    DataType.OCTET_STR,
+    DataType.CHAR_STR,
+    DataType.LONG_OCTET_STR,
+    DataType.LONG_CHAR_STR,
+    DataType.ARRAY,
+    DataType.STRUCT,
+    DataType.SET,
+    DataType.BAG,
+    DataType.CLUSTER_ID,
+    DataType.ATTR_ID,
+    DataType.BAC_OID,
+    DataType.IEEE_ADDR,
+    DataType.SEC_KEY,
+];
+const DATA_TYPE_CLASS_ANALOG = [
+    DataType.UINT8,
+    DataType.UINT16,
+    DataType.UINT24,
+    DataType.UINT32,
+    DataType.UINT40,
+    DataType.UINT48,
+    DataType.UINT56,
+    DataType.INT8,
+    DataType.INT16,
+    DataType.INT24,
+    DataType.INT32,
+    DataType.INT40,
+    DataType.INT48,
+    DataType.INT56,
+    DataType.SEMI_PREC,
+    DataType.SINGLE_PREC,
+    DataType.DOUBLE_PREC,
+    DataType.TOD,
+    DataType.DATE,
+    DataType.UTC,
+];
+
+const FOUNDATION_DISCOVER_RSP_IDS = [
+    Foundation.discoverRsp.ID,
+    Foundation.discoverCommandsRsp.ID,
+    Foundation.discoverCommandsGenRsp.ID,
+    Foundation.discoverExtRsp.ID,
+];
+
+/** Runtime fast lookup */
+const ZCL_CLUSTERS_ID_TO_NAMES = (() => {
+    const map = new Map<number, ClusterName>();
+
+    for (const clusterName in Clusters) {
+        const cluster = Clusters[clusterName as ClusterName];
+
+        map.set(cluster.ID, clusterName as ClusterName);
+    }
+
+    return map;
+})();
+
+export function getDataTypeClass(dataType: DataType): DataTypeClass {
+    if (DATA_TYPE_CLASS_DISCRETE.includes(dataType)) {
+        return DataTypeClass.DISCRETE;
+    }
+
+    if (DATA_TYPE_CLASS_ANALOG.includes(dataType)) {
+        return DataTypeClass.ANALOG;
+    }
+
+    throw new Error(`Don't know value type for '${DataType[dataType]}'`);
+}
+
+function hasCustomClusters(customClusters: CustomClusters): boolean {
+    // XXX: was there a good reason to not set the parameter `customClusters` optional? it would allow simple undefined check
+    // below is twice faster than checking `Object.keys(customClusters).length`
+    for (const _k in customClusters) return true;
+    return false;
+}
+
+function findClusterNameByID(
+    id: number,
+    manufacturerCode: number | undefined,
+    clusters: typeof Clusters | CustomClusters,
+    zcl: boolean,
+): [name: string | undefined, partialMatch: boolean] {
+    let name: string | undefined;
+    // if manufacturer code is given, consider partial match if didn't match against manufacturer code
+    let partialMatch = Boolean(manufacturerCode);
+
+    if (zcl) {
+        const zclName = ZCL_CLUSTERS_ID_TO_NAMES.get(id);
+
+        if (zclName) {
+            const cluster = clusters[zclName];
+
+            // priority on first match when matching only ID
+            if (name === undefined) {
+                name = zclName;
+            }
+
+            if (manufacturerCode && cluster.manufacturerCode === manufacturerCode) {
+                name = zclName;
+                partialMatch = false;
+            } else if (!cluster.manufacturerCode) {
+                name = zclName;
+            }
+        }
+    } else {
+        for (const clusterName in clusters) {
+            const cluster = clusters[clusterName as ClusterName];
+
+            if (cluster.ID === id) {
+                // priority on first match when matching only ID
+                if (name === undefined) {
+                    name = clusterName;
+                }
+
+                if (manufacturerCode && cluster.manufacturerCode === manufacturerCode) {
+                    name = clusterName;
+                    partialMatch = false;
+                    break;
+                }
+
+                if (!cluster.manufacturerCode) {
+                    name = clusterName;
+                    break;
+                }
+            }
+        }
+    }
+
+    return [name, partialMatch];
+}
+
+function getClusterDefinition(
+    key: string | number,
+    manufacturerCode: number | undefined,
+    customClusters: CustomClusters,
+): {name: string; cluster: ClusterDefinition} {
+    let name: string | undefined;
+
+    if (typeof key === "number") {
+        let partialMatch: boolean;
+
+        // custom clusters have priority over Zcl clusters, except in case of better match (see below)
+        [name, partialMatch] = findClusterNameByID(key, manufacturerCode, customClusters, false);
+
+        if (!name) {
+            [name, partialMatch] = findClusterNameByID(key, manufacturerCode, Clusters, true);
+        } else if (partialMatch) {
+            let zclName: string | undefined;
+            [zclName, partialMatch] = findClusterNameByID(key, manufacturerCode, Clusters, true);
+
+            // Zcl clusters contain a better match, use that one
+            if (zclName !== undefined && !partialMatch) {
+                name = zclName;
+            }
+        }
+    } else {
+        name = key;
+    }
+
+    let cluster =
+        name !== undefined && hasCustomClusters(customClusters)
+            ? {
+                  ...Clusters[name as ClusterName],
+                  ...customClusters[name], // should override Zcl clusters
+              }
+            : Clusters[name as ClusterName];
+
+    if (!cluster || cluster.ID === undefined) {
+        if (typeof key === "number") {
+            name = key.toString();
+            cluster = {attributes: {}, commands: {}, commandsResponse: {}, manufacturerCode: undefined, ID: key};
+        } else {
+            name = undefined;
+        }
+    }
+
+    if (!name) {
+        throw new Error(`Cluster with name '${key}' does not exist`);
+    }
+
+    return {name, cluster};
+}
+
+function cloneClusterEntriesWithName<T extends Record<string, unknown>>(entries: Record<string, T>): Record<string, {name: string} & T> {
+    const clone: Record<string, {name: string} & T> = {};
+
+    for (const key in entries) {
+        clone[key] = {...entries[key], name: key};
+    }
+
+    return clone;
+}
+
+function createCluster(name: string, cluster: ClusterDefinition, manufacturerCode?: number): Cluster {
+    const attributes: Record<string, Attribute> = cloneClusterEntriesWithName(cluster.attributes);
+    const commands: Record<string, Command> = cloneClusterEntriesWithName(cluster.commands);
+    const commandsResponse: Record<string, Command> = cloneClusterEntriesWithName(cluster.commandsResponse);
+
+    const getAttribute = (key: number | string): Attribute | undefined => {
+        if (typeof key === "number") {
+            let partialMatchAttr: Attribute | undefined;
+
+            for (const attrKey in attributes) {
+                const attr = attributes[attrKey];
+
+                if (attr.ID === key) {
+                    if (manufacturerCode && attr.manufacturerCode === manufacturerCode) {
+                        return attr;
+                    }
+
+                    if (attr.manufacturerCode === undefined) {
+                        partialMatchAttr = attr;
+                    }
+                }
+            }
+
+            return partialMatchAttr;
+        }
+
+        return attributes[key];
+    };
+
+    const getCommand = (key: number | string): Command => {
+        if (typeof key === "number") {
+            for (const cmdKey in commands) {
+                const cmd = commands[cmdKey];
+
+                if (cmd.ID === key) {
+                    return cmd;
+                }
+            }
+        } else {
+            const cmd = commands[key];
+
+            if (cmd) {
+                return cmd;
+            }
+        }
+
+        throw new Error(`Cluster '${name}' has no command '${key}'`);
+    };
+
+    const getCommandResponse = (key: number | string): Command => {
+        if (typeof key === "number") {
+            for (const cmdKey in commandsResponse) {
+                const cmd = commandsResponse[cmdKey];
+
+                if (cmd.ID === key) {
+                    return cmd;
+                }
+            }
+        } else {
+            const cmd = commandsResponse[key];
+
+            if (cmd) {
+                return cmd;
+            }
+        }
+
+        throw new Error(`Cluster '${name}' has no command response '${key}'`);
+    };
+
+    return {
+        ID: cluster.ID,
+        attributes,
+        manufacturerCode: cluster.manufacturerCode,
+        name,
+        commands,
+        commandsResponse,
+        getAttribute,
+        getCommand,
+        getCommandResponse,
+    };
+}
+
+export function getCluster(key: string | number, manufacturerCode: number | undefined = undefined, customClusters: CustomClusters = {}): Cluster {
+    const {name, cluster} = getClusterDefinition(key, manufacturerCode, customClusters);
+    return createCluster(name, cluster, manufacturerCode);
+}
+
+function getGlobalCommandNameById(id: number): FoundationCommandName {
+    for (const commandName in Foundation) {
+        if (Foundation[commandName as FoundationCommandName].ID === id) {
+            return commandName as FoundationCommandName;
+        }
+    }
+
+    throw new Error(`Global command with id '${id}' does not exist.`);
+}
+
+export function getGlobalCommand(key: number | string): Command {
+    const name = typeof key === "number" ? getGlobalCommandNameById(key) : (key as FoundationCommandName);
+    const command = Foundation[name];
+
+    if (!command) {
+        throw new Error(`Global command with key '${key}' does not exist`);
+    }
+
+    const result: Command = {
+        ID: command.ID,
+        name,
+        parameters: command.parameters,
+    };
+
+    if (command.response !== undefined) {
+        result.response = command.response;
+    }
+
+    return result;
+}
+
+export function isClusterName(name: string): name is ClusterName {
+    return name in Clusters;
+}
+
+export function getFoundationCommand(id: number): FoundationDefinition {
+    for (const commandName in Foundation) {
+        const command = Foundation[commandName as FoundationCommandName];
+
+        if (command.ID === id) {
+            return command;
+        }
+    }
+
+    throw new Error(`Foundation command '${id}' does not exist.`);
+}
+
+export function isFoundationDiscoverRsp(id: number): boolean {
+    return FOUNDATION_DISCOVER_RSP_IDS.includes(id);
+}
+
+/** Check if value is equal to either min, max, minRef or maxRef */
+function isMinOrMax<T>(entry: Attribute | Parameter, value: T): boolean {
+    if (value === entry.max || value === entry.min) {
+        return true;
+    }
+
+    return false;
+}
+
+function processRestrictions<T>(entry: Attribute | Parameter, value: T): void {
+    if (entry.min !== undefined && (value as number) < entry.min) {
+        throw new Error(`${entry.name} requires min of ${entry.min}`);
+    }
+
+    if (entry.minExcl !== undefined && (value as number) <= entry.minExcl) {
+        throw new Error(`${entry.name} requires min exclusive of ${entry.minExcl}`);
+    }
+
+    if (entry.max !== undefined && (value as number) > entry.max) {
+        throw new Error(`${entry.name} requires max of ${entry.max}`);
+    }
+
+    if (entry.maxExcl !== undefined && (value as number) >= entry.maxExcl) {
+        throw new Error(`${entry.name} requires max exclusive of ${entry.maxExcl}`);
+    }
+
+    if (entry.length !== undefined && (value as string | unknown[] | Buffer).length !== entry.length) {
+        throw new Error(`${entry.name} requires length of ${entry.length}`);
+    }
+
+    if (entry.minLen !== undefined && (value as string | unknown[] | Buffer).length < entry.minLen) {
+        throw new Error(`${entry.name} requires min length of ${entry.minLen}`);
+    }
+
+    if (entry.maxLen !== undefined && (value as string | unknown[] | Buffer).length > entry.maxLen) {
+        throw new Error(`${entry.name} requires max length of ${entry.maxLen}`);
+    }
+}
+
+export function processAttributeWrite<T>(attribute: Attribute, value: T): T {
+    if (attribute.write !== true) {
+        throw new Error(`Attribute ${attribute.name} (${attribute.ID}) is not writable`);
+    }
+
+    if (value == null) {
+        return attribute.default !== undefined ? (attribute.default as T) : value /* XXX: dangerous fallback */;
+    }
+
+    // if default, always valid
+    if (attribute.default === value) {
+        return value;
+    }
+
+    if (Number.isNaN(value)) {
+        if (attribute.default === undefined) {
+            const nonValue = ZCL_TYPE_INVALID_BY_TYPE[attribute.type];
+
+            if (nonValue === undefined) {
+                throw new Error(`Attribute ${attribute.name} (${attribute.ID}) does not have a default nor a non-value`);
+            }
+
+            return nonValue as T;
+        }
+
+        return attribute.default as T;
+    }
+
+    processRestrictions(attribute, value);
+
+    return value;
+}
+
+export function processAttributePreRead(attribute: Attribute): void {
+    if (attribute.read === false) {
+        throw new Error(`Attribute ${attribute.name} (${attribute.ID}) is not readable`);
+    }
+}
+
+export function processAttributePostRead<T>(attribute: Attribute, value: T): T {
+    // should never happen?
+    if (value == null) {
+        return value;
+    }
+
+    // if default, always valid
+    if (attribute.default === value) {
+        return value;
+    }
+
+    // if type does not have an `invalid` (undefined) it won't match since value is checked above
+    if (value === ZCL_TYPE_INVALID_BY_TYPE[attribute.type]) {
+        // if value is same as max or min, ignore invalid sentinel
+        if (isMinOrMax(attribute, value)) {
+            return value;
+        }
+
+        // return NaN for both number & bigint to keep logic consistent
+        return Number.NaN as T;
+    }
+
+    processRestrictions(attribute, value);
+
+    return value;
+}
+
+export function processParameterWrite<T>(parameter: Parameter, value: T): T {
+    // should never happen?
+    if (value == null) {
+        return value;
+    }
+
+    if (Number.isNaN(value)) {
+        const nonValue = ZCL_TYPE_INVALID_BY_TYPE[parameter.type];
+
+        if (nonValue === undefined) {
+            throw new Error(`Parameter ${parameter.name} does not have a non-value`);
+        }
+
+        return nonValue as T;
+    }
+
+    processRestrictions(parameter, value);
+
+    return value;
+}
+
+export function processParameterRead<T>(parameter: Parameter, value: T): T {
+    // should never happen?
+    if (value == null) {
+        return value;
+    }
+
+    // if type does not have an `invalid` (undefined) it won't match since value is checked above
+    if (value === ZCL_TYPE_INVALID_BY_TYPE[parameter.type]) {
+        // if value is same as max or min, ignore invalid sentinel
+        if (isMinOrMax(parameter, value)) {
+            return value;
+        }
+
+        // return NaN for both number & bigint to keep logic consistent
+        return Number.NaN as T;
+    }
+
+    processRestrictions(parameter, value);
+
+    return value;
+}