import * as lua from "../../LuaAST";

import { TransformationContext } from "../context";

import { unsupportedProperty, unsupportedSelfFunctionConversion } from "../utils/diagnostics";

import { ContextType, getFunctionContextType } from "../utils/function-context";

import { LuaLibFeature, transformLuaLibFunction } from "../utils/lualib";

import { PropertyCallExpression, transformArguments } from "../visitors/call";

export function transformFunctionPrototypeCall(

context: TransformationContext,

node: PropertyCallExpression

): lua.CallExpression | undefined {

const expression = node.expression;

const callerType = context.checker.getTypeAtLocation(expression.expression);

if (getFunctionContextType(context, callerType) === ContextType.Void) {

context.diagnostics.push(unsupportedSelfFunctionConversion(node));

}

const signature = context.checker.getResolvedSignature(node);

const params = transformArguments(context, node.arguments, signature);

const caller = context.transformExpression(expression.expression);

const expressionName = expression.name.text;

switch (expressionName) {

case "apply":

return transformLuaLibFunction(context, LuaLibFeature.FunctionApply, node, caller, ...params);

case "bind":

return transformLuaLibFunction(context, LuaLibFeature.FunctionBind, node, caller, ...params);

case "call":

return transformLuaLibFunction(context, LuaLibFeature.FunctionCall, node, caller, ...params);

default:

context.diagnostics.push(unsupportedProperty(expression.name, "function", expressionName));

}

}