from ingest.extended_live_adapter import ExtendedLiveAdapter


class AATIPExtendedAdapter(ExtendedLiveAdapter):
    source_name = "AATIP (Public Releases)"
    base_url = "https://www.esd.whs.mil/FOIA/Reading-Room/"

    async def search(self, query: str):
        return await self._safe_search(self.base_url, query)


class SAPExtendedAdapter(AATIPExtendedAdapter):
    source_name = "Special Access Programs (Public)"


class TENCAPExtendedAdapter(AATIPExtendedAdapter):
    source_name = "TEN-CAP (Public)"


class SpecialActivitiesExtendedAdapter(AATIPExtendedAdapter):
    source_name = "Special Activities (Public)"