Enrichment MCP Server

import requests

from string import Template
from urllib.parse import urljoin

import jinja2

from services.base import BaseService


class Shodan(BaseService):
    name: str = "shodan"
    host: str = "https://api.shodan.io/"


class ShodanIP(Shodan):
    endpoint: Template = Template("shodan/host/$ipaddress?key=")

    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 ShodanDomain(Shodan):
    endpoint: Template = Template("dns/domain/$domain?key=")

    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 ""