MapComplete/scripts/ScriptUtils.ts

import * as fs from "fs"
import { existsSync, lstatSync, readdirSync, readFileSync } from "fs"
import { Utils } from "../src/Utils"
import { https } from "follow-redirects"
import { ThemeConfigJson } from "../src/Models/ThemeConfig/Json/ThemeConfigJson"
import { LayerConfigJson } from "../src/Models/ThemeConfig/Json/LayerConfigJson"
import xml2js from "xml2js"

export default class ScriptUtils {
    public static fixUtils() {
        Utils.externalDownloadFunction = ScriptUtils.Download
    }

    /**
     * Returns all files in a directory, recursively reads subdirectories.
     * The returned paths include the path given and subdirectories.
     *
     * @param path
     * @param maxDepth
     */
    public static readDirRecSync(path, maxDepth = 999): string[] {
        const result: string[] = []
        if (maxDepth <= 0) {
            return []
        }
        for (const entry of readdirSync(path)) {
            const fullEntry = path + "/" + entry
            const stats = lstatSync(fullEntry)
            if (stats.isDirectory()) {
                // Subdirectory
                // @ts-ignore
                result.push(...ScriptUtils.readDirRecSync(fullEntry, maxDepth - 1))
            } else {
                result.push(fullEntry)
            }
        }
        return result
    }

    public static DownloadFileTo(url, targetFilePath: string): Promise<void> {
        ScriptUtils.erasableLog("Downloading", url, "to", targetFilePath)
        return new Promise<void>((resolve) => {
            https.get(url, (res) => {
                const filePath = fs.createWriteStream(targetFilePath)
                res.pipe(filePath)
                filePath.on("finish", () => {
                    filePath.close()
                    resolve()
                })
            })
        })
    }

    public static erasableLog(...text) {
        process.stdout.write("\r " + text.join(" ") + "                \r")
    }

    public static sleep(ms: number, text?: string) {
        if (ms <= 0) {
            process.stdout.write("\r                                       \r")
            return
        }
        return new Promise((resolve) => {
            process.stdout.write("\r" + (text ?? "") + " Sleeping for " + ms / 1000 + "s \r")
            setTimeout(resolve, 1000)
        }).then(() => ScriptUtils.sleep(ms - 1000))
    }

    public static getLayerPaths(): string[] {
        return ScriptUtils.readDirRecSync("./assets/layers")
            .filter((path) => path.indexOf(".json") > 0)
            .filter((path) => path.indexOf(".proto.json") < 0)
            .filter((path) => path.indexOf("license_info.json") < 0)
    }

    public static getLayerFiles(): { parsed: LayerConfigJson; path: string }[] {
        return ScriptUtils.readDirRecSync("./assets/layers")
            .filter((path) => path.indexOf(".json") > 0)
            .filter((path) => path.indexOf(".proto.json") < 0)
            .filter((path) => path.indexOf("license_info.json") < 0)
            .map((path) => {
                try {
                    const contents = readFileSync(path, { encoding: "utf8" })
                    if (contents === "") {
                        throw "The file " + path + " is empty, did you properly save?"
                    }

                    const parsed = JSON.parse(contents)
                    return { parsed, path }
                } catch (e) {
                    console.error("Could not parse file ", path, "due to ", e)
                    throw e
                }
            })
    }

    public static getThemePaths(useTranslationPaths = false): string[] {
        const normalFiles = ScriptUtils.readDirRecSync("./assets/themes")
            .filter((path) => path.endsWith(".json") && !path.endsWith(".proto.json"))
            .filter((path) => path.indexOf("license_info.json") < 0)

        if (!useTranslationPaths) {
            return normalFiles
        }
        const specialfiles = ["./assets/themes/mapcomplete-changes/mapcomplete-changes.proto.json"]
        const blacklist = ["assets/themes/mapcomplete-changes/mapcomplete-changes.json"]

        const filtered = normalFiles.filter(
            (path) => !blacklist.some((black) => path.endsWith(black))
        )
        return filtered.concat(specialfiles)
    }

