cartography 0.106.0rc2__py3-none-any.whl → 0.107.0rc2__py3-none-any.whl

cartography/models/airbyte/source.py

@@ -0,0 +1,75 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AirbyteSourceNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("sourceId")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    name: PropertyRef = PropertyRef("name")
+    type: PropertyRef = PropertyRef("sourceType")
+    config_host: PropertyRef = PropertyRef("configuration.host")
+    config_port: PropertyRef = PropertyRef("configuration.port")
+    config_name: PropertyRef = PropertyRef("configuration.name")
+    config_region: PropertyRef = PropertyRef("configuration.region")
+    config_endpoint: PropertyRef = PropertyRef("configuration.endpoint")
+    config_account: PropertyRef = PropertyRef("configuration.account")
+
+
+@dataclass(frozen=True)
+class AirbyteSourceToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteOrganization)-[:RESOURCE]->(:AirbyteSource)
+class AirbyteSourceToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AirbyteSourceToOrganizationRelProperties = (
+        AirbyteSourceToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteSourceToWorkspaceRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteWorkspace)-[:CONTAINS]->(:AirbyteSource)
+class AirbyteSourceToWorkspaceRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteWorkspace"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("workspaceId")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AirbyteSourceToWorkspaceRelProperties = (
+        AirbyteSourceToWorkspaceRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteSourceSchema(CartographyNodeSchema):
+    label: str = "AirbyteSource"
+    properties: AirbyteSourceNodeProperties = AirbyteSourceNodeProperties()
+    sub_resource_relationship: AirbyteSourceToOrganizationRel = (
+        AirbyteSourceToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [AirbyteSourceToWorkspaceRel()]
+    )

cartography/models/airbyte/stream.py

