aio_elin 0.0.1

data/lib/modules/description/style/total_output_drops.rb

@@ -0,0 +1,22 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::Description::TotalOutputDrops < Aio::Module::Description
+  include Aio::Module
+
+  def initialize
+    super({
+      :author     => "Elin",
+      :case       => :total_output_drops,
+      :name       => "出接口丢包",
+      :title      => "出接口丢包检查",
+      :description  => "出接口丢包检查"
+    })  
+  end 
+
+	def to_s
+		"出接口丢包率不应该高出阈值。"
+	end
+end
+

data/lib/modules/input/style/compare_xml.rb

@@ -0,0 +1,89 @@
+#coding=utf-8
+
+require "aio/core"
+require "rexml/document"
+
+class Aio::Module::InputStyle::CompareXML < Aio::Module::InputStyle
+	include Aio::Module
+	include REXML
+	include Aio::Ui::Verbose
+
+	def initialize
+		super({
+			:author				=> "Elin",
+			:description	=> "这个模块用于分析由output/out_compare_xml模块生成的xml文件",
+			:platform			=> "all",
+			:file_suffix	=> Regexp.new('\.xml$'),
+			:pass_file		=> [],
+		})
+	end
+
+	def parse
+		# dir 为 Pathname 类
+		print_good "正在打开文件 #{self.input_file}"
+		file = File.open(self.input_file)
+		print_good "打开文件完成"
+		print_good "正在载入到XML模块..."
+		atime = Time.now
+		doc = Document.new(file)
+		root = doc.root
+		print_good "载入完成。载入用时 #{format("%.2fs", Time.now.to_f - atime.to_f)}"
+		device_hash = {}
+
+		index = 0
+		total = root.elements.size
+
+		print_good "开始比对分析..."
+		# 按每台设备解析XML
+		root.elements.each do |device|
+			index += 1
+			info = {}
+			device_name = device.attributes['device_name']
+			device_hash[device_name] = info
+			
+			# 按照每个cmd解析XML
+			device.elements.each do |cmd|
+				i = {}
+				ch = cmd.name
+				cmd_name = cmd.attributes['cmd']
+				# 当同一个cmd有两个symbol时
+				if info[cmd_name].nil?
+					info[cmd_name] = {ch.to_sym => i}
+				else
+					info[cmd_name][ch.to_sym] = i
+				end
+
+				# 进度条
+				progress_bar(total, index, device_name)
+
+				# 还有一种情况是当<attr cmd='show' val='var'/>
+				# 此时就可以直接将val填入
+				attr_val = cmd.attributes['val']
+				attr_val ? info[cmd_name][ch.to_sym] = attr_val : to_h(cmd, i)
+			end
+		end
+
+		clear_line
+		return device_hash
+	end
+
+	def to_h(xml, hash)
+		xml.elements.each do |e|
+			if e.has_elements?
+				son = {}
+				hash[convert(e.name)] = son
+				to_h(e, son)
+			else
+				hash[convert(e.name)] = e.attributes['val']
+			end
+		end
+	end
+
+	# 转换恢复会出错字符串
+	def convert(e)
+		#/__/.match(e) ? e.gsub!('__', '/').to_sym : e.to_sym
+		e.gsub!('__', '/') if /__/.match(e)
+		e.gsub!(/^_i/, '') if /^_i/.match(e)
+		e.to_sym
+	end
+end

data/lib/modules/input/style/console.rb

