/*
* This file is part of DeskBin.
*
* Copyright (c) 2012, alta189 <http://github.com/alta189/DeskBin/>
* DeskBin is licensed under the GNU Lesser General Public License.
*
* DeskBin is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* DeskBin is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.alta189.deskbin.tasks.service;
import java.awt.Toolkit;
import java.awt.datatransfer.Clipboard;
import java.awt.datatransfer.StringSelection;
import java.awt.datatransfer.Transferable;
import java.awt.image.BufferedImage;
import java.util.List;
import com.alta189.deskbin.services.image.ImageService;
import com.alta189.deskbin.services.image.ImageServiceException;
import com.alta189.deskbin.tasks.Action;
import com.alta189.deskbin.tasks.TaskSnapshot;
import com.alta189.deskbin.util.Keyboard;
import com.alta189.deskbin.util.OptionsMap;
import com.alta189.commons.util.ScreenshotUtil;
public class ScreenshotImageServiceTask extends ServiceTask<ImageService> {
public ScreenshotImageServiceTask(ImageService service) {
this(service, new OptionsMap());
}
public ScreenshotImageServiceTask(ImageService service, OptionsMap options) {
super(service);
setOptions(options);
}
public ScreenshotImageServiceTask(TaskSnapshot snapshot) {
super(snapshot);
}
@Override
public TaskSnapshot generateSnapshot() {
return super.generateSnapshot()
.add("version", 1);
}
@Override
public void run() {
ScreenshotType screenshotType = ScreenshotType.getValue(getOptions().get(String.class, "screenshot-type"));
String url = null;
switch (screenshotType) {
case MAIN_MONITOR:
url = upload(ScreenshotUtil.captureMainMonitor());
break;
case CURRENT_MONITOR:
url = upload(ScreenshotUtil.captureCurrentMonitor());
break;
case ALL_MONITORS:
url = upload(ScreenshotUtil.captureAllMonitors());
break;
case EACH_MONITOR:
url = upload(ScreenshotUtil.captureEachMonitor());
break;
}
if (url != null) {
Action action = Action.getValue(getOptions().get(String.class, "action"));
switch (action) {
case WRITE:
Keyboard.getInstance().type(url);
notify("Screenshot taken and upload url(s) typed.");
break;
case CLIPBOARD:
Keyboard.getInstance();
Clipboard clipboard = Toolkit.getDefaultToolkit().getSystemClipboard();
Transferable transferable = new StringSelection(url);
clipboard.setContents(transferable, null);
notify("Screenshot taken and upload url(s) copied to the clipboard.");
break;
}
}
}
private String upload(BufferedImage image) {
try {
return getService().upload(image);
} catch (ImageServiceException e) {
e.printStackTrace();
}
return null;
}
private String upload(List<BufferedImage> images) {
try {
return getService().upload(images);
} catch (ImageServiceException e) {
e.printStackTrace();
}
return null;
}
public enum ScreenshotType {
MAIN_MONITOR,
CURRENT_MONITOR, // default
ALL_MONITORS,
EACH_MONITOR;
public static ScreenshotType getValue(String str) {
ScreenshotType value = null;
try {
value = valueOf(str);
} catch (Exception ignored) {
}
if (value == null) {
value = CURRENT_MONITOR;
}
return value;
}
}
}