#
# Ronin - A Ruby platform for exploit development and security research.
#
# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#

require 'data_paths'
require 'fileutils'

module Ronin
  module Config
    include DataPaths

    # The users home directory
    HOME = File.expand_path(ENV['HOME'] || ENV['HOMEPATH'])

    # Ronin home directory
    PATH = File.join(HOME,'.ronin')

    # Configuration files directory
    CONFIG_DIR = File.join(PATH,'config')

    # Directory which repositories are installed into
    REPOS_DIR = File.join(PATH,'repos')

    # Temporary file directory
    TMP_DIR = File.join(PATH,'tmp')

    # Directory for storing files for campaigns
    CAMPAIGNS_DIR = File.join(PATH,'campaigns')
    
    FileUtils.mkdir(PATH) unless File.directory?(PATH)
    FileUtils.mkdir(CONFIG_DIR) unless File.directory?(PATH)
    FileUtils.mkdir(TMP_DIR) unless File.directory?(TMP_DIR)
    FileUtils.mkdir(CAMPAIGNS_DIR) unless File.directory?(CAMPAIGNS_DIR)

    #
    # Loads the Ronin configuration file.
    #
    # @param [Symbol, String, nil] name
    #   The optional name of the file to load within +CONFIG_DIR+.
    #
    # @example Load the config file at `~/.ronin/config/ronin.rb`
    #   Config.load
    #   # => true
    #
    # @example Load a specific config file in `~/.ronin/config/`
    #   Config.load :sql
    #   # => true
    #
    def Config.load(name=:ronin)
      path = File.expand_path(File.join(CONFIG_DIR,name.to_s))

      require path if File.file?(path)
    end

    #
    # Auto-creates a directory within {TMP_DIR}.
    #
    # @param [String] sub_path
    #   The sub-path within {TMP_DIR}.
    #
    # @return [String]
    #   The full path within {TMP_DIR}.
    #
    def Config.tmp_dir(sub_path=nil)
      if sub_path
        sub_path = File.expand_path(File.join('',sub_path))
        path = File.join(TMP_DIR,sub_path)

        unless File.exist?(path)
          FileUtils.mkdir_p(path)
        end

        return path
      end

      return TMP_DIR
    end
  end
end