mapcomplete/scripts/ImgurToPanoramax.ts

import Script from "./Script"
import { Overpass } from "../src/Logic/Osm/Overpass"
import { RegexTag } from "../src/Logic/Tags/RegexTag"
import Constants from "../src/Models/Constants"
import { BBox } from "../src/Logic/BBox"
import { existsSync, readFileSync, writeFileSync } from "fs"
import PanoramaxImageProvider, { PanoramaxUploader } from "../src/Logic/ImageProviders/Panoramax"
import { Feature } from "geojson"
import { LicenseInfo } from "../src/Logic/ImageProviders/LicenseInfo"
import { GeoOperations } from "../src/Logic/GeoOperations"
import { Tag } from "../src/Logic/Tags/Tag"
import { Utils } from "../src/Utils"
import ChangeTagAction from "../src/Logic/Osm/Actions/ChangeTagAction"
import { And } from "../src/Logic/Tags/And"
import { Changes } from "../src/Logic/Osm/Changes"
import { ChangeDescription } from "../src/Logic/Osm/Actions/ChangeDescription"
import OsmObjectDownloader from "../src/Logic/Osm/OsmObjectDownloader"
import { OsmObject } from "../src/Logic/Osm/OsmObject"
import { File } from "buffer"
import { open } from "node:fs/promises"
import { UploadableTag } from "../src/Logic/Tags/TagTypes"
import { Imgur } from "../src/Logic/ImageProviders/Imgur"
import { Or } from "../src/Logic/Tags/Or"
import ScriptUtils from "./ScriptUtils"
import { ImmutableStore } from "../src/Logic/UIEventSource"

export class ImgurToPanoramax extends Script {
    private readonly panoramax = new PanoramaxUploader(
        Constants.panoramax.url,
        Constants.panoramax.token
    )
    private licenseChecker = new PanoramaxImageProvider()

    private readonly alreadyUploaded: Record<string, string> = this.readAlreadyUploaded()
    private readonly alreadyUploadedInv: Record<string, string> = Utils.transposeMapSimple(this.alreadyUploaded)
    private _imageDirectory: string
    private _licenseDirectory: string

    private readonly sequenceIds = {
        test: "7f34cf53-27ff-46c9-ac22-78511fa8457a",
        cc0: "1de6f4a1-73ac-4c75-ab7f-2a2aabddf50a", // "f0d6f78a-ff95-4db1-8494-6eb44a17bb37",
        ccby: "288a8052-b475-422c-811a-4f6f1a00015e",
        ccbysa: "f3d02893-b4c1-4cd6-8b27-e27ab57eb59a"
    } as const

    constructor() {
        super(
            "Queries OSM for 'imgur'-images, uploads them to Panoramax and creates a changeset to update OSM"
        )
    }


    private async getRawInfo(imgurUrl): Promise<{ description?: string, datetime: number }> {
        const fallbackpath = this._licenseDirectory + "/raw/" + imgurUrl.replaceAll(/[^a-zA-Z0-9]/g, "_") + ".json"
        if (existsSync(fallbackpath)) {
            console.log("Loaded raw info from fallback path")
            return JSON.parse(readFileSync(fallbackpath, "utf8"))["data"]
        }
        // No local data available; lets ask imgur themselves
        return new Promise((resolve) => {
            Imgur.singleton.DownloadAttribution({ url: imgurUrl },
                raw => {
                    console.log("Writing fallback to", fallbackpath, "(via raw)")
                    writeFileSync(fallbackpath, JSON.stringify(raw), "utf8")
                    resolve(raw["data"])
                })
        })
    }

