milkshake 0.0.1

data/lib/milkshake/app.rb

@@ -0,0 +1,190 @@
+
+begin
+  require 'thor'
+rescue LoadError
+  require 'rubygems'
+  retry
+end
+
+require 'pathname'
+
+module Milkshake
+  class App < Thor
+    
+    autoload :Helpers,  'milkshake/app/helpers'
+    autoload :Actions,  'milkshake/app/actions'
+    autoload :Defaults, 'milkshake/app/defaults'
+    
+    include Helpers
+    include Actions
+    extend  Defaults
+    
+    class_option :app, :default => '.',
+      :desc   => 'Path to the rails application',
+      :banner => 'rails_root', :type => :string,
+      :group  => 'Global'
+    
+    class_option :environment, :default => self.default_environment,
+      :desc   => 'Rails environment',
+      :banner => 'env', :type => :string,
+      :group  => 'Global'
+    
+    desc 'info', 'list all loaded gems'
+    def info
+      goto_rails do
+        load_environment!
+        
+        puts "Loaded gems:"
+        data = Milkshake.environment.gemspecs.collect { |gemspec| [gemspec.name, gemspec.version] }
+        shell.print_table(data)
+      end
+    end
+    
+    desc 'create.app PATH', 'create a new rails app.'
+    method_option :'git-data', :default => false,
+      :desc   => 'Initialize git for the shared directory',
+      :type => :boolean,
+      :group  => 'Command'
+    method_option :'shared-data', :default => nil,
+      :desc   => 'path for the shared directory',
+      :banner => 'path', :type => :string,
+      :group  => 'Command'
+    def create_app(path)
+      install_rails! path
+      install_app!
+      if self.options[:'shared-data']
+        self.options[:'git'] = self.options[:'git-data']
+        ensure_extrernalized_data! self.options[:'shared-data']
+      end
+    end
+    
+    desc 'create.gem PATH', 'create a new milkshake gem.'
+    method_option :summary, :default => self.default_summary,
+      :desc   => 'The gem summary',
+      :banner => 'string', :type => :string,
+      :group  => 'Gem'
+    method_option :description, :default => self.default_description,
+      :desc   => 'The gem description',
+      :banner => 'string', :type => :string,
+      :group  => 'Gem'
+    method_option :website, :default => self.default_website,
+      :desc   => 'The gem website',
+      :banner => 'string', :type => :string,
+      :group  => 'Gem'
+    method_option :author, :default => self.default_author,
+      :desc   => 'The gem author',
+      :banner => 'string', :type => :string,
+      :group  => 'Gem'
+    method_option :email, :default => self.default_email,
+      :desc   => 'The gem author\'s email address',
+      :banner => 'string', :type => :string,
+      :group  => 'Gem'
+    method_option :git, :default => false,
+      :desc   => 'Initialize git',
+      :type => :boolean,
+      :group  => 'Command'
+    method_option :'git-data', :default => false,
+      :desc   => 'Initialize git for the shared directory',
+      :type => :boolean,
+      :group  => 'Command'
+    method_option :'shared-data', :default => nil,
+      :desc   => 'path for the shared directory',
+      :banner => 'path', :type => :string,
+      :group  => 'Command'
+    def create_gem(path)
+      name = File.basename(path)
+      assert_valid_gem_name! name
+      install_rails! path
+      install_app!
+      install_gem! name
+      if self.options[:'shared-data']
+        self.options[:'git'] = self.options[:'git-data']
+        ensure_extrernalized_data! self.options[:'shared-data']
+      end
+    end
+    
+    desc 'create.host PATH', 'create a new milkshake host app.'
+    method_option :'git-data', :default => false,
+      :desc   => 'Initialize git for the shared directory',
+      :type => :boolean,
+      :group  => 'Command'
+    method_option :'shared-data', :default => nil,
+      :desc   => 'path for the shared directory',
+      :banner => 'path', :type => :string,
+      :group  => 'Command'
+    def create_host(path)
+      install_rails! path
+      install_app!
+      install_host!
+      if self.options[:'shared-data']
+        self.options[:'git'] = self.options[:'git-data']
+        ensure_extrernalized_data! self.options[:'shared-data']
+      end
+    end
+    rename_task 'create_host' => 'create.host',
+                'create_app'  => 'create.app',
+                'create_gem'  => 'create.gem'
+    
+    desc 'install.app', 'install the milkshake preinitializer'
+    def install_app
+      install_app!
+    end
+    
+    desc 'install.gem NAME', 'make a milkshake plugin'
+    method_option :summary, :default => self.default_summary,
+      :desc   => 'The gem summary',
+      :banner => 'string', :type => :string,
+      :group  => 'Command'
+    method_option :description, :default => self.default_description,
+      :desc   => 'The gem description',
+      :banner => 'string', :type => :string,
+      :group  => 'Command'
+    method_option :website, :default => self.default_website,
+      :desc   => 'The gem website',
+      :banner => 'string', :type => :string,
+      :group  => 'Command'
+    method_option :author, :default => self.default_author,
+      :desc   => 'The gem author',
+      :banner => 'string', :type => :string,
+      :group  => 'Command'
+    method_option :email, :default => self.default_email,
+      :desc   => 'The gem author\'s email address',
+      :banner => 'string', :type => :string,
+      :group  => 'Command'
+    method_option :git, :default => false,
+      :desc   => 'Initialize git',
+      :type => :boolean,
+      :group  => 'Command'
+    def install_gem(name)
+      assert_valid_gem_name! name
+      install_app!
+      install_gem! name
+    end
+    
+    desc 'install.host', 'make a milkshake host app'
+    def install_host
+      install_app!
+      install_host!
+    end
+    rename_task 'install_host' => 'install.host',
+                'install_app'  => 'install.app',
+                'install_gem'  => 'install.gem'
+    
+    desc 'extract.data SHARED_DIR', 'extract all data'
+    method_option :git, :default => false,
+      :desc   => 'Initialize git',
+      :type => :boolean,
+      :group  => 'Command'
+    def extract_data(path)
+      externalize_data! path
+    end
+    rename_task 'extract_data' => 'extract.data'
+    
+    desc 'link.data SHARED_DIR', 'link all data to an existing shared directory'
+    def link_data(path)
+      link_externalized_data! path
+    end
+    rename_task 'link_data' => 'link.data'
+    
+  end
+end

