import { closestDataStack, mergeProxies } from './scope'
import { injectMagics } from './magics'
import { tryCatch, handleError } from './utils/error'

export let shouldAutoEvaluateFunctions = true

export function dontAutoEvaluateFunctions(callback) {
    let cache = shouldAutoEvaluateFunctions

    shouldAutoEvaluateFunctions = false

    let result = callback()

    shouldAutoEvaluateFunctions = cache

    return result
}

export function evaluate(el, expression, extras = {}) {
    let result

    evaluateLater(el, expression)(value => result = value, extras)

    return result
}

export function evaluateLater(...args) {
    return theEvaluatorFunction(...args)
}

let theEvaluatorFunction = normalEvaluator

export function setEvaluator(newEvaluator) {
    theEvaluatorFunction = newEvaluator
}

let theRawEvaluatorFunction

export function setRawEvaluator(newEvaluator) {
    theRawEvaluatorFunction = newEvaluator
}

export function normalEvaluator(el, expression) {
    let overriddenMagics = {}

    injectMagics(overriddenMagics, el)

    let dataStack = [overriddenMagics, ...closestDataStack(el)]

    let evaluator = (typeof expression === 'function')
        ? generateEvaluatorFromFunction(dataStack, expression)
        : generateEvaluatorFromString(dataStack, expression, el)

    return tryCatch.bind(null, el, expression, evaluator)
}

export function generateEvaluatorFromFunction(dataStack, func) {
    return (receiver = () => {}, { scope = {}, params = [], context } = {}) => {
        // If auto-evaluation is disabled, pass the function itself instead of calling it
        if (! shouldAutoEvaluateFunctions) {
            runIfTypeOfFunction(receiver, func, mergeProxies([scope, ...dataStack]), params)

            return
        }

        let result = func.apply(mergeProxies([scope, ...dataStack]), params)

        runIfTypeOfFunction(receiver, result)
    }
}

let evaluatorMemo = {}

function generateFunctionFromString(expression, el) {
    if (evaluatorMemo[expression]) {
        return evaluatorMemo[expression]
    }

    let AsyncFunction = Object.getPrototypeOf(async function(){}).constructor

    // Some expressions that are useful in Alpine are not valid as the right side of an expression.
    // Here we'll detect if the expression isn't valid for an assignment and wrap it in a self-
    // calling function so that we don't throw an error AND a "return" statement can be used.
    let rightSideSafeExpression = 0
        // Support expressions starting with "if" statements like: "if (...) doSomething()"
        || /^[\n\s]*if.*\(.*\)/.test(expression.trim())
        // Support expressions starting with "let/const" like: "let foo = 'bar'"
        || /^(let|const)\s/.test(expression.trim())
            ? `(async()=>{ ${expression} })()`
            : expression

    const safeAsyncFunction = () => {
        try {
            let func = new AsyncFunction(
                ["__self", "scope"],
                `with (scope) { __self.result = ${rightSideSafeExpression} }; __self.finished = true; return __self.result;`
            )

            Object.defineProperty(func, "name", {
                value: `[Alpine] ${expression}`,
            })

            return func
        } catch ( error ) {
            handleError( error, el, expression )
            return Promise.resolve()
        }
    }
    let func = safeAsyncFunction()

    evaluatorMemo[expression] = func

    return func
}

function generateEvaluatorFromString(dataStack, expression, el) {
    let func = generateFunctionFromString(expression, el)

    return (receiver = () => {}, { scope = {}, params = [], context } = {}) => {
        func.result = undefined
        func.finished = false

        // Run the function.

        let completeScope = mergeProxies([ scope, ...dataStack ])

        if (typeof func === 'function' ) {
            let promise = func.call(context, func, completeScope).catch((error) => handleError(error, el, expression))

            // Check if the function ran synchronously,
            if (func.finished) {
                // Return the immediate result.
                runIfTypeOfFunction(receiver, func.result, completeScope, params, el)
                // Once the function has run, we clear func.result so we don't create
                // memory leaks. func is stored in the evaluatorMemo and every time
                // it runs, it assigns the evaluated expression to result which could
                // potentially store a reference to the DOM element that will be removed later on.
                func.result = undefined
            } else {
                // If not, return the result when the promise resolves.
                promise.then(result => {
                    runIfTypeOfFunction(receiver, result, completeScope, params, el)
                }).catch( error => handleError( error, el, expression ) )
                .finally( () => func.result = undefined )
            }
        }
    }
}

export function runIfTypeOfFunction(receiver, value, scope, params, el) {
    if (shouldAutoEvaluateFunctions && typeof value === 'function') {
        let result = value.apply(scope, params)

        if (result instanceof Promise) {
            result.then(i => runIfTypeOfFunction(receiver, i, scope, params)).catch( error => handleError( error, el, value ) )
        } else {
            receiver(result)
        }
    } else if (typeof value === 'object' && value instanceof Promise) {
        value.then(i => receiver(i))
    } else {
        receiver(value)
    }
}

export function evaluateRaw(...args) {
    return theRawEvaluatorFunction(...args)
}

export function normalRawEvaluator(el, expression, extras = {}) {
    let overriddenMagics = {}

    injectMagics(overriddenMagics, el)

    let dataStack = [overriddenMagics, ...closestDataStack(el)]

    let scope = mergeProxies([extras.scope ?? {}, ...dataStack])

    let params = extras.params ?? []

    if (expression.includes('await')) {
        let AsyncFunction = Object.getPrototypeOf(async function(){}).constructor

        // Some expressions that are useful in Alpine are not valid as the right side of an expression.
        // Here we'll detect if the expression isn't valid for an assignment and wrap it in a self-
        // calling function so that we don't throw an error AND a "return" statement can be used.
        let rightSideSafeExpression = 0
            // Support expressions starting with "if" statements like: "if (...) doSomething()"
            || /^[\n\s]*if.*\(.*\)/.test(expression.trim())
            // Support expressions starting with "let/const" like: "let foo = 'bar'"
            || /^(let|const)\s/.test(expression.trim())
                ? `(async()=>{ ${expression} })()`
                : expression

        let func = new AsyncFunction(
            ["scope"],
            `with (scope) { let __result = ${rightSideSafeExpression}; return __result }`
        )

        let result = func.call(extras.context, scope)

        return result
    } else {
        // Some expressions that are useful in Alpine are not valid as the right side of an expression.
        // Here we'll detect if the expression isn't valid for an assignment and wrap it in a self-
        // calling function so that we don't throw an error AND a "return" statement can be used.
        let rightSideSafeExpression = 0
            // Support expressions starting with "if" statements like: "if (...) doSomething()"
            || /^[\n\s]*if.*\(.*\)/.test(expression.trim())
            // Support expressions starting with "let/const" like: "let foo = 'bar'"
            || /^(let|const)\s/.test(expression.trim())
                ? `(()=>{ ${expression} })()`
                : expression

        let func = new Function(
            ["scope"],
            `with (scope) { let __result = ${rightSideSafeExpression}; return __result }`
        )

        let result = func.call(extras.context, scope)

        // If the result is a function, call it
        if (typeof result === 'function' && shouldAutoEvaluateFunctions) {
            return result.apply(scope, params)
        }

        return result
    }
}
