Skip to content

Use context.storageUri for logs and support None level #4071

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Jul 12, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions package.json
Original file line number Diff line number Diff line change
Expand Up @@ -798,10 +798,11 @@
"Verbose",
"Normal",
"Warning",
"Error"
"Error",
"None"
],
"default": "Normal",
"description": "Sets the logging verbosity level for the PowerShell Editor Services host executable. Valid values are 'Diagnostic', 'Verbose', 'Normal', 'Warning', and 'Error'"
"description": "Sets the logging verbosity level for the PowerShell Editor Services host executable. Valid values are 'Diagnostic', 'Verbose', 'Normal', 'Warning', 'Error', and 'None'"
},
"powershell.developer.editorServicesWaitForDebugger": {
"type": "boolean",
Expand Down
5 changes: 5 additions & 0 deletions src/features/ExternalApi.ts
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ export interface IPowerShellExtensionClient {
unregisterExternalExtension(uuid: string): boolean;
getPowerShellVersionDetails(uuid: string): Promise<IExternalPowerShellDetails>;
waitUntilStarted(uuid: string): Promise<void>;
getStorageUri(): vscode.Uri;
}

/*
Expand Down Expand Up @@ -166,6 +167,10 @@ export class ExternalApiFeature extends LanguageClientConsumer implements IPower
return this.sessionManager.waitUntilStarted();
}

public getStorageUri(): vscode.Uri {
return this.extensionContext.storageUri;
}

public dispose() {
// Nothing to dispose.
}
Expand Down
39 changes: 21 additions & 18 deletions src/logging.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ export enum LogLevel {
Normal,
Warning,
Error,
None,
}

/** Interface for logging operations. New features should use this interface for the "type" of logger.
Expand All @@ -29,19 +30,24 @@ export interface ILogger {

export class Logger implements ILogger {

public logBasePath: string;
public logSessionPath: string;
public logBasePath: vscode.Uri;
public logSessionPath: vscode.Uri;
public MinimumLogLevel: LogLevel = LogLevel.Normal;

private commands: vscode.Disposable[];
private logChannel: vscode.OutputChannel;
private logFilePath: string;
private logFilePath: vscode.Uri;

constructor() {
constructor(logBasePath: vscode.Uri) {
this.logChannel = vscode.window.createOutputChannel("PowerShell Extension Logs");

this.logBasePath = path.resolve(__dirname, "../logs");
utils.ensurePathExists(this.logBasePath);
if (logBasePath === undefined) {
// No workspace, we have to use another folder.
this.logBasePath = vscode.Uri.file(path.resolve(__dirname, "../logs"));
utils.ensurePathExists(this.logBasePath.fsPath);
} else {
this.logBasePath = vscode.Uri.joinPath(logBasePath, "logs");
}

this.commands = [
vscode.commands.registerCommand(
Expand All @@ -59,8 +65,8 @@ export class Logger implements ILogger {
this.logChannel.dispose();
}

public getLogFilePath(baseName: string): string {
return path.resolve(this.logSessionPath, `${baseName}.log`);
public getLogFilePath(baseName: string): vscode.Uri {
return vscode.Uri.joinPath(this.logSessionPath, `${baseName}.log`);
}

public writeAtLevel(logLevel: LogLevel, message: string, ...additionalMessages: string[]) {
Expand Down Expand Up @@ -136,17 +142,16 @@ export class Logger implements ILogger {
}
}

public startNewLog(minimumLogLevel: string = "Normal") {
public async startNewLog(minimumLogLevel: string = "Normal") {
this.MinimumLogLevel = this.logLevelNameToValue(minimumLogLevel.trim());

this.logSessionPath =
path.resolve(
vscode.Uri.joinPath(
this.logBasePath,
`${Math.floor(Date.now() / 1000)}-${vscode.env.sessionId}`);

this.logFilePath = this.getLogFilePath("vscode-powershell");

utils.ensurePathExists(this.logSessionPath);
await vscode.workspace.fs.createDirectory(this.logSessionPath);
}

private logLevelNameToValue(logLevelName: string): LogLevel {
Expand All @@ -156,6 +161,7 @@ export class Logger implements ILogger {
case "normal": return LogLevel.Normal;
case "warning": return LogLevel.Warning;
case "error": return LogLevel.Error;
case "none": return LogLevel.None;
default: return LogLevel.Normal;
}
}
Expand All @@ -168,10 +174,7 @@ export class Logger implements ILogger {
if (this.logSessionPath) {
// Open the folder in VS Code since there isn't an easy way to
// open the folder in the platform's file browser
vscode.commands.executeCommand(
"vscode.openFolder",
vscode.Uri.file(this.logSessionPath),
true);
vscode.commands.executeCommand("vscode.openFolder", this.logSessionPath, true);
}
}

Expand All @@ -181,9 +184,9 @@ export class Logger implements ILogger {
`${now.toLocaleDateString()} ${now.toLocaleTimeString()} [${LogLevel[level].toUpperCase()}] - ${message}`;

this.logChannel.appendLine(timestampedMessage);
if (this.logFilePath) {
if (this.logFilePath && this.MinimumLogLevel !== LogLevel.None) {
fs.appendFile(
this.logFilePath,
this.logFilePath.fsPath,
timestampedMessage + os.EOL,
(err) => {
if (err) {
Expand Down
3 changes: 2 additions & 1 deletion src/main.ts
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<IPower
});

// Setup the logger.
logger = new Logger();
logger = new Logger(context.storageUri);
logger.MinimumLogLevel = LogLevel[extensionSettings.developer.editorServicesLogLevel];

sessionManager =
Expand Down Expand Up @@ -180,6 +180,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<IPower
unregisterExternalExtension: uuid => externalApi.unregisterExternalExtension(uuid),
getPowerShellVersionDetails: uuid => externalApi.getPowerShellVersionDetails(uuid),
waitUntilStarted: uuid => externalApi.waitUntilStarted(uuid),
getStorageUri: () => externalApi.getStorageUri(),
};
}

Expand Down
2 changes: 1 addition & 1 deletion src/process.ts
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ export class PowerShellProcess {
: "";

this.startPsesArgs +=
`-LogPath '${PowerShellProcess.escapeSingleQuotes(editorServicesLogPath)}' ` +
`-LogPath '${PowerShellProcess.escapeSingleQuotes(editorServicesLogPath.fsPath)}' ` +
`-SessionDetailsPath '${PowerShellProcess.escapeSingleQuotes(this.sessionFilePath)}' ` +
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We might want to move sessionFilePath into context.storageUri too. I also need to check if we spawn an instance per workspace or not.

`-FeatureFlags @(${featureFlags}) `;

Expand Down
2 changes: 1 addition & 1 deletion src/session.ts
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ export class SessionManager implements Middleware {
this.sessionSettings.powerShellDefaultVersion = exeNameOverride;
}

this.log.startNewLog(this.sessionSettings.developer.editorServicesLogLevel);
await this.log.startNewLog(this.sessionSettings.developer.editorServicesLogLevel);

// Create the PowerShell executable finder now
this.powershellExeFinder = new PowerShellExeFinder(
Expand Down
9 changes: 7 additions & 2 deletions test/core/paths.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,15 @@ import * as assert from "assert";
import * as fs from "fs";
import * as path from "path";
import * as vscode from "vscode";
import { IPowerShellExtensionClient } from "../../src/features/ExternalApi";
import utils = require("../utils");

describe("Path assumptions", function () {
before(utils.ensureEditorServicesIsConnected);
let storageUri: vscode.Uri;
before(async () => {
const extension: IPowerShellExtensionClient = await utils.ensureEditorServicesIsConnected();
storageUri = extension.getStorageUri();
});

// TODO: This is skipped because it interferes with other tests. Either
// need to find a way to close the opened folder via a Code API, or find
Expand All @@ -22,6 +27,6 @@ describe("Path assumptions", function () {
});

it("Creates the log folder at the correct path", function () {
assert(fs.existsSync(path.resolve(utils.rootPath, "logs")));
assert(fs.existsSync(vscode.Uri.joinPath(storageUri, "logs").fsPath));
});
});
34 changes: 16 additions & 18 deletions test/features/ExternalApi.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,51 +7,50 @@ import { IExternalPowerShellDetails, IPowerShellExtensionClient } from "../../sr

describe("ExternalApi feature", function () {
describe("External extension registration", function () {
let powerShellExtensionClient: IPowerShellExtensionClient;
let extension: IPowerShellExtensionClient;
before(async function () {
const powershellExtension = await utils.ensureExtensionIsActivated();
powerShellExtensionClient = powershellExtension!.exports as IPowerShellExtensionClient;
extension = await utils.ensureExtensionIsActivated();
});

it("Registers and unregisters an extension", function () {
const sessionId: string = powerShellExtensionClient.registerExternalExtension(utils.extensionId);
const sessionId: string = extension.registerExternalExtension(utils.extensionId);
assert.notStrictEqual(sessionId, "");
assert.notStrictEqual(sessionId, null);
assert.strictEqual(
powerShellExtensionClient.unregisterExternalExtension(sessionId),
extension.unregisterExternalExtension(sessionId),
true);
});

it("Registers and unregisters an extension with a version", function () {
const sessionId: string = powerShellExtensionClient.registerExternalExtension(utils.extensionId, "v2");
const sessionId: string = extension.registerExternalExtension(utils.extensionId, "v2");
assert.notStrictEqual(sessionId, "");
assert.notStrictEqual(sessionId, null);
assert.strictEqual(
powerShellExtensionClient.unregisterExternalExtension(sessionId),
extension.unregisterExternalExtension(sessionId),
true);
});

it("Rejects if not registered", async function () {
assert.rejects(
async () => await powerShellExtensionClient.getPowerShellVersionDetails(""))
async () => await extension.getPowerShellVersionDetails(""))
});

it("Throws if attempting to register an extension more than once", async function () {
const sessionId: string = powerShellExtensionClient.registerExternalExtension(utils.extensionId);
const sessionId: string = extension.registerExternalExtension(utils.extensionId);
try {
assert.throws(
() => powerShellExtensionClient.registerExternalExtension(utils.extensionId),
() => extension.registerExternalExtension(utils.extensionId),
{
message: `The extension '${utils.extensionId}' is already registered.`
});
} finally {
powerShellExtensionClient.unregisterExternalExtension(sessionId);
extension.unregisterExternalExtension(sessionId);
}
});

it("Throws when unregistering an extension that isn't registered", async function () {
assert.throws(
() => powerShellExtensionClient.unregisterExternalExtension("not-real"),
() => extension.unregisterExternalExtension("not-real"),
{
message: `No extension registered with session UUID: not-real`
});
Expand All @@ -60,18 +59,17 @@ describe("ExternalApi feature", function () {

describe("PowerShell version details", () => {
let sessionId: string;
let powerShellExtensionClient: IPowerShellExtensionClient;
let extension: IPowerShellExtensionClient;

before(async function () {
const powershellExtension = await utils.ensureExtensionIsActivated();
powerShellExtensionClient = powershellExtension!.exports as IPowerShellExtensionClient;
sessionId = powerShellExtensionClient.registerExternalExtension(utils.extensionId);
extension = await utils.ensureExtensionIsActivated();
sessionId = extension.registerExternalExtension(utils.extensionId);
});

after(function () { powerShellExtensionClient.unregisterExternalExtension(sessionId); });
after(function () { extension.unregisterExternalExtension(sessionId); });

it("Gets non-empty version details from the PowerShell Editor Services", async function () {
const versionDetails: IExternalPowerShellDetails = await powerShellExtensionClient.getPowerShellVersionDetails(sessionId);
const versionDetails: IExternalPowerShellDetails = await extension.getPowerShellVersionDetails(sessionId);

assert.notStrictEqual(versionDetails.architecture, "");
assert.notStrictEqual(versionDetails.architecture, null);
Expand Down
16 changes: 8 additions & 8 deletions test/utils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,16 +14,16 @@ export const rootPath = path.resolve(__dirname, "../../")
const packageJSON: any = require(path.resolve(rootPath, "package.json"));
export const extensionId = `${packageJSON.publisher}.${packageJSON.name}`;

export async function ensureExtensionIsActivated(): Promise<vscode.Extension<any>> {
export async function ensureExtensionIsActivated(): Promise<IPowerShellExtensionClient> {
const extension = vscode.extensions.getExtension(extensionId);
if (!extension.isActive) { await extension.activate(); }
return extension;
return extension!.exports as IPowerShellExtensionClient;
}

export async function ensureEditorServicesIsConnected(): Promise<void> {
const powershellExtension = await ensureExtensionIsActivated();
const client = powershellExtension!.exports as IPowerShellExtensionClient;
const sessionId = client.registerExternalExtension(extensionId);
await client.waitUntilStarted(sessionId);
client.unregisterExternalExtension(sessionId);
export async function ensureEditorServicesIsConnected(): Promise<IPowerShellExtensionClient> {
const extension = await ensureExtensionIsActivated();
const sessionId = extension.registerExternalExtension(extensionId);
await extension.waitUntilStarted(sessionId);
extension.unregisterExternalExtension(sessionId);
return extension;
}