freemium-ajb 0.0.4

data/lib/freemium/subscription_change.rb

@@ -0,0 +1,20 @@
+module Freemium
+  module SubscriptionChange
+
+    def self.included(base)
+      base.class_eval do
+        belongs_to :subscribable, polymorphic: true
+
+        belongs_to :original_subscription_plan, class_name: "SubscriptionPlan"
+        belongs_to :new_subscription_plan, class_name: "SubscriptionPlan"
+
+        composed_of :new_rate, class_name: 'Money', mapping: [ %w(new_rate_cents cents) ], allow_nil: true
+        composed_of :original_rate, class_name: 'Money', mapping: [ %w(original_rate_cents cents) ], allow_nil: true
+
+        validates_presence_of :reason
+        validates_inclusion_of :reason, :in => %w(new upgrade downgrade expiration cancellation)
+      end
+    end
+
+  end
+end

data/lib/freemium/subscription_plan.rb

@@ -0,0 +1,26 @@
+# name
+# key
+# rate_cents
+module Freemium
+  module SubscriptionPlan
+    include Rates
+
+    def self.included(base)
+      base.class_eval do
+        # yes, subscriptions.subscription_plan_id may not be null, but
+        # this at least makes the delete not happen if there are any active.
+        has_many :subscriptions, dependent: :nullify
+        has_and_belongs_to_many :coupons
+
+        composed_of :rate, class_name: 'Money', mapping: [ %w(rate_cents cents) ], allow_nil: true
+
+        validates :name, presence: true
+        validates :key, presence: true
+        validates :rate_cents, presence: true
+
+        serialize :features, Hash
+      end
+    end
+
+  end
+end

data/lib/freemium/testing/app/controllers/application_controller.rb

@@ -0,0 +1,7 @@
+class ApplicationController < ActionController::Base
+  include Freemium::Controller
+
+  def show
+    render :text => '', :layout => 'application'
+  end
+end

data/lib/freemium/testing/application.rb

@@ -0,0 +1,46 @@
+require 'rails/all'
+
+module Freemium
+  module Testing
+    APP_ROOT = File.expand_path('..', __FILE__).freeze
+
+    class Application < Rails::Application
+      config.encoding = "utf-8"
+      config.action_mailer.default_url_options = { :host => 'localhost' }
+
+      if Rails::VERSION::MAJOR >= 3 && Rails::VERSION::MINOR >= 1
+        config.paths['config/database'] = "#{APP_ROOT}/config/database.yml"
+        config.paths['config/routes'] << "#{APP_ROOT}/config/routes.rb"
+        config.paths['app/controllers'] << "#{APP_ROOT}/app/controllers"
+        config.paths['app/views'] << "#{APP_ROOT}/app/views"
+        config.paths['log'] = "tmp/log/development.log"
+        config.assets.enabled = true
+      else
+        config.paths.config.database = "#{APP_ROOT}/config/database.yml"
+        config.paths.config.routes << "#{APP_ROOT}/config/routes.rb"
+        config.paths.app.controllers << "#{APP_ROOT}/app/controllers"
+        config.paths.app.views << "#{APP_ROOT}/app/views"
+        config.paths.log = "tmp/log"
+      end
+
+      config.cache_classes = true
+      config.whiny_nils = true
+      config.consider_all_requests_local = true
+      config.action_controller.perform_caching = false
+      config.action_dispatch.show_exceptions = false
+      config.action_controller.allow_forgery_protection = false
+      config.action_mailer.delivery_method = :test
+      config.active_support.deprecation = :stderr
+      config.secret_token = "SECRET_TOKEN_IS_MIN_30_CHARS_LONG"
+
+      def require_environment!
+        initialize!
+      end
+
+      def initialize!
+        FileUtils.mkdir_p(Rails.root.join("db").to_s)
+        super unless @initialized
+      end
+    end
+  end
+end

data/lib/freemium/testing/config/database.yml

@@ -0,0 +1,11 @@
+development:
+  adapter: sqlite3
+  database: db/development.sqlite3
+  pool: 5
+  timeout: 5000
+
+test:
+  adapter: sqlite3
+  database: db/test.sqlite3
+  pool: 5
+  timeout: 5000

data/lib/freemium/testing/config/routes.rb

@@ -0,0 +1,3 @@
+Rails.application.routes.draw do
+  root :to => "application#show"
+end

data/lib/freemium/transaction.rb

@@ -0,0 +1,15 @@
+module Freemium
+  module Transaction
+
+    def self.included(base)
+      base.class_eval do
+        scope :since, lambda { |time| where(["created_at >= ?", time]) }
+
+        belongs_to :subscription
+
+        composed_of :amount, :class_name => 'Money', :mapping => [ %w(amount_cents cents) ], :allow_nil => true
+      end
+    end
+
+  end
+end

data/lib/freemium/version.rb

@@ -0,0 +1,3 @@
+module Freemium
+  VERSION = '0.0.4'
+end

data/lib/generators/freemium/install/install_generator.rb

