Add newFile command (#159)

* add newFile command

* add all file extensions

* add command to newFile menu of vscode

* support untitled document

* add workspace path if workspace is present

Author: tonka3000

.gitattributes

@@ -0,0 +1 @@
+* text=auto

.prettierignore

@@ -0,0 +1,6 @@
+dist/*
+node_modules/*
+*.json
+*.code-snippets
+webview/dist/*
+webview/node_modules/*

package.json

@@ -39,6 +39,18 @@
                 "title": "Color Theme",
                 "icon": "$(color-mode)"
             },
+            {
+                "command": "excalidraw.newFile",
+                "category": "Excalidraw",
+                "title": "Scene",
+                "icon": "$(color-mode)"
+            },
+            {
+                "command": "excalidraw.newSceneFile",
+                "category": "Excalidraw",
+                "title": "New File",
+                "icon": "$(color-mode)"
+            },
             {
                 "command": "excalidraw.preventDefault",
                 "category": "Excalidraw",
@@ -245,6 +257,10 @@
         ],
         "menus": {
             "commandPalette": [
+                {
+                    "command": "excalidraw.newFile",
+                    "when": "false"
+                },
                 {
                     "command": "excalidraw.updateTheme",
                     "when": "activeCustomEditorId == 'editor.excalidraw'"
@@ -302,6 +318,11 @@
                     "when": "activeCustomEditorId == 'editor.excalidraw'",
                     "group": "navigation"
                 }
+            ],
+            "file/newFile": [
+                {
+                    "command": "excalidraw.newFile"
+                }
             ]
         },
         "customEditors": [
@@ -347,6 +368,8 @@
         "package": "npm run build-webview && webpack --mode production --devtool hidden-source-map",
         "vscode:prepublish": "npm run package",
         "lint": "eslint --ext .ts src",
+        "prettier": "prettier --check .",
+        "prettierfix": "prettier --write .",
         "open-in-browser": "vscode-test-web --extensionDevelopmentPath=. ../examples"
     },
     "capabilities": {

src/commands.ts

@@ -1,4 +1,5 @@
 import * as vscode from "vscode";
+import { newUntitledExcalidrawDocument } from "./utils";
 
 function getConfigurationScope(
   config: vscode.WorkspaceConfiguration,
@@ -89,7 +90,21 @@ function showImage(uri: vscode.Uri, viewColumn?: vscode.ViewColumn) {
   );
 }
 
+async function newFile() {
+  try {
+    await newUntitledExcalidrawDocument();
+  } catch (error) {
+    vscode.window.showErrorMessage(`Failed to create new file: ${error}`);
+  }
+}
+
 export function registerCommands(context: vscode.ExtensionContext) {
+  context.subscriptions.push(
+    vscode.commands.registerCommand("excalidraw.newFile", newFile)
+  );
+  context.subscriptions.push(
+    vscode.commands.registerCommand("excalidraw.newSceneFile", newFile)
+  );
   context.subscriptions.push(
     vscode.commands.registerCommand("excalidraw.updateTheme", updateTheme)
   );
@@ -118,6 +133,6 @@ export function registerCommands(context: vscode.ExtensionContext) {
     )
   );
   context.subscriptions.push(
-    vscode.commands.registerCommand("excalidraw.preventDefault", () => {})
+    vscode.commands.registerCommand("excalidraw.preventDefault", () => { })
   );
 }

src/editor.ts

@@ -84,9 +84,16 @@ export class ExcalidrawEditorProvider
     uri: vscode.Uri,
     openContext: vscode.CustomDocumentOpenContext
   ): Promise<ExcalidrawDocument> {
-    const content = await vscode.workspace.fs.readFile(
-      openContext.backupId ? vscode.Uri.parse(openContext.backupId) : uri
-    );
+    let content: Uint8Array;
+    if (uri.scheme === "untitled") {
+      content = new TextEncoder().encode(
+        JSON.stringify({ type: "excalidraw", elements: [] })
+      );
+    } else {
+      content = await vscode.workspace.fs.readFile(
+        openContext.backupId ? vscode.Uri.parse(openContext.backupId) : uri
+      );
+    }
     const document = new ExcalidrawDocument(uri, content);
 
     const onDidDocumentChange = document.onDidContentChange(() => {

src/utils.ts

@@ -0,0 +1,35 @@
+import * as vscode from "vscode";
+import * as path from 'path';
+
+export function getActiveWorkspace() {
+    const activeEditor = vscode.window.activeTextEditor;
+    if (activeEditor) {
+        const doc = activeEditor.document;
+        const ws = vscode.workspace.getWorkspaceFolder(doc.uri);
+        return ws;
+    }
+
+    const wsf = vscode.workspace.workspaceFolders;
+    if (wsf && wsf.length > 0) {
+        const ws = wsf[0];
+        return ws;
+    }
+    return undefined;
+}
+
+let runningCounter = 0;
+
+export async function newUntitledExcalidrawDocument() {
+    runningCounter += 1;
+    const ws = getActiveWorkspace();
+    let fileName = `Untitled-${runningCounter}.excalidraw`;
+    if (ws) {
+        fileName = path.join(ws.uri.fsPath, fileName);
+    }
+    const uri = vscode.Uri.parse(`untitled:${fileName}`);
+    await vscode.commands.executeCommand(
+        "vscode.openWith",
+        uri,
+        "editor.excalidraw"
+    );
+}