Enrichment MCP Server

import requests

from string import Template
from urllib.parse import urljoin

import jinja2

from services.base import BaseService


class HybridAnalysis(BaseService):
    name: str = "hybridanalysis"
    host: str = "https://www.hybrid-analysis.com/"


class HybridAnalysisIP(HybridAnalysis):
    endpoint: Template = Template("search/terms")

    def get(self, ipaddress: str) -> requests.PreparedRequest:
        return requests.Request(
            method=self.method,
            url=urljoin(self.host, self.endpoint.substitute(ipaddress=ipaddress)),
            headers=self.headers,
        ).prepare()

    def parse_response(self, response: requests.Response, template: jinja2.Template) -> str:
        if response and response.ok and response.json():
            return template.render(
                name=self.name,
                **response.json()
            )
        else:
            return ""


class HybridAnalysisDomain(HybridAnalysis):
    endpoint: Template = Template("search/terms")

    def get(self, domain: str) -> requests.PreparedRequest:
        return requests.Request(
            method=self.method,
            url=urljoin(self.host, self.endpoint.substitute(domain=domain)),
            headers=self.headers,
        ).prepare()

    def parse_response(self, response: requests.Response, template: jinja2.Template) -> str:
        if response and response.ok and response.json():
            return template.render(
                name=self.name,
                **response.json()
            )
        else:
            return ""