refactor: Extracted getBuffer()

Author: ijlee2

src/parse/preprocess.ts

@@ -1,4 +1,4 @@
-import { parse, type Range } from '../utils/content-tag.js';
+import { getBuffer, parse, type Range } from '../utils/content-tag.js';
 
 export interface Template {
   contents: string;
@@ -10,19 +10,8 @@ export interface Template {
   };
 }
 
-const BufferMap: Map<string, Buffer> = new Map();
-
 const PLACEHOLDER = '~';
 
-function getBuffer(s: string): Buffer {
-  let buf = BufferMap.get(s);
-  if (!buf) {
-    buf = Buffer.from(s);
-    BufferMap.set(s, buf);
-  }
-  return buf;
-}
-
 /** Slice string using byte range */
 function sliceByteRange(s: string, a: number, b?: number): string {
   const buf = getBuffer(s);
@@ -35,11 +24,6 @@ function byteToCharIndex(s: string, byteOffset: number): number {
   return buf.subarray(0, byteOffset).toString().length;
 }
 
-/** Calculate byte length */
-function byteLength(s: string): number {
-  return getBuffer(s).length;
-}
-
 function replaceRange(
   s: string,
   start: number,
@@ -84,7 +68,7 @@ export function preprocessTemplateRange(
 
   const tplLength = template.range.end - template.range.start;
   const spaces =
-    tplLength - byteLength(content) - prefix.length - suffix.length;
+    tplLength - getBuffer(content).length - prefix.length - suffix.length;
   const total = prefix + content + ' '.repeat(spaces) + suffix;
 
   return replaceRange(code, template.range.start, template.range.end, total);

src/utils/content-tag.ts

@@ -10,6 +10,19 @@ type Range = {
   start: number;
 };
 
+const BufferMap = new Map<string, Buffer>();
+
+export function getBuffer(string_: string): Buffer {
+  let buffer = BufferMap.get(string_);
+
+  if (!buffer) {
+    buffer = Buffer.from(string_);
+    BufferMap.set(string_, buffer);
+  }
+
+  return buffer;
+}
+
 export function parse(
   file: string,
   options?: PreprocessorOptions,