data/lib/milkshake/automagic.rb

@@ -0,0 +1,3 @@
+
+require 'milkshake'
+Milkshake.extender.inject_milkshake!

data/lib/milkshake/cache.rb

@@ -0,0 +1,50 @@
+
+module Milkshake
+  class Cache
+    
+    attr_accessor :path, :entries, :history
+    
+    def initialize(path)
+      @path = path
+      begin
+        File.open(@path, 'r') { |f| @entries = Marshal.load(f.read) }
+        raise 'wrong type' unless Hash === @entries
+      rescue
+        @entries = {}
+      end
+      @history = []
+    end
+    
+    def key(name)
+      if @entries.key?(name.to_s)
+        @entries[name.to_s]
+      elsif block_given?
+        @entries[name.to_s] = yield
+      else
+        nil
+      end
+    end
+    
+    def [](name)
+      @entries[name.to_s]
+    end
+    
+    def []=(name, value)
+      @entries[name.to_s] = value
+    end
+    
+    def reset!
+      @history << @entries
+      @entries = {}
+    end
+    
+    def restore!
+      @entries = @history.pop || {}
+    end
+    
+    def persist!
+      File.open(@path, 'w+') { |f| f.write Marshal.dump(@entries) }
+    end
+    
+  end
+end

data/lib/milkshake/dependency_resolver.rb

