feat(strings): update padLeft/Right() args

- add optional length arg to explicitly specify string length (e.g. to exclude ANSI control seq chars)
thi-ng · Jan 8, 2021 · 118f97f · 118f97f
1 parent bb62760
commit 118f97f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/packages/strings/src/pad-left.ts b/packages/strings/src/pad-left.ts
@@ -9,16 +9,18 @@ import { repeat } from "./repeat";
 export const padLeft: (
     n: number,
     ch?: string | number
-) => Stringer<any> = memoizeJ<
+) => (x: any, length?: number) => string = memoizeJ<
     number,
     string | number | undefined,
     Stringer<any>
 >((n, ch = " ") => {
     const buf = repeat(String(ch), n);
-    return (x: any) =>
-        x != null
-            ? ((x = x.toString()), x.length < n ? buf.substr(x.length) + x : x)
-            : buf;
+    return (x: any, len?: number) => {
+        if (x == null) return buf;
+        x = x.toString();
+        len = len !== undefined ? len : x.length;
+        return len! < n ? buf.substr(len!) + x : x;
+    };
 });
 
 /**

diff --git a/packages/strings/src/pad-right.ts b/packages/strings/src/pad-right.ts
@@ -9,14 +9,16 @@ import { repeat } from "./repeat";
 export const padRight: (
     n: number,
     ch?: string | number
-) => Stringer<any> = memoizeJ<
+) => (x: any, length?: number) => string = memoizeJ<
     number,
     string | number | undefined,
     Stringer<any>
 >((n, ch = " ") => {
     const buf = repeat(String(ch), n);
-    return (x: any) =>
-        x != null
-            ? ((x = x.toString()), x.length < n ? x + buf.substr(x.length) : x)
-            : buf;
+    return (x, len?: number) => {
+        if (x == null) return buf;
+        x = x.toString();
+        len = len !== undefined ? len : x.length;
+        return len! < n ? x + buf.substr(len!) : x;
+    };
 });