summaryrefslogtreecommitdiff
path: root/lib/oxidized/output/file.rb
blob: 8a53f91acb177cd98d2f3eed4a0dbae47c942910 (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
44
45
46
47
48
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 != 0 # group is explicitly defined by user
      IO.readlines File.join(cfg_dir, group, node)
    else
      if File.exists?("#{cfg_dir}/#{node}") # node configuration file is stored on base directory
        IO.readlines File.join(cfg_dir, node)
      else
        file = Find.find('.').lazy.find{|e|e.match /#{node}/}
        if file
          open(file, 'r').readlines
        else
          "not found."
        end
      end
    end
  end
  
end
end