Add change-staged and change-unstaged events

Task-number: QTQAINFRA-3558
Change-Id: If88527eed80a303f53a6a8cfaa0516336ff9384a
Reviewed-by: Daniel Smith <[email protected]>
Reviewed-by: Paul Wicking <[email protected]>

Author: jujokini

src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtChangeStagedEvent.java

@@ -0,0 +1,20 @@
+//
+// Copyright (C) 2020 The Qt Company
+//
+
+package com.googlesource.gerrit.plugins.qtcodereview;
+
+import com.google.common.base.Supplier;
+import com.google.gerrit.reviewdb.client.Change;
+import com.google.gerrit.server.data.AccountAttribute;
+import com.google.gerrit.server.events.PatchSetEvent;
+
+public class QtChangeStagedEvent extends PatchSetEvent {
+  public static final String TYPE = "change-staged";
+  public Supplier<AccountAttribute> submitter;
+  public String newRev;
+
+  public QtChangeStagedEvent(Change change) {
+    super(TYPE, change);
+  }
+}

src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtChangeUnStagedEvent.java

@@ -0,0 +1,20 @@
+//
+// Copyright (C) 2020 The Qt Company
+//
+
+package com.googlesource.gerrit.plugins.qtcodereview;
+
+import com.google.common.base.Supplier;
+import com.google.gerrit.reviewdb.client.Change;
+import com.google.gerrit.server.data.AccountAttribute;
+import com.google.gerrit.server.events.PatchSetEvent;
+
+public class QtChangeUnStagedEvent extends PatchSetEvent {
+  public static final String TYPE = "change-unstaged";
+  public Supplier<AccountAttribute> submitter;
+  public String newRev;
+
+  public QtChangeUnStagedEvent(Change change) {
+    super(TYPE, change);
+  }
+}

src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtModule.java

@@ -1,5 +1,5 @@
 //
-// Copyright (C) 2019 The Qt Company
+// Copyright (C) 2020 The Qt Company
 //
 
 package com.googlesource.gerrit.plugins.qtcodereview;
@@ -12,11 +12,17 @@
 import com.google.gerrit.extensions.registration.DynamicSet;
 import com.google.gerrit.extensions.restapi.RestApiModule;
 import com.google.gerrit.server.git.ChangeMessageModifier;
