CHGenerator/WebsiteGenerator/Generators/MarkdownProcessor.swift

import Foundation
import Ink
import Splash

struct PageContentGenerator {

    private let factory: TemplateFactory

    private let swift = SyntaxHighlighter(format: HTMLOutputFormat())

    init(factory: TemplateFactory) {
        self.factory = factory
    }

    func generate(page: Element, language: String, content: String) -> (content: String, includesCode: Bool) {
        var hasCodeContent = false

        let imageModifier = Modifier(target: .images) { html, markdown in
            processMarkdownImage(markdown: markdown, html: html, page: page)
        }
        let codeModifier = Modifier(target: .codeBlocks) { html, markdown in
            if markdown.starts(with: "```swift") {
                let code = markdown.between("```swift", and: "```").trimmed
                return "<pre><code>" + swift.highlight(code) + "</pre></code>"
            }
            hasCodeContent = true
            return html
        }
        let linkModifier = Modifier(target: .links) { html, markdown in
            handleLink(page: page, language: language, html: html, markdown: markdown)
        }
        let htmlModifier = Modifier(target: .html) { html, markdown in
            handleHTML(page: page, language: language, html: html, markdown: markdown)
        }

        let parser = MarkdownParser(modifiers: [imageModifier, codeModifier, linkModifier, htmlModifier])
        return (parser.html(from: content), hasCodeContent)
    }

    private func handleLink(page: Element, language: String, html: String, markdown: Substring) -> String {
        let file = markdown.between("(", and: ")")
        if file.hasPrefix("page:") {
            let pageId = file.replacingOccurrences(of: "page:", with: "")
            guard let pagePath = files.getPage(for: pageId) else {
                log.add(warning: "Page id '\(pageId)' not found", source: page.path)
                // Remove link since the page can't be found
                return markdown.between("[", and: "]")
            }
            let fullPath = pagePath + Element.htmlPagePathAddition(for: language)
            // Adjust file path to get the page url
            let url = page.relativePathToOtherSiteElement(file: fullPath)
            return html.replacingOccurrences(of: file, with: url)
        }

        if let filePath = page.nonAbsolutePathRelativeToRootForContainedInputFile(file) {
            // The target of the page link must be present after generation is complete
            files.expect(file: filePath, source: page.path)
        }
        return html
    }

    private func handleHTML(page: Element, language: String, html: String, markdown: Substring) -> String {
        #warning("Check HTML code in markdown for required resources")
        //print("[HTML] Found in page \(page.path):")
        //print(markdown)
        // Things to check:
        // <img src=
        // <a href=
        //
        return html
    }

    private func processMarkdownImage(markdown: Substring, html: String, page: Element) -> String {
        // Split the markdown ![alt](file title)
        // For images: ![left_title](file right_title)
        // For videos: ![option1,option2,...](file)
        // For svg with custom area: ![x,y,width,height](file.svg)
        // For downloads: ![download](file1, text1; file2, text2, ...)
        // External pages: ![external](url1, text1; url2, text2, ...)
        let fileAndTitle = markdown.between("(", and: ")")
        let alt = markdown.between("[", and: "]").nonEmpty
        if alt == "download" {
            return handleDownloadButtons(page: page, content: fileAndTitle)
        }
        if alt == "external" {
            return handleExternalButtons(page: page, content: fileAndTitle)
        }

        let file = fileAndTitle.dropAfterFirst(" ")
        let title = fileAndTitle.contains(" ") ? fileAndTitle.dropBeforeFirst(" ").nonEmpty : nil

        let fileExtension = file.lastComponentAfter(".").lowercased()
        if let _ = ImageType(fileExtension: fileExtension) {
            return handleImage(page: page, file: file, rightTitle: title, leftTitle: alt)
        }
        if let _ = VideoType(rawValue: fileExtension) {
            return handleVideo(page: page, file: file, optionString: alt)
        }
        if fileExtension == "svg" {
            return handleSvg(page: page, file: file, area: alt)
        }
        return handleFile(page: page, file: file, fileExtension: fileExtension)
    }

