enhance(card-cm): replace export column with export selection

Author: ycnmhd

src/lang/lang.ts

@@ -140,7 +140,11 @@ export const lang = {
 
     // export
     cm_export_document: 'Export document',
-    cm_export_column: 'Export column',
+    cm_export_selection: 'Export selection',
+    cm_export_with_subitems: 'With sub-items',
+    cm_export_wo_subitems: 'Without sub-items',
+    cmd_export_selection_with_subitems: 'Export selection with sub-items',
+    cmd_export_selection_wo_subitems: 'Export selection without sub-items',
 
     // document format
     settings_general_default_format: 'Default format',

src/obsidian/commands/add-commands.ts

@@ -11,7 +11,7 @@ import { openSplitNodeModal } from 'src/view/modals/split-node-modal/open-split-
 import { isEditing } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/is-editing';
 import { copyLinkToBlock } from 'src/view/actions/context-menu/card-context-menu/helpers/copy-link-to-block';
 import { extractBranch } from 'src/obsidian/commands/helpers/extract-branch/extract-branch';
-import { exportColumn } from 'src/view/actions/context-menu/card-context-menu/helpers/export-column';
+import { exportSelection } from 'src/view/actions/context-menu/card-context-menu/helpers/export-selection';
 import { exportDocument } from 'src/obsidian/commands/helpers/export-document/export-document';
 import { onPluginError } from 'src/lib/store/on-plugin-error';
 import invariant from 'tiny-invariant';
@@ -165,14 +165,26 @@ const createCommands = (plugin: Lineage) => {
     });
 
     commands.push({
-        name: lang.cm_export_column,
+        name: lang.cmd_export_selection_with_subitems,
         icon: 'file-text',
         checkCallback: (checking) => {
             const view = getActiveLineageView(plugin);
             if (checking) {
                 return Boolean(view);
             }
-            exportColumn(view!);
+            exportSelection(view!, true);
+        },
+    });
+
+    commands.push({
+        name: lang.cmd_export_selection_wo_subitems,
+        icon: 'file-text',
+        checkCallback: (checking) => {
+            const view = getActiveLineageView(plugin);
+            if (checking) {
+                return Boolean(view);
+            }
+            exportSelection(view!, false);
         },
     });
 

src/view/actions/context-menu/card-context-menu/helpers/export-column.ts

@@ -0,0 +1,45 @@
+import { LineageView } from 'src/view/view';
+import { createNewFile } from 'src/obsidian/events/workspace/effects/create-new-file';
+import { openFile } from 'src/obsidian/events/workspace/effects/open-file';
+import { saveNodeContent } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/save-node-content';
+import { getTextOfFlatNodes } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/get-text-of-flat-nodes';
+import { getActiveNodes } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/get-active-nodes';
+import { mapBranchesToText } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/map-branches-to-text';
+
+export const exportSelection = async (
+    view: LineageView,
+    includeSubItems: boolean,
+) => {
+    const viewState = view.viewStore.getValue();
+
+    const isEditing = Boolean(viewState.document.editing.activeNodeId);
+    if (isEditing) {
+        saveNodeContent(view);
+        setTimeout(() => {
+            exportSelection(view, includeSubItems);
+        }, 100);
+        return;
+    }
+
+    let text = '';
+
+    const nodes = getActiveNodes(view, false);
+    if (includeSubItems) {
+        text = mapBranchesToText(
+            view.documentStore.getValue().document,
+            nodes,
+            'unformatted-text',
+        );
+    } else {
+        text = getTextOfFlatNodes(view, nodes, false);
+    }
+
+    const file = view.file!;
+    const newFile = await createNewFile(
+        view.plugin,
+        file.parent!,
+        text,
+        `${file.basename} - exported selection`,
+    );
+    await openFile(view.plugin, newFile, 'split');
+};

src/view/actions/context-menu/card-context-menu/helpers/export-selection.ts

@@ -8,7 +8,7 @@ import { pasteNode } from 'src/view/actions/keyboard-shortcuts/helpers/commands/
 import { openSplitNodeModal } from 'src/view/modals/split-node-modal/open-split-node-modal';
 import { customIcons } from 'src/helpers/load-custom-icons';
 import { copyLinkToBlock } from 'src/view/actions/context-menu/card-context-menu/helpers/copy-link-to-block';
