log.ts

import _ from 'lodash';
import type { WindowsDriver } from '../driver';
import { LogDefRecord, StringRecord } from '@appium/types';

const COLOR_CODE_PATTERN = /\u001b\[(\d+(;\d+)*)?m/g; // eslint-disable-line no-control-regex
const GET_SERVER_LOGS_FEATURE = 'get_server_logs';
const DEFAULT_LOG_LEVEL = 'ALL';

export const supportedLogTypes: LogDefRecord = {
  server: {
    description: 'Appium server logs',
    getter: (self: WindowsDriver): LogEntry[] => {
      self.assertFeatureEnabled(GET_SERVER_LOGS_FEATURE);
      return self.log.unwrap().record.map(nativeLogEntryToSeleniumEntry);
    },
  },
};

function nativeLogEntryToSeleniumEntry(x: StringRecord): LogEntry {
  const msg = _.isEmpty(x.prefix) ? x.message : `[${x.prefix}] ${x.message}`;
  return toLogEntry(
    _.replace(msg, COLOR_CODE_PATTERN, ''),
    x.timestamp ?? Date.now()
  );
}

function toLogEntry(
  message: string,
  timestamp: number,
  level: string = DEFAULT_LOG_LEVEL
): LogEntry {
  return { timestamp, level, message };
}

interface LogEntry {
  timestamp: number;
  level: string;
  message: string;
}