xstate-tsserver/src/queries.ts

XState tsserver plugin for easier navigation in state machines

export const machineWithSetupQuery = `
(call_expression
  function: (member_expression
    object: (call_expression
      function: (identifier) @xstate.setup (#eq? @xstate.setup "setup")
      arguments: (arguments) @xstate.setup.config)
    property: (property_identifier) @xstate.createMachine (#eq? @xstate.createMachine "createMachine"))
  arguments: (arguments
    (object) @xstate.state.config @xstate.root.config)) @xstate.machine
`;

export const setupActionsQuery = `
(pair
  key: (property_identifier) @key (#eq? @key "actions")
  value: (object [
    (pair
      key: (property_identifier) @action.name
      value: (_)
    )
    (shorthand_property_identifier) @action.name
  ])
)
`;

export const machineActionsQuery = `
(pair
  key: (property_identifier) @key
    (#match? @key "actions|entry|exit")
  value: [
    (array (string (string_fragment) @xstate.action.name) @xstate.action)
    (string (string_fragment) @xstate.action.name) @xstate.action
    (array
      (object
        (pair
          key: (property_identifier) @type_key
            (#eq? @type_key "type")
          value: (string (string_fragment) @xstate.action.name) @xstate.action
        )
      )
    )
    (object
      (pair
        key: (property_identifier) @type_key
          (#eq? @type_key "type")
        value: (string (string_fragment) @xstate.action.name) @xstate.action
      )
    )
  ]
)

(call_expression
  function: (identifier) @enqueue (#eq? @enqueue "enqueue")
  arguments: (arguments [ 
    (string (string_fragment) @xstate.action.name) @xstate.action
    (object
      (pair
        key: (property_identifier) @type_key
          (#eq? @type_key "type")
        value: (string (string_fragment) @xstate.action.name) @xstate.action
      )
    )
  ])
)
`;

export const setupGuardsQuery = `
(pair
  key: (property_identifier) @key (#eq? @key "guards")
  value: (object [
    (pair
      key: (property_identifier) @guard.name
      value: (_)
    )
    (shorthand_property_identifier) @guard.name
  ])
)
`;

export const machineGuardsQuery = `
(pair
  key: (property_identifier) @key
    (#eq? @key "guard")
  value: [
    (string (string_fragment) @xstate.guard.name) @xstate.guard
    (object
      (pair
        key: (property_identifier) @type_key
          (#eq? @type_key "type")
        value: (string (string_fragment) @xstate.guard.name) @xstate.guard
      )
    )
  ]
)

(call_expression
  function: (identifier) @check (#match? @check "check|not")
  arguments: (arguments [
    (string (string_fragment) @xstate.guard.name) @xstate.guard
    (object
      (pair
        key: (property_identifier) @type_key
          (#eq? @type_key "type")
        value: (string (string_fragment) @xstate.guard.name) @xstate.guard
      )
    )]
  )
)

(call_expression
  function: (identifier) @hog (#match? @hog "and|or")
  arguments: (arguments 
    (array [
      (string (string_fragment) @xstate.guard.name) @xstate.guard
      (object
        (pair
          key: (property_identifier) @type_key
            (#eq? @type_key "type")
          value: (string (string_fragment) @xstate.guard.name) @xstate.guard
        )
      )]
    )
  )
)
`;

export const setupActorsQuery = `
(pair
  key: (property_identifier) @key (#eq? @key "actors")
  value: (object [
    (pair
      key: (property_identifier) @actor.name
      value: (_)
    )
    (shorthand_property_identifier) @actor.name
  ])
)
`;

export const machineActorsQuery = `
(pair
  key: (property_identifier) @invoke (#eq? @invoke "invoke")
  value: [
    (object
      (pair
        key: (property_identifier) @invoke.src (#eq? @invoke.src "src")
        value: (string (string_fragment) @xstate.actor.name) @xstate.actor
      )
    )
    (array
      (object
        (pair
          key: (property_identifier) @invoke.src (#eq? @invoke.src "src")
          value: (string (string_fragment) @xstate.actor.name) @xstate.actor
        )
      )
    )
  ]
)

(pair
  key: (property_identifier) @key
    (#match? @key "actions|entry|exit")
  value: [
    (call_expression
      function: (identifier) @action.spawnChild (#eq? @action.spawnChild "spawnChild")
      arguments: (arguments (string (string_fragment) @xstate.actor.name)  @xstate.actor)
    )
    (array 
      (call_expression
        function: (identifier) @action.spawnChild (#eq? @action.spawnChild "spawnChild")
        arguments: (arguments (string (string_fragment) @xstate.actor.name)  @xstate.actor)
      )
    )
  ]
)

(call_expression
  function: (identifier) @xstate.spawn (#eq? @xstate.spawn "spawn")
  arguments: (arguments
    (string (string_fragment) @xstate.actor.name) @xstate.actor
  )
)
`;

export const setupDelaysQuery = `
(pair
  key: (property_identifier) @key (#eq? @key "delays")
  value: (object [
    (pair
      key: (property_identifier) @delay.name
      value: (_)
    )
    (shorthand_property_identifier) @delay.name
  ])
)
`;

export const machineDelaysQuery = `
(pair
  key: (property_identifier) @key (#eq? @key "after")
  value: (object
    (pair
      key: (property_identifier) @xstate.delay @xstate.delay.name
      value: (_)
    )
  )
)

(call_expression
	function: (identifier) @action (#match? @action "sendTo|raise")
  arguments: (arguments
		(object
			(pair
				key: (property_identifier) @event (#eq? @event "delay")
				value: (string (string_fragment) @xstate.delay.name) @xstate.delay
			)
		)
  )
)
`;

export const setupImplementableQuery = `
(pair
  key: (property_identifier) @setup.key (#match? @setup.key "actors|actions|guards|delays")
  value: (object [
    (pair
      key: (property_identifier) @implementation.name
      value: (_) @implementation.definition
    )
    (shorthand_property_identifier) @implementation.name @implementation.definition
  ])
)
`;

export const stateQuery = `
(object
  (pair
    key: (property_identifier) @states.key (#eq? @states.key "states")
    value: (object
      (pair
        key: (property_identifier) @xstate.state.name
        value: (object) @xstate.state.config
      ) @xstate.state
    )
  )
)
`;

export const transitionQuery = `
(pair
  key: (property_identifier) @transition.key (#match? @transition.key "after|on$")
  value: (object
    (pair
      key: (_)
      value: [
        (string) @transition.target
        (object
          (pair
            key: (property_identifier) @transition.target.key
            value: (string) @transition.target
          )
        )
        (array
          (object
            (pair
              key: (property_identifier) @transition.target.key (#eq? @transition.target.key "target")
              value: (string) @transition.target
            )
          )
        )
      ]
    )
  )
)

(pair
  key: (property_identifier) @transition.key (#match? @transition.key "onDone|onError|always")
  value: [
    (string) @transition.target
    (object
      (pair
        key: (property_identifier) @transition.target.key
        value: (string) @transition.target
      )
    )
    (array
      (object
        (pair
          key: (property_identifier) @transition.target.key (#eq? @transition.target.key "target")
          value: (string) @transition.target
        )
      )
    )
  ]
)
`;

export const initialStateQuery = `
(pair
  key: (property_identifier) @key (#eq? @key "initial")
  value: (string) @initial.state
)
`;