.gitignore

@@ -1,5 +1,5 @@
 # claude
-.claude/
+CLAUDE.md
 
 # python
 __pycache__/

README.md

@@ -11,23 +11,21 @@ This reads codes from email; it does not generate them (that is `pyotp`'s job).
 `requirements.txt`:
 
 ```
-aiomail @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.5
+aiomail @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.2
 # OAuth token providers (Microsoft / Google) need the extra:
-aiomail[oauth] @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.5
+aiomail[oauth] @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.2
 ```
 
 Direct:
 
 ```bash
-pip install "aiomail @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.5"
+pip install "aiomail @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.2"
-pip install "aiomail[oauth] @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.5"
+pip install "aiomail[oauth] @ git+ssh://git@git.rethinkstudios.io/rethink-public/aiomail.git@v0.1.2"
 ```
 
 Requires `aioimaplib` and `beautifulsoup4` (pulled transitively). The `oauth`
 extra adds `aiohttp` for the refresh-token providers.
 
-Drop the `@v0.1.5` suffix from the line above to install the latest unpinned.
-
 ## Password auth
 
 ```python

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "aiomail"
-version = "0.1.5"
+version = "0.1.2"
 description = "async IMAP one-time-code retrieval with password/OAuth2 auth and dynamic matching"
 requires-python = ">=3.10"
 dependencies = [

src/aiomail/__init__.py

@@ -29,4 +29,4 @@ __all__ = [
     "DEFAULT_FOLDERS",
 ]
 
-__version__ = "0.1.5"
+__version__ = "0.1.2"

src/aiomail/auth.py

@@ -94,9 +94,6 @@ class OAuth2Auth:
         if xoauth2 is not None:
             result, data = await xoauth2(self.user, token)
         elif hasattr(mail, "authenticate"):
-            # escape hatch for a non-aioimaplib client: the shipped aioimaplib IMAP4
-            # always has .xoauth2 and never .authenticate, so this branch never runs
-            # for it; the SASL-callback signature here is untested against any driver
             result, data = await mail.authenticate(
                 "XOAUTH2", lambda _: _sasl_xoauth2(self.user, token)
             )

src/aiomail/client.py

@@ -9,7 +9,6 @@ import asyncio
 import email
 import email.message
 import logging
-import re
 from typing import List, Optional
 
 from aioimaplib import IMAP4, IMAP4_SSL
@@ -18,22 +17,6 @@ from .auth import Auth
 
 log = logging.getLogger(__name__)
 
-# IMAP LIST reply: (flags) "<delim>" <name>  — delim is server-defined (often "/" or
-# "." or NIL); capture the trailing name regardless, quoted or bare
-_LIST_RE = re.compile(rb'^\([^)]*\)\s+(?:"[^"]*"|NIL)\s+(.+)$')
-
-
-def _folder_name(raw: bytes) -> str:
-    """extract the folder name from a LIST reply line, delimiter-agnostic
-
-    parses the real reply form `(flags) "<delim>" <name>` so any server hierarchy
-    delimiter works (not just "/"); falls back to the last quoted/space token if the
-    line doesn't match the canonical shape.
-    """
-    match = _LIST_RE.match(raw.strip())
-    name = match.group(1).decode() if match else raw.decode().rsplit(" ", 1)[-1]
-    return name.strip().strip('"')
-
 
 class IMAPClient:
     """connection-managing IMAP client driven by an injected auth mechanism
@@ -85,33 +68,14 @@ class IMAPClient:
             except Exception as exc:
                 log.warning("connect attempt %d/%d failed: %s", attempt + 1, self.max_retries, exc)
                 if self._mail is not None:
-                    await self._discard_mail(self._mail)
+                    try:
+                        await self._mail.logout()
+                    except Exception as teardown:
+                        log.debug("logout error ignored during failed connect: %s", teardown)
                     self._mail = None
                 await asyncio.sleep(2 * (attempt + 1))
         return False
 
-    @staticmethod
-    async def _discard_mail(mail) -> None:
-        """tear down a half-built IMAP4 without leaking its connect task
-
-        aioimaplib's IMAP4 schedules `create_connection` as a fire-and-forget task it
-        never retrieves; on a refused connection that task raises and asyncio logs a
-        noisy "Task exception was never retrieved" traceback. cancel/await it here (and
-        retrieve its exception) before discarding, so a failed connect stays quiet.
-        """
-        task = getattr(mail, "_client_task", None)
-        if task is not None and not task.done():
-            task.cancel()
-        if task is not None:
-            try:
-                await task
-            except (asyncio.CancelledError, Exception):
-                pass
-        try:
-            await mail.logout()
-        except Exception as teardown:
-            log.debug("logout error ignored during failed connect: %s", teardown)
-
     async def close(self) -> None:
         """log out and drop the connection, swallowing teardown errors"""
         if self._mail is not None:
@@ -131,6 +95,14 @@ class IMAPClient:
         except Exception:
             return await self.connect()
 
+    def is_throttled(self) -> bool:
+        """best-effort detection of a provider throttling response"""
+        return bool(
+            self._mail is not None
+            and getattr(self._mail, "resp", None)
+            and "THROTTLED" in str(self._mail.resp)
+        )
+
     async def get_folders(self) -> List[str]:
         """list mailbox folder names"""
         if not await self.ensure_connection():
@@ -143,7 +115,7 @@ class IMAPClient:
         folders: List[str] = []
         for folder in folder_list or []:
             try:
-                folders.append(_folder_name(folder))
+                folders.append(folder.decode().split(' "/" ')[-1].strip('"'))
             except Exception:
                 continue
         return folders
@@ -197,14 +169,8 @@ class IMAPClient:
         if result != "OK" or not data:
             return None
         for item in data:
-            # aioimaplib stores the literal message payload as the only bytearray in
+            if isinstance(item, (bytes, bytearray)) and len(item) > 20:
-            # the response; every other line (including the `<id> FETCH (...` header)
-            # is plain bytes. select by structure, not length — a length heuristic
-            # mismatches the header line for any 2+ digit id or a BODY[]/UID fetch.
-            if isinstance(item, bytearray):
                 return email.message_from_bytes(bytes(item))
-            # cross-version fallback: aioimaplib 2.0.x never yields tuples here, but an
-            # imaplib-style (header, payload) tuple is handled if a future/alt driver does
             if isinstance(item, tuple) and len(item) > 1:
                 return email.message_from_bytes(item[1])
         return None

src/aiomail/extract.py

@@ -76,7 +76,7 @@ def _scan(text: str, patterns: list[Pattern], lengths: set[int]) -> Optional[str
             return m.group(1) if m.groups() else m.group(0)
     for token in re.split(r"\s+", text):
         digits = "".join(c for c in token if c.isdigit())
-        if digits and len(digits) in lengths:
+        if digits and len(digits) in lengths and digits.isdigit():
             return digits
     return None
 
@@ -121,8 +121,6 @@ def as_predicate(spec: MatchSpec) -> Callable[[Optional[str]], bool]:
     if isinstance(spec, re.Pattern):
         return lambda value: bool(spec.search(value or ""))
     if callable(spec):
-        # coalesce None like the string/regex branches so the documented Optional[str]
+        return spec
-        # predicate contract holds even if a caller's callable assumes a real string
-        return lambda value: bool(spec(value or ""))
     needle = str(spec).lower()
     return lambda value: needle in (value or "").lower()

src/aiomail/oauth.py

@@ -76,17 +76,12 @@ class _RefreshTokenProvider:
                     async with aiohttp.ClientSession(timeout=timeout) as session:
                         async with session.post(endpoint, data=data) as resp:
                             if resp.status == 200:
-                                # content_type=None: some token endpoints return a 200 with
+                                token = (await resp.json()).get("access_token")
-                                # text/plain or text/javascript; default json() would raise
-                                # ContentTypeError and discard a valid token body
-                                token = (await resp.json(content_type=None)).get("access_token")
                                 if token:
                                     self._failures = 0
                                     return token
                             else:
-                                # log a truncated error body only — a token-endpoint
+                                body = await resp.text()
-                                # response can carry sensitive material; never dump it whole
-                                body = (await resp.text())[:200]
                                 log.warning("token endpoint %s -> %s: %s", endpoint, resp.status, body)
                 except Exception as exc:
                     log.warning("token request to %s failed: %s", endpoint, exc)