From 58d26618f67d69bdf83a244d4e34e332876f03ad Mon Sep 17 00:00:00 2001
From: Shadowfacts <shadowfacts@users.noreply.github.com>
Date: Mon, 9 May 2016 17:51:30 -0400
Subject: [PATCH] Add ScreenshotEvent (#2828)

Add ScreenshotEvent
---
 .../util/ScreenShotHelper.java.patch          | 16 +++++
 .../client/ForgeHooksClient.java              | 11 +++
 .../client/event/ScreenshotEvent.java         | 67 +++++++++++++++++++
 3 files changed, 94 insertions(+)
 create mode 100644 patches/minecraft/net/minecraft/util/ScreenShotHelper.java.patch
 create mode 100644 src/main/java/net/minecraftforge/client/event/ScreenshotEvent.java

diff --git a/patches/minecraft/net/minecraft/util/ScreenShotHelper.java.patch b/patches/minecraft/net/minecraft/util/ScreenShotHelper.java.patch
new file mode 100644
index 0000000000..66c5e3cffc
--- /dev/null
+++ b/patches/minecraft/net/minecraft/util/ScreenShotHelper.java.patch
@@ -0,0 +1,16 @@
+--- ../src-base/minecraft/net/minecraft/util/ScreenShotHelper.java
++++ ../src-work/minecraft/net/minecraft/util/ScreenShotHelper.java
+@@ -52,10 +52,13 @@
+                 file2 = new File(file1, p_148259_1_);
+             }
+ 
++            net.minecraftforge.client.event.ScreenshotEvent event = net.minecraftforge.client.ForgeHooksClient.onScreenshot(bufferedimage, file2);
++            if (event.isCanceled()) return event.getCancelMessage(); else file2 = event.getScreenshotFile();
+             ImageIO.write(bufferedimage, "png", (File)file2);
+             ITextComponent itextcomponent = new TextComponentString(file2.getName());
+             itextcomponent.func_150256_b().func_150241_a(new ClickEvent(ClickEvent.Action.OPEN_FILE, file2.getAbsolutePath()));
+             itextcomponent.func_150256_b().func_150228_d(Boolean.valueOf(true));
++            if (event.getResultMessage() != null) return event.getResultMessage();
+             return new TextComponentTranslation("screenshot.success", new Object[] {itextcomponent});
+         }
+         catch (Exception exception)
diff --git a/src/main/java/net/minecraftforge/client/ForgeHooksClient.java b/src/main/java/net/minecraftforge/client/ForgeHooksClient.java
index a0b60506a1..c8d46cae0d 100644
--- a/src/main/java/net/minecraftforge/client/ForgeHooksClient.java
+++ b/src/main/java/net/minecraftforge/client/ForgeHooksClient.java
@@ -6,6 +6,8 @@ import static net.minecraftforge.common.ForgeVersion.Status.BETA_OUTDATED;
 import static org.lwjgl.opengl.GL11.*;
 import static org.lwjgl.opengl.GL20.*;
 
+import java.awt.image.BufferedImage;
+import java.io.File;
 import java.nio.ByteBuffer;
 import java.nio.FloatBuffer;
 import java.util.Collections;
@@ -77,6 +79,7 @@ import net.minecraftforge.client.event.MouseEvent;
 import net.minecraftforge.client.event.RenderGameOverlayEvent;
 import net.minecraftforge.client.event.RenderHandEvent;
 import net.minecraftforge.client.event.RenderWorldLastEvent;
+import net.minecraftforge.client.event.ScreenshotEvent;
 import net.minecraftforge.client.event.TextureStitchEvent;
 import net.minecraftforge.client.event.sound.PlaySoundEvent;
 import net.minecraftforge.client.model.IPerspectiveAwareModel;
@@ -705,4 +708,12 @@ public class ForgeHooksClient
     {
         MinecraftForge.EVENT_BUS.post(new RenderGameOverlayEvent.Post(new RenderGameOverlayEvent(Animation.getPartialTickTime(), res), BOSSINFO));
     }
+
+    public static ScreenshotEvent onScreenshot(BufferedImage image, File screenshotFile)
+    {
+        ScreenshotEvent event = new ScreenshotEvent(image, screenshotFile);
+        MinecraftForge.EVENT_BUS.post(event);
+        return event;
+    }
+
 }
diff --git a/src/main/java/net/minecraftforge/client/event/ScreenshotEvent.java b/src/main/java/net/minecraftforge/client/event/ScreenshotEvent.java
new file mode 100644
index 0000000000..ed2db4b965
--- /dev/null
+++ b/src/main/java/net/minecraftforge/client/event/ScreenshotEvent.java
@@ -0,0 +1,67 @@
+package net.minecraftforge.client.event;
+
+import net.minecraft.util.text.ITextComponent;
+import net.minecraft.util.text.TextComponentString;
+import net.minecraftforge.fml.common.eventhandler.Cancelable;
+import net.minecraftforge.fml.common.eventhandler.Event;
+
+import java.awt.image.BufferedImage;
+import java.io.File;
+
+/**
+ * This event is fired before and after a screenshot is taken
+ * This event is fired on the {@link net.minecraftforge.common.MinecraftForge#EVENT_BUS}
+ * This event is {@link Cancelable}
+ *
+ * {@link #screenshotFile} contains the file the screenshot will be/was saved to
+ * {@link #image} contains the {@link BufferedImage} that will be saved
+ * {@link #resultMessage} contains the {@link ITextComponent} to be returned. If {@code null}, the default vanilla message will be used instead
+ */
+@Cancelable
+public class ScreenshotEvent extends Event
+{
+
+    public static final ITextComponent DEFAULT_CANCEL_REASON = new TextComponentString("Screenshot canceled");
+
+    private BufferedImage image;
+    private File screenshotFile;
+
+    private ITextComponent resultMessage = null;
+
+    public ScreenshotEvent(BufferedImage image, File screenshotFile)
+    {
+        this.image = image;
+        this.screenshotFile = screenshotFile;
+    }
+
+    public BufferedImage getImage()
+    {
+        return image;
+    }
+
+    public File getScreenshotFile()
+    {
+        return screenshotFile;
+    }
+
+    public void setScreenshotFile(File screenshotFile)
+    {
+        this.screenshotFile = screenshotFile;
+    }
+
+    public ITextComponent getResultMessage()
+    {
+        return resultMessage;
+    }
+
+    public void setResultMessage(ITextComponent resultMessage)
+    {
+        this.resultMessage = resultMessage;
+    }
+
+    public ITextComponent getCancelMessage()
+    {
+        return getResultMessage() != null ? getResultMessage() : DEFAULT_CANCEL_REASON;
+    }
+
+}