fix: Tool Registration & Naming Updates (#14)

* fix: Tool Registration & Naming Updates

* fix: Fixed Ruff Linter

Author: willguibr

CHANGELOG.md

@@ -1,5 +1,41 @@
 # Zscaler Integrations MCP Server Changelog
 
+## 0.3.1 (October 28, 2025) - Tool Registration & Naming Updates
+
+### Added
+
+[PR #14](https://github.com/zscaler/zscaler-mcp-server/pull/14)
+
+- **ZIA Tools**: Added missing read-only tools to services registration
+  - `get_zia_dlp_dictionaries` - Manage ZIA DLP dictionaries for data loss prevention
+  - `get_zia_dlp_engines` - Manage ZIA DLP engines for rule processing
+  - `get_zia_user_departments` - Manage ZIA user departments for organizational structure
+  - `get_zia_user_groups` - Manage ZIA user groups for access control
+  - `get_zia_users` - Manage ZIA users for authentication and access control
+
+- **ZPA Tools**: Added missing read-only tools to services registration
+  - `get_zpa_app_protection_profile` - Manage ZPA App Protection Profiles (Inspection Profiles)
+  - `get_zpa_enrollment_certificate` - Manage ZPA Enrollment Certificates
+  - `get_zpa_isolation_profile` - Manage ZPA Cloud Browser Isolation (CBI) profiles
+  - `get_zpa_posture_profile` - Manage ZPA Posture Profiles
+  - `get_zpa_saml_attribute` - Manage ZPA SAML Attributes
+  - `get_zpa_scim_attribute` - Manage ZPA SCIM Attributes
+  - `get_zpa_scim_group` - Manage ZPA SCIM Groups
+  - `get_zpa_app_segments_by_type` - Manage ZPA application segments by type
+  - `get_zpa_trusted_network` - Manage ZPA Trusted Networks
+
+### Changed
+
+- **Tool Naming Convention**: Updated tool names to follow consistent `get_*` pattern for read-only operations
+  - ZIA tools: `zia_*_manager` → `get_zia_*`
+  - ZPA tools: `*_manager` → `get_zpa_*`
+  - Maintains backward compatibility with existing `zia_get_*` and `zpa_get_*` patterns
+
+### Fixed
+
+- **Tool Registration**: Resolved missing tool registrations in `zscaler_mcp/services.py`
+- **Documentation**: Updated README.md with correct tool names and comprehensive tool listings
+
 ## 0.3.0 (October 27, 2025) - Security & Confirmation Release
 
 ### Notes

docsrc/guides/release-notes.rst

@@ -6,6 +6,44 @@ Release Notes
 Zscaler Integrations MCP Server Changelog
 ------------------------------------------
 
+0.3.1 (October 28, 2025) - Tool Registration & Naming Updates
+--------------------------------------------------------------
+
+Added
+------
+
+`PR #14 <https://github.com/zscaler/zscaler-mcp-server/pull/14>`_
+
+- **ZIA Tools**: Added missing read-only tools to services registration
+  - `get_zia_dlp_dictionaries` - Manage ZIA DLP dictionaries for data loss prevention
+  - `get_zia_dlp_engines` - Manage ZIA DLP engines for rule processing
+  - `get_zia_user_departments` - Manage ZIA user departments for organizational structure
+  - `get_zia_user_groups` - Manage ZIA user groups for access control
+  - `get_zia_users` - Manage ZIA users for authentication and access control
+
+- **ZPA Tools**: Added missing read-only tools to services registration
+  - `get_zpa_app_protection_profile` - Manage ZPA App Protection Profiles (Inspection Profiles)
+  - `get_zpa_enrollment_certificate` - Manage ZPA Enrollment Certificates
+  - `get_zpa_isolation_profile` - Manage ZPA Cloud Browser Isolation (CBI) profiles
+  - `get_zpa_posture_profile` - Manage ZPA Posture Profiles
+  - `get_zpa_saml_attribute` - Manage ZPA SAML Attributes
+  - `get_zpa_scim_attribute` - Manage ZPA SCIM Attributes
+  - `get_zpa_scim_group` - Manage ZPA SCIM Groups
+  - `get_zpa_app_segments_by_type` - Manage ZPA application segments by type
+  - `get_zpa_trusted_network` - Manage ZPA Trusted Networks
+
+### Changed
+
+- **Tool Naming Convention**: Updated tool names to follow consistent `get_*` pattern for read-only operations
+  - ZIA tools: `zia_*_manager` → `get_zia_*`
+  - ZPA tools: `*_manager` → `get_zpa_*`
+  - Maintains backward compatibility with existing `zia_get_*` and `zpa_get_*` patterns
+
+### Fixed
+
+- **Tool Registration**: Resolved missing tool registrations in `zscaler_mcp/services.py`
+- **Documentation**: Updated README.md with correct tool names and comprehensive tool listings
+
 0.3.0 (October 27, 2025) - Security & Confirmation Release
 -----------------------------------------------------------
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zscaler-mcp"
-version = "0.3.0"
+version = "0.3.1"
 description = "Official Zscaler Integrations MCP Server"
 readme = "README.md"
 requires-python = ">=3.11,<4.0"

zscaler_mcp/services.py

@@ -71,11 +71,11 @@ def __init__(self, zscaler_client):
     def register_tools(self, server, enabled_tools=None, enable_write_tools=False, write_tools=None):
         """Register ZCC tools with the server."""
         from zscaler_mcp.common.tool_helpers import register_read_tools, register_write_tools
-        
+
         # Register verb-based tools with proper annotations (all read-only)
         read_count = register_read_tools(server, self.read_tools, enabled_tools)
         write_count = register_write_tools(server, self.write_tools, enabled_tools, enable_write_tools, write_tools)
-        
+
         logger.info(f"ZCC Service: Registered {read_count} read tools, {write_count} write tools")
 
 
@@ -136,11 +136,11 @@ def __init__(self, zscaler_client):
     def register_tools(self, server, enabled_tools=None, enable_write_tools=False, write_tools=None):
         """Register ZDX tools with the server."""
         from zscaler_mcp.common.tool_helpers import register_read_tools, register_write_tools
-        
+
         # Register verb-based tools with proper annotations (all read-only)
         read_count = register_read_tools(server, self.read_tools, enabled_tools)
         write_count = register_write_tools(server, self.write_tools, enabled_tools, enable_write_tools, write_tools)
-        
+
         logger.info(f"ZDX Service: Registered {read_count} read tools, {write_count} write tools")
 
 
@@ -212,6 +212,15 @@ def __init__(self, zscaler_client):
             zpa_get_ba_certificate,
             zpa_list_ba_certificates,
         )
+        from .tools.zpa.get_app_protection_profile import app_protection_profile_manager
+        from .tools.zpa.get_enrollment_certificate import enrollment_certificate_manager
+        from .tools.zpa.get_isolation_profile import isolation_profile_manager
+        from .tools.zpa.get_posture_profiles import posture_profile_manager
+        from .tools.zpa.get_saml_attributes import saml_attribute_manager
+        from .tools.zpa.get_scim_attributes import scim_attribute_manager
+        from .tools.zpa.get_scim_groups import scim_group_manager
+        from .tools.zpa.get_segments_by_type import app_segments_by_type_manager
+        from .tools.zpa.get_trusted_networks import trusted_network_manager
         from .tools.zpa.pra_credential import (
             zpa_create_pra_credential,
             zpa_delete_pra_credential,
@@ -287,6 +296,18 @@ def __init__(self, zscaler_client):
             {"func": zpa_get_pra_portal, "name": "zpa_get_pra_portal", "description": "Get a specific ZPA PRA portal by ID (read-only)"},
             {"func": zpa_list_pra_credentials, "name": "zpa_list_pra_credentials", "description": "List ZPA PRA credentials (read-only)"},
             {"func": zpa_get_pra_credential, "name": "zpa_get_pra_credential", "description": "Get a specific ZPA PRA credential by ID (read-only)"},
+            # Profile and Certificate Management
+            {"func": app_protection_profile_manager, "name": "get_zpa_app_protection_profile", "description": "Manage ZPA App Protection Profiles (Inspection Profiles) (read-only)"},
+            {"func": enrollment_certificate_manager, "name": "get_zpa_enrollment_certificate", "description": "Manage ZPA Enrollment Certificates (read-only)"},
+            {"func": isolation_profile_manager, "name": "get_zpa_isolation_profile", "description": "Manage ZPA Cloud Browser Isolation (CBI) profiles (read-only)"},
+            {"func": posture_profile_manager, "name": "get_zpa_posture_profile", "description": "Manage ZPA Posture Profiles (read-only)"},
+            # Identity and Access Management
+            {"func": saml_attribute_manager, "name": "get_zpa_saml_attribute", "description": "Manage ZPA SAML Attributes (read-only)"},
+            {"func": scim_attribute_manager, "name": "get_zpa_scim_attribute", "description": "Manage ZPA SCIM Attributes (read-only)"},
+            {"func": scim_group_manager, "name": "get_zpa_scim_group", "description": "Manage ZPA SCIM Groups (read-only)"},
+            # Network and Segment Management
+            {"func": app_segments_by_type_manager, "name": "get_zpa_app_segments_by_type", "description": "Manage ZPA application segments by type (read-only)"},
+            {"func": trusted_network_manager, "name": "get_zpa_trusted_network", "description": "Manage ZPA Trusted Networks (read-only)"},
         ]
 
         # Define write tools
@@ -340,11 +361,11 @@ def __init__(self, zscaler_client):
     def register_tools(self, server, enabled_tools=None, enable_write_tools=False, write_tools=None):
         """Register ZPA tools with the server."""
         from zscaler_mcp.common.tool_helpers import register_read_tools, register_write_tools
-        
+
         # Register verb-based tools with proper annotations
         read_count = register_read_tools(server, self.read_tools, enabled_tools)
         write_count = register_write_tools(server, self.write_tools, enabled_tools, enable_write_tools, write_tools)
-        
+
         logger.info(f"ZPA Service: Registered {read_count} read tools, {write_count} write tools")
 
 
@@ -398,6 +419,11 @@ def __init__(self, zscaler_client):
             zia_list_ip_source_groups,
             zia_update_ip_source_group,
         )
