feat: introduce AMS support for preflight (#1459)

Author: dzehnder

src/controllers/preflight.js

@@ -90,6 +90,7 @@ function PreflightController(ctx, log, env) {
    * @param {Object} context.data - The request data
    * @param {string[]} context.data.urls - Array of URLs to process
    * @param {string} context.data.step - The audit step
+   * @param {string} context.data.siteId - The siteId, if it's an AMS site
    * @returns {Promise<Object>} The HTTP response object
    */
   const createPreflightJob = async (context) => {
@@ -105,10 +106,20 @@ function PreflightController(ctx, log, env) {
     try {
       const isDev = env.AWS_ENV === 'dev';
       const step = data.step.toLowerCase();
-
+      let site;
       const url = new URL(data.urls[0]);
       const previewBaseURL = `${url.protocol}//${url.hostname}`;
-      const site = await dataAccess.Site.findByPreviewURL(previewBaseURL);
+      if (url.hostname.endsWith('adobecqms.net')) {
+        // AMS
+        site = await dataAccess.Site.findById(data.siteId);
+      } else {
+        // EDS and CS
+        site = await dataAccess.Site.findByPreviewURL(previewBaseURL);
+      }
+
+      if (!site) {
+        throw new Error(`No site found for preview URL: ${previewBaseURL}`);
+      }
       let enableAuthentication = false;
       // check head request for preview url
       const headResponse = await fetch(`${previewBaseURL}`, {
@@ -121,10 +132,6 @@ function PreflightController(ctx, log, env) {
         enableAuthentication = true;
       }
 
-      if (!site) {
-        throw new Error(`No site found for preview URL: ${previewBaseURL}`);
-      }
-
       let promiseTokenResponse;
       if (CS_TYPES.includes(site.getAuthoringType())) {
         try {

test/controllers/preflight.test.js

@@ -186,6 +186,51 @@ describe('Preflight Controller', () => {
       );
     });
 
+    it('creates a preflight job successfully for AMS', async () => {
+      const context = {
+        data: {
+          urls: ['http://author.adobecqms.net/path'],
+          step: 'identify',
+          siteId: 'test-site-123',
+        },
+      };
+      mockDataAccess.Site.findById = sandbox.stub().resolves(mockSite);
+
+      const response = await preflightController.createPreflightJob(context);
+      expect(response.status).to.equal(202);
+
+      const result = await response.json();
+      expect(result).to.deep.equal({
+        jobId,
+        status: 'IN_PROGRESS',
+        createdAt: '2024-03-20T10:00:00Z',
+        pollUrl: `https://spacecat.experiencecloud.live/api/v1/preflight/jobs/${jobId}`,
+      });
+
+      expect(mockDataAccess.AsyncJob.create).to.have.been.calledWith({
+        status: 'IN_PROGRESS',
+        metadata: {
+          payload: {
+            siteId: 'test-site-123',
+            urls: ['http://author.adobecqms.net/path'],
+            step: 'identify',
+            enableAuthentication: true,
+          },
+          jobType: 'preflight',
+          tags: ['preflight'],
+        },
+      });
+
+      expect(mockSqs.sendMessage).to.have.been.calledWith(
+        'https://sqs.test.amazonaws.com/audit-queue',
+        {
+          jobId,
+          type: 'preflight',
+          siteId: 'test-site-123',
+        },
+      );
+    });
+
     it('creates a preflight job successfully in production environment with authentication enabled', async () => {
       if (fetchStub && fetchStub.restore) {
         fetchStub.restore();