microsoft/vscode-react-native

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for details.

import * as vscode from "vscode";
import { IExperiment } from "../IExperiment";
import {
    ExperimentConfig,
    ExperimentParameters,
    ExperimentResult,
    ExperimentStatuses,
} from "../experimentService";
import { PROMPT_TITLES } from "../experimentsStrings";

export default class RNTPreviewPrompt implements IExperiment {
    public async run(
        newExpConfig: ExperimentConfig,
        curExpParameters?: ExperimentParameters,
    ): Promise<ExperimentResult> {
        if (
            curExpParameters &&
            (curExpParameters.promptShown ||
                newExpConfig.popCoveragePercent === curExpParameters.popCoveragePercent)
        ) {
            return {
                resultStatus: ExperimentStatuses.DISABLED,
                updatedExperimentParameters: curExpParameters,
            };
        }

        const updatedExperimentParameters = this.showPrompIfThresholdIsNotExceeded(
            newExpConfig,
            curExpParameters,
        );

        return {
            resultStatus: ExperimentStatuses.ENABLED,
            updatedExperimentParameters,
        };
    }

    private showPrompIfThresholdIsNotExceeded(
        newExpConfig: ExperimentConfig,
        promptParameters?: ExperimentParameters,
    ) {
        if (promptParameters) {
            promptParameters.popCoveragePercent = newExpConfig.popCoveragePercent;
        } else {
            promptParameters = Object.assign({}, newExpConfig, {
                extensionId: "msjsdiag.vscode-react-native-preview",
                promptShown: false,
            });
        }

        if (newExpConfig.popCoveragePercent > Math.random()) {
            const buttonText = "Open extension";
            vscode.window
                .showInformationMessage(PROMPT_TITLES.RNT_PREVIEW_PROMPT, buttonText)
                .then(selection => {
                    if (selection === buttonText && promptParameters) {
                        vscode.commands.executeCommand(
                            "workbench.extensions.search",
                            promptParameters.extensionId,
                        );
                        vscode.commands.executeCommand(
                            "extension.open",
                            promptParameters.extensionId,
                        );
                    }
                });

            promptParameters.promptShown = true;
        } else {
            promptParameters.promptShown = false;
        }

        return promptParameters;
    }
}