feat(utils): 新增 nextUniqueName

FrontEndDev-org · Apr 15, 2023 · f62c3ec · f62c3ec
1 parent 9f65bbf
commit f62c3ec
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/src/utils/string.ts b/src/utils/string.ts
@@ -49,3 +49,12 @@ export function toTypePath(props: string[]): string {
 export function joinSlices(slices: Array<string | undefined>, separator = '\n') {
   return slices.filter(Boolean).join(separator);
 }
+
+export function nextUniqueName(refName: string, nameCountMap: Map<string, number>) {
+  // abc123 -> abc
+  const baseName = refName.replace(/\d+$/, '');
+  const count = nameCountMap.get(baseName) || 0;
+  const nextCount = count + 1;
+  nameCountMap.set(baseName, nextCount);
+  return nextCount === 1 ? baseName : baseName + nextCount;
+}
diff --git a/test/utils/string.test.ts b/test/utils/string.test.ts
@@ -1,4 +1,13 @@
-import { buildName, findOrigin, joinSlices, RefInfo, refToType, toTypePath, varString } from '../../src/utils/string';
+import {
+  buildName,
+  findOrigin,
+  joinSlices,
+  nextUniqueName,
+  RefInfo,
+  refToType,
+  toTypePath,
+  varString,
+} from '../../src/utils/string';
 
 test('buildName', () => {
   expect(buildName('!')).toEqual('unnamed');
@@ -60,3 +69,11 @@ test('joinSlices', () => {
   expect(joinSlices(['', undefined, ''])).toEqual('');
   expect(joinSlices(['1', undefined, '2', ''])).toEqual('1\n2');
 });
+
+test('nextUniqueName', () => {
+  const map = new Map<string, number>();
+  expect(nextUniqueName('abc', map)).toBe('abc');
+  expect(nextUniqueName('abc', map)).toBe('abc2');
+  expect(nextUniqueName('abc', map)).toBe('abc3');
+  expect(nextUniqueName('abc2', map)).toBe('abc4');
+});