package jmri.jmrix.dccpp.dccppovertcp;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.NoSuchElementException;
import jmri.jmrix.dccpp.DCCppCommandStation;
import jmri.jmrix.dccpp.DCCppListener;
import jmri.jmrix.dccpp.DCCppMessage;
import jmri.jmrix.dccpp.DCCppNetworkPortController;
import jmri.jmrix.dccpp.DCCppPacketizer;
import jmri.jmrix.dccpp.DCCppReply;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.annotation.concurrent.GuardedBy;

/**
 * Converts Stream-based I/O to/from DCC++ messages. The "DCCppInterface" side
 * sends/receives DCCppMessage objects. The connection to a
 * DCCppPortnetworkController is via a pair of *Streams, which then carry
 * sequences of characters for transmission.
 * <p>
 * Messages come to this via the main GUI thread, and are forwarded back to
 * listeners in that same thread. Reception and transmission are handled in
 * dedicated threads by RcvHandler and XmtHandler objects. Those are internal
 * classes defined here. The thread priorities are:
 * <ul>
 * <li> RcvHandler - at highest available priority
 * <li> XmtHandler - down one, which is assumed to be above the GUI
 * <li> (everything else)
 * </ul>
 *
 * @author Bob Jacobsen Copyright (C) 2001
 * @author Alex Shepherd Copyright (C) 2003, 2006
 * @author Mark Underwood Copyright (C) 2015
 *
 * Based on jmri.jmrix.loconet.loconetovertcp.LnOverTcpPacketizer
 *
 */
// TODO: Consider ditching the LocoNet-inherited "RECEIVE" and "SEND" prefixes
// and just rely on the already-present "<" and ">" to mark start and end
// of frame.  This would pretty much make DCCppOverTCP redundant with the
// Network Port interface to the Base Station (that is, the "host" JMRI
// application would look just like a Network Base Station to the "client" JMRI
// application).
//
// However, at minimum, this would break backward compatibility for the interface,
// so there is that to consider.  Probably best to do this sooner than later,
// to minimize that impact.
//
public class DCCppOverTcpPacketizer extends DCCppPacketizer {

    static final String OLD_RECEIVE_PREFIX = "RECEIVE ";
    static final String OLD_SEND_PREFIX = "SEND";
    static final String RECEIVE_PREFIX = "<";
    static final String SEND_PREFIX = ""; // Making this an empty string on purpose.
    static final String OLD_SERVER_VERSION_STRING = "VERSION JMRI Server "; // CAREFUL: Changing this could break backward compatibility
    static final String NEW_SERVER_VERSION_STRING = "VERSION DCC++ Server ";

    boolean useOldPrefix = false;

    protected BufferedReader istreamReader = null;

    /**
     * XmtHandler (a local class) object to implement the transmit thread
     */
    @GuardedBy ("xmtHandler")
    final protected Runnable xmtHandler;

    /**
     * RcvHandler (a local class) object to implement the receive thread
     */
    protected Runnable rcvHandler;

    /**
     * Synchronized list used as a transmit queue.
     */
    @GuardedBy ("xmtHandler")
    protected LinkedList<DCCppMessage> xmtList = new LinkedList<>();

    @SuppressFBWarnings(value = "ST_WRITE_TO_STATIC_FROM_INSTANCE_METHOD", justification = "Only used during system initialization")
    public DCCppOverTcpPacketizer(DCCppCommandStation cs) {
        super(cs); // Don't need the command station (?)

        xmtHandler = new XmtHandler();
        rcvHandler = new RcvHandler(this);
        log.debug("DCCppOverTcpPacketizer created.");
    }

    public DCCppNetworkPortController networkController = null;

    public boolean isXmtBusy() {
        return networkController != null;
    }

    /**
     * Make connection to existing DCCppNetworkPortController object.
     *
     * @param p Port networkController for connected. Save this for a later
     *          disconnect call
     */
    public void connectPort(DCCppNetworkPortController p) {
        istream = p.getInputStream();
        istreamReader = new BufferedReader(new InputStreamReader(istream));
        ostream = p.getOutputStream();
        if (networkController != null) {
            log.warn("connectPort: connect called while connected");
        }
        networkController = p;
    }

    /**
     * Break connection to existing DCCppNetworkPortController object. Once broken,
     * attempts to send via "message" member will fail.
     *
     * @param p previously connected port
     */
    public void disconnectPort(DCCppNetworkPortController p) {
        istream = null;
        ostream = null;
        if (networkController != p) {
            log.warn("disconnectPort: disconnect called from non-connected DCCppNetworkPortController");
        }
        networkController = null;
    }

