development/vscode
development/vscode
1
0
Fork 0
mirror of https://github.com/Microsoft/vscode synced 2024-08-26 20:39:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

guess notebook file extension.

Browse source
This commit is contained in:
rebornix 2022-02-17 13:54:01 -08:00
parent c7d204f416
commit 23c5190ea0
No known key found for this signature in database
GPG key ID: 181FC90D15393C20
1 changed files with 23 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/vs/workbench/contrib/notebook/common/notebookEditorInput.ts
View file

@ -23,6 +23,7 @@ import { IResourceEditorInput } from 'vs/platform/editor/common/editor';
import { onUnexpectedError } from 'vs/base/common/errors';
import { VSBuffer } from 'vs/base/common/buffer';
import { IWorkingCopyIdentifier } from 'vs/workbench/services/workingCopy/common/workingCopy';
import { NotebookProviderInfo } from 'vs/workbench/contrib/notebook/common/notebookProvider';
export interface NotebookEditorInputOptions {
startDirty?: boolean;
@ -145,7 +146,7 @@ export class NotebookEditorInput extends AbstractResourceEditorInput {
return undefined;
}
const pathCandidate = this.hasCapability(EditorInputCapabilities.Untitled) ? await this._suggestName(this.labelService.getUriBasenameLabel(this.resource)) : this._editorModelReference.object.resource;
const pathCandidate = this.hasCapability(EditorInputCapabilities.Untitled) ? await this._suggestName(provider, this.labelService.getUriBasenameLabel(this.resource)) : this._editorModelReference.object.resource;
let target: URI | undefined;
if (this._editorModelReference.object.hasAssociatedFilePath()) {
target = pathCandidate;
@ -179,7 +180,27 @@ export class NotebookEditorInput extends AbstractResourceEditorInput {
return await this._editorModelReference.object.saveAs(target);
}
private async _suggestName(suggestedFilename: string) {
private async _suggestName(provider: NotebookProviderInfo, suggestedFilename: string) {
// guess file extensions
const firstSelector = provider.selectors[0];
let selectorStr = firstSelector && typeof firstSelector === 'string' ? firstSelector : undefined;
if (!selectorStr && firstSelector) {
const include = (firstSelector as { include?: string }).include;
if (typeof include === 'string') {
selectorStr = include;
}
}
if (selectorStr) {
const matches = /^\*\.([A-Za-z_-]*)$/.exec(selectorStr);
if (matches && matches.length > 1) {
const fileExt = matches[1];
if (!suggestedFilename.endsWith(fileExt)) {
return joinPath(await this._fileDialogService.defaultFilePath(), suggestedFilename + '.' + fileExt);
}
}
}
return joinPath(await this._fileDialogService.defaultFilePath(), suggestedFilename);
}