|
| 1 | +// src/ODNC/server.js |
| 2 | +const express = require('express'); |
| 3 | +const bodyParser = require('body-parser'); |
| 4 | +const cors = require('cors'); |
| 5 | +const mongoose = require('mongoose'); |
| 6 | +const rateLimit = require('express-rate-limit'); |
| 7 | +const { predictResourceExtraction } = require('./aiModel'); // Placeholder for AI model |
| 8 | +require('dotenv').config(); // Load environment variables |
| 9 | + |
| 10 | +const app = express(); |
| 11 | +const PORT = process.env.PORT || 5000; |
| 12 | + |
| 13 | +// Middleware |
| 14 | +app.use(cors()); |
| 15 | +app.use(bodyParser.json()); |
| 16 | + |
| 17 | +// Rate limiting middleware |
| 18 | +const limiter = rateLimit({ |
| 19 | + windowMs: 1 * 60 * 1000, // 1 minute |
| 20 | + max: 100, // Limit each IP to 100 requests per windowMs |
| 21 | + message: 'Too many requests, please try again later.', |
| 22 | +}); |
| 23 | +app.use(limiter); |
| 24 | + |
| 25 | +// MongoDB connection |
| 26 | +mongoose.connect(process.env.MONGODB_URI, { useNewUrlParser: true, useUnifiedTopology: true }) |
| 27 | + .then(() => console.log('MongoDB connected')) |
| 28 | + .catch(err => console.error('MongoDB connection error:', err)); |
| 29 | + |
| 30 | +// Energy data schema |
| 31 | +const energyDataSchema = new mongoose.Schema({ |
| 32 | + energyType: String, |
| 33 | + amount: Number, |
| 34 | + convertedAmount: Number, |
| 35 | + createdAt: { type: Date, default: Date.now }, |
| 36 | +}); |
| 37 | + |
| 38 | +const EnergyData = mongoose.model('EnergyData', energyDataSchema); |
| 39 | + |
| 40 | +// Endpoint to identify and convert energy |
| 41 | +app.post('/api/convert-energy', async (req, res) => { |
| 42 | + const { energyType, amount } = req.body; |
| 43 | + const conversionResult = convertEnergy(energyType, amount); |
| 44 | + |
| 45 | + if (conversionResult) { |
| 46 | + // Save to database |
| 47 | + const energyEntry = new EnergyData({ energyType, amount, convertedAmount: conversionResult }); |
| 48 | + await energyEntry.save(); |
| 49 | + |
| 50 | + res.json({ success: true, result: conversionResult }); |
| 51 | + } else { |
| 52 | + res.status(400).json({ success: false, message: 'Invalid energy type or conversion failed.' }); |
| 53 | + } |
| 54 | +}); |
| 55 | + |
| 56 | +// Function to convert energy types |
| 57 | +const convertEnergy = (energyType, amount) => { |
| 58 | + // Example conversion logic |
| 59 | + const conversionRates = { |
| 60 | + 'dimensional': 1.5, |
| 61 | + 'quantum': 2.0, |
| 62 | + 'dark': 0.5, |
| 63 | + }; |
| 64 | + |
| 65 | + if (conversionRates[energyType]) { |
| 66 | + return amount * conversionRates[energyType]; |
| 67 | + } |
| 68 | + return null; |
| 69 | +}; |
| 70 | + |
| 71 | +// Endpoint to predict resource extraction |
| 72 | +app.post('/api/predict-extraction', async (req, res) => { |
| 73 | + const { energyData } = req.body; |
| 74 | + try { |
| 75 | + const prediction = await predictResourceExtraction(energyData); |
| 76 | + res.json({ success: true, prediction }); |
| 77 | + } catch (error) { |
| 78 | + console.error('Prediction error:', error); |
| 79 | + res.status(500).json({ success: false, message: 'Prediction failed.' }); |
| 80 | + } |
| 81 | +}); |
| 82 | + |
| 83 | +// Start the server |
| 84 | +app.listen(PORT, () => { |
| 85 | + console.log(`Server is running on port ${PORT}`); |
| 86 | +}); |
0 commit comments