init

Author: Ilanlido

consts.py

@@ -0,0 +1,10 @@
+config_file = "config.json"
+github = "Github"
+providers = [github]
+contexts = ["Cycode: Secrets",
+            "Cycode: IaC",
+            "Cycode: Leaks",
+            "Cycode: CI/CD",
+            "Cycode: SAST",
+            "Cycode: Vulnerable Dependencies",
+            "Cycode: Licenses Compliance", ]

main.py

@@ -0,0 +1,54 @@
+from typing import Dict
+
+from PyInquirer import prompt, Separator
+import logging
+from consts import providers, contexts, config_file
+from menus.menu_base import MenuBase
+from menus.release_block_pr_menu import ReleaseBlockPrMenu
+
+release_block_pr_menu = ReleaseBlockPrMenu()
+menus = [release_block_pr_menu]
+menu_value_to_class: Dict[str, MenuBase] = {menu.get_menu(): menu for menu in menus}
+
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+main_menu = [
+    {
+        'type': 'list',
+        'name': 'menu',
+        'message': 'What do you want to do?',
+        'choices': [menu for menu in menu_value_to_class.keys()],
+
+    },
+    {
+        'type': 'input',
+        'name': 'config_file',
+        "message": "Please provide a configuration file",
+        "default": config_file,
+        'validate': release_block_pr_menu.validate_config_file,
+        "when": release_block_pr_menu.menu_chosen
+    },
+    {
+        'type': 'list',
+        'name': 'provider',
+        "message": "Which provider?",
+        "choices": providers,
+        "when": release_block_pr_menu.menu_chosen,
+    },
+    {
+        'type': 'checkbox',
+        'name': 'contexts',
+        "message": "Which contexts?",
+        "choices": [{"name": context} for context in contexts],
+        "when": release_block_pr_menu.menu_chosen
+    }
+]
+
+if __name__ == '__main__':
+    answer = prompt(main_menu)
+    menu = menu_value_to_class.get(answer.get('menu'))
+    if menu:
+        menu.handle(answer)
+    else:
+        logger.error(f"invalid menu {answer.get('menu')}")

menus/__init__.py

@@ -0,0 +1,14 @@
+from abc import abstractmethod, ABC
+
+
+class MenuBase(ABC):
+    def menu_chosen(self, answers):
+        return answers.get('menu') == self.get_menu()
+
+    @abstractmethod
+    def get_menu(self):
+        pass
+
+    @abstractmethod
+    def handle(self, answers):
+        pass

menus/menu_base.py

@@ -0,0 +1,46 @@
+import os
+from typing import List, Dict
+import pydantic
+import logging
+from consts import github
+from menus.menu_base import MenuBase
+from providers.github_handler import GithubHandler
+from providers.provider_handler import ProviderHandler
+import PyInquirer
+from release_block_pr_config import ReleaseBlockPrConfig
+
+logger = logging.getLogger(__name__)
+
+release_block = "Release Blocked PR's"
+
+
+class ReleaseBlockPrMenu(MenuBase):
+    handlers: Dict[str, ProviderHandler] = {
+        github: GithubHandler()
+    }
+
+    def handle(self, answers):
+        if answers.get("provider") in self.handlers:
+            handler = self.handlers[answers["provider"]]
+            repositories_to_unblock: List[ReleaseBlockPrConfig] = pydantic.parse_file_as(List[ReleaseBlockPrConfig],
+                                                                                         answers["config_file"])
+            for tokens_to_repositories in repositories_to_unblock:
+                for repository in tokens_to_repositories.repositories:
+                    handler.release_branch_protection(tokens_to_repositories.token, repository.organization_name,
+                                                      repository.organization_name, repository.branch,
+                                                      answers["contexts"])
+        else:
+            logger.error(f"Invalid provider selected: {answers.get('provider')}")
+
+    def get_menu(self):
+        return release_block
+
+    @staticmethod
+    def validate_config_file(config_file):
+        if not os.path.isfile(config_file):
+            return "File does not exist"
+        try:
+            pydantic.parse_file_as(List[ReleaseBlockPrConfig], config_file)
+        except Exception as e:
+            return f"Invalid input, not in the expected format {e}"
+        return True

menus/release_block_pr_menu.py

@@ -0,0 +1,11 @@
+import logging
+from typing import List
+
+from providers.provider_handler import ProviderHandler
+
+logger = logging.getLogger(__name__)
+
+
+class GithubHandler(ProviderHandler):
+    def release_branch_protection(self, token, organization_name, repository_name, branch, contexts: List[str]):
+        logger.info(f"Releasing branch protection on repository {repository_name}")

providers/__init__.py

@@ -0,0 +1,8 @@
+from abc import ABC, abstractmethod
+from typing import List
+
+
+class ProviderHandler(ABC):
+    @abstractmethod
+    def release_branch_protection(self, token, organization_name, repository_name, branch, contexts: List[str]):
+        pass

providers/github_handler.py

@@ -0,0 +1,15 @@
+from typing import List
+
+import pydantic
+
+
+class RepositoryConfig(pydantic.BaseModel):
+    repository_name: str
+    organization_name: str
+    branch: str
+
+
+class ReleaseBlockPrConfig(pydantic.BaseModel):
+    token: str
+    provider: str
+    repositories: List[RepositoryConfig]