Class: NetworkDrawer::Source

Inherits:
Object
  • Object
show all
Defined in:
lib/network_drawer/source.rb

Overview

Replesent of source file

Class Method Summary collapse

Class Method Details

.file_type(file_name) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
 
# 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) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
 
# 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) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
 
# 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