feat: copy row as CSV, closes #394

antares-sql · Jul 29, 2022 · 1c3d7aa · 1c3d7aa
1 parent 664d18e
commit 1c3d7aa
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/renderer/components/WorkspaceTabQueryTable.vue b/src/renderer/components/WorkspaceTabQueryTable.vue
@@ -449,6 +449,18 @@ const copyRow = (format: string) => {
       }
       navigator.clipboard.writeText(sqlInserts.join('\n'));
    }
+   else if (format === 'csv') {
+      const csv = [];
+      if (!Array.isArray(contentToCopy)) contentToCopy = [contentToCopy];
+
+      if (contentToCopy.length)
+         csv.push(Object.keys(contentToCopy[0]).join(';'));
+
+      for (const row of contentToCopy)
+         csv.push(Object.values(row).map(col => typeof col === 'string' ? `"${col}"` : col).join(';'));
+
+      navigator.clipboard.writeText(csv.join('\n'));
+   }
 };
 
 const duplicateRow = () => {

diff --git a/src/renderer/components/WorkspaceTabQueryTableContext.vue b/src/renderer/components/WorkspaceTabQueryTableContext.vue
@@ -21,6 +21,11 @@
                   <i class="mdi mdi-18px mdi-table-row text-light pr-1" /> {{ t('word.row', selectedRows.length) }} (JSON)
                </span>
             </div>
+            <div class="context-element" @click="copyRow('csv')">
+               <span class="d-flex">
+                  <i class="mdi mdi-18px mdi-table-row text-light pr-1" /> {{ t('word.row', selectedRows.length) }} (CSV)
+               </span>
+            </div>
             <div class="context-element" @click="copyRow('sql')">
                <span class="d-flex">
                   <i class="mdi mdi-18px mdi-table-row text-light pr-1" /> {{ t('word.row', selectedRows.length) }} (SQL INSERT)