blap/src/widgets/WidgetApi.ts

/*
Copyright 2020 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.
*/

// Dev note: This is largely inspired by Dimension. Used with permission.
// https://github.com/turt2live/matrix-dimension/blob/4f92d560266635e5a3c824606215b84e8c0b19f5/web/app/shared/services/scalar/scalar-widget.api.ts

import { randomString } from "matrix-js-sdk/src/randomstring";

export enum Capability {
    Screenshot = "m.capability.screenshot",
    Sticker = "m.sticker",
    AlwaysOnScreen = "m.always_on_screen",
}

export enum KnownWidgetActions {
    GetSupportedApiVersions = "supported_api_versions",
    TakeScreenshot = "screenshot",
    GetCapabilities = "capabilities",
    SendEvent = "send_event",
    UpdateVisibility = "visibility",
    ReceiveOpenIDCredentials = "openid_credentials",
    SetAlwaysOnScreen = "set_always_on_screen",
}
export type WidgetAction = KnownWidgetActions | string;

export enum WidgetApiType {
    ToWidget = "toWidget",
    FromWidget = "fromWidget",
}

export interface WidgetRequest {
    api: WidgetApiType;
    widgetId: string;
    requestId: string;
    data: any;
    action: WidgetAction;
}

export interface ToWidgetRequest extends WidgetRequest {
    api: WidgetApiType.ToWidget;
}

export interface FromWidgetRequest extends WidgetRequest {
    api: WidgetApiType.FromWidget;
    response: any;
}

/**
 * Handles Riot <--> Widget interactions for embedded/standalone widgets.
 */
export class WidgetApi {
    private origin: string;
    private inFlightRequests: {[requestId: string]: (reply: FromWidgetRequest) => void} = {};
    private readyPromise: Promise<any>;
    private readyPromiseResolve: () => void;

    constructor(currentUrl: string, private widgetId: string, private requestedCapabilities: string[]) {
        this.origin = new URL(currentUrl).origin;

        this.readyPromise = new Promise<any>(resolve => this.readyPromiseResolve = resolve);

        window.addEventListener("message", event => {
            if (event.origin !== this.origin) return; // ignore: invalid origin
            if (!event.data) return; // invalid schema
            if (event.data.widgetId !== this.widgetId) return; // not for us

            const payload = <WidgetRequest>event.data;
            if (payload.api === WidgetApiType.ToWidget && payload.action) {
                console.log(`[WidgetAPI] Got request: ${JSON.stringify(payload)}`);

                if (payload.action === KnownWidgetActions.GetCapabilities) {
                    this.onCapabilitiesRequest(<ToWidgetRequest>payload);
                    this.readyPromiseResolve();
                } else {
                    console.warn(`[WidgetAPI] Got unexpected action: ${payload.action}`);
                }
            } else if (payload.api === WidgetApiType.FromWidget && this.inFlightRequests[payload.requestId]) {
                console.log(`[WidgetAPI] Got reply: ${JSON.stringify(payload)}`);
                const handler = this.inFlightRequests[payload.requestId];
                delete this.inFlightRequests[payload.requestId];
                handler(<FromWidgetRequest>payload);
            } else {
                console.warn(`[WidgetAPI] Unhandled payload: ${JSON.stringify(payload)}`);
            }
        });
    }

    public waitReady(): Promise<any> {
        return this.readyPromise;
    }

    private replyToRequest(payload: ToWidgetRequest, reply: any) {
        if (!window.parent) return;

        const request = JSON.parse(JSON.stringify(payload));
        request.response = reply;

        window.parent.postMessage(request, this.origin);
    }

    private onCapabilitiesRequest(payload: ToWidgetRequest) {
        return this.replyToRequest(payload, {capabilities: this.requestedCapabilities});
    }

    public callAction(action: WidgetAction, payload: any, callback: (reply: FromWidgetRequest) => void) {
        if (!window.parent) return;

        const request: FromWidgetRequest = {
            api: WidgetApiType.FromWidget,
            widgetId: this.widgetId,
            action: action,
            requestId: randomString(160),
            data: payload,
            response: {}, // Not used at this layer - it's used when the client responds
        };
        this.inFlightRequests[request.requestId] = callback;

        console.log(`[WidgetAPI] Sending request: `, request);
        window.parent.postMessage(request, "*");
    }

    public setAlwaysOnScreen(onScreen: boolean): Promise<any> {
        return new Promise<any>(resolve => {
            this.callAction(KnownWidgetActions.SetAlwaysOnScreen, {value: onScreen}, resolve);
        });
    }
}