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

Commit 2d4f8d4

Browse files
author
Alan Shaw
committed
refactor: pass cidBase arg to core
License: MIT Signed-off-by: Alan Shaw <[email protected]>
1 parent 76a1153 commit 2d4f8d4

File tree

19 files changed

+322
-111
lines changed

19 files changed

+322
-111
lines changed

‎src/cli/commands/bitswap/unwant.js‎

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
'use strict'
22

3-
const print = require('../../utils').print
3+
const multibase = require('multibase')
4+
const { print } = require('../../utils')
5+
const { cidToString } = require('../../../utils/cid')
46

57
module.exports = {
68
command: 'unwant <key>',
@@ -12,14 +14,19 @@ module.exports = {
1214
alias: 'k',
1315
describe: 'Key to remove from your wantlist',
1416
type: 'string'
17+
},
18+
'cid-base': {
19+
describe: 'Number base to display CIDs in.',
20+
type: 'string',
21+
choices: multibase.names
1522
}
1623
},
1724
handler (argv) {
1825
argv.ipfs.bitswap.unwant(argv.key, (err) => {
1926
if (err) {
2027
throw err
2128
}
22-
print(`Key ${argv.key} removed from wantlist`)
29+
print(`Key ${cidToString(argv.key, argv.cidBase)} removed from wantlist`)
2330
})
2431
}
2532
}

‎src/cli/commands/bitswap/wantlist.js‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,7 @@ module.exports = {
2828
if (err) {
2929
throw err
3030
}
31-
cids.forEach((cid) => {
32-
print(cid)
33-
})
31+
cids.forEach((cid) => print(cid))
3432
})
3533
}
3634
}

‎src/cli/commands/files/add.js‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@ const waterfall = require('async/waterfall')
1313
const mh = require('multihashes')
1414
const multibase = require('multibase')
1515
const { print, isDaemonOn, createProgressBar } = require('../../utils')
16-
const { cidToString } = require('../../../utils/cid')
1716

