@gugananuvem/aws-local-simulator 1.0.8 → 1.0.10

package/src/services/s3/server.js

@@ -0,0 +1,239 @@
+/**
+ * S3 Server - Servidor HTTP para S3
+ */
+
+const express = require('express');
+const S3Simulator = require('./simulator');
+const logger = require('../../utils/logger');
+
+class S3Server {
+  constructor(port, config) {
+    this.port = port;
+    this.config = config;
+    this.app = express();
+    this.simulator = null;
+    this.server = null;
+    this.setupMiddlewares();
+  }
+
+  setupMiddlewares() {
+    this.app.use(express.json({ limit: '100mb' }));
+    this.app.use(express.raw({ type: 'application/octet-stream', limit: '100mb' }));
+    this.app.use(express.text({ limit: '100mb' }));
+    
+    // Logging de requisições
+    if (logger.currentLogLevel === 'verboso') {
+      this.app.use((req, res, next) => {
+        const start = Date.now();
+        res.on('finish', () => {
+          const duration = Date.now() - start;
+          logger.verboso(`S3: ${req.method} ${req.path} - ${duration}ms`);
+        });
+        next();
+      });
+    }
+  }
+
+  async initialize() {
+    this.simulator = new S3Simulator(this.config);
+    await this.simulator.initialize();
+    this.setupRoutes();
+  }
+
+  setupRoutes() {
+    // Listar buckets
+    this.app.get('/', (req, res) => {
+      const buckets = this.simulator.listBuckets();
+      res.set('Content-Type', 'application/xml');
+      res.send(this.simulator.generateListBucketsResponse(buckets));
+    });
+    
+    // Criar bucket
+    this.app.put('/:bucket', (req, res) => {
+      const result = this.simulator.createBucket(req.params.bucket);
+      if (result.error) {
+        res.status(result.status).send(this.simulator.generateErrorResponse(result.error.code, result.error.message));
+      } else {
+        res.status(200).send();
+      }
+    });
+    
+    // Deletar bucket
+    this.app.delete('/:bucket', (req, res) => {
+      const result = this.simulator.deleteBucket(req.params.bucket);
+      if (result.error) {
+        res.status(result.status).send(this.simulator.generateErrorResponse(result.error.code, result.error.message));
+      } else {
+        res.status(204).send();
+      }
+    });
+    
+    // Head bucket
+    this.app.head('/:bucket', (req, res) => {
+      const bucket = this.simulator.getBucket(req.params.bucket);
+      if (!bucket) {
+        res.status(404).send();
+      } else {
+        res.status(200).send();
+      }
+    });
+    
+    // Upload object
+    this.app.put('/:bucket/*', (req, res) => {
+      const bucket = req.params.bucket;
+      const key = req.params[0];
+      const result = this.simulator.putObject(bucket, key, req.body, req.headers);
+      
+      if (result.error) {
+        res.status(result.status).send(this.simulator.generateErrorResponse(result.error.code, result.error.message));
+      } else {
+        res.set('ETag', `"${result.etag}"`);
+        res.status(200).send();
+      }
+    });
+    
+    // Get object
+    this.app.get('/:bucket/*', (req, res) => {
+      const bucket = req.params.bucket;
+      const key = req.params[0];
+      const result = this.simulator.getObject(bucket, key, req.headers);
+      
+      if (result.error) {
+        res.status(result.status).send(this.simulator.generateErrorResponse(result.error.code, result.error.message));
+      } else {
+        res.set('ETag', `"${result.etag}"`);
+        res.set('Last-Modified', result.lastModified);
+        res.set('Content-Type', result.contentType);
+        res.set('Content-Length', result.size);
+        
+        if (result.metadata) {
+          Object.entries(result.metadata).forEach(([k, v]) => {
+            res.set(`x-amz-meta-${k}`, v);
+          });
+        }
+        
+        res.send(result.content);
+      }
+    });
+    
+    // Head object
+    this.app.head('/:bucket/*', (req, res) => {
+      const bucket = req.params.bucket;
+      const key = req.params[0];
+      const result = this.simulator.headObject(bucket, key);
+      
+      if (result.error) {
+        res.status(result.status).send();
+      } else {
+        res.set('ETag', `"${result.etag}"`);
+        res.set('Last-Modified', result.lastModified);
+        res.set('Content-Type', result.contentType);
+        res.set('Content-Length', result.size);
+        res.status(200).send();
+      }
+    });
+    
+    // Delete object
+    this.app.delete('/:bucket/*', (req, res) => {
+      const bucket = req.params.bucket;
+      const key = req.params[0];
+      const result = this.simulator.deleteObject(bucket, key);
+      
+      if (result.error) {
+        res.status(result.status).send(this.simulator.generateErrorResponse(result.error.code, result.error.message));
+      } else {
+        res.status(204).send();
+      }
+    });
+    
+    // List objects
+    this.app.get('/:bucket', (req, res) => {
+      const bucket = req.params.bucket;
+      const prefix = req.query.prefix || '';
+      const delimiter = req.query.delimiter;
+      const maxKeys = parseInt(req.query['max-keys']) || 1000;
+      const listType = req.query['list-type'];
+      
+      const result = this.simulator.listObjects(bucket, { prefix, delimiter, maxKeys });
+      
+      if (result.error) {
+        res.status(result.status).send(this.simulator.generateErrorResponse(result.error.code, result.error.message));
+      } else {
+        res.set('Content-Type', 'application/xml');
+        if (listType === '2') {
+          res.send(this.simulator.generateListObjectsV2Response(result));
+        } else {
+          res.send(this.simulator.generateListObjectsResponse(result));
+        }
+      }
+    });
+    
+    // Admin endpoints
+    this.setupAdminRoutes();
+  }
+
+  setupAdminRoutes() {
+    this.app.get('/__admin/buckets', (req, res) => {
+      res.json(this.simulator.getBucketsInfo());
+    });
+    
+    this.app.get('/__admin/buckets/:bucket', (req, res) => {
+      const info = this.simulator.getBucketInfo(req.params.bucket);
+      if (info.error) {
+        res.status(404).json(info.error);
+      } else {
+        res.json(info);
+      }
+    });
+    
+    this.app.get('/__admin/buckets/:bucket/objects', (req, res) => {
+      const objects = this.simulator.listAllObjects(req.params.bucket);
+      res.json(objects);
+    });
+    
+    this.app.delete('/__admin/buckets/:bucket', (req, res) => {
+      this.simulator.clearBucket(req.params.bucket);
+      res.json({ message: `Bucket ${req.params.bucket} cleared` });
+    });
+    
+    this.app.delete('/__admin/buckets/:bucket/objects/:key', (req, res) => {
+      this.simulator.deleteObject(req.params.bucket, req.params.key);
+      res.json({ message: 'Object deleted' });
+    });
+    
+    this.app.get('/__admin/stats', (req, res) => {
+      res.json(this.simulator.getStats());
+    });
+  }
+
+  start() {
+    return new Promise((resolve) => {
+      this.server = this.app.listen(this.port, () => {
+        logger.info(`🗄️  S3 rodando em http://localhost:${this.port}`);
+        resolve();
+      });
+    });
+  }
+
+  stop() {
+    return new Promise((resolve) => {
+      if (this.server) {
+        this.server.close(() => resolve());
+      } else {
+        resolve();
+      }
+    });
+  }
+
+  getStatus() {
+    return {
+      running: !!this.server,
+      port: this.port,
+      endpoint: `http://localhost:${this.port}`,
+      bucketsCount: this.simulator?.getBucketsCount() || 0,
+      objectsCount: this.simulator?.getTotalObjectsCount() || 0
+    };
+  }
+}
+
+module.exports = S3Server;