1 files changed, 36 insertions, 0 deletions
diff --git a/billing/app/controllers/customer_controller.rb b/billing/app/controllers/customer_controller.rb
new file mode 100644
index 0000000..dcb94eb
--- /dev/null
+++ b/billing/app/controllers/customer_controller.rb
@@ -0,0 +1,36 @@
+class CustomerController < ApplicationController
+  before_filter :authorize
+
+  def new
+    @tr_data = Braintree::TransparentRedirect.
+                create_customer_data(:redirect_url => confirm_customer_url)
+  end
+
+  def edit
+    current_customer = Customer.find_by_user_id(current_user.id)
+    #current_customer.with_braintree_data!
+    #@credit_card = current_customer.default_credit_card
+    @tr_data = Braintree::TransparentRedirect.
+                update_customer_data(:redirect_url => confirm_customer_url,
+                                     :customer_id => current_customer.braintree_customer_id)
+  end
+
+  def confirm
+    @result = Braintree::TransparentRedirect.confirm(request.query_string)
+
+    if @result.success?
+      # customer = Customer.new(:user_id => current_user.id, :braintree_customer_id =>  @result.customer.id)
+      customer = Customer.new(:braintree_customer_id =>  @result.customer.id)
+      customer.user = current_user
+      customer.save
+      #current_user.save!
+      render :action => "confirm"
+    #elsif current_user.has_payment_info?
+    elsif (current_customer = Customer.find_by_user_id(current_user.id)) and current_customer.has_payment_info?
+      current_customer.with_braintree_data! #todo
+      render :action => "edit"
+    else
+      render :action => "new"
+    end
+  end
+end