com.sikulix.vnc.VNCClipboard Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sikulix2tigervnc Show documentation
... for visual testing and automation (TigerVNC support)
The newest version!
package com.sikulix.vnc;

import java.awt.Toolkit;
import java.awt.datatransfer.ClipboardOwner;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.io.StringReader;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;

public class VNCClipboard {
  /**
   * Enumeration for the text type property in MIME types
   */
  public static class TextType {
    private String type;

    private TextType(String type) {
      this.type = type;
    }

    @Override
    public String toString() {
      return type;
    }
  }

  /**
   * Enumeration for the charset property in MIME types (UTF-8, UTF-16, etc.)
   */
  public static class Charset {
    private String name;

    private Charset(String name) {
      this.name = name;
    }

    @Override
    public String toString() {
      return name;
    }
  }

  /**
   * Enumeration for the transferScript type property in MIME types (InputStream, CharBuffer, etc.)
   */
  public static class TransferType {
    private Class dataClass;

    private TransferType(Class streamClass) {
      this.dataClass = streamClass;
    }

    public Class getDataClass() {
      return dataClass;
    }

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

  private static class TextTransferable implements Transferable, ClipboardOwner {
    private String data;
    private DataFlavor flavor;

    public TextTransferable(String mimeType, String data) {
      flavor = new DataFlavor(mimeType, "Text");
      this.data = data;
    }

    @Override
    public DataFlavor[] getTransferDataFlavors() {
      return new DataFlavor[]{flavor, DataFlavor.stringFlavor};
    }

    @Override
    public boolean isDataFlavorSupported(DataFlavor flavor) {
      boolean b = this.flavor.getPrimaryType().equals(flavor.getPrimaryType());
      return b || flavor.equals(DataFlavor.stringFlavor);
    }

    @Override
    public Object getTransferData(DataFlavor flavor) throws UnsupportedFlavorException, IOException {
      if (flavor.isRepresentationClassInputStream()) {
        return new StringReader(data);
      } else if (flavor.isRepresentationClassReader()) {
        return new StringReader(data);
      } else if (flavor.isRepresentationClassCharBuffer()) {
        return CharBuffer.wrap(data);
      } else if (flavor.isRepresentationClassByteBuffer()) {
        return ByteBuffer.wrap(data.getBytes());
      } else if (flavor.equals(DataFlavor.stringFlavor)) {
        return data;
      }
      throw new UnsupportedFlavorException(flavor);
    }

    @Override
    public void lostOwnership(java.awt.datatransfer.Clipboard clipboard, Transferable contents) {
    }
  }

  public static final TextType HTML = new TextType("text/html");
  public static final TextType PLAIN = new TextType("text/plain");

  public static final Charset UTF8 = new Charset("UTF-8");
  public static final Charset UTF16 = new Charset("UTF-16");
  public static final Charset UNICODE = new Charset("unicode");
  public static final Charset US_ASCII = new Charset("US-ASCII");

  public static final TransferType READER = new TransferType(Reader.class);
  public static final TransferType INPUT_STREAM = new TransferType(InputStream.class);
  public static final TransferType CHAR_BUFFER = new TransferType(CharBuffer.class);
  public static final TransferType BYTE_BUFFER = new TransferType(ByteBuffer.class);

  public static void putTextToClipboard(TextType type, Charset charset, TransferType transferType, CharSequence data) {
    String mimeType = type + "; charset=" + charset + "; class=" + transferType;
    TextTransferable transferable = new TextTransferable(mimeType, data.toString());
    Toolkit.getDefaultToolkit().getSystemClipboard().setContents(transferable, transferable);
  }
}