helper.rb

# frozen_string_literal: true

require 'config'
require 'digest'
require 'fileutils'
require 'logger'
require 'mini_magick'
require 'optparse'
require 'parallel'
require 'sinatra/activerecord'
require 'streamio-ffmpeg'
require 'timeout'
require 'yaml'

require_relative 'lib/helpers'
require_relative 'lib/folders'
require_relative 'lib/thumbs'
require_relative 'lib/images'
require_relative 'lib/models'

Config.load_and_set_settings("#{File.dirname(__FILE__)}/config/settings.yml")

def logger
  @logger ||= Logger.new($stdout)
end

ActiveRecord::Base.establish_connection(
  adapter: Settings.db_adapter,
  database: Settings.db_name,
  password: Settings.db_password,
  username: Settings.db_username,
  host: Settings.db_host,
  encoding: Settings.db_ecnoding,
  collation: Settings.db_collation,
  pool: Settings.db_pool
)

image_root   = Settings.originals_path
thumb_target = Settings.thumb_target
extensions   = Settings.image_extentions + Settings.movie_extentions

@options = {}

OptionParser.new do |opts|
  opts.banner = 'Usage: helper.rb [options]'

  opts.on('--clean-thumbs', TrueClass, 'Clean obsolete Thumbs') do |e|
    @options[:clean_thumbs] = e.nil? ? true : e
  end

  opts.on('--clean-files', TrueClass, 'Clean obsolete Files') do |e|
    @options[:clean_files] = e.nil? ? true : e
  end

  opts.on('--clean-folders', TrueClass, 'Clean obsolete Folders') do |e|
    @options[:clean_folders] = e.nil? ? true : e
  end

  opts.on('--index', TrueClass, 'Start Indexing') do |e|
    @options[:index] = e.nil? ? true : e
  end

  opts.on('--find-duplicates', TrueClass, 'Find Duplicates') do |e|
    @options[:find_duplicates] = e.nil? ? true : e
  end

  opts.on('--ar-logger', TrueClass, 'Activate AcitveRecord Logger') do |e|
    @options[:ar_logger] = e.nil? ? true : e
  end

  opts.on('--add_new_fields', TrueClass, 'add new fields') do |e|
    @options[:add_new_fields] = e.nil? ? true : e
  end

  opts.on('--add_mtime_and_ctime', TrueClass, 'add mtime and ctime') do |e|
    @options[:add_mtime_and_ctime] = e.nil? ? true : e
  end
end.parse!

ActiveRecord::Base.logger = nil unless @options[:ar_logger]

build_index(image_root, thumb_target, extensions) if @options[:index]
remove_thumb(Settings.thumb_target) if @options[:clean_thumbs]
remove_folders                      if @options[:clean_folders]
remove_files(Settings.thumb_target) if @options[:clean_files]
find_duplicates                     if @options[:find_duplicates]

# temporary actions
add_new_fields                      if @options[:add_new_fields]
add_mtime_and_ctime                 if @options[:add_mtime_and_ctime]