aio_elin 0.0.1

data/lib/aio/ui/banner.rb

@@ -0,0 +1,85 @@
+#coding=utf-8
+
+module Aio::Ui
+
+	class Banner
+
+		attr_accessor :mode
+		attr_accessor :type
+		attr_accessor :input_style, :input
+		attr_accessor :ranking
+		attr_accessor :output_style, :output
+		attr_accessor :compare
+		
+		def banner
+			res = %Q^
+Usage: ./aio [option] <var>
+
+Options:
+	-l,   --list              [type]            显示所有或者所选type类型的模块信息
+	-c,   --compare           <path1>           被比较的文件路径
+	-is,  --input-style       <mod>             输入的模块名称
+	-i,   --input             <path>            输入信息的绝对路径
+	-r,   --ranking           <level>           选择命令的级别
+	-os,  --output-style      <mod>             输出的模块名称
+	-o,   --output            <path>            输出信息的路径
+	-h,   --help                                帮助
+^
+			print res
+			exit 0
+		end
+
+		def initialize
+			@type = ""
+		end
+
+		def parse(args=[])
+			banner if args.empty?
+
+			while(!args.empty?) do 
+				arg = args.shift
+				case arg
+				when "-l", "--list"
+					@type = args.shift unless args[0] =~ /-/
+					self.mode = 0
+				when "-is", "--inupt-style"
+					@input_style = check(args.shift)
+					self.mode = 1
+				when "-i", "--input"
+					@input = check(args.shift)
+				when "-r", "--ranking"
+					@ranking = check(args.shift)
+				when "-c", "--compare"
+					@compare = check(args.shift)
+				when "-os", "--output-style"
+					@output_style = check(args.shift)
+					self.mode = 1
+				when "-o", "--output"
+					@output = check(args.shift)
+				when "-h", "--help"
+					banner
+				end
+			end
+
+			self
+		end
+
+		# 检查参数是否合法
+		def check(opt)
+			opt = opt.to_s
+			if opt.empty? and opt =~ /-/
+				banner
+			end
+
+			return opt
+		end
+
+		# 判断是否有 input 信息
+		def has_input?
+			return false if @input_style.nil? and @input.nil?
+			return true
+		end
+			
+	end
+end
+

data/lib/aio/ui/readable_text.rb

@@ -0,0 +1,75 @@
+module Aio::Ui
+
+	class ReadableText
+
+		DefaultIndent	= 2
+
+		class << self
+			
+			def dump_module_cmds(mods, h=nil, indent=DefaultIndent)
+				col = ["cmd name", "ranking", "device", "description"]
+				table = Aio::Base::Toolkit::Table.new(
+					'Header'	=> h ||= 'Aio CMD Modules',
+					'Indent'	=> indent,
+					'Columns'	=> col,
+					'CellPad'	=> 2,
+					'SortIndex'	=> 2)
+
+				# info => { :module_type,
+				# 					:module_layer_2
+				#						:full_path
+				#    				:klass
+				mods.each_pair do |mod_name, info|
+					row = []
+					row << info[:klass].cmd_full
+					row << info[:klass].ranking.to_s
+					row << info[:module_layer_2]
+					row << info[:klass].description
+					table << row
+				end
+
+				table.to_s + "\n"
+			end
+
+			def dump_module_inputs(mods, h=nil, indent=DefaultIndent)
+				col = ["input style", "platform", "description"]
+				table = Aio::Base::Toolkit::Table.new(
+					'Header'		=> h || "Aio Input Modules",
+					'Indent'		=> indent,
+					'Columns'		=> col,
+					'CellPad'		=> 2)
+
+				mods.each_pair do |mod_name, info|
+					row = []
+					row << mod_name
+					row << info[:klass].platform
+					row << info[:klass].description
+					table << row
+				end
+
+				table.to_s + "\n"
+			end
+
+			def dump_module_outputs(mods, h=nil, indent=DefaultIndent)
+				col = ["output style", "suffix", "description"]
+				table = Aio::Base::Toolkit::Table.new(
+					'Header'		=> h || "Aio Output Modules",
+					'Indent'		=> indent,
+					'Columns'		=> col,
+					'CellPad'		=> 2)
+
+				mods.each_pair do |mod_name, info|
+					row = []
+					row << mod_name
+					row << info[:klass].file_suffix
+					row << info[:klass].description
+					table << row
+				end
+
+				table.to_s + "\n"
+			end
+						
+			
+		end # self
+	end
+end

