activegraph 11.0.0.beta.1-java

data/lib/active_graph/transactions.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+module ActiveGraph
+  module Transactions
+    extend ActiveSupport::Concern
+
+    included do
+      thread_mattr_accessor :explicit_session, :tx
+    end
+
+    class_methods do
+      def session(**session_config)
+        ActiveGraph::Base.driver.session(**session_config) do |session|
+          self.explicit_session = session
+          yield session
+          session.last_bookmark
+        end
+      end
+
+      def write_transaction(**config, &block)
+        send_transaction(:write_transaction, **config, &block)
+      end
+
+      def read_transaction(**config, &block)
+        send_transaction(:read_transaction, **config, &block)
+      end
+
+      alias transaction write_transaction
+
+      private
+
+      def send_transaction(method, **config, &block)
+        return yield tx if tx&.open?
+        return run_transaction_work(explicit_session, method, **config, &block) if explicit_session&.open?
+        driver.session do |session|
+          run_transaction_work(session, method, **config, &block)
+        end
+      end
+
+      def run_transaction_work(session, method, **config, &block)
+        implicit = config.delete(:implicit)
+        session.send(method, **config) do |tx|
+          self.tx = tx
+          block.call(tx).tap do |result|
+            if implicit &&
+              [Core::Result, ActiveGraph::Node::Query::QueryProxy, ActiveGraph::Core::Query]
+                .any?(&result.method(:is_a?))
+              result.store
+            end
+          end
+        end.tap { tx.apply_callbacks }
+      rescue ActiveGraph::Rollback
+        # rollbacks are silently swallowed
+      end
+    end
+  end
+end

data/lib/active_graph/type_converters.rb

@@ -0,0 +1,7 @@
+module ActiveGraph
+  module TypeConverters
+    # This exists for legacy purposes. Some gems that the Neo4jrb project does not own
+    # may contain references to this file. We will remove it once that has been dealt with.
+    include ActiveGraph::Shared::TypeConverters
+  end
+end

data/lib/active_graph/undeclared_properties.rb

@@ -0,0 +1,53 @@
+module ActiveGraph
+  # This mixin allows storage and update of undeclared properties in the included class
+  module UndeclaredProperties
+    extend ActiveSupport::Concern
+
+    included do
+      attr_accessor :undeclared_properties
+    end
+
+    def validate_attributes!(_)
+    end
+
+    def read_attribute(name)
+      respond_to?(name) ? super(name) : read_undeclared_property(name.to_sym)
+    end
+    alias [] read_attribute
+
+    def read_undeclared_property(name)
+      _persisted_obj ? _persisted_obj.properties[name] : (undeclared_properties && undeclared_properties[name])
+    end
+
+    def write_attribute(name, value)
+      if respond_to? "#{name}="
+        super(name, value)
+      else
+        add_undeclared_property(name, value)
+      end
+    end
+    alias []= write_attribute
+
+    def skip_update?
+      super && undeclared_properties.blank?
+    end
+
+    def props_for_create
+      super.merge(undeclared_properties!)
+    end
+
+    def props_for_update
+      super.merge(undeclared_properties!)
+    end
+
+    def undeclared_properties!
+      undeclared_properties || {}
+    ensure
+      self.undeclared_properties = nil
+    end
+
+    def add_undeclared_property(name, value)
+      (self.undeclared_properties ||= {})[name] = value
+    end
+  end
+end

data/lib/active_graph/version.rb

@@ -0,0 +1,3 @@
+module ActiveGraph
+  VERSION = '11.0.0.beta.1'
+end

data/lib/active_graph/wrapper.rb

@@ -0,0 +1,4 @@
+module ActiveGraph
+  module ClassWrapper
+  end
+end

data/lib/rails/generators/active_graph/migration/migration_generator.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'active_graph.rb')
+
+module ActiveGraph
+  module Generators
+    class MigrationGenerator < ::Rails::Generators::NamedBase
+      include ::ActiveGraph::Generators::SourcePathHelper
+      include ::ActiveGraph::Generators::MigrationHelper
+
+      def create_migration_file
+        migration_template 'migration.erb'
+      end
+    end
+  end
+end

data/lib/rails/generators/active_graph/migration/templates/migration.erb

@@ -0,0 +1,9 @@
+class <%= @migration_class_name.underscore.camelize %> < ActiveGraph::Migrations::Base
+  def up
+    <%= @content %>
+  end
+
+  def down
+    raise ActiveGraph::IrreversibleMigration
+  end
+end

