sqlite3 1.7.3 → 2.5.0

data/lib/sqlite3/errors.rb

@@ -1,35 +1,88 @@
-require 'sqlite3/constants'
+require "sqlite3/constants"
 
 module SQLite3
   class Exception < ::StandardError
     # A convenience for accessing the error code for this exception.
     attr_reader :code
+
+    # If the error is associated with a SQL query, this is the query
+    attr_reader :sql
+
+    # If the error is associated with a particular offset in a SQL query, this is the non-negative
+    # offset. If the offset is not available, this will be -1.
+    attr_reader :sql_offset
+
+    def message
+      [super, sql_error].compact.join(":\n")
+    end
+
+    private def sql_error
+      return nil unless @sql
+      return @sql.chomp unless @sql_offset >= 0
+
+      offset = @sql_offset
+      sql.lines.flat_map do |line|
+        if offset >= 0 && line.length > offset
+          blanks = " " * offset
+          offset = -1
+          [line.chomp, blanks + "^"]
+        else
+          offset -= line.length if offset
+          line.chomp
+        end
+      end.join("\n")
+    end
   end
 
   class SQLException < Exception; end
+
   class InternalException < Exception; end
+
   class PermissionException < Exception; end
+
   class AbortException < Exception; end
+
   class BusyException < Exception; end
+
   class LockedException < Exception; end
+
   class MemoryException < Exception; end
+
   class ReadOnlyException < Exception; end
+
   class InterruptException < Exception; end
+
   class IOException < Exception; end
+
   class CorruptException < Exception; end
+
   class NotFoundException < Exception; end
+
   class FullException < Exception; end
+
   class CantOpenException < Exception; end
+
   class ProtocolException < Exception; end
+
   class EmptyException < Exception; end
+
   class SchemaChangedException < Exception; end
+
   class TooBigException < Exception; end
+
   class ConstraintException < Exception; end
+
   class MismatchException < Exception; end
+
   class MisuseException < Exception; end
+
   class UnsupportedException < Exception; end
+
   class AuthorizationException < Exception; end
+
   class FormatException < Exception; end
+
   class RangeException < Exception; end
+
   class NotADatabaseException < Exception; end
 end

data/lib/sqlite3/fork_safety.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require "weakref"
+
+module SQLite3
+  # based on Rails's active_support/fork_tracker.rb
+  module ForkSafety
+    module CoreExt # :nodoc:
+      def _fork
+        pid = super
+        if pid == 0
+          ForkSafety.discard
+        end
+        pid
+      end
+    end
+
+    @databases = []
+    @mutex = Mutex.new
+    @suppress = false
+
+    class << self
+      def hook! # :nodoc:
+        ::Process.singleton_class.prepend(CoreExt)
+      end
+
+      def track(database) # :nodoc:
+        @mutex.synchronize do
+          @databases << WeakRef.new(database)
+        end
+      end
+
+      def discard # :nodoc:
+        warned = @suppress
+        @databases.each do |db|
+          next unless db.weakref_alive?
+
+          begin
+            unless db.closed? || db.readonly?
+              unless warned
+                # If you are here, you may want to read
+                # https://github.com/sparklemotion/sqlite3-ruby/pull/558
+                warn("Writable sqlite database connection(s) were inherited from a forked process. " \
+                     "This is unsafe and the connections are being closed to prevent possible data " \
+                     "corruption. Please close writable sqlite database connections before forking.",
+                  uplevel: 0)
+                warned = true
+              end
+              db.close
+            end
+          rescue WeakRef::RefError
+            # GC may run while this method is executing, and that's OK
+          end
+        end
+        @databases.clear
+      end
+
+      # Call to suppress the fork-related warnings.
+      def suppress_warnings!
+        @suppress = true
+      end
+    end
+  end
+end
+
+SQLite3::ForkSafety.hook!