com.bytezone.dm3270.extended.TN3270ExtendedCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dm3270-lib Show documentation

This is a trimmed down version of https://github.com/dmolony/dm3270 to be used as TN3270 client library

There is a newer version: 0.15-lib

package com.bytezone.dm3270.extended;

import com.bytezone.dm3270.Charset;
import com.bytezone.dm3270.buffers.Buffer;
import com.bytezone.dm3270.buffers.MultiBuffer;
import com.bytezone.dm3270.commands.Command;
import com.bytezone.dm3270.display.Screen;
import com.bytezone.dm3270.streams.TelnetState;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

public class TN3270ExtendedCommand extends AbstractExtendedCommand {

  private final Command command;
  private final TelnetState telnetState;
  private final Charset charset;

  public TN3270ExtendedCommand(CommandHeader commandHeader, Command command,
      TelnetState telnetState, Charset charset) {
    super(commandHeader);
    this.command = command;
    this.telnetState = telnetState;
    this.charset = charset;
  }

  public Command getCommand() {
    return command;
  }

  @Override
  public byte[] getData() {
    byte[] combinedData = new byte[command.size() + 5];
    System.arraycopy(commandHeader.getData(), 0, combinedData, 0,
        commandHeader.size());
    System.arraycopy(command.getData(), 0, combinedData, commandHeader.size(),
        command.size());
    return combinedData;
  }

  @Override
  public int size() {
    return command.size() + commandHeader.size();
  }

  @Override
  public void process(Screen screen) {
    commandHeader.process(screen);
    command.process(screen);
  }

  @Override
  public Optional getReply() {
    List buffers = new ArrayList<>();

    Optional headerReply = commandHeader.getReply();
    headerReply.ifPresent(buffers::add);

    // need to add a header for the command reply before the reply
    Optional reply = command.getReply();
    if (reply.isPresent()) {
      byte[] headerBuffer = new byte[5];
      Buffer.packUnsignedShort(telnetState.nextCommandHeaderSeq(), headerBuffer, 3);
      CommandHeader header = new CommandHeader(headerBuffer, charset);
      TN3270ExtendedCommand command = new TN3270ExtendedCommand(header, (Command) reply.get(),
          telnetState, charset);
      buffers.add(command);
    }

    if (buffers.size() == 0) {
      return Optional.empty();
    }

    if (buffers.size() == 1) {
      return Optional.of(buffers.get(0));
    }

    MultiBuffer multiBuffer = new MultiBuffer(charset);
    for (Buffer buffer : buffers) {
      multiBuffer.addBuffer(buffer);
    }
    return Optional.of(multiBuffer);
  }

  @Override
  public byte[] getTelnetData() {
    byte[] headerTelnetBuffer = commandHeader.getTelnetData();
    byte[] commandTelnetBuffer = command.getTelnetData();

    int length = headerTelnetBuffer.length + commandTelnetBuffer.length;
    byte[] returnBuffer = new byte[length];
    System.arraycopy(headerTelnetBuffer, 0, returnBuffer, 0, headerTelnetBuffer.length);
    System.arraycopy(commandTelnetBuffer, 0, returnBuffer, headerTelnetBuffer.length,
        commandTelnetBuffer.length);

    return returnBuffer;
  }

  @Override
  public String getName() {
    return command.getName();
  }

  @Override
  public String toString() {
    return command.toString();
  }

}