feat: add process route

Author: wei

deno.jsonc

@@ -16,7 +16,7 @@
     "@probot/pino": "npm:@probot/pino@^2.5.0",
     "@std/assert": "jsr:@std/assert@1",
     "@std/http": "jsr:@std/http@1",
-    "@wei/probot-scheduler": "jsr:@wei/[email protected].9",
+    "@wei/probot-scheduler": "jsr:@wei/[email protected].10",
     "bullmq": "npm:bullmq@^5.21.2",
     "express": "npm:express@^4.21.1",
     "ioredis": "npm:ioredis@^5.4.1",

deno.lock

@@ -1,6 +1,11 @@
 import express from "express";
 import { createNodeMiddleware, createProbot } from "probot";
-import { JobPriority } from "@wei/probot-scheduler";
+import {
+  createSchedulerService,
+  JobPriority,
+  RepositoryMetadataSchemaType,
+  RepositorySchemaType,
+} from "@wei/probot-scheduler";
 import createSchedulerApp from "@/src/app.ts";
 import { appConfig } from "@/src/configs/app-config.ts";
 import log from "@/src/utils/logger.ts";
@@ -21,21 +26,29 @@ const probot = createProbot({
     log,
   },
 });
+// deno-lint-ignore require-await
+async function getRepositorySchedule(
+  repository: RepositorySchemaType,
+  currentMetadata?: RepositoryMetadataSchemaType,
+) {
+  return {
+    repository_id: repository.id,
+    cron: currentMetadata?.cron ?? getRandomCronSchedule(),
+    job_priority: currentMetadata?.job_priority ?? JobPriority.Normal,
+  };
+}
 const schedulerApp = createSchedulerApp.bind(null, probot, {
   // Optional: Skip the initial full sync
   skipFullSync,
 
   redisClient,
 
   // Define custom repository scheduling
-  // deno-lint-ignore require-await
-  getRepositorySchedule: async (repository, currentMetadata) => {
-    return {
-      repository_id: repository.id,
-      cron: currentMetadata?.cron ?? getRandomCronSchedule(),
-      job_priority: currentMetadata?.job_priority ?? JobPriority.Normal,
-    };
-  },
+  getRepositorySchedule,
+});
+const schedulerService = createSchedulerService(probot, {
+  redisClient,
+  getRepositorySchedule,
 });
 
 const server = express();
@@ -47,7 +60,7 @@ server.use(
     webhooksPath: "/",
   }),
 );
-server.use("/", createRouter(probot));
+server.use("/", createRouter(probot, schedulerService));
 
 server.listen(appConfig.port, () => {
   log.info(`[Express] Server is running on port ${appConfig.port}`);

src/index.ts

@@ -1,10 +1,14 @@
 import type { Request, Response } from "express";
 import type { Probot } from "probot";
 import express from "express";
+import { createSchedulerService } from "@wei/probot-scheduler";
 import getStatsHandlers from "@/src/router/stats.ts";
 import getRepoHandlers from "@/src/router/repo-handler.ts";
 
-const createRouter = (app: Probot) => {
+const createRouter = (
+  app: Probot,
+  schedulerService: ReturnType<typeof createSchedulerService>,
+) => {
   const router = express.Router();
 
   router.get("/", (_req: Request, res: Response) => {
@@ -18,9 +22,12 @@ const createRouter = (app: Probot) => {
   const { probotStatsHandler } = getStatsHandlers(app);
   router.get("/probot/stats", probotStatsHandler);
 
-  const { checkHandler } = getRepoHandlers(app);
+  const { checkHandler, processHandler } = getRepoHandlers(
+    app,
+    schedulerService,
+  );
   router.get("/check/:owner/:repo", checkHandler);
-  // TODO Add process route
+  router.get("/process/:owner/:repo", processHandler);
 
   return router;
 };

src/router/index.ts

@@ -2,12 +2,19 @@ import type { Request, Response } from "express";
 import type { Probot } from "probot";
 import { appConfig } from "@/src/configs/app-config.ts";
 import { getPullConfig } from "@/src/utils/get-pull-config.ts";
-import { JobPriority, RepositoryModel } from "@wei/probot-scheduler";
+import {
+  createSchedulerService,
+  JobPriority,
+  RepositoryModel,
+} from "@wei/probot-scheduler";
 
-function getRepoHandlers(app: Probot) {
+function getRepoHandlers(
+  app: Probot,
+  schedulerService: ReturnType<typeof createSchedulerService>,
+) {
   async function checkHandler(req: Request, res: Response) {
     const full_name = `${req.params.owner}/${req.params.repo}`;
-    app.log.info(`[${full_name}] Checking ${appConfig.configFilename}`);
+    app.log.info({ full_name }, `Checking ${appConfig.configFilename}`);
 
     try {
       // Get Octokit
@@ -57,8 +64,18 @@ function getRepoHandlers(app: Probot) {
     }
   }
 
+  async function processHandler(req: Request, res: Response) {
+    const full_name = `${req.params.owner}/${req.params.repo}`;
+    app.log.info({ full_name }, `Processing`);
+
+    await schedulerService.processRepository({ fullName: full_name }, true);
+
+    res.json({ status: "queued" });
+  }
+
   return {
     checkHandler,
+    processHandler,
   };
 }
 

src/router/repo-handler.ts

@@ -1,4 +1,4 @@
-import { createWorker } from "@wei/probot-scheduler";
+import { createSchedulerWorker } from "@wei/probot-scheduler";
 import { Redis } from "ioredis";
 import { appConfig } from "@/src/configs/app-config.ts";
 import RepoJobProcessor from "@/src/processor/index.ts";
@@ -7,7 +7,7 @@ const redisClient = new Redis(appConfig.redisConfig!, {
   maxRetriesPerRequest: null,
 });
 
-const worker = createWorker(
+const worker = createSchedulerWorker(
   RepoJobProcessor,
   {
     connection: redisClient,