@@ -0,0 +1,74 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AirbyteStreamNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("streamId")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    name: PropertyRef = PropertyRef("name")
+    sync_mode: PropertyRef = PropertyRef("syncMode")
+    cursor_field: PropertyRef = PropertyRef("cursorField")
+    primary_key: PropertyRef = PropertyRef("primaryKey")
+    include_files: PropertyRef = PropertyRef("includeFiles")
+    selected_fields: PropertyRef = PropertyRef("selectedFields")
+    mappers: PropertyRef = PropertyRef("mappers")
+
+
+@dataclass(frozen=True)
+class AirbyteStreamToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteOrganization)-[:RESOURCE]->(:AirbyteStream)
+class AirbyteStreamToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AirbyteStreamToOrganizationRelProperties = (
+        AirbyteStreamToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteStreamToConnectionRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteConnection)-[:HAS]->(:AirbyteStream)
+class AirbyteStreamToConnectionRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteConnection"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("connectionId")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "HAS"
+    properties: AirbyteStreamToOrganizationRelProperties = (
+        AirbyteStreamToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteStreamSchema(CartographyNodeSchema):
+    label: str = "AirbyteStream"
+    properties: AirbyteStreamNodeProperties = AirbyteStreamNodeProperties()
+    sub_resource_relationship: AirbyteStreamToOrganizationRel = (
+        AirbyteStreamToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [AirbyteStreamToConnectionRel()]
+    )

cartography/models/airbyte/tag.py

@@ -0,0 +1,69 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AirbyteTagNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("tagId")
+    name: PropertyRef = PropertyRef("name")
+    color: PropertyRef = PropertyRef("color")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AirbyteTagToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteOrganization)-[:RESOURCE]->(:AirbyteTag)
+class AirbyteTagToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AirbyteTagToOrganizationRelProperties = (
+        AirbyteTagToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteTagToWorkspaceRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteWorkspace)-[:CONTAINS]->(:AirbyteTag)
+class AirbyteTagToWorkspaceRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteWorkspace"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("workspaceId")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "CONTAINS"
+    properties: AirbyteTagToWorkspaceRelProperties = (
+        AirbyteTagToWorkspaceRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteTagSchema(CartographyNodeSchema):
+    label: str = "AirbyteTag"
+    properties: AirbyteTagNodeProperties = AirbyteTagNodeProperties()
+    sub_resource_relationship: AirbyteTagToOrganizationRel = (
+        AirbyteTagToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [AirbyteTagToWorkspaceRel()]
+    )

cartography/models/airbyte/user.py

@@ -0,0 +1,111 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AirbyteUserNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    email: PropertyRef = PropertyRef("email", extra_index=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AirbyteUserToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteOrganization)-[:RESOURCE]->(:AirbyteUser)
+class AirbyteUserToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AirbyteUserToOrganizationRelProperties = (
+        AirbyteUserToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteUserToOrganizationAdminRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteOrganization)<-[:ADMIN_OF]-(:AirbyteUser)
+class AirbyteUserToOrganizationAdminRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("adminOfOrganization", one_to_many=True)},
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "ADMIN_OF"
+    properties: AirbyteUserToOrganizationAdminRelProperties = (
+        AirbyteUserToOrganizationAdminRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteUserToWorkspaceAdminRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteWorkspace)<-[:ADMIN_OF]-(:AirbyteUser)
+class AirbyteUserToWorkspaceAdminRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteWorkspace"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("adminOfWorkspace", one_to_many=True)},
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "ADMIN_OF"
+    properties: AirbyteUserToWorkspaceAdminRelProperties = (
+        AirbyteUserToWorkspaceAdminRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteUserToWorkspaceMemberRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteWorkspace)<-[:MEMBER_OF]-(:AirbyteUser)
+class AirbyteUserToWorkspaceMemberRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteWorkspace"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("memberOfWorkspace", one_to_many=True)},
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "MEMBER_OF"
+    properties: AirbyteUserToWorkspaceMemberRelProperties = (
+        AirbyteUserToWorkspaceMemberRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteUserSchema(CartographyNodeSchema):
+    label: str = "AirbyteUser"
+    properties: AirbyteUserNodeProperties = AirbyteUserNodeProperties()
+    sub_resource_relationship: AirbyteUserToOrganizationRel = (
+        AirbyteUserToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [
+            AirbyteUserToOrganizationAdminRel(),
+            AirbyteUserToWorkspaceAdminRel(),
+            AirbyteUserToWorkspaceMemberRel(),
+        ]
+    )

cartography/models/airbyte/workspace.py

@@ -0,0 +1,46 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AirbyteWorkspaceNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("workspaceId")
+    name: PropertyRef = PropertyRef("name")
+    data_residency: PropertyRef = PropertyRef("dataResidency")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class AirbyteWorkspaceToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:AirbyteOrganization)-[:RESOURCE]->(:AirbyteWorkspace)
+class AirbyteWorkspaceToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "AirbyteOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: AirbyteWorkspaceToOrganizationRelProperties = (
+        AirbyteWorkspaceToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class AirbyteWorkspaceSchema(CartographyNodeSchema):
+    label: str = "AirbyteWorkspace"
+    properties: AirbyteWorkspaceNodeProperties = AirbyteWorkspaceNodeProperties()
+    sub_resource_relationship: AirbyteWorkspaceToOrganizationRel = (
+        AirbyteWorkspaceToOrganizationRel()
+    )

cartography/models/aws/cloudtrail/management_events.py

