aasm 4.5.1 → 5.5.0

data/lib/aasm/state_machine_store.rb

@@ -0,0 +1,76 @@
+require 'concurrent'
+module AASM
+  class StateMachineStore
+    @stores = Concurrent::Map.new
+
+    class << self
+      def stores
+        @stores
+      end
+
+      # do not overwrite existing state machines, which could have been created by
+      # inheritance, see AASM::ClassMethods method inherited
+      def register(klass, overwrite = false, state_machine = nil)
+        raise "Cannot register #{klass}" unless klass.is_a?(Class)
+
+        case name = template = overwrite
+          when FalseClass then stores[klass.to_s] ||= new
+          when TrueClass then stores[klass.to_s] = new
+          when Class then stores[klass.to_s] = stores[template.to_s].clone
+          when Symbol then stores[klass.to_s].register(name, state_machine)
+          when String then stores[klass.to_s].register(name, state_machine)
+          else raise "Don't know what to do with #{overwrite}"
+        end
+      end
+      alias_method :[]=, :register
+
+      def fetch(klass, fallback = nil)
+        stores[klass.to_s] || fallback && begin
+          match = klass.ancestors.find do |ancestor|
+            ancestor.include? AASM and stores[ancestor.to_s]
+          end
+
+          stores[match.to_s]
+        end
+      end
+      alias_method :[], :fetch
+
+      def unregister(klass)
+        stores.delete(klass.to_s)
+      end
+    end
+
+    def initialize
+      @machines = Concurrent::Map.new
+    end
+
+    def clone
+      StateMachineStore.new.tap do |store|
+        @machines.each_pair do |name, machine|
+          store.register(name, machine.clone)
+        end
+      end
+    end
+
+    def machine(name)
+      @machines[name.to_s]
+    end
+    alias_method :[], :machine
+
+    def machine_names
+      @machines.keys
+    end
+    alias_method :keys, :machine_names
+
+    def register(name, machine, force = false)
+      raise "Cannot use #{name.inspect} for machine name" unless name.is_a?(Symbol) or name.is_a?(String)
+      raise "Cannot use #{machine.inspect} as a machine" unless machine.is_a?(AASM::StateMachine)
+
+      if force
+        @machines[name.to_s] = machine
+      else
+        @machines[name.to_s] ||= machine
+      end
+    end
+  end
+end

data/lib/aasm/version.rb

@@ -1,3 +1,3 @@
 module AASM
-  VERSION = "4.5.1"
+  VERSION = "5.5.0"
 end

data/lib/aasm.rb

@@ -1,5 +1,3 @@
-require 'ostruct'
-
 require 'aasm/version'
 require 'aasm/errors'
 require 'aasm/configuration'
@@ -9,7 +7,15 @@ require 'aasm/instance_base'
 require 'aasm/core/transition'
 require 'aasm/core/event'
 require 'aasm/core/state'
+require 'aasm/core/invoker'
+require 'aasm/core/invokers/base_invoker'
+require 'aasm/core/invokers/class_invoker'
+require 'aasm/core/invokers/literal_invoker'
+require 'aasm/core/invokers/proc_invoker'
 require 'aasm/localizer'
+require 'aasm/state_machine_store'
 require 'aasm/state_machine'
 require 'aasm/persistence'
+require 'aasm/persistence/base'
+require 'aasm/persistence/plain_persistence'
 require 'aasm/aasm'

data/lib/generators/aasm/aasm_generator.rb

@@ -0,0 +1,16 @@
+require 'rails/generators/named_base'
+
+module AASM
+  module Generators
+    class AASMGenerator < Rails::Generators::NamedBase
+      namespace "aasm"
+      argument :column_name, type: :string, default: 'aasm_state'
+
+      desc "Generates a model with the given NAME (if one does not exist) with aasm " <<
+      "block and migration to add aasm_state column."
+
+      hook_for :orm
+
+    end
+  end
+end

data/lib/generators/aasm/orm_helpers.rb

@@ -0,0 +1,41 @@
+module AASM
+  module Generators
+    module OrmHelpers
+
+      def model_contents
+        if column_name == 'aasm_state'
+<<RUBY
+  include AASM
+
+  aasm do
+  end
+RUBY
+        else
+<<RUBY
+  include AASM
+
+  aasm :column => '#{column_name}' do
+  end
+RUBY
+        end
+      end
+
+      private
+
+      def column_exists?
+        table_name.singularize.humanize.constantize.column_names.include?(column_name.to_s)
+      rescue NameError
+        false
+      end
+
+      def model_exists?
+        File.exist?(File.join(destination_root, model_path))
+      end
+
+      def model_path
+        @model_path ||= File.join("app", "models", "#{file_path}.rb")
+      end
+
+    end
+  end
+end

data/lib/generators/active_record/aasm_generator.rb

