require 'yaml'

module Forger
  class Setting
    def initialize(check_project=true)
      @check_project = check_project
    end

    # data contains the settings.yml config.  The order or precedence for settings
    # is the project lono/settings.yml and then the ~/.lono/settings.yml.
    @@data = nil
    def data
      return @@data if @@data

      if @check_project && !File.exist?(project_settings_path)
        puts "ERROR: No settings file at #{project_settings_path}.  Are you sure you are in a forger project?".colorize(:red)
        exit 1
      end

      all_envs = load_file(project_settings_path)
      all_envs = merge_base(all_envs)
      @@data = all_envs[Forger.env] || all_envs["base"] || {}
      @@data = flatten_s3_folder(@@data)
      @@data
    end

    # So we can access settings['s3_folder'] transparently
    def flatten_s3_folder(data)
      # data = data.clone
      if data['s3_folder'].is_a?(Hash)
        data['s3_folder'] = s3_folder
      end
      data
    end

    # Special helper method to support multiple formats for s3_folder setting.
    # Format 1: Simple String
    #
    #   development:
    #     s3_folder: mybucket/path/to/folder
    #
    # Format 2: Hash
    #
    #   development:
    #     s3_folder:
    #       default: mybucket/path/to/folder
    #       dev_profile1: mybucket/path/to/folder
    #       dev_profile1: another-bucket/storage/path
    #
    def s3_folder
      s3_folder = data['s3_folder']
      return s3_folder if s3_folder.nil? or s3_folder.is_a?(String)

      # If reach here then the s3_folder is a Hash
      s3_folder[ENV['AWS_PROFILE']] || s3_folder["default"]
    end

  private
    def load_file(path)
      return Hash.new({}) unless File.exist?(path)

      content = RenderMePretty.result(path)
      data = YAML.load(content)
      # If key is is accidentally set to nil it screws up the merge_base later.
      # So ensure that all keys with nil value are set to {}
      data.each do |env, _setting|
        data[env] ||= {}
      end
      data
    end

    # automatically add base settings to the rest of the environments
    def merge_base(all_envs)
      base = all_envs["base"] || {}
      all_envs.each do |env, settings|
        all_envs[env] = base.merge(settings) unless env == "base"
      end
      all_envs
    end

    def project_settings_path
      "#{Forger.root}/config/settings.yml"
    end
  end
end