// ISC License

// Copyright (c) 2021 Alexey Raspopov, Kostiantyn Denysov, Anton Verinov

// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.

// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
//
// https://github.com/alexeyraspopov/picocolors/blob/b6261487e7b81aaab2440e397a356732cad9e342/picocolors.js#L1

const { env, stdout } = globalThis?.process ?? {};

const enabled =
  env &&
  !env.NO_COLOR &&
  (env.FORCE_COLOR || (stdout?.isTTY && !env.CI && env.TERM !== "dumb"));

const replaceClose = (
  str: string,
  close: string,
  replace: string,
  index: number,
): string => {
  const start = str.substring(0, index) + replace;
  const end = str.substring(index + close.length);
  const nextIndex = end.indexOf(close);
  return ~nextIndex
    ? start + replaceClose(end, close, replace, nextIndex)
    : start + end;
};

const formatter = (open: string, close: string, replace = open) => {
  if (!enabled) return String;
  return (input: string) => {
    const string = "" + input;
    const index = string.indexOf(close, open.length);
    return ~index
      ? open + replaceClose(string, close, replace, index) + close
      : open + string + close;
  };
};

export const reset = enabled ? (s: string) => `\x1b[0m${s}\x1b[0m` : String;
export const bold = formatter("\x1b[1m", "\x1b[22m", "\x1b[22m\x1b[1m");
export const dim = formatter("\x1b[2m", "\x1b[22m", "\x1b[22m\x1b[2m");
export const italic = formatter("\x1b[3m", "\x1b[23m");
export const underline = formatter("\x1b[4m", "\x1b[24m");
export const inverse = formatter("\x1b[7m", "\x1b[27m");
export const hidden = formatter("\x1b[8m", "\x1b[28m");
export const strikethrough = formatter("\x1b[9m", "\x1b[29m");
export const black = formatter("\x1b[30m", "\x1b[39m");
export const red = formatter("\x1b[31m", "\x1b[39m");
export const green = formatter("\x1b[32m", "\x1b[39m");
export const yellow = formatter("\x1b[33m", "\x1b[39m");
export const blue = formatter("\x1b[34m", "\x1b[39m");
export const magenta = formatter("\x1b[35m", "\x1b[39m");
export const purple = formatter("\x1b[38;2;173;127;168m", "\x1b[39m");
export const cyan = formatter("\x1b[36m", "\x1b[39m");
export const white = formatter("\x1b[37m", "\x1b[39m");
export const gray = formatter("\x1b[90m", "\x1b[39m");
export const bgBlack = formatter("\x1b[40m", "\x1b[49m");
export const bgRed = formatter("\x1b[41m", "\x1b[49m");
export const bgGreen = formatter("\x1b[42m", "\x1b[49m");
export const bgYellow = formatter("\x1b[43m", "\x1b[49m");
export const bgBlue = formatter("\x1b[44m", "\x1b[49m");
export const bgMagenta = formatter("\x1b[45m", "\x1b[49m");
export const bgCyan = formatter("\x1b[46m", "\x1b[49m");
export const bgWhite = formatter("\x1b[47m", "\x1b[49m");