MapComplete/src/Logic/FeatureSource/Sources/LastClickFeatureSource.ts

import ThemeConfig from "../../../Models/ThemeConfig/ThemeConfig"
import { ImmutableStore, Store, UIEventSource } from "../../UIEventSource"
import { Feature, Point } from "geojson"
import { TagUtils } from "../../Tags/TagUtils"
import BaseUIElement from "../../../UI/BaseUIElement"
import { Utils } from "../../../Utils"
import { OsmTags } from "../../../Models/OsmFeature"
import { FeatureSource } from "../FeatureSource"
import { Lists } from "../../../Utils/Lists"

/**
 * Highly specialized feature source.
 * Based on a lon/lat UIEVentSource, will generate the corresponding feature with the correct properties
 */
export class LastClickFeatureSource implements FeatureSource {
    public readonly renderings: string[]
    private i: number = 0
    private readonly hasPresets: boolean
    private readonly hasNoteLayer: boolean
    public static readonly newPointElementId = "new_point_dialog"
    public readonly features: Store<Feature[]>
    private readonly _features: UIEventSource<Feature[]>
    private _usermode: UIEventSource<string>
    private _enabledAddMorePoints: UIEventSource<boolean>
    constructor(
        layout: ThemeConfig,
        clickSource: Store<{ lon: number; lat: number; mode: "left" | "right" | "middle" }>,
        usermode?: UIEventSource<string>,
        enabledAddMorePoints?: UIEventSource<boolean>
    ) {
        this._usermode = usermode
        this._enabledAddMorePoints = enabledAddMorePoints
        this.hasNoteLayer = layout.hasNoteLayer()
        this.hasPresets = layout.hasPresets()
        const allPresets: BaseUIElement[] = []
        for (const layer of layout.layers)
            for (let i = 0; i < (layer.presets ?? []).length; i++) {
                const preset = layer.presets[i]
                const tags = new ImmutableStore(TagUtils.KVtoProperties(preset.tags))
                const rendering = layer.mapRendering[0]
                if (!rendering) {
                    console.error("NO rendering for preset", layer.id)
                    continue
                }
                const { html } = rendering.RenderIcon(tags, {
                    noSize: true,
                    includeBadges: false,
                })
                allPresets.push(html)
            }

        this.renderings = Lists.dedup(
            allPresets.map((uiElem) =>
                Utils.runningFromConsole ? "" : uiElem.ConstructElement().innerHTML
            )
        )

        this._features = new UIEventSource<Feature[]>([])
        this.features = this._features
        clickSource.addCallbackAndRunD(({ lon, lat, mode }) => {
            this._features.setData([this.createFeature(lon, lat, mode)])
        })
    }

    public createFeature(
        lon: number,
        lat: number,
        mode?: "left" | "right" | "middle"
    ): Feature<Point, OsmTags> {
        const properties: OsmTags = {
            id: LastClickFeatureSource.newPointElementId + "_" + this.i,
            has_note_layer: this.hasNoteLayer ? "yes" : "no",
            has_presets: this.hasPresets ? "yes" : "no",
            renderings: this.renderings.join(""),
            number_of_presets: "" + this.renderings.length,
            first_preset: this.renderings[0],
            mouse_button: mode ?? "none",
            _usermode: this._usermode?.data,
            _addNewEnabled: this._enabledAddMorePoints?.data ?? true ? "yes" : "no",
        }
        this.i++

        return <Feature<Point, OsmTags>>{
            type: "Feature",
            properties,
            geometry: {
                type: "Point",
                coordinates: [lon, lat],
            },
        }
    }

    clear() {
        this._features.setData([])
    }
}