samhuri.net/SiteGenerator/Sources/SiteGenerator/Projects/ProjectsPlugin.swift

//
//  ProjectsPlugin.swift
//  SiteGenerator
//
//  Created by Sami Samhuri on 2019-12-02.
//

import Foundation

private struct Projects: Codable {
    let projects: [Project]

    static func decode(from url: URL) throws -> Projects {
        let json = try Data(contentsOf: url)
        let projects = try JSONDecoder().decode(Projects.self, from: json)
        return projects
    }
}

final class ProjectsPlugin: Plugin {
    let fileManager: FileManager = .default
    let path: String

    var projects: [Project] = []
    var sourceURL: URL!

    init(path: String = "projects") {
        self.path = path
    }

    func setUp(sourceURL: URL) throws {
        self.sourceURL = sourceURL
        let projectsURL = sourceURL.appendingPathComponent("projects.json")
        if fileManager.fileExists(atPath: projectsURL.path) {
            self.projects = try Projects.decode(from: projectsURL).projects.map { project in
                Project(title: project.title, description: project.description, path: "/\(path)/\(project.title).html")
            }
        }
    }

    func render(targetURL: URL, delegate: PluginDelegate) throws {
        guard !projects.isEmpty else {
            return
        }

        let projectsDir = targetURL.appendingPathComponent(path)
        try fileManager.createDirectory(at: projectsDir, withIntermediateDirectories: true, attributes: nil)
        let projectsURL = projectsDir.appendingPathComponent("index.html")
        let projectsHTML = try delegate.renderTemplate(name: "projects", context: ["projects": projects])
        try projectsHTML.write(to: projectsURL, atomically: true, encoding: .utf8)

        for project in projects {
            let filename = "\(project.title).html"
            let projectURL = projectsDir.appendingPathComponent(filename)
            let projectHTML = try delegate.renderTemplate(name: "project", context: ["project": project])
            try projectHTML.write(to: projectURL, atomically: true, encoding: .utf8)
        }
    }
}