+import com.google.gerrit.server.events.EventTypes;
 
 public class QtModule extends FactoryModule {
 
   private static final FluentLogger logger = FluentLogger.forEnclosingClass();
 
+  static {
+    EventTypes.register(QtChangeStagedEvent.TYPE, QtChangeStagedEvent.class);
+    EventTypes.register(QtChangeUnStagedEvent.TYPE, QtChangeUnStagedEvent.class);
+  }
+
   @Override
   protected void configure() {
 

src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtStage.java

@@ -1,5 +1,5 @@
 //
-// Copyright (C) 2019 The Qt Company
+// Copyright (C) 2020 The Qt Company
 //
 
 package com.googlesource.gerrit.plugins.qtcodereview;
@@ -205,6 +205,7 @@ private Change changeToStaging(RevisionResource rsrc, IdentifiedUser submitter,
     change = changeData.reloadChange();
     switch (change.getStatus()) {
       case STAGED:
+        qtUtil.postChangeStagedEvent(change);
         logger.atInfo().log(
             "qtcodereview: changeToStaging %s added to %s", change, stagingBranchKey);
         return change; // this doesn't return data to client, if needed use ChangeJson to convert it

src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtUnStage.java

@@ -1,5 +1,5 @@
 //
-// Copyright (C) 2019 The Qt Company
+// Copyright (C) 2020 The Qt Company
 //
 
 package com.googlesource.gerrit.plugins.qtcodereview;
@@ -149,6 +149,7 @@ private Change removeChangeFromStaging(RevisionResource rsrc, IdentifiedUser sub
       qtUtil.rebuildStagingBranch(git, submitter, projectKey, stagingBranchKey, destBranchShortKey);
 
       change = op.getChange();
+      qtUtil.postChangeUnStagedEvent(change);
       logger.atInfo().log("qtcodereview: unstaged %s from %s", change, stagingBranchKey);
 
     } catch (ResourceConflictException e) {

src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtUtil.java

@@ -1,6 +1,6 @@
 // Copyright (C) 2011 The Android Open Source Project
 // Copyright (C) 2014 Digia Plc and/or its subsidiary(-ies).
-// Copyright (C) 2019 The Qt Company
+// Copyright (C) 2020 The Qt Company
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -16,16 +16,26 @@
 
 package com.googlesource.gerrit.plugins.qtcodereview;
 
+import com.google.common.base.Supplier;
+import com.google.common.base.Suppliers;
 import com.google.common.flogger.FluentLogger;
 import com.google.gerrit.common.FooterConstants;
+import com.google.gerrit.exceptions.StorageException;
+import com.google.gerrit.extensions.registration.DynamicItem;
 import com.google.gerrit.extensions.restapi.RestApiException;
 import com.google.gerrit.reviewdb.client.Branch;
 import com.google.gerrit.reviewdb.client.Change;
 import com.google.gerrit.reviewdb.client.PatchSet;
 import com.google.gerrit.reviewdb.client.Project;
 import com.google.gerrit.server.ChangeMessagesUtil;
 import com.google.gerrit.server.IdentifiedUser;
+import com.google.gerrit.server.data.ChangeAttribute;
 import com.google.gerrit.server.extensions.events.GitReferenceUpdated;
+import com.google.gerrit.server.events.ChangeEvent;
+import com.google.gerrit.server.events.EventDispatcher;
+import com.google.gerrit.server.events.EventFactory;
+import com.google.gerrit.server.notedb.ChangeNotes;
+import com.google.gerrit.server.permissions.PermissionBackendException;
 import com.google.gerrit.server.project.NoSuchRefException;
 import com.google.gerrit.server.query.change.ChangeData;
 import com.google.gerrit.server.query.change.InternalChangeQuery;
@@ -79,6 +89,9 @@ public class QtUtil {
   private final Provider<InternalChangeQuery> queryProvider;
   private final GitReferenceUpdated referenceUpdated;
   private final BatchUpdate.Factory updateFactory;
+  private final ChangeNotes.Factory changeNotesFactory;
+  private final DynamicItem<EventDispatcher> eventDispatcher;
+  private final EventFactory eventFactory;
   private final QtCherryPickPatch qtCherryPickPatch;
   private final QtChangeUpdateOp.Factory qtUpdateFactory;
 
@@ -87,11 +100,17 @@ public class QtUtil {
       Provider<InternalChangeQuery> queryProvider,
       GitReferenceUpdated referenceUpdated,
       BatchUpdate.Factory updateFactory,
+      ChangeNotes.Factory changeNotesFactory,
+      EventFactory eventFactory,
+      DynamicItem<EventDispatcher> eventDispatcher,
       QtCherryPickPatch qtCherryPickPatch,
       QtChangeUpdateOp.Factory qtUpdateFactory) {
     this.queryProvider = queryProvider;
     this.referenceUpdated = referenceUpdated;
     this.updateFactory = updateFactory;
+    this.changeNotesFactory = changeNotesFactory;
+    this.eventDispatcher = eventDispatcher;
+    this.eventFactory = eventFactory;
     this.qtCherryPickPatch = qtCherryPickPatch;
     this.qtUpdateFactory = qtUpdateFactory;
   }
@@ -642,4 +661,38 @@ private static RefUpdate.Result mergeObjectToBranch(
       revWalk.dispose();
     }
   }
+
+  private Supplier<ChangeAttribute> changeAttributeSupplier(Change change, ChangeNotes notes) {
+    return Suppliers.memoize(
+        () -> {
+          try {
+            return eventFactory.asChangeAttribute(change, notes);
+          } catch (StorageException e) {
+            throw new RuntimeException(e);
+          }
+        });
+  }
+
+  public void postChangeStagedEvent(Change change) {
+    try {
+      ChangeNotes notes = changeNotesFactory.createChecked(change.getId());
+      QtChangeStagedEvent event = new QtChangeStagedEvent(change);
+      event.change = changeAttributeSupplier(change, notes);
+      eventDispatcher.get().postEvent(event);
+    } catch (StorageException | PermissionBackendException e) {
+      logger.atWarning().log("qtcodereview: postChangeStagedEvent failed: %s", e);
+    }
+  }
+
+public void postChangeUnStagedEvent(Change change) {
+  try {
+    ChangeNotes notes = changeNotesFactory.createChecked(change.getId());
+    QtChangeUnStagedEvent event = new QtChangeUnStagedEvent(change);
+    event.change = changeAttributeSupplier(change, notes);
+    eventDispatcher.get().postEvent(event);
+  } catch (StorageException | PermissionBackendException e) {
+    logger.atWarning().log("qtcodereview: postChangeUnStagedEvent failed: %s", e);
+  }
+}
+
 }