feat: sqs + direct mode

Author: solaris007

README.md

@@ -27,7 +27,9 @@ SpaceCat Task Processor is a Node.js service that processes messages from the AW
 
 ## Usage
 - The service is designed to run as a serverless function or background worker.
-- It listens for messages on the SQS queue and processes them automatically.
+- It can be invoked in two ways:
+  - **SQS mode:** listens to the `SPACECAT-TASK-PROCESSOR-JOBS` queue and processes messages automatically (default path for existing workflows).
+  - **Direct mode:** the Lambda entrypoint auto-detects single-message payloads (e.g., from AWS Step Functions) and executes the corresponding handler synchronously. This is used by the new agent workflows to obtain immediate results before triggering follow-up actions.
 
 ## Development
 - To run tests:

src/index.js

@@ -54,7 +54,7 @@ function getElapsedSeconds(startTime) {
  * @param {UniversalContext} context the context of the universal serverless function
  * @returns {Response} a response
  */
-async function run(message, context) {
+async function processTask(message, context) {
   const { log } = context;
   const { type, siteId } = message;
 
@@ -80,9 +80,30 @@ async function run(message, context) {
   }
 }
 
-export const main = wrap(run)
+const runSQS = wrap(processTask)
   .with(dataAccess)
   .with(sqsEventAdapter)
   .with(imsClientWrapper)
   .with(secrets, { name: getSecretName })
   .with(helixStatus);
+
+const runDirect = wrap(processTask)
+  .with(dataAccess)
+  .with(imsClientWrapper)
+  .with(secrets, { name: getSecretName })
+  .with(helixStatus);
+
+function isSqsEvent(event, context) {
+  if (Array.isArray(event?.Records)) {
+    return true;
+  }
+  if (Array.isArray(context?.invocation?.event?.Records)) {
+    return true;
+  }
+  return typeof event?.type !== 'string';
+}
+
+export const main = async (event, context) => {
+  const handler = isSqsEvent(event, context) ? runSQS : runDirect;
+  return handler(event, context);
+};

test/index.test.js

@@ -180,4 +180,19 @@ describe('Index Tests', () => {
     expect(resp.status).to.equal(500); // Should return internal server error
     expect(context.log.error.calledWithMatch(sinon.match('demo-url-processor task for test-site failed after'))).to.be.true;
   });
+
+  it('processes direct invocation events without SQS adapter', async () => {
+    const directContext = {
+      ...context,
+      invocation: undefined,
+    };
+    const directEvent = {
+      type: 'dummy',
+      siteId: 'direct-site',
+    };
+
+    const resp = await main(directEvent, directContext);
+    expect(resp.status).to.equal(200);
+    expect(directContext.log.info.calledWith('Found task handler for type: dummy')).to.be.true;
+  });
 });