@@ -0,0 +1,58 @@
+require 'rails/generators/base'
+require 'rails/generators/active_record'
+
+module Freemium
+  module Generators
+    MODELS = %w(transaction coupon coupon_redemption credit_card
+                subscription subscription_change subscription_plan)
+
+    class InstallGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+      source_root File.expand_path('../templates', __FILE__)
+
+      def create_freemium_initializer
+        copy_file 'freemium.rb', 'config/initializers/freemium.rb'
+      end
+
+      def create_models
+        Freemium::Generators::MODELS.each do |model_name|
+          if !File.exists?("app/models/#{model_name}.rb")
+            copy_file "models/#{model_name}.rb", "app/models/#{model_name}.rb"
+            create_migration "create_#{model_name.pluralize}.rb"
+          end
+        end
+      end
+
+      private
+
+      def create_migration(migration_name, config = {})
+        unless migration_exists?(migration_name)
+          migration_template(
+            "db/migrate/#{migration_name}",
+            "db/migrate/#{migration_name}",
+            config
+          )
+        end
+      end
+
+      def migration_exists?(name)
+        existing_migrations.include?(name)
+      end
+
+      def existing_migrations
+        @existing_migrations ||= Dir.glob("db/migrate/*.rb").map do |file|
+          migration_name_without_timestamp(file)
+        end
+      end
+
+      def migration_name_without_timestamp(file)
+        file.sub(%r{^.*(db/migrate/)(?:\d+_)?}, '')
+      end
+
+      # for generating a timestamp when using `create_migration`
+      def self.next_migration_number(dir)
+        ActiveRecord::Generators::Base.next_migration_number(dir)
+      end
+    end
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_coupon_redemptions.rb

@@ -0,0 +1,18 @@
+class CreateCouponRedemptions < ActiveRecord::Migration
+  def self.up
+
+    create_table :coupon_redemptions, :force => true do |t|
+      t.column :subscription_id, :integer, :null => false
+      t.column :coupon_id, :integer, :null => false
+      t.column :redeemed_on, :date, :null => false
+      t.column :expired_on, :date, :null => true
+    end
+
+    add_index :coupon_redemptions, :subscription_id
+
+  end
+
+  def self.down
+    drop_table :coupon_redemptions
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_coupons.rb

@@ -0,0 +1,28 @@
+class CreateCoupons < ActiveRecord::Migration
+  def self.up
+
+    create_table :coupons, :force => true do |t|
+      t.column :description, :string, :null => false
+      t.column :discount_percentage, :integer, :null => false
+      t.column :redemption_key, :string, :null => true
+      t.column :redemption_limit, :integer, :null => true
+      t.column :redemption_expiration, :date, :null => true
+      t.column :duration_in_months, :integer, :null => true
+    end
+
+    create_table :coupons_subscription_plans, :id => false, :force => true do |t|
+      t.column :coupon_id, :integer, :null => false
+      t.column :subscription_plan_id, :integer, :null => false
+    end
+
+
+    add_index :coupons_subscription_plans, :coupon_id, :name => :on_coupon_id
+    add_index :coupons_subscription_plans, :subscription_plan_id, :name => :on_subscription_plan_id
+
+  end
+
+  def self.down
+    drop_table :coupons
+    drop_table :coupons_subscription_plans
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_credit_cards.rb

@@ -0,0 +1,14 @@
+class CreateCreditCards < ActiveRecord::Migration
+  def self.up
+    create_table :credit_cards, :force => true do |t|
+      t.column :display_number, :string, :null => false
+      t.column :card_type, :string, :null => false
+      t.column :expiration_date, :timestamp, :null => false
+      t.column :zip_code, :string, :null => true
+    end
+  end
+
+  def self.down
+    drop_table :credit_cards
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_subscription_changes.rb

@@ -0,0 +1,21 @@
+class CreateSubscriptionChanges < ActiveRecord::Migration
+  def self.up
+    create_table :subscription_changes, :force => true do |t|
+      t.column :subscribable_id, :integer, :null => false
+      t.column :subscribable_type, :string, :null => false
+      t.column :original_subscription_plan_id, :integer, :null => true
+      t.column :new_subscription_plan_id, :integer, :null => true
+      t.column :original_rate_cents, :integer, :null => true
+      t.column :new_rate_cents, :integer, :null => true
+      t.column :reason, :string, :null => false
+      t.column :created_at, :timestamp, :null => false
+    end
+
+    add_index :subscription_changes, :reason
+    add_index :subscription_changes, [:subscribable_id, :subscribable_type], name: :subscription_changes_on_subscribable
+  end
+
+  def self.down
+    drop_table :subscription_changes
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_subscription_plans.rb

@@ -0,0 +1,14 @@
+class CreateSubscriptionPlans < ActiveRecord::Migration
+  def self.up
+    create_table :subscription_plans, :force => true do |t|
+      t.column :name, :string, :null => false
+      t.column :key, :string, :null => false
+      t.column :rate_cents, :integer, :null => false
+      t.column :features, :text
+    end
+  end
+
+  def self.down
+    drop_table :subscription_plans
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_subscriptions.rb

