Skip to content

Commit

Permalink
Switching to Gitlab4j-api
Browse files Browse the repository at this point in the history 
Waiting for gitlab4j/gitlab4j-api#291
  • Loading branch information
@tomasbjerre
tomasbjerre committed Feb 11, 2019
1 parent ff506a3 commit 1f6f39c
Show file tree
Hide file tree
Showing 3 changed files with 95 additions and 68 deletions.
4 changes: 2 additions & 2 deletions build.gradle
View file
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@ apply from: project.buildscript.classLoader.getResource('release.gradle').toURI(


dependencies {
compile 'se.bjurr.violations:violation-comments-lib:1.83'
compile 'com.github.tomasbjerre:java-gitlab-api:comment-on-diff-2'
compile 'se.bjurr.violations:violation-comments-lib:1.85'
compile 'org.gitlab4j:gitlab4j-api:4.9.17-SNAPSHOT'
testCompile 'junit:junit:4.12'
testCompile 'org.assertj:assertj-core:2.3.0'
}
150 changes: 89 additions & 61 deletions src/main/java/se/bjurr/violations/comments/gitlab/lib/GitLabCommentsProvider.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,21 @@
import static java.util.logging.Level.INFO;
import static java.util.logging.Level.SEVERE;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Optional;
import org.gitlab.api.AuthMethod;
import org.gitlab.api.GitlabAPI;
import org.gitlab.api.TokenType;
import org.gitlab.api.models.GitlabCommitDiff;
import org.gitlab.api.models.GitlabMergeRequest;
import org.gitlab.api.models.GitlabNote;
import org.gitlab.api.models.GitlabProject;
import java.util.logging.Level;
import java.util.logging.LogRecord;
import java.util.logging.Logger;
import org.gitlab4j.api.Constants;
import org.gitlab4j.api.GitLabApi;
import org.gitlab4j.api.GitLabApiException;
import org.gitlab4j.api.models.Diff;
import org.gitlab4j.api.models.MergeRequest;
import org.gitlab4j.api.models.Note;
import org.gitlab4j.api.models.Position;
import org.gitlab4j.api.models.Project;
import se.bjurr.violations.comments.lib.CommentsProvider;
import se.bjurr.violations.comments.lib.PatchParser;
import se.bjurr.violations.comments.lib.ViolationsLogger;
Expand All @@ -22,38 +26,39 @@

public class GitLabCommentsProvider implements CommentsProvider {
private final ViolationCommentsToGitLabApi violationCommentsToGitLabApi;

private final GitlabAPI gitlabApi;
private final ViolationsLogger violationsLogger;

private GitlabProject project;

private GitlabMergeRequest mergeRequest;
private final GitLabApi gitLabApi;
private final Project project;
private final MergeRequest mergeRequest;

public GitLabCommentsProvider(
ViolationsLogger violationsLogger,
final ViolationCommentsToGitLabApi violationCommentsToGitLabApi) {
this.violationsLogger = violationsLogger;
final String hostUrl = violationCommentsToGitLabApi.getHostUrl();
final String apiToken = violationCommentsToGitLabApi.getApiToken();
final TokenType tokenType = violationCommentsToGitLabApi.getTokenType();
final AuthMethod method = violationCommentsToGitLabApi.getMethod();
gitlabApi = GitlabAPI.connect(hostUrl, apiToken, tokenType, method);

final boolean ignoreCertificateErrors =
violationCommentsToGitLabApi.isIgnoreCertificateErrors();
gitlabApi.ignoreCertificateErrors(ignoreCertificateErrors);
this.gitLabApi = new GitLabApi(hostUrl, apiToken);
gitLabApi.setIgnoreCertificateErrors(violationCommentsToGitLabApi.isIgnoreCertificateErrors());
gitLabApi.enableRequestResponseLogging(Level.INFO);
gitLabApi.withRequestResponseLogging(
new Logger(GitLabCommentsProvider.class.getName(), null) {
public void log(LogRecord record) {
violationsLogger.log(record.getLevel(), record.getMessage());
}
},
Level.FINE);

final String projectId = violationCommentsToGitLabApi.getProjectId();
try {
project = gitlabApi.getProject(projectId);
} catch (final Throwable e) {
this.project = gitLabApi.getProjectApi().getProject(projectId);
} catch (GitLabApiException e) {
throw new RuntimeException("Could not get project " + projectId, e);
}

final Integer mergeRequestId = violationCommentsToGitLabApi.getMergeRequestIid();
try {
mergeRequest = gitlabApi.getMergeRequestChanges(project.getId(), mergeRequestId);
mergeRequest =
gitLabApi.getMergeRequestApi().getMergeRequest(project.getId(), mergeRequestId);
} catch (final Throwable e) {
throw new RuntimeException("Could not get MR " + projectId + " " + mergeRequestId, e);
}
Expand All @@ -65,43 +70,58 @@ public GitLabCommentsProvider(
public void createCommentWithAllSingleFileComments(final String comment) {
markMergeRequestAsWIP();
try {
gitlabApi.createNote(mergeRequest, comment);
this.gitLabApi
.getNotesApi()
.createMergeRequestNote(project.getId(), this.mergeRequest.getIid(), comment);
} catch (final Throwable e) {
violationsLogger.log(SEVERE, "Could create comment " + comment, e);
}
}

/**
* Set the {@link GitlabMergeRequest} as "Work in Progress" if configured to do so by the
* shouldSetWIP flag.
* Set the merge request as "Work in Progress" if configured to do so by the shouldSetWIP flag.
*/
private void markMergeRequestAsWIP() {
if (!this.violationCommentsToGitLabApi.getShouldSetWIP()) {
return;
}

final String currentTitle = mergeRequest.getTitle();
if (currentTitle.startsWith("WIP:")) {
// To avoid setting WIP again on new comments
return;
}
final Serializable projectId = violationCommentsToGitLabApi.getProjectId();
final Integer mergeRequestIid = violationCommentsToGitLabApi.getMergeRequestIid();
final Integer projectId = this.project.getId();
final Integer mergeRequestIid = this.mergeRequest.getIid();
final String targetBranch = null;
final Integer assigneeId = null;
final String title = "WIP: >>> CONTAINS VIOLATIONS! <<< " + currentTitle;
final String description = null;
final String stateEvent = null;
final Constants.StateEvent stateEvent = null;
final String labels = null;
Integer milestoneId = null;
Boolean removeSourceBranch = null;
Boolean squash = null;
Boolean discussionLocked = null;
Boolean allowCollaboration = null;
try {
mergeRequest.setTitle(title); // To avoid setting WIP again on new comments
gitlabApi.updateMergeRequest(
projectId,
mergeRequestIid,
targetBranch,
assigneeId,
title,
description,
stateEvent,
labels);
mergeRequest.setTitle(title);
gitLabApi
.getMergeRequestApi()
.updateMergeRequest(
projectId,
mergeRequestIid,
targetBranch,
title,
assigneeId,
description,
stateEvent,
labels,
milestoneId,
removeSourceBranch,
squash,
discussionLocked,
allowCollaboration);
} catch (final Throwable e) {
violationsLogger.log(SEVERE, e.getMessage(), e);
}
Expand All @@ -112,9 +132,9 @@ public void createSingleFileComment(
final ChangedFile file, final Integer newLine, final String content) {
markMergeRequestAsWIP();
final Integer projectId = project.getId();
final String baseSha = mergeRequest.getBaseSha();
final String startSha = mergeRequest.getStartSha();
final String headSha = mergeRequest.getHeadSha();
final String baseSha = mergeRequest.getDiff_refs().getBase_sha();
final String startSha = mergeRequest.getDiff_refs().getStart_sha();
final String headSha = mergeRequest.getDiff_refs().getHead_sha();
final String patchString = file.getSpecifics().get(0);
final String oldPath = file.getSpecifics().get(1);
final String newPath = file.getSpecifics().get(2);
Expand All @@ -123,17 +143,21 @@ public void createSingleFileComment(
.findOldLine(newLine) //
.orElse(null);
try {
gitlabApi.createTextDiscussion(
mergeRequest,
content,
null,
baseSha,
startSha,
headSha,
newPath,
newLine,
oldPath,
oldLine);
final Date date = null;
final String positionHash = null;
final Position position = new Position();
position.setPositionType(Position.PositionType.TEXT);
position.setBaseSha(baseSha);
position.setStartSha(startSha);
position.setHeadSha(headSha);
position.setNewLine(newLine);
position.setNewPath(newPath);
position.setOldLine(oldLine);
position.setOldPath(oldPath);
gitLabApi
.getDiscussionsApi()
.createMergeRequestDiscussion(
projectId, mergeRequest.getIid(), content, date, positionHash, position);
} catch (final Throwable e) {
final String lineSeparator = System.lineSeparator();
violationsLogger.log(
Expand Down Expand Up @@ -170,8 +194,11 @@ public void createSingleFileComment(
public List<Comment> getComments() {
final List<Comment> found = new ArrayList<>();
try {
final List<GitlabNote> notes = gitlabApi.getAllNotes(mergeRequest);
for (final GitlabNote note : notes) {

final List<Note> notes =
gitLabApi.getNotesApi().getMergeRequestNotes(project.getId(), mergeRequest.getIid());

for (final Note note : notes) {
final String identifier = note.getId() + "";
final String content = note.getBody();
final String type = "PR";
Expand All @@ -188,7 +215,7 @@ public List<Comment> getComments() {
@Override
public List<ChangedFile> getFiles() {
final List<ChangedFile> changedFiles = new ArrayList<>();
for (final GitlabCommitDiff change : mergeRequest.getChanges()) {
for (final Diff change : mergeRequest.getChanges()) {
final String filename = change.getNewPath();
final List<String> specifics = new ArrayList<>();
final String patchString = change.getDiff();
Expand All @@ -206,17 +233,18 @@ public List<ChangedFile> getFiles() {
public void removeComments(final List<Comment> comments) {
for (final Comment comment : comments) {
try {
final GitlabNote noteToDelete = new GitlabNote();
noteToDelete.setId(Integer.parseInt(comment.getIdentifier()));
gitlabApi.deleteNote(mergeRequest, noteToDelete);
final int noteId = Integer.parseInt(comment.getIdentifier());
this.gitLabApi
.getNotesApi()
.deleteMergeRequestNote(project.getId(), mergeRequest.getIid(), noteId);
} catch (final Throwable e) {
violationsLogger.log(
INFO,
"Exception thrown when delete note "
+ comment.getIdentifier()
+ ". This is probably because of "
+ "https://github.com/timols/java-gitlab-api/issues/321");
// violationsLogger.log(SEVERE, "Could not delete note " + comment, e);
violationsLogger.log(SEVERE, "Could not delete note " + comment, e);
}
}
}
Expand Down
9 changes: 4 additions & 5 deletions src/main/java/se/bjurr/violations/comments/gitlab/lib/ViolationCommentsToGitLabApi.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,7 @@
import java.util.Scanner;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.gitlab.api.AuthMethod;
import org.gitlab.api.TokenType;
import org.gitlab4j.api.Constants.TokenType;
import se.bjurr.violations.comments.lib.CommentsProvider;
import se.bjurr.violations.comments.lib.ViolationsLogger;
import se.bjurr.violations.lib.model.Violation;
Expand All @@ -33,7 +32,7 @@ public static ViolationCommentsToGitLabApi violationCommentsToGitLabApi() {
private String hostUrl;
private String apiToken;
private TokenType tokenType;
private AuthMethod method;
private String method;
private boolean ignoreCertificateErrors;
private String projectId;
private Integer mergeRequestIid;
Expand Down Expand Up @@ -95,11 +94,11 @@ public ViolationCommentsToGitLabApi setTokenType(final TokenType tokenType) {
return this;
}

public AuthMethod getMethod() {
public String getMethod() {
return method;
}

public ViolationCommentsToGitLabApi setMethod(final AuthMethod method) {
public ViolationCommentsToGitLabApi setMethod(final String method) {
this.method = method;
return this;
}
Expand Down

0 comments on commit 1f6f39c

Please sign in to comment.