Upload 3 files

api.js

@@ -3,6 +3,7 @@ const axios = require('axios');
 const jwt = require('jsonwebtoken');
 const vm = require('vm');
 const { JSDOM } = require('jsdom');
+const logger = require('./logger');
 
 const app = express();
 app.use(express.json());
@@ -27,13 +28,20 @@ async function hsw(req, host) {
 
 
 app.post('/hsw', async (req, res) => {
+    const forwarded = req.headers['x-forwarded-for'];
+    const ip = forwarded ? forwarded.split(/, /)[0] : req.connection.remoteAddress;
     const data = req.body;
     const result = await hsw(data.req, data.host);
     res.send(result);
+    logger.info('', { ip: ip, path: '/hsw', statusCode: res.statusCode });
 });
 
 app.get('/ping', (req, res) => {
-    res.send("pong");
+    const forwarded = req.headers['x-forwarded-for'];
+    const ip = forwarded ? forwarded.split(/, /)[0] : req.connection.remoteAddress;
+    res.json({ ip: ip,status:200 });
+    logger.info('', { ip: ip, path: '/ping', statusCode: res.statusCode });
+
 });
 
 app.listen(5000, '0.0.0.0', () => {

logger.js

@@ -0,0 +1,30 @@
+const winston = require('winston');
+
+const myFormat = winston.format.printf(({ level, message, timestamp, meta }) => {
+  return `[${level.toUpperCase()}] ${timestamp} ${meta.ip} ${meta.path} ${meta.statusCode}`;
+});
+
+const logger = winston.createLogger({
+  level: 'info',
+  format: winston.format.combine(
+    winston.format.timestamp(),
+    myFormat
+  ),
+  defaultMeta: { service: 'user-service' },
+  transports: [
+    new winston.transports.File({ filename: 'error.log', level: 'error' }),
+    new winston.transports.File({ filename: 'combined.log' }),
+  ],
+});
+
+if (process.env.NODE_ENV !== 'production') {
+  logger.add(new winston.transports.Console({
+    format: winston.format.combine(
+      winston.format.colorize(),
+      winston.format.timestamp(),
+      myFormat
+    ),
+  }));
+}
+
+module.exports = logger;

package.json

@@ -14,7 +14,8 @@
     "express": "",
     "jsonwebtoken": "",
     "jsdom": "",
-    "vm": ""
+    "vm": "",
+    "winston": ""
   },
   "private": true
 }