statesman_mongoid 0.1.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12b141e7bcaf520970181af415dcdfe5256b148573fd01e3a4639d6585811f97
-  data.tar.gz: 536f5ef0cd30da45fa1abb42ff78fcb80f1be5362fa57257c515580d32676c9a
+  metadata.gz: 150c388e7dc52a4d1cd5f7dc8aec635bb70d254ecbab0928e151463ac9ca43d1
+  data.tar.gz: 4db6ce07d5bca4cb8bb8f60cd07e0165072d23c977b1b165e7358e150eb580e5
 SHA512:
-  metadata.gz: 836739d3e0bc37bebcb07a509bf4e5609715545b3aa564954999d6da5823a0f2442e0f60ef1756e1ce47163a6d1481e284c1591608b6c8f9376f30293fda4805
-  data.tar.gz: 90d396cfbec77f2e354970e214a7f845e5dac8bbeac546ac2262a41cead4abe11da8d9485e3aa9dcdfc576d538c60b66fbc5059cdddcc8cc9da88a2bac18caed
+  metadata.gz: 172df6b8fe89efababe619552ed5391765b7e18f38998f7cebf7fa00861a05a770436659b23c0ce4f59435643ac32ae5d1f8eb48bcb3a6d19fb01f68a7e65424
+  data.tar.gz: 15460af6a52050507a9d8443ab696ce89a74a9bca276351ee78decaaed5279f476eac3d0ba3f8e6104cf4b5884100b5a43dccf7d4af59d4282048527e523cf55

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    statesman_mongoid (0.1.1)
+    statesman_mongoid (0.3.0)
       statesman (~> 10.0)
 
 GEM

data/README.md

@@ -1,6 +1,10 @@
 # Statesman Mongoid
 