@@ -0,0 +1,40 @@
+require 'rails/generators/active_record'
+require 'generators/aasm/orm_helpers'
+
+module ActiveRecord
+  module Generators
+    class AASMGenerator < ActiveRecord::Generators::Base
+      include AASM::Generators::OrmHelpers
+      namespace "active_record:aasm"
+      argument :column_name, type: :string, default: 'aasm_state'
+
+      source_root File.expand_path("../templates", __FILE__)
+
+      def copy_aasm_migration
+        if column_exists?
+          puts "Both model and column exists"
+        elsif model_exists?
+          migration_template "migration_existing.rb", "db/migrate/add_#{column_name}_to_#{table_name}.rb"
+        else
+          migration_template "migration.rb", "db/migrate/aasm_create_#{table_name}.rb"
+        end
+      end
+
+      def generate_model
+        invoke "active_record:model", [name], migration: false unless model_exists?
+      end
+
+      def inject_aasm_content
+        content = model_contents
+
+        class_path = if namespaced?
+                       class_name.to_s.split("::")
+                     else
+                       [class_name]
+                     end
+        inject_into_class(model_path, class_path.last, content) if model_exists?
+      end
+
+    end
+  end
+end

data/lib/generators/active_record/templates/migration.rb

@@ -0,0 +1,8 @@
+class AasmCreate<%= table_name.camelize %> < ActiveRecord::Migration[<%=  ActiveRecord::VERSION::STRING.to_f %>]
+  def change
+    create_table(:<%= table_name %>) do |t|
+      t.string :<%= column_name %>
+      t.timestamps null: false 
+    end
+  end
+end

data/lib/generators/active_record/templates/migration_existing.rb

@@ -0,0 +1,5 @@
+class Add<%= column_name.camelize %>To<%= table_name.camelize %> < ActiveRecord::Migration[<%=  ActiveRecord::VERSION::STRING.to_f %>]
+  def change
+    add_column :<%= table_name %>, :<%= column_name %>, :string
+  end
+end

data/lib/generators/mongoid/aasm_generator.rb

@@ -0,0 +1,28 @@
+require 'rails/generators/named_base'
+require 'generators/aasm/orm_helpers'
+
+module Mongoid
+  module Generators
+    class AASMGenerator < Rails::Generators::NamedBase
+      include AASM::Generators::OrmHelpers
+      namespace "mongoid:aasm"
+      argument :column_name, type: :string, default: 'aasm_state'
+
+      def generate_model
+        invoke "mongoid:model", [name] unless model_exists?
+      end
+
+      def inject_aasm_content
+        inject_into_file model_path, model_contents, after: "include Mongoid::Document\n" if model_exists?
+      end
+
+      def inject_field_types
+        inject_into_file model_path, migration_data, after: "include Mongoid::Document\n" if model_exists?
+      end
+
+      def migration_data
+        "  field :#{column_name}"
+      end
+    end
+  end
+end

data/lib/generators/nobrainer/aasm_generator.rb

@@ -0,0 +1,28 @@
+require 'rails/generators/named_base'
+require 'generators/aasm/orm_helpers'
+
+module NoBrainer
+  module Generators
+    class AASMGenerator < Rails::Generators::NamedBase
+      include AASM::Generators::OrmHelpers
+      namespace 'nobrainer:aasm'
+      argument :column_name, type: :string, default: 'aasm_state'
+
+      def generate_model
+        invoke 'nobrainer:model', [name] unless model_exists?
+      end
+
+      def inject_aasm_content
+        inject_into_file model_path, model_contents, after: "include NoBrainer::Document::Timestamps\n" if model_exists?
+      end
+
+      def inject_field_types
+        inject_into_file model_path, migration_data, after: "include NoBrainer::Document::Timestamps\n" if model_exists?
+      end
+
+      def migration_data
+        "  field :#{column_name}"
+      end
+    end
+  end
+end

data/lib/motion-aasm.rb

@@ -0,0 +1,37 @@
+unless defined?(Motion::Project::App)
+  raise "This must be required from within a RubyMotion Rakefile"
+end
+
+file_dependencies = {
+  'aasm/aasm.rb' => ['aasm/persistence.rb'],
+  'aasm/persistence.rb' => ['aasm/persistence/plain_persistence.rb', 'aasm/persistence/core_data_query_persistence.rb'],
+  'aasm/persistence/base.rb' => ['aasm/base.rb'],
+  'aasm/persistence/core_data_query_persistence.rb' => ['aasm/persistence/base.rb']
+}
+
+exclude_files = [
+  'aasm/rspec.*',
+  'aasm/minitest.*',
+  'aasm/minitest_spec.*',
+  'aasm/persistence/active_record_persistence.rb',
+  'aasm/persistence/dynamoid_persistence.rb',
+  'aasm/persistence/mongoid_persistence.rb',
+  'aasm/persistence/no_brainer_persistence.rb',
+  'aasm/persistence/sequel_persistence.rb',
+  'aasm/persistence/redis_persistence.rb'
+]
+
+Motion::Project::App.setup do |app|
+  parent = File.expand_path File.dirname(__FILE__)
+
+  app.files.unshift Dir.glob(File.join(parent, "aasm/**/*.rb")).reject { |file| exclude_files.any? { |exclude| file.match(exclude) } }
+
+  app.files_dependencies file_dependencies.inject({}, &->(file_dependencies, (file, *dependencies)) do
+    file = File.join(parent, file)
+    dependencies = dependencies.flatten(1).map do |dependency|
+      File.join(parent, dependency)
+    end
+
+    file_dependencies.merge({ file => dependencies })
+  end)
+end