    private func handleImage(page: Element, file: String, rightTitle: String?, leftTitle: String?) -> String {
        let imagePath = page.pathRelativeToRootForContainedInputFile(file)

        let size = files.requireImage(source: imagePath, destination: imagePath, width: configuration.pageImageWidth)

        let imagePath2x = imagePath.insert("@2x", beforeLast: ".")
        let file2x = file.insert("@2x", beforeLast: ".")
        files.requireImage(source: imagePath, destination: imagePath2x, width: 2 * configuration.pageImageWidth)

        let content: [PageImageTemplate.Key : String] = [
            .image: file,
            .image2x: file2x,
            .width: "\(Int(size.width))",
            .height: "\(Int(size.height))",
            .leftText: leftTitle ?? "",
            .rightText: rightTitle ?? ""]
        return factory.image.generate(content)
    }

    private func handleVideo(page: Element, file: String, optionString: String?) -> String {
        let options: [PageVideoTemplate.VideoOption] = optionString.unwrapped { string in
            string.components(separatedBy: " ").compactMap { optionText in
                guard let optionText = optionText.trimmed.nonEmpty else {
                    return nil
                }
                guard let option = PageVideoTemplate.VideoOption(rawValue: optionText) else {
                    log.add(warning: "Unknown video option \(optionText)", source: page.path)
                    return nil
                }
                return option
            }
        } ?? []
        #warning("Check page folder for alternative video versions")
        let sources: [PageVideoTemplate.VideoSource] = [(url: file, type: .mp4)]

        let filePath = page.pathRelativeToRootForContainedInputFile(file)
        files.require(file: filePath)
        return factory.video.generate(sources: sources, options: options)
    }

    private func handleSvg(page: Element, file: String, area: String?) -> String {
        let imagePath = page.pathRelativeToRootForContainedInputFile(file)
        files.require(file: imagePath)

        guard let area = area else {
            return factory.html.svgImage(file: file)
        }
        let parts = area.components(separatedBy: ",").map { $0.trimmed }
        guard parts.count == 4,
              let x = Int(parts[0].trimmed),
              let y = Int(parts[1].trimmed),
              let width = Int(parts[2].trimmed),
              let height = Int(parts[3].trimmed) else {
            log.add(warning: "Invalid area string for svg image", source: page.path)
            return factory.html.svgImage(file: file)
        }

        return factory.html.svgImage(file: file, x: x, y: y, width: width, height: height)
    }

    private func handleFile(page: Element, file: String, fileExtension: String) -> String {
        log.add(warning: "Unhandled file \(file) with extension \(fileExtension)", source: page.path)
        return ""
    }

    private func handleDownloadButtons(page: Element, content: String) -> String {
        let buttons = content
            .components(separatedBy: ";")
            .compactMap { button -> (file: String, text: String, downloadName: String?)? in
                let parts = button.components(separatedBy: ",")
                guard parts.count == 2 || parts.count == 3 else {
                    log.add(warning: "Invalid button definition", source: page.path)
                    return nil
                }
                let file = parts[0].trimmed
                let title = parts[1].trimmed
                let downloadName = parts.count == 3 ? parts[2].trimmed : nil

                // Ensure that file is available
                let filePath = page.pathRelativeToRootForContainedInputFile(file)
                files.require(file: filePath)

                return (file, title, downloadName)
            }
        return factory.html.downloadButtons(buttons)
    }

    private func handleExternalButtons(page: Element, content: String) -> String {
        let buttons = content
            .components(separatedBy: ";")
            .compactMap { button -> (url: String, text: String)? in
                let parts = button.components(separatedBy: ",")
                guard parts.count == 2 else {
                    log.add(warning: "Invalid external link definition", source: page.path)
                    return nil
                }
                let url = parts[0].trimmed
                let title = parts[1].trimmed

                return (url, title)
            }
        return factory.html.externalButtons(buttons)
    }
}