Pzhu/feature custom snooze time (#298934)

* Support custom inline suggestion snooze durations

* update per Copilot review

* Update code per code review. Importantly, add last used custom duration quick selection

* Undo unneeded changes

---------

Co-authored-by: Pengkun ZHU <pengkunzhu@Pengkuns-MacBook-Pro.local>
Co-authored-by: Benjamin Christopher Simmonds <44439583+benibenj@users.noreply.github.com>

Author: 3 people

src/vs/editor/browser/services/inlineCompletionsService.ts

@@ -212,27 +212,71 @@ export class SnoozeInlineCompletion extends Action2 {
 	private async getDurationFromUser(quickInputService: IQuickInputService, storageService: IStorageService): Promise<number | undefined> {
 		const lastSelectedDuration = storageService.getNumber(LAST_SNOOZE_DURATION_KEY, StorageScope.PROFILE, 300_000);
 
-		const items: (IQuickPickItem & { value: number })[] = [
-			{ label: '1 minute', id: '1', value: 60_000 },
-			{ label: '5 minutes', id: '5', value: 300_000 },
-			{ label: '10 minutes', id: '10', value: 600_000 },
-			{ label: '15 minutes', id: '15', value: 900_000 },
-			{ label: '30 minutes', id: '30', value: 1_800_000 },
-			{ label: '60 minutes', id: '60', value: 3_600_000 }
+		const predefinedItems: (IQuickPickItem & { value: number })[] = [
+			{ label: localize('snooze.1minute', "1 minute"), id: '1', value: 60_000 },
+			{ label: localize('snooze.5minutes', "5 minutes"), id: '5', value: 300_000 },
+			{ label: localize('snooze.10minutes', "10 minutes"), id: '10', value: 600_000 },
+			{ label: localize('snooze.15minutes', "15 minutes"), id: '15', value: 900_000 },
+			{ label: localize('snooze.30minutes', "30 minutes"), id: '30', value: 1_800_000 },
+			{ label: localize('snooze.60minutes', "60 minutes"), id: '60', value: 3_600_000 },
 		];
 
+		let items = predefinedItems;
+		if (lastSelectedDuration > 0 && !predefinedItems.some(item => item.value === lastSelectedDuration)) {
+			const minutes = lastSelectedDuration / 60_000;
+			const customItem: IQuickPickItem & { value: number } = {
+				label: localize('snooze.lastCustom', "{0} minutes (Last used)", minutes),
+				id: 'last-custom',
+				value: lastSelectedDuration,
+				description: localize('snooze.lastUsed', "Last used custom duration"),
+			};
+			const index = predefinedItems.findIndex(item => item.value > lastSelectedDuration);
+			if (index === -1) {
+				items = [...predefinedItems, customItem];
+			} else {
+				items = [...predefinedItems.slice(0, index), customItem, ...predefinedItems.slice(index)];
+			}
+		}
+
+		items.push({ label: localize('snooze.custom', "Custom..."), id: 'custom', value: -1 });
+
 		const picked = await quickInputService.pick(items, {
 			placeHolder: localize('snooze.placeholder', "Select snooze duration for Inline Suggestions"),
 			activeItem: items.find(item => item.value === lastSelectedDuration),
 		});
 
 		if (picked) {
+			if (picked.id === 'custom') {
+				return this.getCustomDurationFromUser(quickInputService, storageService);
+			}
 			storageService.store(LAST_SNOOZE_DURATION_KEY, picked.value, StorageScope.PROFILE, StorageTarget.USER);
 			return picked.value;
 		}
 
 		return undefined;
 	}
+
+	private async getCustomDurationFromUser(quickInputService: IQuickInputService, storageService: IStorageService): Promise<number | undefined> {
+		const customMinutes = await quickInputService.input({
+			placeHolder: localize('snooze.customPlaceholder', "Duration in minutes (e.g. 90)"),
+			prompt: localize('snooze.customPrompt', "Enter snooze duration in minutes"),
+			validateInput: async (value) => {
+				const n = Number(value);
+				if (isNaN(n) || n <= 0 || !Number.isFinite(n)) {
+					return localize('snooze.invalidInput', "Please enter a positive number");
+				}
+				return undefined;
+			}
+		});
+
+		if (customMinutes) {
+			const ms = Number(customMinutes) * 60_000;
+			storageService.store(LAST_SNOOZE_DURATION_KEY, ms, StorageScope.PROFILE, StorageTarget.USER);
+			return ms;
+		}
+
+		return undefined;
+	}
 }
 
 export class CancelSnoozeInlineCompletion extends Action2 {