patchwork-mapconverter/src/enhancements/ImportComposer.ts

Executable wrapper for https://github.com/ChiefOfGxBxL/WC3MapTranslator

import { type DirectoryTree } from 'directory-tree'
import { LoggerFactory } from '../logging/LoggerFactory'
import path from 'path'
import { type Import, ImportType } from '../wc3maptranslator/data'
import { FileBlacklist } from './FileBlacklist'

const log = LoggerFactory.createLogger('ImportComposer')

const ImportComposer = {

  composeImportRegistry: function (importsFolderTree: DirectoryTree): Import[] {
    log.info('Composing war3map.imp file from', importsFolderTree.path, 'directory')

    const imports: Import[] = []
    const standardImportFolder = path.join(importsFolderTree.path, 'war3mapImported')
    const fileStack: Array<DirectoryTree<Record<string, unknown>>> = [importsFolderTree]

    while (fileStack.length > 0) {
      const file = fileStack.pop()
      if (file == null) break
      if (FileBlacklist.isDirectoryTreeBlacklisted(file)) continue

      if (file.type === 'directory') {
        const children = file.children

        if (children != null) {
          for (const child of children) {
            fileStack.push(child)
          }
        }
      } else {
        imports.push({
          path: path.relative(importsFolderTree.path, file.path),
          type: file.path.startsWith(standardImportFolder) ? ImportType.Standard : ImportType.Custom
        })
      }
    }

    return imports
  },

  getImportedFilePaths: function (inputPath: string, importRegistry: Import[]): string[] {
    log.info('Reading imported files registry')
    const importedFiles: string[] = []

    for (const entry of importRegistry) {
      importedFiles.push(path.join(inputPath, entry.path))
    }

    log.info('Found total of', importedFiles.length, 'imported files.')
    return importedFiles
  }
}

export default ImportComposer