awslabs.iam-mcp-server 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

awslabs/iam_mcp_server/__init__.py

@@ -12,6 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""AWS IAM MCP Server."""
+"""AWS IAM MCP Server package."""
 
 __version__ = '1.0.0'

awslabs/iam_mcp_server/context.py

@@ -48,3 +48,8 @@ class Context:
     def set_region(cls, region: str):
         """Set the AWS region."""
         cls._region = region
+
+    @classmethod
+    def set_readonly(cls, readonly: bool):
+        """Set the read-only mode."""
+        cls._readonly = readonly

awslabs/iam_mcp_server/models.py

@@ -70,6 +70,16 @@ class IamPolicy(BaseModel):
     update_date: str = Field(..., description='The date and time when the policy was last updated')
 
 
+class IamGroup(BaseModel):
+    """IAM Group model."""
+
+    group_name: str = Field(..., description='The name of the IAM group')
+    group_id: str = Field(..., description='The unique identifier for the group')
+    arn: str = Field(..., description='The Amazon Resource Name (ARN) of the group')
+    path: str = Field(..., description='The path to the group')
+    create_date: str = Field(..., description='The date and time when the group was created')
+
+
 class AccessKey(BaseModel):
     """IAM Access Key model."""
 
@@ -195,3 +205,86 @@ class PolicySimulationResponse(BaseModel):
     is_truncated: bool = Field(False, description='Whether the response is truncated')
     marker: Optional[str] = Field(None, description='Marker for pagination')
     policy_source_arn: str = Field(..., description='ARN of the principal that was simulated')
+
+
+class GroupDetailsResponse(BaseModel):
+    """Response model for detailed group information."""
+
+    group: IamGroup = Field(..., description='Group details')
+    users: List[str] = Field(default_factory=list, description='List of user names in the group')
+    attached_policies: List[AttachedPolicy] = Field(
+        default_factory=list, description='List of attached managed policies'
+    )
+    inline_policies: List[str] = Field(
+        default_factory=list, description='List of inline policy names'
+    )
+
+
+class GroupsListResponse(BaseModel):
+    """Response model for listing groups."""
+
+    groups: List[IamGroup] = Field(..., description='List of IAM groups')
+    is_truncated: bool = Field(False, description='Whether the response is truncated')
+    marker: Optional[str] = Field(None, description='Marker for pagination')
+    count: int = Field(..., description='Number of groups returned')
+
+
+class CreateGroupResponse(BaseModel):
+    """Response model for creating a group."""
+
+    group: IamGroup = Field(..., description='Created group details')
+    message: str = Field(..., description='Success message')
+
+
+class GroupMembershipResponse(BaseModel):
+    """Response model for group membership operations."""
+
+    message: str = Field(..., description='Operation result message')
+    group_name: str = Field(..., description='The name of the group')
+    user_name: str = Field(..., description='The name of the user')
+
+
+class GroupPolicyAttachmentResponse(BaseModel):
+    """Response model for group policy attachment operations."""
+
+    message: str = Field(..., description='Operation result message')
+    group_name: str = Field(..., description='The name of the group')
+    policy_arn: str = Field(..., description='The ARN of the policy')
+
+
+class InlinePolicy(BaseModel):
+    """Inline Policy model."""
+
+    policy_name: str = Field(..., description='The name of the inline policy')
+    policy_document: str = Field(..., description='The policy document in JSON format')
+
+
+class InlinePolicyResponse(BaseModel):
+    """Response model for inline policy operations."""
+
+    policy_name: str = Field(..., description='The name of the policy')
+    policy_document: str = Field(..., description='The policy document in JSON format')
+    user_name: Optional[str] = Field(None, description='The name of the user (for user policies)')
+    role_name: Optional[str] = Field(None, description='The name of the role (for role policies)')
+    message: str = Field(..., description='Operation result message')
+
+
+class InlinePolicyListResponse(BaseModel):
+    """Response model for listing inline policies."""
+
+    policy_names: List[str] = Field(..., description='List of inline policy names')
+    user_name: Optional[str] = Field(None, description='The name of the user (for user policies)')
+    role_name: Optional[str] = Field(None, description='The name of the role (for role policies)')
+    count: int = Field(..., description='Number of policies returned')
+
+
+class ManagedPolicyResponse(BaseModel):
+    """Response model for managed policy document operations."""
+
+    policy_arn: str = Field(..., description='The ARN of the managed policy')
+    policy_name: str = Field(..., description='The name of the policy')
+    version_id: str = Field(..., description='The version ID of the policy')
+    policy_document: str = Field(..., description='The policy document in JSON format')
+    is_default_version: bool = Field(..., description='Whether this is the default version')
+    create_date: str = Field(..., description='The date and time when this version was created')
+    message: str = Field(..., description='Operation result message')