extension/src/goSurvey.ts

/* eslint-disable @typescript-eslint/no-explicit-any */
/*---------------------------------------------------------
 * Copyright 2021 The Go Authors. All rights reserved.
 * Licensed under the MIT License. See LICENSE in the project root for license information.
 *--------------------------------------------------------*/

'use strict';

import vscode = require('vscode');
import { CommandFactory } from './commands';
import { getGoConfig } from './config';
import { extensionId } from './const';
import { GoExtensionContext } from './context';
import {
	DEVELOPER_SURVEY_STATE_KEY,
	getDeveloperSurveyState,
	maybePromptForDeveloperSurvey,
	promptForDeveloperSurvey
} from './developerSurvey/prompt';
import {
	DEVELOPER_SURVEY_CONFIG_STATE_KEY,
	getLatestDeveloperSurvey,
	getDeveloperSurveyConfigState
} from './developerSurvey/config';
import { outputChannel } from './goStatus';
import { getLocalGoplsVersion } from './language/goLanguageServer';
import { getFromGlobalState, getFromWorkspaceState, updateGlobalState } from './stateUtils';
import { getGoVersion } from './util';
import { promptNext4Weeks } from './utils/randomDayutils';

/**
 * GOPLS_SURVEY_KEY is the key for the gopls survey state stored in VSCode
 * memento. It should not be changed to maintain backward compatibility
 * with previous extension versions.
 */
export const GOPLS_SURVEY_STATE_KEY = 'goplsSurveyConfig';

/**
 * GoplsSurveyState is the set of global properties used to determine if
 * we should prompt a user to take the gopls survey.
 * This interface is stored in VS Code's memento. The field names should not
 * be changed as they are key to parsing the stored data from previous releases.
 */
export interface GoplsSurveyState {
	// prompt is true if the user can be prompted to take the survey.
	// It is false if the user has responded "Never" to the prompt.
	prompt?: boolean;

	// promptThisMonth is true if we have used a random number generator
	// to determine if the user should be prompted this month.
	// It is undefined if we have not yet made the determination.
	promptThisMonth?: boolean;

	// dateToPromptThisMonth is the date on which we should prompt the user
	// this month. (It is no longer necessarily in the current month.)
	dateToPromptThisMonth?: Date;

	// dateComputedPromptThisMonth is the date on which the values of
	// promptThisMonth and dateToPromptThisMonth were set.
	dateComputedPromptThisMonth?: Date;

	// lastDatePrompted is the most recent date that the user has been prompted.
	lastDatePrompted?: Date;

	// lastDateAccepted is the most recent date that the user responded "Yes"
	// to the survey prompt. The user need not have completed the survey.
	lastDateAccepted?: Date;
}

export function maybePromptForGoplsSurvey(goCtx: GoExtensionContext) {
	// First, check the value of the 'go.survey.prompt' setting to see
	// if the user has opted out of all survey prompts.
	const goConfig = getGoConfig();
	if (goConfig.get('survey.prompt') === false) {
		return;
	}
	const now = new Date();
	let state = shouldPromptForSurvey(now, getGoplsSurveyState());
	if (!state) {
		return;
	}
	if (!state.dateToPromptThisMonth) {
		return;
	}
	const callback = async () => {
		const currentTime = new Date();
		const { lastUserAction = new Date() } = goCtx;
		// Make sure the user has been idle for at least a minute.
		if (minutesBetween(lastUserAction, currentTime) < 1) {
			setTimeout(callback, 5 * timeMinute);
			return;
		}
		state = await promptForGoplsSurvey(goCtx, state, now);
		if (state) {
			storeSurveyState(GOPLS_SURVEY_STATE_KEY, state);
		}
	};

	// 0 if we passed the computed dateToPromptThisMonth past.
	// If the prompt date was computed a while ago (dateComputedPromptThisMonth),
	// shouldPromptForSurvey should've made a new decision before we get here.
	const delayMs = Math.max(state.dateToPromptThisMonth.getTime() - now.getTime(), 0);
	setTimeout(callback, delayMs);
}

