stripe_saas 0.0.1

data/app/controllers/stripe_saas/application_controller.rb

@@ -0,0 +1,5 @@
+module StripeSaas
+  class ApplicationController < ::ApplicationController
+    helper :application
+  end
+end

data/app/controllers/stripe_saas/subscriptions_controller.rb

@@ -0,0 +1,200 @@
+module StripeSaas
+  class SubscriptionsController < ApplicationController
+    before_filter :load_owner
+    before_filter :show_existing_subscription, only: [:index, :new, :create], unless: :no_owner?
+    before_filter :load_subscription, only: [:show, :cancel, :edit, :update]
+    before_filter :load_plans, only: [:index, :edit]
+
+    def load_plans
+      @plans = ::Plan.order(:price_cents)
+    end
+
+    def unauthorized
+      render status: 401, template: "stripe_saas/subscriptions/unauthorized"
+      false
+    end
+
+    # subscription.subscription_owner
+    def load_owner
+      unless params[:owner_id].nil?
+        if current_owner.present?
+
+          # we need to try and look this owner up via the find method so that we're
+          # taking advantage of any override of the find method that would be provided
+          # by older versions of friendly_id. (support for newer versions default behavior
+          # below.)
+          searched_owner = current_owner.class.find(params[:owner_id]) rescue nil
+
+          # if we couldn't find them that way, check whether there is a new version of
+          # friendly_id in place that we can use to look them up by their slug.
+          # in christoph's words, "why?!" in my words, "warum?!!!"
+          # (we debugged this together on skype.)
+          if searched_owner.nil? && current_owner.class.respond_to?(:friendly)
+            searched_owner = current_owner.class.friendly.find(params[:owner_id]) rescue nil
+          end
+
+          if current_owner.try(:id) == searched_owner.try(:id)
+            @owner = current_owner
+          else
+            customer = Subscription.find_customer(searched_owner)
+            customer_2 = Subscription.find_customer(current_owner)
+            # susbscription we are looking for belongs to the same user but to a different
+            # subscription owner
+            # e.g. user -> account -> subscription
+            # same user but different accounts for example
+
+            if customer_2.try(:id) == customer.try(:id)
+              @owner = searched_owner
+            else
+              return unauthorized
+            end
+          end
+        else
+          return unauthorized
+        end
+      end
+    end
+
+    def no_owner?
+      @owner.nil?
+    end
+
+    def load_subscription
+      # config.subscriptions_owned_by = :account
+      # config.customer_accessor = :owner
+      ownership_attribute = :"#{StripeSaas.subscriptions_owned_by}_id"
+      @subscription = ::Subscription.where(ownership_attribute => current_owner.id).find_by_id(params[:id]) ||
+                      ::Subscription.where(ownership_attribute => @owner.id).find_by_id(params[:id])
+      return @subscription.present? ? @subscription : unauthorized
+    end
+
+    # the following two methods allow us to show the pricing table before someone has an account.
+    # by default these support devise, but they can be overriden to support others.
+    def current_owner
+      # e.g. "self.current_user"
+      send "current_#{StripeSaas.subscriptions_owned_by}"
+    end
+
+    def redirect_to_sign_up
+      # this is a Devise default variable and thus should not change its name
+      # when we change subscription owners from :user to :company
+      session["user_return_to"] = new_subscription_path(plan: params[:plan])
+      devise_scope = (StripeSaas.devise_scope || StripeSaas.subscriptions_owned_by).to_s
+      redirect_to new_registration_path(devise_scope)
+    end
+
+    def index
+      # don't bother showing the index if they've already got a subscription.
+      if current_owner and current_owner.subscription.present?
+        redirect_to stripe_saas.edit_owner_subscription_path(current_owner, current_owner.subscription)
+      end
+
+      # Load all plans.
+      @plans = ::Plan.order(:display_order).all
+
+      # Don't prep a subscription unless a user is authenticated.
+      unless no_owner?
+        # we should also set the owner of the subscription here.
+        @subscription = ::Subscription.new({StripeSaas.owner_id_sym => @owner.id})
+        @subscription.subscription_owner = @owner
+      end
+
+    end
+
+    def new
+      if no_owner?
+
+        if defined?(Devise)
+
+          # by default these methods support devise.
+          if current_owner
+            redirect_to new_owner_subscription_path(current_owner, plan: params[:plan])
+          else
+            redirect_to_sign_up
+          end
+
+        else
+          raise "This feature depends on Devise for authentication."
+        end
+
+      else
+        @subscription = ::Subscription.new
+        @subscription.plan = ::Plan.find_by_stripe_id(params[:plan])
+      end
+    end
+
+    def show_existing_subscription
+      if @owner.subscription.present?
+        redirect_to owner_subscription_path(@owner, @owner.subscription)
+      end
+    end
+
+    def create
+      @subscription = ::Subscription.new(subscription_params)
+      @subscription.subscription_owner = @owner
+
+      if @subscription.save
+        flash[:notice] = after_new_subscription_message
+        redirect_to after_new_subscription_path
+      else
+        flash[:error] = 'There was a problem processing this transaction.'
+        render :new
+      end
+    end
+
+    def show
+    end
+
+    def cancel
+      flash[:notice] = "You've successfully cancelled your subscription."
+      @subscription.plan_id = nil
+      @subscription.save
+      redirect_to owner_subscription_path(@owner, @subscription)
+    end
+
+    def edit
+    end
+
+    def update
+      new_plan_id = subscription_params[:plan_id]
+      new_plan = ::Plan.find(new_plan_id)
+
+      if @subscription.plan.free? && !new_plan.free? && subscription_params[:credit_card_token].nil?
+        flash[:notice] = "Please enter payment information to upgrade."
+        redirect_to edit_owner_subscription_path(@owner, @subscription, update: 'card', plan: new_plan_id)
+      else
+        if @subscription.update_attributes(subscription_params)
+          flash[:notice] = "You've successfully updated your subscription."
+          redirect_to edit_owner_subscription_path(@owner, @subscription)
+        else
+          flash[:error] = 'There was a problem processing this transaction.'
+          render :edit
+        end
+      end
+    end
+
+    private
+
+    def subscription_params
+      # If strong_parameters is around, use that.
+      if defined?(ActionController::StrongParameters)
+        params.require(:subscription).permit(:plan_id, :stripe_id, :current_price, :credit_card_token, :card_type, :last_four)
+      else
+        # Otherwise, let's hope they're using attr_accessible to protect their models!
+        params[:subscription]
+      end
+    end
+
+    def after_new_subscription_path
+      controller = ::ApplicationController.new
+      controller.respond_to?(:after_new_subscription_path) ?
+      controller.try(:after_new_subscription_path, @owner, @subscription) : owner_subscription_path(@owner, @subscription)
+    end
+
+    def after_new_subscription_message
+      controller = ::ApplicationController.new
+      controller.respond_to?(:new_subscription_notice_message) ?
+      controller.try(:new_subscription_notice_message) : "You've been successfully upgraded."
+    end
+  end
+end

