package com.lumaserv.bgp;

import com.lumaserv.bgp.protocol.BGPPacket;
import com.lumaserv.bgp.protocol.message.BGPOpen;
import lombok.Getter;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

public class BGPServer implements Runnable {

    final ServerSocket serverSocket;
    @Getter
    final List<BGPSessionConfiguration> sessionConfigurations = new ArrayList<>();
    final List<BGPSession> sessions = new ArrayList<>();

    public BGPServer() throws IOException {
        this(179);
    }

    public BGPServer(int port) throws IOException {
        serverSocket = new ServerSocket(port);
    }

    private static boolean checkEqual(byte[] a, byte[] b) {
        if(a == b)
            return true;
        if(a == null || b == null)
            return false;
        if(a.length != b.length)
            return false;
        for(int i=0; i<a.length; i++) {
            if(a[i] != b[i])
                return false;
        }
        return true;
    }

    public void run() {
        while (true) {
            try {
                Socket socket = serverSocket.accept();
                System.out.println("Accept");

                BGPSessionConfiguration config = sessionConfigurations.stream()
                        .filter(c -> socket.getInetAddress().equals(c.getRemoteAddr()))
                        .findFirst()
                        .orElse(null);
                System.out.println("Config: " + config);
                if(config == null) {
                    System.out.println("Peer not found:" + socket.getInetAddress());
                    socket.close();
                    continue;
                }

                BGPFsm fsm = new BGPFsm();
                BGPSession session = new BGPSession(socket, config, fsm);
                fsm.setSession(session);
                sessions.add(session);
                System.out.println("automaticStartPassive");
                fsm.getCurrentState().automaticStartPassive();

                // System.out.println("tcpConnectionConfirmed in " + fsm.getCurrentState());
                // fsm.getCurrentState().tcpConnectionConfirmed();
            } catch (IOException ex) {
                ex.printStackTrace();
            }
        }
    }

    public boolean connect(BGPSessionConfiguration config, String host) throws IOException {
        return connect(config, host, 179);
    }

    public boolean connect(BGPSessionConfiguration config, String host, int port) throws IOException {
        try {
            BGPFsm fsm = new BGPFsm();
            BGPSession session = new BGPSession(config, fsm, host, port);
            fsm.setSession(session);
            sessions.add(session);
            fsm.getCurrentState().automaticStart();
            return true;
        } catch (IOException ex) {
            ex.printStackTrace();
            throw(ex);
        }
    }

    public void shutdown() {
        for(BGPSession session : sessions) {
            session.automaticStop();
        }
    }
}