export function shouldPromptForSurvey(now: Date, state: GoplsSurveyState): GoplsSurveyState | undefined {
	// If the prompt value is not set, assume we haven't prompted the user
	// and should do so.
	if (state.prompt === undefined) {
		state.prompt = true;
	}
	storeSurveyState(GOPLS_SURVEY_STATE_KEY, state);
	if (!state.prompt) {
		return;
	}

	// Check if the user has taken the survey in the last year.
	// Don't prompt them if they have been.
	if (state.lastDateAccepted) {
		if (daysBetween(now, state.lastDateAccepted) < 365) {
			return;
		}
	}

	// Check if the user has been prompted for the survey in the last 90 days.
	// Don't prompt them if they have been.
	if (state.lastDatePrompted) {
		if (daysBetween(now, state.lastDatePrompted) < 90) {
			return;
		}
	}

	// Check if the extension has been activated this month.
	if (state.dateComputedPromptThisMonth) {
		// The extension has been activated this month, so we should have already
		// decided if the user should be prompted.
		if (daysBetween(now, state.dateComputedPromptThisMonth) < 28) {
			return state;
		}
	}
	// This is the first activation this month (or ever), so decide if we
	// should prompt the user. This is done by generating a random number in
	// the range [0, 1) and checking if it is < probability.
	// We then randomly pick a day in the next 4 weeks to prompt the user.
	// Probability is set based on the # of responses received, and will be
	// decreased if we begin receiving > 200 responses/month.
	const probability = 0.06;
	state.promptThisMonth = Math.random() < probability;
	if (state.promptThisMonth) {
		state.dateToPromptThisMonth = promptNext4Weeks(now);
	} else {
		state.dateToPromptThisMonth = undefined;
	}
	state.dateComputedPromptThisMonth = now;
	storeSurveyState(GOPLS_SURVEY_STATE_KEY, state);
	return state;
}

async function promptForGoplsSurvey(
	goCtx: GoExtensionContext,
	state: GoplsSurveyState = {},
	now: Date
): Promise<GoplsSurveyState> {
	const selected = await vscode.window.showInformationMessage(
		`Looks like you are using the Go extension for VS Code.
Could you help us improve this extension by filling out a 1-2 minute survey about your experience with it?`,
		'Yes',
		'Not now',
		'Never'
	);

	// Update the time last asked.
	state.lastDatePrompted = now;

	switch (selected) {
		case 'Yes':
			{
				const { latestConfig } = goCtx;
				state.lastDateAccepted = now;
				state.prompt = true;
				const goplsEnabled = latestConfig?.enabled;
				const usersGoplsVersion = await getLocalGoplsVersion(latestConfig);
				const goV = await getGoVersion();
				const goVersion = goV ? (goV.isDevel ? 'devel' : goV.format(true)) : 'na';
				const surveyURL = `https://go.dev/s/ide-hats-survey/?s=c&usingGopls=${goplsEnabled}&gopls=${usersGoplsVersion?.version}&extid=${extensionId}&go=${goVersion}&os=${process.platform}`;
				await vscode.env.openExternal(vscode.Uri.parse(surveyURL));
			}
			break;
		case 'Not now':
			state.prompt = true;

			vscode.window.showInformationMessage("No problem! We'll ask you again another time.");
			break;
		case 'Never': {
			state.prompt = false;

			const selected = await vscode.window.showInformationMessage(
				`No problem! We won't ask again.
To opt-out of all survey prompts, please disable the 'Go > Survey: Prompt' setting.`,
				'Open Settings'
			);
			switch (selected) {
				case 'Open Settings':
					vscode.commands.executeCommand('workbench.action.openSettings', 'go.survey.prompt');
					break;
				default:
					break;
			}
			break;
		}
		default:
			// If the user closes the prompt without making a selection, treat it
			// like a "Not now" response.
			state.prompt = true;

			break;
	}
	return state;
}