-Restored mongoid adapters for the Statesman gem.
+Restored mongoid adapters for [Statesman](https://github.com/gocardless/statesman).
+
+TODO: Restore generators and test units
+
+NOTE: This is an early salvage of the dropped mongoid support with a refitted queries adapter, a couple of fixes and some manual testing, use at your own risk and feel free to contribute.
 
 ## Installation
 
@@ -20,7 +24,14 @@ Or install it yourself as:
 
 ## Usage
 
-TODO: Write usage instructions here
+TODO: Expand usage instructions 
+
+Follow instructions in the Statesman's doc and replace invoked classes with these ones when approriate:
+``` ruby
+Statesman::Adapters::Mongoid
+Statesman::Adapters::MongoidTransition
+Statesman::Adapters::MongoidQueries
+```
 
 ## Development
 
@@ -30,7 +41,7 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/statesman_mongoid.
+Bug reports and pull requests are welcome on GitHub at https://github.com/oz-tal/statesman_mongoid.
 
 ## License
 

data/lib/statesman/adapters/mongoid.rb

@@ -33,18 +33,25 @@ module Statesman
         @last_transition = nil
       end
 
-      def history(*)
+      def history(force_reload: false)
+        reset if force_reload
         transitions_for_parent.asc(:sort_key)
       end
 
       def last(force_reload: false)
         if force_reload
-          @last_transition = history.last
+          @last_transition = history(force_reload: true).last
         else
           @last_transition ||= history.last
         end
       end
 
+      def reset
+        # Aggressive, but the query cache can't be targeted at a more granular level
+        ::Mongoid::QueryCache.clear_cache
+      end
+
+
       private
 
       def transition_class_hash_fields

data/lib/statesman/adapters/mongoid_queries.rb

@@ -0,0 +1,165 @@
+# frozen_string_literal: true
+
+# Based on the somewhat compatible ActiveRecordQueries at commit 455a21dd74bb7d3b7555de0dd66e2ece9461c22d
+
+module Statesman
+  module Adapters
+    module MongoidQueries
+      def self.check_missing_methods!(base)
+        missing_methods = %i[transition_class initial_state].
+          reject { |m| base.respond_to?(m) }
+        return if missing_methods.none?
+
+        raise NotImplementedError,
+              "#{missing_methods.join(', ')} method(s) should be defined on " \
+              "the model. Alternatively, use the new form of `include " \
+              "Statesman::Adapters::MongoidQueries[" \
+              "transition_class: MyTransition, " \
+              "initial_state: :some_state]`"
+      end
+
+      def self.included(base)
+        check_missing_methods!(base)
+
+        base.include(
+          ClassMethods.new(
+            transition_class: base.transition_class,
+            initial_state: base.initial_state,
+            most_recent_transition_alias: base.try(:most_recent_transition_alias),
+            transition_name: base.try(:transition_name),
+          ),
+        )
+      end
+
+      def self.[](**args)
+        ClassMethods.new(**args)
+      end
+
+      class ClassMethods < Module
+        def initialize(**args)
+          @args = args
+        end
+
+        def included(base)
+          ensure_inheritance(base)
+
+          query_builder = QueryBuilder.new(base, **@args)
+
+          define_in_state(base, query_builder)
+          define_not_in_state(base, query_builder)
+
+          define_method(:reload) do |*a|
+            instance = super(*a)
+            if instance.respond_to?(:state_machine, true)
+              instance.state_machine.reset
+            end
+            instance
+          end
+        end
+
+        private
+
+        def ensure_inheritance(base)
+          klass = self
+          existing_inherited = base.method(:inherited)
+          base.define_singleton_method(:inherited) do |subclass|
+            existing_inherited.call(subclass)
+            subclass.send(:include, klass)
+          end
+        end
+
+        def define_in_state(base, query_builder)
+          base.define_singleton_method(:in_state) do |*states|
+            query_builder.states_where(states.flatten)
+          end
+        end
+
+        def define_not_in_state(base, query_builder)
+          base.define_singleton_method(:not_in_state) do |*states|
+            query_builder.states_where_not(states.flatten)
+          end
+        end
+      end
+
+      class QueryBuilder
+        def initialize(model, transition_class:, initial_state:,
+                       most_recent_transition_alias: nil,
+                       transition_name: nil)
+          @model = model
+          @transition_class = transition_class
+          @initial_state = initial_state
+          @most_recent_transition_alias = most_recent_transition_alias
+          @transition_name = transition_name
+        end
+
+        def states_where(states)
+          ids = aggregate_ids_for_most_recent(states, inclusive_match: true)
+          model.where(_id: { '$in' => ids })
+        end
+
+        def states_where_not(states)
+          ids = aggregate_ids_for_most_recent(states, inclusive_match: false)
+          model.where(_id: { '$in' => ids })
+        end
+
+        def aggregate_ids_for_most_recent(states, inclusive_match: true)
+          aggregation = [
+            # Sort most recent
+            { '$sort': { sort_key: -1 } },
+            # Group by foreign key & get most recent states
+            {
+              '$group': {
+                _id: "$#{model_foreign_key}",
+                to_state: { '$first': '$to_state' },
+                model_foreign_key => { '$first': "$#{model_foreign_key}" },
+              },
+            },
+            # Include/exclude states by provided states
+            { '$match': { to_state: { (inclusive_match ? '$in' : '$nin') => states } } },
+            # Trim response to only the foreign key
+            { '$project': { _id: 0, to_state: 0 } },
+          ]
+
+          # Hit the database and return a flat array of ids
+          transition_class.collection.aggregate(aggregation).pluck(model_foreign_key)
+        end
+
+
+        private
+
+        attr_reader :model, :transition_class, :initial_state
+
+        def transition_name
+          @transition_name || transition_class.collection.name.to_sym
+        end
+
+        def transition_reflection
+          model.reflect_on_all_associations(:has_many).each do |value|
+            return value if value.klass == transition_class
+          end
+
+          raise MissingTransitionAssociation,
+                "Could not find has_many association between #{self.class} " \
+                "and #{transition_class}."
+        end
+
+        def model_primary_key
+          transition_reflection.primary_key
+        end
+
+        def model_foreign_key
+          transition_reflection.foreign_key
+        end
+
+        def model_table
+          transition_reflection.name
+        end
+
+        def most_recent_transition_alias
+          @most_recent_transition_alias ||
+            "most_recent_#{transition_name.to_s.singularize}"
+        end
+      end
+    end
+  end
+end

data/lib/statesman_mongoid/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module StatesmanMongoid
-  VERSION = "0.1.1"
+  VERSION = "0.3.0"
 end

data/lib/statesman_mongoid.rb

@@ -1,8 +1,7 @@
 # frozen_string_literal: true
 
-require_relative "statesman_mongoid/version"
-require_relative "statesman/adapters/mongoid"
-require_relative "statesman/adapters/mongoid_transition"
+# require_relative "statesman_mongoid/version"
+Dir[__dir__ + '/statesman/adapters/*'].each &method(:require)
 
 module StatesmanMongoid
   class Error < StandardError; end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: statesman_mongoid
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.3.0
 platform: ruby
 authors:
 - oz-tal
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-03 00:00:00.000000000 Z
+date: 2023-06-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: statesman
@@ -37,6 +37,7 @@ files:
 - README.md
 - Rakefile
 - lib/statesman/adapters/mongoid.rb
+- lib/statesman/adapters/mongoid_queries.rb
 - lib/statesman/adapters/mongoid_transition.rb
 - lib/statesman_mongoid.rb
 - lib/statesman_mongoid/version.rb