/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.jboss.as.test.clustering.cluster.ejb.timer.servlet;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.time.Instant;
import java.util.IdentityHashMap;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import jakarta.servlet.ServletException;
import jakarta.servlet.annotation.WebServlet;
import jakarta.servlet.http.HttpServlet;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;

import org.jboss.as.test.clustering.cluster.ejb.timer.beans.AutoPersistentTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.AutoTransientTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.CalendarPersistentTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.CalendarTransientTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.IntervalPersistentTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.IntervalTransientTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.ManualTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.SingleActionPersistentTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.SingleActionTransientTimerBean;
import org.jboss.as.test.clustering.cluster.ejb.timer.beans.TimerBean;
import org.jboss.as.test.clustering.ejb.EJBDirectory;
import org.jboss.as.test.clustering.ejb.LocalEJBDirectory;

/**
 * @author Paul Ferraro
 */
@WebServlet(urlPatterns = { TimerServlet.SERVLET_PATH })
public class TimerServlet extends HttpServlet {
    private static final long serialVersionUID = 5393197778939188763L;

    private static final String SERVLET_NAME = "timer";
    static final String SERVLET_PATH = "/" + SERVLET_NAME;
    private static final String MODULE = "module";

    public static final Set<Class<? extends TimerBean>> PERSISTENT_TIMER_CLASSES = Set.of(AutoPersistentTimerBean.class, CalendarPersistentTimerBean.class, IntervalPersistentTimerBean.class, SingleActionPersistentTimerBean.class);
    public static final Set<Class<? extends TimerBean>> TRANSIENT_TIMER_CLASSES = Set.of(AutoTransientTimerBean.class, CalendarTransientTimerBean.class, IntervalTransientTimerBean.class, SingleActionTransientTimerBean.class);
    public static final Set<Class<? extends TimerBean>> TIMER_CLASSES = Stream.of(PERSISTENT_TIMER_CLASSES, TRANSIENT_TIMER_CLASSES).flatMap(Set::stream).collect(Collectors.toSet());
    public static final Set<Class<? extends ManualTimerBean>> MANUAL_PERSISTENT_TIMER_CLASSES = Set.of(CalendarPersistentTimerBean.class, IntervalPersistentTimerBean.class, SingleActionPersistentTimerBean.class);
    public static final Set<Class<? extends ManualTimerBean>> MANUAL_TRANSIENT_TIMER_CLASSES = Set.of(CalendarTransientTimerBean.class, IntervalTransientTimerBean.class, SingleActionTransientTimerBean.class);
    public static final Set<Class<? extends ManualTimerBean>> MANUAL_TIMER_CLASSES = Stream.of(MANUAL_PERSISTENT_TIMER_CLASSES, MANUAL_TRANSIENT_TIMER_CLASSES).flatMap(Set::stream).collect(Collectors.toSet());
    public static final Set<Class<? extends TimerBean>> AUTO_TIMER_CLASSES = Set.of(AutoPersistentTimerBean.class, AutoTransientTimerBean.class);
    public static final Set<Class<? extends TimerBean>> SINGLE_ACTION_TIMER_CLASSES = Set.of(SingleActionPersistentTimerBean.class, SingleActionTransientTimerBean.class);

    public static URI createURI(URL baseURL, String module) throws URISyntaxException {
        return baseURL.toURI().resolve(new StringBuilder(SERVLET_NAME).append('?').append(MODULE).append('=').append(module).toString());
    }

    @Override
    protected void service(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        request.getServletContext().log(String.format("%s: http://%s:%s%s?%s", request.getMethod(), request.getServerName(), request.getServerPort(), request.getRequestURI(), request.getQueryString()));
        super.service(request, response);
    }

    @Override
    protected void doDelete(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String module = request.getParameter(MODULE);
        try (EJBDirectory directory = new LocalEJBDirectory(module)) {
            for (Class<? extends ManualTimerBean> beanClass : MANUAL_TIMER_CLASSES) {
                ManualTimerBean bean = directory.lookupSingleton(beanClass, ManualTimerBean.class);
                bean.cancel();
            }
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }

    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String module = request.getParameter(MODULE);
        try (EJBDirectory directory = new LocalEJBDirectory(module)) {
            Map<Class<? extends TimerBean>, TimerBean> beans = new IdentityHashMap<>();
            for (Class<? extends ManualTimerBean> beanClass : MANUAL_TIMER_CLASSES) {
                beans.put(beanClass, directory.lookupSingleton(beanClass, ManualTimerBean.class));
            }
            for (Class<? extends TimerBean> beanClass : AUTO_TIMER_CLASSES) {
                beans.put(beanClass, directory.lookupSingleton(beanClass, TimerBean.class));
            }
            for (Map.Entry<Class<? extends TimerBean>, TimerBean> entry : beans.entrySet()) {
                for (Instant timeout : entry.getValue().getTimeouts()) {
                    response.addDateHeader(entry.getKey().getName(), timeout.toEpochMilli());
                }
            }
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }

    @Override
    protected void doHead(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String module = request.getParameter(MODULE);
        try (EJBDirectory directory = new LocalEJBDirectory(module)) {
            Map<Class<? extends TimerBean>, TimerBean> beans = new IdentityHashMap<>();
            for (Class<? extends ManualTimerBean> beanClass : MANUAL_TIMER_CLASSES) {
                beans.put(beanClass, directory.lookupSingleton(beanClass, ManualTimerBean.class));
            }
            for (Class<? extends TimerBean> beanClass : AUTO_TIMER_CLASSES) {
                beans.put(beanClass, directory.lookupSingleton(beanClass, TimerBean.class));
            }
            for (Map.Entry<Class<? extends TimerBean>, TimerBean> entry : beans.entrySet()) {
                response.addIntHeader(entry.getKey().getName(), entry.getValue().getTimers());
            }
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }

    @Override
    protected void doPut(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String module = request.getParameter(MODULE);
        try (EJBDirectory directory = new LocalEJBDirectory(module)) {
            for (Class<? extends ManualTimerBean> beanClass : MANUAL_TIMER_CLASSES) {
                directory.lookupSingleton(beanClass, ManualTimerBean.class).createTimer();
            }
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }
}
