package com.megatim.fdxconsultation.service.impl.sockets.helpers;

import com.fdx.messaging.common.ConnexionStatut;
import com.megatim.fdxconsultation.core.ifaces.supervision.StandaloneServerStateEntityManager;
import com.megatim.fdxconsultation.model.supervision.StandaloneServerStateEntity;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.ConcurrentHashMap;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.stream.Collectors;
import javax.enterprise.context.ApplicationScoped;
import javax.inject.Inject;
import javax.websocket.EncodeException;
import javax.websocket.Session;

/**
 *
 * @author Gabuntu
 */
@ApplicationScoped
public class ConnexionStatutSessionHandler {

    @Inject
    StandaloneServerStateEntityManager serverStateManager;

    private final Map<String, Session> idToSessions = new ConcurrentHashMap<>();

    private Timer timer = new Timer();

    public void publishMessage(List<ConnexionStatut> connectedStatuts) {
        getSessions().forEach(s -> {
            try {
                s.getBasicRemote().sendObject(connectedStatuts);
            } catch (IOException | EncodeException ex) {
                Logger.getLogger(getClass().getName()).log(Level.SEVERE, null, ex);
            }
        });

        scheduleTask();

    }

    public void addSession(Session session) {
        idToSessions.put(session.getId(), session);
    }

    public void removeSession(Session session) {
        idToSessions.remove(session.getId());
    }

    private List<Session> getSessions() {
        return idToSessions
                .entrySet()
                .stream()
                .map(e -> e.getValue())
                .collect(Collectors.toList());
    }

    private void scheduleTask() {

        timer.cancel();

        final long lastPublishTime = System.nanoTime();
        timer = new Timer();
        timer.schedule(new TimerTask() {
            @Override
            public void run() {
                if (System.nanoTime() - lastPublishTime >= 10_000_000_000L) {
                    StandaloneServerStateEntity stateEntity = serverStateManager.getCurrentStandaloneServerStateEntity();
                    if (stateEntity != null) {
                        List<ConnexionStatut> connectedStatuts = stateEntity.getAgentConfigurations().stream().map(a -> {
                            ConnexionStatut connexionStatut = new ConnexionStatut();
                            connexionStatut.setAgentCode(a.getAgentCode());
                            connexionStatut.setConnected(false);
                            return connexionStatut;
                        }).collect(Collectors.toList());

                        publishMessage(connectedStatuts);
                    }
                }
            }
        }, 0, 5000);

    }
}