images-convert 0.4.1

data/lib/images_convert/configuration.rb

@@ -0,0 +1,303 @@
+# frozen_string_literal: true
+
+require 'yaml'
+require 'fileutils'
+
+module ImagesConvert
+  class Configuration
+    CONFIG_FILE = File.expand_path('~/.images-convert/config.yml')
+
+    def self.load
+      @config ||= begin
+        if File.exist?(CONFIG_FILE)
+          YAML.safe_load(File.read(CONFIG_FILE)) || {}
+        else
+          # 設定ファイルが存在しない場合はデフォルト設定で作成
+          create_default_config
+        end
+      end
+    end
+
+    def self.create_default_config
+      default_config = {
+        'defaults' => {
+          'from' => 'HEIC',
+          'to' => 'JPG',
+          'resize' => '1920x'
+        },
+        'directories' => {
+          'input' => File.expand_path('~/Downloads'),
+          'output' => File.expand_path('~/Desktop')
+        },
+        'selection' => {
+          'latest_mode' => 'mtime'
+        },
+        'waifu2x' => {
+          'scale' => 2,
+          'noise' => 1,
+          'model' => 'photo',
+          'auto_scale_a4' => true,
+          'a4_print' => true,
+          'image_format' => 'webp',
+          'on_error' => 'stop'
+        },
+        'output' => {
+          'image_quality' => 90,
+          'pdf_quality' => 90,
+          'pdf_compression' => 'jpeg',
+          'pdf_density' => 300
+        }
+      }
+
+      FileUtils.mkdir_p(File.dirname(CONFIG_FILE))
+      File.write(CONFIG_FILE, default_config.to_yaml)
+
+      # 初回メッセージ表示（重複防止のため、configコマンドでの表示を抑制）
+      @first_run_message_displayed = true
+      puts "設定ファイルを作成しました: #{CONFIG_FILE}"
+      puts "デフォルト設定:"
+      puts "  入力形式: #{default_config['defaults']['from']}"
+      puts "  出力形式: #{default_config['defaults']['to']}"
+      puts "  リサイズ: #{default_config['defaults']['resize']}"
+      puts "  入力ディレクトリ: #{default_config['directories']['input']}"
+      puts "  出力ディレクトリ: #{default_config['directories']['output']}"
+      puts "  最新判定モード: #{default_config['selection']['latest_mode']}"
+      puts "  waifu2x 既定倍率: #{default_config['waifu2x']['scale']}"
+      puts "  waifu2x ノイズ除去: #{default_config['waifu2x']['noise']}"
+      puts "  waifu2x モデル: #{default_config['waifu2x']['model']}"
+      puts "  出力画像品質: #{default_config['output']['image_quality']}"
+      puts "  PDF 品質: #{default_config['output']['pdf_quality']}"
+      puts "  PDF DPI: #{default_config['output']['pdf_density']}"
+      puts ""
+      puts "設定を変更するには:"
+      puts "  images-convert set FROM TO RESIZE  (または: imgconv set ...)"
+      puts "  images-convert set-dir INPUT_DIR OUTPUT_DIR  (または: imgconv set-dir ...)"
+      puts "  images-convert set-latest-mode MODE (mtime / exif)"
+      puts ""
+
+      default_config
+    end
+
+    def self.reset_to_defaults
+      FileUtils.rm_f(CONFIG_FILE)
+      @config = nil
+      create_default_config
+    end
+
+    def self.default_from
+      cfg = load
+      cfg.dig('defaults', 'from') || 'HEIC'
+    end
+
+    def self.default_to
+      cfg = load
+      cfg.dig('defaults', 'to') || 'JPG'
+    end
+
+    def self.default_resize
+      cfg = load
+      cfg.dig('defaults', 'resize') || '1920x'
+    end
+
+    def self.default_latest_mode
+      cfg = load
+      mode = cfg.dig('selection', 'latest_mode')
+      %w[mtime exif].include?(mode) ? mode : 'mtime'
+    end
+
+    def self.default_waifu2x_scale
+      cfg = load
+      (cfg.dig('waifu2x', 'scale') || 2).to_i
+    end
+
+    def self.default_waifu2x_noise
+      cfg = load
+      (cfg.dig('waifu2x', 'noise') || 1).to_i
+    end
+
+    def self.default_waifu2x_model
+      cfg = load
+      cfg.dig('waifu2x', 'model') || 'photo'
+    end
+
+    def self.default_waifu2x_auto_scale_a4
+      cfg = load
+      cfg.dig('waifu2x', 'auto_scale_a4').nil? ? true : cfg.dig('waifu2x', 'auto_scale_a4')
+    end
+
+    def self.default_waifu2x_a4_print
+      cfg = load
+      cfg.dig('waifu2x', 'a4_print').nil? ? true : cfg.dig('waifu2x', 'a4_print')
+    end
+
+    def self.default_waifu2x_image_format
+      cfg = load
+      cfg.dig('waifu2x', 'image_format') || 'webp'
+    end
+
+    def self.default_waifu2x_on_error
+      cfg = load
+      value = cfg.dig('waifu2x', 'on_error') || 'stop'
+      %w[skip stop].include?(value) ? value : 'stop'
+    end
+
+    def self.default_output_image_quality
+      cfg = load
+      quality = cfg.dig('output', 'image_quality')
+      return 90 if quality.nil?
+      q = quality.to_i
+      q = 100 if q > 100
+      q = 1 if q < 1
+      q
+    end
+
+    def self.default_output_pdf_quality
+      cfg = load
+      quality = cfg.dig('output', 'pdf_quality')
+      return 90 if quality.nil?
+      q = quality.to_i
+      q = 100 if q > 100
+      q = 1 if q < 1
+      q
+    end
+
+    def self.default_output_pdf_compression
+      cfg = load
+      value = cfg.dig('output', 'pdf_compression')
+      return 'jpeg' if value.nil? || value.to_s.strip.empty?
+      normalized = value.to_s.strip.downcase
+      %w[jpeg jpeg2000 zip].include?(normalized) ? normalized : 'jpeg'
+    end
+
+    def self.default_output_pdf_density
+      cfg = load
+      density = cfg.dig('output', 'pdf_density')
+      return 300 if density.nil?
+      d = density.to_i
+      d = 1200 if d > 1200
+      d = 72 if d < 72
+      d
+    end
+
+    def self.default_input_dir
+      cfg = load
+      cfg.dig('directories', 'input') || File.expand_path('~/Downloads')
+    end
+
+    def self.default_output_dir
+      cfg = load
+      cfg.dig('directories', 'output') || File.expand_path('~/Desktop')
+    end
+
+    def self.save_defaults(from:, to:, resize:)
+      config = load
+      config['defaults'] ||= {}
+      config['defaults']['from'] = from
+      config['defaults']['to'] = to
+      config['defaults']['resize'] = resize
+
+      FileUtils.mkdir_p(File.dirname(CONFIG_FILE))
+      File.write(CONFIG_FILE, config.to_yaml)
+      # 変更を即時反映させるためキャッシュを無効化
+      @config = nil
+    end
+
+    def self.save_directories(input_dir:, output_dir:)
+      config = load
+      config['directories'] ||= {}
+      config['directories']['input'] = input_dir
+      config['directories']['output'] = output_dir
+
+      FileUtils.mkdir_p(File.dirname(CONFIG_FILE))
+      File.write(CONFIG_FILE, config.to_yaml)
+      # 変更を即時反映
+      @config = nil
+    end
+
+    def self.save_latest_mode(mode)
+      unless %w[mtime exif].include?(mode)
+        raise ArgumentError, "MODE には mtime または exif を指定してください。"
+      end
+
+      config = load
+      config['selection'] ||= {}
+      config['selection']['latest_mode'] = mode
+
+      FileUtils.mkdir_p(File.dirname(CONFIG_FILE))
+      File.write(CONFIG_FILE, config.to_yaml)
+      @config = nil
+    end
+
+    def self.save_waifu2x_settings(scale:, noise:, model:, auto_scale_a4:, a4_print:, image_format:, on_error:)
+      config = load
+      config['waifu2x'] ||= {}
+      config['waifu2x']['scale'] = scale
+      config['waifu2x']['noise'] = noise
+      config['waifu2x']['model'] = model
+      config['waifu2x']['auto_scale_a4'] = auto_scale_a4
+      config['waifu2x']['a4_print'] = a4_print
+      config['waifu2x']['image_format'] = image_format
+      config['waifu2x']['on_error'] = on_error
+
+      FileUtils.mkdir_p(File.dirname(CONFIG_FILE))
+      File.write(CONFIG_FILE, config.to_yaml)
+      @config = nil
+    end
+
+    def self.save_output_settings(image_quality:, pdf_quality:, pdf_compression: 'jpeg', pdf_density: 300)
+      config = load
+      config['output'] ||= {}
+      config['output']['image_quality'] = image_quality
+      config['output']['pdf_quality'] = pdf_quality
+      config['output']['pdf_compression'] = pdf_compression
+      config['output']['pdf_density'] = pdf_density
+
+      FileUtils.mkdir_p(File.dirname(CONFIG_FILE))
+      File.write(CONFIG_FILE, config.to_yaml)
+      @config = nil
+    end
+
+    def self.show_all_settings
+      config = load
+      {
+        defaults: {
+          from: default_from,
+          to: default_to,
+          resize: default_resize
+        },
+        directories: {
+          input: default_input_dir,
+          output: default_output_dir
+        },
+        selection: {
+          latest_mode: default_latest_mode
+        },
+        output: {
+          image_quality: default_output_image_quality,
+          pdf_quality: default_output_pdf_quality,
+          pdf_compression: default_output_pdf_compression,
+          pdf_density: default_output_pdf_density
+        },
+        waifu2x: {
+          scale: default_waifu2x_scale,
+          noise: default_waifu2x_noise,
+          model: default_waifu2x_model,
+          auto_scale_a4: default_waifu2x_auto_scale_a4,
+          a4_print: default_waifu2x_a4_print,
+          image_format: default_waifu2x_image_format,
+          on_error: default_waifu2x_on_error
+        }
+      }
+    end
+
+    def self.first_run_message_displayed?
+      @first_run_message_displayed ||= false
+    end
+
+    private
+
+    def self.config
+      @config
+    end
+  end
+end