data/lib/aio/ui/verbose.rb

@@ -0,0 +1,29 @@
+#
+# 这个类主要是用来显示进度信息以及进度条
+# 方便Debug 和确认进度
+#
+module Aio::Ui
+	module Verbose
+
+		def progress_bar(total, num, info)
+			percent = ((num.to_f / total.to_f) * 100).to_i
+			buf = "#" * (percent / 2)
+			#printf("#{' ' * 120}\r")
+			printf("[%-50s] [%d%%], [%s]\r", buf, percent, info)
+		end
+			
+		def print_good(str)
+			puts "[+] " << str
+		end
+
+		def print_error(str)
+			puts "[-] " << str
+		end
+
+		def clear_line
+			printf("#{' ' * 100}\r")
+		end
+		
+	end
+
+end

data/lib/aio/ui/version.rb

@@ -0,0 +1,9 @@
+module Aio
+module Ui
+
+module Version
+	RunVersion = "1.0.0"
+end
+end
+end
+

data/lib/aio/ui.rb

@@ -0,0 +1,138 @@
+#coding=utf-8
+
+module Aio::Ui
+	require "aio/ui/version"
+	require "aio/ui/banner"
+	require "aio/ui/readable_text"
+	require "aio/ui/verbose"
+
+	class Console
+
+		include Aio::Ui::Verbose
+		
+		# Aio::ModuleManager
+		attr_accessor :module_manager
+
+		# Aio::DeviceManager
+		attr_accessor :device_manager
+
+		# Aio::Parse::Parser
+		attr_accessor :parser
+
+		# Aio::Ui::Banner
+		attr_accessor :banner
+
+		DefaultCompare = "special/style/compare"
+
+		DefaultInputComapreXML = "input/style/compare_xml"
+
+		def initialize(args)
+			self.banner = Banner.new.parse(args)
+		end
+
+		def process
+			case self.banner.mode
+			when 0
+				proc_0
+			when 1
+				proc_1
+			end
+		end
+
+		# 输出模块表
+		def proc_0
+			case banner.type
+			when "cmd"
+				mods = module_manager.get_modules_by_type("cmd")
+				print ReadableText.dump_module_cmds(mods)
+			when "input"
+				mods = module_manager.get_modules_by_type("input")
+				print ReadableText.dump_module_inputs(mods)
+			when "output"
+				mods = module_manager.get_modules_by_type("output")
+				print ReadableText.dump_module_outputs(mods)
+			else
+				mods = module_manager.get_modules_by_type("cmd")
+				print ReadableText.dump_module_cmds(mods)
+				
+				mods = module_manager.get_modules_by_type("input")
+				print ReadableText.dump_module_inputs(mods)
+
+				mods = module_manager.get_modules_by_type("output")
+				print ReadableText.dump_module_outputs(mods)
+			end
+		end
+
+		# 通过选择的输入输出模块进行操作
+		def proc_1
+
+			# 当完全没有指定input的时候,是可以跳过的
+			if banner.has_input?
+				input_klass = module_manager.get_module_klass_by_name(banner.input_style)
+				input_klass.input_file = banner.input
+				input_klass.ext_info = {:cmds_reg => device_manager.just_cmds_reg}
+
+				parser.input_klass = input_klass
+				parser.parse_by_module
+
+				# 当有compare选项的时候，同时解析compare模块，并合并warning信息
+				if banner.compare 
+					print_good "正在加载比较模块"
+					if banner.compare =~ /\.xml$/
+						input_klass_compare = module_manager.get_module_klass_by_name(DefaultInputComapreXML)
+					else
+						print_error "请确认 -c 参数文件后缀"
+						banner.banner
+					end
+
+					input_klass_compare.input_file = banner.compare
+					compare_klass = module_manager.get_module_klass_by_name(DefaultCompare)
+
+					print_good "比较模块加载成功"
+					print_good "正在进行对比..."
+					parser.input_klass = input_klass_compare
+					parser.parse_by_compare(compare_klass)
+					print_good "对比分析完成"
+					
+				end
+				
+				# Debug 处
+				if Aio::Base::Debug.ui_enable? #or true
+					puts "[Debug UI]"
+					device_manager.devices.each_pair{|name, device_klass|
+						pp name
+						pp device_klass.cmds_useful["show version"]
+						#pp device_klass.device_name
+						#pp device_klass.cmds_useful
+						#pp device_klass.warning_klass.warning_info
+						puts
+					}
+					puts "[/Debug UI]"
+				end
+
+				device_manager.warning_summarize_tidy
+			end	# 判断没有没 input 尾部
+
+			print_good "正在加载输出模块 #{banner.output_style}"
+			output_klass = module_manager.get_module_klass_by_name(banner.output_style)
+
+			pn = Pathname.new(banner.output)
+			dir, base = pn.split
+			base = base.to_s
+			unless base =~ /\./
+				suffix = output_klass.file_suffix
+				unless suffix == "" or /(?i)null/ =~ suffix
+					base << "." + suffix
+				end
+			end
+
+			print_good "正在输出文件 #{dir + base}"
+			output_klass.device_manager = device_manager
+			output_klass.module_manager = module_manager
+			output_klass.output_file = banner.output
+			output_klass.generate
+			print_good "输出文件完成"
+		end
+
+	end
+end

