!759

lib 1003153 // magic.jar

import ma.ma02.*;

import drjava.util.*;
import net.luaos.tb.remote.ServerConnection;
import net.luaos.tb.remote.Snippet;
import net.luaos.tb.remote.SnippetTypes;
import net.luaos.tb.tb07.PasswordUtil;

// That ImageSurface has the selector & the nice zoom stuff
import prophecy.common.image.ImageSurface;
import prophecy.common.image.RGBImage;

p-substance {
  JFrame frame = showFrame();
  hideConsole();
  ScreenshotPanel screenshotPanel = new ScreenshotPanel(frame);
  setFrameContents(frame, screenshotPanel);
}

sclass ScreenshotPanel extends JPanel {
  private JButton btnUpload;
  private JFrame frame;
  private int defaultDelay = 3000;
  private ImageSurface imageSurface;
  private int delayWithFrame = 3000;

  public ScreenshotPanel(JFrame frame) {
    this.frame = frame;
    getContentPane().setLayout(new LetterLayout("B", "I", "I").setBorder(10));

    imageSurface = new ImageSurface() {
      public void setSelection(Rectangle r) {
        super.setSelection(r);
        btnUpload.setText(r != null ? "Upload screenshot (selected area)" : "Upload screenshot");
      }
    };

    JButton btnShoot = new JButton(new AbstractAction("Shoot!") {
      public void actionPerformed(ActionEvent actionEvent) {
        shoot();
      }
    });

    btnUpload = new JButton(new AbstractAction("Upload screenshot") {
      public void actionPerformed(ActionEvent actionEvent) {
        try {
          String user = TBDefaultUser.getInAWTContext();
          user = TBDefaultUser.askUserNameAWT(user);
          if (user == null) return;
          char[] pass = PasswordUtil.readPasswordFromFileOrAWT(user);
          ServerConnection serverConnection = new ServerConnection();
          if (!serverConnection.login(user, pass)) {
            JOptionPane.showMessageDialog(null, "Could not log in");
            return;
          }
          ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
          BufferedImage image = imageSurface.getImage();
          if (imageSurface.getSelection() != null)
            image = new RGBImage(image).clip(imageSurface.getSelection()).getBufferedImage();
          ImageIO.write(image, "png", outputStream);
          String title = "A screenshot";
          title = JOptionPane.showInputDialog(null, "Please enter screenshot title", title);
          if (title == null) return;
          Snippet snippet = new Snippet(title, "", SnippetTypes.SN_IMAGE);
          snippet.isPublic = true;
          snippet.setBlobData(outputStream.toByteArray(), "png");
          try {
            long id = serverConnection.createSnippet(snippet, true);
            String msg = "Screen shot uploaded as #" + id + " (http://tinybrain.de/" + id + ")!";
            JOptionPane.showMessageDialog(null, msg);
          } catch (Exception e) {
            Errors.popup(e);
          }
        } catch (Throwable e) {
          Errors.popup(e);
        }
      }
    });
    btnUpload.setEnabled(false);
    getContentPane().add("B", new Line(btnShoot, btnUpload));
    getContentPane().add("I", GUIUtil.withTitle("Screenshot:", new JScrollPane(imageSurface)));

  }

  public void shoot() {
    hideFrame();
    SwingTimerHelper.runOnce(getDelay(), new Runnable() {
      @Override
      public void run() {
        try {
          BufferedImage image;
          try {
            Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
            Rectangle screenRectangle = new Rectangle(screenSize);
            Robot robot = new Robot();
            image = robot.createScreenCapture(screenRectangle);
          } finally {
            showFrame();
          }
          imageSurface.setImage(image);
          imageSurface.zoomToDisplaySize();
          btnUpload.setEnabled(true);
        } catch (Throwable e) {
          Errors.popup(e);
        }
      }
    });
  }

  private int getDelay() {
    return frame != null ? delayWithFrame : defaultDelay;
  }

  private void hideFrame() {
    if (frame != null)
      frame.setVisible(false);
  }

  private void showFrame() {
    if (frame != null)
      frame.setVisible(true);
  }

  private JPanel getContentPane() {
    return this;
  }
}

Began life as a copy of #1003164