forest_admin_datasource_active_record 1.0.0.pre.beta.82 → 1.0.0.pre.beta.85

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9d83089945909a70c3bc9887643c635bc6ffaba6bc9f595a4dea61e9732dbe6f
-  data.tar.gz: 5bde89f5a4a1c6cbde94c5126150f7e2827b0dec12000d9da8e7c719744bcd4c
+  metadata.gz: 45d904af3bd4dc1f7ad3173832d0114e5514f1ca0c883e188cf90a9738a63434
+  data.tar.gz: 783b4531f3319bd262a5c66b7036a370791e87390580c1593e94ac157334ced3
 SHA512:
-  metadata.gz: 51a055f444cf27aed6ed57083c8c2e449ebb149e32e5bccea8fc9dc5b7cda76a5b82fb3bf461c47ddb129569ecd42dd695f8d75611a9a33a58cf052b35c03945
-  data.tar.gz: 300c2fbabe11109115752dbd7931a2148d00668a4f5675dbf9e5fdd3061f2004bcb74620175a9012768bfe8c15b4126e9ee5f32a844bbd693b78db2b20797821
+  metadata.gz: 42d12bbca254d8da1645dd6c24e6609b7eed0c480bc25c832654378598386913ce851a6a507cdd9adbc9c991a86479b30235e9943c68c0d1789d6a43fc404c3c
+  data.tar.gz: bd48b0696dd2d4093b89c43450002c9e83546df234d597a492b343332b8ebe1501d8f37ea17e238fe9589017ca42f76eeaaa8f26fd545827c5f3d38d2f42c096

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.82"
+  VERSION = "1.0.0-beta.85"
 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.82
+  version: 1.0.0.pre.beta.85
 platform: ruby
 authors:
 - Matthieu
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-12-04 00:00:00.000000000 Z
+date: 2024-12-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord