from typing import Dict

from .base import BaseParser


class OpenGraphParser(BaseParser):
    allowed_og_properties = {
        "og:title",
        "og:description",
        "og:image",
    }

    def extract_data(self) -> Dict[str, str]:
        meta = self._soup.findAll("meta")
        result = {}
        for tag in meta:
            if not tag.has_attr("property"):
                continue
            if tag["property"] not in self.allowed_og_properties:
                continue

            og_property_name = tag["property"][len("og:") :]
            if not tag.has_attr("content"):
                continue

            result[og_property_name] = tag["content"]

        return result