@prosekit/core 0.10.0 → 0.12.0-beta.0

package/src/extensions/events/dom-event.ts

@@ -1,13 +1,13 @@
 import { PluginKey, ProseMirrorPlugin } from '@prosekit/pm/state'
 import type { DOMEventMap, EditorView } from '@prosekit/pm/view'
 
-import { defineFacet, type Facet } from '../../facets/facet'
-import { defineFacetPayload } from '../../facets/facet-extension'
-import type { PlainExtension } from '../../types/extension'
-import type { Setter } from '../../types/setter'
-import { groupEntries } from '../../utils/array-grouping'
-import { combineEventHandlers } from '../../utils/combine-event-handlers'
-import { pluginFacet, type PluginPayload } from '../plugin'
+import { defineFacetPayload } from '../../facets/facet-extension.ts'
+import { defineFacet, type Facet } from '../../facets/facet.ts'
+import type { PlainExtension } from '../../types/extension.ts'
+import type { Setter } from '../../types/setter.ts'
+import { groupEntries } from '../../utils/array-grouping.ts'
+import { combineEventHandlers } from '../../utils/combine-event-handlers.ts'
+import { pluginFacet, type PluginPayload } from '../plugin.ts'
 
 /**
  * A function to handle the events fired on the editable DOM element. Returns

package/src/extensions/events/editor-event.ts

@@ -3,12 +3,12 @@ import type { Node, Slice } from '@prosekit/pm/model'
 import { PluginKey, ProseMirrorPlugin } from '@prosekit/pm/state'
 import type { EditorView } from '@prosekit/pm/view'
 
-import { defineFacet, type Facet } from '../../facets/facet'
-import { defineFacetPayload } from '../../facets/facet-extension'
-import type { PlainExtension } from '../../types/extension'
-import { groupEntries } from '../../utils/array-grouping'
-import { combineEventHandlers } from '../../utils/combine-event-handlers'
-import { pluginFacet, type PluginPayload } from '../plugin'
+import { defineFacetPayload } from '../../facets/facet-extension.ts'
+import { defineFacet, type Facet } from '../../facets/facet.ts'
+import type { PlainExtension } from '../../types/extension.ts'
+import { groupEntries } from '../../utils/array-grouping.ts'
+import { combineEventHandlers } from '../../utils/combine-event-handlers.ts'
+import { pluginFacet, type PluginPayload } from '../plugin.ts'
 
 export type KeyDownHandler = (
   view: EditorView,

package/src/extensions/events/focus.spec.ts

@@ -1,10 +1,10 @@
 import { describe, expect, it, vi } from 'vitest'
 
-import { createEditor } from '../../editor/editor'
-import { union } from '../../editor/union'
-import { defineTestExtension } from '../../testing'
+import { createEditor } from '../../editor/editor.ts'
+import { union } from '../../editor/union.ts'
+import { defineTestExtension } from '../../testing/index.ts'
 
-import { defineFocusChangeHandler } from './focus'
+import { defineFocusChangeHandler } from './focus.ts'
 
 describe('defineFocusChangeHandler', () => {
   it('should call the handler when the editor is focused or blurred', () => {

package/src/extensions/events/focus.ts

@@ -1,6 +1,6 @@
-import type { PlainExtension } from '../../types/extension'
+import type { PlainExtension } from '../../types/extension.ts'
 
-import { defineDomEventFacetPayload } from './dom-event'
+import { defineDomEventFacetPayload } from './dom-event.ts'
 
 /**
  * A function that is called when the editor gains or loses focus.

package/src/extensions/events/plugin-view.ts

@@ -1,10 +1,10 @@
 import { PluginKey, ProseMirrorPlugin, type EditorState } from '@prosekit/pm/state'
 import type { EditorView } from '@prosekit/pm/view'
 
-import { defineFacet } from '../../facets/facet'
-import { defineFacetPayload } from '../../facets/facet-extension'
-import type { PlainExtension } from '../../types/extension'
-import { pluginFacet, type PluginPayload } from '../plugin'
+import { defineFacetPayload } from '../../facets/facet-extension.ts'
+import { defineFacet } from '../../facets/facet.ts'
+import type { PlainExtension } from '../../types/extension.ts'
+import { pluginFacet, type PluginPayload } from '../plugin.ts'
 
 /**
  * A function that is called when the editor view is mounted.

package/src/extensions/history.ts

@@ -1,12 +1,12 @@
 import { history, redo, undo } from '@prosekit/pm/history'
 
-import { union } from '../editor/union'
-import type { Extension } from '../types/extension'
-import { isApple } from '../utils/env'
+import { union } from '../editor/union.ts'
+import type { Extension } from '../types/extension.ts'
+import { isApple } from '../utils/env.ts'
 
-import { defineCommands } from './command'
-import { defineKeymap, type Keymap } from './keymap'
-import { definePlugin } from './plugin'
+import { defineCommands } from './command.ts'
+import { defineKeymap, type Keymap } from './keymap.ts'
+import { definePlugin } from './plugin.ts'
 
 const keymap: Keymap = {
   'Mod-z': undo,

package/src/extensions/keymap-base.spec.ts

@@ -1,12 +1,12 @@
 import { describe, expect, it } from 'vitest'
 import { keyboard } from 'vitest-browser-commands/playwright'
 
-import { union } from '../editor/union'
-import type { TestEditor } from '../test'
-import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing'
-import type { SelectionJSON } from '../types/model'
+import { union } from '../editor/union.ts'
+import type { TestEditor } from '../test/index.ts'
+import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing/index.ts'
+import type { SelectionJSON } from '../types/model.ts'
 
-import { defineBaseKeymap } from './keymap-base'
+import { defineBaseKeymap } from './keymap-base.ts'
 
 describe('Mod-a', () => {
   it('can select the block for the first Mod-a press', async () => {

package/src/extensions/keymap-base.ts

@@ -11,12 +11,12 @@ import {
 } from '@prosekit/pm/commands'
 import { splitSplittableBlock } from 'prosemirror-splittable'
 
-import { selectBlockCommand } from '../commands/select-block'
-import { withPriority } from '../editor/with-priority'
-import type { PlainExtension } from '../types/extension'
-import { Priority } from '../types/priority'
+import { selectBlockCommand } from '../commands/select-block.ts'
+import { withPriority } from '../editor/with-priority.ts'
+import type { PlainExtension } from '../types/extension.ts'
+import type { Priority } from '../types/priority.ts'
 
-import { defineKeymap, type Keymap } from './keymap'
+import { defineKeymap, type Keymap } from './keymap.ts'
 
 // Replace `splitBlock` with `splitSplittableBlock`
 const customEnter = chainCommands(
@@ -68,7 +68,7 @@ export interface BaseKeymapOptions {
  * @public
  */
 export function defineBaseKeymap({
-  priority = Priority.low,
+  priority = (1 satisfies typeof Priority.low),
   preferBlockSelection = true,
 }: BaseKeymapOptions = {}): BaseKeymapExtension {
   const keymap: Keymap = {

package/src/extensions/keymap.spec.ts

@@ -2,12 +2,12 @@ import type { Command } from '@prosekit/pm/state'
 import { describe, expect, it, vi } from 'vitest'
 import { keyboard } from 'vitest-browser-commands/playwright'
 
-import { union } from '../editor/union'
-import { withPriority } from '../editor/with-priority'
-import { defineDoc, defineParagraph, defineText, setupTest, setupTestFromExtension } from '../testing'
-import { Priority } from '../types/priority'
+import { union } from '../editor/union.ts'
+import { withPriority } from '../editor/with-priority.ts'
+import { defineDoc, defineParagraph, defineText, setupTest, setupTestFromExtension } from '../testing/index.ts'
+import { Priority } from '../types/priority.ts'
 
-import { defineKeymap, type Keymap } from './keymap'
+import { defineKeymap, type Keymap } from './keymap.ts'
 
 describe('keymap', () => {
   it('can register and unregister keymap', () => {

package/src/extensions/keymap.ts

@@ -2,11 +2,11 @@ import { keydownHandler } from '@prosekit/pm/keymap'
 import { Plugin, PluginKey, type Command } from '@prosekit/pm/state'
 import type { EditorView } from '@prosekit/pm/view'
 
-import { defineFacet, type Facet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import type { PlainExtension } from '../types/extension'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet, type Facet } from '../facets/facet.ts'
+import type { PlainExtension } from '../types/extension.ts'
 
-import { pluginFacet, type PluginPayload } from './plugin'
+import { pluginFacet, type PluginPayload } from './plugin.ts'
 
 /**
  * A set of keybindings. Please read the

package/src/extensions/mark-spec.spec.ts

@@ -2,10 +2,10 @@ import type { DOMOutputSpec, TagParseRule } from '@prosekit/pm/model'
 import { formatHTML } from 'diffable-html-snapshot'
 import { describe, expect, it } from 'vitest'
 
-import { union } from '../editor/union'
-import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing'
+import { union } from '../editor/union.ts'
+import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing/index.ts'
 
-import { defineMarkAttr, defineMarkSpec } from './mark-spec'
+import { defineMarkAttr, defineMarkSpec } from './mark-spec.ts'
 
 describe('defineMarkSpec', () => {
   it('can merge mark specs', () => {

package/src/extensions/mark-spec.ts

@@ -2,14 +2,14 @@ import { isNotNullish, mapGroupBy } from '@ocavue/utils'
 import type { AttributeSpec, MarkSpec, SchemaSpec } from '@prosekit/pm/model'
 import OrderedMap from 'orderedmap'
 
-import { defineFacet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import { schemaSpecFacet } from '../facets/schema-spec'
-import type { AnyAttrs, AttrSpec } from '../types/attrs'
-import type { Extension } from '../types/extension'
-import { assert } from '../utils/assert'
-import { mergeSpecs } from '../utils/merge-specs'
-import { wrapOutputSpecAttrs, wrapParseRuleAttrs } from '../utils/output-spec'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet } from '../facets/facet.ts'
+import { schemaSpecFacet } from '../facets/schema-spec.ts'
+import type { AnyAttrs, AttrSpec } from '../types/attrs.ts'
+import type { Extension } from '../types/extension.ts'
+import { assert } from '../utils/assert.ts'
+import { mergeSpecs } from '../utils/merge-specs.ts'
+import { wrapOutputSpecAttrs, wrapParseRuleAttrs } from '../utils/output-spec.ts'
 
 /**
  * @public

package/src/extensions/mark-view-effect.ts

@@ -2,11 +2,11 @@ import { isNotNullish } from '@ocavue/utils'
 import { PluginKey, ProseMirrorPlugin } from '@prosekit/pm/state'
 import type { MarkViewConstructor } from '@prosekit/pm/view'
 
-import { defineFacet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import type { Extension } from '../types/extension'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet } from '../facets/facet.ts'
+import type { Extension } from '../types/extension.ts'
 
-import { pluginFacet, type PluginPayload } from './plugin'
+import { pluginFacet, type PluginPayload } from './plugin.ts'
 
 /**
  * @internal

package/src/extensions/mark-view.ts

@@ -1,11 +1,11 @@
 import { PluginKey, ProseMirrorPlugin } from '@prosekit/pm/state'
 import type { MarkViewConstructor } from '@prosekit/pm/view'
 
-import { defineFacet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import type { Extension } from '../types/extension'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet } from '../facets/facet.ts'
+import type { Extension } from '../types/extension.ts'
 
-import { pluginFacet, type PluginPayload } from './plugin'
+import { pluginFacet, type PluginPayload } from './plugin.ts'
 
 export interface MarkViewOptions {
   name: string

package/src/extensions/node-spec.spec.ts

@@ -2,12 +2,12 @@ import type { DOMOutputSpec, Schema, TagParseRule } from '@prosekit/pm/model'
 import { formatHTML } from 'diffable-html-snapshot'
 import { describe, expect, it } from 'vitest'
 
-import { union } from '../editor/union'
-import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing'
+import { union } from '../editor/union.ts'
+import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing/index.ts'
 
-import { defineHistory } from './history'
-import { defineBaseKeymap } from './keymap-base'
-import { defineNodeAttr, defineNodeSpec } from './node-spec'
+import { defineHistory } from './history.ts'
+import { defineBaseKeymap } from './keymap-base.ts'
+import { defineNodeAttr, defineNodeSpec } from './node-spec.ts'
 
 describe('defineNodeSpec', () => {
   it('can merge node specs', () => {

package/src/extensions/node-spec.ts

@@ -2,14 +2,14 @@ import { isNotNullish, mapGroupBy } from '@ocavue/utils'
 import type { AttributeSpec, NodeSpec, SchemaSpec } from '@prosekit/pm/model'
 import OrderedMap from 'orderedmap'
 
-import { defineFacet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import { schemaSpecFacet } from '../facets/schema-spec'
-import type { AnyAttrs, AttrSpec } from '../types/attrs'
-import type { Extension } from '../types/extension'
-import { assert } from '../utils/assert'
-import { mergeSpecs } from '../utils/merge-specs'
-import { wrapOutputSpecAttrs, wrapTagParseRuleAttrs } from '../utils/output-spec'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet } from '../facets/facet.ts'
+import { schemaSpecFacet } from '../facets/schema-spec.ts'
+import type { AnyAttrs, AttrSpec } from '../types/attrs.ts'
+import type { Extension } from '../types/extension.ts'
+import { assert } from '../utils/assert.ts'
+import { mergeSpecs } from '../utils/merge-specs.ts'
+import { wrapOutputSpecAttrs, wrapTagParseRuleAttrs } from '../utils/output-spec.ts'
 
 /**
  * @public

package/src/extensions/node-view-effect.ts

@@ -2,11 +2,11 @@ import { isNotNullish } from '@ocavue/utils'
 import { PluginKey, ProseMirrorPlugin } from '@prosekit/pm/state'
 import type { NodeViewConstructor } from '@prosekit/pm/view'
 
-import { defineFacet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import type { Extension } from '../types/extension'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet } from '../facets/facet.ts'
+import type { Extension } from '../types/extension.ts'
 
-import { pluginFacet, type PluginPayload } from './plugin'
+import { pluginFacet, type PluginPayload } from './plugin.ts'
 
 /**
  * @internal

package/src/extensions/node-view.ts

@@ -1,11 +1,11 @@
 import { PluginKey, ProseMirrorPlugin } from '@prosekit/pm/state'
 import type { NodeViewConstructor } from '@prosekit/pm/view'
 
-import { defineFacet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import type { Extension } from '../types/extension'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet } from '../facets/facet.ts'
+import type { Extension } from '../types/extension.ts'
 
-import { pluginFacet, type PluginPayload } from './plugin'
+import { pluginFacet, type PluginPayload } from './plugin.ts'
 
 export interface NodeViewOptions {
   name: string

package/src/extensions/plugin.spec.ts

@@ -1,12 +1,12 @@
 import { Plugin, PluginKey } from '@prosekit/pm/state'
 import { describe, expect, it, vi } from 'vitest'
 
-import { union } from '../editor/union'
-import { withPriority } from '../editor/with-priority'
-import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing'
-import { Priority } from '../types/priority'
+import { union } from '../editor/union.ts'
+import { withPriority } from '../editor/with-priority.ts'
+import { defineDoc, defineParagraph, defineText, setupTestFromExtension } from '../testing/index.ts'
+import { Priority } from '../types/priority.ts'
 
-import { definePlugin } from './plugin'
+import { definePlugin } from './plugin.ts'
 
 describe('plugin', () => {
   it('maintains plugin order in state based on priority', () => {

package/src/extensions/plugin.ts

@@ -1,11 +1,11 @@
 import type { Schema } from '@prosekit/pm/model'
 import { Plugin, type ProseMirrorPlugin } from '@prosekit/pm/state'
 
-import { ProseKitError } from '../error'
-import { defineFacet, type Facet } from '../facets/facet'
-import { defineFacetPayload } from '../facets/facet-extension'
-import { stateFacet, type StatePayload } from '../facets/state'
-import type { PlainExtension } from '../types/extension'
+import { ProseKitError } from '../error.ts'
+import { defineFacetPayload } from '../facets/facet-extension.ts'
+import { defineFacet, type Facet } from '../facets/facet.ts'
+import { stateFacet, type StatePayload } from '../facets/state.ts'
+import type { PlainExtension } from '../types/extension.ts'
 
 /**
  * Adds a ProseMirror plugin to the editor.

package/src/facets/base-extension.ts

@@ -1,12 +1,12 @@
 import type { Schema } from '@prosekit/pm/model'
 
-import type { Extension, ExtensionTyping } from '../types/extension'
-import { Priority } from '../types/priority'
+import type { Extension, ExtensionTyping } from '../types/extension.ts'
+import type { Priority } from '../types/priority.ts'
 
-import type { Facet } from './facet'
-import type { FacetNode } from './facet-node'
-import type { Tuple5 } from './facet-types'
-import { schemaFacet } from './schema'
+import type { FacetNode } from './facet-node.ts'
+import type { Tuple5 } from './facet-types.ts'
+import type { Facet } from './facet.ts'
+import { schemaFacet } from './schema.ts'
 
 /**
  * @internal
@@ -27,7 +27,7 @@ export abstract class BaseExtension<T extends ExtensionTyping = ExtensionTyping>
    * @internal
    */
   getTree(priority?: Priority): FacetNode {
-    const pri = priority ?? this.priority ?? Priority.default
+    const pri = priority ?? this.priority ?? (2 satisfies typeof Priority.default)
     return (this.trees[pri] ||= this.createTree(pri))
   }
 

package/src/facets/command.ts

@@ -1,7 +1,7 @@
-import type { CommandCreators } from '../types/extension-command'
+import type { CommandCreators } from '../types/extension-command.ts'
 
-import { defineFacet, type Facet } from './facet'
-import { rootFacet, type RootPayload } from './root'
+import { defineFacet, type Facet } from './facet.ts'
+import { rootFacet, type RootPayload } from './root.ts'
 
 type CommandPayload = CommandCreators
 

package/src/facets/facet-extension.spec.ts

@@ -1,13 +1,13 @@
 import { isNotNullish } from '@ocavue/utils'
 import { describe, expect, it, vi } from 'vitest'
 
-import { Priority } from '../types/priority'
+import { Priority } from '../types/priority.ts'
 
-import { defineFacet, Facet } from './facet'
-import { FacetExtensionImpl } from './facet-extension'
-import { subtractFacetNode, unionFacetNode } from './facet-node'
-import type { FacetReducer } from './facet-types'
-import { UnionExtensionImpl } from './union-extension'
+import { FacetExtensionImpl } from './facet-extension.ts'
+import { subtractFacetNode, unionFacetNode } from './facet-node.ts'
+import type { FacetReducer } from './facet-types.ts'
+import { defineFacet, Facet } from './facet.ts'
+import { UnionExtensionImpl } from './union-extension.ts'
 
 describe('facet extension', () => {
   type FooHandler = (foo: string) => void

package/src/facets/facet-extension.ts

@@ -1,25 +1,29 @@
-import type { Extension } from '../types/extension'
-import type { Priority } from '../types/priority'
+import type { Extension } from '../types/extension.ts'
+import type { Priority } from '../types/priority.ts'
 
-import { BaseExtension } from './base-extension'
-import type { Facet } from './facet'
-import { FacetNode } from './facet-node'
-import type { Tuple5 } from './facet-types'
+import { BaseExtension } from './base-extension.ts'
+import { FacetNode } from './facet-node.ts'
+import type { Tuple5 } from './facet-types.ts'
+import type { Facet } from './facet.ts'
 
 /**
  * @internal
  */
 export class FacetExtensionImpl<Input, Output> extends BaseExtension {
   declare extension: Extension
+  readonly facet: Facet<Input, Output>
+  readonly payloads: Input[]
 
   /**
    * @internal
    */
   constructor(
-    readonly facet: Facet<Input, Output>,
-    readonly payloads: Input[],
+    facet: Facet<Input, Output>,
+    payloads: Input[],
   ) {
     super()
+    this.facet = facet
+    this.payloads = payloads
   }
 
   /**

package/src/facets/facet-node.spec.ts

@@ -1,9 +1,9 @@
 import { expect, test } from 'vitest'
 
-import { Priority } from '../types/priority'
+import { Priority } from '../types/priority.ts'
 
-import { Facet } from './facet'
-import { FacetNode, subtractFacetNode, unionFacetNode } from './facet-node'
+import { FacetNode, subtractFacetNode, unionFacetNode } from './facet-node.ts'
+import { Facet } from './facet.ts'
 
 const sum = (input: number[]) => input.reduce((acc, cur) => acc + cur, 0)
 

package/src/facets/facet-node.ts

@@ -1,12 +1,12 @@
 import { isNotNullish } from '@ocavue/utils'
 
-import { Priority } from '../types/priority'
-import { arraySubtract, uniqPush } from '../utils/array'
-import { assert } from '../utils/assert'
+import type { Priority } from '../types/priority.ts'
+import { arraySubtract, uniqPush } from '../utils/array.ts'
+import { assert } from '../utils/assert.ts'
 
-import type { Facet } from './facet'
-import type { FacetReducer, Tuple5 } from './facet-types'
-import type { RootOutput } from './root'
+import type { FacetReducer, Tuple5 } from './facet-types.ts'
+import type { Facet } from './facet.ts'
+import type { RootOutput } from './root.ts'
 
 function zip5<T>(
   a: Tuple5<T>,
@@ -104,18 +104,28 @@ export function subtractFacetNode<I, O>(
 export class FacetNode<I = any, O = any> {
   output: Tuple5<O | null> | null = null
 
+  readonly facet: Facet<I, O>
+  readonly inputs: Tuple5<I[] | null>
+  readonly children: Map<number, FacetNode>
+  readonly reducers: Tuple5<FacetReducer<I, O> | null>
+
   constructor(
-    readonly facet: Facet<I, O>,
-    readonly inputs: Tuple5<I[] | null> = [null, null, null, null, null],
-    readonly children: Map<number, FacetNode> = new Map(),
-    readonly reducers: Tuple5<FacetReducer<I, O> | null> = [
+    facet: Facet<I, O>,
+    inputs: Tuple5<I[] | null> = [null, null, null, null, null],
+    children: Map<number, FacetNode> = new Map<number, FacetNode>(),
+    reducers: Tuple5<FacetReducer<I, O> | null> = [
       null,
       null,
       null,
       null,
       null,
     ],
-  ) {}
+  ) {
+    this.facet = facet
+    this.inputs = inputs
+    this.children = children
+    this.reducers = reducers
+  }
 
   private calcOutput(): Tuple5<O | null> {
     const inputs: Tuple5<I[] | null> = [null, null, null, null, null]
@@ -139,9 +149,9 @@ export class FacetNode<I = any, O = any> {
     }
 
     if (this.facet.singleton) {
-      const reducer = (this.reducers[Priority.default] ||= this.facet.reducer)
+      const reducer = (this.reducers[2 satisfies typeof Priority.default] ||= this.facet.reducer)
       const input: I[] = inputs.filter(isNotNullish).flat()
-      output[Priority.default] = reducer(input)
+      output[2 satisfies typeof Priority.default] = reducer(input)
     } else {
       for (let pri = 0; pri < 5; pri++) {
         const input = inputs[pri]
@@ -164,7 +174,7 @@ export class FacetNode<I = any, O = any> {
 
   getSingletonOutput(): O | null {
     assert(this.facet.singleton)
-    return this.getOutput()[Priority.default]
+    return this.getOutput()[2 satisfies typeof Priority.default]
   }
 
   getRootOutput(): RootOutput {

package/src/facets/facet.spec.ts

@@ -1,6 +1,6 @@
 import { expect, test } from 'vitest'
 
-import { Facet } from './facet'
+import { Facet } from './facet.ts'
 
 type RootInput = {
   count: number

package/src/facets/facet.ts

@@ -1,6 +1,6 @@
-import { assert } from '../utils/assert'
+import { ProseKitError } from '../error.ts'
 
-import type { FacetReducer } from './facet-types'
+import type { FacetReducer } from './facet-types.ts'
 
 let facetCount = 0
 
@@ -27,17 +27,24 @@ export class Facet<Input, Output> {
    */
   readonly path: number[]
 
+  private reduce: () => FacetReducer<Input, Output>
+
   /**
    * @internal
    */
   constructor(
     parent: Facet<Output, any> | null,
     singleton: boolean,
-    private _reducer?: FacetReducer<Input, Output> | undefined,
-    private _reduce?: () => FacetReducer<Input, Output>,
+    reducer?: FacetReducer<Input, Output>,
+    reduce?: () => FacetReducer<Input, Output>,
   ) {
-    // Only one of _reducer or _reduce can be defined
-    assert((_reduce || _reducer) && !(_reduce && _reducer))
+    if (reduce && !reducer) {
+      this.reduce = reduce
+    } else if (reducer && !reduce) {
+      this.reduce = () => reducer
+    } else {
+      throw new ProseKitError('Incorrect reducer')
+    }
 
     this.parent = parent
     this.singleton = singleton
@@ -45,7 +52,7 @@ export class Facet<Input, Output> {
   }
 
   get reducer(): FacetReducer<Input, Output> {
-    return (this._reducer ?? this._reduce?.())!
+    return this.reduce()
   }
 }