class Terraspace::CLI
  class Fmt
    include Concerns::SourceDirs
    include Terraspace::Util::Logging

    def initialize(options={})
      @options = options
      @mod_name = options[:mod]
    end

    @@exit_status = 0
    def run
      logger.info "Formatting terraform files"
      dirs.each do |dir|
        exit_status = format(dir)
        @@exit_status = exit_status if exit_status != 0
      end
      exit(@@exit_status)
    end

    def format(dir)
      Runner.new(dir, @options).format!
    end

  private
    def dirs
      if @mod_name
        type_dirs.select { |p| p.include?(@mod_name) }
      else
        type_dirs
      end
    end

    def type_dirs
      type = @options[:type]
      if type && type != "all"
        app_source_dirs.select { |p| p.include?("/#{type.pluralize}/") }
      else
        app_source_dirs
      end
    end
  end
end