mac_generators 0.0.1 → 0.1.1

data/test/dummy/tmp/Gemfile

@@ -0,0 +1,3 @@
+
+gem "warden", "~> 1.2.0"
+gem "omniauth"

data/test/dummy/tmp/app/controllers/application_controller.rb

@@ -0,0 +1,27 @@
+class ApplicationController < ActionController::Base
+  protect_from_forgery with: :exception
+
+  helper_method :current_identity, :identity_signed_in?, :warden_message
+
+protected
+  def current_identity
+    warden.user(scope: :identity)
+  end
+
+  def identity_signed_in?
+    warden.authenticate?(scope: :identity)
+  end
+
+  def authenticate!
+    redirect_to root_path, notice: t('.not_logged') unless identity_signed_in?
+  end
+
+  def warden_message
+    warden.message
+  end
+
+  def warden
+    request.env['warden']
+  end
+
+end

data/test/dummy/tmp/app/controllers/sessions_controller.rb

@@ -0,0 +1,12 @@
+class SessionsController < ApplicationController
+  def create
+    warden.authenticate!(scope: :identity)
+    redirect_to root_url, notice: t('.logged_in')
+  end
+
+  def destroy
+    warden.logout(:identity)
+    redirect_to root_url, notice: t('.logged_out')
+  end
+end
+

data/test/dummy/tmp/app/models/identity.rb

@@ -0,0 +1,9 @@
+class Identity < ActiveRecord::Base
+
+  class << self
+    def find_identity(uid, provider)
+      where(uid: uid, provider: provider).first
+    end
+  end
+
+end

data/test/dummy/tmp/config/initializers/authentication_domain.rb

@@ -0,0 +1 @@
+Rails.application.config.authentication_domain = ''

data/test/dummy/tmp/config/initializers/omniauth.rb

@@ -0,0 +1,3 @@
+Rails.application.config.middleware.use OmniAuth::Builder do
+  provider :provider, ENV["KEY"], ENV["SECRET"]
+end

data/test/dummy/tmp/config/initializers/warden.rb

@@ -0,0 +1,19 @@
+load File.expand_path("../../../lib/strategies/oauth_authentication.rb",  __FILE__)
+Rails.application.config.middleware.use Warden::Manager do |manager|
+  manager.default_strategies :oauth_authentication
+
+  # TODO: Setup warden's failure app, this will be called everytime that
+  # and authentication failure happen.
+  # Failure app should be a Rack application.
+  # In Rails a controller can be used as a Rack app, just specify the
+  # controller and the action to be called. Example:
+  #manager.failure_app = lambda { |env| HomeController.action(:index).call(env) }
+end
+
+Warden::Manager.serialize_into_session(:identity) do |identity|
+  identity.id
+end
+
+Warden::Manager.serialize_from_session(:identity) do |id|
+  Identity.find(id)
+end

data/test/dummy/tmp/config/locales/en.yml

@@ -0,0 +1,10 @@
+en:
+  sessions:
+    new:
+      log_in: 'Log in'
+    create:
+      unauthorized_domain: 'Sorry but your domain is not authorized'
+      logged_in: 'Welcome back!'
+    destroy:
+      logged_out: 'See you later!'
+

data/test/dummy/tmp/config/routes.rb

@@ -0,0 +1,4 @@
+Dummy::Application.routes.draw do
+  delete '/sessions/destroy' => 'sessions#destroy', as: :log_out
+  get 'auth/:provider/callback' => 'sessions#create', as: :log_in
+end

data/test/dummy/tmp/db/migrate/create_identities.rb

@@ -0,0 +1,12 @@
+class CreateIdentities < ActiveRecord::Migration
+  def change
+    create_table :identities do |t|
+      t.string :provider
+      t.string :uid
+      t.string :name
+      t.string :email
+
+      t.timestamps
+    end
+  end
+end

data/test/dummy/tmp/lib/strategies/oauth_authentication.rb