@@ -0,0 +1,222 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::InputStyle::Console < Aio::Module::InputStyle
+	include Aio::Module
+	include Aio::Ui::Verbose
+
+	class Machine
+
+		attr_accessor :regs, :state
+
+		def initialize
+			@cmd_state = CmdState.new(self)
+			@context_state = ContextState.new(self)
+			@empty_state = EmptyState.new(self)
+			@cmd_context = CmdContext.new
+
+			@state = @empty_state
+		end
+
+		def state
+			@state.class
+		end
+
+		def cmd_context
+			@cmd_context
+		end
+
+		def to_cmd_state
+			@state = @cmd_state
+		end
+
+		def to_context_state
+			@state = @context_state
+		end
+
+		def to_empty_state
+			@state = @empty_state
+		end
+
+		def parse_line(line)
+			@state.parse_line(line)
+		end
+	end
+
+	class CmdContext
+
+		attr_accessor :device_name, :cmd, :context
+
+		# all_info = [[device_name, cmd, context] ... ]
+		attr_accessor :all_info
+
+		def initialize
+			@device_name = ""
+			@cmd = ""
+			@context = []
+			@all_info = []
+		end
+
+		def done
+			# 当之前有cmd内容的时候,比较两个的长度
+			# 长的保留
+			info_idx = has_cmd?
+			if info_idx
+				if @all_info[info_idx][2].size < context.size
+					@all_info[info_idx][2] = context
+				end
+			else
+				
+				# 如果都没有,那么就直接添加
+				@all_info << [device_name, cmd, context]
+			end
+			self.clear
+		end
+
+		def has_cmd?
+			@all_info.each_with_index do |info, idx|
+				next unless info[1] == device_name
+				return idx if info[2] == cmd
+			end
+
+			return nil
+		end
+
+		def clear
+			@device_name = ""
+			@cmd = ""
+			@context = []
+		end
+	end
+
+	class State
+
+		include Aio::Ui::Verbose
+
+		def initialize(machine)
+			@machine = machine
+		end
+
+		def cmd_context
+			@machine.cmd_context
+		end
+	end
+
+	class EmptyState < State
+		def parse_line(line)
+
+			# 当前是空的状态
+			# 判断当前行有没有模块中的命令
+			line = Aio::Base::Toolkit::String.safe(line)
+			return unless @machine.regs.match(line)
+
+			@machine.to_cmd_state
+			@machine.parse_line(line)
+		end
+	end
+
+	class CmdState < State
+		def parse_line(line)
+			# 判断是不是cmd命令行
+			res = /^[\[|<]?(?<name>.*)(#|>|\])(?<cmd>.*)/.match(line)
+			if res.nil?
+				@machine.to_empty_state
+				return
+			end
+
+			# 当已经发现了是cmd行,那么紧接下来就是收集文本内容
+			cmd_context.device_name = res[:name]
+			cmd_context.cmd = res[:cmd]
+			@machine.to_context_state
+		end
+	end
+
+	class ContextState < State
+
+		def parse_line(line)
+			# 如果这行是命令行格式,那么就代表上一个cmd内容结束
+			tmp_reg = ::Regexp.new "^[\\[|<]?#{cmd_context.device_name}(#|>|\\])"
+
+			# 可能会出现UTF-8识别不出摄氏度符号的问题
+			# 已经通过安全字符解决
+			begin
+				if tmp_reg.match(line)
+					cmd_context.done
+					
+					# 完成后将状态设置为空,并且重新分析这一行
+					# 毕竟如果这行也cmd行呢
+					@machine.to_empty_state
+					return @machine.parse_line(line)
+				end
+			rescue => e
+				print_error cmd_context.device_name + " 设备巡检信息编码出现问题"
+				print_error line + " : " + e.message
+			end
+
+
+			cmd_context.context << line
+		end
+	end
+			
+	def initialize
+		super({
+			:author				=> "Elin",
+			:description 	=> "这个模块用于对命令行格式的输入处理",
+			:platform 		=> "all",
+		})
+		@machine = Machine.new
+	end
+
+	def parse
+		dir = self.input_file
+		dir_pn = Pathname.new(dir)
+
+		# 加载所有的cmd正则表达式
+		@machine.regs = merge_regs(ext_info[:cmds_reg])
+
+		# 如果是单个文件的话,直接进入文件解析模式
+		if dir_pn.file?
+			parse_file(dir_pn)
+		else
+		
+			# 如果是文件夹的话,那么就对每个文件进行解析
+			Find.find(dir) do |file|
+				fn = Pathname.new(file)
+				parse_file(file) if fn.file?
+			end
+		end
+
+		@machine.cmd_context.all_info.each do |info|
+			yield info[0], info[1], info[2]
+		end
+	end
+
+	def parse_file(file)
+		fo = File.open(file, "r+", :encoding => "utf-8")
+		
+		fo.each_line do |line|
+			begin 
+				line = Aio::Base::Toolkit::String.safe(line)
+				line = line.strip
+			rescue => e
+				print_error line + " : " + e.message
+			end
+
+			@machine.parse_line(line)
+		end
+		
+		# 如果最后没有done结束,则手动结束
+		if @machine.state == ContextState
+			@machine.cmd_context.done 
+		end
+	end
+
+	def merge_regs(arr_regs)
+		regs = Aio::Base::Toolkit::Regexp.merge(arr_regs).to_s
+		regs.gsub!('(^', '(')
+		::Regexp.new(regs)
+	end
+end
+
+

data/lib/modules/input/style/console.rb.bak

@@ -0,0 +1,114 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::InputStyle::Console < Aio::Module::InputStyle
+	include Aio::Module
+
+	def initialize
+		super({
+			:author				=> "Elin",
+			:description 	=> "这个模块用于对命令行格式的输入处理",
+			:platform 		=> "all",
+		})
+	end
+
+	def parse(&block)
+		dir = self.input_file
+		dir_pn = Pathname.new(dir)
+
+		# 如果是单个文件的话,直接进入文件解析模式
+		if dir_pn.file?
+			parse_file(file) do |info1, info2, info3, info4|
+				block.call(info1, info2, info3, info4)
+			end
+		end
+		
+		# 如果是文件夹的话,那么就对每个文件进行解析
+		Find.find(dir) do |file|
+			fn = Pathname.new(file)
+			if fn.file?
+				parse_file(file) do |info1, info2, info3, info4|
+					block.call(info1, info2, info3, info4)
+				end
+			end
+		end
+	end
+
+	def parse_file(file)
+		fo = File.open(file)
+		regs = merge_regs(ext_info[:cmds_reg])
+
+		cmd = ""
+		context = []
+		device_name = ""
+		last_device_name = ""
+		# all_info = {cmd => [device_name, context]}
+		all_info = {}
+
+		fo_context = fo.each_line.to_a 
+		fo_size = fo_context.size
+		fo_context.each_with_index do |line, idx|
+			line = line.strip
+
+			# 如果不是命令行,首先看下有有没有等待命令
+			# 如果有则将后续信息方法context中
+			unless regs.match(line)
+
+				# 只有当有设备名的时候才进行判断
+				if ! device_name.empty?
+					tmp_reg = ::Regexp.new "^[\\[|<]?#{device_name}(#|>|\\])"
+					# 保证是输入命令的line,而不是其他形式
+				  if tmp_reg.match(line)
+						last_device_name = device_name
+						device_name = ""
+					else	# 其他形式则作为文本输出
+						context << line if not cmd.empty?
+						if idx == fo_size and ! cmd.empty?
+							all_info[cmd] = [last_device_name, context]
+						end
+					end
+				end
+
+				next
+			end
+
+			# 匹配devicename# cmd 的模式
+			res = /^[\[|<]?(?<name>.*)(#|>|\])(?<cmd>.*)/.match(line)
+			next if res.nil?
+
+			# 到这里的是line信息中包含了cmd信息以及设备名信息
+			if not cmd.empty? and not context.empty?
+				# 有一种情况是如果之前有了cmd,后面又出现同样的cmd
+				# 那么比较两个cmd的内容,内容行数多的保留
+				# 感觉有问题,会少一个cmd
+				if all_info.has_key?(cmd)
+					if all_info[cmd][1].size > context.size 
+						device_name = ""
+						cmd = ""
+						context = []
+						next
+					end
+				end
+				all_info[cmd] = [last_device_name, context]
+				device_name = ""
+				cmd = ""
+				context = []
+			end
+			device_name = res[:name]			
+			cmd = res[:cmd]
+		end
+
+		all_info.each_pair do |cmd, info|
+			yield info[0], cmd, info[1], ""
+		end
+	end
+
+	def merge_regs(arr_regs)
+		regs = Aio::Base::Toolkit::Regexp.merge(arr_regs).to_s
+		regs.gsub!('(^', '(')
+		::Regexp.new(regs)
+	end
+end
+
+

data/lib/modules/input/style/xinhua_nat.rb

@@ -0,0 +1,70 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::InputStyle::XinhuaNat < Aio::Module::InputStyle
+	include Aio::Module
+
+	Xinhua_Routing = /^------------------------------------------------------------$/
+
+	def initialize
+		super({
+			:author				=> "Elin",
+			:description	=> "这个模块专门针对中行网络标准化管理NAT软件",
+			:platform			=> "ausware",
+			:file_suffix	=> Regexp.new('\.diag$'),
+			:pass_file		=> [
+							Regexp.new('RoutingTable.*'),
+							Regexp.new('路由状态'),
+						]
+		})
+	end
+
+	def parse
+		# dir 为 Pathname 类
+		dir = self.input_file
+		
+		# 分析第一层： 文件夹名称包含设备名称+管理IP
+		::Dir.foreach(dir) do |device_layer|
+			if device_layer.downcase == "." or device_layer.downcase == ".."
+				next
+			end
+
+			tmp = []
+			device_layer.reverse.split('_', 2).each do |res|
+				tmp << res.reverse
+			end
+
+			manager_ip  = tmp[0]
+			device_name = tmp[1]
+
+
+			# 分析第二层：文件后缀为.diag，此外还有个HSRP文件夹，但是可有一次搞定
+			Find.find(File.join(dir, device_layer)) do |context_layer|
+
+				# 如果文件后缀名不是.diag则跳过
+				if ! self.file_suffix?(context_layer)
+					next
+				end
+
+				# 检查是否是需要略过的文件
+				if self.pass_file?(context_layer)
+					next
+				end
+
+				fo = File.open(context_layer, "r+", :encoding => "utf-8")
+				context = fo.each_line.to_a
+
+				# 分割输出，在路由状态中会有这种情况
+				split(context, Xinhua_Routing).each do |cont|
+					cmd = cont.shift.strip
+					yield device_name, cmd, cont, manager_ip
+				end
+			end
+		end
+	end
+
+	def split(array, pattern)
+		Aio::Base::Toolkit::Array.split(array, pattern)
+	end
+end