require 'iapi-idlc-sdk-pfm/command/base'
require 'mixlib/shellout'

module Pfm
  module Command
    class Plan < Base
      banner 'Usage: pfm plan [options]'

      option :app_release,
             short:        '-a VERSION',
             long:         '--app-release VERSION',
             description:  'Application Version Number to Deploy',
             default:      ''

      option :server_build,
             short:        '-b NUMBER',
             long:         '--server-build NUMBER',
             description:  'Server Build Number to Deploy',
             default:      ENV['SERVER_BUILD']

      option :landscape,
             short:        '-l',
             long:         '--landscape',
             description:  'Format the output with the terraform_landscape gem',
             boolean:      true,
             default:      false

      option :config_file,
             short:        '-c FILE',
             long:         '--config-file FILE',
             description:  'Optional environment metadata file',
             default:      nil

      option :working_dir,
             short:        '-d DIR',
             long:         '--dir DIR',
             description:  'Optional directory of infrastructure configuration to use',
             default:      ''

      def initialize
        super
        @params_valid = true
        @errors = []
      end

      def run(params)
        @params = params
        read_and_validate_params

        if params_valid?
          if (@config[:config_file])
            deploy_setupv2
            plan(@config[:working_dir])
          else
            deploy_setup
            plan(@workspace.tmp_dir)
          end
          # @workspace.cleanup causing bundler issues
          0
        else
          @errors.each { |error| err("Error: #{error}") }
          parse_options(params)
          msg(opt_parser)
          1
        end
      rescue DeploymentFailure => e
        err("PLAN ERROR: #{e.message}\n")
        1
      end

      def plan(dir)
        begin
          if upgraded_terraform?
            Terraform::Binary::Command.run("-chdir #{dir} plan") unless @config[:landscape]
            Terraform::Binary::Command.run("-chdir #{dir} plan | landscape") if @config[:landscape]
          else
            Terraform::Binary.plan(dir.to_s) unless @config[:landscape]
            Terraform::Binary.plan("#{dir} | landscape") if @config[:landscape]
          end
        rescue
          raise DeploymentFailure, 'Finished with errors'
        end
      end

      def read_and_validate_params
        arguments = parse_options(@params)

        case arguments.size
        when 0
          @params_valid = true
        else
          @params_valid = false
        end
      end

      def params_valid?
        @params_valid
      end
    end
  end
end