trusty-layouts-extension 1.0.0

data/vendor/plugins/rails_upgrade/lib/gemfile_generator.rb

@@ -0,0 +1,95 @@
+module Rails
+  module Upgrading
+    class GemfileGenerator
+      def generate_new_gemfile
+        if has_environment?
+          generate_gemfile
+        else
+          raise FileNotFoundError, "Can't find environment.rb [config/environment.rb]!"
+        end
+      end
+      
+      def has_environment?
+        File.exists?("config/environment.rb")
+      end
+      
+      def environment_code
+        File.open("config/environment.rb").read
+      end
+      
+      def generate_gemfile
+        environment_file = environment_code
+        
+        # Get each line that starts with config.gem
+        gem_lines = environment_file.split("\n").select {|l| l =~ /^\s*config\.gem/}
+        
+        # Toss those lines to a generator class; the lines are evaluated in the 
+        # context of that instance.
+        config = GemfileGenerator.new
+        config.instance_eval(gem_lines.join("\n"))
+        
+        config.output
+      end
+    end
+    
+    class GemfileGenerator
+      # Creates a target for the config.gem calls
+      def config
+        self
+      end
+      
+      def initialize
+        @gems = []
+      end
+      
+      # Receive a call to add a gem to the list
+      def gem(name, options={})
+        data = {}
+        
+        # Add new keys from old keys
+        data[:require] = options[:lib] if options[:lib]
+        data[:source] = options[:source] if options[:source]
+        
+        version = options[:version]
+        @gems << [name, version, data]
+      end
+      
+      # Generate the Gemfile output
+      def output
+        # Generic preamble, taken from Yehuda Katz's blog
+        preamble = <<STR
+# Edit this Gemfile to bundle your application's dependencies.
+# This preamble is the current preamble for Rails 3 apps; edit as needed.
+source 'http://rubygems.org'
+
+gem 'rails', '3.0.6'
+
+STR
+        preamble + generate_upgraded_code
+      end
+      
+      # Get Gemfile call for all the gems
+      def generate_upgraded_code   
+        code = @gems.map do |name, version, data|
+          version_string = (version ? "'#{version}'" : nil)
+          source = data.delete(:source)
+          
+          data_string = nil
+          unless data.empty?
+            data_string = data.to_a.map {|k, v| ":#{k} => '#{v}'"}.join(", ")
+          end
+          
+          # If we have a source, generate a call to +source+ then output the
+          # gem call.  Otherwise, just generate the gem requirement.
+          if source
+            str = ["'#{name}'", version_string, data_string].compact.join(", ")
+            "source '#{source}'\ngem #{str}"
+          else
+            str = ["'#{name}'", version_string, data_string].compact.join(", ")
+            "gem #{str}"
+          end
+        end.join("\n")
+      end
+    end
+  end
+end

data/vendor/plugins/rails_upgrade/lib/new_configuration_generator.rb

@@ -0,0 +1,59 @@
+require 'active_support/core_ext/string/inflections'
+
+module Rails
+  module Upgrading
+    class NewConfigurationGenerator
+      def generate_new_configurations
+        if has_environment?
+          generate_new_application_rb
+        else
+          raise FileNotFoundError, "Can't find environment.rb [config/environment.rb]!"
+        end
+      end
+
+      def has_environment?
+        File.exists?("config/environment.rb")
+      end
+
+      def environment_code
+        File.open("config/environment.rb").read
+      end
+
+      def generate_new_application_rb
+        environment_file = environment_code
+
+        initializer_code = ""
+        if matches = environment_file.match(/Rails\:\:Initializer\.run do \|config\|\n(.*)\nend/m)
+          initializer_code = matches[1]
+        else
+          raise "There doesn't seem to be a real environment.rb in your app.  Are you sure config/environment.rb has the right contents?"
+        end
+
+        frame = "# Put this in config/application.rb
+require File.expand_path('../boot', __FILE__)
+
+require 'rails/all'
+
+Bundler.require(:default, Rails.env) if defined?(Bundler)
+
+module #{app_name.classify}
+  class Application < Rails::Application
+    config.autoload_paths += [config.root.join('lib')]
+    config.encoding = 'utf-8'
+%s
+  end
+end"
+
+        frame % [indent(initializer_code)]
+      end
+
+      def indent(text)
+        text.split("\n").map {|l| "  #{l}"}.join("\n")
+      end
+
+      def app_name
+        File.basename(Dir.pwd)
+      end
+    end
+  end
+end

