package: use typeforce

Author: dcousens

package.json

@@ -51,7 +51,8 @@
     "bigi": "^1.1.0",
     "bs58check": "1.0.3",
     "crypto-browserify": "^3.2.6",
-    "ecurve": "1.0.0"
+    "ecurve": "1.0.0",
+    "typeforce": "0.0.2"
   },
   "devDependencies": {
     "async": "^0.9.0",

src/address.js

@@ -1,6 +1,6 @@
 var assert = require('assert')
 var base58check = require('bs58check')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var networks = require('./networks')
 var scripts = require('./scripts')
 
@@ -14,7 +14,7 @@ function findScriptTypeByVersion(version) {
 }
 
 function Address(hash, version) {
-  enforceType('Buffer', hash)
+  typeForce('Buffer', hash)
 
   assert.strictEqual(hash.length, 20, 'Invalid hash length')
   assert.strictEqual(version & 0xff, version, 'Invalid version byte')

src/ecdsa.js

@@ -1,6 +1,6 @@
 var assert = require('assert')
 var crypto = require('crypto')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 
 var BigInteger = require('bigi')
 var ECSignature = require('./ecsignature')
@@ -10,8 +10,8 @@ var ONE = new Buffer([1])
 
 // https://tools.ietf.org/html/rfc6979#section-3.2
 function deterministicGenerateK(curve, hash, d) {
-  enforceType('Buffer', hash)
-  enforceType(BigInteger, d)
+  typeForce('Buffer', hash)
+  typeForce('BigInteger', d)
 
   // sanity check
   assert.equal(hash.length, 32, 'Hash must be 256 bit')

src/eckey.js

@@ -2,7 +2,7 @@ var assert = require('assert')
 var base58check = require('bs58check')
 var crypto = require('crypto')
 var ecdsa = require('./ecdsa')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var networks = require('./networks')
 
 var BigInteger = require('bigi')
@@ -50,7 +50,7 @@ ECKey.makeRandom = function(compressed, rng) {
   rng = rng || crypto.randomBytes
 
   var buffer = rng(32)
-  enforceType('Buffer', buffer)
+  typeForce('Buffer', buffer)
   assert.equal(buffer.length, 32, 'Expected 256-bit Buffer from RNG')
 
   var d = BigInteger.fromBuffer(buffer)

src/ecpubkey.js

@@ -1,6 +1,6 @@
 var crypto = require('./crypto')
 var ecdsa = require('./ecdsa')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var networks = require('./networks')
 
 var Address = require('./address')
@@ -11,8 +11,8 @@ var secp256k1 = ecurve.getCurveByName('secp256k1')
 function ECPubKey(Q, compressed) {
   if (compressed === undefined) compressed = true
 
-  enforceType(ecurve.Point, Q)
-  enforceType('Boolean', compressed)
+  typeForce('Point', Q)
+  typeForce('Boolean', compressed)
 
   this.compressed = compressed
   this.Q = Q

src/ecsignature.js

@@ -1,11 +1,11 @@
 var assert = require('assert')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 
 var BigInteger = require('bigi')
 
 function ECSignature(r, s) {
-  enforceType(BigInteger, r)
-  enforceType(BigInteger, s)
+  typeForce('BigInteger', r)
+  typeForce('BigInteger', s)
 
   this.r = r
   this.s = s

src/hdnode.js

@@ -2,7 +2,7 @@ var assert = require('assert')
 var base58check = require('bs58check')
 var bcrypto = require('./crypto')
 var crypto = require('crypto')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var networks = require('./networks')
 
 var BigInteger = require('bigi')
@@ -29,7 +29,7 @@ function findBIP32NetworkByVersion(version) {
 function HDNode(K, chainCode, network) {
   network = network || networks.bitcoin
 
-  enforceType('Buffer', chainCode)
+  typeForce('Buffer', chainCode)
 
   assert.equal(chainCode.length, 32, 'Expected chainCode length of 32, got ' + chainCode.length)
   assert(network.bip32, 'Unknown BIP32 constants for network')
@@ -53,7 +53,7 @@ HDNode.HIGHEST_BIT = 0x80000000
 HDNode.LENGTH = 78
 
 HDNode.fromSeedBuffer = function(seed, network) {
-  enforceType('Buffer', seed)
+  typeForce('Buffer', seed)
 
   assert(seed.length >= 16, 'Seed should be at least 128 bits')
   assert(seed.length <= 64, 'Seed should be at most 512 bits')

src/script.js

@@ -1,12 +1,12 @@
 var assert = require('assert')
 var bufferutils = require('./bufferutils')
 var crypto = require('./crypto')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var opcodes = require('./opcodes')
 
 function Script(buffer, chunks) {
-  enforceType('Buffer', buffer)
-  enforceType('Array', chunks)
+  typeForce('Buffer', buffer)
+  typeForce('Array', chunks)
 
   this.buffer = buffer
   this.chunks = chunks
@@ -56,7 +56,7 @@ Script.fromBuffer = function(buffer) {
 }
 
 Script.fromChunks = function(chunks) {
-  enforceType('Array', chunks)
+  typeForce('Array', chunks)
 
   var bufferSize = chunks.reduce(function(accum, chunk) {
     if (Buffer.isBuffer(chunk)) {

src/scripts.js

@@ -1,5 +1,5 @@
 var assert = require('assert')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var ops = require('./opcodes')
 
 var ecurve = require('ecurve')
@@ -117,7 +117,7 @@ function isNullDataOutput(script) {
 }
 
 function classifyOutput(script) {
-  enforceType(Script, script)
+  typeForce('Script', script)
 
   if (isPubKeyHashOutput(script)) {
     return 'pubkeyhash'
@@ -135,7 +135,7 @@ function classifyOutput(script) {
 }
 
 function classifyInput(script) {
-  enforceType(Script, script)
+  typeForce('Script', script)
 
   if (isPubKeyHashInput(script)) {
     return 'pubkeyhash'
@@ -161,7 +161,7 @@ function pubKeyOutput(pubKey) {
 
 // OP_DUP OP_HASH160 {pubKeyHash} OP_EQUALVERIFY OP_CHECKSIG
 function pubKeyHashOutput(hash) {
-  enforceType('Buffer', hash)
+  typeForce('Buffer', hash)
 
   return Script.fromChunks([
     ops.OP_DUP,
@@ -174,7 +174,7 @@ function pubKeyHashOutput(hash) {
 
 // OP_HASH160 {scriptHash} OP_EQUAL
 function scriptHashOutput(hash) {
-  enforceType('Buffer', hash)
+  typeForce('Buffer', hash)
 
   return Script.fromChunks([
     ops.OP_HASH160,
@@ -185,7 +185,7 @@ function scriptHashOutput(hash) {
 
 // m [pubKeys ...] n OP_CHECKMULTISIG
 function multisigOutput(m, pubKeys) {
-  enforceType('Array', pubKeys)
+  typeForce('Array', pubKeys)
 
   assert(pubKeys.length >= m, 'Not enough pubKeys provided')
 
@@ -204,14 +204,14 @@ function multisigOutput(m, pubKeys) {
 
 // {signature}
 function pubKeyInput(signature) {
-  enforceType('Buffer', signature)
+  typeForce('Buffer', signature)
 
   return Script.fromChunks([signature])
 }
 
 // {signature} {pubKey}
 function pubKeyHashInput(signature, pubKey) {
-  enforceType('Buffer', signature)
+  typeForce('Buffer', signature)
 
   return Script.fromChunks([signature, pubKey.toBuffer()])
 }

src/transaction.js

@@ -1,7 +1,7 @@
 var assert = require('assert')
 var bufferutils = require('./bufferutils')
 var crypto = require('./crypto')
-var enforceType = require('./types')
+var typeForce = require('typeforce')
 var opcodes = require('./opcodes')
 var scripts = require('./scripts')
 
@@ -96,7 +96,7 @@ Transaction.prototype.addInput = function(hash, index, sequence, script) {
   if (sequence === undefined || sequence === null) {
     sequence = Transaction.DEFAULT_SEQUENCE
   }
-  
+
   script = script || Script.EMPTY
 
   if (typeof hash === 'string') {
@@ -108,10 +108,10 @@ Transaction.prototype.addInput = function(hash, index, sequence, script) {
 
   }
 
-  enforceType('Buffer', hash)
-  enforceType('Number', index)
-  enforceType('Number', sequence)
-  enforceType(Script, script)
+  typeForce('Buffer', hash)
+  typeForce('Number', index)
+  typeForce('Number', sequence)
+  typeForce('Script', script)
 
   assert.equal(hash.length, 32, 'Expected hash length of 32, got ' + hash.length)
 
@@ -144,8 +144,8 @@ Transaction.prototype.addOutput = function(scriptPubKey, value) {
     scriptPubKey = scriptPubKey.toOutputScript()
   }
 
-  enforceType(Script, scriptPubKey)
-  enforceType('Number', value)
+  typeForce('Script', scriptPubKey)
+  typeForce('Number', value)
 
   // Add the output and return the output's index
   return (this.outs.push({
@@ -197,9 +197,9 @@ Transaction.prototype.hashForSignature = function(inIndex, prevOutScript, hashTy
     prevOutScript = tmp
   }
 
-  enforceType('Number', inIndex)
-  enforceType(Script, prevOutScript)
-  enforceType('Number', hashType)
+  typeForce('Number', inIndex)
+  typeForce('Script', prevOutScript)
+  typeForce('Number', hashType)
 
   assert(inIndex >= 0, 'Invalid vin index')
   assert(inIndex < this.ins.length, 'Invalid vin index')