Add PostHog

Author: Twixes

.template.env

@@ -15,3 +15,5 @@ WOLFRAM_ALPHA_APP_ID=
 SENTRY_DSN=
 SENTRY_AUTH_TOKEN=
 SENTRY_ORG=
+POSTHOG_API_KEY=
+POSTHOG_HOST=

configuration.py

@@ -120,6 +120,8 @@ def get(self, key: str) -> Any:
     Setting(
         'disco_max_file_size_in_mib', description='Maksymalny rozmiar pliku utworu disco', unit='MiB', default_value=16
     ),
+    Setting('posthog_api_key', description='Klucz API PostHog', optional=True),
+    Setting('posthog_host', description='Host PostHog', default_value='https://us.i.posthog.com'),
 )
 
 configuration = Configuration(SETTINGS)

docker-compose.dev.yml

@@ -57,6 +57,8 @@ services:
       - CLICKHOUSE_USER=somsiad
       - CLICKHOUSE_PASSWORD=somsiad
       - CLICKHOUSE_DATABASE
+      - POSTHOG_API_KEY
+      - POSTHOG_HOST
       - REDIS_URL=redis://redis:6379/
   web:
     build: web

docker-compose.yml

@@ -32,6 +32,8 @@ services:
       - CLICKHOUSE_USER
       - CLICKHOUSE_PASSWORD
       - CLICKHOUSE_DATABASE
+      - POSTHOG_API_KEY
+      - POSTHOG_HOST
       - REDIS_URL=redis://redis:6379/
   web:
     build: web

plugins/commands.py

@@ -14,6 +14,7 @@
 import discord
 from core import Invocation, is_user_opted_out_of_data_processing
 import datetime as dt
+import posthog
 from somsiad import Somsiad, SomsiadMixin
 
 from discord.ext import commands
@@ -41,6 +42,21 @@ async def on_command(self, ctx: commands.Context):
             )
             session.add(invocation)
 
+            # Track command invocation in PostHog
+            posthog.capture(
+                distinct_id=str(ctx.author.id),
+                event='command_invoked',
+                properties={
+                    'command': ctx.command.qualified_name,
+                    'root_command': ctx.command.root_parent or ctx.command.qualified_name,
+                    'prefix': ctx.prefix,
+                    'server_id': ctx.guild.id if ctx.guild else None,
+                    'server_name': ctx.guild.name if ctx.guild else None,
+                    'channel_id': ctx.channel.id,
+                    'is_dm': ctx.guild is None,
+                }
+            )
+
     @commands.Cog.listener()
     async def on_command_completion(self, ctx: commands.Context):
         with data.session(commit=True) as session:
@@ -50,6 +66,19 @@ async def on_command_completion(self, ctx: commands.Context):
             if invocation is not None:
                 invocation.exited_at = dt.datetime.now()
 
+                # Track command completion in PostHog
+                duration_ms = (invocation.exited_at - invocation.created_at).total_seconds() * 1000
+                posthog.capture(
+                    distinct_id=str(ctx.author.id),
+                    event='command_completed',
+                    properties={
+                        'command': ctx.command.qualified_name,
+                        'root_command': ctx.command.root_parent or ctx.command.qualified_name,
+                        'duration_ms': duration_ms,
+                        'server_id': ctx.guild.id if ctx.guild else None,
+                    }
+                )
+
     @commands.Cog.listener()
     async def on_command_error(self, ctx: commands.Context, error: commands.CommandError):
         with data.session(commit=True) as session:
@@ -58,9 +87,23 @@ async def on_command_error(self, ctx: commands.Context, error: commands.CommandE
             invocation = session.query(Invocation).get(ctx.message.id)
             if invocation is not None:
                 invocation.exited_at = dt.datetime.now()
-                invocation.error = text_snippet(
+                error_message = text_snippet(
                     str(error).replace('Command raised an exception: ', ''), Invocation.MAX_ERROR_LENGTH
                 )
+                invocation.error = error_message
+
+                # Track command error in PostHog
+                posthog.capture(
+                    distinct_id=str(ctx.author.id),
+                    event='command_error',
+                    properties={
+                        'command': ctx.command.qualified_name if ctx.command else 'unknown',
+                        'root_command': ctx.command.root_parent or ctx.command.qualified_name if ctx.command else 'unknown',
+                        'error_type': type(error).__name__,
+                        'error_message': error_message,
+                        'server_id': ctx.guild.id if ctx.guild else None,
+                    }
+                )
 
 
 async def setup(bot: Somsiad):

somsiad.py

@@ -38,6 +38,7 @@
 
 import aiohttp
 import discord
+import posthog
 import psutil
 import sentry_sdk
 import aiochclient
@@ -211,6 +212,13 @@ def __init__(self):
         self.system_channel = None
         self.public_channel = None
 
+        # Initialize PostHog if configured
+        if configuration.get('posthog_api_key'):
+            posthog.api_key = configuration['posthog_api_key']
+            posthog.host = configuration['posthog_host']
+        else:
+            posthog.disabled = True
+
     async def on_ready(self):
         psutil.cpu_percent()
         localize()