    /**
     * Forward a preformatted DCCppMessage to the actual interface.
     *
     * Checksum is computed and overwritten here, then the message is converted
     * to a byte array and queue for transmission
     *
     * @param m Message to send; will be updated with CRC
     */
    @Override
    public void sendDCCppMessage(DCCppMessage m, DCCppListener reply) {
        // update statistics
        //transmittedMsgCount++;

        log.debug("queue DCCpp packet: {}", m);
        // in an atomic operation, queue the request and wake the xmit thread
        try {
            synchronized (xmtHandler) {
                xmtList.addLast(m);
                xmtHandler.notifyAll();
            }
        } catch (Exception e) {
            log.warn("passing to xmit: unexpected exception: ", e);
        }
    }

    /**
     * Invoked at startup to start the threads needed here.
     */
    public void startThreads() {
        int priority = Thread.currentThread().getPriority();
        log.debug("startThreads current priority = {} max available {} default = {} min available = {}",
                priority, Thread.MAX_PRIORITY, Thread.NORM_PRIORITY, Thread.MIN_PRIORITY);

        // make sure that the xmt priority is no lower than the current priority
        int xmtpriority = (Thread.MAX_PRIORITY - 1 > priority ? Thread.MAX_PRIORITY - 1 : Thread.MAX_PRIORITY);
        // start the XmtHandler in a thread of its own
        Thread xmtThread;
        synchronized (xmtHandler) { // never null at this point
            xmtThread = new Thread(xmtHandler, "DCC++ transmit handler");
        }
        log.debug("Xmt thread starts at priority {}", xmtpriority);
        xmtThread.setDaemon(true);
        xmtThread.setPriority(Thread.MAX_PRIORITY - 1);
        xmtThread.start();

        // start the RcvHandler in a thread of its own
        if (rcvHandler == null) {
            rcvHandler = new RcvHandler(this);
        }
        Thread rcvThread = new Thread(rcvHandler, "DCC++ receive handler");
        rcvThread.setDaemon(true);
        rcvThread.setPriority(Thread.MAX_PRIORITY);
        rcvThread.start();
    }

    /**
     * Captive class to handle incoming characters. This is a permanent loop,
     * looking for input messages in character form on the stream connected to
     * the DCCppNetworkPortController via <code>connectPort</code>.
     */
    class RcvHandler implements Runnable {

        /**
         * Remember the DCCppPacketizer object.
         *
         * @param lt the DCCppOverTcpPacketizer trafficController to run over
         */
        public RcvHandler(DCCppOverTcpPacketizer lt) {
            //trafficController = lt;
        }
        //DCCppOverTcpPacketizer trafficController;

