require 'sparkle_formation'
require 'sfn'

module Sfn
  class Command
    # Validate command
    class Validate < Command

      include Sfn::CommandModule::Base
      include Sfn::CommandModule::Template
      include Sfn::CommandModule::Stack

      def execute!
        print_only_original = config[:print_only]
        config[:print_only] = true
        file = load_template_file
        ui.info "#{ui.color("Template Validation (#{provider.connection.provider}): ", :bold)} #{config[:file].sub(Dir.pwd, '').sub(%r{^/}, '')}"
        config[:print_only] = print_only_original

        raw_template = _format_json(parameter_scrub!(template_content(file)))

        if(config[:print_only])
          ui.puts raw_template
        else
          validate_stack(
            file.respond_to?(:dump) ? file.dump : file,
            sparkle_collection.get(:template, config[:file])[:name]
          )
        end
      end

      # Validate template with remote API and unpack nested templates if required
      #
      # @param template [Hash] template data structure
      # @param name [String] name of template
      # @return [TrueClass]
      def validate_stack(template, name)
        resources = template.fetch('Resources', {})
        nested_stacks = resources.find_all do |r_name, r_value|
          r_value.is_a?(Hash) &&
            provider.connection.data[:stack_types].include?(r_value['Type'])
        end
        nested_stacks.each do |n_name, n_resource|
          validate_stack(n_resource.fetch('Properties', {}).fetch('Stack', {}), "#{name} > #{n_name}")
          n_resource['Properties'].delete('Stack')
        end
        begin
          ui.info "Validating: #{ui.color(name, :bold)}"
          if(config[:upload_root_template])
            upload_result = store_template('validation-stack', template, Smash.new)
            stack = provider.connection.stacks.build(
              :name => 'validation-stack',
              :template_url => upload_result[:url]
            )
          else
            stack = provider.connection.stacks.build(
              :name => 'validation-stack',
              :template => parameter_scrub!(template)
            )
          end
          result = api_action!(:api_stack => stack) do
            stack.validate
          end
          ui.info ui.color('  -> VALID', :bold, :green)
          true
        rescue => e
          ui.info ui.color('  -> INVALID', :bold, :red)
          ui.fatal e.message
          raise e
        end
      end

    end
  end
end