azuredatastudio/samples/extensionSamples/src/controllers/mainController.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

'use strict';

import * as sqlops from 'sqlops';
import * as Utils from '../utils';
import ControllerBase from './controllerBase';
import * as fs from 'fs';
import * as path from 'path';

/**
 * The main controller class that initializes the extension
 */
export default class MainController extends ControllerBase {
    // PUBLIC METHODS //////////////////////////////////////////////////////
    /**
     * Deactivates the extension
     */
    public deactivate(): void {
        Utils.logDebug('Main controller deactivated');
    }

    public activate(): Promise<boolean> {
        const webviewExampleHtml = fs.readFileSync(path.join(__dirname, 'webviewExample.html')).toString();
        const buttonHtml = fs.readFileSync(path.join(__dirname, 'button.html')).toString();
        const counterHtml = fs.readFileSync(path.join(__dirname, 'counter.html')).toString();

        let countWidget: sqlops.DashboardWebview;
        let buttonWidget: sqlops.DashboardWebview;
        let count = 0;

        let dialog: sqlops.ModalDialog = sqlops.window.createDialog('Flyout extension');
        dialog.html = '<div>This is a flyout extension.</div>';

        sqlops.dashboard.registerWebviewProvider('webview-count', e => {
            e.html = counterHtml;
            countWidget = e;
        });
        sqlops.dashboard.registerWebviewProvider('webview-button', e => {
            e.html = buttonHtml;
            buttonWidget = e;
            e.onMessage(event => {
                if (event === 'openFlyout') {
                    dialog.open();
                } else {
                    count++;
                    countWidget.postMessage(count);
                }
            });
        });
        sqlops.dashboard.registerWebviewProvider('webviewExample', e => {
            e.html = webviewExampleHtml;
        });

        return Promise.resolve(true);
    }
}