        // readline is deprecated, but there are no problems
        // with multi-byte characters here.
        @Override
        public void run() {

            String rxLine;
            while (true) {   // loop permanently, program close will exit
                try {
                    // start by looking for a complete line

                    if (istreamReader == null) {
                        log.error("istreamReader not initialized!");
                        return;
                    }
                    rxLine = istreamReader.readLine(); // Note: This uses BufferedReader for safer data handling
                    if (rxLine == null) {
                        log.warn("run: input stream returned null, exiting loop");
                        return;
                    }

                    log.debug("Received: {}", rxLine);

                    // Legacy support. If this message is the old JMRI version
                    // handshake, flag us as in "old mode"
                    if (rxLine.startsWith(OLD_SERVER_VERSION_STRING)) {
                        useOldPrefix = true;
                    }

                    // Legacy support. If the old receive prefix is present
                    // remove it.
                    if (rxLine.startsWith(OLD_RECEIVE_PREFIX)) {
                        final int trim = OLD_RECEIVE_PREFIX.length();
                        rxLine = rxLine.substring(trim);
                    }

                    if (!rxLine.startsWith(RECEIVE_PREFIX)) {
                        // Not a valid Tcp packet
                        log.debug("Wrong Prefix: {}", rxLine);
                        continue;
                    }

                    // Strip the prefix off.
                    //final int trim = RECEIVE_PREFIX.length();
                    //rxLine = rxLine.substring(trim);

                    int firstidx = rxLine.indexOf("<");
                    int lastidx = rxLine.lastIndexOf(">");
                    log.debug("String {} Index1 {} Index 2{}", rxLine, firstidx, lastidx);

                    // BUG FIX: Incoming DCCppOverTCP messages are already formatted for DCC++ and don't
                    // need to be parsed. Indeed, trying to parse them will screw them up.
                    // So instead, we de-deprecated the string constructor so that we can
                    // directly create a DCCppReply from the incoming string without translation/parsing.

                    //  Note: the substring call below also strips off the "< >"
                    DCCppReply msg = DCCppReply.parseDCCppReply(rxLine.substring(rxLine.indexOf("<") + 1,
                                                                                 rxLine.lastIndexOf(">")));
                    //DCCppReply msg = new DCCppReply(rxLine.substring(rxLine.indexOf("<") + 1,
                    //                                rxLine.lastIndexOf(">")));

                    if (!msg.isValidReplyFormat()) {
                        log.warn("Invalid Reply Format: {}", msg.toString());
                        continue;
                    }
                    // message is complete, dispatch it !!
                    log.debug("queue reply for notification");

                    final DCCppReply thisMsg = msg;
                    //final DCCppPacketizer thisTc = trafficController;
                    // return a notification via the queue to ensure end
                    Runnable r = new Runnable() {
                            final DCCppReply msgForLater = thisMsg;

                            @Override
                            public void run() {
                                notifyReply(msgForLater, null);
                            }
                        };
                    javax.swing.SwingUtilities.invokeLater(r);
                    // done with this one
                    //} catch (DCCppMessageException e) {
                    // just let it ride for now
                    //  log.warn("run: unexpected DCCppMessageException: ", e);
                } catch (java.io.EOFException e) {
                    // posted from idle port when enableReceiveTimeout used
                    log.debug("EOFException, is DCC++ serial I/O using timeouts?");
                } catch (java.io.IOException e) {
                    // fired when write-end of HexFile reaches end
                    log.debug("IOException, should only happen with HexFile: ", e);
                    log.info("End of file");
                    //  disconnectPort(networkController);
                    return;
                } // normally, we don't catch the unnamed Exception, but in this
                // permanently running loop it seems wise.
                catch (Exception e) {
                    log.warn("run: unexpected Exception: ", e);
                }
            } // end of permanent loop
        }
    }

    /**
     * Captive class to handle transmission.
     */
    class XmtHandler implements Runnable {

        @Override
        public void run() {

            while (!threadStopRequest) {   // loop until asked to stop
                // any input?
                try {
                    // get content; failure is a NoSuchElementException
                    log.debug("check for input");
                    DCCppMessage msg;
                    synchronized (xmtHandler) {
                        msg = xmtList.removeFirst();
                    }

                    // input - now send
                    try {
                        if (ostream != null) {
                            //Commented out as the original LnPortnetworkController always returned true.
                            //if (!networkController.okToSend()) log.warn(DCCpp port not ready to receive"); // TCP, not RS232, so message is a real warning
                            log.debug("start write to network stream");
                            StringBuilder packet = new StringBuilder(msg.length() + SEND_PREFIX.length() + 2);
                            if (useOldPrefix) {
                                packet.append(OLD_SEND_PREFIX);
                            }
                            packet.append("<").append(msg.toString()).append(">");
                            if (log.isDebugEnabled()) { // avoid building a String when not needed
                                log.debug("Write to LbServer: {}", packet.toString());
                            }
                            packet.append("\r\n");
                            ostream.write(packet.toString().getBytes());
                            ostream.flush();
                            log.debug("end write to stream");
                        } else {
                            // no stream connected
                            log.warn("sendDCCppMessage: no connection established");
                        }
                    } catch (java.io.IOException e) {
                        log.warn("sendDCCppMessage: IOException: {}", e.toString());
                    }
                } catch (NoSuchElementException e) {
                    // message queue was empty, wait for input
                    log.debug("start wait");

                    new jmri.util.WaitHandler(this);  // handle synchronization, spurious wake, interruption

                    log.debug("end wait");
                }
            }
        }
    }

    /**
     * Terminate the receive and transmit threads.
     * <p>
     * This is intended to be used only by testing subclasses.
     */
    @Override
    public void terminateThreads() {
        threadStopRequest = true;
        if (xmtThread != null) {
            xmtThread.interrupt();
            try {
                xmtThread.join();
            } catch (InterruptedException ie){
                // interrupted during cleanup.
            }
        }

        if (rcvThread != null) {
            rcvThread.interrupt();
            try {
                rcvThread.join();
            } catch (InterruptedException ie){
                // interrupted during cleanup.
            }
        }
    }

    private final static Logger log = LoggerFactory.getLogger(DCCppOverTcpPacketizer.class);

}