feat: add flow annotations

.babelrc

@@ -0,0 +1,34 @@
+{
+  "env": {
+    "development": {
+      "sourceMaps": "inline",
+      "comments": false,
+      "presets": [
+        [
+          "env",
+          {
+            "targets": {
+              "node": "current"
+            }
+          }
+        ],
+        "flow-node"
+      ]
+    },
+    "umd": {
+      "comments": false,
+      "presets": [
+        [
+          "env",
+          {
+            "modules": false,
+            "targets": {
+              "browsers": "last 2 versions"
+            }
+          }
+        ],
+        "flow-node"
+      ]
+    }
+  }
+}

.flowconfig

@@ -0,0 +1,9 @@
+[ignore]
+.*/node_modules/documentation/*
+
+[libs]
+
+[include]
+
+[options]
+suppress_comment= \\(.\\|\n\\)*\\@FlowIgnore

package.json

@@ -4,22 +4,38 @@
   "description": "multiple hash functions",
   "main": "src/index.js",
   "scripts": {
-    "lint": "aegir lint",
-    "build": "aegir build",
-    "test": "aegir test -t node -t browser -t webworker",
+    "lint": "lint-staged && npm run type-check",
+    "type-check": "flow check",
+    "build": "npm run build:node && BABEL_ENV=umd aegir build",
+    "test": "npm run build:node && aegir test -t node -t browser -t webworker",
     "test:node": "aegir test -t node",
     "test:browser": "aegir test -t browser",
     "test:webworker": "aegir test -t webworker",
     "release": "aegir release -t node -t browser",
     "release-minor": "aegir release --type minor -t node -t browser",
     "release-major": "aegir release --type major -t node -t browser",
     "coverage": "aegir coverage",
-    "coverage-publish": "aegir coverage --provider coveralls"
+    "coverage-publish": "aegir coverage --provider coveralls",
+    "build:types": "flow-copy-source --verbose src lib",
+    "build:lib": "babel --out-dir lib src",
+    "build:node": "npm run build:types && npm run build:lib",
+    "start": "flow-copy-source --watch --verbose src lib & babel --watch --out-dir lib src"
+  },
+  "lint-staged": {
+    "*.js": [
+      "prettier --no-semi --write",
+      "git add"
+    ]
   },
   "pre-commit": [
     "lint",
     "test"
   ],
+  "standard": {
+    "ignore": [
+      "dist"
+    ]
+  },
   "repository": {
     "type": "git",
     "url": "git://github.com/multiformats/js-multihashing.git"
@@ -33,16 +49,26 @@
     "url": "https://github.com/multiformats/js-multihashing/issues"
   },
   "dependencies": {
+    "babel-cli": "^6.26.0",
+    "babel-core": "^6.26.0",
+    "babel-loader": "^7.1.4",
     "blakejs": "^1.1.0",
     "js-sha3": "^0.7.0",
     "multihashes": "~0.4.12",
     "webcrypto": "~0.1.1"
   },
   "devDependencies": {
     "aegir": "^12.3.0",
+    "babel-preset-flow-node": "^2.0.1",
     "chai": "^4.1.2",
     "dirty-chai": "^2.0.1",
-    "pre-commit": "^1.2.2"
+    "flow-bin": "^0.69.0",
+    "flow-copy-source": "^1.3.0",
+    "lint-staged": "^7.0.2",
+    "pre-commit": "^1.2.2",
+    "prettier": "^1.11.1",
+    "rollup.config.flow": "^1.0.0",
+    "source-map-support": "^0.5.4"
   },
   "homepage": "https://github.com/multiformats/js-multihashing",
   "contributors": [

src/blake.js

@@ -1,56 +1,79 @@
-'use strict'
+// @flow
+"use strict"
 
-const blake = require('blakejs')
-const minB = 0xb201
-const minS = 0xb241
+import * as blake from "blakejs"
+import type { Hash, HashTable, HashBuilder } from "./types"
+import type { Code } from "multihashes/lib/constants"
 
-var blake2b = {
+const minB: Code = 0xb201
+const minS: Code = 0xb241
+
+type BlakeCtx = {
+  b: Uint8Array,
+  h: Uint32Array,
+  t: number,
+  c: number,
+  outlen: number
+}
+type Blake2Hash = Buffer
+type BlakeHasher = {
+  init(size: number, key: ?number): BlakeCtx,
+  update(ctx: BlakeCtx, input: Uint8Array): void,
+  digest(ctx: BlakeCtx): Uint8Array
+}
+
+const blake2b: BlakeHasher = {
   init: blake.blake2bInit,
   update: blake.blake2bUpdate,
   digest: blake.blake2bFinal
 }
 
-var blake2s = {
+blake2b.init(1, 1)
+
+const blake2s: BlakeHasher = {
   init: blake.blake2sInit,
   update: blake.blake2sUpdate,
   digest: blake.blake2sFinal
 }
 
-class B2Hash {
-  constructor (size, hashFunc) {
+class B2Hash implements Hash {
+  ctx: BlakeCtx | null
+  hf: BlakeHasher
+
+  constructor(size, hashFunc) {
     this.hf = hashFunc
     this.ctx = this.hf.init(size, null)
   }
 
-  update (buf) {
+  update(buf: Buffer): Hash {
     if (this.ctx === null) {
-      throw new Error('blake2 context is null. (already called digest?)')
+      throw new Error("blake2 context is null. (already called digest?)")
     }
     this.hf.update(this.ctx, buf)
     return this
   }
 
-  digest () {
+  digest(): Blake2Hash {
     const ctx = this.ctx
     this.ctx = null
+    if (ctx === null) {
+      throw Error("blake2 context is null. (already called digest?)")
+    }
     return Buffer.from(this.hf.digest(ctx))
   }
 }
 
-function addFuncs (table) {
-  function mkFunc (size, hashFunc) {
-    return () => new B2Hash(size, hashFunc)
+export const addFuncs = (table: HashTable) => {
+  const mkFunc = (size: number, hashFunc: BlakeHasher): HashBuilder => {
+    return (): Hash => new B2Hash(size, hashFunc)
   }
 
-  var i
+  // I don't like using any here but the only way I could get the types to work here.
+  let i
   for (i = 0; i < 64; i++) {
-    table[minB + i] = mkFunc(i + 1, blake2b)
+    table[(minB + i: any)] = mkFunc(i + 1, blake2b)
   }
   for (i = 0; i < 32; i++) {
-    table[minS + i] = mkFunc(i + 1, blake2s)
+    table[(minS + i: any)] = mkFunc(i + 1, blake2s)
   }
 }
-
-module.exports = {
-  addFuncs: addFuncs
-}

src/index.js

@@ -1,23 +1,34 @@
-'use strict'
+// @flow
+"use strict"
 
-const multihash = require('multihashes')
-const blake = require('./blake')
-const sha3 = require('./sha3')
-const crypto = require('webcrypto')
+import * as multihash from "multihashes"
+import type { Multihash } from "multihashes"
+import type { Name, Code } from "multihashes/lib/constants"
+import type { HashTable, Hash } from "./types"
+import * as blake from "./blake"
+import * as sha3 from "./sha3"
+import * as crypto from "webcrypto"
 
-const mh = module.exports = Multihashing
+const mh = (module.exports = Multihashing)
 
 mh.Buffer = Buffer // for browser things
 
-function Multihashing (buf, func, length) {
+function Multihashing(
+  buf: Buffer,
+  func: Name | Code,
+  length: number
+): Multihash {
   return multihash.encode(mh.digest(buf, func, length), func, length)
 }
 
 // expose multihash itself, to avoid silly double requires.
 mh.multihash = multihash
 
-mh.digest = function (buf, func, length) {
-  let digest = mh.createHash(func).update(buf).digest()
+mh.digest = function(buf: Buffer, func: Name | Code, length: ?number): Buffer {
+  let digest = mh
+    .createHash(func)
+    .update(buf)
+    .digest()
 
   if (length) {
     digest = digest.slice(0, length)
@@ -26,38 +37,38 @@ mh.digest = function (buf, func, length) {
   return digest
 }
 
-mh.createHash = function (func, length) {
+mh.createHash = function(func: Name | Code): Hash {
   func = multihash.coerceCode(func)
   if (!mh.functions[func]) {
-    throw new Error('multihash function ' + func + ' not yet supported')
+    throw new Error("multihash function " + func + " not yet supported")
   }
 
   return mh.functions[func]()
 }
 
-mh.verify = function (hash, buf) {
+mh.verify = function verify(hash: Multihash, buf: Buffer): boolean {
   const decoded = multihash.decode(hash)
   const encoded = mh(buf, decoded.name, decoded.length)
   return encoded.equals(hash)
 }
 
 mh.functions = {
-  0x11: gsha1,
-  0x12: gsha2256,
-  0x13: gsha2512
+  [0x11]: gsha1,
+  [0x12]: gsha2256,
+  [0x13]: gsha2512
 }
 
 blake.addFuncs(mh.functions)
 sha3.addFuncs(mh.functions)
 
-function gsha1 () {
-  return crypto.createHash('sha1')
+function gsha1(): Hash {
+  return crypto.createHash("sha1")
 }
 
-function gsha2256 () {
-  return crypto.createHash('sha256')
+function gsha2256(): Hash {
+  return crypto.createHash("sha256")
 }
 
-function gsha2512 () {
-  return crypto.createHash('sha512')
+function gsha2512(): Hash {
+  return crypto.createHash("sha512")
 }

src/sha3.js

@@ -1,6 +1,8 @@
-'use strict'
+// @flow
+"use strict"
 
-const sha3 = require('js-sha3')
+import * as sha3 from "js-sha3"
+import type { Hash, HashTable } from "./types"
 
 const functions = [
   [0x14, sha3.sha3_512],
@@ -9,44 +11,53 @@ const functions = [
   [0x17, sha3.sha3_224],
   [0x18, sha3.shake128, 256],
   [0x19, sha3.shake256, 512],
-  [0x1A, sha3.keccak224],
-  [0x1B, sha3.keccak256],
-  [0x1C, sha3.keccak384],
-  [0x1D, sha3.keccak512]
+  [0x1a, sha3.keccak224],
+  [0x1b, sha3.keccak256],
+  [0x1c, sha3.keccak384],
+  [0x1d, sha3.keccak512]
 ]
 
-class Hasher {
-  constructor (hashFunc, arg) {
+type HexString = string
+type Sha3Hash = Buffer
+type ShaHasher = (input: string | Buffer, length?: number) => HexString
+
+class ShaHash implements Hash {
+  hf: ShaHasher
+  input: Buffer | null
+  arg: number
+
+  constructor(hashFunc, arg?: number) {
     this.hf = hashFunc
-    this.arg = arg
+    if (arg) {
+      this.arg = arg
+    }
     this.input = null
   }
 
-  update (buf) {
+  update(buf: Buffer): Hash {
     this.input = buf
     return this
   }
 
-  digest () {
+  digest(): Sha3Hash {
+    if (!this.input) {
+      throw Error("Missing an input to hash")
+    }
     const input = this.input
     const arg = this.arg
-    return Buffer.from(this.hf(input, arg), 'hex')
+    return Buffer.from(this.hf(input, arg), "hex")
   }
 }
 
-function addFuncs (table) {
+export const addFuncs = (table: HashTable) => {
   for (const info of functions) {
     const code = info[0]
     const fn = info[1]
 
     if (info.length === 3) {
-      table[code] = () => new Hasher(fn, info[2])
+      table[code] = () => new ShaHash(fn, info[2])
     } else {
-      table[code] = () => new Hasher(fn)
+      table[code] = () => new ShaHash(fn)
     }
   }
 }
-
-module.exports = {
-  addFuncs: addFuncs
-}