From 36098673a92d1a132db9bbd5a1eb657d2e4a8765 Mon Sep 17 00:00:00 2001
From: Andrej Podhradsky <apodhrad@redhat.com>
Date: Fri, 22 Sep 2023 18:16:40 +0200
Subject: [PATCH] Add support for AWS Resource Explorer

Signed-off-by: Andrej Podhradsky <apodhrad@redhat.com>
---
 wrapanapi/systems/ec2.py | 98 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)

diff --git a/wrapanapi/systems/ec2.py b/wrapanapi/systems/ec2.py
index 3f15b27e..e7c4a352 100644
--- a/wrapanapi/systems/ec2.py
+++ b/wrapanapi/systems/ec2.py
@@ -1,6 +1,7 @@
 import base64
 import os
 import re
+import typing
 
 import boto3
 from boto3 import client as boto3client
@@ -481,6 +482,67 @@ def cleanup(self):
         """
         return self.delete()
 
+class ResourceExplorerResource():
+    """
+    This class represents a resource returned by Resource Explorer.
+    """
+    def __init__(self, arn, region, resource_type, service, properties=[]):
+        self.arn = arn
+        self.region = region
+        self.resource_type = resource_type
+        self.service = service
+        self.properties = properties
+
+    def get_tag_value(self, key) -> str:
+        """
+        Returns a tag value for a given tag key.
+        Tags are taken from the resource properties.
+
+        Args:
+            key: a tag key
+        """
+        tags = self.get_tags(regex=f"^{key}$")
+        if len(tags) > 0:
+            return tags[0].get("Value")
+        return None
+    
+    def get_tags(self, regex="") -> typing.List[dict]:
+        """
+        Returns a list of tags (a dict with keys 'Key' and 'Value').
+        Tags are taken from the resource properties.
+
+        Args:
+            regex: a regular expressions for keys, default is ""
+        """
+        list = []
+        for property in self.properties:
+            data = property.get("Data")
+            for tag in data:
+                key = tag.get("Key")
+                if re.match(regex, key):
+                    list.append(tag)
+        return list
+    
+    @property
+    def id(self) -> str:
+        """
+        Returns the last part of the arn.
+        This part is used as id in aws cli.
+        """
+        if self.arn:
+            return self.arn.split(":")[-1]
+        return None
+    
+    @property
+    def name(self) -> str:
+        """
+        Returns a name for the resource derived from the associated tag with key 'Name'.
+        If there is no such tag then the name is the id from arn.
+        """
+        name = self.get_tag_value("Name")
+        if not name:
+            name = self.id
+        return name
 
 class EC2System(System, VmMixin, TemplateMixin, StackMixin, NetworkMixin):
     """EC2 Management System, powered by boto
@@ -534,6 +596,7 @@ def __init__(self, **kwargs):
         self.ssm_connection = boto3client("ssm", **connection_kwargs)
         self.sns_connection = boto3client("sns", **connection_kwargs)
         self.cw_events_connection = boto3client("events", **connection_kwargs)
+        self.resource_explorer_connection = boto3client("resource-explorer-2", **connection_kwargs)
 
         self.kwargs = kwargs
 
@@ -1743,3 +1806,38 @@ def cleanup_resources(self):
         self.remove_all_unused_nics()
         self.remove_all_unused_volumes()
         self.remove_all_unused_ips()
+
+    def list_resources(self, query="", view="") -> typing.List[ResourceExplorerResource]:
+        """
+        Lists resources using AWS Resource Explorer (resource-explorer-2).
+
+        Args:
+            query: keywords and filters for resources; default is "" (all)
+            view: arn of the view to use for the query; default is "" (default view)
+
+        Return:
+            a list of resources satisfying the query
+
+        Examples:
+            Use query "tag.key:kubernetes.io/cluster/*" to list OCP resources
+        """
+        args = {
+            "QueryString":query
+        }
+        if view:
+            args["ViewArn"] = view
+        list = []
+        paginator = self.resource_explorer_connection.get_paginator("search")
+        page_iterator = paginator.paginate(**args)
+        for page in page_iterator:
+            resources = page.get("Resources")
+            for r in resources:
+                resource = ResourceExplorerResource(
+                    arn=r.get("Arn"),
+                    region=r.get("Region"),
+                    service=r.get("Service"),
+                    properties=r.get("Properties"),
+                    resource_type=r.get("ResourceType")
+                )
+                list.append(resource)
+        return list