Merge pull request #7 from lumalabs/theop/uniqueId-v2

Add public setUniqueIdGenerator for custom ID generation

Author: kaznaan

packages/utils/api-report.api.md

@@ -463,6 +463,9 @@ export function setInLocalStorage(key: string, value: string): void;
 // @internal
 export function setInSessionStorage(key: string, value: string): void;
 
+// @public
+export function setUniqueIdGenerator(fn: (size?: number) => string): void;
+
 // @internal
 export function sleep(ms: number): Promise<void>;
 

packages/utils/src/index.ts

@@ -33,7 +33,7 @@ export { ExecutionQueue } from './lib/ExecutionQueue'
 export { FileHelpers } from './lib/file'
 export { noop, omitFromStackTrace } from './lib/function'
 export { getHashForBuffer, getHashForObject, getHashForString, lns } from './lib/hash'
-export { mockUniqueId, restoreUniqueId, uniqueId } from './lib/id'
+export { mockUniqueId, restoreUniqueId, setUniqueIdGenerator, uniqueId } from './lib/id'
 export { getFirstFromIterable } from './lib/iterable'
 export type { JsonArray, JsonObject, JsonPrimitive, JsonValue } from './lib/json-value'
 export {

packages/utils/src/lib/id.ts

@@ -55,6 +55,26 @@ function nanoid(size = 21) {
 }
 
 let impl = nanoid
+
+/**
+ * Override the unique ID generator with a custom implementation.
+ *
+ * This allows you to customize how IDs are generated across tldraw,
+ * affecting all ID generation including shapes, pages, and other records.
+ *
+ * @param fn - A function that returns a string ID. Takes an optional size parameter.
+ * @example
+ * ```ts
+ * import { setUniqueIdGenerator } from '@tldraw/editor'
+ *
+ * setUniqueIdGenerator(() => `id-${Date.now()}-${Math.random()}`)
+ * ```
+ * @public
+ */
+export function setUniqueIdGenerator(fn: (size?: number) => string) {
+	impl = fn
+}
+
 /**
  * Mock the unique ID generator with a custom implementation for testing.
  *

scripts/pack-and-copy.sh

@@ -3,7 +3,7 @@ set -e
 
 REPO_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
 DEST="$HOME/work/core/shared/js/tldraw"
-PACKAGES=(tldraw editor)
+PACKAGES=(utils tldraw editor)
 
 mkdir -p "$DEST"