@@ -0,0 +1,36 @@
+module Strategies
+  class OauthAuthentication < ::Warden::Strategies::Base
+    def valid?
+      request.env['omniauth.auth'].present?
+    end
+
+    def authenticate!
+      auth = request.env['omniauth.auth']
+
+      if authorized_domain?(auth)
+        identity = Identity.find_identity(auth['uid'], auth['provider']) || create_identity(auth)
+        return success! identity
+      end
+
+      fail! I18n.t('sessions.create.unauthorized_domain')
+    end
+
+  private
+    def authorized_domain?(auth)
+      if Rails.application.config.respond_to?(:authentication_domain) && Rails.application.config.authentication_domain.present?
+        return auth['info']['email'].split('@').last == Rails.application.config.authentication_domain
+      end
+
+      true
+    end
+
+    def create_identity(auth)
+      params = { uid: auth['uid'], provider: auth['provider'],
+                name: auth['info']['name'], email: auth['info']['email'] }
+
+      Identity.create! params
+    end
+  end
+end
+
+Warden::Strategies.add(:oauth_authentication, Strategies::OauthAuthentication)

data/test/fixtures/application_controller.rb

@@ -0,0 +1,3 @@
+class ApplicationController < ActionController::Base
+  protect_from_forgery with: :exception
+end

data/test/fixtures/en.yml

@@ -0,0 +1 @@
+en:

data/test/fixtures/routes.rb

@@ -0,0 +1,2 @@
+Dummy::Application.routes.draw do
+end

data/test/generators/authentication_email_generator_test.rb

@@ -0,0 +1,141 @@
+require 'test_helper'
+require 'generators/authentication/email/email_generator'
+
+describe Authentication::Generators::EmailGenerator do
+  include GeneratorsTestHelper
+
+  before do
+    copy_routes
+    copy_locales
+    copy_gemfile
+    copy_application_controller
+    make_migrations_dir
+  end
+
+  it 'copies controllers templates for user signup and session' do
+    run_generator
+
+    assert_file 'app/controllers/identities_controller.rb' do |m|
+      assert_match 'IdentitiesController', m
+      assert_match 'def new', m
+      assert_match 'def create', m
+    end
+
+    assert_file 'app/controllers/sessions_controller.rb' do |m|
+      assert_match 'SessionsController', m
+      assert_match 'def new', m
+      assert_match 'def create', m
+      assert_match 'def destroy', m
+    end
+  end
+
+  it 'copies user_controller template for user class' do
+    run_generator ['user']
+
+    assert_file 'app/controllers/users_controller.rb' do |m|
+      assert_match 'UsersController', m
+      assert_match 'def new', m
+      assert_match 'def create', m
+      assert_match '@user = User.new', m
+    end
+  end
+
+  it 'copies erb templates' do
+    run_generator
+
+    assert_file 'app/views/identities/new.html.erb' do |m|
+      assert_match 'form_for @identity, url: identity_path', m
+    end
+
+    assert_file 'app/views/sessions/new.html.erb' do |m|
+      assert_match 'form_for @identity, url: sessions_path', m
+    end
+  end
+
+  it 'copies haml templates' do
+    run_generator ['--haml']
+
+    assert_file 'app/views/identities/new.html.haml' do |m|
+      assert_match 'form_for @identity, url: identity_path', m
+    end
+
+    assert_file 'app/views/sessions/new.html.haml' do |m|
+      assert_match 'form_for @identity, url: sessions_path', m
+    end
+  end
+
+  it 'add routes' do
+    run_generator
+
+    assert_file 'config/routes.rb' do |m|
+      assert_match "get 'sign_up' => 'identities#new'", m
+      assert_match "get 'log_in' => 'sessions#new'", m
+      assert_match "delete 'log_out' => 'sessions#destroy'", m
+
+      assert_match "resource :identity, only: [:create, :new]", m
+      assert_match "resource :sessions, only: [:create, :new]", m
+    end
+  end
+
+  it 'generate migration' do
+    Authentication::Generators::EmailGenerator.class_eval do
+      def migration_name
+        "create_#{resource_pluralize}.rb"
+      end
+    end
+
+    run_generator
+
+    assert_file 'db/migrate/create_identities.rb' do |m|
+      assert_match 'create_table :identities', m
+      assert_match 't.string :email', m
+      assert_match 't.string :password_hash', m
+      assert_match 't.string :password_digest', m
+    end
+
+    assert_file 'app/models/identity.rb' do |m|
+      assert_match 'class Identity < ActiveRecord::Base', m
+      assert_match 'has_secure_password validations: true', m
+    end
+  end
+
+  it 'add helper methods to application controller' do
+    run_generator
+
+    assert_file 'app/controllers/application_controller.rb' do |m|
+      assert_match 'helper_method :current_identity, :identity_signed_in?', m
+      assert_match 'def current_identity', m
+      assert_match 'def identity_signed_in?', m
+      assert_match 'def authenticate!', m
+    end
+  end
+
+  it 'add warden gem to Gemfile' do
+    run_generator
+
+    assert_file 'Gemfile' do |m|
+      assert_match "gem \"warden\", \"~> 1.2.0\"", m
+      assert_match "gem \"bcrypt-ruby\"", m
+    end
+  end
+
+  it 'copy warden configuration' do
+    run_generator
+
+    assert_file 'config/initializers/warden.rb' do |m|
+      assert_match 'manager.default_strategies :database_authentication', m
+      assert_match 'Warden::Manager.serialize_into_session(:identity)', m
+      assert_match 'Warden::Manager.serialize_from_session(:identity)', m
+    end
+  end
+
+  it 'copy warden strategies' do
+    run_generator
+
+    assert_file 'lib/strategies/database_authentication.rb' do |m|
+      assert_match 'def valid?', m
+      assert_match 'authenticate!', m
+    end
+  end
+end
+

