106 lines
3.1 KiB
Python
106 lines
3.1 KiB
Python
from typing import Iterator, Dict, Any
|
|
|
|
from googleapiclient.discovery import build
|
|
from googleapiclient.errors import HttpError
|
|
|
|
from mail_intake.adapters.base import MailIntakeAdapter
|
|
from mail_intake.exceptions import MailIntakeAdapterError
|
|
from mail_intake.auth.base import MailIntakeAuthProvider
|
|
|
|
|
|
class MailIntakeGmailAdapter(MailIntakeAdapter):
|
|
"""
|
|
Gmail read-only adapter.
|
|
|
|
This class is the ONLY place where:
|
|
- googleapiclient is imported
|
|
- Gmail REST semantics are known
|
|
- .execute() is called
|
|
|
|
It must remain thin and dumb by design.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
auth_provider: MailIntakeAuthProvider,
|
|
user_id: str = "me",
|
|
):
|
|
self._auth_provider = auth_provider
|
|
self._user_id = user_id
|
|
self._service = None
|
|
|
|
@property
|
|
def service(self):
|
|
if self._service is None:
|
|
try:
|
|
creds = self._auth_provider.get_credentials()
|
|
self._service = build("gmail", "v1", credentials=creds)
|
|
except Exception as exc:
|
|
raise MailIntakeAdapterError(
|
|
"Failed to initialize Gmail service"
|
|
) from exc
|
|
return self._service
|
|
|
|
def iter_message_refs(self, query: str) -> Iterator[Dict[str, str]]:
|
|
"""
|
|
Iterate over message references matching the query.
|
|
|
|
Yields:
|
|
{
|
|
"message_id": "...",
|
|
"thread_id": "..."
|
|
}
|
|
"""
|
|
try:
|
|
request = (
|
|
self.service.users()
|
|
.messages()
|
|
.list(userId=self._user_id, q=query)
|
|
)
|
|
|
|
while request is not None:
|
|
response = request.execute()
|
|
|
|
for msg in response.get("messages", []):
|
|
yield {
|
|
"message_id": msg["id"],
|
|
"thread_id": msg["threadId"],
|
|
}
|
|
|
|
request = (
|
|
self.service.users()
|
|
.messages()
|
|
.list_next(request, response)
|
|
)
|
|
|
|
except HttpError as exc:
|
|
raise MailIntakeAdapterError(
|
|
"Gmail API error while listing messages"
|
|
) from exc
|
|
|
|
def fetch_message(self, message_id: str) -> Dict[str, Any]:
|
|
try:
|
|
return (
|
|
self.service.users()
|
|
.messages()
|
|
.get(userId=self._user_id, id=message_id)
|
|
.execute()
|
|
)
|
|
except HttpError as exc:
|
|
raise MailIntakeAdapterError(
|
|
f"Gmail API error while fetching message {message_id}"
|
|
) from exc
|
|
|
|
def fetch_thread(self, thread_id: str) -> Dict[str, Any]:
|
|
try:
|
|
return (
|
|
self.service.users()
|
|
.threads()
|
|
.get(userId=self._user_id, id=thread_id)
|
|
.execute()
|
|
)
|
|
except HttpError as exc:
|
|
raise MailIntakeAdapterError(
|
|
f"Gmail API error while fetching thread {thread_id}"
|
|
) from exc
|