CHGenerator/Sources/Generator/Generators/SiteGenerator.swift

import Foundation

struct SiteGenerator {

    let templates: TemplateFactory

    init() throws {
        let templatesFolder = files.urlInContentFolder("templates")
        self.templates = try TemplateFactory(templateFolder: templatesFolder)
    }

    func generate(site: Element) {
        site.languages.forEach {
            generate(site: site, metadata: $0)
        }
    }

    private func generate(site: Element, metadata: Element.LocalizedMetadata) {
        let language = metadata.language
        let template = LocalizedSiteTemplate(
            factory: templates,
            language: language,
            site: site)

        // Generate sections
        let overviewGenerator = OverviewPageGenerator(factory: template)
        let pageGenerator = PageGenerator(factory: template)
        
        var elementsToProcess: [Element] = [site]
        while let element = elementsToProcess.popLast() {
            // Move recursively down to all pages
            elementsToProcess.append(contentsOf: element.elements)

            processAllFiles(for: element)

            if !element.elements.isEmpty {
                overviewGenerator.generate(section: element, language: language)
            } else {
                #warning("Determine previous and next pages (with relative links)")
                pageGenerator.generate(
                    page: element,
                    language: language,
                    nextPage: nil,
                    previousPage: nil)
            }
        }
    }

    private func processAllFiles(for element: Element) {
        element.requiredFiles.forEach(files.require)
        element.externalFiles.forEach(files.exclude)
        element.images.forEach {
            files.requireImage(
                source: $0.sourcePath,
                destination: $0.destinationPath,
                requiredBy: element.path,
                width: $0.desiredWidth,
                desiredHeight: $0.desiredHeight)
        }
    }
}