require 'json'
require 'ap'
require 'yaml'
require 'slop'
require 'aws/cfn/dsl/base'
require 'aws/cfn/dsl/template'

module Aws
  module Cfn
    module Compiler
      class Base < ::Aws::Cfn::Dsl::Base
        attr_accessor :items
        attr_accessor :opts
        attr_accessor :spec

        def initialize
          super
          @items = {}
          @dynamic_items = {}
          @dynamic_references = {}
          @all_sections.each do |section|
            @dynamic_items[section]      ||= {}
            @dynamic_references[section] ||= []
          end
        end

        def dynamic_item(section,resource,hash)
          abort! "Invalid section '#{section}'\nValid sections are: #{@all_sections.join(',')}" unless @all_sections.include?(section)
          @dynamic_items[section][resource] = hash
        end

        def dynamic_reference(section,resource)
          abort! "Invalid section '#{section}'\nValid sections are: #{@all_sections.join(',')}" unless @all_sections.include?(section)
          @dynamic_references[section] << resource
        end

        def _uniq_names(arr)
          names = {}
          arr.map { |n| names[n] = nil }
          names.keys
        end

        def validate(compiled)
          abort! 'No Resources!?' unless compiled['Resources']
          logStep 'Validating template'

         # --- Parameters ----------------------------------------------------------------------------------------------
          prms  = compiled['Parameters'].keys rescue []
          if compiled['Parameters']
            compiled['Parameters'].each do |name,hash|
              abort! "Parameter #{name} has an invalid compiled block!\n#{hash.ai}" unless hash.is_a?(Hash)
              if hash['Type']
                unless %w(String Number CommaDelimitedList).include?(hash['Type'])
                  abort! "Parameter #{name} has an invalid type: #{hash['Type']}"
                end
              end
              if hash['Default']
                unless hash['Default'].is_a?(String)
                  abort! "Parameter #{name} has an invalid default (Must be string): #{hash['Default']}"
                end
              end
            end
            @logger.info '  Parameters validated'
          end

          # --- functions ----------------------------------------------------------------------------------------------
          funs  = find_fns(compiled) #.select { |a| !(a =~ /^AWS::/) }

          bad = []
          funs.each do |fn|
            unless @valid_functions.include?(fn)
              bad << fn
            end
          end
          if bad.size > 0
            abort! "Encountered unsupported function(s) ...\n#{bad.ai}\nSupported functions are:\n#{@valid_functions.ai}"
          end
          @logger.info '  Functions validated'

          # --- Mappings ----------------------------------------------------------------------------------------------
          mappings  = find_maps(compiled)
          mpgs  = compiled['Mappings'].nil? ? [] : compiled['Mappings'].keys
          names = _uniq_names(mpgs+@dynamic_items['Mappings'].keys)

          maps = {}
          mappings.each{|m| maps[m[:mapping]] = m[:source] }
          mapnames = maps.keys+@dynamic_references['Mappings']
          net = (mapnames-names)
          unless net.empty?
            @logger.error '!!! Unknown mappings !!!'
            net.each do |name|
              @logger.error "  #{name} (Location in template: #{maps[name].join('/')})"
            end
            abort!
          end
          net = (names-mapnames)
          unless net.empty?
            @logger.warn '!!! Unused mappings !!!'
            net.each do |name|
              @logger.warn "  #{name}"
            end
          end
          @logger.info '  Mappings validated'

          # --- Conditions ----------------------------------------------------------------------------------------------
          cond  = find_conditions(compiled)+@dynamic_references['Conditions']
          cnds  = compiled['Conditions'].keys rescue []
          names = _uniq_names(cnds+@dynamic_items['Conditions'].keys)

          net = (cond-names)
          unless net.empty?
            @logger.error '!!! Unknown conditions !!!'
            net.each do |name|
              @logger.error "  #{name}"
            end
            abort!
          end
          net = (names-cond)
          unless net.empty?
            @logger.warn '!!! Unused conditions !!!'
            net.each do |name|
              @logger.warn "  #{name}"
            end
          end
          @logger.info '  Conditions validated'

          # --- References ----------------------------------------------------------------------------------------------
          # Parameters => Resources => Outputs
          refs  = find_refs(compiled).select { |a,_| !(a =~ /^AWS::/) }
          rscs  = compiled['Resources'].keys
          names = _uniq_names((mpgs+rscs+prms+cnds)+@dynamic_items['Mappings'].keys+@dynamic_items['Conditions'].keys+@dynamic_items['Parameters'].keys+@dynamic_items['Resources'].keys)

          net = (_uniq_names(refs.keys+@dynamic_references['Mappings']+@dynamic_references['Parameters']+@dynamic_references['Resources'])-names)
          unless net.empty?
            @logger.error '!!! Unknown references !!!'
            net.each do |name|
              @logger.error "  #{name} from #{refs[name][0]}:#{refs[name][1]}"
            end
            abort!
          end
          net = (prms+@dynamic_items['Parameters'].keys-refs.keys-@dynamic_references['Parameters'])
          unless net.empty?
            @logger.warn '!!! Unused Parameters !!!'
            net.sort.each do |name|
              @logger.warn "  #{name}"
            end
          end
          net = (rscs+@dynamic_items['Resources'].keys-refs.keys-@dynamic_references['Resources'])
          unless net.empty?
            @logger.info '!!! Unreferenced Resources !!!'
            net.sort.each do |name|
              @logger.info "  #{name}"
            end
          end
          @logger.info '  References validated'
        end

        def save_template(output_file,compiled)
          filn = output_file
          file = output_file
          file = File.expand_path(output_file) if @config[:expandedpaths]
          if File.exists?(file)
            file = File.realpath(file)
          end
          logStep "Writing compiled file to #{filn}..."
          begin
            hash = {}
            compiled.each do |item,value|
              unless value.nil?
                if (not value.is_a?(Hash)) or (value.count > 0)
                  hash[item] = value
                end
              end
            end

            dir  = File.dirname(output_file)
            file = File.basename(output_file)
            sect = dir == '.' ? Dir.pwd : dir
            sect = File.basename(sect) unless @config[:expandedpaths]
            save_section(dir, file, @config[:format], sect, hash, '', 'template')

            parameters = []
            if (@config[:parametersfile] or @config[:stackinifile]) and hash.has_key?('Parameters')
              hash['Parameters'].each do |par,hsh|
                # noinspection RubyStringKeysInHashInspection
                parameters <<   {
                    'ParameterKey' => par,
                    'ParameterValue' => hsh.has_key?('Default') ? hsh['Default'] : '',
                    # 'UsePreviousValue' => false,
                }
              end
            end

            if @config[:parametersfile] and parameters.size > 0
              dir  = File.dirname(@config[:parametersfile])
              file = File.basename(@config[:parametersfile])
              sect = dir == '.' ? Dir.pwd : dir
              sect = File.basename(sect) unless @config[:expandedpaths]

              save_section(dir, file, @config[:format], sect, parameters, '', 'parameters')
            end

            if @config[:stackinifile] and parameters.size > 0
              dir  = File.dirname(@config[:stackinifile])
              file = File.basename(@config[:stackinifile])
              sect = dir == '.' ? Dir.pwd : dir
              sect = File.basename(sect) unless @config[:expandedpaths]

              save_inifile(dir, file, sect, parameters, 'parameters')
            end
            @logger.info '  Compiled file written.'
          rescue
            abort! "!!! Could not write compiled file: #{$!}"
          end
        end

        def load_spec(spec=nil)
          if spec
            abs = nil
            [spec, File.join(@config[:brick_path_list],spec)].flatten.each do |p|
              begin
                abs = File.realpath(File.absolute_path(File.expand_path(p)))
                break if File.exists?(abs)
              rescue => e
                @logger.debug e
                # pass
              end
            end

            if not abs.nil? and File.exists?(abs)
              logStep "Loading specification #{@config[:expandedpaths] ? abs : spec}..."
              unless abs =~ /\.(json|ya?ml|jts|yts)\z/i
                abort! "Unsupported specification file type: #{spec}=>#{abs}\n\tSupported types are: json,yaml,jts,yts\n"
              end

              spec = File.read(abs)

              case File.extname(File.basename(abs)).downcase
                when /json|jts/
                  @spec = JSON.parse(spec)
                when /yaml|yts/
                  begin
                    @spec = YAML.load(spec)
                  rescue Psych::SyntaxError => e
                    i = 0
                    abort! "Error in the template specification: #{e.message}\n#{spec.split(/\n/).map{|l| "#{i+=1}: #{l}"}.join("\n")}"
                  end
                else
                  abort! "Unsupported file type for specification: #{spec}"
              end
            else
              abort! 'Unable to open specification'+ (abs.nil? ? " or {.,#{@config[:brick_path_list].join(',')}/}#{spec} not found" : ": #{abs}")
            end
          else
            abort! 'No specification provided'
          end
        end

        protected

        require 'aws/cfn/compiler/mixins/options'
        include Aws::Cfn::Compiler::Options

        require 'aws/cfn/compiler/mixins/load'
        include Aws::Cfn::Compiler::Load

        require 'aws/cfn/compiler/mixins/parse'
        include Aws::Cfn::Compiler::Parse

        require 'aws/cfn/compiler/mixins/compile'
        include Aws::Cfn::Compiler::Compile

        require 'aws/cfn/compiler/mixins/save'
        include Aws::Cfn::Compiler::Save

      end
    end
  end
end