@rawnodes/logger 2.5.0 → 2.6.0

package/dist/index.mjs

@@ -683,6 +683,7 @@ function createFormattedFilterStream(format, level, rules, store, destination) {
 }
 function createStreams(config, store) {
   const streams = [];
+  const transports = [];
   const consoleStream = createFormattedFilterStream(
     config.console.format,
     config.console.level,
@@ -730,6 +731,7 @@ function createStreams(config, store) {
   }
   for (const discordConfig of toArray(config.discord)) {
     const transport = new DiscordTransport(discordConfig);
+    transports.push(transport);
     const discordStream = createHttpTransportStream(transport, discordConfig.level, discordConfig.rules, store);
     streams.push({
       level: "trace",
@@ -738,6 +740,7 @@ function createStreams(config, store) {
   }
   for (const telegramConfig of toArray(config.telegram)) {
     const transport = new TelegramTransport(telegramConfig);
+    transports.push(transport);
     const telegramStream = createHttpTransportStream(transport, telegramConfig.level, telegramConfig.rules, store);
     streams.push({
       level: "trace",
@@ -746,13 +749,17 @@ function createStreams(config, store) {
   }
   for (const cloudwatchConfig of toArray(config.cloudwatch)) {
     const transport = new CloudWatchTransport(cloudwatchConfig, config.hostname);
+    transports.push(transport);
     const cwStream = createHttpTransportStream(transport, cloudwatchConfig.level, cloudwatchConfig.rules, store);
     streams.push({
       level: "trace",
       stream: cwStream
     });
   }
-  return pino.multistream(streams);
+  return {
+    destination: pino.multistream(streams),
+    transports
+  };
 }
 function toArray(value) {
   if (!value) return [];
@@ -837,14 +844,14 @@ function createState(config, store) {
     levelOverrides.set(key, rule);
   }
   const { contextIndex, complexRules } = buildIndexes(levelOverrides);
-  const streams = createStreams(config, loggerStore);
+  const { destination, transports } = createStreams(config, loggerStore);
   const options = {
     level: "trace",
     // Accept all, we filter in shouldLog()
     customLevels: CUSTOM_LEVELS,
     base: { hostname: config.hostname ?? hostname() }
   };
-  const pinoLogger = pino(options, streams);
+  const pinoLogger = pino(options, destination);
   let callerConfig;
   if (config.caller === true) {
     callerConfig = {};
@@ -858,7 +865,8 @@ function createState(config, store) {
     levelOverrides,
     contextIndex,
     complexRules,
-    callerConfig
+    callerConfig,
+    transports
   };
 }
 function rebuildIndexes(state) {
@@ -1004,6 +1012,10 @@ var CallerConfigSchema = z.object({
   depth: z.number().int().nonnegative().optional(),
   fullPath: z.boolean().optional()
 });
+var AutoShutdownConfigSchema = z.object({
+  timeout: z.number().int().positive().optional(),
+  signals: z.array(z.string()).optional()
+});
 var LoggerConfigSchema = z.object({
   level: LevelConfigSchema,
   console: ConsoleConfigSchema,
@@ -1012,7 +1024,8 @@ var LoggerConfigSchema = z.object({
   telegram: z.union([TelegramConfigSchema, z.array(TelegramConfigSchema)]).optional(),
   cloudwatch: z.union([CloudWatchConfigSchema, z.array(CloudWatchConfigSchema)]).optional(),
   caller: z.union([z.boolean(), CallerConfigSchema]).optional(),
-  hostname: z.string().optional()
+  hostname: z.string().optional(),
+  autoShutdown: z.union([z.boolean(), AutoShutdownConfigSchema]).optional()
 });
 function validateConfig(config) {
   return LoggerConfigSchema.parse(config);
@@ -1081,6 +1094,53 @@ function formatCallerInfo(info) {
   return location;
 }
 
+// src/utils/shutdown.ts
+var DEFAULT_OPTIONS2 = {
+  timeout: 5e3,
+  exitCode: 0,
+  signals: ["SIGTERM", "SIGINT"]
+};
+var registered = false;
+function registerShutdown(logger, options = {}) {
+  const opts = { ...DEFAULT_OPTIONS2, ...options };
+  const shutdown = async (signal) => {
+    if (signal) {
+      console.error(`[Logger] Received ${signal}, shutting down gracefully...`);
+    }
+    try {
+      if (opts.onShutdown) {
+        await opts.onShutdown();
+      }
+      const timeoutPromise = new Promise((_, reject) => {
+        setTimeout(() => {
+          reject(new Error(`Shutdown timed out after ${opts.timeout}ms`));
+        }, opts.timeout);
+      });
+      await Promise.race([
+        logger.shutdown(),
+        timeoutPromise
+      ]);
+      if (signal) {
+        console.error("[Logger] Graceful shutdown completed");
+        process.exit(opts.exitCode);
+      }
+    } catch (error) {
+      console.error("[Logger] Shutdown error:", error instanceof Error ? error.message : error);
+      if (signal) {
+        process.exit(1);
+      }
+    }
+  };
+  if (!registered) {
+    for (const signal of opts.signals) {
+      process.on(signal, () => void shutdown(signal));
+    }
+    process.on("beforeExit", () => void shutdown());
+    registered = true;
+  }
+  return shutdown;
+}
+
 // src/logger.ts
 var Logger = class _Logger {
   constructor(state, context) {
@@ -1091,7 +1151,12 @@ var Logger = class _Logger {
   static create(config, store) {
     const validatedConfig = validateConfig(config);
     const state = createState(validatedConfig, store);
-    return new _Logger(state, "APP");
+    const logger = new _Logger(state, "APP");
+    if (validatedConfig.autoShutdown) {
+      const shutdownConfig = typeof validatedConfig.autoShutdown === "object" ? validatedConfig.autoShutdown : {};
+      registerShutdown(logger, shutdownConfig);
+    }
+    return logger;
   }
   for(context) {
     return new _Logger(this.state, context);
@@ -1133,6 +1198,15 @@ var Logger = class _Logger {
   getLevelOverrides() {
     return Array.from(this.state.levelOverrides.values());
   }
+  // Shutdown
+  /**
+   * Gracefully shutdown the logger, flushing all pending messages.
+   * Should be called before process exit to ensure no logs are lost.
+   */
+  async shutdown() {
+    const closePromises = this.state.transports.map((transport) => transport.close());
+    await Promise.all(closePromises);
+  }
   // Profiling
   profile(id, meta) {
     const existing = this.profileTimers.get(id);
@@ -1399,16 +1473,28 @@ function createMasker(options = {}) {
 }
 
 // src/formatters.ts
-function flattenObject(obj, prefix = "") {
+var MAX_FLATTEN_DEPTH = 10;
+var CIRCULAR_REF = "[Circular]";
+function flattenObject(obj, prefix = "", ancestors = /* @__PURE__ */ new WeakSet(), depth = 0) {
   const result = {};
+  if (depth > MAX_FLATTEN_DEPTH) {
+    result[prefix || "value"] = "[Max depth exceeded]";
+    return result;
+  }
+  ancestors.add(obj);
   for (const [key, value] of Object.entries(obj)) {
     const newKey = prefix ? `${prefix}.${key}` : key;
     if (value !== null && typeof value === "object" && !Array.isArray(value) && !(value instanceof Error)) {
-      Object.assign(result, flattenObject(value, newKey));
+      if (ancestors.has(value)) {
+        result[newKey] = CIRCULAR_REF;
+      } else {
+        Object.assign(result, flattenObject(value, newKey, ancestors, depth + 1));
+      }
     } else {
       result[newKey] = value;
     }
   }
+  ancestors.delete(obj);
   return result;
 }
 function formatLogfmtValue(value) {
@@ -1437,6 +1523,6 @@ function formatLogfmt(data) {
   return Object.entries(flattened).filter(([, value]) => value !== void 0 && value !== null).map(([key, value]) => `${key}=${formatLogfmtValue(value)}`).join(" ");
 }
 
-export { BaseHttpTransport, CallerConfigSchema, CloudWatchConfigSchema, CloudWatchTransport, ConsoleConfigSchema, DiscordConfigSchema, DiscordTransport, FileConfigSchema, HttpTransportBaseConfigSchema, LOG_LEVELS, LevelConfigObjectSchema, LevelConfigSchema, LevelRuleSchema, LogFormatSchema, LogLevelSchema, LogStreamNameSchema, LogStreamPatternConfigSchema, LogStreamPatternSchema, LogStreamTemplateConfigSchema, Logger, LoggerConfigSchema, LoggerStore, MessageBuffer, TelegramConfigSchema, TelegramTransport, assertLogLevel, createMasker, createSingletonLogger, extractRequestId, flattenObject, formatCallerInfo, formatLogfmt, formatLogfmtValue, generateRequestId, getCallerInfo, getOrGenerateRequestId, isValidLogLevel, maskSecrets, matchesContext, measureAsync, measureSync, safeValidateConfig, validateConfig };
+export { AutoShutdownConfigSchema, BaseHttpTransport, CallerConfigSchema, CloudWatchConfigSchema, CloudWatchTransport, ConsoleConfigSchema, DiscordConfigSchema, DiscordTransport, FileConfigSchema, HttpTransportBaseConfigSchema, LOG_LEVELS, LevelConfigObjectSchema, LevelConfigSchema, LevelRuleSchema, LogFormatSchema, LogLevelSchema, LogStreamNameSchema, LogStreamPatternConfigSchema, LogStreamPatternSchema, LogStreamTemplateConfigSchema, Logger, LoggerConfigSchema, LoggerStore, MessageBuffer, TelegramConfigSchema, TelegramTransport, assertLogLevel, createMasker, createSingletonLogger, extractRequestId, flattenObject, formatCallerInfo, formatLogfmt, formatLogfmtValue, generateRequestId, getCallerInfo, getOrGenerateRequestId, isValidLogLevel, maskSecrets, matchesContext, measureAsync, measureSync, registerShutdown, safeValidateConfig, validateConfig };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map