Reset pk sequences for the models that import IDs (#311)

Author: dcollie2

config/brakeman.ignore

@@ -3,21 +3,21 @@
     {
       "warning_type": "SQL Injection",
       "warning_code": 0,
-      "fingerprint": "b9be381566d68ef14f46a175c99ec4814565753eb616f259e5ddcdd9aff5f819",
+      "fingerprint": "574a24662084f6c06b2ab956a5ab6b850248892cae89bb5fbeb85337895940fc",
       "check_name": "SQL",
       "message": "Possible SQL injection",
       "file": "lib/autorequire/data_import.rb",
-      "line": 148,
+      "line": 155,
       "link": "https://brakemanscanner.org/docs/warning_types/sql_injection/",
-      "code": "ActiveRecord::Base.connection.execute(\"SELECT setval('topics_id_seq', #{((Topic.maximum(:id) or 0) + 1)}, false)\")",
+      "code": "ActiveRecord::Base.connection.execute(\"SELECT setval('#{\"#{model.table_name}_#{model.primary_key}_seq\"}', #{((model.maximum(model.primary_key) or 0) + 1)}, false)\")",
       "render_path": null,
       "location": {
         "type": "method",
         "class": "DataImport",
-        "method": "s(:self).reset_topic_id_starting_value"
+        "method": "s(:self).reset_pk_sequence_for"
       },
-      "user_input": "Topic.maximum(:id)",
-      "confidence": "High",
+      "user_input": "((model.maximum(model.primary_key) or 0) + 1)",
+      "confidence": "Medium",
       "cwe_id": [
         89
       ],

lib/autorequire/data_import.rb

@@ -103,6 +103,8 @@ def self.import_providers
       user.save! if user.new_record?
       puts "User #{user.email} #{user.new_record? ? "created" : "already exists"}"
     end
+
+    reset_pk_sequence_for(Provider)
   end
 
   def self.import_languages
@@ -141,14 +143,17 @@ def self.import_topics
       puts "#{topic.title} #{topic.new_record? ? "created" : "already exists"}"
       topic.save!
     end
-    reset_topic_id_starting_value
+    reset_pk_sequence_for(Topic)
   end
 
-  def self.reset_topic_id_starting_value
-    max_id = Topic.maximum(:id) || 0
+  def self.reset_pk_sequence_for(model)
+    table = model.table_name
+    pk = model.primary_key
+    seq = "#{table}_#{pk}_seq"
+    max_id = model.maximum(pk) || 0
     new_start_value = max_id + 1
-      ActiveRecord::Base.connection.execute("SELECT setval('topics_id_seq', #{new_start_value}, false)")
-    puts "Reset topics ID starting value to #{new_start_value}"
+    ActiveRecord::Base.connection.execute("SELECT setval('#{seq}', #{new_start_value}, false)")
+    puts "Reset #{table} ID starting value to #{new_start_value}"
   end
 
   def self.import_tags
@@ -165,6 +170,8 @@ def self.import_tags
       end
     end
     puts "Tags import completed"
+
+    reset_pk_sequence_for(Tag)
   end
 
   def self.import_topic_tags