feat(StringTemplate): 支持截断省略

fjc0k · Sep 11, 2024 · cae281c · cae281c
1 parent ad237a8
commit cae281c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/src/utils/StringTemplate.test.ts b/src/utils/StringTemplate.test.ts
@@ -120,4 +120,22 @@ describe('StringTemplate', () => {
     expect(StringTemplate.render('{{false}}', {}, { onlyCode: true })).toBe('')
     expect(StringTemplate.render('{{0}}', {}, { onlyCode: true })).toBe('0')
   })
+
+  test('支持截断省略', () => {
+    expect(StringTemplate.render('{x#5}', { x: '0123456789' })).toBe('01...')
+    expect(StringTemplate.render('{x#0}', { x: '0123456789' })).toBe(
+      '0123456789',
+    )
+    expect(StringTemplate.render('{x#1}', { x: '0123456789' })).toBe('...')
+    expect(StringTemplate.render('{x#8}', { x: '0123456789' })).toBe('01234...')
+    expect(
+      StringTemplate.render(
+        '{x#8}',
+        { x: '0123456789' },
+        {
+          code: true,
+        },
+      ),
+    ).toBe('01234...')
+  })
 })
diff --git a/src/utils/StringTemplate.ts b/src/utils/StringTemplate.ts
@@ -1,3 +1,5 @@
+import { toSingleLineString } from './toSingleLineString'
+
 export interface StringTemplateRenderOptions {
   /** 是否启用代码渲染，需环境支持 eval */
   code?: boolean
@@ -13,6 +15,7 @@ export class StringTemplate {
    * 渲染字符串模板。语法：
    *
    * - 用 `{key}` 直接替换；
+   * - 用 `{key#10}` 直接替换，取前10个字符省略；
    * - 用 `{key:param1,param2}` 执行函数替换；
    * - 用 `{{key==='test'?'hi':'hello'}}` 执行代码替换（内部使用 eval 实现，需开启选项里的 `code` 参数）。
    *
@@ -44,10 +47,15 @@ export class StringTemplate {
               )
             : enableCode
             ? template.replace(
-                new RegExp(`(?<!\\$)\\{${key}\\}`, 'g'),
-                data[key],
+                new RegExp(`(?<!\\$)\\{${key}(?:#(\\d+))?\\}`, 'g'),
+                (_, len) =>
+                  len ? toSingleLineString(String(data[key]), +len) : data[key],
+              )
+            : template.replace(
+                new RegExp(`{${key}(?:#(\\d+))?\\}`, 'g'),
+                (_, len) =>
+                  len ? toSingleLineString(String(data[key]), +len) : data[key],
               )
-            : template.replaceAll(`{${key}}`, data[key])
       }
     }
     if (enableCode) {