foobara-postgresql-crud-driver 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 3372860139e831774b1d0dcf96d276475e9c128758383e0d43eaf3b0aa43191e
+  data.tar.gz: 271a58763b91f23402447a8790d454b30c16420d72821d58a9244068a8923b54
+SHA512:
+  metadata.gz: d43d752a836148695674b2886d5a117831867a14498cbcce06b229f725a64d141beaa3bf883e3a0a62cc0abc1aee6d7775529dcff7f693d8754f7e2ba6cb326a
+  data.tar.gz: 2c578ece66b3b77dcb3c63d7ff8368407a9e2b33a0ea4f4502dcc33b4fb9b4249df5fe787b661ddb0959770224f6517b88ccaf9f72c02f3d790efd84cf6f8c54

data/CHANGELOG.md

@@ -0,0 +1,7 @@
+## [0.0.1] - 2025-06-16
+
+- Initial release
+
+## [0.0.0] - 2025-05-08
+
+- Project birth

data/LICENSE-MPL-2.0.txt

@@ -0,0 +1,373 @@
+Mozilla Public License Version 2.0
+==================================
+
+1. Definitions
+--------------
+
+1.1. "Contributor"
+    means each individual or legal entity that creates, contributes to
+    the creation of, or owns Covered Software.
+
+1.2. "Contributor Version"
+    means the combination of the Contributions of others (if any) used
+    by a Contributor and that particular Contributor's Contribution.
+
+1.3. "Contribution"
+    means Covered Software of a particular Contributor.
+
+1.4. "Covered Software"
+    means Source Code Form to which the initial Contributor has attached
+    the notice in Exhibit A, the Executable Form of such Source Code
+    Form, and Modifications of such Source Code Form, in each case
+    including portions thereof.
+
+1.5. "Incompatible With Secondary Licenses"
+    means
+
+    (a) that the initial Contributor has attached the notice described
+        in Exhibit B to the Covered Software; or
+
+    (b) that the Covered Software was made available under the terms of
+        version 1.1 or earlier of the License, but not also under the
+        terms of a Secondary License.
+
+1.6. "Executable Form"
+    means any form of the work other than Source Code Form.
+
+1.7. "Larger Work"
+    means a work that combines Covered Software with other material, in
+    a separate file or files, that is not Covered Software.
+
+1.8. "License"
+    means this document.
+
+1.9. "Licensable"
+    means having the right to grant, to the maximum extent possible,
+    whether at the time of the initial grant or subsequently, any and
+    all of the rights conveyed by this License.
+
+1.10. "Modifications"
+    means any of the following:
+
+    (a) any file in Source Code Form that results from an addition to,
+        deletion from, or modification of the contents of Covered
+        Software; or
+
+    (b) any new file in Source Code Form that contains any Covered
+        Software.
+
+1.11. "Patent Claims" of a Contributor
+    means any patent claim(s), including without limitation, method,
+    process, and apparatus claims, in any patent Licensable by such
+    Contributor that would be infringed, but for the grant of the
+    License, by the making, using, selling, offering for sale, having
+    made, import, or transfer of either its Contributions or its
+    Contributor Version.
+
+1.12. "Secondary License"
+    means either the GNU General Public License, Version 2.0, the GNU
+    Lesser General Public License, Version 2.1, the GNU Affero General
+    Public License, Version 3.0, or any later versions of those
+    licenses.
+
+1.13. "Source Code Form"
+    means the form of the work preferred for making modifications.
+
+1.14. "You" (or "Your")
+    means an individual or a legal entity exercising rights under this
+    License. For legal entities, "You" includes any entity that
+    controls, is controlled by, or is under common control with You. For
+    purposes of this definition, "control" means (a) the power, direct
+    or indirect, to cause the direction or management of such entity,
+    whether by contract or otherwise, or (b) ownership of more than
+    fifty percent (50%) of the outstanding shares or beneficial
+    ownership of such entity.
+
+2. License Grants and Conditions
+--------------------------------
+
+2.1. Grants
+
+Each Contributor hereby grants You a world-wide, royalty-free,
+non-exclusive license:
+
+(a) under intellectual property rights (other than patent or trademark)
+    Licensable by such Contributor to use, reproduce, make available,
+    modify, display, perform, distribute, and otherwise exploit its
+    Contributions, either on an unmodified basis, with Modifications, or
+    as part of a Larger Work; and
+
+(b) under Patent Claims of such Contributor to make, use, sell, offer
+    for sale, have made, import, and otherwise transfer either its
+    Contributions or its Contributor Version.
+
+2.2. Effective Date
+
+The licenses granted in Section 2.1 with respect to any Contribution
+become effective for each Contribution on the date the Contributor first
+distributes such Contribution.
+
+2.3. Limitations on Grant Scope
+
+The licenses granted in this Section 2 are the only rights granted under
+this License. No additional rights or licenses will be implied from the
+distribution or licensing of Covered Software under this License.
+Notwithstanding Section 2.1(b) above, no patent license is granted by a
+Contributor:
+
+(a) for any code that a Contributor has removed from Covered Software;
+    or
+
+(b) for infringements caused by: (i) Your and any other third party's
+    modifications of Covered Software, or (ii) the combination of its
+    Contributions with other software (except as part of its Contributor
+    Version); or
+
+(c) under Patent Claims infringed by Covered Software in the absence of
+    its Contributions.
+
+This License does not grant any rights in the trademarks, service marks,
+or logos of any Contributor (except as may be necessary to comply with
+the notice requirements in Section 3.4).
+
+2.4. Subsequent Licenses
+
+No Contributor makes additional grants as a result of Your choice to
+distribute the Covered Software under a subsequent version of this
+License (see Section 10.2) or under the terms of a Secondary License (if
+permitted under the terms of Section 3.3).
+
+2.5. Representation
+
+Each Contributor represents that the Contributor believes its
+Contributions are its original creation(s) or it has sufficient rights
+to grant the rights to its Contributions conveyed by this License.
+
+2.6. Fair Use
+
+This License is not intended to limit any rights You have under
+applicable copyright doctrines of fair use, fair dealing, or other
+equivalents.
+
+2.7. Conditions
+
+Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted
+in Section 2.1.
+
+3. Responsibilities
+-------------------
+
+3.1. Distribution of Source Form
+
+All distribution of Covered Software in Source Code Form, including any
+Modifications that You create or to which You contribute, must be under
+the terms of this License. You must inform recipients that the Source
+Code Form of the Covered Software is governed by the terms of this
+License, and how they can obtain a copy of this License. You may not
+attempt to alter or restrict the recipients' rights in the Source Code
+Form.
+
+3.2. Distribution of Executable Form
+
+If You distribute Covered Software in Executable Form then:
+
+(a) such Covered Software must also be made available in Source Code
+    Form, as described in Section 3.1, and You must inform recipients of
+    the Executable Form how they can obtain a copy of such Source Code
+    Form by reasonable means in a timely manner, at a charge no more
+    than the cost of distribution to the recipient; and
+
+(b) You may distribute such Executable Form under the terms of this
+    License, or sublicense it under different terms, provided that the
+    license for the Executable Form does not attempt to limit or alter
+    the recipients' rights in the Source Code Form under this License.
+
+3.3. Distribution of a Larger Work
+
+You may create and distribute a Larger Work under terms of Your choice,
+provided that You also comply with the requirements of this License for
+the Covered Software. If the Larger Work is a combination of Covered
+Software with a work governed by one or more Secondary Licenses, and the
+Covered Software is not Incompatible With Secondary Licenses, this
+License permits You to additionally distribute such Covered Software
+under the terms of such Secondary License(s), so that the recipient of
+the Larger Work may, at their option, further distribute the Covered
+Software under the terms of either this License or such Secondary
+License(s).
+
+3.4. Notices
+
+You may not remove or alter the substance of any license notices
+(including copyright notices, patent notices, disclaimers of warranty,
+or limitations of liability) contained within the Source Code Form of
+the Covered Software, except that You may alter any license notices to
+the extent required to remedy known factual inaccuracies.
+
+3.5. Application of Additional Terms
+
+You may choose to offer, and to charge a fee for, warranty, support,
+indemnity or liability obligations to one or more recipients of Covered
+Software. However, You may do so only on Your own behalf, and not on
+behalf of any Contributor. You must make it absolutely clear that any
+such warranty, support, indemnity, or liability obligation is offered by
+You alone, and You hereby agree to indemnify every Contributor for any
+liability incurred by such Contributor as a result of warranty, support,
+indemnity or liability terms You offer. You may include additional
+disclaimers of warranty and limitations of liability specific to any
+jurisdiction.
+
+4. Inability to Comply Due to Statute or Regulation
+---------------------------------------------------
+
+If it is impossible for You to comply with any of the terms of this
+License with respect to some or all of the Covered Software due to
+statute, judicial order, or regulation then You must: (a) comply with
+the terms of this License to the maximum extent possible; and (b)
+describe the limitations and the code they affect. Such description must
+be placed in a text file included with all distributions of the Covered
+Software under this License. Except to the extent prohibited by statute
+or regulation, such description must be sufficiently detailed for a
+recipient of ordinary skill to be able to understand it.
+
+5. Termination
+--------------
+
+5.1. The rights granted under this License will terminate automatically
+if You fail to comply with any of its terms. However, if You become
+compliant, then the rights granted under this License from a particular
+Contributor are reinstated (a) provisionally, unless and until such
+Contributor explicitly and finally terminates Your grants, and (b) on an
+ongoing basis, if such Contributor fails to notify You of the
+non-compliance by some reasonable means prior to 60 days after You have
+come back into compliance. Moreover, Your grants from a particular
+Contributor are reinstated on an ongoing basis if such Contributor
+notifies You of the non-compliance by some reasonable means, this is the
+first time You have received notice of non-compliance with this License
+from such Contributor, and You become compliant prior to 30 days after
+Your receipt of the notice.
+
+5.2. If You initiate litigation against any entity by asserting a patent
+infringement claim (excluding declaratory judgment actions,
+counter-claims, and cross-claims) alleging that a Contributor Version
+directly or indirectly infringes any patent, then the rights granted to
+You by any and all Contributors for the Covered Software under Section
+2.1 of this License shall terminate.
+
+5.3. In the event of termination under Sections 5.1 or 5.2 above, all
+end user license agreements (excluding distributors and resellers) which
+have been validly granted by You or Your distributors under this License
+prior to termination shall survive termination.
+
+************************************************************************
+*                                                                      *
+*  6. Disclaimer of Warranty                                           *
+*  -------------------------                                           *
+*                                                                      *
+*  Covered Software is provided under this License on an "as is"       *
+*  basis, without warranty of any kind, either expressed, implied, or  *
+*  statutory, including, without limitation, warranties that the       *
+*  Covered Software is free of defects, merchantable, fit for a        *
+*  particular purpose or non-infringing. The entire risk as to the     *
+*  quality and performance of the Covered Software is with You.        *
+*  Should any Covered Software prove defective in any respect, You     *
+*  (not any Contributor) assume the cost of any necessary servicing,   *
+*  repair, or correction. This disclaimer of warranty constitutes an   *
+*  essential part of this License. No use of any Covered Software is   *
+*  authorized under this License except under this disclaimer.         *
+*                                                                      *
+************************************************************************
+
+************************************************************************
+*                                                                      *
+*  7. Limitation of Liability                                          *
+*  --------------------------                                          *
+*                                                                      *
+*  Under no circumstances and under no legal theory, whether tort      *
+*  (including negligence), contract, or otherwise, shall any           *
+*  Contributor, or anyone who distributes Covered Software as          *
+*  permitted above, be liable to You for any direct, indirect,         *
+*  special, incidental, or consequential damages of any character      *
+*  including, without limitation, damages for lost profits, loss of    *
+*  goodwill, work stoppage, computer failure or malfunction, or any    *
+*  and all other commercial damages or losses, even if such party      *
+*  shall have been informed of the possibility of such damages. This   *
+*  limitation of liability shall not apply to liability for death or   *
+*  personal injury resulting from such party's negligence to the       *
+*  extent applicable law prohibits such limitation. Some               *
+*  jurisdictions do not allow the exclusion or limitation of           *
+*  incidental or consequential damages, so this exclusion and          *
+*  limitation may not apply to You.                                    *
+*                                                                      *
+************************************************************************
+
+8. Litigation
+-------------
+
+Any litigation relating to this License may be brought only in the
+courts of a jurisdiction where the defendant maintains its principal
+place of business and such litigation shall be governed by laws of that
+jurisdiction, without reference to its conflict-of-law provisions.
+Nothing in this Section shall prevent a party's ability to bring
+cross-claims or counter-claims.
+
+9. Miscellaneous
+----------------
+
+This License represents the complete agreement concerning the subject
+matter hereof. If any provision of this License is held to be
+unenforceable, such provision shall be reformed only to the extent
+necessary to make it enforceable. Any law or regulation which provides
+that the language of a contract shall be construed against the drafter
+shall not be used to construe this License against a Contributor.
+
+10. Versions of the License
+---------------------------
+
+10.1. New Versions
+
+Mozilla Foundation is the license steward. Except as provided in Section
+10.3, no one other than the license steward has the right to modify or
+publish new versions of this License. Each version will be given a
+distinguishing version number.
+
+10.2. Effect of New Versions
+
+You may distribute the Covered Software under the terms of the version
+of the License under which You originally received the Covered Software,
+or under the terms of any subsequent version published by the license
+steward.
+
+10.3. Modified Versions
+
+If you create software not governed by this License, and you want to
+create a new license for such software, you may create and use a
+modified version of this License if you rename the license and remove
+any references to the name of the license steward (except to note that
+such modified license differs from this License).
+
+10.4. Distributing Source Code Form that is Incompatible With Secondary
+Licenses
+
+If You choose to distribute Source Code Form that is Incompatible With
+Secondary Licenses under the terms of this version of the License, the
+notice described in Exhibit B of this License must be attached.
+
+Exhibit A - Source Code Form License Notice
+-------------------------------------------
+
+  This Source Code Form is subject to the terms of the Mozilla Public
+  License, v. 2.0. If a copy of the MPL was not distributed with this
+  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+If it is not possible or desirable to put the notice in a particular
+file, then You may include the notice in a location (such as a LICENSE
+file in a relevant directory) where a recipient would be likely to look
+for such a notice.
+
+You may add additional accurate notices of copyright ownership.
+
+Exhibit B - "Incompatible With Secondary Licenses" Notice
+---------------------------------------------------------
+
+  This Source Code Form is "Incompatible With Secondary Licenses", as
+  defined by the Mozilla Public License, v. 2.0.

