api.v1.workers.ts

import { json, TypedResponse } from "@remix-run/server-runtime";
import {
  WorkersCreateRequestBody,
  WorkersCreateResponseBody,
  WorkersListResponseBody,
} from "@trigger.dev/core/v3";
import {
  createActionApiRoute,
  createLoaderApiRoute,
} from "~/services/routeBuilders/apiBuilder.server";
import { WorkerGroupService } from "~/v3/services/worker/workerGroupService.server";

export const loader = createLoaderApiRoute(
  {
    corsStrategy: "all",
    findResource: async () => 1, // This is a dummy function, we don't need to find a resource
  },
  async ({
    authentication,
  }): Promise<TypedResponse<WorkersListResponseBody | { error: string }>> => {
    if (authentication.environment.project.engine !== "V2") {
      return json({ error: "Not supported for V1 projects" }, { status: 400 });
    }

    const service = new WorkerGroupService();
    const workers = await service.listWorkerGroups({
      projectId: authentication.environment.projectId,
    });

    // Reuse the trigger path's resolution so isDefault matches where runs
    // actually route: env default -> project default -> global default.
    const defaultWorkerGroup = await service.getDefaultWorkerGroupForProject({
      projectId: authentication.environment.projectId,
      environmentDefaultWorkerGroupId: authentication.environment.defaultWorkerGroupId,
    });

    return json(
      workers.map((w) => ({
        type: w.type,
        name: w.name,
        description: w.description,
        isDefault: w.id === defaultWorkerGroup?.id,
        updatedAt: w.updatedAt,
      }))
    );
  }
);

export const { action } = createActionApiRoute(
  {
    corsStrategy: "all",
    body: WorkersCreateRequestBody,
  },
  async ({
    authentication,
    body,
  }): Promise<TypedResponse<WorkersCreateResponseBody | { error: string }>> => {
    if (authentication.environment.project.engine !== "V2") {
      return json({ error: "Not supported" }, { status: 400 });
    }

    const service = new WorkerGroupService();
    const { workerGroup, token } = await service.createWorkerGroup({
      projectId: authentication.environment.projectId,
      organizationId: authentication.environment.organizationId,
      name: body.name,
      description: body.description,
    });

    return json({
      token: {
        plaintext: token.plaintext,
      },
      workerGroup: {
        name: workerGroup.name,
        description: workerGroup.description,
      },
    });
  }
);