add ai function

.gitignore

@@ -1,2 +1,3 @@
 node_modules
-dbdata
+dbdata
+.env

backend/package.json

@@ -11,9 +11,11 @@
   "license": "ISC",
   "description": "API for a simple notes web app",
   "dependencies": {
+    "@google/generative-ai": "^0.24.1",
     "bcrypt": "^6.0.0",
     "cookie-parser": "^1.4.7",
     "cors": "^2.8.5",
+    "dotenv": "^17.2.1",
     "express": "^5.1.0",
     "jsonwebtoken": "^9.0.2",
     "passport": "^0.7.0",

backend/src/app.js

@@ -1,3 +1,4 @@
+import 'dotenv/config';
 import express from 'express';
 import cors from 'cors';
 import { notesRouter } from './routes/notesRouter.js';
@@ -12,30 +13,38 @@ import { addLogger } from './config/logger.config.js';
 import { sequelize } from './config/database.js';
 import './models/models.js';
 import { createAdminUserIfNotExists } from './utils.js';
+import { aiRouter } from './routes/aiRouter.js';
 
 const PORT = process.env.PORT || 3000;
 
 const app = express();
 
+const allowedOrigins = [`http://localhost:${PORT}`, 'http://localhost:3001', 'http://localhost:3004', 'http://localhost:8080'];
+
+console.log("CORS is configured for these origins1:", allowedOrigins);
+
+app.use(cors({
+    origin: allowedOrigins,
+    credentials: true
+}));
+
+
+
 app.use(express.json());
 app.use(express.urlencoded({ extended: true }));
 app.use(cookieParser());
 initializePassport();
 app.use(passport.initialize());
 
-const allowedOrigins = [`http://localhost:${PORT}`, `http://localhost:3001`];
 
-app.use(cors({
-    origin: allowedOrigins,
-    credentials: true
-}));
 
 
 app.use(addLogger);
 app.use('/api/notes', notesRouter.getRouter());
 app.use('/api/categories', categoriesRouter.getRouter());
 app.use('/api/sessions', sessionsRouter.getRouter());
 app.use('/api/docs', swaggerUiExpress.serve, swaggerUiExpress.setup(docsSpecs, { swaggerOptions: { withCredentials: true } }));
+app.use('/api/ai', aiRouter.getRouter());
 
 sequelize.authenticate()
     .then(async () => {
@@ -49,4 +58,5 @@ sequelize.authenticate()
     .catch((error) => {
         console.error('Unable to connect to the database:', error);
         process.exit(1);
-    });
+    });
+

backend/src/controllers/ai.controller.js

@@ -0,0 +1,21 @@
+import { aiService } from '../services/ai.service.js';
+
+class AIController {
+    async generate(req, res) {
+        const { prompt } = req.body;
+
+        if (!prompt) {
+            return res.sendBadRequestError("'prompt' is required in the request body.");
+        }
+
+        try {
+            const generatedText = await aiService.generateContent(prompt);
+            res.sendOk({ content: generatedText });
+        } catch (error) {
+            req.logger.error(`AI content generation failed: ${error.message}`);
+            res.sendInternalServerError("Error generating content with AI.");
+        }
+    }
+}
+
+export const aiController = new AIController();

backend/src/routes/aiRouter.js

@@ -0,0 +1,11 @@
+import Router from './router.js';
+import { aiController } from '../controllers/ai.controller.js';
+
+class AIRouter extends Router {
+    init() {
+        // Only authenticated users can access the AI feature.
+        this.post('/generate', ["USER"], async (req, res) => this.controller.generate(req, res));
+    }
+}
+
+export const aiRouter = new AIRouter(aiController);