"use client";

import { useEffect } from "react";
import posthog from "posthog-js";
import { PostHogProvider as PHProvider } from "posthog-js/react";
import { useSession } from "@/utils/auth-client";
import { usePathname, useSearchParams } from "next/navigation";
import { env } from "@/env";
import { useAccount } from "@/providers/EmailAccountProvider";
import {
  getAppPageViewProperties,
  PRODUCT_ANALYTICS_EVENTS,
} from "@/utils/analytics/product";

// based on: https://posthog.com/docs/libraries/next-js

export function PostHogPageview() {
  const pathname = usePathname();
  const searchParams = useSearchParams();

  useEffect(() => {
    if (pathname) {
      let url = window.origin + pathname;
      if (searchParams?.toString()) {
        url = `${url}?${searchParams.toString()}`;
      }
      posthog.capture("$pageview", {
        $current_url: url,
      });

      const appPageProperties = getAppPageViewProperties({
        pathname,
        searchParams,
      });
      if (appPageProperties) {
        posthog.capture(PRODUCT_ANALYTICS_EVENTS.pageViewed, appPageProperties);
      }
    }
  }, [pathname, searchParams]);

  return null;
}

export function PostHogIdentify() {
  const { data: session } = useSession();
  const { emailAccount } = useAccount();

  useEffect(() => {
    if (session?.user.email)
      posthog.identify(session.user.email, {
        email: session.user.email,
      });
  }, [session?.user.email]);

  useEffect(() => {
    // Set super properties that will be included with all events
    posthog.register({
      email_account_id: emailAccount?.id,
      email_account_email: emailAccount?.email,
      email_account_provider: emailAccount?.account?.provider,
    });

    // Most users only use one email account, and it's helpful to have the provider on the person property
    if (emailAccount) {
      posthog.setPersonProperties(
        {},
        {
          default_email_account_provider: emailAccount?.account?.provider,
        },
      );
    }
  }, [emailAccount]);

  return null;
}

if (typeof window !== "undefined" && env.NEXT_PUBLIC_POSTHOG_KEY) {
  posthog.init(env.NEXT_PUBLIC_POSTHOG_KEY, {
    api_host: env.NEXT_PUBLIC_POSTHOG_API_HOST, // https://posthog.com/docs/advanced/proxy/nextjs
    capture_pageview: false, // Disable automatic pageview capture, as we capture manually
  });
}

export function PostHogProvider({ children }: { children: React.ReactNode }) {
  return <PHProvider client={posthog}>{children}</PHProvider>;
}