Merge pull request #438 from PowerShell/daviwil/remote-files

Add "remote files" feature that handles files from remote sessions

Author: daviwil

src/features/RemoteFiles.ts

@@ -0,0 +1,67 @@
+/*---------------------------------------------------------
+ * Copyright (C) Microsoft Corporation. All rights reserved.
+ *--------------------------------------------------------*/
+
+import os = require('os');
+import path = require('path');
+import vscode = require('vscode');
+import { IFeature } from '../feature';
+import { LanguageClient, RequestType, NotificationType } from 'vscode-languageclient';
+
+export class RemoteFilesFeature implements IFeature {
+
+ private tempSessionPathPrefix: string;
+
+ constructor() {
+ // Get the common PowerShell Editor Services temporary file path
+ // so that remote files from previous sessions can be closed.
+ this.tempSessionPathPrefix =
+ path.join(os.tmpdir(), 'PSES-')
+ .toLowerCase();
+
+ // At startup, close any lingering temporary remote files
+ this.closeRemoteFiles();
+
+ // TEMPORARY: Register for the onDidSave event so that we can alert
+ // the user when they attempt to save a remote file. We don't
+ // currently propagate saved content back to the remote session.
+ vscode.workspace.onDidSaveTextDocument(doc => {
+ if (this.isDocumentRemote(doc)) {
+ vscode.window.showWarningMessage(
+ "Changes to remote files are not yet saved back in the remote session, coming in 0.10.0.");
+ }
+ })
+ }
+
+ public setLanguageClient(languageclient: LanguageClient) {
+ }
+
+ public dispose() {
+ // Close any leftover remote files before exiting
+ this.closeRemoteFiles();
+ }
+
+ private isDocumentRemote(doc: vscode.TextDocument) {
+ return doc.languageId === "powershell" &&
+ doc.fileName.toLowerCase().startsWith(this.tempSessionPathPrefix);
+ }
+
+ private closeRemoteFiles() {
+ var remoteDocuments =
+ vscode.workspace.textDocuments.filter(
+ doc => this.isDocumentRemote(doc));
+
+ function innerCloseFiles(): Thenable<{}> {
+ if (remoteDocuments.length > 0) {
+ var doc = remoteDocuments.pop();
+
+ return vscode.window
+ .showTextDocument(doc)
+ .then(editor => vscode.commands.executeCommand("workbench.action.closeActiveEditor"))
+ .then(_ => innerCloseFiles());
+ }
+ };
+
+ innerCloseFiles();
+ }
+}

src/main.ts

@@ -16,6 +16,7 @@ import { OpenInISEFeature } from './features/OpenInISE';
 import { ExpandAliasFeature } from './features/ExpandAlias';
 import { ShowHelpFeature } from './features/ShowOnlineHelp';
 import { CodeActionsFeature } from './features/CodeActions';
+import { RemoteFilesFeature } from './features/RemoteFiles';
 import { DebugSessionFeature } from './features/DebugSession';
 import { PickPSHostProcessFeature } from './features/DebugSession';
 import { SelectPSSARulesFeature } from './features/SelectPSSARules';
@@ -106,6 +107,7 @@ export function activate(context: vscode.ExtensionContext): void {
  new CodeActionsFeature(),
  new NewFileOrProjectFeature(),
  new DocumentFormatterFeature(),
+ new RemoteFilesFeature(),
  new DebugSessionFeature(),
  new PickPSHostProcessFeature()
  ];