data/app/helpers/stripe_saas/application_helper.rb

@@ -0,0 +1,26 @@
+module StripeSaas
+  module ApplicationHelper
+    include MoneyRails::ActionViewExtension
+
+    def plan_price(plan)
+      "#{humanized_money_with_symbol(plan.price, :no_cents => false)}/#{plan_interval(plan)}"
+    end
+
+    def plan_interval(plan)
+      case plan.interval
+      when "month"
+        "month"
+      when "year"
+        "year"
+      when "week"
+        "week"
+      when "6-month"
+        "half-year"
+      when "3-month"
+        "quarter"
+      else
+        "month"
+      end
+    end
+  end
+end

data/app/views/stripe_saas/subscriptions/_card.html.erb

@@ -0,0 +1,51 @@
+<% content_for :stripe_saas do %>
+<script type="text/javascript" src="https://js.stripe.com/v2/"></script>
+<% end %>
+
+<%= render 'card_form', url: url, title: title %>
+
+<script type="text/javascript">
+  Stripe.setPublishableKey("<%= StripeSaas.stripe_publishable_key %>");
+
+  var stripeResponseHandler = function (status, response) {
+
+    if (response.error) {
+      // show the errors on the form
+      $(".payment-errors").text(response.error.message).show();
+      $(".submit-button").removeAttr("disabled");
+    } else {
+      var $form = $("#payment-form");
+      // token contains id, last4, and card type
+      // insert the token into the form so it gets submitted to the server
+      $cc_token_field = $('<input type="hidden" name="subscription[credit_card_token]" />').val(response.id);
+      $last_four_field = $('<input type="hidden" name="subscription[last_four]" />').val(response.card.last4);
+      $card_brand = $('<input type="hidden" name="subscription[card_type]" />').val(response.card.brand);
+      //
+      $form.append($cc_token_field, $last_four_field, $card_brand);
+
+      $form.get(0).submit();
+    }
+  }
+
+  $(document).ready(function() {
+    // By default, don't show errors.
+    $(".payment-errors").hide()
+
+    $("#payment-form").submit(function(event) {
+
+      // disable the submit button to prevent repeated clicks
+      $('.submit-button').attr("disabled", "disabled");
+
+      Stripe.createToken({
+        number: $('.card-number').val(),
+        cvc: $('.card-cvc').val(),
+        exp_month: $('.card-expiry-month').val(),
+        exp_year: $('.card-expiry-year').val()
+      }, stripeResponseHandler);
+
+      // prevent the form from submitting with the default action
+      event.preventDefault();
+    });
+  });
+
+</script>

