aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/java/com/googlesource/gerrit/plugins/qtcodereview/QtChangeUpdateOp.java
blob: 491e89656ca2dd9ef5ad4f72f1e4d8cf39e7df32 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
//
// Copyright (C) 2018 The Qt Company
//

package com.googlesource.gerrit.plugins.qtcodereview;

import com.google.common.base.Strings;
import com.google.gerrit.common.Nullable;
import com.google.gerrit.extensions.restapi.ResourceConflictException;
import com.google.gerrit.reviewdb.client.Change;
import com.google.gerrit.reviewdb.client.ChangeMessage;
import com.google.gerrit.reviewdb.client.PatchSet;
import com.google.gerrit.server.ChangeMessagesUtil;
import com.google.gerrit.server.notedb.ChangeUpdate;
import com.google.gerrit.server.update.BatchUpdateOp;
import com.google.gerrit.server.update.ChangeContext;
import com.google.gwtorm.server.OrmException;
import com.google.inject.Inject;
import com.google.inject.assistedinject.Assisted;
import java.io.IOException;

public class QtChangeUpdateOp implements BatchUpdateOp {

    public interface Factory {
        QtChangeUpdateOp create(Change.Status newStatus,
                                @Assisted("defaultMessage") String defaultMessage,
                                @Assisted("inputMessage") String inputMessage,
                                @Assisted("tag") String tag);
    }

    private final Change.Status newStatus;
    private final String defaultMessage;
    private final String inputMessage;
    private final String tag;

    private Change change;

    private final ChangeMessagesUtil cmUtil;


    @Inject
    QtChangeUpdateOp(ChangeMessagesUtil cmUtil,
                     @Nullable @Assisted Change.Status newStatus,
                     @Nullable @Assisted("defaultMessage") String defaultMessage,
                     @Nullable @Assisted("inputMessage") String inputMessage,
                     @Nullable @Assisted("tag") String tag) {
        this.cmUtil = cmUtil;
        this.newStatus = newStatus;
        this.defaultMessage = defaultMessage;
        this.inputMessage = inputMessage;
        this.tag = tag;
    }

    public Change getChange() {
        return change;
    }

    @Override
    public boolean updateChange(ChangeContext ctx) throws OrmException, IOException, ResourceConflictException {
        boolean updated = false;
        change = ctx.getChange();
        PatchSet.Id psId = change.currentPatchSetId();
        ChangeUpdate update = ctx.getUpdate(psId);

        if (newStatus != null) {
            change.setStatus(newStatus);
            updated = true;
        }

        if (updated == true) {
            change.setLastUpdatedOn(ctx.getWhen());
        }

        if (defaultMessage != null) {
            cmUtil.addChangeMessage(ctx.getDb(), update, newMessage(ctx));
            updated = true;
        }

        return updated;
    }

    private ChangeMessage newMessage(ChangeContext ctx) {
        StringBuilder msg = new StringBuilder();
        msg.append(defaultMessage);
        if (!Strings.nullToEmpty(inputMessage).trim().isEmpty()) {
            msg.append("\n\n");
            msg.append(inputMessage.trim());
        }
        return ChangeMessagesUtil.newMessage(ctx, msg.toString(), tag);
    }

}