    public static getThemeFiles(useTranslationPaths = false): {
        parsed: ThemeConfigJson
        path: string
        raw: string
    }[] {
        return this.getThemePaths(useTranslationPaths).map((path) => {
            try {
                const contents = readFileSync(path, { encoding: "utf8" })
                if (contents === "") {
                    throw "The file " + path + " is empty, did you properly save?"
                }
                const parsed = JSON.parse(contents)
                return { parsed: parsed, path: path, raw: contents }
            } catch (e) {
                console.error("Could not read file ", path, "due to ", e)
                throw e
            }
        })
    }

    public static TagInfoHistogram(key: string): Promise<{
        data: { count: number; value: string; fraction: number }[]
    }> {
        const url = `https://taginfo.openstreetmap.org/api/4/key/values?key=${key}&filter=all&lang=en&sortname=count&sortorder=desc&page=1&rp=17&qtype=value`
        return ScriptUtils.DownloadJSON(url)
    }

    public static async ReadSvg(path: string): Promise<SVGElement> {
        if (!existsSync(path)) {
            throw "File not found: " + path
        }
        const root = await xml2js.parseStringPromise(readFileSync(path, { encoding: "utf8" }))
        return root.svg
    }

    public static ReadSvgSync(path: string, callback: (svg: any) => void): any {
        xml2js.parseString(
            readFileSync(path, { encoding: "utf8" }),
            { async: false },
            (err, root) => {
                if (err) {
                    throw err
                }
                callback(root["svg"])
            }
        )
    }

    private static async DownloadJSON(url: string, headers?: any): Promise<any> {
        const data = await ScriptUtils.Download(url, headers)
        return JSON.parse(data["content"])
    }

    public static async DownloadFetch(
        url: string,
        headers?: any
    ): Promise<{ content: string } | { redirect: string }> {
        console.log("Fetching", url)
        const req = await fetch(url, { headers })
        const data = await req.text()
        console.log("Fetched", url, data)
        return { content: data }
    }

    public static Download(
        url: string,
        headers?: any
    ): Promise<{ content: string } | { redirect: string }>
    public static Download(
        url: string,
        headers?: any,
        timeoutSecs?: number
    ): Promise<{ content: string } | { redirect: string } | "timeout">
    public static Download(
        url: string,
        headers?: any,
        timeoutSecs?: number
    ): Promise<{ content: string } | { redirect: string } | "timeout"> {
        if (url.startsWith("./assets")) {
            return Promise.resolve({ content: readFileSync("./public/" + url, "utf8") })
        }
        if (url.startsWith("./")) {
            return Promise.resolve({ content: readFileSync(url, "utf8") })
        }

        const requestPromise = new Promise((resolve, reject) => {
            try {
                headers = headers ?? {}
                if (!headers.Accept) {
                    headers.accept ??= "application/json"
                }
                ScriptUtils.erasableLog(" > ScriptUtils.Download(", url, ")")
                const urlObj = new URL(url)
                const request = https.get(
                    {
                        host: urlObj.host,
                        path: urlObj.pathname + urlObj.search,

                        port: urlObj.port,
                        headers: headers,
                    },
                    (res) => {
                        const parts: string[] = []
                        res.setEncoding("utf8")
                        res.on("data", function (chunk) {
                            // @ts-ignore
                            parts.push(chunk)
                        })

                        res.addListener("end", function () {
                            if (res.statusCode === 301 || res.statusCode === 302) {
                                console.log("Got a redirect:", res.headers.location)
                                resolve({ redirect: res.headers.location })
                            }
                            if (res.statusCode >= 400) {
                                console.log(
                                    "Error while fetching ",
                                    url,
                                    "due to",
                                    res.statusMessage
                                )
                                reject(res.statusCode)
                            }
                            resolve({ content: parts.join("") })
                        })
                    }
                )
                request.on("error", function (e) {
                    reject(e)
                })
            } catch (e) {
                reject(e)
            }
        })
        const timeoutPromise = new Promise<any>((resolve, reject) => {
            setTimeout(() => {
                if (timeoutSecs === undefined) {
                    return // No resolve
                }
                resolve("timeout")
            }, (timeoutSecs ?? 10) * 1000)
        })
        return Promise.race([requestPromise, timeoutPromise])
    }
}