data/test/generators/authentication_omniauth_generator_test.rb

@@ -0,0 +1,107 @@
+require 'test_helper'
+require 'generators/authentication/omniauth/omniauth_generator'
+
+describe Authentication::Generators::OmniauthGenerator do
+  include GeneratorsTestHelper
+
+  before do
+    copy_routes
+    copy_locales
+    copy_gemfile
+    copy_application_controller
+    make_migrations_dir
+  end
+
+  it 'copies controllers templates for session' do
+    run_generator
+
+    assert_file 'app/controllers/sessions_controller.rb' do |m|
+      assert_match 'SessionsController', m
+      assert_match 'def create', m
+      assert_match 'def destroy', m
+    end
+  end
+
+  it 'add routes' do
+    run_generator
+
+    assert_file 'config/routes.rb' do |m|
+      assert_match "get 'auth/:provider/callback' => 'sessions#create'", m
+      assert_match "delete '/sessions/destroy' => 'sessions#destroy'", m
+    end
+  end
+
+  it 'generate migration' do
+    Authentication::Generators::OmniauthGenerator.class_eval do
+      def migration_name
+        "create_#{resource_pluralize}.rb"
+      end
+    end
+
+    run_generator
+
+    assert_file 'db/migrate/create_identities.rb' do |m|
+      assert_match 'create_table :identities', m
+      assert_match 't.string :provider', m
+      assert_match 't.string :uid', m
+      assert_match 't.string :name', m
+      assert_match 't.string :email', m
+    end
+
+    assert_file 'app/models/identity.rb' do |m|
+      assert_match 'class Identity < ActiveRecord::Base', m
+      assert_match 'def find_identity(uid, provider)', m
+    end
+  end
+
+  it 'add helper methods to application controller' do
+    run_generator
+
+    assert_file 'app/controllers/application_controller.rb' do |m|
+      assert_match 'helper_method :current_identity, :identity_signed_in?', m
+      assert_match 'def current_identity', m
+      assert_match 'def identity_signed_in?', m
+      assert_match 'def authenticate!', m
+    end
+  end
+
+  it 'add warden gem to Gemfile' do
+    run_generator
+
+    assert_file 'Gemfile' do |m|
+      assert_match "gem \"warden\", \"~> 1.2.0\"", m
+      assert_match "gem \"omniauth\"", m
+    end
+  end
+
+  it 'copy warden configuration' do
+    run_generator
+
+    assert_file 'config/initializers/warden.rb' do |m|
+      assert_match 'manager.default_strategies :oauth_authentication', m
+      assert_match 'Warden::Manager.serialize_into_session(:identity)', m
+      assert_match 'Warden::Manager.serialize_from_session(:identity)', m
+    end
+  end
+
+  it 'copy domain authentication configuration' do
+    run_generator
+
+    assert_file 'config/initializers/authentication_domain.rb'
+  end
+
+  it 'copy omniauth configuration' do
+    run_generator
+
+    assert_file 'config/initializers/omniauth.rb'
+  end
+
+  it 'copy warden strategies' do
+    run_generator
+
+    assert_file 'lib/strategies/oauth_authentication.rb' do |m|
+      assert_match 'def valid?', m
+      assert_match 'authenticate!', m
+    end
+  end
+end