data/LICENSE.txt

@@ -0,0 +1,9 @@
+This project is licensed under the MPL-2.0 license.
+
+
+  MPL-2.0 License: LICENSE-MPL-2.0.txt or https://www.mozilla.org/MPL/2.0/
+
+
+This is equivalent to the following SPDX License Expression: MPL-2.0
+
+Copyright (c) 2025 Miles Georgi

data/README.md

@@ -0,0 +1,47 @@
+# Foobara::PostgresqlCrudDriver
+
+Allows persisting Foobara Entity records in a Postgres database
+
+## Installation
+
+Typical stuff: add `gem "foobara-postgresql-crud-driver"` to your Gemfile or .gemspec file. Or even just
+`gem install foobara-postgresql-crud-driver` if just playing with it directly in scripts.
+
+## Usage
+
+You can initialize a Foobara::PostgresqlCrudDriver instance with a URL, a credentials hash, or nothing at all
+which will use the contents of `ENV["DATABASE_URL"]` by default.
+
+```ruby
+crud_driver = Foobara::PostgresqlCrudDriver.new("postgres://testuser:testpassword@localhost/testdb")
+Foobara::Persistence.default_crud_driver = crud_driver
+```
+
+Note: There is not currently a Foobara migrations feature or system. So you will have to either manually
+migrate or leach off of migrations of an existing system such as Rails by using the `foobara-rails-command-connector`
+gem. If you think it would be fun to design/implement a migrations system for Foobara, please get in touch!!
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub
+at https://github.com/foobara/postgresql-crud-driver
+
+## Development
+
+If not using Docker, you can use the system postgres and setup a test database user. On the command line you can:
+
+```
+$ sudo -u postgres psql
+```
+
+And then:
+
+```sql
+CREATE USER testuser WITH PASSWORD 'testpassword' CREATEDB;
+```
+
+TODO: add instructions for use with docker and docker-compose.yml file
+
+## License
+
+This project is licensed under the MPL-2.0 license. Please see LICENSE.txt for more info.

