MapComplete/scripts/fixTheme.ts

/*
 * This script attempt to automatically fix some basic issues when a theme from the custom generator is loaded
 */
import {Utils} from "../Utils"
import {readFileSync, writeFileSync} from "fs";
import SmallLicense from "../Models/smallLicense";
import ScriptUtils from "./ScriptUtils";
import AllImageProviders from "../Logic/ImageProviders/AllImageProviders";
import {LayoutConfigJson} from "../Models/ThemeConfig/Json/LayoutConfigJson";
import LayerConfig from "../Models/ThemeConfig/LayerConfig";

ScriptUtils.fixUtils()

if (process.argv.length == 2) {
    console.log("USAGE: ts-node scripts/fixTheme <path to theme>")
    throw "No path specified"
}

const path = process.argv[2]
const dir = path.substring(0, path.lastIndexOf("/"))

console.log("Fixing up ", path)

const themeConfigJson: LayoutConfigJson = JSON.parse(readFileSync(path, "UTF8"))

const licenses: SmallLicense[] = []

const replacements: { source: string, destination: string }[] = []

for (const layerConfigJson of themeConfigJson.layers) {
    if (typeof (layerConfigJson) === "string") {
        continue;
    }
    if (layerConfigJson["overpassTags"] !== undefined) {
        const tags = layerConfigJson["overpassTags"];
        layerConfigJson["overpassTags"] = undefined;
        layerConfigJson["source"] = {osmTags: tags}
    }
    // @ts-ignore
    const layerConfig = new LayerConfig(layerConfigJson, "fix theme", true)
    const images: string[] = Array.from(layerConfig.ExtractImages())
    const remoteImages = images.filter(img => img.startsWith("http"))

    for (const remoteImage of remoteImages) {


        const filename = remoteImage.substring(remoteImage.lastIndexOf("/"))
        ScriptUtils.DownloadFileTo(remoteImage, dir + "/" + filename)


        const imgPath = remoteImage.substring(remoteImage.lastIndexOf("/") + 1)

        for (const attributionSrc of AllImageProviders.ImageAttributionSource) {
            try {
                attributionSrc.GetAttributionFor(remoteImage).addCallbackAndRun(license => {
                    console.log("Downloaded an attribution!")
                    licenses.push({
                        path: imgPath,
                        license: license?.license ?? "",
                        authors: Utils.NoNull([license?.artist]),
                        sources: [remoteImage]
                    })
                })
            } catch (e) {
                // Hush hush
            }
        }

        replacements.push({source: remoteImage, destination: `${dir}/${imgPath}`})
    }
}

let fixedThemeJson = JSON.stringify(themeConfigJson, null, "  ")
for (const replacement of replacements) {
    fixedThemeJson = fixedThemeJson.replace(new RegExp(replacement.source, "g"), replacement.destination)
}

writeFileSync(dir + "/generated.license_info.json", JSON.stringify(licenses, null, "  "))
writeFileSync(path + ".autofixed.json", fixedThemeJson)

console.log(`IMPORTANT:
 1) Copy generated.license_info.json over into license_info.json and add the missing attributions and authors
 2) Verify ${path}.autofixed.json as theme, and rename it to ${path}`)