Skip to content
This repository has been archived by the owner on Feb 12, 2024. It is now read-only.

Commit

Permalink
feat: pipe to add
Browse files Browse the repository at this point in the history
Allows `echo "hello" | jsipfs add`

resolves #1829

License: MIT
Signed-off-by: Alan Shaw <alan.shaw@protocol.ai>
  • Loading branch information
Alan Shaw committed Jan 19, 2019
1 parent 7d9b006 commit 507d161
Show file tree
Hide file tree
Showing 3 changed files with 32 additions and 17 deletions.
35 changes: 19 additions & 16 deletions src/cli/commands/add.js
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ const byteman = require('byteman')
const reduce = require('async/reduce')
const mh = require('multihashes')
const multibase = require('multibase')
const toPull = require('stream-to-pull-stream')
const { print, isDaemonOn, createProgressBar } = require('../utils')
const { cidToString } = require('../../utils/cid')
const globSource = require('../../utils/files/glob-source')
Expand All @@ -20,15 +21,9 @@ function getTotalBytes (paths, cb) {
}, cb)
}

function addPipeline (paths, addStream, options) {
const {
recursive,
quiet,
quieter,
silent
} = options
function addPipeline (source, addStream, options) {
pull(
globSource(...paths, { recursive }),
source,
addStream,
pull.collect((err, added) => {
if (err) {
Expand All @@ -39,14 +34,15 @@ function addPipeline (paths, addStream, options) {
throw err
}

if (silent) return
if (quieter) return print(added.pop().hash)
if (options.silent) return
if (options.quieter) return print(added.pop().hash)

sortBy(added, 'path')
.reverse()
.map((file) => {
const log = [ 'added', cidToString(file.hash, { base: options.cidBase }) ]
if (!quiet && file.path.length > 0) log.push(file.path)
const log = options.quiet ? [] : ['added']
log.push(cidToString(file.hash, { base: options.cidBase }))
if (!options.quiet && file.path.length > 0) log.push(file.path)
return log.join(' ')
})
.forEach((msg) => print(msg))
Expand All @@ -55,7 +51,7 @@ function addPipeline (paths, addStream, options) {
}

module.exports = {
command: 'add <file...>',
command: 'add [file...]',

describe: 'Add a file to IPFS using the UnixFS data format',

Expand Down Expand Up @@ -163,8 +159,15 @@ module.exports = {
throw new Error('Error: Enabling the sharding experiment should be done on the daemon')
}

if (!argv.progress) {
return addPipeline(argv.file, ipfs.addPullStream(options), argv)
const source = argv.file
? globSource(...argv.file, { recursive: argv.recursive })
: toPull.source(process.stdin) // Pipe directly to ipfs.add

const adder = ipfs.addPullStream(options)

// No progress or piping directly to ipfs.add: no need to getTotalBytes
if (!argv.progress || !argv.file) {
return addPipeline(source, adder, argv)
}

getTotalBytes(argv.file, (err, totalBytes) => {
Expand All @@ -176,7 +179,7 @@ module.exports = {
bar.update(byteLength / totalBytes, { progress: byteman(byteLength, 2, 'MB') })
}

addPipeline(argv.file, ipfs.addPullStream(options), argv)
addPipeline(source, adder, argv)
})
}
}
13 changes: 12 additions & 1 deletion test/cli/files.js
Original file line number Diff line number Diff line change
Expand Up @@ -277,13 +277,24 @@ describe('files', () => runOnAndOff((thing) => {
})
})

it('add from pipe', () => {
const proc = ipfs('add')
proc.stdin.write(Buffer.from('hello\n'))
proc.stdin.end()
return proc
.then(out => {
expect(out)
.to.eql('added QmZULkCELmmk5XNfCgTnCyFgAVxBRBXyDHGGMVoLFLiXEN QmZULkCELmmk5XNfCgTnCyFgAVxBRBXyDHGGMVoLFLiXEN\n')
})
})

it('add --quiet', function () {
this.timeout(30 * 1000)

return ipfs('add -q src/init-files/init-docs/readme')
.then((out) => {
expect(out)
.to.eql('added QmPZ9gcCEpqKTo6aq61g2nXGUhM4iCL3ewB6LDXZCtioEB\n')
.to.eql('QmPZ9gcCEpqKTo6aq61g2nXGUhM4iCL3ewB6LDXZCtioEB\n')
})
})

Expand Down
1 change: 1 addition & 0 deletions test/utils/ipfs-exec.js
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@ module.exports = (repoPath, opts) => {
})

res.kill = cp.kill.bind(cp)
res.stdin = cp.stdin
res.stdout = cp.stdout
res.stderr = cp.stderr

Expand Down

0 comments on commit 507d161

Please sign in to comment.