implement discount on courses/ebooks

iliakan · iliakan · commit d7ca2c434574 · 2015-05-28T17:46:38.000+03:00
diff --git a/fixture/init/payments.js b/fixture/init/payments.js
@@ -3,6 +3,10 @@ require('payments');
 
 exports.Transaction = [];
 exports.TransactionLog = [];
+exports.Discount = [{
+  code: "DISCOUNT",
+  discount: 1
+}];
 exports.Order = [];
 
 exports.OrderTemplate = [
@@ -12,7 +16,7 @@ exports.OrderTemplate = [
     slug:        "js",
     module:      'ebook',
     weight:      1,
-    amount:      1,
+    amount:      2,
     data:        {
       file: "tutorial/js.zip"
     }
@@ -23,7 +27,7 @@ exports.OrderTemplate = [
     slug:        "ui",
     module:      'ebook',
     weight:      2,
-    amount:      1,
+    amount:      2,
     data:        {
       file: "tutorial/ui.zip"
     }
@@ -34,7 +38,7 @@ exports.OrderTemplate = [
     slug:        "js-ui",
     module:      'ebook',
     weight:      3,
-    amount:      1,
+    amount:      2,
     data:        {
       file: "tutorial/js-ui.zip"
     }
diff --git a/handlers/courses/lib/createOrderFromTemplate.js b/handlers/courses/lib/createOrderFromTemplate.js
@@ -1,4 +1,5 @@
 var Order = require('payments').Order;
+var Discount = require('payments').Discount;
 var OrderCreateError = require('payments').OrderCreateError;
 var CourseGroup = require('../models/courseGroup');
 var pluralize = require('textUtil/pluralize');
@@ -42,9 +43,17 @@ module.exports = function*(orderTemplate, user, requestBody) {
     throw new OrderCreateError("Вы не авторизованы.");
   }
 
+
+  var price = group.price;
+  if (requestBody.discountCode) {
+    var discount = yield* Discount.findByCodeAndModule(requestBody.discountCode, 'courses');
+    if (discount) price = discount.apply(price);
+  }
+
+
   var order = new Order({
     title:  group.title,
-    amount: orderData.count * group.price,
+    amount: orderData.count * price,
     module: orderTemplate.module,
     data:   orderData,
     email:  user.email,
diff --git a/handlers/ebook/client/orderForm.js b/handlers/ebook/client/orderForm.js
@@ -14,7 +14,6 @@ class OrderForm {
 
     this.elem.addEventListener('submit', (e) => this.onSubmit(e));
 
-
     this.delegate('[data-order-payment-change]', 'click', function(e) {
       e.preventDefault();
       this.elem.querySelector('[data-order-form-step-payment]').style.display = 'block';
diff --git a/handlers/ebook/lib/createOrderFromTemplate.js b/handlers/ebook/lib/createOrderFromTemplate.js
@@ -1,14 +1,21 @@
 var Order = require('payments').Order;
+var Discount = require('payments').Discount;
 
 // middleware
 // create order from template,
 // use the incoming data if needed
 module.exports = function* (orderTemplate, user, requestBody) {
 
+  var amount = orderTemplate.amount;
+  if (requestBody.discountCode) {
+    var discount = yield* Discount.findByCodeAndModule(requestBody.discountCode, 'ebook');
+    if (discount) amount = discount.apply(amount);
+  }
+
   var order = new Order({
     title:       orderTemplate.title,
     description: orderTemplate.description,
-    amount:      orderTemplate.amount,
+    amount:      amount,
     module:      orderTemplate.module,
     data:        orderTemplate.data
   });
diff --git a/handlers/payments/common/client/formPayment.js b/handlers/payments/common/client/formPayment.js
@@ -76,6 +76,11 @@ class FormPayment {
       orderData[key] = paymentData[key];
     }
 
+    var discountCode = window.location.search.match(/[?&]discountCode=(\w+)/);
+    if (discountCode) {
+      orderData.discountCode = discountCode[1];
+    }
+
     // response status must be 200
     var request = xhr({
       method:         'POST',
diff --git a/handlers/payments/index.js b/handlers/payments/index.js
@@ -9,6 +9,7 @@ exports.loadTransaction = require('./lib/loadTransaction');
 exports.getOrderInfo = require('./lib/getOrderInfo');
 
 var Order = exports.Order = require('./models/order');
+var Discount = exports.Discount = require('./models/discount');
 var OrderTemplate = exports.OrderTemplate = require('./models/orderTemplate');
 var Transaction = exports.Transaction = require('./models/transaction');
 var TransactionLog = exports.TransactionLog = require('./models/transactionLog');
diff --git a/handlers/payments/models/discount.js b/handlers/payments/models/discount.js
@@ -0,0 +1,58 @@
+var mongoose = require('mongoose');
+var Schema = mongoose.Schema;
+var autoIncrement = require('mongoose-auto-increment');
+var OrderTemplate = require('./orderTemplate');
+var Transaction = require('./transaction');
+var _ = require('lodash');
+
+var schema = new Schema({
+  discount:      {
+    type:     Number,
+    required: true
+  },
+
+
+  onlyModule: {
+    type: String
+  },
+
+  code: {
+    type: String,
+    required: true,
+    unique: true,
+    default: function() {
+      return Math.random().toString(36).slice(2, 10);
+    }
+  },
+
+  isActive: {
+    type:    Boolean,
+    default: true
+  },
+
+  created:  {
+    type:    Date,
+    default: Date.now
+  }
+
+});
+
+/**
+ * find active discount with the code
+ * if discount has onlyModule set, ensure the match
+ * @param code
+ * @param onlyModule
+ * @returns {*}
+ */
+schema.statics.findByCodeAndModule = function*(code, onlyModule) {
+  var discount = yield Discount.findOne({code: code, isActive: true}).exec();
+  if (discount.onlyModule && discount.onlyModule != onlyModule) return null;
+  return discount;
+};
+
+schema.methods.apply = function(amount) {
+  return this.discount < 1 ? amount * this.discount : this.discount;
+};
+
+var Discount = module.exports = mongoose.model('Discount', schema);
+
