/*
Copyright 2019 Michael Telatynski <7t3chguy@gmail.com>
Copyright 2015, 2016, 2019, 2020, 2021 The Matrix.org Foundation C.I.C.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

import React from "react";
import { IFieldType, IInstance, IProtocol, IPublicRoomsChunkRoom } from "matrix-js-sdk/src/client";
import { Visibility } from "matrix-js-sdk/src/@types/partials";
import { IRoomDirectoryOptions } from "matrix-js-sdk/src/@types/requests";
import { logger } from "matrix-js-sdk/src/logger";

import { MatrixClientPeg } from "../../MatrixClientPeg";
import dis from "../../dispatcher/dispatcher";
import Modal from "../../Modal";
import { _t } from '../../languageHandler';
import SdkConfig from '../../SdkConfig';
import { instanceForInstanceId, protocolNameForInstanceId } from '../../utils/DirectoryUtils';
import Analytics from '../../Analytics';
import NetworkDropdown, { ALL_ROOMS, Protocols } from "../views/directory/NetworkDropdown";
import SettingsStore from "../../settings/SettingsStore";
import { IDialogProps } from "../views/dialogs/IDialogProps";
import AccessibleButton, { ButtonEvent } from "../views/elements/AccessibleButton";
import ErrorDialog from "../views/dialogs/ErrorDialog";
import QuestionDialog from "../views/dialogs/QuestionDialog";
import BaseDialog from "../views/dialogs/BaseDialog";
import DirectorySearchBox from "../views/elements/DirectorySearchBox";
import ScrollPanel from "./ScrollPanel";
import Spinner from "../views/elements/Spinner";
import { getDisplayAliasForAliasSet } from "../../Rooms";
import { Action } from "../../dispatcher/actions";
import PosthogTrackers from "../../PosthogTrackers";
import { ViewRoomPayload } from "../../dispatcher/payloads/ViewRoomPayload";
import { PublicRoomTile } from "../views/rooms/PublicRoomTile";

const LAST_SERVER_KEY = "mx_last_room_directory_server";
const LAST_INSTANCE_KEY = "mx_last_room_directory_instance";

function track(action: string) {
    Analytics.trackEvent('RoomDirectory', action);
}

interface IProps extends IDialogProps {
    initialText?: string;
}

interface IState {
    publicRooms: IPublicRoomsChunkRoom[];
    loading: boolean;
    protocolsLoading: boolean;
    error?: string;
    instanceId: string;
    roomServer: string;
    filterString: string;
}

export default class RoomDirectory extends React.Component<IProps, IState> {
    private unmounted = false;
    private nextBatch: string = null;
    private filterTimeout: number;
    private protocols: Protocols;

    constructor(props) {
        super(props);

        let protocolsLoading = true;
        if (!MatrixClientPeg.get()) {
            // We may not have a client yet when invoked from welcome page
            protocolsLoading = false;
        } else {
            MatrixClientPeg.get().getThirdpartyProtocols().then((response) => {
                this.protocols = response;
                const myHomeserver = MatrixClientPeg.getHomeserverName();
                const lsRoomServer = localStorage.getItem(LAST_SERVER_KEY);
                const lsInstanceId = localStorage.getItem(LAST_INSTANCE_KEY);

                let roomServer = myHomeserver;
                if (
                    SdkConfig.getObject("room_directory")?.get("servers")?.includes(lsRoomServer) ||
                    SettingsStore.getValue("room_directory_servers")?.includes(lsRoomServer)
                ) {
                    roomServer = lsRoomServer;
                }

                let instanceId: string = null;
                if (roomServer === myHomeserver && (
                    lsInstanceId === ALL_ROOMS ||
                    Object.values(this.protocols).some(p => p.instances.some(i => i.instance_id === lsInstanceId))
                )) {
                    instanceId = lsInstanceId;
                }

                // Refresh the room list only if validation failed and we had to change these
                if (this.state.instanceId !== instanceId || this.state.roomServer !== roomServer) {
                    this.setState({
                        protocolsLoading: false,
                        instanceId,
                        roomServer,
                    });
                    this.refreshRoomList();
                    return;
                }
                this.setState({ protocolsLoading: false });
            }, (err) => {
                logger.warn(`error loading third party protocols: ${err}`);
                this.setState({ protocolsLoading: false });
                if (MatrixClientPeg.get().isGuest()) {
                    // Guests currently aren't allowed to use this API, so
                    // ignore this as otherwise this error is literally the
                    // thing you see when loading the client!
                    return;
                }
                track('Failed to get protocol list from homeserver');
                const brand = SdkConfig.get().brand;
                this.setState({
                    error: _t(
                        '%(brand)s failed to get the protocol list from the homeserver. ' +
                        'The homeserver may be too old to support third party networks.',
                        { brand },
                    ),
                });
            });
        }

        this.state = {
            publicRooms: [],
            loading: true,
            error: null,
            instanceId: localStorage.getItem(LAST_INSTANCE_KEY),
            roomServer: localStorage.getItem(LAST_SERVER_KEY),
            filterString: this.props.initialText || "",
            protocolsLoading,
        };
    }

    componentDidMount() {
        this.refreshRoomList();
    }

    componentWillUnmount() {
        if (this.filterTimeout) {
            clearTimeout(this.filterTimeout);
        }
        this.unmounted = true;
    }

    private refreshRoomList = () => {
        this.nextBatch = null;
        this.setState({
            publicRooms: [],
            loading: true,
        });
        this.getMoreRooms();
    };

    private getMoreRooms(): Promise<boolean> {
        if (!MatrixClientPeg.get()) return Promise.resolve(false);

        this.setState({
            loading: true,
        });

        const filterString = this.state.filterString;
        const roomServer = this.state.roomServer;
        // remember the next batch token when we sent the request
        // too. If it's changed, appending to the list will corrupt it.
        const nextBatch = this.nextBatch;
        const opts: IRoomDirectoryOptions = { limit: 20 };
        if (roomServer != MatrixClientPeg.getHomeserverName()) {
            opts.server = roomServer;
        }
        if (this.state.instanceId === ALL_ROOMS) {
            opts.include_all_networks = true;
        } else if (this.state.instanceId) {
            opts.third_party_instance_id = this.state.instanceId as string;
        }
        if (this.nextBatch) opts.since = this.nextBatch;
        if (filterString) opts.filter = { generic_search_term: filterString };
        return MatrixClientPeg.get().publicRooms(opts).then((data) => {
            if (
                filterString != this.state.filterString ||
                roomServer != this.state.roomServer ||
                nextBatch != this.nextBatch) {
                // if the filter or server has changed since this request was sent,
                // throw away the result (don't even clear the busy flag
                // since we must still have a request in flight)
                return false;
            }

            if (this.unmounted) {
                // if we've been unmounted, we don't care either.
                return false;
            }

            this.nextBatch = data.next_batch;
            this.setState((s) => ({
                ...s,
                publicRooms: [...s.publicRooms, ...(data.chunk || [])],
                loading: false,
            }));
            return Boolean(data.next_batch);
        }, (err) => {
            if (
                filterString != this.state.filterString ||
                roomServer != this.state.roomServer ||
                nextBatch != this.nextBatch) {
                // as above: we don't care about errors for old requests either
                return false;
            }

            if (this.unmounted) {
                // if we've been unmounted, we don't care either.
                return false;
            }

            logger.error("Failed to get publicRooms: %s", JSON.stringify(err));
            track('Failed to get public room list');
            const brand = SdkConfig.get().brand;
            this.setState({
                loading: false,
                error: (
                    _t('%(brand)s failed to get the public room list.', { brand }) +
                    (err && err.message) ? err.message : _t('The homeserver may be unavailable or overloaded.')
                ),
            });
        });
    }

    /**
     * A limited interface for removing rooms from the directory.
     * Will set the room to not be publicly visible and delete the
     * default alias. In the long term, it would be better to allow
     * HS admins to do this through the RoomSettings interface, but
     * this needs SPEC-417.
     */
    private removeFromDirectory = (room: IPublicRoomsChunkRoom) => {
        const alias = getDisplayAliasForRoom(room);
        const name = room.name || alias || _t('Unnamed room');

        let desc;
        if (alias) {
            desc = _t('Delete the room address %(alias)s and remove %(name)s from the directory?', { alias, name });
        } else {
            desc = _t('Remove %(name)s from the directory?', { name: name });
        }

        Modal.createTrackedDialog('Remove from Directory', '', QuestionDialog, {
            title: _t('Remove from Directory'),
            description: desc,
            onFinished: (shouldDelete: boolean) => {
                if (!shouldDelete) return;

                const modal = Modal.createDialog(Spinner);
                let step = _t('remove %(name)s from the directory.', { name: name });

                MatrixClientPeg.get().setRoomDirectoryVisibility(room.room_id, Visibility.Private).then(() => {
                    if (!alias) return;
                    step = _t('delete the address.');
                    return MatrixClientPeg.get().deleteAlias(alias);
                }).then(() => {
                    modal.close();
                    this.refreshRoomList();
                }, (err) => {
                    modal.close();
                    this.refreshRoomList();
                    logger.error("Failed to " + step + ": " + err);
                    Modal.createTrackedDialog('Remove from Directory Error', '', ErrorDialog, {
                        title: _t('Error'),
                        description: (err && err.message)
                            ? err.message
                            : _t('The server may be unavailable or overloaded'),
                    });
                });
            },
        });
    };

    private onOptionChange = (server: string, instanceId?: string) => {
        // clear next batch so we don't try to load more rooms
        this.nextBatch = null;
        this.setState({
            // Clear the public rooms out here otherwise we needlessly
            // spend time filtering lots of rooms when we're about to
            // to clear the list anyway.
            publicRooms: [],
            roomServer: server,
            instanceId: instanceId,
            error: null,
        }, this.refreshRoomList);
        // We also refresh the room list each time even though this
        // filtering is client-side. It hopefully won't be client side
        // for very long, and we may have fetched a thousand rooms to
        // find the five gitter ones, at which point we do not want
        // to render all those rooms when switching back to 'all networks'.
        // Easiest to just blow away the state & re-fetch.

        // We have to be careful here so that we don't set instanceId = "undefined"
        localStorage.setItem(LAST_SERVER_KEY, server);
        if (instanceId) {
            localStorage.setItem(LAST_INSTANCE_KEY, instanceId);
        } else {
            localStorage.removeItem(LAST_INSTANCE_KEY);
        }
    };

    private onFillRequest = (backwards: boolean) => {
        if (backwards || !this.nextBatch) return Promise.resolve(false);

        return this.getMoreRooms();
    };

    private onFilterChange = (alias: string) => {
        this.setState({
            filterString: alias?.trim() || "",
        });

        // don't send the request for a little bit,
        // no point hammering the server with a
        // request for every keystroke, let the
        // user finish typing.
        if (this.filterTimeout) {
            clearTimeout(this.filterTimeout);
        }
        this.filterTimeout = setTimeout(() => {
            this.filterTimeout = null;
            this.refreshRoomList();
        }, 700);
    };

    private onFilterClear = () => {
        // update immediately
        this.setState({
            filterString: "",
        }, this.refreshRoomList);

        if (this.filterTimeout) {
            clearTimeout(this.filterTimeout);
        }
    };

    private onJoinFromSearchClick = (alias: string) => {
        // If we don't have a particular instance id selected, just show that rooms alias
        if (!this.state.instanceId || this.state.instanceId === ALL_ROOMS) {
            // If the user specified an alias without a domain, add on whichever server is selected
            // in the dropdown
            if (alias.indexOf(':') == -1) {
                alias = alias + ':' + this.state.roomServer;
            }
            this.showRoomAlias(alias, true);
        } else {
            // This is a 3rd party protocol. Let's see if we can join it
            const protocolName = protocolNameForInstanceId(this.protocols, this.state.instanceId);
            const instance = instanceForInstanceId(this.protocols, this.state.instanceId);
            const fields = protocolName
                ? this.getFieldsForThirdPartyLocation(alias, this.protocols[protocolName], instance)
                : null;
            if (!fields) {
                const brand = SdkConfig.get().brand;
                Modal.createTrackedDialog('Unable to join network', '', ErrorDialog, {
                    title: _t('Unable to join network'),
                    description: _t('%(brand)s does not know how to join a room on this network', { brand }),
                });
                return;
            }
            MatrixClientPeg.get().getThirdpartyLocation(protocolName, fields).then((resp) => {
                if (resp.length > 0 && resp[0].alias) {
                    this.showRoomAlias(resp[0].alias, true);
                } else {
                    Modal.createTrackedDialog('Room not found', '', ErrorDialog, {
                        title: _t('Room not found'),
                        description: _t('Couldn\'t find a matching Matrix room'),
                    });
                }
            }, (e) => {
                Modal.createTrackedDialog('Fetching third party location failed', '', ErrorDialog, {
                    title: _t('Fetching third party location failed'),
                    description: _t('Unable to look up room ID from server'),
                });
            });
        }
    };

    private onCreateRoomClick = (ev: ButtonEvent) => {
        this.onFinished();
        dis.dispatch({
            action: 'view_create_room',
            public: true,
            defaultName: this.state.filterString.trim(),
        });
        PosthogTrackers.trackInteraction("WebRoomDirectoryCreateRoomButton", ev);
    };

    private showRoomAlias(alias: string, autoJoin = false) {
        this.showRoom(null, alias, autoJoin);
    }

    private showRoom = (room: IPublicRoomsChunkRoom, roomAlias?: string, autoJoin = false, shouldPeek = false) => {
        this.onFinished();
        const payload: ViewRoomPayload = {
            action: Action.ViewRoom,
            auto_join: autoJoin,
            should_peek: shouldPeek,
            metricsTrigger: "RoomDirectory",
        };
        if (room) {
            // Don't let the user view a room they won't be able to either
            // peek or join: fail earlier so they don't have to click back
            // to the directory.
            if (MatrixClientPeg.get().isGuest()) {
                if (!room.world_readable && !room.guest_can_join) {
                    dis.dispatch({ action: 'require_registration' });
                    return;
                }
            }

            if (!roomAlias) {
                roomAlias = getDisplayAliasForRoom(room);
            }

            payload.oob_data = {
                avatarUrl: room.avatar_url,
                // XXX: This logic is duplicated from the JS SDK which
                // would normally decide what the name is.
                name: room.name || roomAlias || _t('Unnamed room'),
            };

            if (this.state.roomServer) {
                payload.via_servers = [this.state.roomServer];
            }
        }
        // It's not really possible to join Matrix rooms by ID because the HS has no way to know
        // which servers to start querying. However, there's no other way to join rooms in
        // this list without aliases at present, so if roomAlias isn't set here we have no
        // choice but to supply the ID.
        if (roomAlias) {
            payload.room_alias = roomAlias;
        } else {
            payload.room_id = room.room_id;
        }
        dis.dispatch(payload);
    };
    private stringLooksLikeId(s: string, fieldType: IFieldType) {
        let pat = /^#[^\s]+:[^\s]/;
        if (fieldType && fieldType.regexp) {
            pat = new RegExp(fieldType.regexp);
        }

        return pat.test(s);
    }

    private getFieldsForThirdPartyLocation(userInput: string, protocol: IProtocol, instance: IInstance) {
        // make an object with the fields specified by that protocol. We
        // require that the values of all but the last field come from the
        // instance. The last is the user input.
        const requiredFields = protocol.location_fields;
        if (!requiredFields) return null;
        const fields = {};
        for (let i = 0; i < requiredFields.length - 1; ++i) {
            const thisField = requiredFields[i];
            if (instance.fields[thisField] === undefined) return null;
            fields[thisField] = instance.fields[thisField];
        }
        fields[requiredFields[requiredFields.length - 1]] = userInput;
        return fields;
    }

    private onFinished = () => {
        this.props.onFinished(false);
    };

    render() {
        let content;
        if (this.state.error) {
            content = this.state.error;
        } else if (this.state.protocolsLoading) {
            content = <Spinner />;
        } else {
            const cells = (this.state.publicRooms || [])
                .map(room =>
                    <PublicRoomTile
                        key={room.room_id}
                        room={room}
                        showRoom={this.showRoom}
                        removeFromDirectory={this.removeFromDirectory}
                    />,
                );
            // we still show the scrollpanel, at least for now, because
            // otherwise we don't fetch more because we don't get a fill
            // request from the scrollpanel because there isn't one

            let spinner;
            if (this.state.loading) {
                spinner = <Spinner />;
            }

            const createNewButton = <>
                <hr />
                <AccessibleButton kind="primary" onClick={this.onCreateRoomClick} className="mx_RoomDirectory_newRoom">
                    { _t("Create new room") }
                </AccessibleButton>
            </>;

            let scrollPanelContent;
            let footer;
            if (cells.length === 0 && !this.state.loading) {
                footer = <>
                    <h5>{ _t('No results for "%(query)s"', { query: this.state.filterString.trim() }) }</h5>
                    <p>
                        { _t("Try different words or check for typos. " +
                            "Some results may not be visible as they're private and you need an invite to join them.") }
                    </p>
                    { createNewButton }
                </>;
            } else {
                scrollPanelContent = <div className="mx_RoomDirectory_table">
                    { cells }
                </div>;
                if (!this.state.loading && !this.nextBatch) {
                    footer = createNewButton;
                }
            }
            content = <ScrollPanel
                className="mx_RoomDirectory_tableWrapper"
                onFillRequest={this.onFillRequest}
                stickyBottom={false}
                startAtBottom={false}
            >
                { scrollPanelContent }
                { spinner }
                { footer && <div className="mx_RoomDirectory_footer">
                    { footer }
                </div> }
            </ScrollPanel>;
        }

        let listHeader;
        if (!this.state.protocolsLoading) {
            const protocolName = protocolNameForInstanceId(this.protocols, this.state.instanceId);
            let instanceExpectedFieldType;
            if (
                protocolName &&
                this.protocols &&
                this.protocols[protocolName] &&
                this.protocols[protocolName].location_fields.length > 0 &&
                this.protocols[protocolName].field_types
            ) {
                const lastField = this.protocols[protocolName].location_fields.slice(-1)[0];
                instanceExpectedFieldType = this.protocols[protocolName].field_types[lastField];
            }

            let placeholder = _t('Find a room…');
            if (!this.state.instanceId || this.state.instanceId === ALL_ROOMS) {
                placeholder = _t("Find a room… (e.g. %(exampleRoom)s)", {
                    exampleRoom: "#example:" + this.state.roomServer,
                });
            } else if (instanceExpectedFieldType) {
                placeholder = instanceExpectedFieldType.placeholder;
            }

            let showJoinButton = this.stringLooksLikeId(this.state.filterString, instanceExpectedFieldType);
            if (protocolName) {
                const instance = instanceForInstanceId(this.protocols, this.state.instanceId);
                if (this.getFieldsForThirdPartyLocation(
                    this.state.filterString,
                    this.protocols[protocolName],
                    instance,
                ) === null) {
                    showJoinButton = false;
                }
            }

            listHeader = <div className="mx_RoomDirectory_listheader">
                <DirectorySearchBox
                    className="mx_RoomDirectory_searchbox"
                    onChange={this.onFilterChange}
                    onClear={this.onFilterClear}
                    onJoinClick={this.onJoinFromSearchClick}
                    placeholder={placeholder}
                    showJoinButton={showJoinButton}
                    initialText={this.props.initialText}
                />
                <NetworkDropdown
                    protocols={this.protocols}
                    onOptionChange={this.onOptionChange}
                    selectedServerName={this.state.roomServer}
                    selectedInstanceId={this.state.instanceId}
                />
            </div>;
        }
        const explanation =
            _t("If you can't find the room you're looking for, ask for an invite or <a>create a new room</a>.", null,
                { a: sub => (
                    <AccessibleButton kind="link_inline" onClick={this.onCreateRoomClick}>
                        { sub }
                    </AccessibleButton>
                ) },
            );

        const title = _t("Explore rooms");
        return (
            <BaseDialog
                className="mx_RoomDirectory_dialog"
                hasCancel={true}
                onFinished={this.onFinished}
                title={title}
                screenName="RoomDirectory"
            >
                <div className="mx_RoomDirectory">
                    { explanation }
                    <div className="mx_RoomDirectory_list">
                        { listHeader }
                        { content }
                    </div>
                </div>
            </BaseDialog>
        );
    }
}

// Similar to matrix-react-sdk's MatrixTools.getDisplayAliasForRoom
// but works with the objects we get from the public room list
export function getDisplayAliasForRoom(room: IPublicRoomsChunkRoom) {
    return getDisplayAliasForAliasSet(room.canonical_alias, room.aliases);
}