data/lib/foobara/postgresql_crud_driver.rb

@@ -0,0 +1,4 @@
+require "foobara/all"
+require "pg"
+
+Foobara::Util.require_directory "#{__dir__}/../../src"

data/src/connection_pool.rb

@@ -0,0 +1,46 @@
+module Foobara
+  class PostgresqlCrudDriver < Persistence::EntityAttributesCrudDriver
+    class ConnectionPool
+      class TooManyConnectionsError < StandardError; end
+
+      attr_accessor :max_connections, :connection_proc, :available_connections, :in_use_connections
+
+      def initialize(max_connections: 5, &connection_proc)
+        self.max_connections = max_connections
+        self.connection_proc = connection_proc
+        self.available_connections = []
+        self.in_use_connections = []
+      end
+
+      def checkout
+        connection = available_connections.pop
+
+        unless connection
+          if in_use_connections.size >= max_connections
+            # :nocov:
+            raise TooManyConnectionsError, "#{in_use_connections.size} connections in use, cannot allocate more."
+            # :nocov:
+          end
+
+          connection = connection_proc.call
+        end
+
+        in_use_connections << connection
+
+        connection
+      end
+
+      def checkin(connection)
+        in_use_connections.delete(connection)
+        available_connections << connection
+      end
+
+      def clear(&)
+        available_connections.each(&)
+        in_use_connections.each(&)
+        available_connections.clear
+        in_use_connections.clear
+      end
+    end
+  end
+end