@@ -0,0 +1,64 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_source_node_matcher
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import SourceNodeMatcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class AssumedRoleRelProperties(CartographyRelProperties):
+    """
+    Properties for the ASSUMED_ROLE relationship representing role assumption events.
+    Matches the cloudtrail_management_events spec and adds enhanced temporal precision.
+    """
+
+    # Mandatory fields for MatchLinks
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    _sub_resource_label: PropertyRef = PropertyRef(
+        "_sub_resource_label", set_in_kwargs=True
+    )
+    _sub_resource_id: PropertyRef = PropertyRef("_sub_resource_id", set_in_kwargs=True)
+
+    # CloudTrail-specific relationship properties
+    last_used: PropertyRef = PropertyRef("last_used")
+    times_used: PropertyRef = PropertyRef("times_used")
+    first_seen_in_time_window: PropertyRef = PropertyRef("first_seen_in_time_window")
+
+    # Event type tracking properties
+    event_types: PropertyRef = PropertyRef("event_types")
+    assume_role_count: PropertyRef = PropertyRef("assume_role_count")
+    saml_count: PropertyRef = PropertyRef("saml_count")
+    web_identity_count: PropertyRef = PropertyRef("web_identity_count")
+
+
+@dataclass(frozen=True)
+class AssumedRoleMatchLink(CartographyRelSchema):
+    """
+    MatchLink schema for ASSUMED_ROLE relationships from CloudTrail events.
+    Creates relationships like: (AWSUser|AWSRole|AWSPrincipal)-[:ASSUMED_ROLE]->(AWSRole)
+
+    This MatchLink handles role assumption relationships discovered via CloudTrail management events.
+    It supports multiple source node types and aggregated relationship properties.
+    """
+
+    # MatchLink-specific fields
+    source_node_label: str = (
+        "AWSPrincipal"  # Base type that covers AWSUser, AWSRole, AWSPrincipal
+    )
+    source_node_matcher: SourceNodeMatcher = make_source_node_matcher(
+        {"arn": PropertyRef("source_principal_arn")},
+    )
+
+    # Standard CartographyRelSchema fields
+    target_node_label: str = "AWSRole"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"arn": PropertyRef("destination_principal_arn")},
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "ASSUMED_ROLE"
+    properties: AssumedRoleRelProperties = AssumedRoleRelProperties()

cartography/models/aws/codebuild/project.py

@@ -0,0 +1,49 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class CodeBuildProjectNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("arn")
+    arn: PropertyRef = PropertyRef("arn", extra_index=True)
+    region: PropertyRef = PropertyRef("Region", set_in_kwargs=True)
+    created: PropertyRef = PropertyRef("created")
+    environment_variables: PropertyRef = PropertyRef("environmentVariables")
+    source_type: PropertyRef = PropertyRef("sourceType")
+    source_location: PropertyRef = PropertyRef("sourceLocation")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CodeBuildProjectToAwsAccountRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class CodeBuildProjectToAWSAccountRel(CartographyRelSchema):
+    target_node_label: str = "AWSAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("AWS_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: CodeBuildProjectToAwsAccountRelProperties = (
+        CodeBuildProjectToAwsAccountRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class CodeBuildProjectSchema(CartographyNodeSchema):
+    label: str = "CodeBuildProject"
+    properties: CodeBuildProjectNodeProperties = CodeBuildProjectNodeProperties()
+    sub_resource_relationship: CodeBuildProjectToAWSAccountRel = (
+        CodeBuildProjectToAWSAccountRel()
+    )

cartography/models/aws/ecs/containers.py

@@ -66,6 +66,24 @@ class ECSContainerToTaskRel(CartographyRelSchema):
     properties: ECSContainerToTaskRelProperties = ECSContainerToTaskRelProperties()
 
 
+@dataclass(frozen=True)
+class ECSContainerToECRImageRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class ECSContainerToECRImageRel(CartographyRelSchema):
+    target_node_label: str = "ECRImage"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"digest": PropertyRef("imageDigest")}
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "HAS_IMAGE"
+    properties: ECSContainerToECRImageRelProperties = (
+        ECSContainerToECRImageRelProperties()
+    )
+
+
 @dataclass(frozen=True)
 class ECSContainerSchema(CartographyNodeSchema):
     label: str = "ECSContainer"
@@ -76,5 +94,6 @@ class ECSContainerSchema(CartographyNodeSchema):
     other_relationships: OtherRelationships = OtherRelationships(
         [
             ECSContainerToTaskRel(),
+            ECSContainerToECRImageRel(),
         ]
     )

cartography/models/aws/ecs/task_definitions.py

@@ -83,6 +83,42 @@ class ECSTaskDefinitionToECSTaskRel(CartographyRelSchema):
     )
 
 