data/lib/aio.rb

@@ -0,0 +1,13 @@
+#!/usr/bin/env ruby
+
+cwd = File.expand_path(File.dirname(__FILE__))
+#cwd_lib = File.join(cwd, "lib")
+$: << cwd
+
+BaseFile = cwd
+
+# core
+require File.join(cwd, "aio", "core.rb")
+
+# config
+require File.join(cwd, "../config.rb")

data/lib/modules/cmd/cisco/show_clock.rb

@@ -0,0 +1,32 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::Cmd::Cisco::ShowClock < Aio::Module::Cmd::Cisco
+  include Aio::Module
+
+  def initialize
+    super({
+      :cmd_full     => "show clock",
+      :cmd_short    => "sh clo",
+      :author       => "Elin",
+      :description  => "This is Cisco Command# show clock",
+      :ranking      => Ranking_1,
+      :platform     => "all",
+    })
+	end
+
+	def parse
+		context = self.context.dup
+		clock = {}
+		useful[:clock] = clock
+		context.readline_match_block(/(?<time>.*)\.\d+ (?<zone>[^\s]+) (?<week>[^\s]+) (?<month>[^\s]+) (?<day>[^\s]+) (?<year>[^\s]+)/) do |block|
+			block.update(clock, :time)
+			block.update(clock, :zone)
+			block.update(clock, :week)
+			block.update(clock, :year)
+			block.update(clock, :month, Aio::Base::Toolkit::Date.month_to_i(block[:month]))
+			block.update(clock, :day)
+		end
+	end
+end

data/lib/modules/cmd/cisco/show_cpu.rb

@@ -0,0 +1,32 @@
+#coding=utf-8
+
+class Aio::Module::Cmd::Cisco::ShowCPU < Aio::Module::Cmd::Cisco
+	include Aio::Module
+	
+	def initialize
+		super({
+			:cmd_full			=> "show cpu",
+			:cmd_short		=> "sh cpu",
+			:author				=> "Elin",
+			:description	=> "This is Cisco ASA Command# show cpu",
+			:ranking			=> Ranking_1,
+			:platform			=> "asa",
+			:benchmark		=> {
+					:seconds_5	=> ["<", 50],
+					:minute_1		=> ["<", 50],
+					:minutes_5	=> ["<", 50],
+				}
+		})
+	end
+
+	def parse
+		cont = self.context.dup
+		cpu = {}
+		useful[:cpu] = cpu
+		cont.readline_match_block(/1 minute: (?<minute_1>\d+)%; 5 minutes: (?<minutes_5>\d+)%/) do |block|
+		  block.warning_serious(cpu, :minute_1, self)
+		  block.warning_serious(cpu, :minutes_5, self)
+		end
+	end
+end
+

data/lib/modules/cmd/cisco/show_env.rb

