perf: improve base64url encoding when available in Node.js

panva · Jan 18, 2021 · d5af559 · d5af559
1 parent 35f5613
commit d5af559
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/lib/help/base64url.js b/lib/help/base64url.js
@@ -1,13 +1,12 @@
-const fromBase64 = (base64) => {
-  return base64.replace(/=/g, '').replace(/\+/g, '-').replace(/\//g, '_')
-}
-
-const encode = (input, encoding = 'utf8') => {
-  return fromBase64(Buffer.from(input, encoding).toString('base64'))
-}
-
-const encodeBuffer = (buf) => {
-  return fromBase64(buf.toString('base64'))
+let encode;
+let encodeBuffer;
+if (Buffer.isEncoding('base64url')) {
+  encode = (input, encoding = 'utf8') => Buffer.from(input, encoding).toString('base64url');
+  encodeBuffer = (buf) => buf.toString('base64url');
+} else {
+  const fromBase64 = (base64) => base64.replace(/=/g, '').replace(/\+/g, '-').replace(/\//g, '_');
+  encode = (input, encoding = 'utf8') => fromBase64(Buffer.from(input, encoding).toString('base64'));
+  encodeBuffer = (buf) => fromBase64(buf.toString('base64'));
 }
 
 const decodeToBuffer = (input) => {