MapComplete/src/Logic/Geocoding/ThemeSearch.ts

import GeocodingProvider, { SearchResult, GeocodingOptions } from "./GeocodingProvider"
import * as themeOverview from "../../assets/generated/theme_overview.json"
import { MinimalLayoutInformation } from "../../Models/ThemeConfig/LayoutConfig"
import { SpecialVisualizationState } from "../../UI/SpecialVisualization"
import { Utils } from "../../Utils"
import MoreScreen from "../../UI/BigComponents/MoreScreen"
import { ImmutableStore, Store } from "../UIEventSource"

export default class ThemeSearch implements GeocodingProvider {

    private static allThemes: MinimalLayoutInformation[] = (themeOverview["default"] ?? themeOverview)
    private readonly _state: SpecialVisualizationState
    private readonly _knownHiddenThemes: Store<Set<string>>
    private readonly _suggestionLimit: number

    constructor(state: SpecialVisualizationState, suggestionLimit: number) {
        this._state = state
        this._suggestionLimit = suggestionLimit
        this._knownHiddenThemes = MoreScreen.knownHiddenThemes(this._state.osmConnection)
    }

    async search(query: string): Promise<SearchResult[]> {
        return this.searchWrapped(query, 99)
    }

    suggest(query: string, options?: GeocodingOptions): Store<SearchResult[]> {
        return new ImmutableStore(this.searchWrapped(query, this._suggestionLimit ?? 4))
    }


    private searchWrapped(query: string, limit: number): SearchResult[] {
        return this.searchDirect(query, limit).map(match => <SearchResult>{
            payload: match,
            category: "theme",
            osm_id: match.id
        })
    }

    public searchDirect(query: string, limit: number): MinimalLayoutInformation[] {
        if (query.length < 1) {
            return []
        }
        query = Utils.simplifyStringForSearch(query)
        return ThemeSearch.allThemes
            .filter(th => !th.hideFromOverview || this._knownHiddenThemes.data.has(th.id))
            .filter(th => th.id !== this._state.layout.id)
            .filter(th => MoreScreen.MatchesLayout(th, query))
            .slice(0, limit)
    }


}