data/lib/images_convert/mini_magick_stub.rb

@@ -0,0 +1,121 @@
+# frozen_string_literal: true
+
+require 'fileutils'
+require 'yaml'
+
+module MiniMagick
+  class Image
+    attr_reader :source_path, :width, :height, :quality_value
+
+    class << self
+      def stub_exif_store
+        @stub_exif_store ||= {}
+      end
+
+      def stub_dimension_store
+        @stub_dimension_store ||= {}
+      end
+
+      def set_stub_exif(path, data)
+        normalized = data.transform_keys(&:to_s)
+        stub_exif_store[path] = normalized
+        File.write(sidecar_path(path), normalized.to_yaml)
+      end
+
+      def set_stub_dimensions(path, width:, height:)
+        stub_dimension_store[path] = { width: width.to_i, height: height.to_i }
+      end
+
+      def reset_stub!
+        @stub_exif_store = {}
+        @stub_dimension_store = {}
+      end
+
+      def sidecar_path(path)
+        "#{path}.exif"
+      end
+
+      def dimension_sidecar_path(path)
+        "#{path}.dims"
+      end
+
+      def ping(path)
+        new(path)
+      end
+    end
+
+    def self.open(path)
+      unless File.exist?(path)
+        raise Errno::ENOENT, "No such file or directory - #{path}"
+      end
+      new(path)
+    end
+
+    def initialize(path)
+      @source_path = path
+      @resize_value = nil
+      @format = File.extname(path).delete('.').to_s.downcase
+      @quality_value = nil
+      dims = self.class.stub_dimension_store[path]
+      unless dims
+        dim_sidecar = self.class.dimension_sidecar_path(path)
+        if File.exist?(dim_sidecar)
+          begin
+            loaded = YAML.safe_load(File.read(dim_sidecar)) || {}
+            width = loaded['width'] || loaded[:width]
+            height = loaded['height'] || loaded[:height]
+            dims = { width: width.to_i, height: height.to_i }
+            self.class.stub_dimension_store[path] = dims
+          rescue StandardError
+            dims = nil
+          end
+        end
+      end
+      @width = dims ? dims[:width].to_i : 0
+      @height = dims ? dims[:height].to_i : 0
+    end
+
+    def resize(value)
+      @resize_value = value
+    end
+
+    def format(value)
+      @format = value.to_s.downcase
+    end
+
+    def quality(value)
+      @quality_value = value.to_i
+    end
+
+    def write(destination)
+      FileUtils.mkdir_p(File.dirname(destination))
+      File.open(destination, 'wb') do |f|
+        f.write("mini_magick_stub\n")
+        f.write("source: #{@source_path}\n")
+        f.write("resize: #{@resize_value}\n")
+        f.write("format: #{@format}\n")
+        f.write("quality: #{@quality_value}\n") if @quality_value
+      end
+    end
+
+    def [](key)
+      data = self.class.stub_exif_store[@source_path]
+      unless data
+        sidecar = self.class.sidecar_path(@source_path)
+        if File.exist?(sidecar)
+          begin
+            loaded = YAML.safe_load(File.read(sidecar)) || {}
+            data = loaded.transform_keys(&:to_s)
+            self.class.stub_exif_store[@source_path] = data
+          rescue StandardError
+            data = {}
+          end
+        else
+          data = {}
+        end
+      end
+
+      data[key.to_s]
+    end
+  end
+end