function getGoplsSurveyState(): GoplsSurveyState {
	return getStateConfig(GOPLS_SURVEY_STATE_KEY) as GoplsSurveyState;
}

export const resetSurveyStates: CommandFactory = () => () => {
	storeSurveyState(GOPLS_SURVEY_STATE_KEY, null);
	storeSurveyState(DEVELOPER_SURVEY_STATE_KEY, null);
	storeSurveyState(DEVELOPER_SURVEY_CONFIG_STATE_KEY, null);
};

export function storeSurveyState(key: string, state: any) {
	if (state) {
		updateGlobalState(key, JSON.stringify(state));
	} else {
		updateGlobalState(key, null); // reset
	}
}

/**
 * Retrieves and parses a JSON state object from either workspace or global storage.
 *
 * This function fetches a string from the state, safely parses it as JSON, and
 * automatically converts any values with keys containing "date" or "timestamp"
 * into `Date` objects. It returns an empty object if no state is found or if
 * parsing fails.
 */
export function getStateConfig(globalStateKey: string, workspace?: boolean): any {
	let saved: any;
	if (workspace === true) {
		saved = getFromWorkspaceState(globalStateKey);
	} else {
		saved = getFromGlobalState(globalStateKey);
	}
	if (saved === undefined) {
		return {};
	}
	try {
		const state = JSON.parse(saved, (key: string, value: any) => {
			// Make sure values that should be dates are correctly converted.
			if (key.toLowerCase().includes('date') || key.toLowerCase().includes('timestamp')) {
				return new Date(value);
			}
			return value;
		});
		return state || {};
	} catch (err) {
		console.log(`Error parsing JSON from ${saved}: ${err}`);
		return {};
	}
}

export const showSurveyStates: CommandFactory = (_, goCtx) => async () => {
	outputChannel.info('HaTs Survey State');
	outputChannel.info(JSON.stringify(getGoplsSurveyState(), null, 2));
	outputChannel.show();

	outputChannel.info('Developer Survey State');
	outputChannel.info(JSON.stringify(getDeveloperSurveyState(), null, 2));
	outputChannel.show();

	outputChannel.info('Latest Developer Survey Configuration');
	outputChannel.info(JSON.stringify(getDeveloperSurveyConfigState(), null, 2));
	outputChannel.show();

	const now = new Date();
	let selected = await vscode.window.showInformationMessage('Prompt for HaTS survey?', 'Yes', 'Maybe', 'No');
	switch (selected) {
		case 'Yes':
			promptForGoplsSurvey(goCtx, getGoplsSurveyState(), now);
			break;
		case 'Maybe':
			maybePromptForGoplsSurvey(goCtx);
			break;
		default:
			break;
	}
	selected = await vscode.window.showInformationMessage('Prompt for Developer survey?', 'Yes', 'Maybe', 'No');
	switch (selected) {
		case 'Yes': {
			const config = await getLatestDeveloperSurvey(now);
			if (!config) {
				return;
			}
			promptForDeveloperSurvey(now, getDeveloperSurveyState(), config);
			break;
		}
		case 'Maybe':
			maybePromptForDeveloperSurvey(goCtx);
			break;
		default:
			break;
	}
};

export const timeMinute = 1000 * 60;
const timeHour = timeMinute * 60;
export const timeDay = timeHour * 24;

// daysBetween returns the number of days between a and b.
export function daysBetween(a: Date, b: Date): number {
	return msBetween(a, b) / timeDay;
}

// minutesBetween returns the number of minutes between a and b.
export function minutesBetween(a: Date, b: Date): number {
	return msBetween(a, b) / timeMinute;
}

export function msBetween(a: Date, b: Date): number {
	return Math.abs(a.getTime() - b.getTime());
}