fix(send): correctly typing optional payloads

Author: sandros94

playground/app/pages/index.vue

@@ -65,10 +65,10 @@ const items = ['notifications', 'chat'] as const
 const channels = ref<Array<typeof items[number]>>([])
 
 const { states, status, send, open, close } = useWS<{
-  notifications: {
+  notifications?: {
     message: string
   }
-  chat: {
+  chat?: {
     user?: string
     text: string
   }[]

src/runtime/app/composables/ws.ts

@@ -147,7 +147,7 @@ export function useWS<
    */
   function send<
     Type extends 'subscribe' | 'unsubscribe',
-    Topic extends keyof T,
+    Topic extends keyof Required<T>,
   >(type: Type, topic: Topic): boolean
   /**
    * Sends a payload to a specific WebSocket topic.
@@ -167,8 +167,8 @@ export function useWS<
    */
   function send<
     Type extends 'publish',
-    Topic extends keyof T,
-    Payload extends T[Topic] extends Array<infer U> ? U : T[Topic],
+    Topic extends keyof Required<T>,
+    Payload extends Required<T>[Topic] extends Array<infer U> ? U : Required<T>[Topic],
   >(type: Type, topic: Topic, payload: Payload): boolean
   function send<
     M extends (

src/runtime/types/index.ts

@@ -38,12 +38,12 @@ export interface UseWSReturn<T extends Record<string | AllTopics, any>, D> exten
   >(payload: Payload): boolean
   send<
     Type extends 'subscribe' | 'unsubscribe',
-    Topic extends keyof T,
+    Topic extends keyof Required<T>,
   >(type: Type, topic: Topic): boolean
   send<
     Type extends 'publish',
-    Topic extends keyof T,
-    Payload extends T[Topic] extends Array<infer U> ? U : T[Topic],
+    Topic extends keyof Required<T>,
+    Payload extends Required<T>[Topic] extends Array<infer U> ? U : Required<T>[Topic],
   >(type: Type, topic: Topic, payload: Payload): boolean
   /**
    * Upstream, mainly for internal use.