Update: add import/export methods

Author: dpaez

lib/clients/drive.js

@@ -1,7 +1,9 @@
+const {resolve} = require('path')
 const grpc = require('@grpc/grpc-js')
 const maybe = require('call-me-maybe')
 const collectStream = require('stream-collector')
 const pump = require('pump')
+const mirrorFolder = require('mirror-folder')
 const codecs = require('codecs')
 const { Writable, Transform } = require('streamx')
 
@@ -27,6 +29,11 @@ const {
   fromDriveInfo,
   toRPCMetadata: toMetadata
 } = require('../common')
+const DriveWatcher = require('../driveWatcher')
+const {
+  importKeyFilePath: IMPORT_KEY_FILE_PATH,
+  exportKeyFilePath: EXPORT_KEY_FILE_PATH
+} = require('../constants')
 
 module.exports = class DriveClient {
   constructor (endpoint, token) {
@@ -98,6 +105,60 @@ module.exports = class DriveClient {
       })
     }))
   }
+
+  import (dir, drive) {
+    if (!dir) {
+      dir = process.cwd()
+    } else {
+      dir = resolve(dir)
+    }
+
+    if (!drive.writable) {
+      console.error('The target drive is not writable!')
+      return process.exit(1)
+    }
+
+    function shouldIgnore (name) {
+      if (!name) return true
+      if (name.indexOf(EXPORT_KEY_FILE_PATH) !== -1) return true
+      else if (name.indexOf(IMPORT_KEY_FILE_PATH) !== -1) return true
+      return false
+    }
+
+    return mirrorFolder(dir, { fs: drive, name: '/' }, {
+      watch: true,
+      dereference: true,
+      keepExisting: true,
+      ignore: (file, stat, cb) => {
+        if (shouldIgnore(file)) return process.nextTick(cb, null, true)
+        return process.nextTick(cb, null, false)
+      }
+    })
+  }
+
+  export (drive, dir, opts = { recursive: false }) {
+    if (!drive) {
+      throw new Error('missing argument: drive')
+    }
+
+    if (!dir) {
+      throw new Error('missing argument: dir')
+    }
+
+    dir = resolve(dir)
+
+    const driveWatcher = new DriveWatcher(this._client, drive, {
+      recursive: opts.recursive
+    })
+
+    mirrorFolder({ fs: drive, name: '/' }, dir, {
+      watch: driveWatcher.watch.bind(driveWatcher),
+      keepExisting: true,
+      ensureParents: true
+    })
+
+    return driveWatcher
+  }
 }
 
 class RemoteHyperdrive {