falcondev-oss
diff --git a/‎src/capabilties.ts
Lines changed: 3 additions & 3 deletions b/‎src/capabilties.ts
Lines changed: 3 additions & 3 deletions
diff --git a/‎tests/capabilities.test.ts
Lines changed: 25 additions & 2 deletions b/‎tests/capabilities.test.ts
Lines changed: 25 additions & 2 deletions
@@ -18,7 +18,7 @@ export interface ContextOptions {
   createError?: (args: { capability: string }) => unknown
 }
 
-export function createActor<Actor>() {
+export function createActor<Actor extends object>() {
   return {
     build<D>(builder: (cap: ReturnType<typeof createCapability<Actor>>) => D) {
       return (actor: Actor, opts?: ContextOptions) => {
@@ -120,7 +120,7 @@ function createQuery<Actor, Subject, Capabilities extends string, Args>({
         })
       },
     }),
-    subjects: (subjects: Subject[]) => ({
+    subjects: <const S extends Subject>(subjects: S[]) => ({
       canSome: <Capability extends Capabilities>(
         capability: Capability,
         ...args: HasArgs<Capability>
@@ -146,7 +146,7 @@ function createQuery<Actor, Subject, Capabilities extends string, Args>({
           },
         }
       },
-      filter: <FilterCaps extends Capabilities>(
+      filter: <const FilterCaps extends Capabilities>(
         capabilities: FilterCaps[],
         args: IfEmptyObject<Args, void, Partial<Args>>,
       ) => {
 
@@ -1,6 +1,7 @@
-import { describe, expect, expectTypeOf, test } from 'vitest'
+import { range } from 'remeda'
 
-import { arg, createActor } from '../src/'
+import { describe, expect, expectTypeOf, test } from 'vitest'
+import { arg, createActor, mode, type Modes } from '../src/'
 
 test('minimal example', () => {
   let i = 0
@@ -28,6 +29,28 @@ test('minimal example', () => {
   expect(caps.a.list()).toEqual(['yield', 'return'])
 })
 
+test('modes', () => {
+  const useActor = createActor().build((cap) => ({
+    a: cap
+      .subject<
+        Modes<{
+          a: { a: number }
+          b: { b: number }
+        }>
+      >()
+      .define(function* () {
+        yield ['read']
+        return []
+      }),
+  }))
+
+  const caps = useActor({})
+
+  const filtered = caps.a.subjects(range(1, 5).map((i) => mode('b', { b: i }))).filter(['read'])
+
+  expectTypeOf(filtered).toMatchTypeOf<{ b: number }[]>()
+})
+
 describe('user management demo', () => {
   type Role = 'user' | 'moderator' | 'admin'