data/test/support/generators_test_helper.rb

@@ -0,0 +1,51 @@
+module GeneratorsTestHelper
+  def self.included(base)
+    base.class_eval do
+      destination File.join(Rails.root, "tmp")
+      setup :prepare_destination
+
+      begin
+        base.tests Rails::Generators.const_get(base.name.sub(/Test$/, ''))
+      rescue
+      end
+    end
+  end
+
+  def copy_routes
+    routes = File.expand_path("../../fixtures/routes.rb", __FILE__)
+    destination = File.join(destination_root, "config")
+
+    copy_file routes, destination
+  end
+
+  def copy_gemfile
+    gemfile = File.expand_path("../../fixtures/Gemfile", __FILE__)
+
+    copy_file gemfile, destination_root
+  end
+
+  def copy_application_controller
+    controller = File.expand_path("../../fixtures/application_controller.rb", __FILE__)
+    destination = File.join(destination_root, 'app', 'controllers')
+
+    copy_file controller, destination
+  end
+
+  def copy_locales
+    controller = File.expand_path("../../fixtures/en.yml", __FILE__)
+    destination = File.join(destination_root, 'config', 'locales')
+
+    copy_file controller, destination
+  end
+
+  def copy_file(file, destination)
+    FileUtils.mkdir_p(destination)
+    FileUtils.cp file, destination
+  end
+
+  def make_migrations_dir
+    destination = File.join(destination_root, 'db', 'migrate')
+
+    FileUtils.mkdir_p(destination)
+  end
+end

data/test/test_helper.rb

@@ -3,6 +3,9 @@ ENV["RAILS_ENV"] = "test"
 
 require File.expand_path("../dummy/config/environment.rb",  __FILE__)
 require "rails/test_help"
+require "minitest/rails"
+require "minitest/focus"
+require "minitest/colorize"
 
 Rails.backtrace_cleaner.remove_silencers!
 
@@ -13,3 +16,20 @@ Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each { |f| require f }
 if ActiveSupport::TestCase.method_defined?(:fixture_path=)
   ActiveSupport::TestCase.fixture_path = File.expand_path("../fixtures", __FILE__)
 end
+
+require 'rails/generators/test_case'
+class Rails::Generators::TestCase
+  register_spec_type(self) do |desc|
+    Class === desc && desc < Rails::Generators::Base
+  end
+
+  register_spec_type(/Generator( ?Test)?\z/i, self)
+
+  def self.determine_default_generator(name)
+    generator = determine_constant_from_test_name(name) do |constant|
+      Class === constant && constant < Rails::Generators::Base
+    end
+    raise NameError.new("Unable to resolve generator for #{name}") if generator.nil?
+    generator
+  end
+end