Fix jwt-token

data-retention-service/database.js

@@ -0,0 +1,222 @@
+/**
+ * Database connection and models for Data Retention Service
+ */
+
+const { Sequelize, DataTypes } = require('sequelize');
+
+let sequelize;
+let models = {};
+
+/**
+ * Initialize database connection
+ */
+async function initializeDatabase() {
+  // Database connection
+  sequelize = new Sequelize(
+    process.env.DB_NAME || 'drone_detection',
+    process.env.DB_USER || 'postgres',
+    process.env.DB_PASSWORD || 'password',
+    {
+      host: process.env.DB_HOST || 'localhost',
+      port: process.env.DB_PORT || 5432,
+      dialect: 'postgres',
+      logging: process.env.NODE_ENV === 'development' ? console.log : false,
+      pool: {
+        max: 5,
+        min: 0,
+        acquire: 30000,
+        idle: 10000
+      }
+    }
+  );
+
+  // Test connection
+  await sequelize.authenticate();
+  
+  // Define models
+  defineModels();
+  
+  return sequelize;
+}
+
+/**
+ * Define database models
+ */
+function defineModels() {
+  // Tenant model
+  models.Tenant = sequelize.define('Tenant', {
+    id: {
+      type: DataTypes.INTEGER,
+      primaryKey: true,
+      autoIncrement: true
+    },
+    slug: {
+      type: DataTypes.STRING(50),
+      unique: true,
+      allowNull: false
+    },
+    name: {
+      type: DataTypes.STRING(100),
+      allowNull: false
+    },
+    features: {
+      type: DataTypes.JSONB,
+      defaultValue: {}
+    },
+    is_active: {
+      type: DataTypes.BOOLEAN,
+      defaultValue: true
+    }
+  }, {
+    tableName: 'tenants',
+    timestamps: true,
+    createdAt: 'created_at',
+    updatedAt: 'updated_at'
+  });
+
+  // DroneDetection model
+  models.DroneDetection = sequelize.define('DroneDetection', {
+    id: {
+      type: DataTypes.INTEGER,
+      primaryKey: true,
+      autoIncrement: true
+    },
+    tenant_id: {
+      type: DataTypes.INTEGER,
+      allowNull: false
+    },
+    device_id: {
+      type: DataTypes.STRING(50),
+      allowNull: false
+    },
+    timestamp: {
+      type: DataTypes.DATE,
+      allowNull: false
+    },
+    drone_type: {
+      type: DataTypes.INTEGER,
+      allowNull: true
+    },
+    rssi: {
+      type: DataTypes.FLOAT,
+      allowNull: true
+    },
+    frequency: {
+      type: DataTypes.FLOAT,
+      allowNull: true
+    }
+  }, {
+    tableName: 'drone_detections',
+    timestamps: false,
+    indexes: [
+      {
+        fields: ['tenant_id', 'timestamp']
+      },
+      {
+        fields: ['timestamp']
+      }
+    ]
+  });
+
+  // Heartbeat model
+  models.Heartbeat = sequelize.define('Heartbeat', {
+    id: {
+      type: DataTypes.INTEGER,
+      primaryKey: true,
+      autoIncrement: true
+    },
+    tenant_id: {
+      type: DataTypes.INTEGER,
+      allowNull: false
+    },
+    device_id: {
+      type: DataTypes.STRING(50),
+      allowNull: false
+    },
+    timestamp: {
+      type: DataTypes.DATE,
+      allowNull: false
+    },
+    status: {
+      type: DataTypes.STRING(20),
+      defaultValue: 'online'
+    }
+  }, {
+    tableName: 'heartbeats',
+    timestamps: false,
+    indexes: [
+      {
+        fields: ['tenant_id', 'timestamp']
+      },
+      {
+        fields: ['timestamp']
+      }
+    ]
+  });
+
+  // SecurityLog model (optional, might not exist in all installations)
+  models.SecurityLog = sequelize.define('SecurityLog', {
+    id: {
+      type: DataTypes.INTEGER,
+      primaryKey: true,
+      autoIncrement: true
+    },
+    tenant_id: {
+      type: DataTypes.INTEGER,
+      allowNull: true
+    },
+    timestamp: {
+      type: DataTypes.DATE,
+      allowNull: false
+    },
+    level: {
+      type: DataTypes.STRING(20),
+      allowNull: false
+    },
+    message: {
+      type: DataTypes.TEXT,
+      allowNull: false
+    },
+    metadata: {
+      type: DataTypes.JSONB,
+      defaultValue: {}
+    }
+  }, {
+    tableName: 'security_logs',
+    timestamps: false,
+    indexes: [
+      {
+        fields: ['tenant_id', 'timestamp']
+      },
+      {
+        fields: ['timestamp']
+      }
+    ]
+  });
+}
+
+/**
+ * Get models
+ */
+async function getModels() {
+  if (!sequelize) {
+    await initializeDatabase();
+  }
+  return models;
+}
+
+/**
+ * Close database connection
+ */
+async function closeDatabase() {
+  if (sequelize) {
+    await sequelize.close();
+  }
+}
+
+module.exports = {
+  initializeDatabase,
+  getModels,
+  closeDatabase,
+  sequelize: () => sequelize
+};