Infer method for b64 to call based on input type

lukechilds · May 23, 2017 · 97a6430 · 97a6430
1 parent f0b9838
commit 97a6430
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -13,11 +13,12 @@ const validateOpts = (opts, chunkMultiple) => {
 };
 
 const b64 = (input, opts) => {
-	if (opts && ['encode', 'decode'].indexOf(opts.method) > -1) {
-		return b64[opts.method](input, opts);
+	if (input instanceof Buffer || typeof input === 'string') {
+		const method = input instanceof Buffer ? 'encode' : 'decode';
+		return b64[method](input, opts);
 	}
 
-	return Promise.reject(new Error('opts.method must be \'encode\' or \'decode\'.'));
+	return Promise.reject(new Error('input must be a buffer or string'));
 };
 
 b64.encode = (input, opts) => new Promise(resolve => {