1817
function checkPath (inPath, recursive) {
1918
// This function is to check for the following possible inputs
@@ -90,7 +89,7 @@ function addPipeline (index, addStream, list, argv) {
9089
sortBy(added, 'path')
9190
.reverse()
9291
.map((file) => {
93-
const log = [ 'added', cidToString(file.hash, argv.cidBase) ]
92+
const log = [ 'added', file.hash ]
9493

9594
if (!quiet && file.path.length > 0) log.push(file.path)
9695

@@ -199,6 +198,7 @@ module.exports = {
199198
? argv.shardSplitThreshold
200199
: Infinity,
201200
cidVersion: argv.cidVersion,
201+
cidBase: argv.cidBase,
202202
rawLeaves: argv.rawLeaves,
203203
onlyHash: argv.onlyHash,
204204
hashAlg: argv.hash,

‎src/cli/commands/ls.js‎

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
const multibase = require('multibase')
44
const { print, rightpad } = require('../utils')
5-
const { cidToString } = require('../../utils/cid')
65

76
module.exports = {
87
command: 'ls <key>',
@@ -35,13 +34,11 @@ module.exports = {
3534
},
3635

3736
handler (argv) {
38-
argv.ipfs.ls(argv.key, { recursive: argv.recursive }, (err, links) => {
37+
argv.ipfs.ls(argv.key, { recursive: argv.recursive, cidBase: argv.cidBase }, (err, links) => {
3938
if (err) {
4039
throw err
4140
}
4241

43-
links = links.map(l => Object.assign(l, { hash: cidToString(l.hash, argv.cidBase) }))
44-
4542
if (argv.headers) {
4643
links = [{hash: 'Hash', size: 'Size', name: 'Name'}].concat(links)
4744
}

‎src/cli/commands/pin/add.js‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
const multibase = require('multibase')
44
const { print } = require('../../utils')
5-
const { cidToString } = require('../../../utils/cid')
65

76
module.exports = {
87
command: 'add <ipfsPath...>',
@@ -24,12 +23,13 @@ module.exports = {
2423
},
2524

2625
handler (argv) {
27-
const recursive = argv.recursive
26+
const { recursive, cidBase } = argv
2827
const type = recursive ? 'recursive' : 'direct'
29-
argv.ipfs.pin.add(argv.ipfsPath, { recursive: recursive }, (err, results) => {
28+
29+
argv.ipfs.pin.add(argv.ipfsPath, { recursive, cidBase }, (err, results) => {
3030
if (err) { throw err }
3131
results.forEach((res) => {
32-
print(`pinned ${cidToString(res.hash, argv.cidBase)} ${type}ly`)
32+
print(`pinned ${res.hash} ${type}ly`)
3333
})
3434
})
3535
}

‎src/cli/commands/pin/ls.js‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
const multibase = require('multibase')
44
const { print } = require('../../utils')
5-
const { cidToString } = require('../../../utils/cid')
65

76
module.exports = {
87
// bracket syntax with '...' tells yargs to optionally accept a list
@@ -33,13 +32,12 @@ module.exports = {
3332

3433
handler: (argv) => {
3534
const paths = argv.ipfsPath
36-
const type = argv.type
37-
const quiet = argv.quiet
35+
const { type, quiet, cidBase } = argv
3836

39-
argv.ipfs.pin.ls(paths, { type }, (err, results) => {
37+
argv.ipfs.pin.ls(paths, { type, cidBase }, (err, results) => {
4038
if (err) { throw err }
4139
results.forEach((res) => {
42-
let line = cidToString(res.hash, argv.cidBase)
40+
let line = res.hash
4341
if (!quiet) {
4442
line += ` ${res.type}`
4543
}

‎src/cli/commands/pin/rm.js‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,9 @@ module.exports = {
2424
},
2525

2626
handler: (argv) => {
27-
const recursive = argv.recursive
28-
argv.ipfs.pin.rm(argv.ipfsPath, { recursive: recursive }, (err, results) => {
27+
const { recursive, cidBase } = argv
28+
29+
argv.ipfs.pin.rm(argv.ipfsPath, { recursive, cidBase }, (err, results) => {
2930
if (err) { throw err }
3031
results.forEach((res) => {
3132
print(`unpinned ${cidToString(res.hash, argv.cidBase)}`)

‎src/cli/commands/resolve.js‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,8 @@ module.exports = {
2222
},
2323

2424
handler (argv) {
25-
argv.ipfs.resolve(argv.name, {
26-
recursive: argv.recursive,
27-
cidBase: argv.cidBase
28-
}, (err, res) => {
25+
const { recursive, cidBase } = argv
26+
argv.ipfs.resolve(argv.name, { recursive, cidBase }, (err, res) => {
2927
if (err) throw err
3028
print(res)
3129
})

‎src/core/components/bitswap.js‎

Lines changed: 22 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,11 @@ const Big = require('big.js')
77
const CID = require('cids')
88
const PeerId = require('peer-id')
99
const errCode = require('err-code')
10+
const multibase = require('multibase')
11+
const { cidToString } = require('../../utils/cid')
1012

11-
function wantlistToCidArray (list) {
12-
return Array.from(list).map((e) => e[1].cid)
13+
function formatWantlist (list, cidBase) {
14+
return Array.from(list).map((e) => cidToString(e[1].cid, cidBase))
1315
}
1416

1517
module.exports = function bitswap (self) {
@@ -24,10 +26,18 @@ module.exports = function bitswap (self) {
2426
options = {}
2527
}
2628

29+
options = options || {}
30+
2731
if (!self.isOnline()) {
2832
return setImmediate(() => callback(new Error(OFFLINE_ERROR)))
2933
}
3034

35+
if (options.cidBase && !multibase.names.includes(options.cidBase)) {
36+
return setImmediate(() => {
37+
callback(errCode(new Error('invalid multibase'), 'ERR_INVALID_MULTIBASE'))
38+
})
39+
}
40+
3141
let list
3242
if (peerId) {
3343
try {
@@ -43,7 +53,7 @@ module.exports = function bitswap (self) {
4353
list = self._bitswap.getWantlist()
4454
}
4555

46-
setImmediate(() => callback(null, wantlistToCidArray(list)))
56+
setImmediate(() => callback(null, formatWantlist(list, options.cidBase)))
4757
}),
4858

4959
stat: promisify((options, callback) => {
@@ -52,17 +62,25 @@ module.exports = function bitswap (self) {
5262
options = {}
5363
}
5464

65+
options = options || {}
66+
5567
if (!self.isOnline()) {
5668
return setImmediate(() => callback(new Error(OFFLINE_ERROR)))
5769
}
5870

71+
if (options.cidBase && !multibase.names.includes(options.cidBase)) {
72+
return setImmediate(() => {
73+
callback(errCode(new Error('invalid multibase'), 'ERR_INVALID_MULTIBASE'))
74+
})
75+
}
76+
5977
const snapshot = self._bitswap.stat().snapshot
6078

6179
setImmediate(() => {
6280
callback(null, {
6381
provideBufLen: parseInt(snapshot.providesBufferLength.toString()),
6482
blocksReceived: new Big(snapshot.blocksReceived),
65-
wantlist: wantlistToCidArray(self._bitswap.getWantlist()),
83+
wantlist: formatWantlist(self._bitswap.getWantlist(), options.cidBase),
6684
peers: self._bitswap.peers().map((id) => id.toB58String()),
6785
dupBlksReceived: new Big(snapshot.dupBlksReceived),
6886
dupDataReceived: new Big(snapshot.dupDataReceived),

‎src/core/components/block.js‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ const waterfall = require('async/waterfall')
77
const setImmediate = require('async/setImmediate')
88
const promisify = require('promisify-es6')
99
const errCode = require('err-code')
10+
const multibase = require('multibase')
11+
const { cidToString } = require('../../utils/cid')
1012

1113
module.exports = function block (self) {
1214
return {
@@ -102,6 +104,12 @@ module.exports = function block (self) {
102104
return setImmediate(() => callback(errCode(err, 'ERR_INVALID_CID')))
103105
}
104106

107+
if (options.cidBase && !multibase.names.includes(options.cidBase)) {
108+
return setImmediate(() => {
109+
callback(errCode(new Error('invalid multibase'), 'ERR_INVALID_MULTIBASE'))
110+
})
111+
}
112+
105113
if (options.preload !== false) {
106114
self._preload(cid)
107115
}
@@ -111,7 +119,7 @@ module.exports = function block (self) {
111119
return callback(err)
112120
}
113121
callback(null, {
114-
key: cid,
122+
key: cidToString(cid, options.cidBase),
115123
size: block.data.length
116124
})
117125
})

0 commit comments

Comments
 (0)