import type { TransformContext, RuntimeContext, Config } from '../types.js';
import { join, extname } from 'path';
import mm from 'micromatch';
import { stringify, TYPE_YAML, TYPE_JSON, TYPE_MARKDOWN, TYPE_TOML } from '../converter/index.js';
const { contains } = mm;

const formatMapping = {
  '.md': TYPE_MARKDOWN,
  '.json': TYPE_JSON,
  '.yaml': TYPE_YAML,
  '.yml': TYPE_YAML,
  '.toml': TYPE_TOML,
};

export const getFormat = async (
  transformContext: TransformContext,
  runtimeContext: RuntimeContext,
  config: Config,
) => {
  const directory = await getDirectory(transformContext, runtimeContext, config);
  let format = TYPE_YAML;
  if (typeof config.format === 'string') {
    format = config.format;
  } else if (typeof config.format === 'function') {
    format = await config.format(transformContext, runtimeContext, TYPE_YAML);
  } else {
    [format = TYPE_YAML] =
      Object.entries(config.format || {}).find(([, glob]) => contains(directory, glob)) || [];
  }

  return format;
};

const getDirectory = async (
  transformContext: TransformContext,
  runtimeContext: RuntimeContext,
  config: Config,
) => {
  const { contentTypeId } = transformContext;

  const contentTypeDirectory = await runtimeContext.hooks.mapDirectory(
    transformContext,
    contentTypeId,
  );
  return join(config.directory || process.cwd(), contentTypeDirectory);
};

export const getFilepath = async (
  transformContext: TransformContext,
  runtimeContext: RuntimeContext,
  config: Config,
) => {
  const { id, locale } = transformContext;

  const directory = await getDirectory(transformContext, runtimeContext, config);
  // Use CLDR codes for locales (http://www.localeplanet.com/icu/)
  const localeAddon = locale.default ? '' : `@${locale.code.replace('-', '_')}`;

  const format = await getFormat(transformContext, runtimeContext, config);

  const filename = await runtimeContext.hooks.mapFilename(
    transformContext,
    `${id}${localeAddon}.${format}`,
  );

  return join(directory, filename);
};

export const write = async (
  transformContext: TransformContext,
  runtimeContext: RuntimeContext,
  config: Config,
) => {
  const { content: data } = transformContext;

  const format = await getFormat(transformContext, runtimeContext, config);
  const filepath = await getFilepath(transformContext, runtimeContext, config);
  const ext = extname(filepath);
  // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
  const content = stringify(data, formatMapping?.[ext] ?? format);

  await runtimeContext.fileManager.writeFile(filepath, content);
};