summaryrefslogtreecommitdiff
path: root/lib/oxidized/output/file.rb
blob: ee8a9a6541825aa8e5dec5aa86b8629227b8b395 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
module Oxidized
class OxFile < Output
  require 'fileutils'

  def initialize
    @cfg = CFG.output[:file]
  end

  def setup
    if not @cfg
      CFG.output[:file] = {
        :directory => File.join(Config::Root, 'configs')
      }
      CFG.save
    end
  end

  def store node, data, opt={}
    file = @cfg[:directory]
    if opt[:group]
      file = File.join File.dirname(file), opt[:group]
    end
    FileUtils.mkdir_p file
    file = File.join file, node
    open(file, 'w') { |fh| fh.write data }
  end

  def fetch node, group
    cfg_dir = @cfg[:directory]
    if group # group is explicitly defined by user
      IO.readlines File.join(cfg_dir, group, node)
    else
      if File.exists? File.join(cfg_dir, node) # node configuration file is stored on base directory
        IO.readlines File.join(cfg_dir, node)
      else
        path = Dir.glob File.join(cfg_dir, '**', node) # fetch node in all groups
        open(path[0], 'r').readlines
      end
    end
  end

end
end