    private async getLicenseFor(imgurUrl: string): Promise<LicenseInfo> {
        const imageName = imgurUrl.split("/").at(-1)
        const licensePath: string = this._licenseDirectory + "/" + imageName
        if (existsSync(licensePath)) {
            const rawText = readFileSync(licensePath, "utf8")
            if (rawText?.toLowerCase() === "cc0" || rawText?.toLowerCase().startsWith("cc0")) {
                return { licenseShortName: "CC0", artist: "Unknown" }
            }
            try {

                const licenseText: LicenseInfo = JSON.parse(rawText)
                if (licenseText.licenseShortName) {
                    return licenseText
                }
                console.log("<<< No valid license found in text", rawText)
                return undefined
            } catch (e) {
                console.error("Could not read ", rawText.slice(0, 20), "as json for image", imgurUrl, "from", licensePath)
            }
        }


        // We didn't find the expected license in the expected location; search for the fallback (raw) license
        const fallbackpath = this._licenseDirectory + "/raw/" + imgurUrl.replaceAll(/[^a-zA-Z0-9]/g, "_") + ".json"
        if (existsSync(fallbackpath)) {
            const fallbackRaw: string = JSON.parse(readFileSync(fallbackpath, "utf8"))["data"]?.description
            if (fallbackRaw?.toLowerCase()?.startsWith("cc0") || fallbackRaw?.toLowerCase()?.indexOf("#cc0") >= 0) {
                return { licenseShortName: "CC0", artist: "Unknown" }
            }
           const license = Imgur.parseLicense(fallbackRaw)
            if(license){
                return license
            }
            console.log("No (fallback) license found for (but file exists), not uploading", imgurUrl, fallbackRaw)
            return undefined
        }


        // No local data available; lets ask imgur themselves
        const attr = await Imgur.singleton.DownloadAttribution({ url: imgurUrl },
            raw => {
                console.log("Writing fallback to", fallbackpath)
                writeFileSync(fallbackpath, JSON.stringify(raw), "utf8")
            })
        console.log("Got license via API:", attr.licenseShortName)
        await ScriptUtils.sleep(500)
        if (attr.licenseShortName) {
            return attr
        }
        return undefined
    }

    async uploadImage(
        key: string,
        feat: Feature
    ): Promise<UploadableTag | undefined> {
        const v = feat.properties[key]
        if (!v) {
            return undefined
        }

        const imageHash = v.split("/").at(-1).split(".").at(0)
        {
            const panohash = this.alreadyUploaded[imageHash]
            if (panohash) {
                console.log("Already uploaded", panohash)
                return new And([new Tag(key.replace("image", "panoramax"), panohash), new Tag(key, "")])
            }
        }

        let path: string = undefined
        if (existsSync(this._imageDirectory + "/" + imageHash + ".jpg")) {
            path = this._imageDirectory + "/" + imageHash + ".jpg"
        } else if (existsSync(this._imageDirectory + "/" + imageHash + ".jpeg")) {
            path = this._imageDirectory + "/" + imageHash + ".jpeg"
        }
        if (!path) {
            return undefined
        }
        const license: LicenseInfo = await this.getLicenseFor(v)
        if (license === undefined) {
            return undefined
        }
        const sequence = this.sequenceIds[license.licenseShortName?.toLowerCase()]
        const handle = await open(path)
        const stat = await handle.stat()

        class MyFile extends File {
            // we should set correct size
            // otherwise we will encounter UND_ERR_REQ_CONTENT_LENGTH_MISMATCH
            size = stat.size
            stream = undefined
        }

        const file = new MyFile([], path)

        file.stream = function() {
            return handle.readableWebStream()
        }

        const licenseRaw = await this.getRawInfo(v)
        const date = new Date(licenseRaw.datetime * 1000)

        console.log("Uploading", imageHash, sequence)
        const result = await this.panoramax.uploadImage(
            <any>file,
            GeoOperations.centerpointCoordinates(feat),
            license.artist,
            true,
            sequence,
            date.toISOString()
        )
        await handle.close()
        this.alreadyUploaded[imageHash] = result.value
        this.writeAlreadyUploaded()
        return new And([new Tag(key.replace("image", result.key), result.value), new Tag(key, "")])
    }

    private writeAlreadyUploaded() {
        writeFileSync("uploaded_images.json", JSON.stringify(this.alreadyUploaded))
    }

    private readAlreadyUploaded() {
        const uploaded = JSON.parse(readFileSync("uploaded_images.json", "utf8"))
        console.log("Detected ", Object.keys(uploaded).length, "previously uploaded images")
        return uploaded
    }