@@ -0,0 +1,96 @@
+
+require 'tsort'
+
+module Milkshake
+  class DependencyResolver
+    
+    attr_reader :names, :specs, :gems
+    
+    def self.load_for(gems)
+      dependency_loader = self.new(gems)
+      dependency_loader.add_dependecies!
+      dependency_loader.order_by_dependecies!
+      return dependency_loader
+    end
+    
+    def initialize(gems)
+      specs = gems.collect do |k,r|
+        find_gemspec(k, r)
+      end.compact
+      
+      @names = specs.collect { |spec| spec.name }
+      @specs = specs.inject({}) { |h, spec| h[spec.name] = spec ; h }
+      @gems  = gems
+    end
+    
+    def add_dependecies!
+      @specs.values.each do |spec|
+        add_dependecies_for spec
+      end
+    end
+    
+    def order_by_dependecies!
+      @names = tsort
+    end
+    
+    def each
+      @names.each do |name|
+        yield(@specs[name])
+      end
+    end
+    
+    def reverse_each
+      @names.reverse.each do |name|
+        yield(@specs[name])
+      end
+    end
+    
+  private
+    
+    def find_gemspec(name, options={})
+      requirement = options[:version] || options["version"] || Gem::Requirement.default
+      dep   = Gem::Dependency.new(name, requirement)
+      specs = gemspec_index.search(dep)
+      
+      return nil if specs.empty?
+      
+      specs.sort! do |a,b|
+        b.version <=> a.version
+      end
+      
+      gemspec = specs.first
+      
+      gemspec
+    end
+    
+    def gemspec_index
+      @gemspec_index ||= Gem::SourceIndex.from_installed_gems
+    end
+    
+    include TSort
+    
+    def add_dependecies_for(spec)
+      deps = (spec.rails_dependencies || {})
+      deps.each do |name, options|
+        next if @names.include?(name)
+        
+        gemspec = find_gemspec(name, options)
+        @specs[gemspec.name] = gemspec
+        @names.push(gemspec.name)
+        add_dependecies_for(gemspec)
+      end
+      
+      @gems = @gems.merge(deps)
+    end
+    
+    def tsort_each_node(&block)
+      @names.each(&block)
+    end
+    
+    def tsort_each_child(node, &block)
+      deps = (@specs[node].rails_dependencies || {})
+      deps.keys.each(&block)
+    end
+    
+  end
+end

data/lib/milkshake/environment.rb

@@ -0,0 +1,82 @@
+
+module Milkshake
+  class Environment
+    
+    attr_reader :gems, :options, :cache
+    
+    def self.load(cache, path)
+      if File.exist?(path)
+        new cache, YAML.load_file(path)
+      else
+        new cache
+      end
+    end
+    
+    def initialize(cache, options={})
+      @cache   = cache
+      @options = {'gems' => {}}.merge(options)
+      
+      Milkshake.extender.extend_rubygems!
+      
+      reload!
+    end
+    
+    def reload!
+      resolver  = nil
+      @gems = @cache.key('environment.gems') do
+        resolver ||= DependencyResolver.load_for(@options['gems'])
+        resolver.gems
+      end
+      
+      @gemspecs = @cache.key('environment.gemspecs') do
+        resolver ||= DependencyResolver.load_for(@options['gems'])
+        specs = resolver.specs
+        specs.each { |(name, spec)| spec.store_persistent_load_information! }
+        specs
+      end
+      @gemspecs.each { |(name, spec)| spec.use_persistent_load_information! }
+      
+      @order = @cache.key('environment.gems.order') do
+        resolver ||= DependencyResolver.load_for(@options['gems'])
+        resolver.names
+      end
+      
+      @gemspecs = @ordered_gemspecs = nil
+    end
+    
+    def gemspecs_by_name
+      @gemspecs ||= {}
+    end
+    
+    def gemspecs
+      @ordered_gemspecs ||= @order.collect do |name|
+        gemspecs_by_name[name]
+      end
+    end
+    
+    def gem_dependencies
+      @order.inject([]) do |g, name|
+        g << Rails::GemDependency.new(name, @gems[name])
+      end
+    end
+    
+    def locale_paths
+      @cache.key('environment.locale_paths') do
+        if Gem::Version.new(Rails.version) < Gem::Version.new("2.3.4")
+          locale_paths = self.gemspecs.inject([]) do |paths, gemspec|
+            locale_path = File.join(gemspec.full_gem_path, 'config', 'locales')
+            if File.directory?(locale_path)
+              paths.push(Dir[File.join(locale_path, '*.{rb,yml}')])
+            else
+              paths
+            end
+          end
+          locale_paths.flatten!
+        else
+          []
+        end
+      end
+    end
+    
+  end
+end

