mirror of
https://github.com/espressif/esp-matter.git
synced 2026-04-27 11:03:05 +00:00
Mahesh Pimpale
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
# Copyright 2026 Espressif Systems (Shanghai) PTE LTD
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""
|
|
Parse context: pre-loaded metadata used during cluster parsing.
|
|
Loads JSON artifacts once per run to avoid repeated file I/O.
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
|
|
from utils.config import ARTIFACTS_DIR, FileNames
|
|
from utils.conversion_utils import hex_to_int
|
|
|
|
|
|
def load_cluster_parse_context(output_dir: str):
|
|
"""Load all cluster-related metadata from output_dir into a single context."""
|
|
|
|
def _load_json(filename: str, base_dir: str = output_dir):
|
|
path = os.path.join(base_dir, filename)
|
|
with open(path, "r") as f:
|
|
return json.load(f)
|
|
|
|
return ClusterParseContext(
|
|
delegate_clusters=_load_json(FileNames.DELEGATE_CLUSTERS.value),
|
|
plugin_init_cb_clusters=_load_json(FileNames.PLUGIN_INIT_CB_CLUSTERS.value),
|
|
migrated_clusters=_load_json(
|
|
FileNames.MIGRATED_CLUSTERS.value, base_dir=ARTIFACTS_DIR
|
|
)["migrated_cluster"],
|
|
zap_filter_list=_load_json(FileNames.ZAP_FILTER_LIST.value),
|
|
internally_managed_attributes=_load_json(
|
|
FileNames.INTERNALLY_MANAGED_ATTRIBUTES.value
|
|
),
|
|
)
|
|
|
|
|
|
class ClusterParseContext:
|
|
"""Holds pre-loaded metadata for cluster parsing. Built once per run."""
|
|
|
|
def __init__(
|
|
self,
|
|
delegate_clusters: list,
|
|
plugin_init_cb_clusters: list,
|
|
migrated_clusters: list,
|
|
zap_filter_list: dict,
|
|
internally_managed_attributes: dict,
|
|
):
|
|
self.delegate_clusters = delegate_clusters
|
|
self.plugin_init_cb_clusters = plugin_init_cb_clusters
|
|
self.migrated_clusters = migrated_clusters
|
|
self.zap_filter_list = zap_filter_list
|
|
self.internally_managed_attributes = internally_managed_attributes
|
|
|
|
def get_allowed_attributes(self, cluster_id: str):
|
|
"""Return allowed attributes for cluster_id."""
|
|
cluster_info = self.zap_filter_list.get("clusters", {}).get(cluster_id, {})
|
|
if not cluster_info:
|
|
return []
|
|
return hex_to_int(list(cluster_info.get("Attributes", {}).values()))
|
|
|
|
def get_allowed_commands(self, cluster_id: str):
|
|
"""Return allowed commands for cluster_id."""
|
|
cluster_info = self.zap_filter_list.get("clusters", {}).get(cluster_id, {})
|
|
if not cluster_info:
|
|
return []
|
|
return hex_to_int(list(cluster_info.get("Commands", {}).values()))
|
|
|
|
def get_allowed_events(self, cluster_id: str):
|
|
"""Return allowed events for cluster_id."""
|
|
cluster_info = self.zap_filter_list.get("clusters", {}).get(cluster_id, {})
|
|
if not cluster_info:
|
|
return []
|
|
return hex_to_int(list(cluster_info.get("Events", {}).values()))
|
|
|
|
def get_internally_managed_attributes(self, esp_name: str) -> list:
|
|
"""Return list of internally managed attribute names for the cluster."""
|
|
return self.internally_managed_attributes.get(esp_name, [])
|