# frozen_string_literal: true

require "yaml"
require "pathname"
require "thor"
require "thor/actions"
require "thor_plus/actions"

module Pragmater
  # The Command Line Interface (CLI) for the gem.
  class CLI < Thor
    include Thor::Actions
    include ThorPlus::Actions

    package_name Pragmater::Identity.version_label

    def initialize args = [], options = {}, config = {}
      super args, options, config
    end

    desc "-a, [--add=ADD]", "Add pragma comments to source file(s)."
    map %w(-a --add) => :add
    method_option :comments, aliases: "-c", desc: "Pragma comments", type: :array, default: []
    method_option :extensions, aliases: "-e", desc: "File extension whitelist", type: :array, default: [".rb", ".rake"]
    def add path
      write path, options[:comments], options[:extensions], :add
    end

    desc "-r, [--remove=REMOVE]", "Remove pragma comments from source file(s)."
    map %w(-r --remove) => :remove
    method_option :comments, aliases: "-c", desc: "Pragma comments", type: :array, default: []
    method_option :extensions, aliases: "-e", desc: "File extension whitelist", type: :array, default: [".rb", ".rake"]
    def remove path
      write path, options[:comments], options[:extensions], :remove
    end

    desc "-e, [--edit]", "Edit #{Pragmater::Identity.label} settings in default editor."
    map %w(-e --edit) => :edit
    def edit
      resource_file = File.join ENV["HOME"], Pragmater::Identity.file_name
      info "Editing: #{resource_file}..."
      `#{editor} #{resource_file}`
    end

    desc "-v, [--version]", "Show #{Pragmater::Identity.label} version."
    map %w(-v --version) => :version
    def version
      say Pragmater::Identity.version_label
    end

    desc "-h, [--help=HELP]", "Show this message or get help for a command."
    map %w(-h --help) => :help
    def help task = nil
      say && super
    end

    private

    def update_file path, comments, action
      Writer.new(path, comments).public_send action
      say "Updated: #{path}."
    end

    def write path, comments, extensions, action
      pathname = Pathname path

      case
        when pathname.file?
          update_file pathname, comments, action
        when pathname.directory?
          files = Pathname.glob %(#{pathname}/**/*{#{extensions.join ","}})
          files.each { |file_path| update_file file_path, comments, action }
        else
          error "Invalid path: #{path}."
      end
    end
  end
end