SpringRole · Santan24 · Jun 10, 2021 · Jun 10, 2021 · Jun 10, 2021 · Jun 10, 2021
diff --git a/app.js b/app.js
@@ -4,11 +4,12 @@ const cors = require("cors");
 const bcryptjs = require("bcryptjs");
 require("dotenv").config();
 require("./db/connectionDB");
+require('./models/user')
+require('./models/Survey')
 const authRoutes = require("./routes/auth");
 const surveyRoutes = require("./routes/survey");
 
 const app = express();
-
 app.use(express.urlencoded({ extended: true }));
 app.use(express.json());
 app.use(cors());

diff --git a/controllers/auth.js b/controllers/auth.js
@@ -3,13 +3,7 @@ const nodemailer = require("nodemailer");
 const bcryptjs = require("bcryptjs");
 const jwt = require("jsonwebtoken");
 
-const transporter = nodemailer.createTransport({
-  service: "gmail",
-  auth: {
-    user: "[email protected]",
-    pass: "*Saheb13*",
-  },
-});
+const {transporter} = require('../utils/transporter')
 
 const signup = async (req, res) => {
   const { username, password, email, phoneNo } = req.body;
@@ -33,7 +27,7 @@ const signup = async (req, res) => {
         from: "[email protected]",
         to: email,
         subject: "Account activation link",
-        html: `<h2>PLease click on given link to activate your account</h2>
+        html: `<h2>Account activation link</h2>
               <p>http://localhost:2000/api/activate/${token}</p>
        `,
       };

diff --git a/controllers/survey.js b/controllers/survey.js
@@ -1,8 +1,46 @@
 const mongoose = require("mongoose");
 const { Survey } = require("../models/survey");
+const {transporter} = require('../utils/transporter')
 
 const createSurvey = async (req, res) => {
-  const survey = await Survey.create(req.body);
+  const { title, category,subject, body, recipients } = req.body;
+  const survey = await Survey.create({
+    title,
+    category,
+    subject,
+    body,
+    recipients,
+    dateSent: Date.now(),
+  });
+
+  const data = {
+    from: "[email protected]",
+    to: recipients,
+    subject: "Account activation link",
+    html: `    <html>
+    <body>
+      <div style="text-align: center;">
+        <h3>I'd like your input!</h3>
+        <p>Please answer the following question:</p>
+        <p>${survey.body}</p>
+        <div>
+          <a href="/">Yes</a>
+        </div>
+        <div>
+          <a href="/">No</a>
+        </div>
+      </div>
+    </body>
+  </html>
+   `,
+  };
+
+  try {
+    await transporter.sendMail(data);
+    console.log("Email sent Successfully! survey");
+  } catch (err) {
+    console.log("error occured while sending email!", err);
+  }
   res.send(survey);
 };
-module.exports = { createSurvey };
+module.exports = { createSurvey };
diff --git a/middlewares/requireCredits.js b/middlewares/requireCredits.js
@@ -0,0 +1,6 @@
+module.exports = (req, res, next) => {
+  if (!req.user.credits <1) {
+    return res.status(403).send({ error: " You do Not have enough Credits!" });
+  }
+  next();
+};
diff --git a/middlewares/requireLogin.js b/middlewares/requireLogin.js
@@ -0,0 +1,8 @@
+module.exports = (req, res, next) => {
+    if (!req.user) {
+      return res.status(401).send({ error: 'You must log in!' });
+    }
+
+    next();
+  };
+
diff --git a/models/recipient.js b/models/recipient.js
@@ -0,0 +1,8 @@
+const mongoose = require("mongoose");
+const { Schema } = mongoose;
+const recipientSchema = new Schema({
+  email: String,
+  responded: { type: Boolean, default: false },
+});
+
+module.exports = recipientSchema;
diff --git a/models/survey.js b/models/survey.js
@@ -1,26 +1,35 @@
-
-const mongoose=require('mongoose')
-
-const surveySchema=mongoose.Schema({
-   title:{
-       type:String
-   },
-   type:{
-       type:String
-   },
-
-   questionsSet:[
-    {
-        ques:{
-            type: String
-        },
-        ans:{
-            type: [Boolean]
-        }
-    }
-]
-
-})
-//created the survey api now 
-const Survey = mongoose.model("Survey",surveySchema)
-module.exports={Survey}
+const mongoose = require("mongoose");
+const recipientSchema= require('./recipient')
+const surveySchema = mongoose.Schema({
+  title: {
+    type: String,
+  },
+  category: {
+    type: String,
+  },
+  body: {
+    type: String,
+  },
+  subject: {
+    type: String,
+  },
+  recipients:[
+       String
+    ],
+  yes: {
+    type: Number,
+    default: 0,
+  },
+  No: {
+    type: Number,
+    default: 0,
+  },
+  _user:{
+    type:mongoose.Schema.Types.ObjectId,
+    ref:'User'
+  },
+   dateSent: Date,
+   lastResponded: Date
+});
+const Survey = mongoose.models.Survey||mongoose.model("Survey", surveySchema);
+module.exports = { Survey };
diff --git a/models/user.js b/models/user.js
@@ -1,10 +1,8 @@
 const mongoose = require("mongoose");
-
 const userSchema = mongoose.Schema({
   username: {
     type: String,
     required: true,
-    max: 64,
   },
   password: {
     type: String,
@@ -30,6 +28,5 @@ const userSchema = mongoose.Schema({
     default: null,
   },
 });
-
-const User = mongoose.model("User", userSchema);
+const User = mongoose.models.User || mongoose.model("User", userSchema);
 module.exports = { User };
diff --git a/routes/survey.js b/routes/survey.js
@@ -1,7 +1,7 @@
 const express = require("express");
-const { createSurvey } = require("../controllers/survey");
 const router = express.Router();
+const { createSurvey } = require("../controllers/survey");
 
 router.post("/createsurvey", createSurvey);
 
-module.exports = router;
+module.exports = router;
diff --git a/utils/transporter.js b/utils/transporter.js
@@ -0,0 +1,13 @@
+const nodemailer = require('nodemailer')
+
+const transporter = nodemailer.createTransport({
+  service: "gmail",
+  auth: {
+    user: "[email protected]",
+    pass: "*Saheb13*",
+  },
+});
+
+module.exports ={
+    transporter
+}