data/lib/rails/generators/active_graph/model/model_generator.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+require File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'active_graph.rb')
+
+class ActiveGraph::Generators::ModelGenerator < Rails::Generators::NamedBase #:nodoc:
+  include ::ActiveGraph::Generators::SourcePathHelper
+  include ::ActiveGraph::Generators::MigrationHelper
+
+  argument :attributes, type: :array, default: [], banner: 'field:type field:type'
+
+  check_class_collision
+
+  class_option :timestamps, type: :boolean
+  class_option :parent,     type: :string, desc: 'The parent class for the generated model'
+  class_option :indices,    type: :array,  desc: 'The properties which should be indexed'
+  class_option :has_one,    type: :array,  desc: 'A list of has_one relationships'
+  class_option :has_many,   type: :array,  desc: 'A list of has_many relationships'
+
+  def create_model_file
+    template 'model.erb', File.join('app/models', class_path, "#{singular_name}.rb")
+    migration_template 'migration.erb', 'create_'
+  end
+
+  protected
+
+  def migration?
+    false
+  end
+
+  def timestamps?
+    options[:timestamps]
+  end
+
+  # rubocop:disable Naming/PredicateName
+  def has_many?
+    options[:has_many]
+  end
+
+  def has_many_statements
+    options[:has_many].each_with_object('') do |key, txt|
+      txt << has_x('has_many', key)
+    end
+  end
+
+  def has_one?
+    options[:has_one]
+  end
+
+  def has_x(method, key)
+    to, from = key.split(':')
+    (from ? "\n  #{method}(:#{to}).from(:#{from})\n" : "\n  #{method} :#{to}")
+  end
+
+  def has_one_statements
+    txt = ''
+    options[:has_one].each do |key|
+      txt << has_x('has_one', key)
+    end
+    txt
+  end
+  # rubocop:enable Naming/PredicateName
+
+  def indices?
+    options[:indices]
+  end
+
+  def index_fragment(property)
+    return if !options[:indices] || !options[:indices].include?(property)
+
+    "index :#{property}"
+  end
+
+  def parent?
+    options[:parent]
+  end
+
+  def timestamp_statements
+    '
+  property :created_at, type: DateTime
+  # property :created_on, type: Date
+
+  property :updated_at, type: DateTime
+  # property :updated_on, type: Date
+
+'
+  end
+
+  hook_for :test_framework
+end

data/lib/rails/generators/active_graph/model/templates/migration.erb

@@ -0,0 +1,11 @@
+class Create<%= @migration_class_name.underscore.camelize %> < ActiveGraph::Migrations::Base
+  disable_transactions!
+
+  def up
+    add_constraint :<%= class_name %>, :uuid
+  end
+
+  def down
+    drop_constraint :<%= class_name %>, :uuid
+  end
+end

data/lib/rails/generators/active_graph/model/templates/model.erb

@@ -0,0 +1,15 @@
+class <%= class_name %> <%= parent? ? "#{options[:parent].classify}" : "" %>
+  include ActiveGraph::Node
+<% attributes.reject(&:reference?).each do |attribute| -%>
+  property :<%= attribute.name %><%= ", type: #{attribute.type_class}" unless attribute.type_class == 'any' %><%= "\n  " + index_fragment if index_fragment = index_fragment(attribute.name) %>
+<% end -%>
+
+<% attributes.select(&:reference?).each do |attribute| -%>
+  has_one :in_or_out_or_both, :<%= attribute.name %>, type: :FILL_IN_RELATIONSHIP_TYPE_HERE
+<% end -%>
+
+<%= has_many_statements if has_many? -%>
+<%= has_one_statements if has_one? -%>
+
+<%= timestamp_statements if timestamps? -%>
+end

data/lib/rails/generators/active_graph/upgrade_v8/templates/migration.erb

@@ -0,0 +1,17 @@
+class <%= @migration_class_name.underscore.camelize %> < ActiveGraph::Migrations::Base
+  def up
+<% @schema.each do |type, data|
+     data.each do |element| %>
+    add_<%= type %> <%= element[:label].inspect %>, <%= element[:property_name].inspect %>, force: true
+  <% end
+   end %>
+  end
+
+  def down
+<% @schema.each do |type, data|
+     data.each do |element| %>
+    drop_<%= type %> <%= element[:label].inspect %>, <%= element[:property_name].inspect %>
+  <% end
+   end %>
+  end
+end

