forest_admin_datasource_active_record 1.0.0.pre.beta.83 → 1.0.0.pre.beta.86

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b525730d06962b290addae1d5f289763cc3caa8924f48bca63a6204ea52f11ec
-  data.tar.gz: b4894584a3618c8a9cbca7f2d828a2ee399a5ca152def20b791aa05099896c1f
+  metadata.gz: 7c6bf7b65420c985052b2b2a96214666a0ee3fbb3437ede7f5ae2ab952f78087
+  data.tar.gz: e57a2b8dbe63b4ebe0c25f0bc62c7f6c2d6a7433662da8899e2dc975ce491914
 SHA512:
-  metadata.gz: 07e5364b10e6158fc7a6dd115821f5f0b5981c12803bbe98b882f8a82614356f12bb7254b58b5a6876b4b3711a99f64e8759cb9afd0b18767a115fd5dbe210ad
-  data.tar.gz: 489b1ee0ec0ea6183f16454ae0ec57a2e2ee6803e233d207317b926f160438c835ef7530ea69cdb67771cb4805bee325f0b89deb6d0ffc0889a723de6ac5bac0
+  metadata.gz: 988f20ccd3b4b7a4dc0ba60d35f6a5551cda3bd56eefd35e14189dd31ff4bedeb195049dbdfa9a2a03ed298a745ad0c1af9c1dda98edf75f0bf2dde279e549a5
+  data.tar.gz: f26f7ae357400d10b9ab884f82c169a8c1c43b18aed4c2d6ffe200e9b70835f2faaad822c896d7c7c226366f455de5fa59aa0efcdb2123f08164f13a1b67270d

data/lib/forest_admin_datasource_active_record/datasource.rb

@@ -4,15 +4,56 @@ module ForestAdminDatasourceActiveRecord
   class Datasource < ForestAdminDatasourceToolkit::Datasource
     attr_reader :models
 
-    def initialize(db_config = {}, support_polymorphic_relations: false)
+    def initialize(
+      db_config = {},
+      support_polymorphic_relations: false,
+      live_query_connections: nil
+    )
       super()
       @models = []
       @support_polymorphic_relations = support_polymorphic_relations
       @habtm_models = {}
+      @connection_drivers = {}
+
+      @live_query_connections = if live_query_connections.is_a?(String)
+                                  { live_query_connections => 'primary' }
+                                elsif live_query_connections.is_a?(Hash)
+                                  live_query_connections
+                                else
+                                  {}
+                                end
+
       init_orm(db_config)
       generate
     end
 
+    def execute_native_query(connection_name, query, binds)
+      unless @live_query_connections[connection_name]
+        raise ForestAdminAgent::Http::Exceptions::NotFoundError,
+              "Native query connection '#{connection_name}' is unknown."
+      end
+
+      begin
+        connection = @live_query_connections[connection_name]
+
+        result = ActiveRecord::Base.connects_to(database: { reading: connection.to_sym }).first.connection
+                                   .exec_query(query, "SQL Native Query on '#{connection_name}'", binds)
+
+        ForestAdminDatasourceToolkit::Utils::HashHelper.convert_keys(result.to_a)
+      rescue StandardError => e
+        raise ForestAdminDatasourceToolkit::Exceptions::ForestException,
+              "Error when executing SQL query: '#{e.full_message}'"
+      end
+    end
+
+    def build_binding_symbol(connection_name, binds)
+      if @connection_drivers[@live_query_connections[connection_name]] == 'postgresql'
+        "$#{binds.size + 1}"
+      else
+        '?'
+      end
+    end
+
     private
 
     def generate
@@ -43,6 +84,17 @@ module ForestAdminDatasourceActiveRecord
 
     def init_orm(db_config)
       ActiveRecord::Base.establish_connection(db_config)
+      current_config = ActiveRecord::Base.connection_db_config.env_name
+      configurations = ActiveRecord::Base.configurations
+                                         .configurations
+                                         .group_by(&:env_name)
+                                         .transform_values do |configs|
+        configs.to_h do |config|
+          [config.name, config.adapter]
+        end
+      end.to_h
+
+      @connection_drivers = configurations[current_config]
     end
 
     def build_habtm(model)

data/lib/forest_admin_datasource_active_record/version.rb

@@ -1,3 +1,3 @@
 module ForestAdminDatasourceActiveRecord
-  VERSION = "1.0.0-beta.83"
+  VERSION = "1.0.0-beta.86"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: forest_admin_datasource_active_record
 version: !ruby/object:Gem::Version
-  version: 1.0.0.pre.beta.83
+  version: 1.0.0.pre.beta.86
 platform: ruby
 authors:
 - Matthieu
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-12-13 00:00:00.000000000 Z
+date: 2025-01-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord