/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.camel.quarkus.component.fhir.deployment;

import java.lang.reflect.Modifier;
import java.util.function.BooleanSupplier;

import ca.uhn.fhir.rest.client.apache.ApacheRestfulClientFactory;
import ca.uhn.fhir.rest.client.api.IClientInterceptor;
import ca.uhn.fhir.rest.server.exceptions.BaseServerResponseException;
import ca.uhn.fhir.util.jar.DependencyLogImpl;
import ca.uhn.fhir.validation.schematron.SchematronBaseValidator;
import io.quarkus.deployment.GeneratedClassGizmoAdaptor;
import io.quarkus.deployment.annotations.BuildProducer;
import io.quarkus.deployment.annotations.BuildStep;
import io.quarkus.deployment.builditem.AdditionalApplicationArchiveMarkerBuildItem;
import io.quarkus.deployment.builditem.CombinedIndexBuildItem;
import io.quarkus.deployment.builditem.ExtensionSslNativeSupportBuildItem;
import io.quarkus.deployment.builditem.FeatureBuildItem;
import io.quarkus.deployment.builditem.GeneratedClassBuildItem;
import io.quarkus.deployment.builditem.nativeimage.NativeImageResourceBundleBuildItem;
import io.quarkus.deployment.builditem.nativeimage.ReflectiveClassBuildItem;
import io.quarkus.deployment.pkg.steps.NativeOrNativeSourcesBuild;
import io.quarkus.gizmo.ClassCreator;
import io.quarkus.gizmo.Gizmo;
import io.quarkus.gizmo.MethodCreator;
import org.apache.camel.quarkus.component.fhir.IsSchematronAbsent;
import org.jboss.jandex.DotName;
import org.jboss.jandex.IndexView;

final class FhirProcessor {
    private static final String FEATURE = "camel-fhir";
    private static final String RESTFUL_SERVER_UTILS_CLASS_NAME = "ca.uhn.fhir.rest.server.RestfulServerUtils";
    private static final String[] INTERCEPTOR_CLASSES = {
            IClientInterceptor.class.getName(),
    };

    @BuildStep
    FeatureBuildItem feature() {
        return new FeatureBuildItem(FEATURE);
    }

    @BuildStep
    ExtensionSslNativeSupportBuildItem activateSslNativeSupport() {
        return new ExtensionSslNativeSupportBuildItem(FEATURE);
    }

    @BuildStep()
    NativeImageResourceBundleBuildItem hapiMessages() {
        return new NativeImageResourceBundleBuildItem("ca.uhn.fhir.i18n.hapi-messages");
    }

    @BuildStep
    AdditionalApplicationArchiveMarkerBuildItem hl7ArchiveMarker() {
        return new AdditionalApplicationArchiveMarkerBuildItem("org/hl7/fhir");
    }

    @BuildStep
    AdditionalApplicationArchiveMarkerBuildItem hapiArchiveMarker() {
        return new AdditionalApplicationArchiveMarkerBuildItem("ca/uhn/fhir");
    }

    @BuildStep()
    void registerForReflection(
            CombinedIndexBuildItem combinedIndex,
            BuildProducer<ReflectiveClassBuildItem> reflectiveClass) {

        IndexView index = combinedIndex.getIndex();
        index.getAllKnownSubclasses(DotName.createSimple(BaseServerResponseException.class.getName()))
                .stream()
                .map(classInfo -> ReflectiveClassBuildItem.builder(classInfo.name().toString())
                        .build())
                .forEach(reflectiveClass::produce);

        String[] clientInterceptors = index.getAllKnownImplementations(DotName.createSimple(IClientInterceptor.class.getName()))
                .stream()
                .map(classInfo -> classInfo.name().toString())
                .toArray(String[]::new);

        reflectiveClass.produce(ReflectiveClassBuildItem.builder(clientInterceptors).methods().build());
        reflectiveClass.produce(ReflectiveClassBuildItem.builder(INTERCEPTOR_CLASSES).methods().build());
        reflectiveClass.produce(ReflectiveClassBuildItem.builder(DependencyLogImpl.class)
                .fields().build());
        reflectiveClass.produce(ReflectiveClassBuildItem.builder(ApacheRestfulClientFactory.class)
                .methods().fields().build());
    }

    @BuildStep(onlyIfNot = IsSchematronAbsent.class)
    void registerSchematronForReflection(BuildProducer<ReflectiveClassBuildItem> reflectiveClass) {
        reflectiveClass.produce(ReflectiveClassBuildItem.builder(SchematronBaseValidator.class)
                .fields().build());
    }

    @BuildStep(onlyIf = { NativeOrNativeSourcesBuild.class, IsFhirServerAbsent.class })
    void generateRestfulServerUtils(BuildProducer<GeneratedClassBuildItem> generatedClass) {
        // Avoid having redundant hapi-fhir-server on the classpath and generate RestfulServerUtils.createEtag to satisfy native compilation.
        try (ClassCreator classCreator = ClassCreator.builder()
                .className(RESTFUL_SERVER_UTILS_CLASS_NAME)
                .classOutput(new GeneratedClassGizmoAdaptor(generatedClass, false))
                .setFinal(true)
                .superClass(Object.class.getName())
                .build()) {

            /*
             * Original implementation of createETag is:
             *
             * 	public static String createEtag(String theVersionId) {
             * 		return "W/\"" + theVersionId + '"';
             *  }
             */
            try (MethodCreator methodCreator = classCreator.getMethodCreator("createEtag", String.class, String.class)) {
                methodCreator.setModifiers(Modifier.PUBLIC | Modifier.STATIC);

                Gizmo.StringBuilderGenerator stringBuilder = Gizmo.newStringBuilder(methodCreator);
                stringBuilder.append("W/");
                stringBuilder.append('"');
                stringBuilder.append(methodCreator.getMethodParam(0));
                stringBuilder.append('"');

                methodCreator.returnValue(stringBuilder.callToString());
            }
        }
    }

    static final class IsFhirServerAbsent implements BooleanSupplier {
        @Override
        public boolean getAsBoolean() {
            try {
                Class.forName(RESTFUL_SERVER_UTILS_CLASS_NAME);
                return false;
            } catch (ClassNotFoundException e) {
                return true;
            }
        }
    }
}