-import { exportColumn } from 'src/view/actions/context-menu/card-context-menu/helpers/export-column';
+import { exportSelection } from 'src/view/actions/context-menu/card-context-menu/helpers/export-selection';
 import {
     MenuItemObject,
     renderContextMenu,
@@ -181,10 +181,21 @@ export const showCardContextMenu = (event: MouseEvent, view: LineageView) => {
             disabled: multipleNodesAreSelected || isInSidebar,
         },
         {
-            title: lang.cm_export_column,
+            title: lang.cm_export_selection,
             icon: 'file-text',
-            action: () => exportColumn(view),
-            disabled: multipleNodesAreSelected || isInSidebar,
+            disabled: isInSidebar,
+            submenu: [
+                {
+                    title: lang.cm_export_with_subitems,
+                    icon: 'file-text',
+                    action: () => exportSelection(view, true),
+                },
+                {
+                    title: lang.cm_export_wo_subitems,
+                    icon: 'file-text',
+                    action: () => exportSelection(view, false),
+                },
+            ],
         },
     ];
 

src/view/actions/context-menu/card-context-menu/show-card-context-menu.ts

@@ -1,11 +1,11 @@
 import { LineageView } from 'src/view/view';
 import { getActiveNodes } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/get-active-nodes';
-import { copyNodesToClipboard } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-nodes-to-clipboard';
+import { copyFlatNodesToClipboard } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-flat-nodes-to-clipboard';
 
 export const copyActiveNodesToClipboard = async (
     view: LineageView,
     isInSidebar: boolean,
 ) => {
     const nodes = getActiveNodes(view, isInSidebar);
-    await copyNodesToClipboard(view, nodes);
+    await copyFlatNodesToClipboard(view, nodes);
 };

src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-active-nodes-to-clipboard.ts

@@ -0,0 +1,11 @@
+import { LineageView } from 'src/view/view';
+import { getTextOfFlatNodes } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/get-text-of-flat-nodes';
+
+export const copyFlatNodesToClipboard = async (
+    view: LineageView,
+    nodes: string[],
+    copyAsOutline = false,
+) => {
+    const text = getTextOfFlatNodes(view, nodes, copyAsOutline);
+    await navigator.clipboard.writeText(text);
+};

src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-flat-nodes-to-clipboard.ts

@@ -1,12 +1,8 @@
 import { LineageView } from 'src/view/view';
-import { sortNodeIdsBySectionNumber } from 'src/lib/tree-utils/sort/sort-node-ids-by-section-number';
-import { copyNodesToClipboard } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-nodes-to-clipboard';
+import { copyFlatNodesToClipboard } from 'src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-flat-nodes-to-clipboard';
 
 export const copyFlatSearchResultsToClipboard = async (view: LineageView) => {
     const results = Array.from(view.viewStore.getValue().search.results.keys());
-    const sortedResults = sortNodeIdsBySectionNumber(
-        view.documentStore.getValue().sections,
-        results,
-    );
-    await copyNodesToClipboard(view, sortedResults, true);
+
+    await copyFlatNodesToClipboard(view, results, true);
 };

src/view/actions/keyboard-shortcuts/helpers/commands/commands/helpers/clipboard/copy-flat-search-results-to-clipboard.ts

@@ -0,0 +1,22 @@
+import { LineageView } from 'src/view/view';
+import { sortNodeIdsBySectionNumber } from 'src/lib/tree-utils/sort/sort-node-ids-by-section-number';
+
+export const getTextOfFlatNodes = (
+    view: LineageView,
+    nodes: string[],
+    copyAsOutline = false,
+) => {
+    if (nodes.length === 1) copyAsOutline = false;
+    const documentState = view.documentStore.getValue();
+    const documentContent = documentState.document.content;
+    const sortedNodes = sortNodeIdsBySectionNumber(
+        documentState.sections,
+        nodes,
+    );
+    return sortedNodes
+        .map((id) => {
+            const content = documentContent[id].content;
+            return (copyAsOutline ? '- ' : '') + content;
+        })
+        .join(copyAsOutline ? '\n' : '\n\n');
+};