refactor(hooks,providers): rewrite useAckee hook and AckeeProvider

5 files changed, 143 insertions, 59 deletions

diff --git a/src/utils/providers/ackee-provider/ackee-provider.test.tsx b/src/utils/providers/ackee-provider/ackee-provider.test.tsx
new file mode 100644
index 0000000..7ba11dc
--- /dev/null
+++ b/src/utils/providers/ackee-provider/ackee-provider.test.tsx
@@ -0,0 +1,46 @@
+import { describe, expect, it } from '@jest/globals';
+import { render, screen as rtlScreen } from '@testing-library/react';
+import { type FC, useContext } from 'react';
+import type { AckeeTrackerValue } from '../../../types';
+import { AckeeContext, AckeeProvider } from './ackee-provider';
+
+const bodyPrefix = 'Tracking is set to:';
+
+const ComponentTest: FC = () => {
+  const { tracking } = useContext(AckeeContext);
+
+  return (
+    <div>
+      {bodyPrefix} {tracking}
+    </div>
+  );
+};
+
+describe('AckeeProvider', () => {
+  it('uses the default value when the provider is not used', () => {
+    render(<ComponentTest />);
+
+    expect(rtlScreen.getByText(new RegExp(bodyPrefix))).toHaveTextContent(
+      `${bodyPrefix} full`
+    );
+  });
+
+  it('provides the given tracking value to its children', () => {
+    const tracking: AckeeTrackerValue = 'partial';
+
+    render(
+      <AckeeProvider
+        domainId="some-id"
+        server="https://example.com"
+        storageKey="facilis"
+        tracking={tracking}
+      >
+        <ComponentTest />
+      </AckeeProvider>
+    );
+
+    expect(rtlScreen.getByText(new RegExp(bodyPrefix))).toHaveTextContent(
+      `${bodyPrefix} ${tracking}`
+    );
+  });
+});
diff --git a/src/utils/providers/ackee-provider/ackee-provider.tsx b/src/utils/providers/ackee-provider/ackee-provider.tsx
new file mode 100644
index 0000000..8cd29cd
--- /dev/null
+++ b/src/utils/providers/ackee-provider/ackee-provider.tsx
@@ -0,0 +1,95 @@
+import { useRouter } from 'next/router';
+import {
+  type FC,
+  type ReactNode,
+  createContext,
+  type Dispatch,
+  type SetStateAction,
+  useMemo,
+} from 'react';
+import { useAckee } from 'use-ackee';
+import type { AckeeTrackerValue } from '../../../types';
+import { useLocalStorage } from '../../hooks';
+
+type AckeeContextProps = {
+  tracking: AckeeTrackerValue;
+  setTracking: Dispatch<SetStateAction<AckeeTrackerValue>>;
+};
+
+export const AckeeContext = createContext<AckeeContextProps>({
+  setTracking: (value) => value,
+  tracking: 'full',
+});
+
+const validator = (value: unknown): value is AckeeTrackerValue =>
+  value === 'full' || value === 'partial';
+
+export type AckeeProviderProps = {
+  /**
+   * The provider children.
+   */
+  children?: ReactNode;
+  /**
+   *  The id given by Ackee for this domain.
+   */
+  domainId: string;
+  /**
+   * Should we track visits from localhost?
+   *
+   * @default false
+   */
+  isLocalhostTracked?: boolean;
+  /**
+   * Should we track our own visits?
+   *
+   * @default false
+   */
+  isOwnVisitsTracked?: boolean;
+  /**
+   * An URL that points to your Ackee installation (without trailing slash).
+   */
+  server: string;
+  /**
+   * The key to use in local storage.
+   */
+  storageKey: string;
+  /**
+   * Should the tracking be detailed (full) or not (partial)?
+   */
+  tracking: AckeeTrackerValue;
+};
+
+export const AckeeProvider: FC<AckeeProviderProps> = ({
+  children,
+  domainId,
+  isLocalhostTracked = false,
+  isOwnVisitsTracked = false,
+  server,
+  storageKey,
+  tracking: tracker,
+}) => {
+  const [tracking, setTracking] = useLocalStorage<AckeeTrackerValue>(
+    storageKey,
+    tracker,
+    validator
+  );
+  const { asPath } = useRouter();
+
+  useAckee(
+    asPath,
+    { domainId, server },
+    {
+      detailed: tracking === 'full',
+      ignoreLocalhost: !isLocalhostTracked,
+      ignoreOwnVisits: !isOwnVisitsTracked,
+    }
+  );
+
+  const value = useMemo(() => {
+    return { setTracking, tracking };
+  }, [setTracking, tracking]);
+
+  return (
+    <AckeeContext.Provider value={value}>{children}</AckeeContext.Provider>
+  );
+};
diff --git a/src/utils/providers/ackee-provider/index.ts b/src/utils/providers/ackee-provider/index.ts
new file mode 100644
index 0000000..10f7a26
--- /dev/null
+++ b/src/utils/providers/ackee-provider/index.ts
@@ -0,0 +1 @@
+export * from './ackee-provider';
diff --git a/src/utils/providers/ackee.tsx b/src/utils/providers/ackee.tsx
deleted file mode 100644
index 0cb0166..0000000
--- a/src/utils/providers/ackee.tsx
+++ /dev/null
@@ -1,58 +0,0 @@
-import { useRouter } from 'next/router';
-import { createContext, FC, ReactNode, useContext, useState } from 'react';
-import useAckee from 'use-ackee';
-
-export type AckeeProps = {
-  domain: string;
-  siteId: string;
-  detailed?: boolean;
-  setDetailed: (isDetailed: boolean) => void;
-};
-
-export type AckeeProviderProps = {
-  children: ReactNode;
-  domain: string;
-  siteId: string;
-  ignoreLocalhost?: boolean;
-  ignoreOwnVisits?: boolean;
-};
-
-export const AckeeContext = createContext<AckeeProps>({
-  domain: '',
-  siteId: '',
-  setDetailed: (_) => {
-    // Do nothing.
-  },
-});
-
-export const useAckeeTracker = () => useContext(AckeeContext);
-
-export const AckeeProvider: FC<AckeeProviderProps> = ({
-  domain,
-  siteId,
-  children,
-  ignoreLocalhost = true,
-  ignoreOwnVisits = true,
-}) => {
-  const [detailed, setDetailed] = useState<boolean>(false);
-  const { asPath } = useRouter();
-
-  useAckee(
-    asPath,
-    { server: domain, domainId: siteId },
-    { detailed, ignoreLocalhost, ignoreOwnVisits }
-  );
-
-  return (
-    <AckeeContext.Provider
-      value={{
-        domain,
-        siteId,
-        detailed,
-        setDetailed,
-      }}
-    >
-      {children}
-    </AckeeContext.Provider>
-  );
-};
diff --git a/src/utils/providers/index.ts b/src/utils/providers/index.ts
index 43641a1..640730f 100644
--- a/src/utils/providers/index.ts
+++ b/src/utils/providers/index.ts
@@ -1,2 +1,2 @@
-export * from './ackee';
+export * from './ackee-provider';
 export * from './prism-theme';