module Xml2Go

# File lib/xml2go.rb, line 9
def self.get_const_name(str)
  final_str = ""
  return str.gsub(/([A-Z])/){|c| "_#{c}"}.gsub(/^_/, "").upcase
end

# File lib/xml2go.rb, line 14
def self.load(file_handle)
  @@doc = Nokogiri::XML(file_handle)
  @@structs = {}
end

# File lib/xml2go.rb, line 19
def self.parse(config)
  @@config = config
  parse_element(@@doc)
end

# File lib/xml2go.rb, line 24
def self.parse_element(element)
  @@parser = Xml2Go::Parser.new(@@config)
  @@parser.parse_element(element)
  @@structs = @@parser.structs
  return
end

# File lib/xml2go.rb, line 38
def self.write_mocks_to_file(filename)
  file_handle = File.new(filename, "w")
  file_handle.write("package main\n\n")
  consts = @@structs.values.map{ |v| v.get_consts }
  file_handle.write(consts.join("\n"))
  blobs = @@structs.values.map { |v| v.to_declaration}
  file_handle.write(blobs.join("\n"))
  file_handle.close()
end

# File lib/xml2go.rb, line 31
def self.write_to_file(filename)
  file_handle = File.new(filename, "w")
  file_handle.write("package main\n\n")
  file_handle.write(@@structs.values.join("\n"))
  file_handle.close()
end