feat: improve utils

Author: antfu

packages/core/src/node/__tests__/host-docks.test.ts

@@ -167,17 +167,15 @@ describe('devToolsDockHost', () => {
     it('should preserve dock in values() after update', () => {
       const host = new DevToolsDockHost(mockContext)
 
-      host.register({
+      const { update } = host.register({
         type: 'iframe',
         id: 'test',
         title: 'Original',
         icon: 'icon',
         url: 'http://localhost:3000',
       })
 
-      host.update({
-        type: 'iframe',
-        id: 'test',
+      update({
         title: 'Updated',
         icon: 'newicon',
         url: 'http://localhost:3001',

packages/core/src/node/host-docks.ts

@@ -14,12 +14,23 @@ export class DevToolsDockHost implements DevToolsDockHostType {
     return Array.from(this.views.values())
   }
 
-  register(view: DevToolsDockUserEntry, force?: boolean): void {
+  register<T extends DevToolsDockUserEntry>(view: T, force?: boolean): {
+    update: (patch: Partial<T>) => void
+  } {
     if (this.views.has(view.id) && !force) {
       throw new Error(`Dock with id "${view.id}" is already registered`)
     }
     this.views.set(view.id, view)
     this.events.emit('dock:entry:updated', view)
+
+    return {
+      update: (patch) => {
+        if (patch.id && patch.id !== view.id) {
+          throw new Error(`Cannot change the id of a dock. Use register() to add new docks.`)
+        }
+        this.update(Object.assign(this.views.get(view.id)!, patch))
+      },
+    }
   }
 
   update(view: DevToolsDockUserEntry): void {

packages/kit/src/types/docks.ts

@@ -1,12 +1,14 @@
 import type { EventEmitter } from './events'
 
 export interface DevToolsDockHost {
-  views: Map<string, DevToolsDockUserEntry>
-  events: EventEmitter<{
+  readonly views: Map<string, DevToolsDockUserEntry>
+  readonly events: EventEmitter<{
     'dock:entry:updated': (entry: DevToolsDockUserEntry) => void
   }>
 
-  register: (entry: DevToolsDockUserEntry) => void
+  register: <T extends DevToolsDockUserEntry>(entry: T, force?: boolean) => {
+    update: (patch: Partial<T>) => void
+  }
   update: (entry: DevToolsDockUserEntry) => void
   values: () => DevToolsDockUserEntry[]
 }