data/lib/rails/generators/active_graph/upgrade_v8/upgrade_v8_generator.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'active_graph.rb')
+
+module ActiveGraph
+  module Generators
+    class UpgradeV8Generator < ::Rails::Generators::Base
+      include ::ActiveGraph::Generators::SourcePathHelper
+      include ::ActiveGraph::Generators::MigrationHelper
+
+      def create_upgrade_v8_file
+        @schema = load_all_models_schema!
+        migration_template 'migration.erb'
+      end
+
+      def file_name
+        'upgrate_to_v8'
+      end
+
+      private
+
+      def load_all_models_schema!
+        Rails.application.eager_load!
+        initialize_all_models!
+        ActiveGraph::ModelSchema.legacy_model_schema_informations
+      end
+
+      def initialize_all_models!
+        models = ActiveGraph::Node.loaded_classes
+        models.map(&:ensure_id_property_info!)
+      end
+    end
+  end
+end

data/lib/rails/generators/active_graph_generator.rb

@@ -0,0 +1,121 @@
+# frozen_string_literal: true
+
+require 'rails/generators/named_base'
+require 'rails/generators/active_model'
+
+module ActiveGraph
+  module Generators #:nodoc:
+  end
+end
+
+module ActiveGraph::Generators::MigrationHelper
+  extend ActiveSupport::Concern
+
+  def base_migration_file_name(file_name, prefix = '')
+    "#{prefix}#{file_name.parameterize}"
+  end
+
+  def migration_file_name(file_name, prefix = '')
+    "#{Time.now.utc.strftime('%Y%m%d%H%M%S')}_#{base_migration_file_name(file_name, prefix)}.rb"
+  end
+
+  def migration_lookup_at(dirname)
+    Dir.glob("#{dirname}/[0-9]*_*.rb")
+  end
+
+  # Stolen from https://github.com/rails/rails/blob/30767f980faa2d7a0531774ddf040471db74a23b/railties/lib/rails/generators/migration.rb#L20
+  def existing_migration(dirname, file_name)
+    migration_lookup_at(dirname).grep(/\d+_#{file_name}.rb$/).first
+  end
+
+  # :revoke happens when task is invoked with `rails destroy model ModelName`
+  def migration_template(template_name, prefix = '')
+    real_file_name = case @behavior
+                     when :revoke
+                       existing_migration(
+                         'db/neo4j/migrate',
+                         base_migration_file_name(file_name, prefix)
+                       )
+                     else
+                       migration_file_name(file_name, prefix)
+                     end
+
+    # If running with :revoke and migration doesn't exist, real_file_name = nil
+    return if !real_file_name
+
+    @migration_class_name = file_name.camelize
+
+    # template() method is still run on revoke but it doesn't generate anything
+    # other than a consol message indicating the filepath.
+    # (this appears to be behavior provided by rails)
+    template template_name, File.join('db/neo4j/migrate', real_file_name)
+
+    # On revoke, we need to manually remove the file
+    FileUtils.rm(real_file_name) if @behavior == :revoke
+  end
+end
+
+module ActiveGraph::Generators::SourcePathHelper
+  extend ActiveSupport::Concern
+
+  module ClassMethods
+    def source_root
+      @_neo4j_source_root ||= File.expand_path(File.join(File.dirname(__FILE__),
+                                                         'active_graph', generator_name, 'templates'))
+    end
+  end
+end
+
+
+class ActiveGraph::Generators::ActiveModel < Rails::Generators::ActiveModel #:nodoc:
+  def self.all(klass)
+    "#{klass}.all"
+  end
+
+  def self.find(klass, params = nil)
+    "#{klass}.find(#{params})"
+  end
+
+  def self.build(klass, params = nil)
+    if params
+      "#{klass}.new(#{params})"
+    else
+      "#{klass}.new"
+    end
+  end
+
+  def save
+    "#{name}.save"
+  end
+
+  def update_attributes(params = nil)
+    "#{name}.update_attributes(#{params})"
+  end
+
+  def errors
+    "#{name}.errors"
+  end
+
+  def destroy
+    "#{name}.destroy"
+  end
+end
+
+
+module Rails
+  module Generators
+    class GeneratedAttribute #:nodoc:
+      def type_class
+        case type.to_s.downcase
+        when 'any' then 'any'
+        when 'datetime' then 'DateTime'
+        when 'date' then 'Date'
+        when 'integer', 'number', 'fixnum' then 'Integer'
+        when 'float' then 'Float'
+        else
+          'String'
+        end
+      end
+    end
+  end
+end