fix useRoomState() types to accept either a Schema or Room.

Author: endel

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@colyseus/react",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "type": "module",
   "scripts": {
     "dev": "tsdown --watch --format esm,cjs",

src/schema/useRoomState.ts

@@ -29,11 +29,24 @@ import { useColyseusState } from './useColyseusState';
  * const players = useRoomState(room, (s) => s.players);
  * ```
  */
+
+export function useRoomState<T, U = any>(
+    selector: (state: InferState<T, never>) => U
+): Snapshot<U> | undefined;
+
 export function useRoomState<T = any, State = InferState<T, never>, U = State>(
     room: Room<T, State> | null | undefined,
-    selector: (state: State) => U = (s) => s as unknown as U
-): Snapshot<U> | undefined {
+    selector?: (state: State) => U
+): Snapshot<U> | undefined;
+
+export function useRoomState(
+    roomOrSelector?: Room | ((state: any) => any) | null,
+    selector?: (state: any) => any
+): any {
+    const room = typeof roomOrSelector === 'function' ? undefined : roomOrSelector;
+    const sel = typeof roomOrSelector === 'function' ? roomOrSelector : selector;
+
     const serializer = room?.serializer as SchemaSerializer<any> | undefined;
 
-    return useColyseusState(room?.state, serializer?.decoder, selector);
+    return useColyseusState(room?.state, serializer?.decoder, sel);
 }

src/tests/useRoomState.test.ts

@@ -2,8 +2,20 @@ import 'reflect-metadata';
 import { renderHook } from '@testing-library/react';
 import { describe, expect, test } from 'vitest'
 import { useRoomState } from '../schema/useRoomState';
+import { Schema, type } from '@colyseus/schema';
 
 describe('falsy room argument', () => {
+    test('types', () => {
+        class MySchema extends Schema {
+            @type("string") myField: string = "hello";
+        }
+        class Room {
+            '~state': MySchema;
+        }
+        const schemaField = renderHook(() => useRoomState<MySchema>((s) => s.myField));
+        const roomField = renderHook(() => useRoomState<Room>((s) => s.myField));
+    })
+
     test('does not crash when room is null', () => {
         const { result } = renderHook(() => useRoomState(null));
         expect(result.current).toBeUndefined();