data/lib/images_convert/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module ImagesConvert
+  VERSION = "0.4.1"
+end

data/lib/images_convert/waifu2x_test_stub.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'fileutils'
+require 'tmpdir'
+
+module Waifu2x
+  module TestInstrumentation
+    class << self
+      def enabled?
+        path = ENV['IMGCONV_TEST_WAIFU2X_CALL_LOG']
+        path && !path.empty?
+      end
+
+      def log(payload)
+        path = ENV['IMGCONV_TEST_WAIFU2X_CALL_LOG']
+        return unless path && !path.empty?
+
+        FileUtils.mkdir_p(File.dirname(path))
+        File.open(path, 'a') do |f|
+          f.write(JSON.dump(payload))
+          f.write("\n")
+        end
+      end
+
+      def fake_paths
+        dir = ENV['IMGCONV_TEST_WAIFU2X_FAKE_ROOT'] || Dir.tmpdir
+        {
+          bin: File.expand_path('waifu2x-ncnn-vulkan', dir),
+          models: File.expand_path('waifu2x-models', dir)
+        }
+      end
+    end
+  end
+end
+
+if defined?(Waifu2x::Processor)
+  class << Waifu2x::Processor
+    unless method_defined?(:process_images_without_test_stub)
+      alias_method :process_images_without_test_stub, :process_images
+    end
+
+    def process_images(input_dir, output_dir, **options)
+      if Waifu2x::TestInstrumentation.enabled?
+        FileUtils.mkdir_p(output_dir) unless Dir.exist?(output_dir)
+        format = options[:image_format].to_s
+        format = 'png' if format.empty?
+        stub_path = File.join(output_dir, "waifu2x_stub_output.#{format}")
+        File.write(stub_path, 'waifu2x-stub')
+        Waifu2x::TestInstrumentation.log({ input_dir: input_dir, output_dir: output_dir, options: options })
+        return
+      end
+
+      process_images_without_test_stub(input_dir, output_dir, **options)
+    end
+  end
+end
+
+if defined?(Waifu2x::Setup)
+  class << Waifu2x::Setup
+    unless method_defined?(:resolve_paths_without_test_stub)
+      alias_method :resolve_paths_without_test_stub, :resolve_paths
+    end
+
+    def resolve_paths(**options)
+      if Waifu2x::TestInstrumentation.enabled?
+        fake = Waifu2x::TestInstrumentation.fake_paths
+        return { bin: fake[:bin], models: fake[:models] }
+      end
+
+      resolve_paths_without_test_stub(**options)
+    end
+  end
+end
+
+if defined?(Waifu2x::PDFBuilder)
+  class << Waifu2x::PDFBuilder
+    unless method_defined?(:create_pdf_without_test_stub)
+      alias_method :create_pdf_without_test_stub, :create_pdf
+    end
+
+    def create_pdf(image_paths, output_path, options = {})
+      if Waifu2x::TestInstrumentation.enabled?
+        FileUtils.mkdir_p(File.dirname(output_path))
+        File.write(output_path, "waifu2x-pdf-stub\n")
+        Waifu2x::TestInstrumentation.log({ pdf_output: output_path, images: image_paths, options: options })
+        return
+      end
+
+      create_pdf_without_test_stub(image_paths, output_path, options)
+    end
+  end
+end