development/vscode
development/vscode
1
0
Fork 0
mirror of https://github.com/Microsoft/vscode synced 2024-07-17 02:57:19 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix for attachment naming within notebook metadata

Browse source
This commit is contained in:
Michael Lively 2022-08-25 15:04:56 -07:00
parent 74acf9c038
commit 16f842ea76
1 changed files with 17 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
extensions/ipynb/src/notebookImagePaste.ts
View file

@ -31,13 +31,13 @@ class CopyPasteEditProvider implements vscode.DocumentPasteEditProvider {
}
// get filename data from paste
const pasteFilename = dataItem.asFile()?.name;
if (!pasteFilename) {
const clipboardFilename = dataItem.asFile()?.name;
if (!clipboardFilename) {
return undefined;
}
const separatorIndex = pasteFilename?.lastIndexOf('.');
const filename = pasteFilename?.slice(0, separatorIndex);
const filetype = pasteFilename?.slice(separatorIndex);
const separatorIndex = clipboardFilename?.lastIndexOf('.');
const filename = clipboardFilename?.slice(0, separatorIndex);
const filetype = clipboardFilename?.slice(separatorIndex);
if (!filename || !filetype) {
return undefined;
}
@ -46,13 +46,12 @@ class CopyPasteEditProvider implements vscode.DocumentPasteEditProvider {
if (!currentCell) {
return undefined;
}
const notebookUri = currentCell.notebook.uri;
// create updated metadata for cell (prep for WorkspaceEdit)
const b64string = encodeBase64(fileDataAsUint8);
const startingAttachments = currentCell.metadata.custom?.attachments;
const newMetadata = buildMetadata(b64string, currentCell, pasteFilename, filetype, startingAttachments);
const [newMetadata, revisedFilename] = buildMetadata(b64string, currentCell, filename, filetype, startingAttachments);
// build edits
const nbEdit = vscode.NotebookEdit.updateCellMetadata(currentCell.index, newMetadata);
@ -62,8 +61,8 @@ class CopyPasteEditProvider implements vscode.DocumentPasteEditProvider {
// create a snippet for paste
const pasteSnippet = new vscode.SnippetString();
pasteSnippet.appendText('![');
pasteSnippet.appendPlaceholder(`${pasteFilename}`);
pasteSnippet.appendText(`](attachment:${pasteFilename})`);
pasteSnippet.appendPlaceholder(`${revisedFilename}`);
pasteSnippet.appendText(`](attachment:${revisedFilename})`);
return { insertText: pasteSnippet, additionalEdit: workspaceEdit };
}
@ -123,27 +122,27 @@ function encodeBase64(buffer: Uint8Array, padded = true, urlSafe = false) {
return output;
}
function buildMetadata(b64: string, cell: vscode.NotebookCell, filename: string, filetype: string, startingAttachments: any): { [key: string]: any } {
function buildMetadata(b64: string, cell: vscode.NotebookCell, filename: string, filetype: string, startingAttachments: any): [{ [key: string]: any }, string] {
const outputMetadata = { ...cell.metadata };
let tempFilename = filename + filetype;
if (!outputMetadata.custom) {
outputMetadata['custom'] = { 'attachments': { [filename]: { 'image/png': b64 } } };
outputMetadata['custom'] = { 'attachments': { [tempFilename]: { 'image/png': b64 } } };
} else if (!outputMetadata.custom.attachments) {
outputMetadata.custom['attachments'] = { [filename]: { 'image/png': b64 } };
outputMetadata.custom['attachments'] = { [tempFilename]: { 'image/png': b64 } };
} else {
for (let appendValue = 2; filename in startingAttachments; appendValue++) {
const objEntries = Object.entries(startingAttachments[filename]);
for (let appendValue = 2; tempFilename in startingAttachments; appendValue++) {
const objEntries = Object.entries(startingAttachments[tempFilename]);
if (objEntries.length) { // check that mime:b64 are present
const [, attachmentb64] = objEntries[0];
if (attachmentb64 !== b64) { // append a "-#" here. same name, diff data. this matches jupyter behavior
filename = filename.concat(`-${appendValue}`) + filetype;
tempFilename = filename.concat(`-${appendValue}`) + filetype;
}
}
}
outputMetadata.custom.attachments[filename] = { 'image/png': b64 };
outputMetadata.custom.attachments[tempFilename] = { 'image/png': b64 };
}
return outputMetadata;
return [outputMetadata, tempFilename];
}
export function imagePasteSetup() {