import { defineComponent, withApiName, validateCustomApiName, type ComponentDefinition } from './component';
import {
  fieldReferences,
  type ObjectScoped,
  manifestObject,
  manifestField,
  type ObjectReference,
  type DefinitionFieldReference,
} from './custom-object';
import type { Field } from '../field';
import type { AnyCustomObjectDefinition, FieldValue } from '../../../core/platform/custom-objects/types';
import { Rule, type RuleProps, type RuleAction } from '../rule';

/** Flow actions accept shared writable field definitions as well as existing manifest fields. */
export type DefinedRuleAction<TObject extends ObjectReference> =
  | (Omit<RuleAction, 'targetField'> & { targetField: ObjectScoped<TObject, Field> })
  | (TObject extends AnyCustomObjectDefinition ?
      {
        [K in keyof TObject['fields']]: TypedRuleAction<TObject['fields'][K]>;
      }[keyof TObject['fields']]
    : never);

export type DefineRecordTriggeredFlowProps<TObject extends ObjectReference> = Omit<
  RuleProps,
  'flowId' | 'apiName' | 'actions'
> & {
  apiName: `${string}__c`;
  actions?: readonly DefinedRuleAction<NoInfer<TObject>>[];
};

export function defineRecordTriggeredFlow<TObject extends ObjectReference>(
  object: TObject,
  props: DefineRecordTriggeredFlowProps<TObject>,
): Rule {
  validateCustomApiName(props.apiName);
  const { actions: inputActions, ...rest } = props;
  const actions = inputActions?.map((action) => ({
    ...action,
    targetField: manifestField(object, action.targetField as DefinitionFieldReference<TObject>),
  }));
  return recordTriggeredFlowComponent.register(
    withApiName(
      new Rule(manifestObject(object), {
        ...rest,
        flowId: rest.apiName,
        ...(actions != null ? { actions } : {}),
      }),
      rest.apiName,
    ),
    object,
    props,
  );
}

type TypedRuleAction<TField> =
  TField extends { writable: true } ?
    {
      targetField: TField;
      fixedValue?: FieldValue<TField>;
      rqlFormula?: string;
    }
  : never;

const recordTriggeredFlowComponent: ComponentDefinition<typeof defineRecordTriggeredFlow> = defineComponent({
  type: 'CUSTOM_OBJECT_RECORD_TRIGGERED_FLOW',
  define: defineRecordTriggeredFlow,
  references: (_definition, object, props) => [
    object,
    ...(props.actions ?? []).flatMap((action) => fieldReferences(object, action.targetField)),
  ],
});