import { FilesOutput } from '../../../../parsersEngine/index.js';

export type RenderBarrelFileOptions = {
  useSemicolon?: boolean;
  useFileExtensionsInImport?: boolean;
  quoteCharacter?: "'" | '"'; // single or double quote
  removePatternInOutputFilePath?: string;
  matchOutputFilePathPattern?: string;
  fileExtensionsMapping?: Record<string, string>;
};

const nativeEsmExtensionsMapping: Record<string, string> = {
  js: 'js',
  jsx: 'js',
  ts: 'js',
  tsx: 'js',
};

function getMappedExtension(
  extension: string,
  useFileExtensionsInImport: Exclude<
    RenderBarrelFileOptions['useFileExtensionsInImport'],
    undefined
  >,
  extensionMapping: Exclude<RenderBarrelFileOptions['fileExtensionsMapping'], undefined>,
) {
  if (!useFileExtensionsInImport) {
    return '';
  }
  return extensionMapping[extension] ?? extension;
}

export function renderBarrelFile(
  files: FilesOutput['files'],
  {
    useSemicolon = true,
    useFileExtensionsInImport = false,
    quoteCharacter = '"',
    removePatternInOutputFilePath = '',
    matchOutputFilePathPattern = '.(js|jsx|ts|tsx)$',
    fileExtensionsMapping = nativeEsmExtensionsMapping,
  }: RenderBarrelFileOptions,
) {
  const entries = files
    .filter(file => new RegExp(matchOutputFilePathPattern, 'g').test(file.path))
    .map(file => {
      const fileExtension = file.path.split('.').pop() ?? '';
      const fullExtensionLength = fileExtension ? (fileExtension as string).length + 1 : 0; // +1 for the dot character

      const extensionLessPath = file.path.slice(0, -fullExtensionLength);
      const mappedExtension = getMappedExtension(
        fileExtension,
        useFileExtensionsInImport,
        fileExtensionsMapping,
      );

      const path =
        mappedExtension.length > 0 ? `${extensionLessPath}.${mappedExtension}` : extensionLessPath;

      const withMask = removePatternInOutputFilePath
        ? path.replace(removePatternInOutputFilePath, '')
        : path;

      const pathWithPrefix =
        withMask.startsWith('./') || withMask.startsWith('/') ? withMask : `./${withMask}`;

      return `export * from ${quoteCharacter}${pathWithPrefix}${quoteCharacter}${useSemicolon ? ';' : ''}`;
    });

  return entries.length > 0 ? entries.join('\n') + '\n' : `export {}${useSemicolon ? ';' : ''}\n`;
}