watermelontools · EstebanDalelR · Jul 12, 2023 · Jul 12, 2023 · Jul 12, 2023 · Jul 12, 2023
diff --git a/app/layout.tsx b/app/layout.tsx
@@ -1,11 +1,18 @@
 import "@primer/css/index.scss";
 
-import AuthProvider from "../lib/auth/AuthProvider";
-import Navbar from "../components/Navbar";
 import { getServerSession } from "next-auth";
+
+import Navbar from "../components/Navbar";
 import Header from "../components/Header";
-import { authOptions } from "./api/auth/[...nextauth]/route";
 import LogInBtn from "../components/login-btn";
+
+import AuthProvider from "../lib/auth/AuthProvider";
+
+import { PHProvider, PostHogPageview } from "./providers";
+
+import { authOptions } from "./api/auth/[...nextauth]/route";
+import { ReactNode, Suspense } from "react";
+
 export const metadata = {
   title: {
     template: "%s | Watermelon",
@@ -17,26 +24,31 @@ export const metadata = {
 export default async function RootLayout({
   children,
 }: {
-  children: React.ReactNode;
+  children: ReactNode;
 }) {
   const session = await getServerSession(authOptions);
   let userEmail = session?.user?.email;
   let userName = session?.user?.name;
 
   return (
     <html lang="en" data-color-mode="dark" data-dark-theme="dark">
-      <body>
-        {userEmail ? (
-          <>
-            <Header userEmail={userEmail} userToken={userName} />
-            <Navbar>
-              <AuthProvider>{children}</AuthProvider>
-            </Navbar>
-          </>
-        ) : (
-          <LogInBtn />
-        )}
-      </body>
+      <Suspense fallback={null}>
+        <PostHogPageview />
+      </Suspense>
+      <PHProvider>
+        <body>
+          {userEmail ? (
+            <>
+              <Header userEmail={userEmail} userToken={userName} />
+              <Navbar>
+                <AuthProvider>{children}</AuthProvider>
+              </Navbar>
+            </>
+          ) : (
+            <LogInBtn />
+          )}
+        </body>
+      </PHProvider>
     </html>
   );
 }
diff --git a/app/providers.tsx b/app/providers.tsx
@@ -0,0 +1,34 @@
+"use client";
+import posthog from "posthog-js";
+import { PostHogProvider } from "posthog-js/react";
+import { usePathname, useSearchParams } from "next/navigation";
+import { useEffect } from "react";
+
+if (typeof window !== "undefined") {
+  posthog.init(process.env.NEXT_PUBLIC_POSTHOG_KEY!, {
+    api_host: process.env.NEXT_PUBLIC_POSTHOG_HOST!,
+  });
+}
+
+export function PostHogPageview(): JSX.Element {
+  const pathname = usePathname();
+  const searchParams = useSearchParams();
+
+  useEffect(() => {
+    if (pathname) {
+      let url = window.origin + pathname;
+      if (searchParams && searchParams.toString()) {
+        url = url + `?${searchParams.toString()}`;
+      }
+      posthog.capture("$pageview", {
+        $current_url: url,
+      });
+    }
+  }, [pathname, searchParams]);
+
+  return <></>;
+}
+
+export function PHProvider({ children }: { children: React.ReactNode }) {
+  return <PostHogProvider client={posthog}>{children}</PostHogProvider>;
+}
diff --git a/lib/track/posthogTracker.ts b/lib/track/posthogTracker.ts
@@ -0,0 +1,10 @@
+import "server-only";
+
+import { PostHog } from "posthog-node";
+
+export default function PostHogTracker() {
+  const posthogClient = new PostHog(process.env.NEXT_PUBLIC_POSTHOG_KEY!, {
+    host: process.env.NEXT_PUBLIC_POSTHOG_HOST!,
+  });
+  return posthogClient;
+}
diff --git a/package.json b/package.json
@@ -26,6 +26,8 @@
     "nodemailer": "^6.7.8",
     "octokit": "^2.0.9",
     "openai": "^3.2.1",
+    "posthog-js": "^1.70.2",
+    "posthog-node": "^3.1.1",
     "react": "^18.2.0",
     "react-dom": "^18.2.0",
     "server-only": "0.0.1",

diff --git a/pages/api/actions/github.ts b/pages/api/actions/github.ts
@@ -18,6 +18,7 @@ import linearMarkdown from "../../../utils/actions/markdownHelpers/linear";
 import countMarkdown from "../../../utils/actions/markdownHelpers/count";
 
 import addActionLog from "../../../utils/db/github/addActionLog";
+import PostHogTracker from "../../../lib/track/posthogTracker";
 const app = new App({
   appId: process.env.GITHUB_APP_ID!,
   privateKey: process.env.GITHUB_PRIVATE_KEY!,
@@ -301,12 +302,10 @@ export default async (req, res) => {
             searchStringSet
               .concat(` ${title.split("/").join(" ")}`)
               .concat(` ${body}`.split("\n").join(" "))
-              .split("\n")
-              .flatMap((line) => line.split(","))
-              .map((commit: string) => commit.toLowerCase())
-              .filter((commit) => !stopwords.includes(commit))
-              .join(" ")
               .split(" ")
+              .flatMap((word) => word.split(","))
+              .map((word: string) => word.toLowerCase())
+              .filter((word) => !stopwords.includes(word))
           )
         ).join(" ");
         // select six random words from the search string
@@ -376,7 +375,8 @@ export default async (req, res) => {
           if (businessLogicSummary) {
             textToWrite += businessLogicSummary;
           } else {
-            textToWrite += "Error getting summary" + businessLogicSummary.error + "\n";
+            textToWrite +=
+              "Error getting summary" + businessLogicSummary.error + "\n";
           }
         } else {
           textToWrite += `AI Summary deactivated by ${userLogin} \n`;
@@ -412,6 +412,19 @@ export default async (req, res) => {
           isPrivateRepo: repository.private,
           repoName: repo,
         });
+
+        PostHogTracker().capture({
+          distinctId: watermelon_user,
+          event: "GitHub Action",
+          properties: {
+            user: userLogin,
+            owner,
+            repo,
+            action: payload.action,
+            //@ts-ignore
+            issue_number: number,
+          },
+        });
         await addActionLog({
           randomWords,
           ghValue,
@@ -428,7 +441,6 @@ export default async (req, res) => {
           count,
           watermelon_user,
         });
-
         // Fetch all comments on the PR
         const comments = await octokit.request(
           "GET /repos/{owner}/{repo}/issues/{issue_number}/comments?sort=created&direction=desc",

diff --git a/pages/api/extension/getContext.ts b/pages/api/extension/getContext.ts
@@ -4,6 +4,7 @@ import getAllData from "../../../utils/db/user/getAllData";
 import updateTokensFromJira from "../../../utils/jira/updateTokens";
 import updateTokens from "../../../utils/db/jira/updateTokens";
 import searchMessageByText from "../../../utils/slack/searchMessageByText";
+import PostHogTracker from "../../../lib/track/posthogTracker";
 function replaceSpecialChars(inputString) {
   const specialChars = /[!"#$%&/()=?_"{}¨*]/g; // Edit this list to include or exclude characters
   return inputString.toLowerCase().replace(specialChars, " ");
@@ -183,12 +184,24 @@ export default async function handler(req, res) {
   }
 
   const githubIssues = await fetchGitHubIssues(userTokens, owner, repo);
-  let PRTitles = githubIssues.map((issue) => issue.title);
-  const jiraIssues = await fetchJiraTickets({ user, ...userTokens }, PRTitles);
-  const slackConversations = await fetchSlackConversations(
-    userTokens,
-    PRTitles
-  );
+  const PRTitles = githubIssues.map((issue) => issue.title);
+  const [jiraIssues, slackConversations] = await Promise.all([
+    fetchJiraTickets({ user, ...userTokens }, PRTitles),
+    fetchSlackConversations(userTokens, PRTitles),
+  ]);
+  PostHogTracker().capture({
+    distinctId: user,
+    event: "extensionContext",
+    properties: {
+      githubIssues,
+      jiraIssues,
+      slackConversations,
+      user,
+      repo,
+      owner,
+      gitSystem,
+    },
+  });
 
   return res.send({
     github: githubIssues,

diff --git a/pages/api/hover/getHoverData.ts b/pages/api/hover/getHoverData.ts
@@ -4,10 +4,15 @@ import { trackEvent } from "../../../utils/analytics/azureAppInsights";
 import updateTokensFromJira from "../../../utils/jira/updateTokens";
 import updateTokens from "../../../utils/db/jira/updateTokens";
 import searchMessageByText from "../../../utils/slack/searchMessageByText";
+import PostHogTracker from "../../../lib/track/posthogTracker";
 function replaceSpecialChars(inputString) {
   const specialChars = /[!"#$%&/()=?_"{}¨*]/g; // Edit this list to include or exclude characters
   return inputString.toLowerCase().replace(specialChars, " ");
 }
+function handleRejection(reason) {
+  console.error(reason);
+  return { error: reason };
+}
 export default async function handler(req, res) {
   const { user, gitSystem, repo, owner, commitTitle } = req.body;
 
@@ -122,12 +127,38 @@ export default async function handler(req, res) {
     }
     return slackValue;
   }
-  const githubIssues = await fetchGitHubIssues(userTokens, owner, repo);
-  const jiraTickets = await fetchJiraTickets(userTokens, commitTitle);
-  const slackConversations = await fetchSlackConversations(
-    userTokens,
-    commitTitle
-  );
+  const [githubResult, jiraResult, slackResult] = await Promise.allSettled([
+    fetchGitHubIssues(userTokens, owner, repo),
+    fetchJiraTickets(userTokens, commitTitle),
+    fetchSlackConversations(userTokens, commitTitle),
+  ]);
+
+  const githubIssues =
+    githubResult.status === "fulfilled"
+      ? githubResult.value
+      : handleRejection(githubResult.reason);
+  const jiraTickets =
+    jiraResult.status === "fulfilled"
+      ? jiraResult.value
+      : handleRejection(jiraResult.reason);
+  const slackConversations =
+    slackResult.status === "fulfilled"
+      ? slackResult.value
+      : handleRejection(slackResult.reason);
+
+  PostHogTracker().capture({
+    distinctId: user,
+    event: "unifiedHoverData",
+    properties: {
+      gitSystem,
+      repo,
+      owner,
+      commitTitle,
+      githubIssues,
+      jiraTickets,
+      slackConversations,
+    },
+  });
   trackEvent({
     name: "unifiedHoverData",
     properties: { user, gitSystem, repo, owner, commitTitle },