forest_admin_datasource_active_record 1.0.0.pre.beta.83 → 1.0.0.pre.beta.85

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b525730d06962b290addae1d5f289763cc3caa8924f48bca63a6204ea52f11ec
-  data.tar.gz: b4894584a3618c8a9cbca7f2d828a2ee399a5ca152def20b791aa05099896c1f
+  metadata.gz: 45d904af3bd4dc1f7ad3173832d0114e5514f1ca0c883e188cf90a9738a63434
+  data.tar.gz: 783b4531f3319bd262a5c66b7036a370791e87390580c1593e94ac157334ced3
 SHA512:
-  metadata.gz: 07e5364b10e6158fc7a6dd115821f5f0b5981c12803bbe98b882f8a82614356f12bb7254b58b5a6876b4b3711a99f64e8759cb9afd0b18767a115fd5dbe210ad
-  data.tar.gz: 489b1ee0ec0ea6183f16454ae0ec57a2e2ee6803e233d207317b926f160438c835ef7530ea69cdb67771cb4805bee325f0b89deb6d0ffc0889a723de6ac5bac0
+  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.83"
+  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.83
+  version: 1.0.0.pre.beta.85
 platform: ruby
 authors:
 - Matthieu
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-12-13 00:00:00.000000000 Z
+date: 2024-12-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord