Instrument the cli by calling a proxied cfapi endpoint that calls posthog

codeflash/api/cfapi.py

@@ -1,3 +1,4 @@
+import logging
 from typing import Optional, Dict, Any
 
 import requests
@@ -5,6 +6,7 @@ from requests import Response
 
 from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.github.PrComment import PrComment
+from codeflash.version import __version__, __version_tuple__
 
 CFAPI_BASE_URL = "https://app.codeflash.ai"
 # CFAPI_BASE_URL = "http://localhost:3001"
@@ -30,6 +32,25 @@ def make_cfapi_request(
     return response
 
 
+def log_posthog_event(event: str, properties: Dict[str, Any] = None) -> None:
+    """
+    Log an event to PostHog.
+    :param event: The name of the event.
+    :param properties: A dictionary of properties to attach to the event.
+    """
+    if properties is None:
+        properties = {}
+
+    properties["cli_version"] = __version__
+    properties["cli_version_tuple"] = __version_tuple__
+
+    response = make_cfapi_request(
+        "/posthog-capture-from-cli", "POST", {"event": event, "properties": properties}
+    )
+    if response.status_code != 200:
+        logging.debug(f"Failed to log event to PostHog: {response.text}")
+
+
 def suggest_changes(
     owner: str,
     repo: str,

codeflash/cli_cmds/cmd_init.py

@@ -6,6 +6,7 @@ import click
 import tomlkit
 from git import Repo
 
+from codeflash.api.cfapi import log_posthog_event
 from codeflash.cli_cmds.cli import CODEFLASH_LOGO
 from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.code_utils.git_utils import get_github_secrets_page_url
@@ -38,14 +39,20 @@ def init_codeflash():
         "    codeflash --help to see all options"
     )
 
+    log_posthog_event("cli-installation-successful")
+
 
 def collect_setup_info(setup_info: dict[str, str]):
     setup_info["project_root"] = click.prompt(
         "What's your project's root directory?", default=f"{os.getcwd()}"
     )
+    log_posthog_event("cli-project-root-provided")
     setup_info["test_framework"] = click.prompt(
         "Which test framework do you use?", default="pytest"
     )  # TODO options pytest/unittest
+    log_posthog_event(
+        "cli-test-framework-provided", {"test_framework": setup_info["test_framework"]}
+    )
     test_subdir = "tests"  # maybe different defaults for pytest vs unittest?
     # TODO discover test dir, if we can't ask for it
     tests_root = click.prompt(
@@ -53,6 +60,7 @@ def collect_setup_info(setup_info: dict[str, str]):
         default=os.path.join(setup_info["project_root"], test_subdir),
     )
     setup_info["tests_root"] = os.path.relpath(tests_root, setup_info["project_root"])
+    log_posthog_event("cli-tests-root-provided", {"tests_root": setup_info["tests_root"]})
     # Ask for paths to ignore and update the setup_info dictionary
     # ignore_paths_input = click.prompt("Are there any paths CodeFlash should ignore? (comma-separated, no spaces)",
     #                                   default='', show_default=False)
@@ -72,7 +80,9 @@ def prompt_github_action(setup_info: dict[str, str]):
         .lower()
         .strip()
     )
-    if optimize_prs.startswith("y"):
+    optimize_yes = optimize_prs.startswith("y")
+    log_posthog_event("cli-github-optimization-choice", {"optimize_prs": optimize_yes})
+    if optimize_yes:
         repo = Repo(setup_info["project_root"], search_parent_directories=True)
         git_root = repo.git.rev_parse("--show-toplevel")
         workflows_path = os.path.join(git_root, ".github", "workflows")
@@ -87,7 +97,12 @@ def prompt_github_action(setup_info: dict[str, str]):
             .lower()
             .strip()
         )
-        if confirm_creation.startswith("y"):
+        confirm_creation_yes = confirm_creation.startswith("y")
+        log_posthog_event(
+            "cli-github-optimization-confirm-workflow-creation",
+            {"confirm_creation": confirm_creation_yes},
+        )
+        if confirm_creation_yes:
             os.makedirs(workflows_path, exist_ok=True)
             from importlib.resources import read_text
 
@@ -108,6 +123,10 @@ def prompt_github_action(setup_info: dict[str, str]):
                 show_default=False,
             )
             click.launch(get_github_secrets_page_url(repo))
+            log_posthog_event("cli-github-workflow-created")
+        else:
+            click.echo("Skipping GitHub workflow creation.")
+            log_posthog_event("cli-github-workflow-skipped")
 
 
 # Create or update the pyproject.toml file with the CodeFlash dependency & configuration
@@ -182,12 +201,15 @@ def prompt_api_key() -> bool:
             show_default=False,
         )
         if use_existing_key == "":
+            log_posthog_event("cli-existing-api-key-used")
             return False
         else:
             enter_api_key_and_save_to_rc()
+            log_posthog_event("cli-new-api-key-entered")
             return True
     else:
         enter_api_key_and_save_to_rc()
+        log_posthog_event("cli-new-api-key-entered")
         return True
 
 

codeflash/version.py

@@ -1,3 +1,3 @@
 # These version placeholders will be replaced by poetry-dynamic-versioning during `poetry build`.
-__version__ = "0.1.1"
-__version_tuple__ = (0, 1, 1)
+__version__ = "0.1.1.post30.dev0+ae81de0"
+__version_tuple__ = (0, 1, 1, "post30", "dev0", "ae81de0")