    private async patchDate(panokey: string) {
        const imgurkey = this.alreadyUploadedInv[panokey]
        const license = await this.getRawInfo("https://i.imgur.com/" + imgurkey + ".jpg")
        const date = new Date(license.datetime * 1000)
        const panolicense = await this.panoramax.panoramax.search({
            ids: [panokey]
        })
        const panodata = panolicense[0]
        const collection: string = panodata.collection
        console.log({ imgurkey, date, panodata, datetime: license.datetime })
        const p = this.panoramax.panoramax
        const url = p.host+"/collections/" + collection + "/items/" + panokey
        const result = await p.fetch(url, {
            method: "PATCH",
            headers: { "content-type": "application/json" },
            body: JSON.stringify({
                ts: date.getTime(),
            })
        })
        console.log("Patched date of ", p.createViewLink({
            imageId: panokey,
        }), url, "result is", result.status, await result.text())
    }


    async main(args: string[]): Promise<void> {
        this._imageDirectory = args[0] ?? "/home/pietervdvn/data/imgur-image-backup"
        this._licenseDirectory = args[1] ?? "/home/pietervdvn/git/MapComplete-data/ImageLicenseInfo"

      //  await this.panoramax.panoramax.createCollection("CC0 - part 2")
      //  return
      /*  for (const panohash in this.alreadyUploadedInv) {
            await this.patchDate(panohash)
            break
        }*/



        const bounds = new BBox([
            [
                1.7217767788980893,
                41.00219164121438
            ],
            [
                2.7238939148245436,
                41.9258679932085
            ]
        ])
        const maxcount = 10000
        const overpassfilters: RegexTag[] = []
        const r = /^https:\/\/i.imgur.com\/.*/
        for (const k of ["image", "image:menu", "image:streetsign"]) {
            overpassfilters.push(new RegexTag(k, r))
            for (let i = 0; i < 20; i++) {
                overpassfilters.push(new RegexTag(k + ":" + i, r))
            }
        }
        const overpass = new Overpass(new Or(overpassfilters), [], Constants.defaultOverpassUrls[0], new ImmutableStore(500) )
        const features = (await overpass.queryGeoJson(bounds))[0].features
        const featuresCopy = [...features]
        let converted = 0

        const total = features.length
        const changes: ChangeDescription[] = []

        do {
            const f = features.shift()
            if (!f) {
                break
            }
            if (converted % 100 === 0) {
                console.log("Converted:", converted, "total:", total, "progress:", Math.round(converted * 100 / total) + "%")
            }

            let changedTags: (UploadableTag | undefined)[] = []
            console.log("Handling "+f.properties.id)
            for (const k of ["image", "image:menu", "image:streetsign"]) {
                changedTags.push(await this.uploadImage(k, f))
                for (let i = 0; i < 20; i++) {
                    changedTags.push(await this.uploadImage(k + ":" + i, f))
                }
            }
            changedTags = Utils.NoNull(changedTags)
            if (changedTags.length > 0) {
                const action = new ChangeTagAction(
                    f.properties.id,
                    new And(changedTags),
                    f.properties,
                    {
                        theme: "image-mover",
                        changeType: "link-image"
                    }
                )
                changes.push(...(await action.CreateChangeDescriptions()))
            }
            converted++
        } while (converted < maxcount)

        console.log("Uploaded images for", converted, "items; now creating the changeset")

        const modif: string[] = Utils.Dedup(changes.map((ch) => ch.type + "/" + ch.id))
        const modifiedObjectsFresh: OsmObject[] = []
        const dloader = new OsmObjectDownloader()
        for (let i = 0; i < modif.length; i++) {
            if (i % 100 === 0) {
                console.log("Downloaded osm object", i, "/", modif.length, "(" + Math.round(i * 100 / modif.length) + "%)")
            }
            const id = modif[i]
            const obj = await dloader.DownloadObjectAsync(id)
            if (obj === "deleted") {
                continue
            }
            modifiedObjectsFresh.push(obj)
        }
        const modifiedObjects = Changes.createChangesetObjectsStatic(
            changes,
            modifiedObjectsFresh,
            false,
            []
        )
        const cs = Changes.buildChangesetXML("0", modifiedObjects)
        writeFileSync("imgur_to_panoramax.osc", cs, "utf8")


        const usernames = featuresCopy.map(f => f.properties.user)
        const hist : Record<string,number> = {}
        for (const username of usernames) {
            hist[username] = (hist[username] ?? 0)+ 1
        }
        console.log(hist)
    }
}

new ImgurToPanoramax().run()