index.ts

export const effectCreator = `ClassProperty[value.callee.name='createEffect']`

export const effectDecorator = `Decorator[expression.callee.name='Effect']`
export const classPropertyWithEffectDecorator =
  `ClassDeclaration > ClassBody > ClassProperty > ${effectDecorator}` as const

export const actionCreator = `CallExpression[callee.name='createAction']`
export const actionCreatorWithLiteral =
  `${actionCreator}[arguments.0.type='Literal']` as const
export const actionCreatorProps = `${actionCreator} CallExpression` as const
export const actionCreatorPropsComputed =
  `${actionCreatorProps} > TSTypeParameterInstantiation > :matches(TSTypeReference[typeName.name!='Readonly'], [type=/^TS(.*)(Keyword|Type)$/])` as const

export const constructorExit = `MethodDefinition[kind='constructor']:exit`

export const dispatchInEffects = (storeName: string) =>
  `ClassProperty > CallExpression:has(Identifier[name='createEffect']) CallExpression > MemberExpression:has(Identifier[name='dispatch']):has(MemberExpression > Identifier[name='${storeName}'])`

export const injectedStore = `MethodDefinition[kind='constructor'] Identifier[typeAnnotation.typeAnnotation.typeName.name='Store']`
export const typedStore = `MethodDefinition[kind='constructor'] Identifier>TSTypeAnnotation>TSTypeReference[typeName.name='Store'][typeParameters.params]`

export const ngModuleDecorator = `ClassDeclaration > Decorator > CallExpression[callee.name='NgModule']`

export const ngModuleProviders =
  `${ngModuleDecorator} ObjectExpression Property[key.name='providers'] > ArrayExpression Identifier` as const

export const ngModuleImports =
  `${ngModuleDecorator} ObjectExpression Property[key.name='imports'] > ArrayExpression CallExpression[callee.object.name='EffectsModule'][callee.property.name=/forRoot|forFeature/] ArrayExpression > Identifier` as const

export const actionDispatch = (storeName: string) =>
  `ExpressionStatement > CallExpression:matches([callee.object.name='${storeName}'][callee.property.name='dispatch'], [callee.object.object.type='ThisExpression'][callee.object.property.name='${storeName}'][callee.property.name='dispatch'])`

export const storeExpression = (storeName: string) =>
  `CallExpression:matches([callee.object.name='${storeName}'], [callee.object.object.type='ThisExpression'][callee.object.property.name='${storeName}'])`

export const storeExpressionCallable = (storeName: string) =>
  `CallExpression:matches([callee.object.callee.object.name='${storeName}'], [callee.object.callee.object.object.type='ThisExpression'][callee.object.callee.object.property.name='${storeName}'])`

export const storePipe = (storeName: string) =>
  `${storeExpression(storeName)}[callee.property.name='pipe']`

export const pipeableSelect = (storeName: string) =>
  `${storePipe(storeName)} CallExpression[callee.name='select']`

export const storeSelect = (storeName: string) =>
  `${storeExpression(storeName)}[callee.property.name='select']`

export const createReducer = `CallExpression[callee.name='createReducer']`

export const onFunctionWithoutType =
  `${createReducer} CallExpression[callee.name='on'] > ArrowFunctionExpression:not([returnType.typeAnnotation],:has(CallExpression))` as const

export const storeActionReducerMap =
  `${ngModuleDecorator} ObjectExpression Property[key.name='imports'] > ArrayExpression CallExpression[callee.object.name='StoreModule'][callee.property.name=/forRoot|forFeature/] > ObjectExpression:first-child > Property` as const

export const actionReducerMap = `VariableDeclarator[id.typeAnnotation.typeAnnotation.typeName.name='ActionReducerMap'] > ObjectExpression > Property`

export const createEffectExpression = `ClassProperty > CallExpression[callee.name='createEffect']`

const mapOperators = '(concat|exhaust|flat|merge|switch)Map'
const mapToOperators = '(concat|merge|switch)MapTo'
const mapOperatorsExpression =
  `${createEffectExpression} CallExpression[callee.name=/^${mapOperators}$/]` as const
const mapToOperatorsExpression =
  `${createEffectExpression} CallExpression[callee.name=/^${mapToOperators}$/]` as const

export const createEffectBody =
  `${createEffectExpression} > ArrowFunctionExpression` as const

export const effectsImplicitReturn =
  `${mapOperatorsExpression} > ArrowFunctionExpression > ArrayExpression, ${mapToOperatorsExpression} ArrayExpression` as const

export const effectsReturn =
  `${mapOperatorsExpression} ReturnStatement` as const