Files
metasploit-gs/lib/bindata/registry.rb
T
Ramon de C Valle f124597a56 Code cleanups
git-svn-id: file:///home/svn/framework3/trunk@5773 4d416f70-5f16-0410-b530-b9f4589650da
2008-10-19 21:03:39 +00:00

42 lines
1.1 KiB
Ruby

require 'singleton'
module BinData
# This registry contains a register of name -> class mappings.
class Registry
include Singleton
def initialize
@registry = {}
end
# Convert camelCase +name+ to underscore style.
def underscore_name(name)
name.to_s.sub(/.*::/, "").
gsub(/([A-Z]+)([A-Z][a-z])/,'\1_\2').
gsub(/([a-z\d])([A-Z])/,'\1_\2').
tr("-", "_").
downcase
end
# Registers the mapping of +name+ to +klass+. +name+ is converted
# from camelCase to underscore style.
# Returns the converted name
def register(name, klass)
# convert camelCase name to underscore style
key = underscore_name(name)
# warn if replacing an existing class
if $VERBOSE and (existing = @registry[key])
warn "warning: replacing registered class #{existing} with #{klass}"
end
@registry[key] = klass
key.dup
end
# Returns the class matching a previously registered +name+.
def lookup(name)
@registry[name.to_s]
end
end
end