clipboard.java

基于Junit的 功能和单元测试的的测试工具。只支持Swing.

package org.uispec4j;

import java.awt.*;
import java.awt.datatransfer.*;
import java.io.*;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;

/**
 * Utility for changing and checking the contents of the system's clipboard.
 */
public class Clipboard {

  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);

  private Clipboard() {
  }

  /**
   * Dumps a given text (either String or StringBuffer) into the Clipboard, with a default MIME type
   */
  public static void putText(CharSequence data) {
    StringSelection copy = new StringSelection(data.toString());
    getSystemClipboard().setContents(copy, copy);
  }

  /**
   * Dumps a given text (either String or StringBuffer) into the Clipboard with a specified MIME type
   */
  public static void putText(TextType type, Charset charset, TransferType transferType, CharSequence data) {
    String mimeType = type + "; charset=" + charset + "; class=" + transferType;
    TextTransferable transferable = new TextTransferable(mimeType, data.toString());
    getSystemClipboard().setContents(transferable, transferable);
  }

  private static java.awt.datatransfer.Clipboard getSystemClipboard() {
    return Toolkit.getDefaultToolkit().getSystemClipboard();
  }

  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;
    }

    public DataFlavor[] getTransferDataFlavors() {
      return new DataFlavor[]{flavor};
    }

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

    public Object getTransferData(DataFlavor flavor) throws UnsupportedFlavorException, IOException {
      if (flavor.isRepresentationClassInputStream()) {
        return new StringBufferInputStream(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());
      }
      throw new UnsupportedFlavorException(flavor);
    }

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

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

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

    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;
    }

    public String toString() {
      return name;
    }
  }

  /**
   * Enumeration for the transfert 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;
    }

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

}