/*
 * Copyright 2014-2017 Hewlett-Packard Enterprise Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.frontend.find.core.beanconfiguration;

import org.apache.catalina.WebResourceRoot;
import org.apache.catalina.connector.Connector;
import org.apache.catalina.webresources.StandardRoot;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.context.embedded.EmbeddedServletContainerFactory;
import org.springframework.boot.context.embedded.tomcat.TomcatEmbeddedServletContainerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class TomcatConfig {
    @Value("${server.reverseProxy}")
    private boolean useReverseProxy;

    @Value("${server.ajp.port}")
    private int ajpPort;

    @Value("${server.tomcat.resources.max-cache-kb}")
    private long webResourcesCacheSize;

    @Value("${server.tomcat.connector.max-post-size}")
    private int connectorMaxPostSize;

    @Bean
    public EmbeddedServletContainerFactory servletContainer() {
        final TomcatEmbeddedServletContainerFactory tomcat = new TomcatEmbeddedServletContainerFactory();

        if(useReverseProxy) {
            tomcat.addAdditionalTomcatConnectors(createAjpConnector());
        }

        // Set the web resources cache size (this defaults to 10MB but that is too small for Find)
        tomcat.addContextCustomizers(context -> {
            final WebResourceRoot resources = new StandardRoot(context);
            resources.setCacheMaxSize(webResourcesCacheSize);
            context.setResources(resources);
        });

        tomcat.addConnectorCustomizers(connector -> {
            connector.setMaxPostSize(connectorMaxPostSize);
        });

        return tomcat;
    }

    private Connector createAjpConnector() {
        final Connector connector = new Connector("AJP/1.3");
        connector.setPort(ajpPort);
        connector.setAttribute("tomcatAuthentication", false);
        return connector;
    }
}