mochi-analytics 1.0.0__tar.gz

mochi_analytics-1.0.0/.gitignore

@@ -0,0 +1,10 @@
+__pycache__/
+*.py[cod]
+*.egg-info/
+build/
+dist/
+.venv/
+venv/
+.pytest_cache/
+.ruff_cache/
+.mypy_cache/

mochi_analytics-1.0.0/CHANGELOG.md

@@ -0,0 +1,8 @@
+# Changelog
+
+## 1.0.0 (2026-07-09)
+
+
+### Features
+
+* initial public release of mochi-analytics and the discord.py adapter ([d0b2eaa](https://github.com/mochi-analytics/mochi-py/commit/d0b2eaa1a71da5acfbf1afcc79c9bd5ae51f2b24))

mochi_analytics-1.0.0/LICENSE

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

mochi_analytics-1.0.0/PKG-INFO

@@ -0,0 +1,74 @@
+Metadata-Version: 2.4
+Name: mochi-analytics
+Version: 1.0.0
+Summary: Core client for Mochi - self-hosted analytics for Discord bots
+Project-URL: Homepage, https://github.com/mochi-analytics/mochi-py
+Project-URL: Repository, https://github.com/mochi-analytics/mochi-py
+Project-URL: Bug Tracker, https://github.com/mochi-analytics/mochi-py/issues
+Author-email: Mochi Analytics <contact@bwmp.dev>
+License-Expression: Apache-2.0
+License-File: LICENSE
+Keywords: analytics,bot,discord,mochi
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Requires-Dist: aiohttp>=3.9
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest>=8; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# mochi-analytics
+
+Core client for Mochi — self-hosted analytics for Discord bots. Asyncio-native,
+batching, and non-blocking: `track()` never blocks or raises into your bot.
+
+## Install
+
+```sh
+pip install mochi-analytics
+```
+
+## Usage
+
+```python
+from mochi_analytics import MochiClient, MochiEvent, MochiSnapshot
+
+mochi = MochiClient(url="https://mochi.example.com", api_key="mochi_sk_...")
+
+# Queue events — sent in the background, batched.
+mochi.track(MochiEvent(type="command", name="play", user_id="123"))
+mochi.track_command("play", guild_id="456")
+
+# Send a health snapshot immediately.
+await mochi.snapshot(MochiSnapshot(guild_count=1204, ws_ping_ms=38))
+
+# On shutdown, flush anything still queued.
+await mochi.shutdown()
+```
+
+## Design guarantees
+
+- Events are batched (flushed every 5 s or 100 events) and sent in the
+  background — `track()` never blocks or raises.
+- Transient failures (429 / 5xx / network) retry with exponential backoff;
+  the queue is bounded (oldest dropped first), so a dead Mochi instance can
+  never leak memory or crash the bot.
+- Errors are routed to the optional `on_error` callback, never raised.
+
+## Options
+
+| Argument | Default | Notes |
+|---|---|---|
+| `flush_interval` | `5.0` | Seconds between background flushes. |
+| `max_batch_size` | `100` | Events per request (server limit is 100). |
+| `max_queue_size` | `10000` | Events beyond this are dropped oldest-first. |
+| `max_retries` | `3` | Retry attempts for retryable failures. |
+| `on_error` | no-op | `Callable[[Exception], None]` for dropped/failed sends. |
+| `transport` | aiohttp | Injectable `async (url, body) -> (status, text)` for testing. |
+
+## License
+
+Apache-2.0

mochi_analytics-1.0.0/README.md

@@ -0,0 +1,52 @@
+# mochi-analytics
+
+Core client for Mochi — self-hosted analytics for Discord bots. Asyncio-native,
+batching, and non-blocking: `track()` never blocks or raises into your bot.
+
+## Install
+
+```sh
+pip install mochi-analytics
+```
+
+## Usage
+
+```python
+from mochi_analytics import MochiClient, MochiEvent, MochiSnapshot
+
+mochi = MochiClient(url="https://mochi.example.com", api_key="mochi_sk_...")
+
+# Queue events — sent in the background, batched.
+mochi.track(MochiEvent(type="command", name="play", user_id="123"))
+mochi.track_command("play", guild_id="456")
+
+# Send a health snapshot immediately.
+await mochi.snapshot(MochiSnapshot(guild_count=1204, ws_ping_ms=38))
+
+# On shutdown, flush anything still queued.
+await mochi.shutdown()
+```
+
+## Design guarantees
+
+- Events are batched (flushed every 5 s or 100 events) and sent in the
+  background — `track()` never blocks or raises.
+- Transient failures (429 / 5xx / network) retry with exponential backoff;
+  the queue is bounded (oldest dropped first), so a dead Mochi instance can
+  never leak memory or crash the bot.
+- Errors are routed to the optional `on_error` callback, never raised.
+
+## Options
+
+| Argument | Default | Notes |
+|---|---|---|
+| `flush_interval` | `5.0` | Seconds between background flushes. |
+| `max_batch_size` | `100` | Events per request (server limit is 100). |
+| `max_queue_size` | `10000` | Events beyond this are dropped oldest-first. |
+| `max_retries` | `3` | Retry attempts for retryable failures. |
+| `on_error` | no-op | `Callable[[Exception], None]` for dropped/failed sends. |
+| `transport` | aiohttp | Injectable `async (url, body) -> (status, text)` for testing. |
+
+## License
+
+Apache-2.0

mochi_analytics-1.0.0/pyproject.toml

@@ -0,0 +1,35 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "mochi-analytics"
+version = "1.0.0"
+description = "Core client for Mochi - self-hosted analytics for Discord bots"
+readme = "README.md"
+license = "Apache-2.0"
+requires-python = ">=3.9"
+keywords = ["discord", "bot", "analytics", "mochi"]
+authors = [{ name = "Mochi Analytics", email = "contact@bwmp.dev" }]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: Apache Software License",
+    "Framework :: AsyncIO",
+    "Intended Audience :: Developers",
+]
+dependencies = ["aiohttp>=3.9"]
+
+[project.urls]
+Homepage = "https://github.com/mochi-analytics/mochi-py"
+Repository = "https://github.com/mochi-analytics/mochi-py"
+"Bug Tracker" = "https://github.com/mochi-analytics/mochi-py/issues"
+
+[project.optional-dependencies]
+dev = ["pytest>=8", "pytest-asyncio>=0.23"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/mochi_analytics"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]

mochi_analytics-1.0.0/src/mochi_analytics/__init__.py

@@ -0,0 +1,23 @@
+"""Core client for Mochi - self-hosted analytics for Discord bots."""
+
+from mochi_analytics.client import (
+    MochiChannelType,
+    MochiClient,
+    MochiError,
+    MochiEvent,
+    MochiEventType,
+    MochiSnapshot,
+    Transport,
+)
+
+__all__ = [
+    "MochiChannelType",
+    "MochiClient",
+    "MochiError",
+    "MochiEvent",
+    "MochiEventType",
+    "MochiSnapshot",
+    "Transport",
+]
+
+__version__ = "1.0.0"  # x-release-please-version

mochi_analytics-1.0.0/src/mochi_analytics/client.py

@@ -0,0 +1,327 @@
+"""Batching, non-blocking analytics client for Mochi.
+
+Failures never propagate into the caller: analytics must not be able to crash
+a bot. Mirrors the design of ``@mochi-analytics/core`` but is asyncio-native.
+"""
+
+from __future__ import annotations
+
+import asyncio
+from dataclasses import dataclass, field
+from datetime import datetime, timezone
+from typing import Any, Awaitable, Callable, Optional
+
+try:  # aiohttp is optional at import time so tests can inject a transport.
+    import aiohttp
+except ImportError:  # pragma: no cover - exercised only without aiohttp
+    aiohttp = None  # type: ignore[assignment]
+
+MochiEventType = str  # "command" | "guild_join" | "guild_leave" | "error" | "custom"
+MochiChannelType = str  # "guild_text" | "guild_voice" | "thread" | "dm" | "group_dm" | "other"
+
+RETRYABLE_STATUS = frozenset({429, 500, 502, 503, 504})
+
+#: A transport sends one request and returns ``(status_code, body_text)`` or
+#: ``(status_code, body_text, headers)``. Headers are optional and only used to
+#: read ``Retry-After`` on a 429. The default uses aiohttp; tests inject a fake.
+Transport = Callable[[str, Any], Awaitable[Any]]
+
+
+class MochiError(Exception):
+    """Raised internally on a permanent request failure; routed to ``on_error``."""
+
+
+def _iso_now() -> str:
+    return datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+
+
+def _header(headers: Any, name: str) -> Optional[str]:
+    """Case-insensitive header lookup tolerant of plain dicts and CIMultiDicts."""
+    if headers is None:
+        return None
+    getter = getattr(headers, "get", None)
+    if getter is None:
+        return None
+    value = getter(name)
+    if value is None:
+        value = getter(name.lower())
+    return value
+
+
+def _parse_retry_after(value: Optional[str]) -> Optional[float]:
+    """Parse a Retry-After header (delta-seconds or HTTP-date) into seconds."""
+    if not value:
+        return None
+    value = value.strip()
+    try:
+        seconds = float(value)
+        return seconds if seconds >= 0 else None
+    except ValueError:
+        pass
+    from email.utils import parsedate_to_datetime
+
+    try:
+        when = parsedate_to_datetime(value)
+    except (TypeError, ValueError):
+        return None
+    if when is None:
+        return None
+    if when.tzinfo is None:
+        when = when.replace(tzinfo=timezone.utc)
+    return max((when - datetime.now(timezone.utc)).total_seconds(), 0.0)
+
+
+@dataclass
+class MochiEvent:
+    """A single analytics event.
+
+    Attributes use idiomatic snake_case; they are serialized to the camelCase
+    wire format by :meth:`to_wire`.
+    """
+
+    type: MochiEventType
+    name: Optional[str] = None
+    guild_id: Optional[str] = None
+    #: Raw Discord user id - hashed server-side with a per-bot salt, never stored.
+    user_id: Optional[str] = None
+    channel_type: Optional[MochiChannelType] = None
+    shard_id: Optional[int] = None
+    success: Optional[bool] = None
+    duration_ms: Optional[int] = None
+    meta: Optional[dict[str, Any]] = None
+    #: ISO 8601 timestamp; defaults to send time when omitted.
+    ts: Optional[str] = None
+
+    def to_wire(self) -> dict[str, Any]:
+        wire: dict[str, Any] = {"type": self.type}
+        if self.name is not None:
+            wire["name"] = self.name
+        if self.guild_id is not None:
+            wire["guildId"] = self.guild_id
+        if self.user_id is not None:
+            wire["userId"] = self.user_id
+        if self.channel_type is not None:
+            wire["channelType"] = self.channel_type
+        if self.shard_id is not None:
+            wire["shardId"] = self.shard_id
+        if self.success is not None:
+            wire["success"] = self.success
+        if self.duration_ms is not None:
+            wire["durationMs"] = self.duration_ms
+        if self.meta is not None:
+            wire["meta"] = self.meta
+        wire["ts"] = self.ts or _iso_now()
+        return wire
+
+
+@dataclass
+class MochiSnapshot:
+    """A guild-count / health sample."""
+
+    guild_count: int
+    shard_id: Optional[int] = None
+    total_shards: Optional[int] = None
+    approximate_member_sum: Optional[int] = None
+    ws_ping_ms: Optional[int] = None
+    ts: Optional[str] = None
+
+    def to_wire(self) -> dict[str, Any]:
+        wire: dict[str, Any] = {"guildCount": self.guild_count}
+        if self.shard_id is not None:
+            wire["shardId"] = self.shard_id
+        if self.total_shards is not None:
+            wire["totalShards"] = self.total_shards
+        if self.approximate_member_sum is not None:
+            wire["approximateMemberSum"] = self.approximate_member_sum
+        if self.ws_ping_ms is not None:
+            wire["wsPingMs"] = self.ws_ping_ms
+        if self.ts is not None:
+            wire["ts"] = self.ts
+        return wire
+
+
+class MochiClient:
+    """Batching, non-blocking analytics client.
+
+    ``track`` is synchronous and returns immediately - events are queued and
+    flushed by a background asyncio task (every ``flush_interval`` seconds or
+    whenever the batch fills). Transient failures retry with exponential
+    backoff; the queue is bounded so a dead Mochi instance can never leak
+    memory or crash the bot.
+
+    The background flush loop starts lazily the first time ``track`` /
+    ``snapshot`` / ``flush`` is called while an event loop is running, so a
+    client can be constructed outside of a running loop.
+    """
+
+    def __init__(
+        self,
+        url: str,
+        api_key: str,
+        *,
+        flush_interval: float = 5.0,
+        max_batch_size: int = 100,
+        max_queue_size: int = 10_000,
+        max_retries: int = 3,
+        on_error: Optional[Callable[[Exception], None]] = None,
+        transport: Optional[Transport] = None,
+    ) -> None:
+        base = url.rstrip("/")
+        self._ingest_url = f"{base}/api/v1/ingest"
+        self._snapshot_url = f"{base}/api/v1/snapshot"
+        self._api_key = api_key
+        self._flush_interval = flush_interval
+        self._max_batch_size = min(max_batch_size, 100)
+        self._max_queue_size = max_queue_size
+        self._max_retries = max_retries
+        self._on_error = on_error or (lambda _err: None)
+        self._transport = transport or self._default_transport
+
+        self._queue: list[dict[str, Any]] = []
+        self._flush_loop_task: Optional[asyncio.Task[None]] = None
+        self._flushing: Optional[asyncio.Task[None]] = None
+        self._shutdown = False
+        self._session: Optional[Any] = None
+
+    # -- public API -----------------------------------------------------
+
+    def track(self, event: MochiEvent) -> None:
+        """Queue an event. Returns immediately; sending happens in the background."""
+        if self._shutdown:
+            return
+        self._queue.append(event.to_wire())
+        overflow = len(self._queue) - self._max_queue_size
+        if overflow > 0:
+            del self._queue[:overflow]
+            self._report(MochiError("mochi: event queue overflow, dropped oldest"))
+        self._ensure_loop()
+        if len(self._queue) >= self._max_batch_size:
+            self._schedule_flush()
+
+    def track_command(self, name: str, **context: Any) -> None:
+        """Convenience wrapper for a ``command`` event."""
+        self.track(MochiEvent(type="command", name=name, **context))
+
+    async def snapshot(self, snapshot: MochiSnapshot) -> None:
+        """Send a guild-count / health snapshot immediately (with retries)."""
+        try:
+            await self._send(self._snapshot_url, snapshot.to_wire())
+        except Exception as error:  # never propagate into the caller
+            self._report(error)
+
+    async def flush(self) -> None:
+        """Drain the queue. Safe to call concurrently; flushes are serialized."""
+        if self._flushing and not self._flushing.done():
+            await self._flushing
+            return
+        self._flushing = asyncio.ensure_future(self._drain())
+        try:
+            await self._flushing
+        finally:
+            self._flushing = None
+
+    async def shutdown(self) -> None:
+        """Stop the background loop and flush remaining events. Call on exit."""
+        self._shutdown = True
+        if self._flush_loop_task:
+            self._flush_loop_task.cancel()
+            try:
+                await self._flush_loop_task
+            except asyncio.CancelledError:
+                pass
+            self._flush_loop_task = None
+        await self.flush()
+        if self._session is not None:
+            await self._session.close()
+            self._session = None
+
+    # -- internals ------------------------------------------------------
+
+    def _ensure_loop(self) -> None:
+        if self._flush_loop_task is not None or self._shutdown:
+            return
+        try:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            return  # no loop yet; loop starts on the next call inside one
+        self._flush_loop_task = loop.create_task(self._flush_loop())
+
+    async def _flush_loop(self) -> None:
+        while not self._shutdown:
+            await asyncio.sleep(self._flush_interval)
+            await self.flush()
+
+    def _schedule_flush(self) -> None:
+        if self._flushing and not self._flushing.done():
+            return
+        try:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            return
+        self._flushing = loop.create_task(self._drain())
+
+    async def _drain(self) -> None:
+        while self._queue:
+            batch = self._queue[: self._max_batch_size]
+            del self._queue[: self._max_batch_size]
+            try:
+                await self._send(self._ingest_url, {"events": batch})
+            except Exception as error:
+                # Batch is dropped; don't spin on a failing endpoint.
+                self._report(error)
+                return
+
+    async def _send(self, url: str, body: Any) -> None:
+        last_error: Optional[Exception] = None
+        retry_after: Optional[float] = None  # seconds, from a prior 429
+        for attempt in range(self._max_retries + 1):
+            if attempt > 0:
+                # Honor a server-provided Retry-After in preference to the
+                # computed backoff; otherwise use exponential backoff.
+                delay = retry_after if retry_after is not None else 0.5 * 2 ** (attempt - 1)
+                retry_after = None
+                await asyncio.sleep(delay)
+            try:
+                # Transport returns (status, text) or (status, text, headers).
+                status, text, *rest = await self._transport(url, body)
+            except Exception as error:  # network error -> retry
+                last_error = error
+                continue
+            if 200 <= status < 300:
+                return
+            if status not in RETRYABLE_STATUS:
+                raise MochiError(f"mochi: request rejected ({status}) {text}")
+            if status == 429 and rest:
+                retry_after = _parse_retry_after(_header(rest[0], "Retry-After"))
+            last_error = MochiError(f"mochi: server returned {status}")
+        raise last_error or MochiError("mochi: request failed")
+
+    async def _default_transport(self, url: str, body: Any) -> "tuple[int, str]":
+        if aiohttp is None:  # pragma: no cover
+            raise MochiError(
+                "mochi: aiohttp is not installed; install mochi-analytics with its "
+                "default dependencies or pass a custom transport"
+            )
+        session = await self._get_session()
+        async with session.post(
+            url,
+            json=body,
+            headers={
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {self._api_key}",
+            },
+        ) as response:
+            text = await response.text()
+            # response.headers is a case-insensitive CIMultiDict.
+            return response.status, text, response.headers
+
+    async def _get_session(self) -> Any:
+        if self._session is None:
+            self._session = aiohttp.ClientSession()
+        return self._session
+
+    def _report(self, error: Exception) -> None:
+        try:
+            self._on_error(error)
+        except Exception:  # an error handler must never take down the bot
+            pass

mochi_analytics-1.0.0/tests/test_client.py

@@ -0,0 +1,180 @@
+"""Mirrors the @mochi-analytics/core vitest suite."""
+
+from __future__ import annotations
+
+import time
+from typing import Any, Callable
+
+import pytest
+
+from mochi_analytics import MochiClient, MochiEvent, MochiSnapshot
+
+
+class Call:
+    def __init__(self, url: str, body: Any) -> None:
+        self.url = url
+        self.body = body
+
+
+def make_transport(responder: Callable[[Call, int], int]):
+    """Build a fake transport that records calls and returns a status per call."""
+    calls: list[Call] = []
+
+    async def transport(url: str, body: Any) -> "tuple[int, str]":
+        call = Call(url, body)
+        calls.append(call)
+        status = responder(call, len(calls) - 1)
+        return status, "{}"
+
+    return calls, transport
+
+
+def make_client(transport, **extra):
+    errors: list[Exception] = []
+    client = MochiClient(
+        "http://localhost:9999/",
+        "mochi_sk_test",
+        flush_interval=60.0,  # effectively disabled; tests flush manually
+        max_retries=2,
+        on_error=errors.append,
+        transport=transport,
+        **extra,
+    )
+    return client, errors
+
+
+async def test_batches_events_into_one_request_with_auth():
+    calls, transport = make_transport(lambda *_: 202)
+    client, _ = make_client(transport)
+
+    client.track(MochiEvent(type="command", name="play", user_id="1"))
+    client.track(MochiEvent(type="guild_join", guild_id="2"))
+    await client.flush()
+
+    assert len(calls) == 1
+    assert calls[0].url == "http://localhost:9999/api/v1/ingest"
+    assert len(calls[0].body["events"]) == 2
+    assert calls[0].body["events"][0]["ts"]
+    assert calls[0].body["events"][0]["userId"] == "1"
+    await client.shutdown()
+
+
+async def test_auto_flushes_when_batch_size_reached():
+    calls, transport = make_transport(lambda *_: 202)
+    client, _ = make_client(transport, max_batch_size=5)
+
+    for _ in range(5):
+        client.track(MochiEvent(type="command", name="x"))
+    await client.flush()
+
+    total = sum(len(c.body["events"]) for c in calls)
+    assert total == 5
+    await client.shutdown()
+
+
+async def test_splits_oversized_queue_into_multiple_batches():
+    calls, transport = make_transport(lambda *_: 202)
+    client, _ = make_client(transport, max_batch_size=10)
+
+    for _ in range(25):
+        client.track(MochiEvent(type="command", name="x"))
+    await client.flush()
+
+    assert len(calls) >= 3
+    total = sum(len(c.body["events"]) for c in calls)
+    assert total == 25
+    await client.shutdown()
+
+
+async def test_retries_retryable_failure_then_succeeds():
+    calls, transport = make_transport(lambda _call, index: 503 if index == 0 else 202)
+    client, errors = make_client(transport)
+
+    client.track(MochiEvent(type="command", name="play"))
+    await client.flush()
+
+    assert len(calls) == 2
+    assert errors == []
+    await client.shutdown()
+
+
+async def test_drops_batch_and_reports_on_non_retryable_error():
+    calls, transport = make_transport(lambda *_: 400)
+    client, errors = make_client(transport)
+
+    client.track(MochiEvent(type="command", name="play"))
+    await client.flush()
+
+    assert len(calls) == 1  # no retries on 400
+    assert len(errors) == 1
+    assert "400" in str(errors[0])
+    await client.shutdown()
+
+
+async def test_drops_oldest_events_on_queue_overflow():
+    calls, transport = make_transport(lambda *_: 202)
+    client, errors = make_client(transport, max_queue_size=3, max_batch_size=100)
+
+    for i in range(5):
+        client.track(MochiEvent(type="custom", name=f"event-{i}"))
+    await client.flush()
+
+    assert len(errors) > 0
+    names = [e["name"] for c in calls for e in c.body["events"]]
+    assert names == ["event-2", "event-3", "event-4"]
+    await client.shutdown()
+
+
+async def test_sends_snapshots_immediately():
+    calls, transport = make_transport(lambda *_: 202)
+    client, _ = make_client(transport)
+
+    await client.snapshot(MochiSnapshot(guild_count=42, ws_ping_ms=30))
+
+    assert len(calls) == 1
+    assert calls[0].url == "http://localhost:9999/api/v1/snapshot"
+    assert calls[0].body["guildCount"] == 42
+    await client.shutdown()
+
+
+async def test_honors_retry_after_on_429():
+    timestamps: list[float] = []
+
+    async def transport(url: str, body: Any):
+        timestamps.append(time.monotonic())
+        if len(timestamps) == 1:
+            return 429, "{}", {"Retry-After": "1"}
+        return 202, "{}"
+
+    client, errors = make_client(transport)
+
+    start = time.monotonic()
+    client.track(MochiEvent(type="command", name="play"))
+    await client.flush()
+
+    assert len(timestamps) == 2
+    assert errors == []
+    # Retry-After: 1s must dominate the 0.5s computed backoff.
+    assert time.monotonic() - start >= 0.9
+    await client.shutdown()
+
+
+async def test_never_raises_when_on_error_raises():
+    calls, transport = make_transport(lambda *_: 202)
+
+    def boom(_err: Exception) -> None:
+        raise RuntimeError("handler blew up")
+
+    client = MochiClient(
+        "http://localhost:9999/",
+        "mochi_sk_test",
+        flush_interval=60.0,
+        max_queue_size=1,
+        on_error=boom,
+        transport=transport,
+    )
+
+    # Overflow reports from inside track(); a throwing handler must not escape.
+    client.track(MochiEvent(type="custom", name="a"))
+    client.track(MochiEvent(type="custom", name="b"))
+    await client.shutdown()