@gugananuvem/aws-local-simulator 1.0.34 → 1.0.39

package/src/services/sns/server.js

@@ -1,3 +1,4 @@
+const { setupServiceMiddleware } = require('../../middleware');
 /**
  * @fileoverview SNS HTTP Server — Query Protocol completo
  * Compatível com AWS SDK v3 SNS Client
@@ -45,6 +46,14 @@ function createSNSServer(simulator, config, logger) {
     next();
   });
 
+  // ── IAM enforcement (applied conditionally via injectDependencies) ──
+  app.use((req, res, next) => {
+    if (app.iamMiddleware && !req.path.startsWith('/__admin')) {
+      return app.iamMiddleware(req, res, next);
+    }
+    next();
+  });
+
   // ── Endpoint principal (Query Protocol) ─────────────────────
   app.post('/', async (req, res) => {
     const body   = req.body || {};

package/src/services/sqs/index.js

@@ -94,6 +94,44 @@ class SQSService {
   getServer() {
     return this.server;
   }
+
+  injectDependencies(server) {
+    const lambdaService = server.getService('lambda');
+    if (lambdaService) {
+      this.lambdaService = lambdaService;
+      if (this.simulator) {
+        this.simulator.setLambdaService(lambdaService);
+        this.simulator.reattachLambdaHandlers();
+      }
+      if (this.server) {
+        this.server.lambdaService = lambdaService;
+      }
+    }
+
+    const iamService = server.getService('iam');
+    if (iamService && iamService.getSimulator()) {
+      const { createIAMEnforcementMiddleware } = require('../iam/middleware');
+      const iamSimulator = iamService.getSimulator();
+
+      const middleware = createIAMEnforcementMiddleware({
+        iamSimulator,
+        servicePrefix: 'sqs',
+        extractAction: (req) => req.body?.Action || req.query?.Action || 'Unknown',
+        extractResource: (req) => {
+          const queueUrl = req.body?.QueueUrl || req.query?.QueueUrl || '';
+          const queueName = queueUrl.split('/').pop() || req.body?.QueueName || 'unknown';
+          return `arn:aws:sqs:us-east-1:123456789012:${queueName}`;
+        },
+        formatError: (res, code, message) => {
+          res.status(403).set('Content-Type', 'text/xml').send(
+            `<ErrorResponse><Error><Type>Sender</Type><Code>${code}</Code><Message>${message}</Message></Error></ErrorResponse>`
+          );
+        }
+      });
+
+      this.server.iamMiddleware = middleware;
+    }
+  }
 }
 
 module.exports = SQSService;

package/src/services/sqs/server.js

@@ -1,3 +1,4 @@
+const { setupServiceMiddleware } = require('../../middleware');
 /**
  * SQS Server - Servidor HTTP para SQS
  */
@@ -9,7 +10,9 @@ const SQSSimulator = require('./simulator');
 const logger = require('../../utils/logger');
 
 class SQSServer {
-  constructor(port, config, lambdaService) {
+  constructor(port, config, lambdaService) {
+    this.organizationsService = null;
+    this.stsService = null;
     this.port = port;
     this.config = config;
     this.lambdaService = lambdaService;
@@ -67,6 +70,14 @@ class SQSServer {
       next();
     });
 
+    // IAM enforcement (applied conditionally via injectDependencies)
+    this.app.use((req, res, next) => {
+      if (this.iamMiddleware && !req.path.startsWith('/__admin')) {
+        return this.iamMiddleware(req, res, next);
+      }
+      next();
+    });
+
     this.app.post('/', (req, res) => {
       const action = req.query.Action || req.body.Action ||
         (req.headers['x-amz-target'] && req.headers['x-amz-target'].split('.')[1]);
@@ -161,13 +172,16 @@ class SQSServer {
             MessageId: m.MessageId,
             ReceiptHandle: m.ReceiptHandle,
             MD5OfBody: m.MD5OfBody,
-            Body: m.Body
+            Body: m.Body,
+            Attributes: m.Attributes
           }))
         };
       case 'DeleteMessage':
         return {};
       case 'GetQueueUrl':
         return { QueueUrl: result.queueUrl };
+      case 'GetQueueAttributes':
+        return { Attributes: result.attributes };
       case 'ListQueues':
         return { QueueUrls: (result.queues || []).map(q => q.url) };
       default:
@@ -189,8 +203,9 @@ class SQSServer {
         return this.generateDeleteMessageResponse();
       case 'SetQueueAttributes':
         return this.generateSetQueueAttributesResponse();
+      case 'GetQueueAttributes':
+        return this.generateGetQueueAttributesResponse(result.attributes);
       case 'GetQueueUrl':
-
         return this.generateGetQueueUrlResponse(result.queueUrl);
       case 'ListQueues':
         return this.generateListQueuesResponse(result.queues);
@@ -206,6 +221,29 @@ class SQSServer {
 </SetQueueAttributesResponse>`;
   }
 
+  generateGetQueueAttributesResponse(attributes) {
+    let xml = `<?xml version="1.0" encoding="UTF-8"?>
+<GetQueueAttributesResponse xmlns="http://queue.amazonaws.com/doc/2012-11-05/">
+  <GetQueueAttributesResult>
+    <Attribute>`;
+    
+    for (const [key, value] of Object.entries(attributes)) {
+      xml += `
+      <Name>${key}</Name>
+      <Value>${this.escapeXml(value)}</Value>
+    </Attribute>
+    <Attribute>`;
+    }
+    
+    xml = xml.slice(0, -13);  // Remove last <Attribute>
+    
+    xml += `
+  </GetQueueAttributesResult>
+  <ResponseMetadata><RequestId>${crypto.randomUUID()}</RequestId></ResponseMetadata>
+</GetQueueAttributesResponse>`;
+    
+    return xml;
+  }
 
   generateCreateQueueResponse(queueUrl) {
     return `<?xml version="1.0" encoding="UTF-8"?>
@@ -270,7 +308,22 @@ class SQSServer {
       <MessageId>${msg.MessageId}</MessageId>
       <ReceiptHandle>${msg.ReceiptHandle}</ReceiptHandle>
       <MD5OfBody>${msg.MD5OfBody}</MD5OfBody>
-      <Body>${this.escapeXml(msg.Body)}</Body>
+      <Body>${this.escapeXml(msg.Body)}</Body>`;
+      
+      // Add message attributes if present
+      if (msg.Attributes && Object.keys(msg.Attributes).length > 0) {
+        for (const [attrKey, attrValue] of Object.entries(msg.Attributes)) {
+          if (attrValue !== null && attrValue !== undefined) {
+            xml += `
+      <Attribute>
+        <Name>${attrKey}</Name>
+        <Value>${this.escapeXml(String(attrValue))}</Value>
+      </Attribute>`;
+          }
+        }
+      }
+      
+      xml += `
     </Message>`;
     }