forked from MapComplete/MapComplete
376 lines
14 KiB
TypeScript
376 lines
14 KiB
TypeScript
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: "e9bcb8c0-8ade-4ac9-bc9f-cfa464221fd6", // "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 isPng = v.endsWith(".png")
|
|
|
|
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 (isPng) {
|
|
path = this._imageDirectory + "/../imgur_png_images/jpg/" + imageHash + ".jpg"
|
|
} else 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
|
|
}
|
|
let license: LicenseInfo
|
|
try {
|
|
license = await this.getLicenseFor(v)
|
|
} catch (e) {
|
|
console.error("Could not fetch license due to", e)
|
|
if (e === 404) {
|
|
console.log("NOT FOUND")
|
|
return new Tag(key, "")
|
|
}
|
|
throw e
|
|
}
|
|
if (license === undefined) {
|
|
return undefined
|
|
}
|
|
const sequence = this.sequenceIds[license.licenseShortName?.toLowerCase()]
|
|
console.log("Reading ", path)
|
|
if (!existsSync(path)) {
|
|
return undefined
|
|
}
|
|
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([
|
|
[-180, -90],
|
|
[180, 90],
|
|
])
|
|
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(converted + "/" + total, " 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()
|