import type {
  ArrowFunctionExpression,
  CallExpression,
  Expression,
  FunctionExpression,
  Identifier,
  Literal,
  NodeListenerMap,
  ObjectExpression,
  ParamNode,
  Property,
  SpreadElement,
  TemplateLiteral,
  TSCallSignatureDeclaration,
  TSFunctionType,
  TSLiteralType,
  TSMethodSignature,
  TSPropertySignature
} from './ast';
export type VueObjectType = 'mark' | 'export' | 'definition' | 'instance';
export type VueObjectData = {
  node: ObjectExpression;
  type: VueObjectType;
  parent: VueObjectData | null;
  functional: boolean;
};
type VueVisitorBase = {
  [T in keyof NodeListenerMap]?: (node: NodeListenerMap[T], obj: VueObjectData) => void;
};
export interface VueVisitor extends VueVisitorBase {
  [query: string]: ((node: ParamNode, obj: VueObjectData) => void) | undefined;
  onVueObjectEnter?(node: ObjectExpression, obj: VueObjectData): void;
  onVueObjectExit?(node: ObjectExpression, obj: VueObjectData): void;
  onSetupFunctionEnter?(
    node: (FunctionExpression | ArrowFunctionExpression) & { parent: Property },
    obj: VueObjectData
  ): void;
  onSetupFunctionExit?(
    node: (FunctionExpression | ArrowFunctionExpression) & { parent: Property },
    obj: VueObjectData
  ): void;
  onRenderFunctionEnter?(
    node: (FunctionExpression | ArrowFunctionExpression) & { parent: Property },
    obj: VueObjectData
  ): void;
}

type ScriptSetupVisitorBase = {
  [T in keyof NodeListenerMap]?: (node: NodeListenerMap[T]) => void;
};
export interface ScriptSetupVisitor extends ScriptSetupVisitorBase {
  [query: string]:
    | ((node: ParamNode) => void)
    | ((node: CallExpression, props: Array<ComponentArrayProp | ComponentObjectProp | ComponentTypeProp>) => void)
    | ((node: CallExpression, props: Array<ComponentArrayEmit | ComponentObjectEmit | ComponentTypeEmit>) => void)
    | undefined;
  onDefinePropsEnter?(
    node: CallExpression,
    props: Array<ComponentArrayProp | ComponentObjectProp | ComponentTypeProp>
  ): void;
  onDefinePropsExit?(
    node: CallExpression,
    props: Array<ComponentArrayProp | ComponentObjectProp | ComponentTypeProp>
  ): void;
  onDefineEmitsEnter?(
    node: CallExpression,
    props: Array<ComponentArrayEmit | ComponentObjectEmit | ComponentTypeEmit>
  ): void;
  onDefineEmitsExit?(
    node: CallExpression,
    props: Array<ComponentArrayEmit | ComponentObjectEmit | ComponentTypeEmit>
  ): void;
}

type ComponentArrayPropDetectName = {
  type: 'array';
  key: Literal | TemplateLiteral;
  propName: string;
  value: null;
  node: Expression | SpreadElement;
};
type ComponentArrayPropUnknownName = {
  type: 'array';
  key: null;
  propName: null;
  value: null;
  node: Expression | SpreadElement;
};
export type ComponentArrayProp = ComponentArrayPropDetectName | ComponentArrayPropUnknownName;

type ComponentObjectPropDetectName = {
  type: 'object';
  key: Expression;
  propName: string;
  value: Expression;
  node: Property;
};
type ComponentObjectPropUnknownName = {
  type: 'object';
  key: null;
  propName: null;
  value: Expression;
  node: Property;
};
export type ComponentObjectProp = ComponentObjectPropDetectName | ComponentObjectPropUnknownName;

export type ComponentTypeProp = {
  type: 'type';
  key: Identifier;
  propName: string;
  value: null;
  node: TSPropertySignature | TSMethodSignature;

  required: boolean;
  types: string[];
};

type ComponentArrayEmitDetectName = {
  type: 'array';
  key: Literal | TemplateLiteral;
  emitName: string;
  value: null;
  node: Expression | SpreadElement;
};
type ComponentArrayEmitUnknownName = {
  type: 'array';
  key: null;
  emitName: null;
  value: null;
  node: Expression | SpreadElement;
};
export type ComponentArrayEmit = ComponentArrayEmitDetectName | ComponentArrayEmitUnknownName;
type ComponentObjectEmitDetectName = {
  type: 'object';
  key: Expression;
  emitName: string;
  value: Expression;
  node: Property;
};
type ComponentObjectEmitUnknownName = {
  type: 'object';
  key: null;
  emitName: null;
  value: Expression;
  node: Property;
};
export type ComponentObjectEmit = ComponentObjectEmitDetectName | ComponentObjectEmitUnknownName;

export type ComponentTypeEmit = {
  type: 'type';
  key: TSLiteralType;
  emitName: string;
  value: null;
  node: TSCallSignatureDeclaration | TSFunctionType;
};