data/lib/milkshake/extender.rb

@@ -0,0 +1,58 @@
+
+module Milkshake
+  class Extender
+    
+    def inject_milkshake!
+      if defined?(::PhusionPassenger)
+        extend_rails!
+      else
+        extend_boot!
+      end
+      extend_rubygems!
+    end
+    
+    # for passenger
+    def extend_rails!
+      Object.const_set('Rails', Module.new)
+      r = Object.const_get('Rails')
+      def r.singleton_method_added(m)
+        if (m.to_s == 'boot!') and !@injected_milkshake
+          @injected_milkshake = true
+          k = (class << self ; self ; end)
+          k.send :alias_method, "milkshakeless_#{m}", m
+          k.send :define_method, m do
+            milkshakeless_boot!
+            Milkshake.load!
+            Milkshake.extender.extend_railties!
+          end
+        end
+      end
+    end
+    
+    # for others
+    def extend_boot!
+      include_module Rails::VendorBoot, Milkshake::RailsExtentions::VendorBoot
+      include_module Rails::GemBoot,    Milkshake::RailsExtentions::GemBoot
+    end
+    
+    def extend_railties!
+      include_module Rails::Configuration, Milkshake::RailsExtentions::Configuration
+      include_module Rails::Initializer,   Milkshake::RailsExtentions::Initializer
+    end
+    
+    def extend_frameworks!
+      include_module ActiveRecord::Migrator, Milkshake::RailsExtentions::Migrator
+    end
+    
+    def extend_rubygems!
+      include_module Gem::Specification, Milkshake::RubygemsExtentions::Specification
+    end
+    
+  private
+    
+    def include_module(base, mod)
+      base.send :include, mod
+    end
+    
+  end
+end

data/lib/milkshake/linker.rb

@@ -0,0 +1,64 @@
+
+require 'snapshots'
+
+module Milkshake
+  class Linker
+    
+    attr_reader :environment, :validator, :cache, :current_snapshot
+    
+    def initialize(environment, validator, cache)
+      @environment = environment
+      @validator   = validator
+      @cache       = cache
+    end
+    
+    def link!
+      link_only_once do
+        if validator.relink?
+          @current_snapshot = Snapshots.dump
+          
+          link_public_directories!
+          
+          run_migrations!
+          
+          validator.persist!
+          cache.persist!
+        end
+      end
+    end
+    
+  private
+    
+    def run_migrations!
+      ActiveRecord::Migrator.migrate("db/migrate/", nil)
+    end
+    
+    def link_public_directories!
+      public_vendor_path = File.join(Rails.public_path, 'vendor')
+      FileUtils.rm_rf(public_vendor_path)
+      FileUtils.mkdir_p(public_vendor_path)
+      
+      self.environment.gemspecs.each do |gemspec|
+        public_path = File.join(gemspec.full_gem_path, 'public')
+        if File.directory?(public_path)
+          FileUtils.ln_s(public_path, File.join(public_vendor_path, gemspec.name))
+        end
+      end
+    end
+    
+    def link_only_once
+      lock_path  = Milkshake.cache_file
+      FileUtils.touch(lock_path)
+      lock_file  = File.new(lock_path)
+      acquired   = !!lock_file.flock(File::LOCK_EX | File::LOCK_NB)
+      if acquired
+        yield
+      else
+        lock_file.flock(File::LOCK_EX)
+      end
+    ensure
+      lock_file.flock(File::LOCK_UN)
+    end
+    
+  end
+end

data/lib/milkshake/loader.rb

