/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jdbi.v3.sqlobject.internal;

import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.function.BiConsumer;
import java.util.function.Supplier;
import java.util.function.UnaryOperator;

import org.jdbi.v3.core.config.ConfigRegistry;
import org.jdbi.v3.core.extension.ExtensionMethod;
import org.jdbi.v3.core.extension.HandleSupplier;
import org.jdbi.v3.core.internal.Invocations;
import org.jdbi.v3.core.internal.MemoizingSupplier;
import org.jdbi.v3.core.internal.exceptions.Sneaky;
import org.jdbi.v3.sqlobject.GenerateSqlObject;
import org.jdbi.v3.sqlobject.Handler;
import org.jdbi.v3.sqlobject.SqlObject;
import org.jdbi.v3.sqlobject.UnableToCreateSqlObjectException;

public final class SqlObjectInitData {
    private static final Object[] NO_ARGS = new Object[0];
    public static final ThreadLocal<SqlObjectInitData> INIT_DATA = new ThreadLocal<>();

    private final boolean concrete;
    private final Class<?> extensionType;
    private final UnaryOperator<ConfigRegistry> instanceConfigurer;
    private final Map<Method, UnaryOperator<ConfigRegistry>> methodConfigurers;
    private final Map<Method, Handler> methodHandlers;

    public SqlObjectInitData(
            Class<?> extensionType,
            UnaryOperator<ConfigRegistry> instanceConfigurer,
            Map<Method, UnaryOperator<ConfigRegistry>> methodConfigurers,
            Map<Method, Handler> methodHandlers) {
        concrete = isConcrete(extensionType);
        this.extensionType = extensionType;
        this.instanceConfigurer = instanceConfigurer;
        this.methodConfigurers = methodConfigurers;
        this.methodHandlers = methodHandlers;
    }

    public static boolean isConcrete(Class<?> extensionType) {
        return extensionType.getAnnotation(GenerateSqlObject.class) != null;
    }

    public static SqlObjectInitData initData() {
        final SqlObjectInitData result = INIT_DATA.get();
        if (result == null) {
            throw new IllegalStateException("Implemented SqlObject types must be initialized by SqlObjectFactory");
        }
        return result;
    }

    public static Method lookupMethod(String methodName, Class<?>... parameterTypes) {
        return lookupMethod(initData().extensionType, methodName, parameterTypes);
    }

    private static Method lookupMethod(Class<?> klass, String methodName, Class<?>... parameterTypes) {
        try {
            return klass.getMethod(methodName, parameterTypes);
        } catch (NoSuchMethodException | SecurityException e) {
            try {
                return klass.getDeclaredMethod(methodName, parameterTypes);
            } catch (Exception x) {
                e.addSuppressed(x);
                try {
                    return SqlObject.class.getMethod(methodName, parameterTypes);
                } catch (Exception x2) {
                    e.addSuppressed(x2);
                }
            }
            throw new IllegalStateException(
                    String.format("can't find %s#%s%s", klass.getName(), methodName, Arrays.asList(parameterTypes)), e);
        }
    }

    public boolean isConcrete() {
        return concrete;
    }

    public Class<?> extensionType() {
        return extensionType;
    }

    public <E> E instantiate(Class<E> passExtensionType, HandleSupplier handle, ConfigRegistry instanceConfig) {
        if (!extensionType.equals(passExtensionType)) {
            throw new IllegalArgumentException("mismatch extension type");
        }
        try {
            return Invocations.invokeWith(SqlObjectInitData.INIT_DATA, this, () ->
                passExtensionType.cast(
                    Class.forName(extensionType.getPackage().getName() + "." + extensionType.getSimpleName() + "Impl")
                        .getConstructor(HandleSupplier.class, ConfigRegistry.class)
                        .newInstance(handle, instanceConfig)));
        } catch (Exception | ExceptionInInitializerError e) {
            throw new UnableToCreateSqlObjectException(e);
        }
    }

    public ConfigRegistry configureInstance(ConfigRegistry config) {
        return instanceConfigurer.apply(config);
    }

    public void forEachMethodHandler(BiConsumer<Method, Handler> action) {
        methodHandlers.forEach(action);
    }

    public Supplier<InContextInvoker> lazyInvoker(Object target, Method method, HandleSupplier handle, ConfigRegistry instanceConfig) {
        return MemoizingSupplier.of(() -> {
            ExtensionMethod extensionMethod = new ExtensionMethod(extensionType, method);
            ConfigRegistry methodConfig = methodConfigurers.get(method).apply(instanceConfig.createCopy());
            Handler methodHandler = methodHandlers.get(method);
            methodHandler.warm(methodConfig);
            return new InContextInvoker() {
                @Override
                public Object invoke(Object[] args) {
                    return call(() -> methodHandler.invoke(target, args == null ? NO_ARGS : args, handle));
                }

                @Override
                public Object call(Callable<?> task) {
                    try {
                        return handle.invokeInContext(
                                extensionMethod,
                                methodConfig,
                                task);
                    } catch (Exception x) {
                        throw Sneaky.throwAnyway(x);
                    }
                }
            };
        });
    }

    public interface InContextInvoker {
        Object invoke(Object[] args);
        Object call(Callable<?> task);
        default Object call(Runnable task) { // NOPMD
            return call(() -> {
                task.run();
                return null;
            });
        }
    }
}

@@ -978,54 +978,8 @@
 er;%0A
-import org.jdbi.v3.core.internal.Invocations;%0A
 impo
@@ -4056,32 +4056,33 @@
 pe%22);%0A        %7D%0A
+%0A
         try %7B%0A  
@@ -4095,38 +4095,8 @@
     
-return Invocations.invokeWith(
 SqlO
@@ -4122,21 +4122,19 @@
 DATA
-, this, () -%3E
+.set(this);
 %0A   
@@ -4142,19 +4142,22 @@
         
-   
+return
  passExt
@@ -4173,20 +4173,16 @@
 e.cast(%0A
-    
         
@@ -4308,20 +4308,16 @@
         
-   
- 
 .getCons
@@ -4388,20 +4388,16 @@
         
-    
 .newInst
@@ -4425,17 +4425,16 @@
 Config))
-)
 ;%0A      
@@ -4544,24 +4544,94 @@
 ception(e);%0A
+        %7D finally %7B%0A            SqlObjectInitData.INIT_DATA.remove();%0A
         %7D%0A  

Remove invocations execution