Add PatchsetHook extensions

2 files changed, 86 insertions, 0 deletions

diff --git a/src/main/java/com/gitblit/extensions/PatchsetHook.java b/src/main/java/com/gitblit/extensions/PatchsetHook.java
new file mode 100644
index 00000000..e465f507
--- /dev/null
+++ b/src/main/java/com/gitblit/extensions/PatchsetHook.java
@@ -0,0 +1,53 @@
+/*
+ * Copyright 2014 gitblit.com.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.gitblit.extensions;
+
+import ro.fortsoft.pf4j.ExtensionPoint;
+
+import com.gitblit.models.TicketModel;
+
+/**
+ * Extension point for plugins to respond to Ticket patchset changes.
+ *
+ * @author James Moger
+ *
+ */
+public abstract class PatchsetHook implements ExtensionPoint {
+
+	/**
+	 * Called after a new patchset has been created.  This patchset
+	 * may not be the first patchset of the ticket.  The ticket may be a new
+	 * proposal or it may be a existing ticket that now has a new patchset.
+	 *
+	 * @param ticket
+	 */
+	public abstract void onNewPatchset(TicketModel ticket);
+
+	/**
+	 * Called after a patchset has been FAST-FORWARD updated.
+	 *
+	 * @param ticket
+	 */
+	public abstract void onUpdatePatchset(TicketModel ticket);
+
+	/**
+	 * Called after a patchset has been merged to the integration branch specified
+	 * in the ticket.
+	 *
+	 * @param ticket
+	 */
+	public abstract void onMergePatchset(TicketModel ticket);
+}
diff --git a/src/main/java/com/gitblit/git/PatchsetReceivePack.java b/src/main/java/com/gitblit/git/PatchsetReceivePack.java
index 2c3b957b..77d5a71e 100644
--- a/src/main/java/com/gitblit/git/PatchsetReceivePack.java
+++ b/src/main/java/com/gitblit/git/PatchsetReceivePack.java
@@ -51,6 +51,7 @@ import org.slf4j.LoggerFactory;
 

 import com.gitblit.Constants;

 import com.gitblit.Keys;

+import com.gitblit.extensions.PatchsetHook;

 import com.gitblit.manager.IGitblit;

 import com.gitblit.models.RepositoryModel;

 import com.gitblit.models.TicketModel;

@@ -716,6 +717,15 @@ public class PatchsetReceivePack extends GitblitReceivePack {
 				RefLogUtils.updateRefLog(user, getRepository(),

 						Arrays.asList(new ReceiveCommand(cmd.getOldId(), cmd.getNewId(), cmd.getRefName())));

 

+				// call any patchset hooks

+				for (PatchsetHook hook : gitblit.getExtensions(PatchsetHook.class)) {

+					try {

+						hook.onNewPatchset(ticket);

+					} catch (Exception e) {

+						LOGGER.error("Failed to execute extension", e);

+					}

+				}

+

 				return ticket;

 			} else {

 				sendError("FAILED to create ticket");

@@ -743,6 +753,20 @@ public class PatchsetReceivePack extends GitblitReceivePack {
 				RefLogUtils.updateRefLog(user, getRepository(),

 					Arrays.asList(new ReceiveCommand(cmd.getOldId(), cmd.getNewId(), cmd.getRefName())));

 

+				// call any patchset hooks

+				final boolean isNewPatchset = change.patchset.rev == 1;

+				for (PatchsetHook hook : gitblit.getExtensions(PatchsetHook.class)) {

+					try {

+						if (isNewPatchset) {

+							hook.onNewPatchset(ticket);

+						} else {

+							hook.onUpdatePatchset(ticket);

+						}

+					} catch (Exception e) {

+						LOGGER.error("Failed to execute extension", e);

+					}

+				}

+

 				// return the updated ticket

 				return ticket;

 			} else {

@@ -1167,6 +1191,15 @@ public class PatchsetReceivePack extends GitblitReceivePack {
 						ObjectId.fromString(mergeResult.sha), oldRef.getName());

 				RefLogUtils.updateRefLog(user, getRepository(), Arrays.asList(cmd));

 			}

+

+			// call patchset hooks

+			for (PatchsetHook hook : gitblit.getExtensions(PatchsetHook.class)) {

+				try {

+					hook.onMergePatchset(ticket);

+				} catch (Exception e) {

+					LOGGER.error("Failed to execute extension", e);

+				}

+			}

 			return mergeResult.status;

 		} else {

 			LOGGER.error("FAILED to resolve ticket {} by merge from web ui", ticketId);