Merge pull request #86 from xmtp-coder-agent/fix/issue-85

feat: add codex template support

Author: neekolas

src/config.ts

@@ -11,6 +11,7 @@ const AppConfigSchema = z.object({
 	coderToken: z.string().min(1),
 	coderTaskNamePrefix: z.string().min(1).default("gh"),
 	coderTemplateName: z.string().min(1).default("task-template"),
+	coderTemplateNameCodex: z.string().min(1).default("task-template-codex"),
 	coderTemplatePreset: z.string().min(1).optional(),
 	coderOrganization: z.string().min(1).default("default"),
 	logFormat: z.string().optional(),
@@ -35,6 +36,7 @@ export function loadConfig(env: Record<string, string | undefined>): AppConfig {
 		coderToken: env.CODER_TOKEN,
 		coderTaskNamePrefix: env.CODER_TASK_NAME_PREFIX,
 		coderTemplateName: env.CODER_TEMPLATE_NAME,
+		coderTemplateNameCodex: env.CODER_TEMPLATE_NAME_CODEX,
 		coderTemplatePreset: env.CODER_TEMPLATE_PRESET,
 		coderOrganization: env.CODER_ORGANIZATION,
 		logFormat: env.LOG_FORMAT,

src/handler-dispatcher.test.ts

@@ -30,6 +30,7 @@ const mockConfig: AppConfig = {
 	coderToken: "coder-token",
 	coderTaskNamePrefix: "gh",
 	coderTemplateName: "task-template",
+	coderTemplateNameCodex: "task-template-codex",
 	coderOrganization: "default",
 	port: 3000,
 };
@@ -92,6 +93,8 @@ describe("HandlerDispatcher", () => {
 		const createTaskContext = {
 			issueNumber: 42,
 			issueUrl: "https://github.com/xmtp/test-repo/issues/42",
+			issueTitle: "Fix some bug",
+			issueLabels: [] as string[],
 			repoName: "test-repo",
 			repoOwner: "xmtp",
 			senderLogin: "human-dev",

src/handler-dispatcher.ts

@@ -51,6 +51,7 @@ export class HandlerDispatcher {
 			coderToken: this.options.config.coderToken,
 			coderTaskNamePrefix: this.options.config.coderTaskNamePrefix,
 			coderTemplateName: this.options.config.coderTemplateName,
+			coderTemplateNameCodex: this.options.config.coderTemplateNameCodex,
 			coderTemplatePreset: this.options.config.coderTemplatePreset,
 			coderOrganization: this.options.config.coderOrganization,
 			agentGithubUsername: this.options.config.agentGithubUsername,
@@ -73,6 +74,8 @@ export class HandlerDispatcher {
 						repo: ctx.repoName,
 						issueNumber: ctx.issueNumber,
 						issueUrl: ctx.issueUrl,
+						issueTitle: ctx.issueTitle,
+						issueLabels: ctx.issueLabels,
 						senderLogin: ctx.senderLogin,
 					},
 					logger,

src/handlers/close-task.test.ts

@@ -16,6 +16,7 @@ const baseInputs: HandlerConfig = {
 	coderUsername: "coder-agent",
 	coderTaskNamePrefix: "gh",
 	coderTemplateName: "task-template",
+	coderTemplateNameCodex: "task-template-codex",
 	coderOrganization: "default",
 	agentGithubUsername: "xmtp-coder-agent",
 };

src/handlers/create-task.test.ts

@@ -15,6 +15,7 @@ const baseInputs: HandlerConfig = {
 	coderUsername: "coder-agent",
 	coderTaskNamePrefix: "gh",
 	coderTemplateName: "task-template",
+	coderTemplateNameCodex: "task-template-codex",
 	coderOrganization: "default",
 	agentGithubUsername: "xmtp-coder-agent",
 };
@@ -24,6 +25,8 @@ const issueContext = {
 	repo: "libxmtp",
 	issueNumber: 42,
 	issueUrl: "https://github.com/xmtp/libxmtp/issues/42",
+	issueTitle: "Fix some bug",
+	issueLabels: [],
 	senderLogin: "human-dev",
 };
 
@@ -185,6 +188,94 @@ describe("CreateTaskHandler", () => {
 		expect(String(taskNameArg)).toBe("gh-libxmtp-42");
 	});
 
+	test("uses codex template when issue title contains codex", async () => {
+		github.checkActorPermission.mockResolvedValue(true);
+		coder.getTask.mockResolvedValue(null);
+		coder.createTask.mockResolvedValue(mockTask);
+
+		const ctx = {
+			...issueContext,
+			issueTitle: "Add Codex Support",
+		};
+		const handler = new CreateTaskHandler(
+			coder,
+			github as unknown as import("../github-client").GitHubClient,
+			baseInputs,
+			ctx,
+			logger,
+		);
+		await handler.run();
+
+		const templateCall = coder.getTemplateByOrganizationAndName.mock
+			.calls[0] as unknown as [string, string];
+		expect(templateCall[1]).toBe("task-template-codex");
+	});
+
+	test("uses codex template when issue has codex label", async () => {
+		github.checkActorPermission.mockResolvedValue(true);
+		coder.getTask.mockResolvedValue(null);
+		coder.createTask.mockResolvedValue(mockTask);
+
+		const ctx = {
+			...issueContext,
+			issueLabels: ["enhancement", "codex"],
+		};
+		const handler = new CreateTaskHandler(
+			coder,
+			github as unknown as import("../github-client").GitHubClient,
+			baseInputs,
+			ctx,
+			logger,
+		);
+		await handler.run();
+
+		const templateCall = coder.getTemplateByOrganizationAndName.mock
+			.calls[0] as unknown as [string, string];
+		expect(templateCall[1]).toBe("task-template-codex");
+	});
+
+	test("uses default template when no codex indicator", async () => {
+		github.checkActorPermission.mockResolvedValue(true);
+		coder.getTask.mockResolvedValue(null);
+		coder.createTask.mockResolvedValue(mockTask);
+
+		const handler = new CreateTaskHandler(
+			coder,
+			github as unknown as import("../github-client").GitHubClient,
+			baseInputs,
+			issueContext,
+			logger,
+		);
+		await handler.run();
+
+		const templateCall = coder.getTemplateByOrganizationAndName.mock
+			.calls[0] as unknown as [string, string];
+		expect(templateCall[1]).toBe("task-template");
+	});
+
+	test("codex match in title is case insensitive", async () => {
+		github.checkActorPermission.mockResolvedValue(true);
+		coder.getTask.mockResolvedValue(null);
+		coder.createTask.mockResolvedValue(mockTask);
+
+		const ctx = {
+			...issueContext,
+			issueTitle: "Use CODEX for processing",
+		};
+		const handler = new CreateTaskHandler(
+			coder,
+			github as unknown as import("../github-client").GitHubClient,
+			baseInputs,
+			ctx,
+			logger,
+		);
+		await handler.run();
+
+		const templateCall = coder.getTemplateByOrganizationAndName.mock
+			.calls[0] as unknown as [string, string];
+		expect(templateCall[1]).toBe("task-template-codex");
+	});
+
 	test("logs task name via injected logger", async () => {
 		github.checkActorPermission.mockResolvedValue(true);
 		coder.getTask.mockResolvedValue(null);

src/handlers/create-task.ts

@@ -10,6 +10,8 @@ export interface IssueContext {
 	repo: string;
 	issueNumber: number;
 	issueUrl: string;
+	issueTitle: string;
+	issueLabels: string[];
 	senderLogin: string;
 }
 
@@ -88,9 +90,10 @@ export class CreateTaskHandler {
 			: this.context.issueUrl;
 
 		// 5. Get template and create task
+		const templateName = this.resolveTemplateName();
 		const template = await this.coder.getTemplateByOrganizationAndName(
 			this.inputs.coderOrganization,
-			this.inputs.coderTemplateName,
+			templateName,
 		);
 
 		const presets = await this.coder.getTemplateVersionPresets(
@@ -136,6 +139,20 @@ export class CreateTaskHandler {
 		};
 	}
 
+	private resolveTemplateName(): string {
+		const titleHasCodex = /codex/i.test(this.context.issueTitle);
+		const labelsHaveCodex = this.context.issueLabels.some(
+			(label) => label.toLowerCase() === "codex",
+		);
+		if (titleHasCodex || labelsHaveCodex) {
+			this.logger.info(
+				`Using codex template: ${this.inputs.coderTemplateNameCodex}`,
+			);
+			return this.inputs.coderTemplateNameCodex;
+		}
+		return this.inputs.coderTemplateName;
+	}
+
 	private generateTaskUrl(coderUsername: string, taskId: string): string {
 		const baseURL = this.inputs.coderURL.replace(/\/$/, "");
 		return `${baseURL}/tasks/${coderUsername}/${taskId}`;

src/handlers/failed-check.test.ts

@@ -15,6 +15,7 @@ const baseInputs: HandlerConfig = {
 	coderUsername: "coder-agent",
 	coderTaskNamePrefix: "gh",
 	coderTemplateName: "task-template",
+	coderTemplateNameCodex: "task-template-codex",
 	coderOrganization: "default",
 	agentGithubUsername: "xmtp-coder-agent",
 };

src/handlers/issue-comment.test.ts

@@ -16,6 +16,7 @@ const baseInputs: HandlerConfig = {
 	coderUsername: "coder-agent",
 	coderTaskNamePrefix: "gh",
 	coderTemplateName: "task-template",
+	coderTemplateNameCodex: "task-template-codex",
 	coderOrganization: "default",
 	agentGithubUsername: "xmtp-coder-agent",
 };

src/handlers/pr-comment.test.ts

@@ -16,6 +16,7 @@ const baseInputs: HandlerConfig = {
 	coderUsername: "coder-agent",
 	coderTaskNamePrefix: "gh",
 	coderTemplateName: "task-template",
+	coderTemplateNameCodex: "task-template-codex",
 	coderOrganization: "default",
 	agentGithubUsername: "xmtp-coder-agent",
 };

src/schemas.ts

@@ -7,6 +7,7 @@ export interface HandlerConfig {
 	coderToken: string;
 	coderTaskNamePrefix: string;
 	coderTemplateName: string;
+	coderTemplateNameCodex: string;
 	coderTemplatePreset?: string;
 	coderOrganization: string;
 	agentGithubUsername: string; // replaces coderGithubUsername