add subscriptions

1 files changed, 44 insertions, 79 deletions

diff --git a/engines/billing/app/controllers/subscriptions_controller.rb b/engines/billing/app/controllers/subscriptions_controller.rb
index 65db21b..1d29cac 100644
--- a/engines/billing/app/controllers/subscriptions_controller.rb
+++ b/engines/billing/app/controllers/subscriptions_controller.rb
@@ -1,107 +1,72 @@
 class SubscriptionsController < BillingBaseController
   before_filter :require_login
-  before_filter :confirm_cancel_subscription, :only => [:destroy]
+  before_filter :assign_user
+  before_filter :confirm_cancel_subscription, only: [:destroy]
+  before_filter :generate_client_token, only: [:show]
+  before_filter :get_braintree_customer, only: [:subscribe]
 
   def index
-    @subscriptions = Braintree::Plan.all
+    if @user.subscription_id
+      @subscription = Braintree::Subscription.find @user.subscription_id
+      @plan = Braintree::Plan.all.select{ |plan| plan.id == @subscription.plan_id }.first
+    else
+      @subscriptions = Braintree::Plan.all
+    end
   end
 
   def show
-    @subscriptions = Braintree::Plan.all
-    @subscriptions = Braintree::Plan.all.find params[:subscription_id]
-  end
-
-  def search
-    @subscription = Braintree::Subscription.search params[:status]
-  end
-
-  def new
-    if current_user.braintree_customer_id
-      @client_token = Braintree::ClientToken.generate(customer_id: current_user.braintree_customer_id)
-    else
-     @client_token = Braintree::ClientToken.generate
-    end
-    @subscriptions = Braintree::Plan.all
+    @plan = Braintree::Plan.all.select{ |plan| plan.id == params[:id] }.first
   end
 
-  def create
-    @result = Braintree::Subscription.create(
-      payment_method_token: braintree_customer.payment_methods.first.token,
-      plan_id: params[:plan_id],
-    )
+  def subscribe
+    @result = Braintree::Subscription.create(payment_method_token: @customer.payment_methods.first.token,
+                                             plan_id: params[:id])
     if @result.success?
-      flash[:success] = "Congratulations! Your transaction has been successfully!"
+      @user.update_attributes subscription_id: @result.subscription.id
+      flash[:success] = I18n.t(:subscription_sucess)
     else
-      flash[:error] = "Something went wrong while processing your donation. Please try again!"
+      flash[:error] = I18n.t(:subscription_not_sucess)
     end
-    redirect_to action: :new, locale: params[:locale]
+    redirect_to action: :index, locale: params[:locale]
   end
 
-  def braintree_customer
-    if current_user.braintree_customer_id
-      Braintree::Customer.find current_user.braintree_customer_id
+  def unsubscribe
+    @result = Braintree::Subscription.cancel(@user.subscription_id)
+    if @result.success?
+      @user.update_attributes subscription_id: nil
+      flash[:success] = I18n.t(:unsubscription_sucess)
     else
-      customer = Braintree::Customer.create(payment_method_nonce: params[:payment_method_nonce]).customer
-      current_user.update_attributes braintree_customer_id: customer.id
-      customer
+      flash[:error] = I18n.t(:unsubscription_not_sucess)
     end
+    redirect_to action: :index, locale: params[:locale]
   end
 
-  def confirm
-    @result = Braintree::Subscription.sale(
-      payment_method_token: params[:payment_method_nonce],
-      plan_id: params[:plan_id],
-    )
+  private
+  def assign_user
+    @user = current_user
   end
 
-  def _confirm
-    make_subscription
-    if @result.success?
-      flash[:success] = "Congratulations! Your transaction has been successfully!"
+  def generate_client_token
+    if current_user.braintree_customer_id
+      @client_token = Braintree::ClientToken.generate(customer_id: current_user.braintree_customer_id)
     else
-      flash[:error] = "Something went wrong while processing your donation. Please try again!"
+     @client_token = Braintree::ClientToken.generate
     end
-  redirect_to action: :new, locale: params[:locale]
   end
 
-private
-
-  def make_subscription
-    unless current_user.has_payment_info?
-      subs_with_user_info
+  def get_braintree_customer
+    if current_user.braintree_customer_id
+      @customer = Braintree::Customer.find(current_user.braintree_customer_id)
     else
-      subs_without_user_info
+      @customer = Braintree::Customer.create(
+                              payment_method_nonce: params[:payment_method_nonce],
+                              first_name: params[:first_name],
+                              last_name: params[:last_name],
+                              company: params[:company],
+                              email: current_user.email,
+                              phone: params[:phone]
+                            ).customer
+      current_user.update_attributes braintree_customer_id: @customer.id
     end
   end
-
-  def subs_with_user_info
-    # don't show link to subscribe if they are already subscribed?
-    @result = Braintree::Subscription.sale(
-    payment_method_token: params[:payment_method_nonce],
-    plans_id: Braintree::Plan.all,
-    customer: {
-       first_name: params[:first_name],
-       last_name: params[:last_name],
-       company: params[:company],
-       email: current_user.email,
-       phone: params[:phone]
-       },
-    options: {
-    store_in_vault: true
-    })
-    current_user.update_attributes(braintree_customer_id: @result.transaction.customer_details.id) if @result.success?
-  end
-
-  def subs_without_user_info
-    @result = Braintree::Subscription.sale(
-              payment_method_token: params[:payment_method_nonce],
-              plans_id: Braintree::Plan.all
-    )
-  end
-
-  def destroy
-    @subscriptions = Braintree::Subscription.cancel params[:id]
-  end
-
-
 end