wow

Author: n1ru4l

packages/libraries/core/src/client/agent.ts

@@ -123,7 +123,7 @@ export function createAgent<TEvent>(
           ? null
           : pluginOptions.circuitBreaker,
   };
-  const logger = createHiveLogger(pluginOptions.logger ?? console, '[agent]', pluginOptions.debug);
+  const logger = createHiveLogger(pluginOptions.logger ?? console, '[agent]');
 
   let circuitBreaker: CircuitBreakerInterface<
     Parameters<typeof sendHTTPCall>,

packages/libraries/core/src/client/client.ts

@@ -10,14 +10,61 @@ import { createPersistedDocuments } from './persisted-documents.js';
 import { createReporting } from './reporting.js';
 import type { HiveClient, HiveInternalPluginOptions, HivePluginOptions } from './types.js';
 import { createUsage } from './usage.js';
-import { createHiveLogger, isLegacyAccessToken } from './utils.js';
+import { createHiveLogger, HiveLogger, isLegacyAccessToken } from './utils.js';
 
-export function createHive(options: HivePluginOptions): HiveClient {
-  const logger = createHiveLogger(
+function chooseDefaultLogger(options: HivePluginOptions): HiveLogger {
+  if (options.logger === 'debug') {
+    return createHiveLogger(
+      {
+        debug(...args) {
+          console.debug(...args);
+        },
+        info(...args) {
+          console.info(...args);
+        },
+        error(...args) {
+          console.error(...args);
+        },
+      },
+      '[hive]',
+    );
+  }
+  if (options.logger === 'info') {
+    return createHiveLogger(
+      {
+        debug() {},
+        info(...args) {
+          console.info(...args);
+        },
+        error(...args) {
+          console.error(...args);
+        },
+      },
+      '[hive]',
+    );
+  }
+  if (options.logger === 'error') {
+    return createHiveLogger(
+      {
+        debug() {},
+        info() {},
+        error(...args) {
+          console.error(...args);
+        },
+      },
+      '[hive]',
+    );
+  }
+
+  return createHiveLogger(
     options?.logger ?? options?.agent?.logger ?? console,
     '[hive]',
     options.debug,
   );
+}
+
+export function createHive(options: HivePluginOptions): HiveClient {
+  const logger = chooseDefaultLogger(options);
   let enabled = options.enabled ?? true;
 
   if (enabled === false && !options.experimental__persistedDocuments) {

packages/libraries/core/src/client/types.ts

@@ -217,9 +217,9 @@ export type HivePluginOptions = OptionalWhenFalse<
     /**
      * Custom logger.
      *
-     * Default: console based logger
+     * Default: 'info'
      */
-    logger?: Logger;
+    logger?: Logger | 'error' | 'info' | 'debug';
     /**
      * Access Token for usage reporting
      */

packages/libraries/core/src/client/utils.ts

@@ -210,9 +210,6 @@ export function createHiveLogger(baseLogger: Logger, prefix: string, debug = tru
 
   return {
     [hiveSymbol]: context,
-    info: (message: string) => {
-      logger.info(printPath(path) + message);
-    },
     error: (error: any, ...data: any[]) => {
       if (error.stack) {
         const pth = printPath(path);
@@ -223,6 +220,9 @@ export function createHiveLogger(baseLogger: Logger, prefix: string, debug = tru
         logger.error(printPath(path) + String(error), ...data);
       }
     },
+    info: (message: string) => {
+      logger.info(printPath(path) + message);
+    },
     debug: (message: string) => {
       if (!context.debug) {
         return;