require 'fileutils'
require 'json'
require 'yaml'
require 'hieracles/utils'

module Hieracles
  # configuration singleton
  module Config
    include Hieracles::Utils
    extend self

    attr_reader :extraparams, :server, :classpath, :scope,
      :modulepath, :hierafile, :basepath, :encpath, :format

    def load(options)
      @optionfile = options[:config] || defaultconfig
      @extraparams = extract_params(options[:params])
      initconfig(@optionfile) unless File.exist? @optionfile
      values = YAML.load_file(@optionfile)
      @server = values['server']
      @classpath = values['classpath']
      @modulepath = values['modulepath'] || 'modules'
      @encpath = options[:encpath] || values['encpath'] || 'enc'
      @basepath = File.expand_path(options[:basepath] || values['basepath'] || '.')
      @hierafile = options[:hierafile] || values['hierafile'] || 'hiera.yaml'
      @format = (options[:format] || values['format'] || 'console').capitalize
      facts_file = options[:yaml_facts] || options[:json_facts]
      facts_format = options[:json_facts] ? :json : :yaml
      @scope = sym_keys((facts_file && load_facts(facts_file, facts_format)) || values['defaultscope'] || {})
    end

    def initconfig(file)
      FileUtils.mkdir_p(File.dirname(file))
      File.open(file, 'w') do |f|
        f.puts '---'
        f.puts '# uncomment if you use the CGI method for discovery'
        f.puts '# server: puppetserver.example.com'
        f.puts 'classpath: manifests/classes/%s.pp'
        f.puts 'modulepath: modules'
        f.puts 'encpath: enc'
        f.puts 'hierafile: hiera.yaml'
      end
    end

    def defaultconfig
      File.join(ENV['HOME'], '.config', 'hieracles', 'config.yml')
    end

    # str is like: something=xxx;another=yyy  
    def extract_params(str)
      return {} unless str
      str.split(';').reduce({}) do |a, k|
        a["#{k[/^[^=]*/]}".to_sym] = k[/[^=]*$/]
        a
      end
    end

    def path(what)
      File.join(@basepath, send(what.to_sym))
    end

    def load_facts(file, format)
      if format == :json
        JSON.parse(File.read(file))
      else
        YAML.load_file(file)
      end
    end

  end
end