data/app/views/stripe_saas/subscriptions/_card_form.html.erb

@@ -0,0 +1,42 @@
+<%= form_for @subscription, url: url, html: {id: 'payment-form', class: 'form-horizontal'} do |f| %>
+
+<fieldset>
+
+  <legend><%= title %></legend>
+
+  <div class="control-group">
+    <label class="control-label">Card Number</label>
+    <div class="controls">
+      <input type="text" size="20" autocomplete="off" class="card-number"/>
+    </div>
+  </div>
+
+  <div class="control-group">
+    <label class="control-label">Expiration (MM/YYYY)</label>
+    <div class="controls">
+      <input type="text" size="2" class="card-expiry-month input-mini"/>
+      <span> / </span>
+      <input type="text" size="4" class="card-expiry-year input-mini"/>
+    </div>
+  </div>
+
+  <div class="control-group">
+    <label class="control-label">CVC</label>
+    <div class="controls">
+      <input type="text" size="4" autocomplete="off" class="card-cvc input-small"/>
+    </div>
+  </div>
+
+  <div class="alert alert-error payment-errors"></div>
+  <%= f.hidden_field :plan_id, value: params[:plan] || @subscription.plan_id %>
+
+</fieldset>
+
+<div class="control-group">
+  <div class="controls">
+    <button type="submit" class="btn btn-primary submit-button">Upgrade Your Account</button>
+    <%= link_to "Cancel", owner_subscriptions_path(@owner), class: 'btn' %>
+  </div>
+</div>
+
+<% end %>

data/app/views/stripe_saas/subscriptions/_pricing_table.html.erb

@@ -0,0 +1,43 @@
+<ul class="thumbnails pricing-table">
+  <% @plans.each do |plan| %>
+    <li class="span3 plan <%= 'plan-primary' if plan.highlight? %>">
+      <div class="thumbnail">
+        <div class="caption">
+          <h3><%= plan.name %></h3>
+          <h4><%= plan_price(plan) %></h4>
+          <div class="call-to-action">
+            <% if @subscription.nil? %>
+              <%= link_to 'Sign Up', stripe_saas.new_subscription_path(plan: plan.stripe_id), class: "btn btn-success btn-large" %>
+            <% elsif @subscription.persisted? %>
+              <% if @subscription.plan == plan %>
+                <%= form_for @subscription, url: owner_subscription_path(@owner, @subscription) do |f| %>
+                  <%= f.submit 'Selected', class: "btn btn-large", disabled: 'disabled' %>
+                <% end %>
+              <% else %>
+                <%= form_for @subscription, url: owner_subscription_path(@owner, @subscription) do |f| %>
+                  <%= f.hidden_field :plan_id, value: plan.id %>
+                  <%= f.submit @subscription.describe_difference(plan), class: "btn btn-success btn-large" %>
+                <% end %>
+              <% end %>
+            <% else %>
+              <%= link_to 'Upgrade', new_owner_subscription_path(@owner, plan: plan.stripe_id), class: "btn btn-success btn-large" %>
+            <% end %>
+          </div>
+          <ul class="features">
+            <ul>
+              <% plan.plan_features.each do |plan_feature| %>
+                <li>
+                <% if plan_feature.feature.feature_type == 'boolean' %>
+                  <%= plan_feature.value > 0 ? number_with_delimiter(plan_feature.value) : "No" %> <%= plan_feature.feature.description %>
+                <% else %>
+                  <%= pf.value ? '&#10004;' : '&#10006;' %> <%= pf.feature.description %>
+                <% end %>
+                </li>
+              <% end %>
+            </ul>
+          </ul>
+        </div>
+      </div>
+    </li>
+  <% end %>
+</ul>

data/app/views/stripe_saas/subscriptions/edit.html.erb

