netconfgen 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e74877df80cf028688b27e0712cc3d56b15caa44
-  data.tar.gz: 37c95750d391b9f41217cd724f1e08e8e21f5184
+  metadata.gz: 8aa9d22b0e9d275cf0ef86fde23168d0453a605d
+  data.tar.gz: aec475534ce8cdae5c7f67469a19ef8c122ad906
 SHA512:
-  metadata.gz: 12e26b6ae98233cca7b0534c3affcf99405410c02adb99cf60ea167feba85865e6d294c182709e036ccdd79c342688f2fe4ec19f2b9a84c65c29310eef39616c
-  data.tar.gz: e334f1c2070bde07d19a111362fc19ceb87a9778c63da993524ed5d2d5d5cebdbf8d5d8f304ddbed051f155203250b34893c52958895c9ea0c4585601535a8b1
+  metadata.gz: c5e043ea9977c620995ac432d23e67860887a112f53fd2948e6de74b1955db2fe768537f8ea3c9c4f3c3b9f6e331306a14879ff348b7f3f7fd30b7fb9a6a318b
+  data.tar.gz: c7f2b03d00e9d483a23a732e523c461aafe95ccbd4dc094b7b683ee658388741c1d90cbeb76ede63efd499e086dde3bc65858d37d77df30b48b27be3be562e28

data/bin/confgen

@@ -11,6 +11,7 @@ require 'json'
 
 CONFIG_FILE = ["/etc/netconfgen.yaml", "~/.netconfgen.yaml", "netconfgen.yaml"]
 config = {}
+
 CONFIG_FILE.each do |fname|
   fname = File.expand_path(fname)
   if File.exists?(fname)
@@ -61,6 +62,11 @@ if config[:listen] && config[:port] < 1024 && !Process.euid.zero?
   abort 'Please run tftpd as root via sudo!'
 end
 
+
+if config[:verbose]
+  PP.pp(config, STDERR)
+end
+
 log = Logger.new(STDOUT)
 log.level = config[:verbose] ? "DEBUG" : "INFO"
 log.formatter = lambda do |s, d, p, m|
@@ -68,13 +74,10 @@ log.formatter = lambda do |s, d, p, m|
 end
 config[:logger] = log
 
-if config[:verbose]
-  PP.pp(config, STDERR)
-end
 
 if !config[:listen]
-  STDERR.puts "Rendering block #{config[:name]}" if config[:verbose]
-  br = NetConfGen::BlockEngine.new(config[:root])
+  STDERR.puts "Rendering block #{config[:name]} from #{config[:root]}" if config[:verbose]
+  br = NetConfGen::BlockEngine.new(config[:root], config[:settings])
 
   config[:variables].each do |k, v|
     br.set(k, v)

data/lib/netconfgen/netconfgen.rb

@@ -2,7 +2,10 @@
 #
 
 require 'erb'
-
+require 'net/http'
+require 'uri'
+require 'json'
+require 'pp'
 
 module NetConfGen
 
@@ -27,20 +30,100 @@ module NetConfGen
   end
 
   class BlockContext
-    def initialize(blockengine)
+    def initialize(blockengine, settings=nil)
       @blockengine = blockengine
+      @settings = settings
     end
 
     def include(name)
       block = @blockengine.load(name)
       return block.render
     end
+
+    # Creates an array of individual ports from a port range string
+    def portrange(portrange)
+      if m = portrange.match(/((?:Gi|Fa|Te)(?:[0-9]+\/)+)(\d+)(?:-(\d+))?/) # eg. Gi1/0/27-28 or Gi1/0/27
+        ports = []
+        if m[3]
+          (m[2]..m[3]).each do |i|
+            ports << (m[1] + i)
+          end
+        else
+          ports << (m[1] + m[2])
+        end
+
+        return ports
+      elsif m = portrange.match(/(Gi|Fa|Te)\[(\d)(\d)\](\/\d+)/) # eg. Gi[34]/1
+        ports = []
+        (m[2]..m[3]).each do |i|
+          ports << (m[1] + i + m[4])
+        end
+        return ports
+      end
+      return []
+    end
+
+    def megaexcel_find_row_by_column_value(data, column_definition_row_number, column_name, column_value)
+
+      # The 3rd row is known to be containing the column names
+      columns = data["values"][column_definition_row_number]
+
+      # Set i to the INDEX of the column which we are searching
+      i = columns.index(column_name)
+      if i == nil
+        raise "Column #{column_name} not found from megaexcel"
+      end
+
+      # Find the row where the INDEX column contains the value we are looking for
+      row = data["values"].find do |x|
+        x[i] == column_value
+      end
+
+      # Convert the indexed row into an object which keys are the column names
+      obj = {}
+      columns.each_with_index do |key, j|
+        obj[key] = row[j]
+      end
+
+      return obj
+    end
+
+    def megaexcel(name)
+      url = @settings["megaexcel"]["url"]
+      name_field = @settings["megaexcel"]["name_field"]
+      column_definition_row_number = @settings["megaexcel"]["column_definition_row_number"]
+
+      ret = Net::HTTP.get(URI.parse(url))
+      data = JSON.parse(ret)
+
+      row = megaexcel_find_row_by_column_value(data, column_definition_row_number, name_field, name)
+      pp row
+    end
+
+    def megaexcel_vlans(data)
+      vlans_column_name = @settings["megaexcel"]["vlans_column_name"]
+      str = data[vlans_column_name]
+      vlans = []
+      str.each_line do |line|
+        if m = line.match(/((?:Fa|Gi|Te)[^ ]+).+?vlan\s?(\d+)/)
+          vlans << {
+            "ports" => m[1],
+            "vlan" => m[2],
+            "description" => line.chomp.strip,
+          }
+        else
+          # Warning. unknown vlan setup #{line}
+        end
+      end
+      return vlans
+    end
   end
 
   class BlockEngine
-    attr_reader :context
-    def initialize(basepath)
+    attr_reader :context, :settings
+    def initialize(basepath, settings=nil)
       @basepath = basepath
+      @settings = settings
       if !File.directory?(basepath)
         raise "Basepath #{basepath} does not exists"
       end
@@ -48,7 +131,7 @@ module NetConfGen
 
       @blocks = {}
 
-      @context = BlockContext.new(self)
+      @context = BlockContext.new(self, @settings)
     end
 
     def set(key, value)

data/netconfgen.gemspec

@@ -3,7 +3,7 @@
 
 Gem::Specification.new do |s|
   s.name          = 'netconfgen'
-  s.version       = '0.0.1'
+  s.version       = '0.0.2'
   s.date          = Time.now
 
   s.summary       = %q{Template based config generation}

data/netconfgen.yaml

@@ -2,3 +2,4 @@ verbose: true
 variables:
   foo: 2
   test: test
+settings:

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: netconfgen
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Juho Mäkinen juho.makinen@gmail.com
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-07-06 00:00:00.000000000 Z
+date: 2019-07-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubygems-tasks