Fix cosmos new document could not be saved with default content

Author: Flanker-MSFT

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/actions/CreateNewDocumentAction.java

@@ -27,6 +27,8 @@
 import java.util.UUID;
 import java.util.function.Function;
 
+import static com.microsoft.azure.toolkit.intellij.common.fileexplorer.VirtualFileActions.FILE_CHANGED;
+
 public class CreateNewDocumentAction {
 
     private static final String DEFAULT_CONTENT =
@@ -39,6 +41,7 @@ public class CreateNewDocumentAction {
     public static void create(@Nonnull final ICosmosDocumentContainer<?> container, @Nonnull final Project project) {
         final FileEditorManager fileEditorManager = FileEditorManager.getInstance(project);
         final VirtualFile virtualFile = getOrCreateVirtualFile(container, fileEditorManager);
+        virtualFile.putUserData(FILE_CHANGED, true); // mark file changed to prompt when close with default content
         final Function<String, Boolean> onSave = content -> {
             AzureTaskManager.getInstance().runInBackground(new AzureTask<>(OperationBundle.description("internal/cosmos.create_document.container", container.getName()), () -> {
                 try {
@@ -62,7 +65,7 @@ private static synchronized VirtualFile getOrCreateVirtualFile(@Nonnull final IC
 
     @SneakyThrows
     private static VirtualFile createVirtualFile(@Nonnull final ICosmosDocumentContainer<?> container, final FileEditorManager manager) {
-        final File tempFile = FileUtil.createTempFile(UUID.randomUUID().toString(), ".json", true);
+        final File tempFile = FileUtil.createTempFile(String.format("%s:New Document", container.getName()), ".json", true);
         FileUtil.writeToFile(tempFile, DEFAULT_CONTENT);
         return VirtualFileActions.createVirtualFile(container.getId(), tempFile.getName(), tempFile, manager);
     }

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-lib/src/main/java/com/microsoft/azure/toolkit/intellij/common/fileexplorer/VirtualFileActions.java

@@ -48,8 +48,8 @@
 import java.util.function.Function;
 
 public class VirtualFileActions {
+    public static final Key<Boolean> FILE_CHANGED = new Key<>("FILE_CHANGED");
     private static final Key<String> FILE_ID = new Key<>("FILE_ID");
-    private static final Key<Boolean> FILE_CHANGED = new Key<>("FILE_CHANGED");
     private static final String FILE_EDITING = "Save File";
     private static final String SAVE_CHANGES = "Do you want to save your changes?";
     private static final String SUCCESS_DOWNLOADING = "File %s is successfully downloaded to %s.";