Merge pull request #139 from diffix/cristian/misc

Set a default file path for export dialog.

Author: cristianberneanu

src/AnonymizationStep/AnonymizationStep.tsx

@@ -63,7 +63,8 @@ function AnonymizationSummary({ result, loading }: CommonProps) {
 }
 
 async function exportResult(schema: TableSchema, bucketColumns: BucketColumn[]) {
-  const filePath = await window.selectExportFile();
+  const defaultPath = schema.file.path.replace(/\.\w*$/, '') + '_anonymized.csv';
+  const filePath = await window.selectExportFile(defaultPath);
   if (!filePath) return;
 
   message.loading({ content: `Exporting anonymized data to ${filePath}...`, key: filePath, duration: 0 });

src/FileLoadStep/FileLoadStep.tsx

@@ -28,7 +28,7 @@ export const FileLoadStep: FunctionComponent<FileLoadStepProps> = ({ children, o
         <NotebookNavAnchor step={NotebookNavStep.CsvImport} status={file ? 'done' : 'active'} />
         <Title level={3}>Import data to anonymize</Title>
         <Dragger
-          accept=".csv"
+          accept=".csv,.tsv,.txt"
           fileList={[]}
           beforeUpload={(file) => {
             // Uploader is mid state update. We push the op to next frame to avoid react warning.

src/main.ts

@@ -94,8 +94,9 @@ ipcMain.handle('call_service', (_event, taskId: string, request: string) =>
   }),
 );
 
-ipcMain.handle('select_export_file', async (_event) => {
+ipcMain.handle('select_export_file', async (_event, defaultPath: string) => {
   const options = {
+    defaultPath: defaultPath,
     filters: [
       { name: 'CSV', extensions: ['csv'] },
       { name: 'All Files', extensions: ['*'] },

src/preload.ts

@@ -27,8 +27,8 @@ window.callService = (request: unknown, signal: AbortSignal) =>
     return json ? JSON.parse(json) : null;
   });
 
-window.selectExportFile = () => {
-  return ipcRenderer.invoke('select_export_file');
+window.selectExportFile = (defaultPath: string) => {
+  return ipcRenderer.invoke('select_export_file', defaultPath);
 };
 
 window.hashFile = (fileName: string, signal: AbortSignal) =>

src/types/index.ts

@@ -114,7 +114,7 @@ export {};
 declare global {
   interface Window {
     callService(request: unknown, signal: AbortSignal): Promise<QueryResult>;
-    selectExportFile(): Promise<string | null>;
+    selectExportFile(defaultPath: string): Promise<string | null>;
     hashFile(fileName: string, signal: AbortSignal): Promise<string>;
   }
 }