summaryrefslogtreecommitdiff
path: root/lib/oxidized/source/sql.rb
blob: f7e65101835ea69dd174cc5497f6542278b3f416 (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
module Oxidized
class SQL < Source
  require 'sequel'

  def initialize
    super
    @cfg = CFG.source[:sql]
  end

  def setup
    if not @cfg
      CFG.source[:sql] = {
        :adapter => 'sqlite',
        :file    => File.join(Config::Root, 'sqlite.db'),
        :table   => 'devices',
        :map     => {
          :name      => 'name',
          :model     => 'rancid',
        }
      }
    end
    CFG.save
  end

  def load
    nodes = []
    case @cfg[:adapter]
    when 'sqlite'
      require 'sqlite3'
      Sequel.sqlite @cfg[:file]
    end
    klass = Class.new(Sequel::Model @cfg[:table].to_sym)
    SQL.send :remove_const, :Nodes if SQL.const_defined? :Nodes
    SQL.const_set :Nodes, klass
    @cfg[:map].each { |new,old| Nodes.class_eval "alias #{new.to_sym} #{old.to_sym}" }
    Nodes.each do |node|
      keys = {}
      @cfg[:map].each { |key, _| keys[key] = node.send(key.to_sym) }
      keys[:model] = map_model keys[:model] if keys.key? :model
      nodes << keys
    end
    nodes
  end

end
end