Adds the PR Label Bot

AffectedArc07 · AffectedArc07 · commit dcd043ebd9ba · 2025-10-02T17:58:54.000+01:00
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -0,0 +1,15 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "type": "java",
+            "name": "Attach",
+            "request": "attach",
+            "hostName": "localhost",
+            "port": "8000"
+        },
+    ]
+}
diff --git a/README.MD b/README.MD
@@ -12,6 +12,7 @@ It handles:
 - IP2ASN
 - Profiler Daemon Ingest
 - Profiler DB Cleanup
+- PR Label Bot
 
 ### ACL Cleanup
 
@@ -39,6 +40,10 @@ This project itself does not provide a data extraction method, but that can be f
 
 Just wipes rows older than 7 days out the profiler table.
 
+### PR Label Bot
+
+Handles labelling PRs on the GitHub based on TM status, assigned type, votes and other stuff.
+
 ## Provided Projects
 
 - `TaskDaemon.Core` - Core application which handles running tasks.
@@ -58,7 +63,7 @@ To build, run `mvn package` to generate the JAR file. This will then appear in t
 
 1. Compile the codebase (See `Building`).
 2. Copy the jar and the example `config.toml` in the repository root to a running location.
-3. Import the `schemas/profiler_schema.sql` schema to your database server.
+3. Import the relevant schemas from the `schemas` dir to your database server.
 4. Setup the configuration as per section labels.
 5. Start the application up, it will now run tasks in the background.
 
diff --git a/TaskDaemon.Core/pom.xml b/TaskDaemon.Core/pom.xml
@@ -27,6 +27,11 @@
             <artifactId>taskdaemon-database-profiler</artifactId>
             <version>dev-SNAPSHOT</version>
         </dependency>
+        <dependency>
+            <groupId>me.aa07.paradise</groupId>
+            <artifactId>taskdaemon-database-pullrequests</artifactId>
+            <version>dev-SNAPSHOT</version>
+        </dependency>
         <dependency>
             <groupId>org.apache.logging.log4j</groupId>
             <artifactId>log4j-core</artifactId>
@@ -55,6 +60,10 @@
             <groupId>org.apache.logging.log4j</groupId>
             <artifactId>log4j-slf4j2-impl</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.kohsuke</groupId>
+            <artifactId>github-api</artifactId>
+        </dependency>
     </dependencies>
 
     <build>
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/Core.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/Core.java
@@ -9,6 +9,7 @@
 import me.aa07.paradise.taskdaemon.core.modules.bouncerrestart.BouncerRestartJob;
 import me.aa07.paradise.taskdaemon.core.modules.devrank.DevRankJob;
 import me.aa07.paradise.taskdaemon.core.modules.ip2asn.Ip2AsnJob;
+import me.aa07.paradise.taskdaemon.core.modules.prlabel.PrLabelJob;
 import me.aa07.paradise.taskdaemon.core.modules.profilercleanup.ProfilerCleanupJob;
 import me.aa07.paradise.taskdaemon.core.modules.profileringest.ProfilerWorker;
 import me.aa07.paradise.taskdaemon.core.redis.RedisManager;
