vikingeducation · KaraThrash · Aug 16, 2017 · Aug 16, 2017 · Aug 16, 2017 · Aug 16, 2017
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+node_modules
diff --git a/README.md b/README.md
@@ -1,2 +1,5 @@
+Greg and Dan
+
+
 # Ponz.io
 Building Ponz.io, with its endearingly upside-down-triangle-shaped business model.
diff --git a/index.html b/index.html
@@ -0,0 +1,14 @@
+<!doctype html>
+<html>
+
+
+  <script src="https://code.jquery.com/jquery-1.11.1.min.js"></script>
+  <link rel="stylesheet" href="styles.css">
+
+
+<div id="trapezoid1"> <p>1</p></div>
+<div id="trapezoid2"><p>1</p> </div>
+<div id="trapezoid3"> <p>1</p></div>
+<div id="trapezoid4"> <p>1</p></div>
+<div id="trapezoid5"> <p>1</p></div>
+</html>
diff --git a/index.js b/index.js
@@ -0,0 +1,119 @@
+const express = require("express");
+const app = express();
+const cookieParser = require("cookie-parser");
+const bodyParser = require("body-parser");
+const expressSession = require("express-session");
+const exphbs = require("express-handlebars");
+const shop = require("./routers/shop");
+const pyramid = require("./pyramid");
+const {
+  createSignedSessionId,
+  loginMiddleware,
+  loggedInOnly,
+  loggedOutOnly
+} = require("./session/Session.js");
+
+var flash = require("express-flash-messages");
+app.use(flash());
+var payOut = require("./moneyLogic");
+app.use(cookieParser());
+app.use(bodyParser.urlencoded({ extended: false }));
+app.use(express.static(`${__dirname}/public`));
+app.use(
+  expressSession({
+    secret: process.env.secret || "keyboard cat",
+    saveUninitialized: false,
+    resave: false
+  })
+);
+
+app.engine(
+  "handlebars",
+  exphbs({ defaultLayout: "main", partialsDir: "views/partials" })
+);
+app.set("view engine", "handlebars");
+app.use(loginMiddleware);
+const User = require("./models/User");
+const mongoose = require("mongoose");
+mongoose.connect("mongodb://localhost/project_ponz");
+
+app.use("/shop", shop);
+
+app.get("/", loggedInOnly, (req, res) => {
+  console.log(req.user);
+  User.findOne({ username: req.user.username }).then(user => {
+    pyramid(user._id).then(results => {
+      console.log(JSON.stringify(results, null, 2));
+      return res.render("index2", { results, user });
+    });
+  });
+});
+app.get("/pyramid", loggedInOnly, (req, res) => {
+  console.log(req.user);
+  User.findOne({ username: req.user.username }).then(user => {
+    pyramid(user._id).then(results => {
+      console.log(JSON.stringify(results, null, 2));
+      return res.render("index", { results, user });
+    });
+  });
+});
+app.get("/login", loggedOutOnly, (req, res) => {
+  res.render("login");
+});
+app.get("/money", loggedInOnly, (req, res) => {
+  payOut(req.user.referrerId);
+
+  // res.render("login");
+});
+app.get("/logout", loggedInOnly, (req, res) => {
+  res.cookie("sessionId", "");
+  res.redirect("/");
+});
+
+app.get("/register/:id", (req, res) => {
+  res.render("register", { id: req.params.id });
+});
+
+app.post("/register/:id", loggedOutOnly, (req, res) => {
+  User.findOne({ username: req.body.username }).then(foundUser => {
+    if (foundUser === null) {
+      let newUser = {};
+      newUser.username = req.body.username;
+      newUser.password = req.body.password;
+      newUser.referrals = [];
+      newUser.AnkhMorporkDollars = -100;
+      if (req.params.id !== "new") {
+        newUser.referrerId = req.params.id;
+      } else {
+        newUser.referrerId = 1;
+      }
+      User.create(newUser).then(newUser => {
+        const sessionId = createSignedSessionId(newUser.username);
+        res.cookie("sessionId", sessionId);
+        payOut(newUser.referrerId, newUser.id);
+        return res.redirect("/");
+      });
+    } else {
+      return res.redirect("/login");
+    }
+  });
+});
+
+app.post("/login", loggedOutOnly, (req, res) => {
+  User.findOne({ username: req.body.username }).then(foundUser => {
+    if (foundUser === null) {
+      return res.redirect("/login");
+    }
+    if (foundUser.validatePassword(req.body.password)) {
+      const sessionId = createSignedSessionId(foundUser.username);
+      res.cookie("sessionId", sessionId);
+      return res.redirect("/");
+    } else {
+      return res.redirect("/login");
+    }
+  });
+});
+
+app.listen(3000, "0.0.0.0", (req, res) => {
+  console.log("listening on port 3000");
+});
diff --git a/models/User.js b/models/User.js
@@ -0,0 +1,34 @@
+const mongoose = require("mongoose");
+var bluebird = require("bluebird");
+mongoose.Promise = bluebird;
+const Schema = mongoose.Schema;
+const bcrypt = require("bcrypt");
+const uniqueValidator = require("mongoose-unique-validator");
+
+const UserSchema = new Schema(
+  {
+    username: { type: String, unique: true, required: true },
+    passwordHash: { type: String, require: true },
+    referrerId: String,
+    referrals: [{ type: Schema.Types.ObjectId, ref: "User" }],
+    AnkhMorporkDollars: Number,
+    goodies: []
+  },
+  {
+    timestamps: true
+  }
+);
+
+UserSchema.plugin(uniqueValidator);
+
+UserSchema.virtual("password").set(function(value) {
+  this.passwordHash = bcrypt.hashSync(value, 2);
+});
+
+UserSchema.methods.validatePassword = function(password) {
+  return bcrypt.compareSync(password, this.passwordHash);
+};
+
+const User = mongoose.model("User", UserSchema);
+
+module.exports = User;
diff --git a/moneyLogic.js b/moneyLogic.js
@@ -0,0 +1,27 @@
+const User = require("./models/User");
+var payOut = async function(referrerId, referralId) {
+  try {
+    var zref = await User.update(
+      { _id: referrerId },
+      { $push: { referrals: referralId } }
+    );
+  } catch (e) {}
+  let currentPayout = 40;
+  while (referrerId != 1) {
+    let foundReferrer = await User.findOne({ _id: referrerId });
+    let newTotal = foundReferrer.AnkhMorporkDollars + currentPayout;
+
+    try {
+      var z = await User.update(
+        { _id: referrerId },
+        { AnkhMorporkDollars: newTotal }
+      );
+    } catch (e) {}
+    referrerId = foundReferrer.referrerId;
+    currentPayout *= 0.5;
+    if (currentPayout < 1) {
+      currentPayout = 1;
+    }
+  }
+};
+module.exports = payOut;