@@ -0,0 +1,34 @@
+
+module Milkshake
+  class Loader
+    
+    attr_accessor :environment, :cache
+    
+    def initialize(environment, cache)
+      @environment = environment
+      @cache       = cache
+    end
+    
+    def load_gem_initializers!
+      initializers.each do |initializer|
+        load(initializer)
+      end
+    end
+    
+  private
+    
+    def initializers
+      @cache.key('loader.initializers') do
+        relative_path = %w( rails initializers ** *.rb )
+        paths = []
+        self.environment.gemspecs.each do |gemspec|
+          paths.concat Dir.glob(File.join(gemspec.full_gem_path, *relative_path))
+        end
+        paths.concat Dir.glob(File.join(Rails.root, *relative_path))
+        paths
+      end
+    end
+    
+  end
+end
+

data/lib/milkshake/rails_extentions/boot.rb

@@ -0,0 +1,36 @@
+
+module Milkshake
+  module RailsExtentions
+    
+    module VendorBoot
+      def self.included(base)
+        base.module_eval do
+          alias_method :load_initializer_without_milkshake, :load_initializer
+          alias_method :load_initializer, :load_initializer_with_milkshake
+        end
+      end
+      
+      def load_initializer_with_milkshake
+        load_initializer_without_milkshake
+        Milkshake.load!
+        Milkshake.extender.extend_railties!
+      end
+    end
+    
+    module GemBoot
+      def self.included(base)
+        base.module_eval do
+          alias_method :load_initializer_without_milkshake, :load_initializer
+          alias_method :load_initializer, :load_initializer_with_milkshake
+        end
+      end
+      
+      def load_initializer_with_milkshake
+        load_initializer_without_milkshake
+        Milkshake.load!
+        Milkshake.extender.extend_railties!
+      end
+    end
+    
+  end
+end

data/lib/milkshake/rails_extentions/configuration.rb

@@ -0,0 +1,36 @@
+
+module Milkshake
+  module RailsExtentions
+    module Configuration
+      
+      def self.included(base)
+        %w( default_gems default_i18n default_load_paths ).each do |m|
+          base.send :alias_method, "#{m}_without_milkshake", m
+          base.send :alias_method, m, "#{m}_with_milkshake"
+        end
+      end
+      
+      # inject gem dependecies
+      def default_gems_with_milkshake
+        default_gems = default_gems_without_milkshake
+        default_gems.concat(Milkshake.environment.gem_dependencies)
+        default_gems
+      end
+      
+      # inject locales from gem dependecies
+      def default_i18n_with_milkshake
+        default_i18n = default_i18n_without_milkshake
+        default_i18n.load_path.concat(Milkshake.environment.locale_paths)
+        default_i18n.load_path.uniq!
+        default_i18n
+      end
+      
+      # inject fallback application controller
+      def default_load_paths_with_milkshake
+        path = File.expand_path(File.join(File.dirname(__FILE__), *%w( .. rails_fallbacks )))
+        default_load_paths_without_milkshake.push(path)
+      end
+      
+    end
+  end
+end

data/lib/milkshake/rails_extentions/initializer.rb

@@ -0,0 +1,50 @@
+
+module Milkshake
+  module RailsExtentions
+    module Initializer
+      
+      def self.included(base)
+        %w( require_frameworks check_for_unbuilt_gems load_application_initializers process ).each do |m|
+          base.send :alias_method, "#{m}_without_milkshake", m
+          base.send :alias_method, m, "#{m}_with_milkshake"
+        end
+      end
+      
+      def require_frameworks_with_milkshake
+        require_frameworks_without_milkshake
+        Milkshake.extender.extend_frameworks!
+      end
+      
+      def check_for_unbuilt_gems_with_milkshake
+        check_for_unbuilt_gems_without_milkshake
+        
+        Milkshake.linker.link!
+      end
+      
+      def load_application_initializers_with_milkshake
+        Milkshake.loader.load_gem_initializers!
+        Milkshake.persist!
+        
+        load_application_initializers_without_milkshake
+      end
+      
+      def process_with_milkshake
+        process_without_milkshake
+      rescue Exception => e
+        Milkshake.cache.restore!
+        Milkshake.environment.reload!
+        if Milkshake.linker.current_snapshot
+          Snapshots.load(Milkshake.linker.current_snapshot)
+        end
+        
+        if @restore_relink
+          raise e
+        else
+          @restore_relink = true
+          retry
+        end
+      end
+      
+    end
+  end
+end