modify webhook to support custom plans

1 files changed, 18 insertions, 6 deletions

diff --git a/server/lib/webhook/webhook.js b/server/lib/webhook/webhook.js
index 58a13ca..bd51dac 100644
--- a/server/lib/webhook/webhook.js
+++ b/server/lib/webhook/webhook.js
@@ -26,10 +26,11 @@ var xml_bodyparser = require('express-xml-bodyparser');
 var parser = new xml2js.Parser();
 
 var subscribe = module.exports = {
-  plan_level: {
+  plan_levels: {
     free: 0,
     basic: 1,
     pro: 2,
+    custom: 3,
   },
   
   callbacks: {
@@ -62,19 +63,30 @@ var subscribe = module.exports = {
       Subscription.findOne({ "uuid": uuid }, function(err, old_subscriber){
         // if (err) return;
        
-        var plan = subscrip.plan[0].plan_code[0].split("-")
-        var plan_type = plan[0]
-        var plan_period = plan[1]
+        var plan, plan_type, plan_code
+        var plan_code = subscrip.plan[0].plan_code[0]
         
+        if (plan_code.indexOf("-") !== -1) {
+          plan = plan_code.split("-")
+          plan_type = plan[0]
+          plan_period = plan[1]
+        }
+        else {
+          plan_type = "custom"
+          plan_period = "monthly"
+        }
+        
+        user.plan_code = plan_code
         user.plan_type = plan_type
-        user.plan_level = subscribe.plan_level[plan_type]
+        user.plan_level = subscribe.plan_levels[plan_type]
 
         var subscriber = old_subscriber || new Subscription ()
         subscriber.uuid = uuid
         subscriber.user_id = user._id
+        subscriber.plan_code = plan_code
         subscriber.plan_type = plan_type
         subscriber.plan_period = plan_period
-        subscriber.plan_level = subscribe.plan_level[plan_type]
+        subscriber.plan_level = subscribe.plan_levels[plan_type]
         subscriber.add_ons = []
         var add_ons = subscrip.subscription_add_ons[0].subscription_add_on
         if (add_ons) {