module Thor::Util

def self.constant_from_thor_path(str)
  make_constant(to_constant(str))
rescue NameError => e
  raise e unless e.message =~ /^uninitialized constant (.*)$/
  raise Error, "There was no available namespace `#{str}'."
end

def self.constant_to_thor_path(str, remove_default = true)
  str = snake_case(str.to_s).squeeze(":")
  str.gsub!(/^default/, '') if remove_default
  str
end

def self.constants_in_contents(str)
  klasses = ObjectSpace.classes.dup
  Module.new.class_eval(str)
  klasses = ObjectSpace.classes - klasses
  klasses = klasses.select {|k| k < Thor }
  klasses.map! {|k| k.to_s.gsub(/#<Module:\w+>::/, '')}
end

def self.make_constant(str)
  list = str.split("::").inject(Object) {|obj, x| obj.const_get(x)}
end

def self.snake_case(str)
  return str.downcase if str =~ /^[A-Z_]+$/
  str.gsub(/\B[A-Z]/, '_\&').squeeze('_') =~ /_*(.*)/
  return $+.downcase
end  

def self.to_constant(str)
  str = 'default' if str.empty?
  str.gsub(/:(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase }
end