data/vendor/plugins/rails_upgrade/lib/routes_upgrader.rb

@@ -0,0 +1,344 @@
+# TODO: Fix formatting on member/collection methods
+
+module Rails
+  module Upgrading
+    module FakeRouter
+      module ActionController
+        module Routing
+          class Routes
+            def self.setup
+              @redrawer = Rails::Upgrading::RouteRedrawer.new
+            end
+
+            def self.redrawer
+              @redrawer
+            end
+
+            def self.draw
+              yield @redrawer
+            end
+          end
+        end
+      end
+    end
+    
+    class RoutesUpgrader
+      def generate_new_routes
+        if has_routes_file?
+          upgrade_routes
+        else
+          raise FileNotFoundError, "Can't find your routes file [config/routes.rb]!"
+        end
+      end
+      
+      def has_routes_file?
+        File.exists?("config/routes.rb")
+      end
+      
+      def routes_code
+        File.read("config/routes.rb")
+      end
+      
+      def upgrade_routes
+        FakeRouter::ActionController::Routing::Routes.setup
+
+        # Read and eval the file; our fake route mapper will capture
+        # the calls to draw routes and generate new route code
+        FakeRouter.module_eval(routes_code)
+
+        # Give the route set to the code generator and get its output
+        generator = RouteGenerator.new(FakeRouter::ActionController::Routing::Routes.redrawer.routes)
+        generator.generate
+      end
+    end
+    
+    class RouteRedrawer
+      attr_accessor :routes
+
+      def self.stack
+        @stack
+      end
+      
+      def self.stack=(val)
+        @stack = val
+      end
+      
+      def initialize
+        @routes = []
+
+        # The old default route was actually two routes; we generate the new style
+        # one only if we haven't generated it for the first old default route.
+        @default_route_generated = false
+        
+        # Setup the stack for parents; used use proper indentation
+        self.class.stack = [@routes]
+      end
+      
+      def root(options)
+        debug "mapping root"
+        @routes << FakeRoute.new("/", options)
+      end
+      
+      def connect(path, options={})
+        debug "connecting #{path}"
+        
+        if (path == ":controller/:action/:id.:format" || path == ":controller/:action/:id")
+          if !@default_route_generated
+            current_parent << FakeRoute.new("/:controller(/:action(/:id))", {:default_route => true})
+          
+            @default_route_generated = true
+          end
+        else
+          current_parent << FakeRoute.new(path, options)
+        end
+      end
+      
+      def resources(*args, &block)
+        _res(FakeResourceRoute, args, &block)
+      end
+
+      def resource(*args, &block)
+        _res(FakeSingletonResourceRoute, args, &block)
+      end
+
+      def _res(klass, args)
+        if args.last.is_a?(Hash)
+          options = args.pop
+          debug "options  #{options.inspect}"
+        end
+
+        args.each do |a|
+          current_parent << klass.new(a, options || {})
+          debug "mapping resources #{current_parent.last.name}"
+
+          if block_given?
+            parent = current_parent.last
+
+            parent = stack(parent) do
+              yield(self)
+            end
+          end
+        end
+      end
+
+      def namespace(name, options = {})
+        debug "mapping namespace #{name}"
+        namespace = FakeNamespace.new(name, options)
+        
+        namespace = stack(namespace) do
+          yield(self)
+        end
+        
+        current_parent << namespace
+      end
+      
+      def method_missing(m, *args)
+        debug "named route: #{m}"
+        current_parent << FakeRoute.new(args.shift, args.pop, m.to_s)
+      end
+      
+      def self.indent
+        ' ' * ((stack.length) * 2)
+      end
+
+    private
+      def debug(txt)
+        puts txt if ENV['DEBUG']
+      end
+      
+      def stack(obj)
+        self.class.stack << obj
+        yield
+        self.class.stack.pop
+      end
+    
+      def current_parent
+        self.class.stack.last
+      end
+    end
+    
+    class RouteObject
+      def indent_lines(code_lines)
+        if code_lines.length > 1
+          code_lines.flatten.map {|l| "#{@indent}#{l.chomp}"}.join("\n") + "\n"
+        else
+          "#{@indent}#{code_lines.shift}"
+        end
+      end
+      
+      def opts_to_string(opts)
+        opts.is_a?(Hash) ? opts.map {|k, v|
+          ":#{k} => " + (v.is_a?(Hash) ? ('{ ' + opts_to_string(v) + ' }') : "#{value_to_string(v)}")
+        }.join(", ") : opts.to_s
+      end
+      
+      def value_to_string(value)
+        case value
+        when Regexp, Symbol, Array then value.inspect
+        when String then "'" + value.to_s + "'"
+        else value.to_s
+        end
+      end
+    end
+    
+    class FakeNamespace < RouteObject
+      attr_accessor :routes, :name, :options
+      
+      def initialize(name, options = {})
+        @routes = []
+        @name, @options = name, options
+        @indent = RouteRedrawer.indent
+      end
+      
+      def to_route_code
+        if !@options.empty?
+          options = ', ' + opts_to_string(@options)
+        else
+          options = ''
+        end
+
+        lines = ["namespace :#{@name}#{options} do", @routes.map {|r| r.to_route_code}, "end"]
+        
+        indent_lines(lines)
+      end
+      
+      def <<(val)
+        @routes << val
+      end
+      
+      def last
+        @routes.last
+      end
+    end
+    
+    class FakeRoute < RouteObject
+      attr_accessor :name, :path, :options
+      
+      def initialize(path, options, name = "")
+        @path = path
+        @options = options || {}
+        @name = name
+        @indent = RouteRedrawer.indent
+      end
+      
+      def to_route_code
+        if @options[:default_route]
+          indent_lines ["match '#{@path}'"]
+        else
+          base = "match '%s' => '%s#%s'"
+          extra_options = []
+        
+          if not name.empty?
+            extra_options << ":as => :#{name}"
+          end
+          
+          if @options[:requirements]
+            @options[:constraints] = @options.delete(:requirements)
+          end
+          
+          if @options[:conditions]
+            @options[:via] = @options.delete(:conditions).delete(:method)
+          end
+
+          @options ||= {}
+          base = (base % [@path, @options.delete(:controller), (@options.delete(:action) || "index")])
+          opts = opts_to_string(@options)
+
+          route_pieces = ([base] + extra_options + [opts])
+          route_pieces.delete("")
+          
+          indent_lines [route_pieces.join(", ")]
+        end
+      end
+    end
+    
+    class FakeResourceRoute < RouteObject
+      attr_accessor :name, :children
+      
+      def initialize(name, options = {})
+        @name = name
+        @children = []
+        @options = options
+        @indent = RouteRedrawer.indent
+      end
+      
+      def to_route_code
+        # preserve :only & :except options
+        copied_options = @options.reject { |k,v| ![:only, :except].member?(k) }
+        unless copied_options.empty?
+          copied_options_str = ", " + copied_options.map { |k, v| "#{k.inspect} => #{v.inspect}" }.join(",")
+        end
+        
+        if !@children.empty? || @options.has_key?(:collection) || @options.has_key?(:member)
+          prefix = ["#{route_method} :#{@name}#{copied_options_str} do"]
+          lines = prefix + custom_methods + [@children.map {|r| r.to_route_code}.join("\n"), "end"]
+        
+          indent_lines(lines)
+        else
+          base = "#{route_method} :%s#{copied_options_str}"
+          indent_lines [base % [@name]]
+        end
+      end
+      
+      def custom_methods
+        collection_code = generate_custom_methods_for(:collection)
+        member_code = generate_custom_methods_for(:member)
+        [collection_code, member_code]
+      end
+      
+      def generate_custom_methods_for(group)
+        return "" unless @options[group]
+        
+        method_code = []
+        
+        RouteRedrawer.stack << self
+        @options[group].each do |name, methods|
+          [*methods].each do |method|
+            method_code << "#{method} :#{name}"
+          end
+        end
+        RouteRedrawer.stack.pop
+        
+        indent_lines ["#{group} do", method_code, "end"].flatten
+      end
+      
+      def route_method
+        "resources"
+      end
+      
+      def <<(val)
+        @children << val
+      end
+      
+      def last
+        @children.last
+      end
+    end
+    
+    class FakeSingletonResourceRoute < FakeResourceRoute
+      def route_method
+        "resource"
+      end
+    end
+    
+    class RouteGenerator
+      def initialize(routes)
+        @routes = routes
+        @new_code = ""
+      end
+      
+      def generate
+        @new_code = @routes.map do |r|
+          r.to_route_code
+        end.join("\n")
+        
+        "#{app_name.underscore.classify}::Application.routes.draw do\n#{@new_code}\nend\n"
+      end
+      
+    private
+      def app_name
+        File.basename(Dir.pwd)
+      end
+    end
+  end
+end