@@ -153,12 +154,28 @@ private void setupJobs(Scheduler scheduler, DbCore dbCore, ConfigHolder config,
             .withSchedule(CronScheduleBuilder.cronSchedule("0 0 8 * * ?")) // Every day - 8AM
             .build();
 
+        // PR Label bot
+        JobDataMap jdm_pullrequests = new JobDataMap();
+        jdm_pullrequests.put("LOGGER", logger);
+        jdm_pullrequests.put("DBCORE", dbCore);
+        jdm_pullrequests.put("TGSCFG", config.tgs);
+        jdm_pullrequests.put("GITHUBCFG", config.github);
+        JobDetail jd_pullrequests = JobBuilder.newJob(PrLabelJob.class)
+                .withIdentity("pullrequests", "pullrequests")
+                .usingJobData(jdm_pullrequests)
+                .build();
+        CronTrigger ct_pullrequests = TriggerBuilder.newTrigger()
+                .withIdentity("pullrequests", "pullrequests")
+                .withSchedule(CronScheduleBuilder.cronSchedule("0 0 * * * ?")) // Every hour
+                .build();
+
         // Schedule all
         scheduler.scheduleJob(jd_aclcleanup, ct_aclcleanup);
         scheduler.scheduleJob(jd_bouncerrestart, ct_bouncerrestart);
         scheduler.scheduleJob(jd_devrank, ct_devrank);
         scheduler.scheduleJob(jd_ip2asn, ct_ip2asn);
         scheduler.scheduleJob(jd_profilercleanup, ct_profilercleanup);
+        scheduler.scheduleJob(jd_pullrequests, ct_pullrequests);
     }
 
     private void launchAll() {
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/config/ConfigHolder.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/config/ConfigHolder.java
@@ -3,9 +3,11 @@
 public class ConfigHolder {
     public DatabaseConfig forumsDatabase;
     public DatabaseConfig gameDatabase;
+    public GithubConfig github;
     public Ip2AsnSerivceConfig ip2asn;
     public PfsenseConfig pfsense;
     public DatabaseConfig profilerDatabase;
+    public DatabaseConfig pullRequestsDatabase;
     public RedisConfig redis;
     public TgsConfig tgs;
 }
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/config/GithubConfig.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/config/GithubConfig.java
@@ -0,0 +1,9 @@
+package me.aa07.paradise.taskdaemon.core.config;
+
+public class GithubConfig {
+    public String githubToken;
+    public long repoId;
+    public String orgName;
+    public String reviewTeamSlug;
+    public String reviewTeamToken;
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/database/DatabaseType.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/database/DatabaseType.java
@@ -1,5 +1,5 @@
 package me.aa07.paradise.taskdaemon.core.database;
 
 public enum DatabaseType {
-    ProfilerDb, GameDb, Forums
+    GameDb, Forums, ProfilerDb, PullRequests
 }
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/database/DbCore.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/database/DbCore.java
@@ -54,6 +54,7 @@ private void establishConnections(ConfigHolder config) {
         db_types.put(DatabaseType.Forums, config.forumsDatabase);
         db_types.put(DatabaseType.GameDb, config.gameDatabase);
         db_types.put(DatabaseType.ProfilerDb, config.profilerDatabase);
+        db_types.put(DatabaseType.PullRequests, config.pullRequestsDatabase);
 
         for (DatabaseType dbtype : db_types.keySet()) {
             DatabaseConfig cfg = db_types.get(dbtype);
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/prlabel/PullRequest.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/prlabel/PullRequest.java
@@ -0,0 +1,66 @@
+package me.aa07.paradise.taskdaemon.core.models.prlabel;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.HashMap;
+import me.aa07.paradise.taskdaemon.database.pullrequests.enums.VotesNewVoteType;
+import me.aa07.paradise.taskdaemon.database.pullrequests.enums.VotesNewVotingGroup;
+import org.kohsuke.github.GHPullRequest;
+
+public class PullRequest {
+    public int pullNumber;
+    public int testmergeRequests;
+    public String prTypes;
+    public GHPullRequest prObject;
+    public HashMap<VotesNewVotingGroup, HashMap<VotesNewVoteType, Integer>> votes;
+    public ArrayList<String> labels;
+
+    public PullRequest() {
+        votes = new HashMap<VotesNewVotingGroup, HashMap<VotesNewVoteType, Integer>>();
+        labels = new ArrayList<String>();
+    }
+
+    public int sumTotalApprovals() {
+        return sumTotalByVoteType(VotesNewVoteType.APPROVE);
+    }
+
+    public int sumTotalObjections() {
+        return sumTotalByVoteType(VotesNewVoteType.OBJECT);
+    }
+
+    // Safety caching - stops eating API calls
+    public boolean safeAddLabel(String label) throws IOException {
+        if (labels.contains(label)) {
+            return false;
+        }
+
+        labels.add(label);
+        prObject.addLabels(label);
+        return true;
+    }
+
+    public boolean safeDelLabel(String label) throws IOException {
+        if (!labels.contains(label)) {
+            return false;
+        }
+
+        labels.add(label);
+        prObject.removeLabels(label);
+        return true;
+    }
+
+    // Private methods
+    private int sumTotalByVoteType(VotesNewVoteType voteType) {
+        int total = 0;
+
+        for (VotesNewVotingGroup group : votes.keySet()) {
+            HashMap<VotesNewVoteType, Integer> vote_map = votes.get(group);
+
+            if (vote_map.containsKey(voteType)) {
+                total += vote_map.get(voteType);
+            }
+        }
+
+        return total;
+    }
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/CompileJobResponse.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/CompileJobResponse.java
@@ -0,0 +1,5 @@
+package me.aa07.paradise.taskdaemon.core.models.tgs;
+
+public class CompileJobResponse {
+    public RevisionInformation revisionInformation;
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/DreamDaemonResponse.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/DreamDaemonResponse.java
@@ -0,0 +1,5 @@
+package me.aa07.paradise.taskdaemon.core.models.tgs;
+
+public class DreamDaemonResponse {
+    public CompileJobResponse activeCompileJob;
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/RevisionInformation.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/RevisionInformation.java
@@ -0,0 +1,7 @@
+package me.aa07.paradise.taskdaemon.core.models.tgs;
+
+import java.util.List;
+
+public class RevisionInformation {
+    public List<TestMerge> activeTestMerges;
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/TestMerge.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/models/tgs/TestMerge.java
@@ -0,0 +1,5 @@
+package me.aa07.paradise.taskdaemon.core.models.tgs;
+
+public class TestMerge {
+    public Integer number;
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/FilesChangedManager.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/FilesChangedManager.java
@@ -0,0 +1,46 @@
+package me.aa07.paradise.taskdaemon.core.modules.prlabel;
+
+import java.io.IOException;
+import me.aa07.paradise.taskdaemon.core.models.prlabel.PullRequest;
+import org.apache.logging.log4j.Logger;
+import org.kohsuke.github.GHPullRequestFileDetail;
+
+public class FilesChangedManager implements IPullRequestProcessor {
+    private Logger logger;
+
+    public FilesChangedManager(Logger logger) {
+        this.logger = logger;
+    }
+
+    // Dont change these
+    private static final String LABEL_CONFIG = "Configuration Change";
+    private static final String LABEL_MAP_EDIT = "Map Edit";
+    private static final String LABEL_RUSTLIBS = "Rustlibs";
+    private static final String LABEL_SOUND = "Sound";
+    private static final String LABEL_SPRITES = "Sprites";
+    private static final String LABEL_SQL = "SQL Change";
+    private static final String LABEL_TGUI = "TGUI";
+
+    @Override
+    public void processPr(PullRequest pr) throws IOException {
+        for (GHPullRequestFileDetail fd : pr.prObject.listFiles()) {
+            internalProcessPr(pr, fd, LABEL_CONFIG, "config/");
+            internalProcessPr(pr, fd, LABEL_MAP_EDIT, "_maps/map_files/");
+            internalProcessPr(pr, fd, LABEL_RUSTLIBS, "rust/");
+            internalProcessPr(pr, fd, LABEL_SOUND, "sound/");
+            internalProcessPr(pr, fd, LABEL_SPRITES, "icons/");
+            internalProcessPr(pr, fd, LABEL_SQL, "SQL/");
+            internalProcessPr(pr, fd, LABEL_TGUI, "tgui/");
+        }
+    }
+
+    private void internalProcessPr(PullRequest pr, GHPullRequestFileDetail fd, String label, String path)
+            throws IOException {
+        if (fd.getFilename().startsWith(path)) {
+            if (pr.safeAddLabel(label)) {
+                logger.info(
+                        String.format("[PrLabelJob] [FilesChangedManager] Adding '%s' to #%s", label, pr.pullNumber));
+            }
+        }
+    }
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/IPullRequestProcessor.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/IPullRequestProcessor.java
@@ -0,0 +1,8 @@
+package me.aa07.paradise.taskdaemon.core.modules.prlabel;
+
+import java.io.IOException;
+import me.aa07.paradise.taskdaemon.core.models.prlabel.PullRequest;
+
+public interface IPullRequestProcessor {
+    public void processPr(PullRequest pr) throws IOException;
+}
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/PrLabelJob.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/PrLabelJob.java
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/StatusManager.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/StatusManager.java
diff --git a/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/TestmergeManager.java b/TaskDaemon.Core/src/me/aa07/paradise/taskdaemon/core/modules/prlabel/TestmergeManager.java
diff --git a/config.toml.example b/config.toml.example
diff --git a/pom.xml b/pom.xml
diff --git a/start.sh b/start.sh

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,5 @@`
`1`	`1`	`package me.aa07.paradise.taskdaemon.core.database;`
`2`	`2`
`3`	`3`	`public enum DatabaseType {`
`4`		`- ProfilerDb, GameDb, Forums`
	`4`	`+ GameDb, Forums, ProfilerDb, PullRequests`
`5`	`5`	`}`