drallgood · Aug 15, 2014 · Aug 14, 2014 · Aug 14, 2014 · Aug 15, 2014 · Aug 15, 2014
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,12 @@
 # Changelog
 All notable changes to this project will be documented in this file.
 
+## NEXT
+
+### Added
+
+* Support for [notification headers](https://app.zencoder.com/docs/api/encoding/notifications/notification-headers)
+
 ## 0.0.1 - 2014-XX-XX
 ### Added
 - Initial version

diff --git a/pom.xml b/pom.xml
@@ -141,6 +141,18 @@
 					<target>1.6</target>
 				</configuration>
 			</plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-source-plugin</artifactId>
+                <executions>
+                    <execution>
+                        <id>attach-sources</id>
+                        <goals>
+                            <goal>jar</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
 			<plugin>
 				<groupId>org.apache.maven.plugins</groupId>
 				<artifactId>maven-surefire-plugin</artifactId>

diff --git a/src/main/java/de/bitzeche/video/transcoding/zencoder/job/ZencoderNotification.java b/src/main/java/de/bitzeche/video/transcoding/zencoder/job/ZencoderNotification.java
@@ -27,10 +27,13 @@
 
 import de.bitzeche.video.transcoding.zencoder.util.XmlUtility;
 
+import java.util.Map;
+
 public class ZencoderNotification {
 
 	private String notificationString;
 	private String format;
+	private Map<String, String> headers;
 
 	public ZencoderNotification(String notificationDestination) {
 		this.notificationString = notificationDestination;
@@ -45,6 +48,16 @@ public Element createXML(Document document) {
 			root.appendChild(nfNode);
 		}
 
+        if (this.headers != null) {
+            Node headersNode = document.createElement("headers");
+            root.appendChild(headersNode);
+            for (Map.Entry<String, String> headerEntry : headers.entrySet()) {
+                Node header = document.createElement(headerEntry.getKey());
+                header.setTextContent(headerEntry.getValue());
+                headersNode.appendChild(header);
+            }
+        }
+
 		Node urlNode = document.createElement("url");
 		urlNode.setTextContent(this.notificationString);
 		root.appendChild(urlNode);
@@ -60,6 +73,17 @@ public void setFormat(String format) {
 		this.format = format;
 	}
 
+    /**
+     * @param headers see https://app.zencoder.com/docs/api/encoding/notifications/notification-headers
+     */
+    public void setHeaders(Map<String, String> headers) {
+        this.headers = headers;
+    }
+
+    public Map<String, String> getHeaders() {
+        return headers;
+    }
+
 	public String toString() {
 		Element elem;
 		Document document;
@@ -84,5 +108,4 @@ public String toString() {
 		}
 		return this.getClass().getSimpleName();
 	}
-
 }
diff --git a/src/test/java/de/bitzeche/video/transcoding/zencoder/test/ZencoderNotificationTest.java b/src/test/java/de/bitzeche/video/transcoding/zencoder/test/ZencoderNotificationTest.java
@@ -23,6 +23,8 @@
 
 import de.bitzeche.video.transcoding.zencoder.job.ZencoderNotification;
 
+import java.util.HashMap;
+
 public class ZencoderNotificationTest {
 
 	@Test
@@ -42,4 +44,16 @@ public void testWithOptions() throws ParserConfigurationException {
 		// System.out.println(doc);
 		Assert.assertEquals(doc, expected);
 	}
+
+    @Test
+    public void testWithHeaders() throws ParserConfigurationException {
+        ZencoderNotification notif = new ZencoderNotification("[email protected]");
+        notif.setHeaders(new HashMap<String, String>(){{
+            put("cloudfront_url", "http://asdjaosidjas.cloudfront.net");
+        }});
+        String doc = StringUtil.stripSpacesAndLineBreaksFrom(notif.toString());
+        String expected = "<?xmlversion=\"1.0\"encoding=\"UTF-8\"?><notification><headers><cloudfront_url>http://asdjaosidjas.cloudfront.net</cloudfront_url></headers><url>[email protected]</url></notification>";
+        // System.out.println(doc);
+        Assert.assertEquals(doc, expected);
+    }
 }