1 files changed, 46 insertions, 0 deletions
diff --git a/lib/oxidized/source/sql.rb b/lib/oxidized/source/sql.rb
new file mode 100644
index 0000000..f7e6510
--- /dev/null
+++ b/lib/oxidized/source/sql.rb
@@ -0,0 +1,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