copy commands, history picker

Author: jrieken

src/locations/index.ts

@@ -5,7 +5,7 @@
 
 import * as vscode from 'vscode';
 import { SymbolsTree } from '../tree';
-import { LocationTreeInput, ReferenceItem } from './model';
+import { FileItem, LocationsModel, LocationTreeInput, ReferenceItem } from './model';
 
 export function register(tree: SymbolsTree, context: vscode.ExtensionContext): void {
 
@@ -28,6 +28,41 @@ export function register(tree: SymbolsTree, context: vscode.ExtensionContext): v
 					preserveFocus
 				});
 			}
-		})
+		}),
+		vscode.commands.registerCommand('references-view.copy', copyCommand),
+		vscode.commands.registerCommand('references-view.copyAll', copyAllCommand),
+		vscode.commands.registerCommand('references-view.copyPath', copyPathCommand),
 	);
 }
+
+const copyAllCommand = async (item: ReferenceItem | FileItem | unknown) => {
+	if (item instanceof ReferenceItem) {
+		copyCommand(item.file.model);
+	} else if (item instanceof FileItem) {
+		copyCommand(item.model);
+	}
+};
+
+const copyCommand = async (item: LocationsModel | ReferenceItem | FileItem | unknown) => {
+	let val: string | undefined;
+	if (item instanceof LocationsModel) {
+		val = await item.asCopyText();
+	} else if (item instanceof ReferenceItem) {
+		val = await item.asCopyText();
+	} else if (item instanceof FileItem) {
+		val = await item.asCopyText();
+	}
+	if (val) {
+		await vscode.env.clipboard.writeText(val);
+	}
+};
+
+const copyPathCommand = (item: FileItem | unknown) => {
+	if (item instanceof FileItem) {
+		if (item.uri.scheme === 'file') {
+			vscode.env.clipboard.writeText(item.uri.fsPath);
+		} else {
+			vscode.env.clipboard.writeText(item.uri.toString(true));
+		}
+	}
+};

src/locations/model.ts

@@ -45,7 +45,7 @@ export class LocationTreeInput implements SymbolTreeInput {
 	}
 }
 
-class LocationsModel implements SymbolItemNavigation<FileItem | ReferenceItem>, SymbolItemEditorHighlights<FileItem | ReferenceItem> {
+export class LocationsModel implements SymbolItemNavigation<FileItem | ReferenceItem>, SymbolItemEditorHighlights<FileItem | ReferenceItem> {
 
 	private _onDidChange = new vscode.EventEmitter<FileItem | ReferenceItem | undefined>();
 	readonly onDidChangeTreeData = this._onDidChange.event;
@@ -292,7 +292,7 @@ class LocationsTreeDataProvider implements Required<vscode.TreeDataProvider<File
 	}
 }
 
-class FileItem {
+export class FileItem {
 
 	constructor(
 		readonly uri: vscode.Uri,

src/tree.ts

@@ -180,11 +180,16 @@ class TreeDataProviderDelegate implements vscode.TreeDataProvider<undefined> {
 // --- history
 
 class HistoryItem {
+
+	readonly description: string;
+
 	constructor(
 		readonly word: string,
 		readonly anchor: WordAnchor,
 		readonly input: SymbolTreeInput,
-	) { }
+	) {
+		this.description = `${vscode.workspace.asRelativePath(input.uri)} • ${input.title.toLocaleLowerCase()}`;
+	}
 }
 
 class TreeInputHistory implements vscode.TreeDataProvider<HistoryItem>{
@@ -221,7 +226,22 @@ class TreeInputHistory implements vscode.TreeDataProvider<HistoryItem>{
 					const position = item.anchor.getPosition() ?? item.input.position;
 					return vscode.commands.executeCommand('vscode.open', item.input.uri, { selection: new vscode.Range(position, position) });
 				}
-			})
+			}),
+			vscode.commands.registerCommand('references-view.pickFromHistory', async () => {
+				interface HistoryPick extends vscode.QuickPickItem {
+					item: HistoryItem;
+				}
+				const entries = await this.getChildren();
+				const picks = entries.map(item => <HistoryPick>{
+					label: item.word,
+					description: item.description,
+					item
+				});
+				const pick = await vscode.window.showQuickPick(picks, { placeHolder: 'Select previous reference search' });
+				if (pick) {
+					this._reRunHistoryItem(pick.item);
+				}
+			}),
 		);
 	}
 
@@ -264,10 +284,10 @@ class TreeInputHistory implements vscode.TreeDataProvider<HistoryItem>{
 
 	// --- tree data provider
 
-	getTreeItem(element: HistoryItem): vscode.TreeItem {
-		const result = new vscode.TreeItem(element.word);
-		result.description = `${vscode.workspace.asRelativePath(element.input.uri)} • ${element.input.title.toLocaleLowerCase()}`;
-		// result.command = { command: 'references-view.SHOW', arguments: [element], title: 'Rerun' };
+	getTreeItem(item: HistoryItem): vscode.TreeItem {
+		const result = new vscode.TreeItem(item.word);
+		result.description = item.description;
+		result.command = { command: '_references-view.showHistoryItem', arguments: [item], title: 'Rerun' };
 		result.collapsibleState = vscode.TreeItemCollapsibleState.None;
 		result.contextValue = 'history-item';
 		return result;