@@ -0,0 +1,31 @@
+class CreateSubscriptions < ActiveRecord::Migration
+  def self.up
+    create_table :subscriptions, :force => true do |t|
+      t.column :subscribable_id, :integer, :null => false
+      t.column :subscribable_type, :string, :null => false
+      t.column :billing_key, :string, :null => true
+      t.column :credit_card_id, :integer, :null => true
+      t.column :subscription_plan_id, :integer, :null => false
+      t.column :paid_through, :date, :null => true
+      t.column :expire_on, :date, :null => true
+      t.column :billing_key, :string, :null => true
+      t.column :started_on, :date, :null => true
+      t.column :last_transaction_at, :datetime, :null => true
+      t.column :last_transaction_success, :boolean, :null => true
+      t.column :in_trial, :boolean, :null => false, :default => false
+    end
+    # for polymorphic association queries
+    add_index :subscriptions, :subscribable_id
+    add_index :subscriptions, :subscribable_type
+    add_index :subscriptions, [:subscribable_id, :subscribable_type], name: :subscriptions_on_subscribable
+    # for finding due, pastdue, and expiring subscriptions
+    add_index :subscriptions, :paid_through
+    add_index :subscriptions, :expire_on
+    # for applying transactions from automated recurring billing
+    add_index :subscriptions, :billing_key
+  end
+
+  def self.down
+    drop_table :subscriptions
+  end
+end

data/lib/generators/freemium/install/templates/db/migrate/create_transactions.rb

@@ -0,0 +1,17 @@
+class CreateTransactions < ActiveRecord::Migration
+  def self.up
+    create_table :transactions, :force => true do |t|
+      t.column :subscription_id, :integer, :null => false
+      t.column :success, :boolean, :null => false
+      t.column :billing_key, :string, :null => false
+      t.column :amount_cents, :integer, :null => false
+      t.column :message, :string, :null => true
+      t.column :created_at, :timestamp, :null => false
+    end
+    add_index :transactions, :subscription_id
+  end
+
+  def self.down
+    drop_table :transactions
+  end
+end

data/lib/generators/freemium/install/templates/freemium.rb

@@ -0,0 +1,16 @@
+Freemium.configure do |c|
+  if Rails.env.test?
+    c.gateway = Freemium::Gateways::Test.new
+  else
+    c.gateway = Freemium::Gateways::BrainTree.new
+    c.gateway.username = "demo"
+    c.gateway.password = "password"
+  end
+
+  c.admin_report_recipients = ["your@email.com"]
+
+  # c.days_grace = 3
+  # c.days_free_trial = 0
+  # c.mailer = FreemiumMailer
+  # c.expired_plan = nil
+end

data/lib/generators/freemium/install/templates/models/coupon.rb

@@ -0,0 +1,3 @@
+class Coupon < ActiveRecord::Base
+  include Freemium::Coupon
+end

data/lib/generators/freemium/install/templates/models/coupon_redemption.rb

@@ -0,0 +1,3 @@
+class CouponRedemption < ActiveRecord::Base
+  include Freemium::CouponRedemption
+end

data/lib/generators/freemium/install/templates/models/credit_card.rb

@@ -0,0 +1,3 @@
+class CreditCard < ActiveRecord::Base
+  include Freemium::CreditCard
+end

data/lib/generators/freemium/install/templates/models/subscription.rb

@@ -0,0 +1,3 @@
+class Subscription < ActiveRecord::Base
+  include Freemium::Subscription
+end

data/lib/generators/freemium/install/templates/models/subscription_change.rb

@@ -0,0 +1,3 @@
+class SubscriptionChange < ActiveRecord::Base
+  include Freemium::SubscriptionChange
+end

data/lib/generators/freemium/install/templates/models/subscription_plan.rb

@@ -0,0 +1,3 @@
+class SubscriptionPlan < ActiveRecord::Base
+  include Freemium::SubscriptionPlan
+end

data/lib/generators/freemium/install/templates/models/transaction.rb

@@ -0,0 +1,3 @@
+class Transaction < ActiveRecord::Base
+  include Freemium::Transaction
+end

data/lib/generators/views/USAGE

@@ -0,0 +1,3 @@
+Description:
+  Override the default freemum views. This generator will copy all off the
+  base freemium views into your project.

data/lib/generators/views/views_generator.rb

@@ -0,0 +1,39 @@
+require 'rails/generators/base'
+
+module Freemium
+  module Generators
+    class ViewsGenerator < Rails::Generators::Base
+      source_root Freemium.root
+
+      def create_views
+        views.each do |view|
+          copy_file view
+        end
+      end
+
+      def create_locales
+        locales.each do |locale|
+          copy_file locale
+        end
+      end
+
+      private
+
+      def views
+        files_within_root('.', 'app/views/**/*.*')
+      end
+
+      def locales
+        files_within_root('.', 'config/locales/**/*.*')
+      end
+
+      def files_within_root(prefix, glob)
+        root = "#{self.class.source_root}/#{prefix}"
+
+        Dir["#{root}/#{glob}"].sort.map do |full_path|
+          full_path.sub(root, '.').gsub('/./', '/')
+        end
+      end
+    end
+  end
+end