org.yamcs.simulation.simulator.TelnetServerHandler Maven / Gradle / Ivy

Go to download
package org.yamcs.simulation.simulator;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.TimeZone;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;

public class TelnetServerHandler extends SimpleChannelInboundHandler {

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

    private Simulator simulator;

    public TelnetServerHandler(Simulator simulator) {
        this.simulator = simulator;
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        log.info("Telnet client connected: " + ctx.channel().remoteAddress());
    }

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String scpi) throws Exception {
        String[] commands = scpi.trim().split(";");
        List responses = new ArrayList<>();

        for (String command : commands) {
            if (command.trim().isEmpty()) {
                continue;
            }

            String[] parts = command.trim().split("\\s+", 2);
            switch (parts[0].toUpperCase()) {
            case "*IDN?":
                responses.add("SPACEAPPS,Demo Simulator");
                break;
            case ":BATTERY1:VOLTAGE?":
                responses.add(Float.toString(simulator.powerDataHandler.getBattery1Voltage()));
                break;
            case ":BATTERY2:VOLTAGE?":
                responses.add(Float.toString(simulator.powerDataHandler.getBattery2Voltage()));
                break;
            case ":BATTERY3:VOLTAGE?":
                responses.add(Float.toString(simulator.powerDataHandler.getBattery3Voltage()));
                break;
            case ":LOS?":
                responses.add((simulator.isLOS() ? "1" : "0"));
                break;
            case ":LOS:STAR":
            case ":LOS:START":
                simulator.setLOS();
                break;
            case ":LOS:STOP":
                simulator.setAOS();
                break;
            case ":LOS:START:DATE?":
            case ":LOS:STAR:DATE?":
                Date startDate = simulator.getLastLosStart();
                if (startDate == null) {
                    responses.add("");
                } else {
                    responses.add(formatDate(startDate));
                }
                break;
            case ":LOS:START:TIME?":
            case ":LOS:STAR:TIME?":
                Date startTime = simulator.getLastLosStart();
                if (startTime == null) {
                    responses.add("");
                } else {
                    responses.add(formatTime(startTime));
                }
                break;
            case ":LOS:STOP:DATE?":
                Date stopDate = simulator.getLastLosStop();
                if (stopDate == null) {
                    responses.add("");
                } else {
                    responses.add(formatDate(stopDate));
                }
                break;
            case ":LOS:STOP:TIME?":
                Date stopTime = simulator.getLastLosStop();
                if (stopTime == null) {
                    responses.add("");
                } else {
                    responses.add(formatTime(stopTime));
                }
                break;
            case ":DATE?":
                responses.add(formatDate(new Date()));
                break;
            case ":TIME?":
                responses.add(formatTime(new Date()));
                break;
            default:
                responses.add("unrecognized command");
            }
        }

        if (!responses.isEmpty()) {
            ctx.write(String.join(";", responses));
            ctx.writeAndFlush("\r\n");
        }
    }

    private static String formatDate(Date date) {
        SimpleDateFormat df = new SimpleDateFormat("yyyy,MM,dd");
        df.setTimeZone(TimeZone.getTimeZone("UTC"));
        return df.format(date);
    }

    private static String formatTime(Date date) {
        SimpleDateFormat df = new SimpleDateFormat("HH,mm,ss");
        df.setTimeZone(TimeZone.getTimeZone("UTC"));
        return df.format(date);
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        log.error("Closing channel due to exception", cause);
        ctx.close();
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        log.info("Telnet client disconnected: " + ctx.channel().remoteAddress());
    }
}