From 9d217025fac3e335c308f02e7377e14ccfdc0e66 Mon Sep 17 00:00:00 2001
From: Saku Ytti <saku@ytti.fi>
Date: Wed, 17 Apr 2013 17:48:50 +0300
Subject: Initial commit

Silly for shit-and-giggles attempt at rancid
---
 lib/oxidized/source/sql.rb | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 lib/oxidized/source/sql.rb

(limited to 'lib/oxidized/source/sql.rb')

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
-- 
cgit v1.2.1