1+ # frozen_string_literal: true
2+
13require "sqlite3/errors"
24
35module SQLite3
@@ -58,11 +60,20 @@ def get_enum_pragma(name)
5860 # have duplicate values. See #synchronous, #default_synchronous,
5961 # #temp_store, and #default_temp_store for usage examples.
6062 def set_enum_pragma ( name , mode , enums )
61- match = enums . find { |p | p . find { |i | i . to_s . downcase == mode . to_s . downcase } }
63+ match = if enums . is_a? ( Array )
64+ # maybe deprecate this?
65+ enums . find { |p | p . find { |i | i . to_s . downcase == mode . to_s . downcase } }
66+ elsif mode . is_a? ( String )
67+ enums . fetch ( mode . downcase )
68+ else
69+ mode
70+ end
71+
6272 unless match
6373 raise SQLite3 ::Exception , "unrecognized #{ name } #{ mode . inspect } "
6474 end
65- execute ( "PRAGMA #{ name } ='#{ match . first . upcase } '" )
75+
76+ execute ( "PRAGMA #{ name } ='#{ match } '" )
6677 end
6778
6879 # Returns the value of the given pragma as an integer.
@@ -77,26 +88,57 @@ def set_int_pragma(name, value)
7788 end
7889
7990 # The enumeration of valid synchronous modes.
80- SYNCHRONOUS_MODES = [ [ "full" , 2 ] , [ "normal" , 1 ] , [ "off" , 0 ] ]
91+ SYNCHRONOUS_MODES = {
92+ "full" => 2 ,
93+ "normal" => 1 ,
94+ "off" => 0
95+ } . freeze
8196
8297 # The enumeration of valid temp store modes.
83- TEMP_STORE_MODES = [ [ "default" , 0 ] , [ "file" , 1 ] , [ "memory" , 2 ] ]
98+ TEMP_STORE_MODES = {
99+ "default" => 0 ,
100+ "file" => 1 ,
101+ "memory" => 2
102+ } . freeze
84103
85104 # The enumeration of valid auto vacuum modes.
86- AUTO_VACUUM_MODES = [ [ "none" , 0 ] , [ "full" , 1 ] , [ "incremental" , 2 ] ]
105+ AUTO_VACUUM_MODES = {
106+ "none" => 0 ,
107+ "full" => 1 ,
108+ "incremental" => 2
109+ } . freeze
87110
88111 # The list of valid journaling modes.
89- JOURNAL_MODES = [ [ "delete" ] , [ "truncate" ] , [ "persist" ] , [ "memory" ] ,
90- [ "wal" ] , [ "off" ] ]
112+ JOURNAL_MODES = {
113+ "delete" => "delete" ,
114+ "truncate" => "truncate" ,
115+ "persist" => "persist" ,
116+ "memory" => "memory" ,
117+ "wal" => "wal" ,
118+ "off" => "off"
119+ } . freeze
91120
92121 # The list of valid locking modes.
93- LOCKING_MODES = [ [ "normal" ] , [ "exclusive" ] ]
122+ LOCKING_MODES = {
123+ "normal" => "normal" ,
124+ "exclusive" => "exclusive"
125+ } . freeze
94126
95127 # The list of valid encodings.
96- ENCODINGS = [ [ "utf-8" ] , [ "utf-16" ] , [ "utf-16le" ] , [ "utf-16be" ] ]
128+ ENCODINGS = {
129+ "utf-8" => "utf-8" ,
130+ "utf-16" => "utf-16" ,
131+ "utf-16le" => "utf-16le" ,
132+ "utf-16be" => "utf-16be"
133+ } . freeze
97134
98135 # The list of valid WAL checkpoints.
99- WAL_CHECKPOINTS = [ [ "passive" ] , [ "full" ] , [ "restart" ] , [ "truncate" ] ]
136+ WAL_CHECKPOINTS = {
137+ "passive" => "passive" ,
138+ "full" => "full" ,
139+ "restart" => "restart" ,
140+ "truncate" => "truncate"
141+ } . freeze
100142
101143 def application_id
102144 get_int_pragma "application_id"
@@ -227,7 +269,7 @@ def encoding
227269 end
228270
229271 def encoding = ( mode )
230- set_enum_pragma "encoding" , mode , ENCODINGS
272+ set_string_pragma "encoding" , mode , ENCODINGS
231273 end
232274
233275 def foreign_key_check ( *table , &block ) # :yields: row
@@ -295,7 +337,7 @@ def journal_mode
295337 end
296338
297339 def journal_mode = ( mode )
298- set_enum_pragma "journal_mode" , mode , JOURNAL_MODES
340+ set_string_pragma "journal_mode" , mode , JOURNAL_MODES
299341 end
300342
301343 def journal_size_limit
@@ -319,7 +361,7 @@ def locking_mode
319361 end
320362
321363 def locking_mode = ( mode )
322- set_enum_pragma "locking_mode" , mode , LOCKING_MODES
364+ set_string_pragma "locking_mode" , mode , LOCKING_MODES
323365 end
324366
325367 def max_page_count
@@ -525,7 +567,7 @@ def wal_checkpoint
525567 end
526568
527569 def wal_checkpoint = ( mode )
528- set_enum_pragma "wal_checkpoint" , mode , WAL_CHECKPOINTS
570+ set_string_pragma "wal_checkpoint" , mode , WAL_CHECKPOINTS
529571 end
530572
531573 def writable_schema = ( mode )
@@ -568,6 +610,13 @@ def table_info table
568610
569611 private
570612
613+ def set_string_pragma ( pragma_name , value , valid_values )
614+ valid_values . fetch ( value . to_s . downcase ) {
615+ raise SQLite3 ::Exception , "unrecognized #{ pragma_name } #{ value . inspect } "
616+ }
617+ set_enum_pragma ( pragma_name , value , valid_values )
618+ end
619+
571620 # Compares two version strings
572621 def version_compare ( v1 , v2 )
573622 v1 = v1 . split ( "." ) . map { |i | i . to_i }
0 commit comments