update builder

Signed-off-by: ふぁ <yuki@yuki0311.com>

tools/hooks.py

@@ -0,0 +1,193 @@
+import yaml
+import json
+
+
+class HookBase:
+    PLACEHOLDER: dict
+
+    def __init__(self):
+        with open("src/config/placeholder.json", mode="r", encoding="utf-8") as f:
+            self.PLACEHOLDER = json.load(f)
+
+    def placeholder_to_yaml(self, obj, default=False, example=False) -> dict:
+        fn = lambda x: self.placeholder_to_yaml(x, default=default, example=example)
+        if type(obj) is dict:
+            properties = {i: fn(obj[i]) for i in obj}
+            value = {
+                "type": "object",
+                "required": [i for i in obj],
+                "properties": properties,
+            }
+            value.update({"default": properties} if default else {})
+            value.update({"example": properties} if example else {})
+            return value
+        elif type(obj) is list and len(obj) > 0:
+            properties = fn(obj[0])
+            value = {
+                "type": "array",
+                "items": properties,
+            }
+            value.update({"default": [properties]} if default else {})
+            value.update({"example": [properties]} if example else {})
+            return value
+        elif type(obj) is list and len(obj) == 0:
+            value = {
+                "type": "array",
+                "items": {"type": "object"},
+            }
+            value.update({"default": []} if default else {})
+            value.update({"example": []} if example else {})
+            return value
+        elif type(obj) is str:
+            return {"type": "string", "example": obj, "default": obj}
+        elif type(obj) is bool:
+            return {"type": "boolean", "example": obj, "default": obj}
+        elif type(obj) is int:
+            return {"type": "integer", "example": obj, "default": obj}
+
+    def load_component(self, name: str) -> dict:
+        with open(f"src/config/component/{name}.yaml", mode="r", encoding="utf-8") as f:
+            return yaml.safe_load(f)
+
+
+class OpenapiHookBase(HookBase):
+    def hook(self, value: dict):
+        return value
+
+
+class RequestHookBase(HookBase):
+    def hook(self, path: str, value: dict):
+        return path, value
+
+
+class AddSecuritySchemesOnSecuritySchemes(HookBase):
+    def hook(self, value: dict):
+        component = self.load_component("security_schemes")
+        param = component["components"]["securitySchemes"]
+        value["components"]["securitySchemes"].extend(param)
+        return value
+
+
+class AddSecuritySchemesOnHeader(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        value["parameters"] = value.get("parameters", [])
+        component = self.load_component("security_schemes")
+        param = component["paths"]["/parameters"]["get"]["parameters"]
+        value["parameters"].extend(param)
+        return path, value
+
+
+class ReplaceQueryIdPlaceholder(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        path_name = path.split("/")[-1]
+        new = self.PLACEHOLDER[path_name]["queryId"]
+        return path.replace(r"{{queryId}}", new), value
+
+
+class SetResponsesHeader(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        component = self.load_component("response_header")
+        value["responses"]["200"]["headers"] = component["components"]["headers"]
+        return path, value
+
+
+class AddParametersOnParametersAsString(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        value["parameters"] = value.get("parameters", [])
+        path_name = path.split("/")[-1]
+        for key in self.PLACEHOLDER[path_name].keys():
+            example = json.dumps(self.PLACEHOLDER[path_name][key])
+            value["parameters"].append(
+                {
+                    "name": key,
+                    "in": "query",
+                    "required": True,
+                    "schema": {
+                        "type": "string",
+                        "default": example,
+                        "example": example,
+                    },
+                }
+            )
+        return path, value
+
+
+class AddParametersOnParametersAsObject(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        value["parameters"] = value.get("parameters", [])
+        path_name = path.split("/")[-1]
+        for key in self.PLACEHOLDER[path_name].keys():
+            example = json.dumps(self.PLACEHOLDER[path_name][key])
+            value["parameters"].append(
+                {
+                    "name": key,
+                    "in": "query",
+                    "required": True,
+                    "schema": {
+                        "type": "object",
+                        "default": example,
+                        "example": example,
+                    },
+                }
+            )
+        return path, value
+
+
+class AddParametersOnContent(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        value["parameters"] = value.get("parameters", [])
+        path_name = path.split("/")[-1]
+        for key in self.PLACEHOLDER[path_name].keys():
+            value["parameters"].append(
+                {
+                    "name": key,
+                    "in": "query",
+                    "required": True,
+                    "content": {
+                        "application/json": {
+                            "schema": self.placeholder_to_yaml(
+                                self.PLACEHOLDER[path_name][key]
+                            ),
+                        },
+                    },
+                }
+            )
+        return path, value
+
+
+class AddParametersOnParameters(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        value["parameters"] = value.get("parameters", [])
+        path_name = path.split("/")[-1]
+        for key in self.PLACEHOLDER[path_name].keys():
+            value["parameters"].append(
+                {
+                    "name": key,
+                    "in": "query",
+                    "required": True,
+                    "schema": self.placeholder_to_yaml(
+                        self.PLACEHOLDER[path_name][key]
+                    ),
+                }
+            )
+        return path, value
+
+
+class AddParametersOnBody(RequestHookBase):
+    def hook(self, path: str, value: dict):
+        value["parameters"] = value.get("parameters", [])
+        path_name = path.split("/")[-1]
+        data = self.PLACEHOLDER[path_name]
+        schema = {i: self.placeholder_to_yaml(data[i]) for i in data.keys()}
+        value["requestBody"] = {
+            "description": "body",
+            "required": True,
+            "content": {
+                "application/json": {
+                    "schema": {
+                        "properties": schema,
+                    },
+                }
+            },
+        }
+        return path, value