Merge pull request #375 from scientist-softserv/i867-make-imagemagick…

…-limits-configurable

i876 - Make memory, map, and disk limits in `im_identify` configurable

lib/iiif_print/configuration.rb

@@ -4,6 +4,12 @@ class Configuration
     attr_writer :after_create_fileset_handler
 
     attr_writer :ingest_queue_name
+
+    # Example settings in the initializer:
+    # config.memory_limit = "512MiB"  # Limit for memory usage
+    # config.map_limit = "1GiB"       # Limit for map usage
+    # config.disk_limit = "20KP"      # Limit for disk usage
+    attr_accessor :memory_limit, :map_limit, :disk_limit
     ##
     # @return [Symbol, Proc]
     def ingest_queue_name

lib/iiif_print/image_tool.rb

@@ -74,7 +74,18 @@ def im_identify_geometry(lines)
 
     # @return [Array<String>] lines of output from imagemagick `identify`
     def im_identify
-      cmd = "identify -limit memory 8GiB -limit map 16GiB -limit disk 50GiB -format 'Geometry: %G\nDepth: %[bit-depth]\nColorspace: %[colorspace]\nAlpha: %A\nMIME type: %m\n' #{path}"
+      memory_limit = IiifPrint.config.memory_limit
+      map_limit = IiifPrint.config.map_limit
+      disk_limit = IiifPrint.config.disk_limit
+
+      cmd = "identify"
+
+      cmd += " -limit memory #{memory_limit}" if memory_limit.present?
+      cmd += " -limit map #{map_limit}" if map_limit.present?
+      cmd += " -limit disk #{disk_limit}" if disk_limit.present?
+
+      cmd += " -format 'Geometry: %G\nDepth: %[bit-depth]\nColorspace: %[colorspace]\nAlpha: %A\nMIME type: %m\n' #{path}"
+
       output, status = Open3.capture2(cmd)
       Rails.logger.info "Identify command output: #{output}"
       Rails.logger.info "Identify command status: #{status}"