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, language: language) } let codeModifier = Modifier(target: .codeBlocks) { html, markdown in if markdown.starts(with: "```swift") { let code = markdown.between("```swift", and: "```").trimmed return "
" + swift.highlight(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: // String { // Split the markdown ![alt](file title) // There are several known shorthand commands // 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, ...) // For a simple boxes: ![box](title;body) // A fancy page link: ![page](page_id) // External pages: ![external](url1, text1; url2, text2, ...) let fileAndTitle = markdown.between("(", and: ")") let alt = markdown.between("[", and: "]").nonEmpty if let alt = alt, let command = ShorthandMarkdownKey(rawValue: alt) { return handleShortHandCommand(command, page: page, language: language, 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 handleShortHandCommand(_ command: ShorthandMarkdownKey, page: Element, language: String, content: String) -> String { switch command { case .downloadButtons: return handleDownloadButtons(page: page, content: content) case .externalLink: return handleExternalButtons(page: page, content: content) case .includedHtml: return handleExternalHTML(page: page, file: content) case .box: return handleSimpleBox(page: page, content: content) case .pageLink: return handlePageLink(page: page, language: language, pageId: content) } } private func handleImage(page: Element, file: String, rightTitle: String?, leftTitle: String?) -> String { let imagePath = page.pathRelativeToRootForContainedInputFile(file) let size = files.requireMultiVersionImage( source: imagePath, destination: imagePath, requiredBy: page.path, width: configuration.pageImageWidth) let content: [PageImageTemplate.Key : String] = [ .image: file.dropAfterLast("."), .imageExtension: file.lastComponentAfter("."), .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) } private func handleExternalHTML(page: Element, file: String) -> String { let url = page.inputFolder.appendingPathComponent(file) guard url.exists else { log.add(error: "File \(file) not found", source: page.path) return "" } do { return try String(contentsOf: url) } catch { log.add(error: "File \(file) could not be read", source: page.path, error: error) return "" } } private func handleSimpleBox(page: Element, content: String) -> String { let parts = content.components(separatedBy: ";") guard parts.count > 1 else { log.add(error: "Invalid box specification", source: page.path) return "" } let title = parts[0] let text = parts.dropFirst().joined(separator: ";") return factory.makePlaceholder(title: title, text: text) } private func handlePageLink(page: Element, language: String, pageId: String) -> String { guard let linkedPage = siteRoot.find(pageId) else { log.add(warning: "Page id '\(pageId)' not found", source: page.path) // Remove link since the page can't be found return "" } var content = [PageLinkTemplate.Key: String]() content[.title] = linkedPage.title(for: language) let fullThumbnailPath = linkedPage.thumbnailFilePath(for: language).destination let relativeImageUrl = page.relativePathToOtherSiteElement(file: fullThumbnailPath) let metadata = linkedPage.localized(for: language) if linkedPage.state.hasThumbnailLink { let fullPageUrl = linkedPage.fullPageUrl(for: language) let relativePageUrl = page.relativePathToOtherSiteElement(file: fullPageUrl) content[.url] = "href=\"\(relativePageUrl)\"" } content[.image] = relativeImageUrl.dropAfterLast(".") if let suffix = metadata.thumbnailSuffix { content[.title] = factory.html.make(title: metadata.title, suffix: suffix) } else { content[.title] = metadata.title } let path = linkedPage.makePath(language: language, from: siteRoot) content[.path] = factory.pageLink.makePath(components: path) content[.description] = metadata.relatedContentText if let parent = linkedPage.findParent(from: siteRoot), parent.thumbnailStyle == .large { content[.className] = " related-page-link-large" } // We assume that the thumbnail images are already required by overview pages. return factory.pageLink.generate(content) } }