+        from .tools.zia.list_dlp_dictionaries import zia_dlp_dictionary_manager
+        from .tools.zia.list_dlp_engines import zia_dlp_engine_manager
+        from .tools.zia.list_user_departments import zia_user_department_manager
+        from .tools.zia.list_user_groups import zia_user_group_manager
+        from .tools.zia.list_users import zia_users_manager
         from .tools.zia.location_management import (
             zia_create_location,
             zia_delete_location,
@@ -470,6 +496,14 @@ def __init__(self, zscaler_client):
             {"func": zia_list_web_dlp_rules, "name": "zia_list_web_dlp_rules", "description": "List ZIA web DLP rules (read-only)"},
             {"func": zia_list_web_dlp_rules_lite, "name": "zia_list_web_dlp_rules_lite", "description": "List ZIA web DLP rules in lite format (read-only)"},
             {"func": zia_get_web_dlp_rule, "name": "zia_get_web_dlp_rule", "description": "Get a specific ZIA web DLP rule by ID (read-only)"},
+            # DLP Dictionaries
+            {"func": zia_dlp_dictionary_manager, "name": "get_zia_dlp_dictionaries", "description": "Manage ZIA DLP dictionaries for data loss prevention pattern and phrase matching (read-only)"},
+            # DLP Engines
+            {"func": zia_dlp_engine_manager, "name": "get_zia_dlp_engines", "description": "Manage ZIA DLP engines for data loss prevention rule processing (read-only)"},
+            # User Management
+            {"func": zia_user_department_manager, "name": "get_zia_user_departments", "description": "Manage ZIA user departments for organizational structure (read-only)"},
+            {"func": zia_user_group_manager, "name": "get_zia_user_groups", "description": "Manage ZIA user groups for access control and policy assignment (read-only)"},
+            {"func": zia_users_manager, "name": "get_zia_users", "description": "Manage ZIA users for authentication and access control (read-only)"},
             # IP Source Groups
             {"func": zia_list_ip_source_groups, "name": "zia_list_ip_source_groups", "description": "List ZIA IP source groups (read-only)"},
             {"func": zia_get_ip_source_group, "name": "zia_get_ip_source_group", "description": "Get a specific ZIA IP source group by ID (read-only)"},
@@ -575,11 +609,11 @@ def __init__(self, zscaler_client):
     def register_tools(self, server, enabled_tools=None, enable_write_tools=False, write_tools=None):
         """Register ZIA tools with the server."""
         from zscaler_mcp.common.tool_helpers import register_read_tools, register_write_tools
-        
+
         # Register verb-based tools with proper annotations
         read_count = register_read_tools(server, self.read_tools, enabled_tools)
         write_count = register_write_tools(server, self.write_tools, enabled_tools, enable_write_tools, write_tools)
-        
+
         logger.info(f"ZIA Service: Registered {read_count} read tools, {write_count} write tools")
 
 
@@ -637,11 +671,11 @@ def __init__(self, zscaler_client):
     def register_tools(self, server, enabled_tools=None, enable_write_tools=False, write_tools=None):
         """Register ZTW tools with the server."""
         from zscaler_mcp.common.tool_helpers import register_read_tools, register_write_tools
-        
+
         # Register verb-based tools with proper annotations
         read_count = register_read_tools(server, self.read_tools, enabled_tools)
         write_count = register_write_tools(server, self.write_tools, enabled_tools, enable_write_tools, write_tools)
-        
+
         logger.info(f"ZTW Service: Registered {read_count} read tools, {write_count} write tools")
 
 
@@ -685,11 +719,11 @@ def __init__(self, zscaler_client):
     def register_tools(self, server, enabled_tools=None, enable_write_tools=False, write_tools=None):
         """Register ZIdentity tools with the server."""
         from zscaler_mcp.common.tool_helpers import register_read_tools, register_write_tools
-        
+
         # Register verb-based tools with proper annotations (all read-only)
         read_count = register_read_tools(server, self.read_tools, enabled_tools)
         write_count = register_write_tools(server, self.write_tools, enabled_tools, enable_write_tools, write_tools)
-        
+
         logger.info(f"ZIdentity Service: Registered {read_count} read tools, {write_count} write tools")