feat(codebuild): codebuild service and checks (#1467)

providers/aws/services/codebuild/codebuild_service.py

@@ -0,0 +1,85 @@
+import datetime
+import threading
+from dataclasses import dataclass
+
+from lib.logger import logger
+from providers.aws.aws_provider import generate_regional_clients
+
+
+################### Codebuild
+class Codebuild:
+    def __init__(self, audit_info):
+        self.service = "codebuild"
+        self.session = audit_info.audit_session
+        self.audited_account = audit_info.audited_account
+        self.regional_clients = generate_regional_clients(self.service, audit_info)
+        self.projects = []
+        self.__threading_call__(self.__list_projects__)
+        self.__list_builds_for_project__()
+
+    def __get_session__(self):
+        return self.session
+
+    def __threading_call__(self, call):
+        threads = []
+        for regional_client in self.regional_clients.values():
+            threads.append(threading.Thread(target=call, args=(regional_client,)))
+        for t in threads:
+            t.start()
+        for t in threads:
+            t.join()
+
+    def __list_projects__(self, regional_client):
+        logger.info("Codebuild - listing projects")
+        try:
+            list_projects_paginator = regional_client.get_paginator("list_projects")
+            for page in list_projects_paginator.paginate():
+                for project in page["projects"]:
+                    self.projects.append(
+                        CodebuildProject(
+                            name=project,
+                            region=regional_client.region,
+                            last_invoked_time=None,
+                            buildspec=None,
+                        )
+                    )
+
+        except Exception as error:
+            logger.error(
+                f"{regional_client.region} -- {error.__class__.__name__}: {error}"
+            )
+
+    def __list_builds_for_project__(self):
+        logger.info("Codebuild - listing builds from projects")
+        try:
+            for project in self.projects:
+                for region, client in self.regional_clients.items():
+                    if project.region == region:
+                        ids = client.list_builds_for_project(projectName=project.name)
+                        if "ids" in ids:
+                            builds = client.batch_get_builds(ids=[ids["ids"][0]])
+                            if "builds" in builds:
+                                project.last_invoked_time = builds["builds"][0][
+                                    "endTime"
+                                ]
+
+                        project.buildspec = client.batch_get_projects(
+                            names=[project.name]
+                        )["projects"][0]["source"]["buildspec"]
+
+        except Exception as error:
+            logger.error(f"{client.region} -- {error.__class__.__name__}: {error}")
+
+
+@dataclass
+class CodebuildProject:
+    name: str
+    region: str
+    last_invoked_time: datetime
+    buildspec: str
+
+    def __init__(self, name, region, last_invoked_time, buildspec):
+        self.name = name
+        self.region = region
+        self.last_invoked_time = last_invoked_time
+        self.buildspec = buildspec