/*
 * MIT License
 *
 * Copyright (c) 2019 nest-mods
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

import { Logger } from '@nestjs/common';
import * as _ from 'lodash';
import { Helpers } from './helpers.util';

/**
 * @see Logger.constructor
 */
export interface InjectLoggerOptions {
  context?: string;
  isTimeDiffEnabled?: boolean;
}

export function Log(
  prefix?: string,
  options?: InjectLoggerOptions,
): PropertyDecorator;
export function Log(options?: InjectLoggerOptions): PropertyDecorator;
export function Log(
  prefixOrOptions?: string | InjectLoggerOptions,
  options?: InjectLoggerOptions,
): PropertyDecorator {
  let prefix;
  if (_.isString(prefixOrOptions)) {
    prefix = prefixOrOptions;
  } else {
    prefix = Helpers.DEFAULT_PREFIX;
    options = prefixOrOptions || options;
  }
  return (target: object, propertyKey: string | symbol) => {
    const context = options?.context || Helpers.getClassName(target);
    const isTimeDiffEnabled = options?.isTimeDiffEnabled || true;
    target[propertyKey] = new Logger(`${prefix}:${context}`, {
      timestamp: isTimeDiffEnabled,
    });
  };
}