data/src/pg_transaction.rb

@@ -0,0 +1,11 @@
+module Foobara
+  class PostgresqlCrudDriver < Persistence::EntityAttributesCrudDriver
+    class PgTransaction
+      attr_accessor :connection
+
+      def initialize(connection)
+        self.connection = connection
+      end
+    end
+  end
+end

data/src/postgresql_crud_driver.rb

@@ -0,0 +1,412 @@
+module Foobara
+  class PostgresqlCrudDriver < Persistence::EntityAttributesCrudDriver
+    class NoDatabaseUrlError < StandardError; end
+    class NoSuchColumnOrTableError < StandardError; end
+
+    class UnsupportedPgColumnTypeError < StandardError
+      def initialize(pg_type, attribute_name, entity_class)
+        # :nocov:
+        super("Unsupported column type #{pg_type} for attribute #{attribute_name} on #{entity_class.entity_name}")
+        # :nocov:
+      end
+    end
+
+    class << self
+      def get_transaction_number
+        @get_transaction_number ||= 0
+        @get_transaction_number += 1
+        if @get_transaction_number > 65_535
+          # TODO: test this codepath somehow
+          # :nocov:
+          @get_transaction_number = 1
+          # :nocov:
+        end
+        @get_transaction_number
+      end
+
+      def has_real_transactions?
+        true
+      end
+
+      def escape_identifier(identifier)
+        if identifier.is_a?(::Symbol)
+          identifier = identifier.to_s
+        end
+
+        "\"#{PG::Connection.escape(identifier).gsub('"', '""')}\""
+      end
+    end
+
+    attr_accessor :creds, :transaction_number
+
+    # We intentionally don't call super because super would open a connection which we do not want
+    # since we want one connection per transaction.
+    # rubocop:disable Lint/MissingSuper
+    def initialize(connection_or_credentials = nil)
+      self.transaction_number = self.class.get_transaction_number
+      self.tables = {}
+      self.creds = case connection_or_credentials
+                   when ::String, ::Hash
+                     connection_or_credentials
+                   when nil
+                     ENV["DATABASE_URL"] || raise(
+                       NoDatabaseUrlError,
+                       'Must set ENV["DATABASE_URL"] if trying to initialize PostgresqlCrudDriver with no arguments'
+                     )
+                   else
+                     # :nocov:
+                     raise ArgumentError, "Expected a String or Hash with connection creds or nil with DATABASE_URL set"
+                     # :nocov:
+                   end
+    end
+    # rubocop:enable Lint/MissingSuper
+
+    def open_connection
+      PG.connect(creds)
+    end
+
+    def connection_pool
+      @connection_pool ||= ConnectionPool.new(max_connections: 5) { open_connection }
+    end
+
+    def open_transaction
+      connection = connection_pool.checkout
+      tx = PgTransaction.new(connection)
+      connection.exec("BEGIN")
+      flush_transaction(tx)
+      tx
+    end
+
+    def flush_transaction(raw_tx)
+      raw_tx.connection.exec("SAVEPOINT foobara_crud_driver_revert_point_#{transaction_number}")
+    end
+
+    def revert_transaction(raw_tx)
+      raw_tx.connection.exec("ROLLBACK TO foobara_crud_driver_revert_point_#{transaction_number}")
+    end
+
+    def rollback_transaction(raw_tx)
+      raw_tx.connection.exec("ROLLBACK")
+      connection_pool.checkin(raw_tx.connection)
+    end
+
+    def commit_transaction(raw_tx)
+      raw_tx.connection.exec("COMMIT")
+      connection_pool.checkin(raw_tx.connection)
+    end
+
+    class Table < Persistence::EntityAttributesCrudDriver::Table
+      ARRAY_ELEMENT_ENCODERS = {
+        "_int4" => PG::TextEncoder::Integer.new,
+        "_int8" => PG::TextEncoder::Integer.new,
+        "_text" => PG::TextEncoder::String.new,
+        "_varchar" => PG::TextEncoder::String.new,
+        "_bool" => PG::TextEncoder::Boolean.new,
+        "_float4" => PG::TextEncoder::Float.new,
+        "_float8" => PG::TextEncoder::Float.new
+      }.freeze
+
+      def all(page_size: nil)
+        Enumerator.new do |yielder|
+          after = nil
+
+          loop do
+            page = fetch_page(after:, page_size:)
+
+            break if page.empty?
+
+            page.each do |record|
+              yielder << record
+            end
+
+            after = DataPath.value_at(entity_class.primary_key_attribute, page.last)
+          end
+        end.lazy
+      end
+
+      def fetch_page(after: nil, page_size: nil, order_by: entity_class.primary_key_attribute)
+        page_size ||= 100
+
+        sql = <<~SQL
+          SELECT *
+          FROM #{PostgresqlCrudDriver.escape_identifier(table_name)}
+        SQL
+
+        if after
+          column, value = normalize_attribute(order_by, after)
+          sql += " WHERE #{column} > #{value} "
+        else
+          column = PostgresqlCrudDriver.escape_identifier(order_by)
+        end
+
+        sql += <<~SQL
+          ORDER BY #{column}
+          LIMIT #{page_size}
+        SQL
+
+        result_data = raw_connection.exec(sql)
+
+        result_data.map do |attributes|
+          restore_pg_attributes(attributes)
+        end
+      end
+
+      def insert(attributes)
+        columns = []
+        values = []
+
+        attributes.each_pair do |attribute_name, value|
+          column_name, value = normalize_attribute(attribute_name, value)
+
+          columns << column_name
+          values << value
+        end
+
+        sql = <<~SQL
+          INSERT INTO #{PostgresqlCrudDriver.escape_identifier(table_name)}
+          (#{columns.join(", ")})
+          VALUES (#{values.join(", ")})
+          RETURNING #{PostgresqlCrudDriver.escape_identifier(entity_class.primary_key_attribute)}
+        SQL
+
+        record_data = raw_connection.exec(sql)
+        record_id = record_data.first[entity_class.primary_key_attribute.to_s]
+
+        find(record_id)
+      rescue PG::UniqueViolation => e
+        raise CannotInsertError.new(record_id, "already exists: #{e.message}")
+      end
+
+      def count
+        sql = <<~SQL
+          SELECT COUNT(*)
+          FROM #{PostgresqlCrudDriver.escape_identifier(table_name)}
+        SQL
+
+        raw_connection.exec(sql).first["count"].to_i
+      end
+
+      def find(record_id)
+        column, value = normalize_attribute(entity_class.primary_key_attribute, record_id)
+
+        sql = <<~SQL
+          SELECT *
+          FROM #{PostgresqlCrudDriver.escape_identifier(table_name)}
+          WHERE #{column} = #{value}
+        SQL
+
+        attributes = raw_connection.exec(sql).first
+
+        restore_pg_attributes(attributes)
+      end
+
+      def update(attributes)
+        record_id = record_id_for(attributes)
+
+        unless exists?(record_id)
+          # :nocov:
+          raise CannotUpdateError.new(record_id, "does not exist")
+          # :nocov:
+        end
+
+        set_expressions = []
+
+        attributes.each_pair do |attribute_name, value|
+          column_name, value = normalize_attribute(attribute_name, value)
+
+          set_expressions << "#{column_name} = #{value}"
+        end
+
+        primary_key, record_id = normalize_attribute(entity_class.primary_key_attribute, record_id)
+
+        sql = <<~SQL
+          UPDATE #{PostgresqlCrudDriver.escape_identifier(table_name)}
+          SET #{set_expressions.join(", ")}
+          WHERE #{primary_key} = #{record_id}
+        SQL
+
+        raw_connection.exec(sql)
+        find(record_id)
+      end
+
+      def hard_delete(record_id)
+        unless exists?(record_id)
+          # :nocov:
+          raise CannotDeleteError.new(record_id, "does not exist")
+          # :nocov:
+        end
+
+        primary_key, record_id = normalize_attribute(entity_class.primary_key_attribute, record_id)
+
+        sql = <<~SQL
+          DELETE FROM #{PostgresqlCrudDriver.escape_identifier(table_name)}
+          WHERE #{primary_key} = #{record_id}
+        SQL
+
+        raw_connection.exec(sql)
+      end
+
+      def hard_delete_all
+        sql = <<~SQL
+          DELETE FROM #{PostgresqlCrudDriver.escape_identifier(table_name)}
+        SQL
+
+        raw_connection.exec(sql)
+      end
+
+      private
+
+      def raw_connection
+        # Feels so weird to grab the transaction this way, hmmm
+        entity_class.current_transaction.raw_tx.connection
+      end
+
+      def restore_pg_attributes(attributes)
+        return attributes if attributes.nil?
+
+        attributes = attributes.to_h do |attribute_name, value|
+          info = column_info[attribute_name.to_s]
+
+          value = case info[:type]
+                  when "integer", "text", "timestamp without time zone"
+                    value
+                  when "jsonb"
+                    if value.nil?
+                      unless info[:is_nullable]
+                        # :nocov:
+                        raise "Unexpected nil value for #{attribute_name}"
+                        # :nocov:
+                      end
+
+                      nil
+                    else
+                      JSON.parse(value)
+                    end
+                  when "ARRAY"
+                    decoder = PG::TextDecoder::Array.new(name: info[:element_type])
+                    decoder.decode(value)
+                  else
+                    # :nocov:
+                    raise UnsupportedPgColumnTypeError.new(info[:type], attribute_name, entity_class)
+                    # :nocov:
+                  end
+
+          [attribute_name, value]
+        end
+
+        # TODO: don't do this?
+        entity_class.attributes_type.process_value!(attributes)
+      end
+
+      def normalize_attribute(attribute_name, value)
+        info = column_info[attribute_name.to_s]
+
+        unless info
+          # :nocov:
+          raise NoSuchColumnOrTableError, "Either #{table_name} or #{table_name}.#{attribute_name} does not exist"
+          # :nocov:
+        end
+
+        pg_type = info[:type]
+        foobara_type = entity_class.model_type.element_types.element_types[attribute_name]
+
+        value = if value.nil?
+                  if info[:is_nullable]
+                    "NULL"
+                  else
+                    # :nocov:
+                    raise "Unexpected nil value for #{attribute_name}"
+                    # :nocov:
+                  end
+                elsif foobara_type.extends?(:number)
+                  case pg_type
+                  when "integer"
+                    value.to_i
+                  else
+                    # :nocov:
+                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                    # :nocov:
+                  end
+                elsif foobara_type.extends?(:string) || foobara_type.extends?(:symbol)
+                  case pg_type
+                  when "text"
+                    "'#{PG::Connection.escape(value.to_s)}'"
+                  else
+                    # :nocov:
+                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                    # :nocov:
+                  end
+                elsif foobara_type.extends?(:datetime)
+                  case pg_type
+                  when "timestamp without time zone"
+                    "'#{PG::Connection.escape(value.inspect)}'"
+                  else
+                    # :nocov:
+                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                    # :nocov:
+                  end
+                elsif foobara_type.extends?(:model) || foobara_type.extends?(:attributes)
+                  case pg_type
+                  when "jsonb"
+                    "'#{PG::Connection.escape(JSON.fast_generate(value))}'"
+                  else
+                    # :nocov:
+                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                    # :nocov:
+                  end
+                elsif foobara_type.extends?(:array)
+                  element_type = foobara_type.element_type
+
+                  if element_type.extends?(:detached_entity)
+                    case pg_type
+                    when "ARRAY"
+                      elements_type = ARRAY_ELEMENT_ENCODERS[info[:element_type]]
+                      array_string = PG::TextEncoder::Array.new(elements_type:).encode(value)
+                      escaped = PG::Connection.escape(array_string)
+
+                      "'#{escaped}'"
+                    else
+                      # :nocov:
+                      raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                      # :nocov:
+                    end
+                  else
+                    # :nocov:
+                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                    # :nocov:
+                  end
+                else
+                  # :nocov:
+                  raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
+                  # :nocov:
+                end
+
+        [PostgresqlCrudDriver.escape_identifier(attribute_name), value]
+      end
+
+      def column_info
+        return @column_info if @column_info
+
+        sql = <<~HERE
+          SELECT column_name, data_type, is_nullable, udt_name
+          FROM information_schema.columns
+          WHERE table_schema = 'public' AND table_name = '#{raw_connection.escape(table_name)}'
+          ORDER BY ordinal_position
+        HERE
+
+        result = raw_connection.exec(sql)
+
+        @column_info = result.to_h do |row|
+          [
+            row["column_name"],
+            {
+              type: row["data_type"],
+              is_nullable: row["is_nullable"] == "YES",
+              element_type: row["udt_name"]
+            }
+          ]
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: foobara-postgresql-crud-driver
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Miles Georgi
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: foobara
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.131
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.131
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+email:
+- azimux@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE-MPL-2.0.txt
+- LICENSE.txt
+- README.md
+- lib/foobara/postgresql_crud_driver.rb
+- src/connection_pool.rb
+- src/pg_transaction.rb
+- src/postgresql_crud_driver.rb
+homepage: https://github.com/foobara/postgresql-crud-driver
+licenses:
+- MPL-2.0
+metadata:
+  homepage_uri: https://github.com/foobara/postgresql-crud-driver
+  source_code_uri: https://github.com/foobara/postgresql-crud-driver
+  changelog_uri: https://github.com/foobara/postgresql-crud-driver/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.4.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Provides a CRUD driver for reading/writing entities with a PostgreSQL database
+test_files: []