class NetworkDrawer::Source

Replesent of source file

Public Class Methods

file_type(file_name) click to toggle source
# File lib/network_drawer/source.rb, line 21
def self.file_type(file_name)
  return nil unless file_name
  case file_name
  when /\.json$/
    :json
  when /\.ya?ml$/
    :yaml
  else
    nil
  end
end
read(file_name) click to toggle source
# File lib/network_drawer/source.rb, line 7
def self.read(file_name)
  source = {}
  case file_type(file_name)
  when :json
    source = JSON.parse(File.read(file_name), symbolize_names: true)
  when :yaml
    source = symbolize(YAML.load(File.read(file_name)))
  else
    puts 'Incorrect file type'
    exit 1
  end
  source
end
symbolize(obj) click to toggle source
# File lib/network_drawer/source.rb, line 32
def self.symbolize(obj)
  return obj.inject({}) do |memo, (k, v)|
    memo[k.to_sym] =  symbolize(v)
    memo
  end if obj.is_a? Hash
  return obj.inject([]) do |memo, v|
    memo << symbolize(v)
    memo
  end if obj.is_a? Array
  return obj
end