Allow uploading zipped db (#336)

* allow uploading zipped db

* increase the max upload size to 100mb

Author: naveed-ahmad

app/controllers/segments/dashboard_controller.rb

@@ -8,8 +8,11 @@ def setup_db
       uploaded_file = params[:file]
 
       if uploaded_file.present?
-        save_path = upload_db(uploaded_file)
-        redirect_to segments_dashboard_path, notice: "DB uploaded successfully to #{save_path}."
+        if (save_path = upload_db(uploaded_file))
+          redirect_to segments_dashboard_path, notice: "DB uploaded successfully to #{save_path}."
+        else
+          redirect_to segments_setup_db_path, alert: "Failed to upload the database. Please ensure the file is a valid zip containing a .db file."
+        end
       else
         redirect_to segments_setup_db_path, alert: "Please select a file."
       end
@@ -95,7 +98,7 @@ def authorize_access!
   end
 
   def check_segments_database
-    models = segment_modals
+    models = segment_models
     missing_models = models.reject { |model| model.table_exists? }
 
     if missing_models.any?
@@ -106,7 +109,7 @@ def check_segments_database
     redirect_to segments_setup_db_path, alert: "Segments database is missing" and return
   end
 
-  def segment_modals
+  def segment_models
     [
       SegmentStats::DetectionStat,
       SegmentStats::FailureStat,
@@ -117,13 +120,37 @@ def segment_modals
   end
 
   def upload_db(uploaded_file)
-    save_path = Rails.root.join("tmp", "segments_database.db")
+    require "zip"
+    tmp_dir = Rails.root.join("tmp")
+    db_output_path = tmp_dir.join("segments_database.db")
+    zip_temp_path = tmp_dir.join("segments_upload.zip")
+
+    File.open(zip_temp_path, "wb") { |f| f.write(uploaded_file.read) }
+
+    db_file = nil
+    Zip::File.open(zip_temp_path) do |zip_file|
+      zip_file.each do |entry|
+        if entry.name.ends_with?(".db")
+          entry.extract(db_output_path) { true }
+          db_file = db_output_path
+          break
+        end
+      end
+    end
+
+    File.delete(zip_temp_path) if File.exist?(zip_temp_path)
+
+    if db_file.present?
+      SegmentStats::Base.establish_connection(
+        adapter: 'sqlite3',
+        database: db_file.to_s
+      )
 
-    File.open(save_path, "wb") do |file|
-      file.write(uploaded_file.read)
+      segment_models.each(&:reset_column_information)
     end
 
-    segment_modals.each(&:reset_column_information)
-    save_path
+    db_file
+  rescue StandardError => e
+    # Do nothing
   end
 end

docker/qul.tarteel.ai

@@ -1,5 +1,5 @@
 server {
-  client_max_body_size 20M;
+  client_max_body_size 100M;
   listen   3000;
   server_name qul.tarteel.ai;
   root   /home/app/qul/public;