import { createLogger, format, transports } from 'winston';

// Define some custom colors for your custom tokens
const customColors = {
  user_id: 'magenta',
  user_email: 'magenta',
  route: 'cyan',
  method: 'cyan',
  // etc.
};

// Create a colorizer with addColors
const colorizer = format.colorize();
colorizer.addColors(customColors);

const colorLevel = (level) => {
  switch (level) {
    case 'info':
      // Green
      return `\x1b[32m${level}\x1b[0m`;
    case 'warn':
      // Yellow
      return `\x1b[33m${level}\x1b[0m`;
    case 'error':
      // Red
      return `\x1b[31m${level}\x1b[0m`;
    default:
      // No color for other levels
      return `\x1b[32m${level}\x1b[0m`;
  }
};

// The function that replicates your printf logic, but returns info
// (instead of a final string).
const mutateMessageFormat = format((info) => {
  const { level, message, timestamp, route, method, user_id, user_email } =
    info;

  // Colorize the level
  const coloredLevel = colorLevel(level);

  // Build your custom base string, with the colored level in brackets
  const base = `${timestamp} [${coloredLevel}]`;

  // Conditionally append user info if present
  const userInfo = user_id
    ? ` ${colorizer.colorize(
        'user_id',
        `user:${user_id}`
      )}${user_email ? colorizer.colorize('user_email', ` ${user_email}`) : ''}`
    : '';

  // Conditionally append request info if present
  const requestInfo =
    route && method
      ? ` (${colorizer.colorize('method', method)} ${colorizer.colorize(
          'route',
          route
        )})`
      : '';

  // Mutate the log message in place, combining your custom string logic
  info.message = `${base}${userInfo}${requestInfo} - ${message}`;

  // Return `info` so Winston passes it along the pipeline
  return info;
});

const getFormattedTimestamp = () => {
  const now = new Date();
  const utcTime = now.toLocaleString('en-US', {
    timeZone: 'UTC',
    hour: '2-digit',
    minute: '2-digit',
    second: '2-digit',
    hour12: false,
  });
  const dubaiTime = now.toLocaleString('en-US', {
    timeZone: 'Asia/Dubai',
    hour: '2-digit',
    minute: '2-digit',
    hour12: false,
  });
  return `${utcTime} UTC (${dubaiTime} GST)`;
};

// Create a Winston logger that ultimately logs in JSON format
const logger = createLogger({
  level: 'info',
  format: format.combine(
    // Attach a timestamp
    format.timestamp({ format: getFormattedTimestamp }),
    // Mutate the message with your fancy string
    mutateMessageFormat(),
    // Finally, serialize EVERYTHING to JSON
    format.json()
  ),
  transports: [
    new transports.Console({
      stderrLevels: ['error', 'warn'],
    }),
  ],
});

export default logger;