perf: use 'base64url' encoding when available in Node.js runtime

panva · Jan 18, 2021 · 808f06c · 808f06c
1 parent 0294114
commit 808f06c
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/src/runtime/node/base64url.ts b/src/runtime/node/base64url.ts
@@ -1,12 +1,21 @@
 import { decoder } from '../../lib/buffer_utils.js'
 
-export const encode = (input: Uint8Array | string) =>
-  Buffer.from(input).toString('base64').replace(/=/g, '').replace(/\+/g, '-').replace(/\//g, '_')
+let encodeImpl: (input: Uint8Array | string) => string
 
-export const decode = (input: Uint8Array | string) => {
+function normalize(input: string | Uint8Array) {
   let encoded = input
   if (encoded instanceof Uint8Array) {
     encoded = decoder.decode(encoded)
   }
-  return new Uint8Array(Buffer.from(encoded, 'base64'))
+  return encoded
 }
+
+if (Buffer.isEncoding('base64url')) {
+  encodeImpl = (input) => Buffer.from(input).toString(<BufferEncoding>'base64url')
+} else {
+  encodeImpl = (input) =>
+    Buffer.from(input).toString('base64').replace(/=/g, '').replace(/\+/g, '-').replace(/\//g, '_')
+}
+
+export const encode = encodeImpl
+export const decode = (input: Uint8Array | string) => Buffer.from(normalize(input), 'base64')