use Buffer.from when avail

mafintosh · Apr 26, 2018 · eebe20e · eebe20e
1 parent e70eedf
commit eebe20e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -1,8 +1,14 @@
 module.exports = toBuffer
 
+var makeBuffer = Buffer.from && Buffer.from !== Uint8Array.from ? Buffer.from : bufferFrom
+
+function bufferFrom (buf, enc) {
+  return new Buffer(buf, enc)
+}
+
 function toBuffer (buf, enc) {
   if (Buffer.isBuffer(buf)) return buf
-  if (typeof buf === 'string') return new Buffer(buf, enc)
-  if (Array.isArray(buf)) return new Buffer(buf)
+  if (typeof buf === 'string') return makeBuffer(buf, enc)
+  if (Array.isArray(buf)) return makeBuffer(buf)
   throw new Error('Input should be a buffer or a string')
 }