Add more Repository attributes

ls1intum · Aug 31, 2024 · f9af54a · f9af54a
1 parent cf50f37
commit f9af54a
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/...tion-server/src/main/java/de/tum/in/www1/hephaestus/codereview/repository/Repository.java b/...tion-server/src/main/java/de/tum/in/www1/hephaestus/codereview/repository/Repository.java
@@ -33,9 +33,17 @@ public class Repository extends BaseGitServiceEntity {
     @NonNull
     private String description;
 
+    @NonNull
+    String defaultBranch;
+
+    @NonNull
+    private RepositoryVisibility visibility;
+
     @NonNull
     private String url;
 
+    String homepage;
+
     @OneToMany(cascade = CascadeType.ALL, mappedBy = "repository", fetch = FetchType.EAGER)
     @ToString.Exclude
     private Set<PullRequest> pullRequests = new HashSet<>();

diff --git a/...er/src/main/java/de/tum/in/www1/hephaestus/codereview/repository/RepositoryConverter.java b/...er/src/main/java/de/tum/in/www1/hephaestus/codereview/repository/RepositoryConverter.java
@@ -1,6 +1,7 @@
 package de.tum.in.www1.hephaestus.codereview.repository;
 
 import org.kohsuke.github.GHRepository;
+import org.kohsuke.github.GHRepository.Visibility;
 import org.springframework.data.convert.ReadingConverter;
 import org.springframework.lang.NonNull;
 import org.springframework.lang.Nullable;
@@ -21,7 +22,20 @@ public Repository convert(@NonNull GHRepository source) {
         repository.setNameWithOwner(source.getFullName());
         repository.setDescription(source.getDescription());
         repository.setUrl(source.getHtmlUrl().toString());
+        repository.setDefaultBranch(source.getDefaultBranch());
+        repository.setVisibility(convertVisibility(source.getVisibility()));
+        repository.setHomepage(source.getHomepage());
         return repository;
     }
 
+    private RepositoryVisibility convertVisibility(Visibility visibility) {
+        switch (visibility) {
+            case PRIVATE:
+                return RepositoryVisibility.PRIVATE;
+            case PUBLIC:
+                return RepositoryVisibility.PUBLIC;
+            default:
+                return RepositoryVisibility.PRIVATE;
+        }
+    }
 }
diff --git a/...r/src/main/java/de/tum/in/www1/hephaestus/codereview/repository/RepositoryVisibility.java b/...r/src/main/java/de/tum/in/www1/hephaestus/codereview/repository/RepositoryVisibility.java
@@ -0,0 +1,5 @@
+package de.tum.in.www1.hephaestus.codereview.repository;
+
+public enum RepositoryVisibility {
+    PUBLIC, PRIVATE
+}