+@dataclass(frozen=True)
+class ECSTaskDefinitionToTaskRoleRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class ECSTaskDefinitionToTaskRoleRel(CartographyRelSchema):
+    target_node_label: str = "AWSRole"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"arn": PropertyRef("taskRoleArn")}
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "HAS_TASK_ROLE"
+    properties: ECSTaskDefinitionToTaskRoleRelProperties = (
+        ECSTaskDefinitionToTaskRoleRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class ECSTaskDefinitionToExecutionRoleRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class ECSTaskDefinitionToExecutionRoleRel(CartographyRelSchema):
+    target_node_label: str = "AWSRole"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"arn": PropertyRef("executionRoleArn")}
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "HAS_EXECUTION_ROLE"
+    properties: ECSTaskDefinitionToExecutionRoleRelProperties = (
+        ECSTaskDefinitionToExecutionRoleRelProperties()
+    )
+
+
 @dataclass(frozen=True)
 class ECSTaskDefinitionSchema(CartographyNodeSchema):
     label: str = "ECSTaskDefinition"
@@ -93,5 +129,7 @@ class ECSTaskDefinitionSchema(CartographyNodeSchema):
     other_relationships: OtherRelationships = OtherRelationships(
         [
             ECSTaskDefinitionToECSTaskRel(),
+            ECSTaskDefinitionToTaskRoleRel(),
+            ECSTaskDefinitionToExecutionRoleRel(),
         ]
     )

cartography/models/aws/inspector/findings.py

@@ -7,8 +7,10 @@ from cartography.models.core.nodes import ExtraNodeLabels
 from cartography.models.core.relationships import CartographyRelProperties
 from cartography.models.core.relationships import CartographyRelSchema
 from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_source_node_matcher
 from cartography.models.core.relationships import make_target_node_matcher
 from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import SourceNodeMatcher
 from cartography.models.core.relationships import TargetNodeMatcher
 
 
@@ -135,6 +137,40 @@ class InspectorFindingToECRImageRel(CartographyRelSchema):
     )
 
 
+@dataclass(frozen=True)
+class InspectorFindingToPackageRelRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    _sub_resource_label: PropertyRef = PropertyRef(
+        "_sub_resource_label", set_in_kwargs=True
+    )
+    _sub_resource_id: PropertyRef = PropertyRef("_sub_resource_id", set_in_kwargs=True)
+    # The following properties live in vulnerablePackages from AWS API
+    # Adding them here to avoid multiple repetion of packages
+    filepath: PropertyRef = PropertyRef("filePath")
+    fixedinversion: PropertyRef = PropertyRef("fixedInVersion")
+    remediation: PropertyRef = PropertyRef("remediation")
+    sourcelayerhash: PropertyRef = PropertyRef("sourceLayerHash")
+    sourcelambdalayerarn: PropertyRef = PropertyRef("sourceLambdaLayerArn")
+
+
+@dataclass(frozen=True)
+# (:AWSInspectorFinding)-[:HAS]->(:AWSInspectorPackage)
+class InspectorFindingToPackageMatchLink(CartographyRelSchema):
+    target_node_label: str = "AWSInspectorPackage"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("packageid")},
+    )
+    source_node_label: str = "AWSInspectorFinding"
+    source_node_matcher: SourceNodeMatcher = make_source_node_matcher(
+        {"id": PropertyRef("findingarn")},
+    )
+    properties: InspectorFindingToPackageRelRelProperties = (
+        InspectorFindingToPackageRelRelProperties()
+    )
+    direction: LinkDirection = LinkDirection.OUTWARD
+    rel_label: str = "HAS"
+
+
 @dataclass(frozen=True)
 class AWSInspectorFindingSchema(CartographyNodeSchema):
     label: str = "AWSInspectorFinding"
@@ -146,6 +182,7 @@ class AWSInspectorFindingSchema(CartographyNodeSchema):
     other_relationships: OtherRelationships = OtherRelationships(
         [
             InspectorFindingToEC2InstanceRel(),
+            # TODO: Fix ECRRepository and ECRImage relationships
             InspectorFindingToECRRepositoryRel(),
             InspectorFindingToECRImageRel(),
             InspectorFindingToAWSAccountRelDelegateRel(),