#
# Copyright (c) 2012 RightScale Inc
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

require 'rbconfig'

module RightSupport::Ruby
  module StringExtensions
    if require_succeeds?('active_support/core_ext')
      ACTIVE_SUPPORT_WORKALIKES = false
    else
      ACTIVE_SUPPORT_WORKALIKES = true
    end

    # Convert to snake case.
    #
    #   "FooBar".snake_case           #=> "foo_bar"
    #   "HeadlineCNNNews".snake_case  #=> "headline_cnn_news"
    #   "CNN".snake_case              #=> "cnn"
    #
    # @return [String] Receiver converted to snake case.
    #
    # @api public
    def snake_case
      return downcase if match(/\A[A-Z]+\z/)
      gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2').
      gsub(/([a-z])([A-Z])/, '\1_\2').
      downcase
    end
  
    # Convert a constant name to a path, assuming a conventional structure.
    #
    #   "FooBar::Baz".to_const_path # => "foo_bar/baz"
    #
    # @return [String] Path to the file containing the constant named by receiver
    #   (constantized string), assuming a conventional structure.
    #
    # @api public
    def to_const_path
      snake_case.gsub(/::/, "/")
    end
  
    # Convert constant name to constant
    #
    #    "FooBar::Baz".to_const => FooBar::Baz
    #
    # @return [Constant] Constant corresponding to given name or nil if no
    #   constant with that name exists
    #
    # @api public
    def to_const
      names = split('::')
      names.shift if names.empty? || names.first.empty?
  
      constant = Object
      names.each do |name|
        # modified to return nil instead of raising an const_missing error
        constant = constant && constant.const_defined?(name) ? constant.const_get(name) : nil
      end
      constant
    end

    # Reverse operation of snake case:
    #
    #    "some_string/some_other_string" => "SomeString::SomeOtherString"
    #
    # @return [String] Camelized string
    #
    # @api public
    if !String.public_method_defined?(:camelize) && ACTIVE_SUPPORT_WORKALIKES
      def camelize(first_letter = :upper)
        case first_letter
          when :upper then gsub(/\/(.?)/) { "::" + $1.upcase }.gsub(/(^|_)(.)/) { $2.upcase }
          when :lower then first + camelize(self)[1..-1]
        end
      end
    end

    # Add ability to output colored text to console
    # e.g.: puts "Hello".red
    def bold; colorize("\e[1m\e[29m"); end
    def grey; colorize("\e[30m"); end
    def red; colorize("\e[1m\e[31m"); end
    def dark_red; colorize("\e[31m"); end
    def green; colorize("\e[1m\e[32m"); end
    def dark_green; colorize("\e[32m"); end
    def yellow; colorize("\e[1m\e[33m"); end
    def blue; colorize("\e[1m\e[34m"); end
    def dark_blue; colorize("\e[34m"); end
    def pur; colorize("\e[1m\e[35m"); end
    def colorize(color_code)
      # Doesn't work with the Windows prompt...
      @@windows ||= RbConfig::CONFIG['host_os'] =~ /mswin|win32|dos|mingw|cygwin/i
      (@@windows || !$stdout.isatty) ? to_s : "#{color_code}#{to_s}\e[0m"
    end
  
  end
end

class String
  include RightSupport::Ruby::StringExtensions
end