fix: alow to run tasks via nitro cli (#4188)

Co-authored-by: Rihan Arfan <me@file.properties>

Author: jkniest

src/runtime/internal/routes/dev-tasks.ts

@@ -1,8 +1,21 @@
-import { H3 } from "h3";
+import { H3, type H3Event } from "h3";
 import { runTask } from "../task.ts";
 
 import { scheduledTasks, tasks } from "#nitro/virtual/tasks";
 
+const taskHandler = async (event: H3Event) => {
+  const name = event.context.params?.name;
+  const body = (await event.req.json().catch(() => ({}))) as Record<string, unknown>;
+  const payload = {
+    ...Object.fromEntries(event.url.searchParams.entries()),
+    ...((body.payload as Record<string, unknown>) ?? body),
+  };
+  return await runTask(name!, {
+    context: { waitUntil: event.req.waitUntil },
+    payload,
+  });
+};
+
 const app: H3 = new H3()
   .get("/_nitro/tasks", async () => {
     const _tasks = await Promise.all(
@@ -16,17 +29,7 @@ const app: H3 = new H3()
       scheduledTasks,
     };
   })
-  .get("/_nitro/tasks/:name", async (event) => {
-    const name = event.context.params?.name;
-    const body = (await event.req.json().catch(() => ({}))) as Record<string, unknown>;
-    const payload = {
-      ...Object.fromEntries(event.url.searchParams.entries()),
-      ...body,
-    };
-    return await runTask(name!, {
-      context: { waitUntil: event.req.waitUntil },
-      payload,
-    });
-  });
+  .get("/_nitro/tasks/:name", taskHandler)
+  .post("/_nitro/tasks/:name", taskHandler);
 
 export default app;

test/presets/nitro-dev.test.ts

@@ -22,6 +22,58 @@ describe("nitro:preset:nitro-dev", async () => {
         expect(status).toBe(200);
       });
 
+      describe("tasks", () => {
+        it("GET /_nitro/tasks lists tasks", async () => {
+          const { data, status } = await callHandler({ url: "/_nitro/tasks" });
+          expect(status).toBe(200);
+          expect(data.tasks).toBeTypeOf("object");
+          expect(data.tasks.test).toMatchObject({ description: "task to debug" });
+          expect(data.tasks["db:migrate"]).toMatchObject({
+            description: "Run database migrations",
+          });
+          expect(data.scheduledTasks).toEqual(
+            expect.arrayContaining([
+              expect.objectContaining({ cron: "* * * * *", tasks: ["test"] }),
+            ])
+          );
+        });
+
+        it("GET /_nitro/tasks/:name runs a task", async () => {
+          const { data, status } = await callHandler({
+            url: "/_nitro/tasks/db:migrate",
+          });
+          expect(status).toBe(200);
+          expect(data.result).toBe("Success");
+        });
+
+        it("POST /_nitro/tasks/:name runs a task", async () => {
+          const { data, status } = await callHandler({
+            url: "/_nitro/tasks/db:migrate",
+            method: "POST",
+          });
+          expect(status).toBe(200);
+          expect(data.result).toBe("Success");
+        });
+
+        it("POST /_nitro/tasks/:name accepts payload", async () => {
+          const { data, status } = await callHandler({
+            url: "/_nitro/tasks/test",
+            method: "POST",
+            body: JSON.stringify({ payload: { key: "value" } }),
+          });
+          expect(status).toBe(200);
+          expect(data.result.payload.key).toBe("value");
+        });
+
+        it("GET /_nitro/tasks/:name accepts query params as payload", async () => {
+          const { data, status } = await callHandler({
+            url: "/_nitro/tasks/test?key=value",
+          });
+          expect(status).toBe(200);
+          expect(data.result.payload.key).toBe("value");
+        });
+      });
+
       describe("openAPI", () => {
         let spec: OpenAPI3;
         it("/_openapi.json", async () => {