@@ -0,0 +1,7 @@
+<% if params['update'] == 'card' %>
+  <%= render 'card', title: "Update Your Payment Information", url: owner_subscription_path(@owner, @subscription) %>
+<% else %>
+  <h1>What Plan Is Best For You?</h1>
+  <%= render 'pricing_table' %>
+  <p>You can also <%= link_to 'cancel your subscription', cancel_owner_subscription_path(@owner, @subscription), method: :post %>.</p>
+<% end %>

data/app/views/stripe_saas/subscriptions/index.html.erb

@@ -0,0 +1,2 @@
+<h1>What Plan Is Best For You?</h1>
+<%= render 'pricing_table' %>

data/app/views/stripe_saas/subscriptions/new.html.erb

@@ -0,0 +1 @@
+<%= render 'card', title: "Upgrade Your Account", url: owner_subscriptions_path(@owner) %>

data/app/views/stripe_saas/subscriptions/show.html.erb

@@ -0,0 +1,15 @@
+<% if @subscription.plan.present? %>
+  <h2>You're Subscribed!</h2>
+  <p>You're currently subscribed to the <%= @subscription.plan.name %> plan.</p>
+  <%= link_to 'Choose Another Plan', edit_owner_subscription_path(@owner, @subscription), class: 'btn' %>
+<% else %>
+  <h2>No Subscription</h2>
+  <p>You are not subscribed to a paid plan.</p>
+  <%= link_to 'Choose A Plan', edit_owner_subscription_path(@owner, @subscription), class: 'btn' %>
+<% end %>
+
+<br><br>
+
+<h4>Payment Information</h4>
+<p>The card on file for your account ends with <%= @subscription.last_four %>.</p>
+<%= link_to 'Update Payment Information', edit_owner_subscription_path(@owner, @subscription, update: 'card'), class: 'btn' %>

data/app/views/stripe_saas/subscriptions/unauthorized.html.erb

@@ -0,0 +1 @@
+<h1>Unauthorized</h1>

data/config/routes.rb

@@ -0,0 +1,10 @@
+StripeSaas::Engine.routes.draw do
+  resources :subscriptions, only: [:new]
+  resources StripeSaas.owner_resource, as: :owner do
+    resources :subscriptions do
+      member do
+        post :cancel
+      end
+    end
+  end
+end

data/lib/generators/stripe_saas/install_generator.rb

@@ -0,0 +1,60 @@
+require 'rails/generators'
+
+module StripeSaas
+  class InstallGenerator < Rails::Generators::Base
+
+    def self.source_paths
+      [StripeSaas::Engine.root, File.expand_path("../templates", __FILE__)]
+    end
+
+    include Rails::Generators::Migration
+
+    argument :subscription_owner_model, :type => :string, :required => true, :desc => "Owner of the subscription"
+    desc "StripeSaas installation generator"
+
+    # Override the attr_accessor generated by 'argument' so that
+    # subscription_owner_model is always returned lowercase.
+    def subscription_owner_model
+      @subscription_owner_model.downcase
+    end
+
+    def install
+
+      unless defined?(StripeSaas)
+        gem("stripe_saas")
+      end
+
+      require "securerandom"
+      @api_key = SecureRandom.uuid
+
+      template "config/initializers/stripe_saas.rb"
+
+      # Generate subscription.
+      generate("model", "subscription stripe_id:string plan_id:integer last_four:string card_type:string current_price_cents:integer #{subscription_owner_model}_id:integer")
+      template "app/models/subscription.rb"
+
+      # Add the plans.
+      generate("model", "plan stripe_id:string name:string price_cents:integer interval:string interval_count:integer trial_period_days:integer metadata_as_json:text statement_descriptor:text highlight:boolean display_order:integer")
+      template "app/models/plan.rb"
+
+      # Add features
+      generate("model", "feature name:string description:string feature_type:string unit:string display_order:integer")
+      template "app/models/feature.rb"
+
+      # Add Plan Features
+      generate("model", "plan_feature value:string display_value:string plan_id:integer feature_id:integer")
+      template "app/models/plan_feature.rb"
+
+      # Update the owner relationship.
+      inject_into_class "app/models/#{subscription_owner_model}.rb", Plan, "has_one :subscription\n"
+
+      route <<-RUBY
+      mount StripeSaas::Engine, at: 'stripe_saas'
+      scope module: 'stripe_saas' do
+        get 'pricing' => 'subscriptions#index', as: 'pricing'
+      end
+      RUBY
+    end
+
+  end
+end