import { mkdir } from 'fs/promises'
import {
    GirModule,
    Logger,
    START_MODULE,
    FILE_PARSING_DONE,
    TSDATA_PARSING_DONE,
    GENERATING_TYPES_DONE,
    ERROR_NO_MODULE_SPECIFIED,
} from '@ts-for-gir/lib'
import { GeneratorType, Generator } from '@ts-for-gir/generator-base'
import { TypeDefinitionGenerator } from '@ts-for-gir/generator-typescript'
import { HtmlDocGenerator } from '@ts-for-gir/generator-html-doc'

import type { OptionsGeneration, NSRegistry } from '@ts-for-gir/lib'

export class GenerationHandler {
    log: Logger
    generator: Generator

    constructor(
        private readonly config: OptionsGeneration,
        type: GeneratorType,
    ) {
        this.log = new Logger(config.verbose, 'GenerationHandler')

        switch (type) {
            case GeneratorType.TYPES:
                this.generator = new TypeDefinitionGenerator(config)
                break
            case GeneratorType.HTML_DOC:
                this.generator = new HtmlDocGenerator(config)
                break
            default:
                throw new Error('Unknown Generator')
        }
    }

    public async start(girModules: GirModule[], registry: NSRegistry): Promise<void> {
        this.log.info(START_MODULE)

        if (girModules.length == 0) {
            this.log.error(ERROR_NO_MODULE_SPECIFIED)
        }

        this.log.info(FILE_PARSING_DONE)

        this.log.info(TSDATA_PARSING_DONE)

        if (this.config.outdir) {
            await mkdir(this.config.outdir, { recursive: true })
        }

        // TODO: Put this somewhere that makes sense
        registry.transform({
            inferGenerics: true,
            verbose: this.config.verbose,
        })

        await this.generator.start(registry)

        for (const girModule of girModules) {
            this.log.log(` - ${girModule.packageName} ...`)
            await this.generator.generate(registry, girModule)
        }

        await this.generator.finish(registry, girModules)

        this.log.success(GENERATING_TYPES_DONE)
    }
}