bitdrive-cli/bin/commands/import.js

A Bitspace service that for managing Bitdrives over FUSE.

const p = require('path')

const cliProgress = require('cli-progress')
const { flags } = require('@oclif/command')

const BitdriveServiceCommand = require('../../lib/cli')

class ImportCommand extends BitdriveServiceCommand {
  static usage = 'import [dir] [key]'
  static description = 'Import a directory into a Bitdrive.'
  static args = [
    {
      name: 'dir',
      description: 'The directory you would like to upload to the drive.',
      default: process.cwd(),
      required: true,
      parse: dir => {
        return p.resolve(dir)
      }
    },
    BitdriveServiceCommand.keyArg({
      description: 'The drive key.',
      required: false
    })
  ]
  static flags = {
    'no-seed': flags.boolean({
      description: 'Do not seed the new drive on the Bitdrive network',
      default: false
    }),
    'watch': flags.boolean({
      description: 'Stay running and continue importing new changes.',
      default: false
    })
  }

  async run () {
    const { args, flags } = this.parse(ImportCommand)
    if (flags['no-seed']) flags.noSeed = flags['no-seed']
    await super.run()

    var total = 0
    var uploaded = 0

    const { progress, drive, cleanup } = await this.client.import(args.key, args.dir, flags)

    process.on('SIGINT', onend)
    process.on('SIGTERM', onend)

    const bar = new cliProgress.SingleBar({
      format: `Importing | {bar} | {percentage}% | {value}/{total} Files`
    })
    console.log(`Importing ${args.dir} into ${drive.key.toString('hex')} (Ctrl+c to exit)...`)
    console.log()

    // TODO: This should be upstreamed into mirror-folder.
    progress.on('pending-not-ignored', name => {
      bar.setTotal(++total)
    })
    progress.on('put', () => {
      bar.update(++uploaded)
    })
    progress.on('del', () => {
      bar.update(++uploaded)
    })
    // TODO: This should be upstreamed into mirror-folder.
    progress.on('skip-not-ignored', (src, dst) => {
      bar.update(++uploaded)
    })
    progress.on('end', onend)
    bar.start(1, 0)

    async function onend () {
      // Make sure the events are fully processed.
      await new Promise(resolve => setTimeout(resolve, 500))
      await cleanup()
      console.log('\nImport completed or stopped by user. Exiting...')
      process.exit(0)
    }
  }
}

module.exports = ImportCommand