@@ -0,0 +1,71 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::Cmd::Cisco::ShowEnv < Aio::Module::Cmd::Cisco
+	include Aio::Module
+
+	def initialize
+		super({
+			:cmd_full 	=> "show environment",
+			:cmd_short	=> "sh env",
+			:author			=> "Elin",
+			:description	=> "This is Cisco Command# show environment",
+			:ranking		=> Ranking_1,
+			:platform		=> "all",
+			:benchmark	=> {
+				:power_status	=> ["==", "Normal"],
+				:fan_status		=> ["==", "OK"],
+				:cpu_status		=> ["==", "Normal"],
+			}
+		})
+	end
+
+	def parse
+		cont = self.context.dup
+
+		# 是有版本区别的，要分开对待
+		case device_template
+		when Template_1, Template_2
+			parse_tmp_1(cont)
+		end
+	end
+
+	def parse_tmp_1(cont)
+		useful[:power] = {}
+		cont.readline_match_block(/SYSTEM POWER SUPPLY STATUS/)
+		cont.readline_match_block(/==========================/)
+		cont.readline_range_loop(/Internal Power/, reg_blank) do |cont_layer|
+			power = {}
+			cont_layer.readline_match_block(/Internal Power Supply (?<power_id>\d+) Type: (?<type>.*)/) do |block|
+				power_id = block[:power_id]
+				useful[:power]["slot_#{power_id}"] = power
+				block.update(power, :power_id)
+				block.update(power, :type)
+			end
+			cont_layer.readline_match_block(/Output Status: (?<power_status>.*)/) do |block|
+				block.warning_serious(power, :power_status, self)
+			end
+		end
+
+		useful[:fan] = {}
+		cont.readline_match_block(/SYSTEM FAN STATUS/)
+		cont.readline_match_block(/=================/)
+		cont.readline_match_block_loop(/Fan (?<fan_id>\d+) (?<fan_status>.*),/, reg_blank) do |block|
+			fan = {}
+			fan_id = block[:fan_id]
+			useful[:fan]["slot_#{fan_id}"] = fan
+			block.update(fan, :fan_id)
+			block.warning_serious(fan, :fan_status, self)
+		end
+
+		cont.readline_match_block(/SYSTEM TEMPERATURE STATUS/)
+		cont.readline_match_block(/=========================/)
+		cont.readline_match_block(/CPU temperature: (?<cpu_temperature>\d+) Celsius, (?<cpu_status>.*)/) do |block|
+			useful[:temperature] = {}
+			block.update(useful[:temperature], :cpu_temperature)
+			block.warning_serious(useful[:temperature], :cpu_status, self)
+		end
+
+	end
+end

data/lib/modules/cmd/cisco/show_failover.rb

@@ -0,0 +1,39 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::Cmd::Cisco::ShowFailover < Aio::Module::Cmd::Cisco
+	
+	include Aio::Module
+
+	def initialize
+		super({
+			:cmd_full				=> "show failover",
+			:cmd_short			=> "sh fail",
+			:author					=> "Elin",
+			:description		=> "This is Cisco ASA Command# show failover",
+			:ranking				=> Ranking_1,
+			:platform				=> FW,
+		})
+	end
+
+	def parse
+		context = self.context.dup
+		failover = {}
+		useful[:failover] = failover
+		context.readline_match_block(/Failover (?<status>.*)/) do |block|
+			block.update(failover, :status)
+		end
+
+		if failover[:status] =~ /On/
+			context.readline_match_block(/Failover unit (?<standby>.*)/) do |block|
+				block.update(failover, :standby)
+			end
+			context.readline_match_block(/Failover LAN Interface: failover (?<interface>[^\s]+) \((?<interface_status>[^\)]+)\)/) do |block|
+				block.update(failover, :interface)
+				block.update(failover, :interface_status)
+			end
+		end
+	end
+end
+

data/lib/modules/cmd/cisco/show_interface_ip_brief.rb

@@ -0,0 +1,24 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::Cmd::Cisco::ShowInterfaceIpBrief < Aio::Module::Cmd::Cisco
+	include Aio::Module
+	include Aio::Ui::Verbose
+
+	def initialize
+		super({
+			:cmd_full				=> "show interface ip brief",
+			:cmd_short			=> "sh int ip b",
+			:author					=> "Elin",
+			:description		=> "This is Cisco ASA Command# show interface ip brief",
+			:ranking				=> Ranking_1,
+			:platform				=> FW,
+		})
+	end
+
+	def parse
+		call_cmd(Aio::Module::Cmd::Cisco::ShowIpInterfaceBrief)
+	end
+end
+