aio_elin 0.0.1

data/lib/aio/core/module/input_style.rb

@@ -0,0 +1,98 @@
+#coding=utf-8
+
+require "pathname"
+
+module Aio::Module
+	class InputStyle
+		
+		class ErrorModuleInput < StandardError
+			def to_s; "请重载此方法"; end
+		end
+
+		# 重要！
+		# 输入文件
+		attr_accessor :input_file
+
+		# 信息
+		attr_accessor 	:input_info
+
+		# 外加信息,用于针对特殊情况
+		attr_accessor :ext_info
+
+		def initialize(info={})
+			@input_info = info.dup
+			set_defaults()
+		end
+
+		def set_defaults
+			self.input_info = {
+				:author				=> nil,
+				:description 	=> "No Input Description",
+				# 产生文件的平台
+				:platform			=> nil,
+				:file_suffix	=> nil,
+				:pass_file		=> nil,
+				:license   		=> ::Aio::AIO_LICENSE,
+			}.update(@input_info)
+
+		end
+
+		def type
+			Aio::Module::InputStyle
+		end
+
+		# 保证输出的file为Pathname类
+		def check_file(file)
+			if file.class != Pathname
+				case file.class
+				when String
+					file = Pathname.new(file)
+				when File
+					file = Pathname.new(file.path)
+				end
+			end
+			file
+		end
+
+		# 请重载此方法
+		def parse
+			raise ErrorModuleInput.new			
+		end
+
+		# 检查是否符合文件后缀，符合返回真
+		def file_suffix?(file)
+			if self.file_suffix.match(file)
+				return true
+			end
+			return false
+		end
+
+		# 检查是否符合pass_file， 符合返回真
+		def pass_file?(file)
+			self.pass_file.each do |pass|
+				if pass.match(Aio::Base::Toolkit::String.safe(file))
+					return true
+				end
+			end
+			return false
+		end
+
+		def input_file
+			@input_info[:input_file]
+		end
+
+		def input_file=(file)
+			file = check_file(file)
+			@input_info[:input_file] = file
+		end
+
+		def author;    		self.input_info[:author];    		end 
+		def description;  self.input_info[:description];  end 
+		def platform;  		self.input_info[:platform];  		end 
+		def file_suffix;	self.input_info[:file_suffix];	end
+		def license;   		self.input_info[:license];   		end 
+		def pass_file;		self.input_info[:pass_file];		end
+
+
+	end	# class
+end	# module

data/lib/aio/core/module/output_style.rb

@@ -0,0 +1,80 @@
+#coding=utf-8
+
+module Aio::Module
+	class OutputStyle
+		
+		class OutputMethodError < StandardError
+			def to_s
+				"请重载此方法"
+			end
+		end
+
+		# 输出的文件名
+		attr_accessor :output_file
+
+		# 信息
+		attr_accessor :output_info
+
+		# device_manager 加载
+		attr_accessor :device_manager
+
+		# module_manager 加载
+		attr_accessor :module_manager
+
+		def initialize(info={})
+			@output_info = info.dup
+			set_defaults()
+		end
+
+		def set_defaults
+			self.output_info = {
+				:author				=> nil,
+				:description  => "No Output Description",
+				:file_suffix  => nil,
+				:license			=> ::Aio::AIO_LICENSE,
+			}.update(@output_info)
+		end
+
+		def author;				self.output_info[:author]; end
+		def description;	self.output_info[:description]; end
+		def file_suffix;	self.output_info[:file_suffix]; end
+		def license;			self.output_info[:license];	end
+
+		def type
+			Aio::Module::OutputStyle
+		end
+
+		# 输入的文件名，判断是否有后缀
+		# 如果没有则添加后缀
+		def output_file=(file_name)
+			file_pathname = Pathname.new(file_name)
+			if file_pathname.extname.empty?
+			  base, name = file_pathname.split
+				suffix = output_info[:file_suffix]
+
+				# 当不需要suffix, 将其设为空
+				unless /^\./ =~ suffix
+					if suffix == "" or /(?i)null/ =~ suffix
+						suffix = ""
+					else
+						suffix = "." + suffix
+					end
+				end
+			  name = name.to_s + suffix
+			  file_pathname = Pathname.new(base + name)
+			end
+			@output_file = file_pathname
+		end
+
+		def each_devices_with_useful
+			self.device_manager.each_devices_with_useful do |k, v|
+				yield k, v
+			end
+		end
+
+		# 生成文件
+		def generate
+			raise OutputMethodError.new
+		end
+	end
+end

data/lib/aio/core/module/platform.rb

@@ -0,0 +1,30 @@
+module Aio::Module
+
+	class Platform
+		attr_reader :platform
+	end
+
+	class RT < Platform
+		def self.to_s; "route"; end
+	end
+
+	class SW < Platform
+		def self.to_s; "switch"; end
+	end
+
+	class FW < Platform
+		def self.to_s; "failwall"; end
+	end
+
+	class RS < Platform
+		def self.to_s; "route & switch"; end
+	end
+
+	class RF < Platform
+		def self.to_s; "route & failwall"; end
+	end
+
+	class ALL < Platform
+		def self.to_s; "route & switch & failwall"; end
+	end
+end

data/lib/aio/core/module/ranking.rb

@@ -0,0 +1,24 @@
+module Aio::Module
+
+	class	Ranking
+		attr_reader :ranking
+	end
+
+	class Ranking_1 < Ranking
+		def self.to_s; "level-1"; end
+	end
+	class Ranking_2 < Ranking
+		def self.to_s; "level-2"; end
+	end
+	class Ranking_3 < Ranking
+		def self.to_s; "level-3"; end
+	end
+	class Ranking_4 < Ranking
+		def self.to_s; "level-4"; end
+	end
+	class Ranking_5 < Ranking
+		def self.to_s; "level-5"; end
+	end
+
+end
+	

data/lib/aio/core/module/special_style.rb

@@ -0,0 +1,30 @@
+#coding=utf-8
+
+module Aio::Module
+	class SpecialStyle
+		class MethodError < StandardError; end
+
+		attr_accessor :special_info
+
+		def initialize(info={})
+			@special_info = info.dup
+			set_defaults()
+		end
+
+		def set_defaults
+			self.special_info = {
+				:author 		=> nil,
+				:description => nil,
+				:license			=> ::Aio::AIO_LICENSE,
+			}.update(@special_info)
+		end
+
+		def type
+			Aio::Module::SpecialStyle
+		end
+
+		def parse
+			raise MethodError, "请重载此方法"
+		end
+	end
+end

data/lib/aio/core/module/template.rb

@@ -0,0 +1,13 @@
+module Aio::Module
+
+require "singleton"
+class Template
+	include Singleton
+end
+class Template_1 < Template; end
+class Template_2 < Template; end
+class Template_3 < Template; end
+class Template_4 < Template; end
+class Template_5 < Template; end
+
+end

data/lib/aio/core/module.rb

@@ -0,0 +1,15 @@
+module Aio
+module Module
+
+	require "aio/core/module/ranking"
+	require "aio/core/module/platform"
+	require "aio/core/module/template"
+	require "aio/core/module/cmd"
+	require "aio/core/module/input_style"
+	require "aio/core/module/output_style"
+	require "aio/core/module/special_style"
+	require "aio/core/module/description"
+
+end
+end
+

data/lib/aio/core/module_loader.rb

@@ -0,0 +1,161 @@
+#这一部分用来加载Module下各模块
+
+module Aio::Module
+	require "find"
+	require "pathname"
+
+	class Loader
+
+	Reg_klass_filter = /class (.*)[\s]+<[\s]+Aio::Module/
+	Reg_ruby_file = /.rb$/
+
+	attr_accessor :module_manager
+
+		def initialize(module_manager)
+			@module_manager = module_manager
+		end
+			
+		# 分析各个文件夹目录
+		# 目录格式 ~/aio/lib/modules/cisco/show_version.rb
+		# 目录格式 ~/aio/lib/modules/cmd/cisco/show_version.rb
+		# 目录格式包括了 cmd, input, output
+		# cmd 放置各个命令模块
+    # input 放置各种输入方式的模块
+		# output 放置各种输出方式的模块
+		def each_module_reference_name(path, opts={})
+			::Dir.foreach(path) do |entry|
+				if entry.downcase == "." or entry.downcase == ".."
+					next
+				end
+
+				full_entry_path = ::File.join(path, entry)
+				module_type = entry
+				
+				# 第一层目录结构，判断是否为cmd, input, output 的模块类型
+				unless ::File.directory?(full_entry_path) and
+										module_manager.module_type_enable?(module_type)
+					next
+				end
+
+				# module_type = "cmd"
+				# full_entry_path = "~/aio/lib/modules/cmd"
+
+				full_entry_pathname = Pathname.new(full_entry_path)
+
+				Find.find(full_entry_path) do |entry_descendant_path|
+					if File.directory?(entry_descendant_path)
+						next
+					end
+
+					# 判断是不是.rb 结尾，而不是 .rb.swp 结尾
+					unless vaild_ruby_file?(entry_descendant_path)
+						next
+					end
+
+					# entry_descendant_path 为ruby文件的完整绝对路径
+					entry_descendant_pathname = Pathname.new(entry_descendant_path)
+
+					# 获得 modules 的二级目录名称, cmd/cisco 中的cisco 
+					# 或者 input/style 中的style
+					base, _i = entry_descendant_pathname.split
+					_i, module_layer_2 = base.split
+					module_layer_2 = module_layer_2.to_s
+
+					# 查询是否有效并且取得模块的类名
+					module_klass_name = get_module_klass_name(entry_descendant_pathname)
+					if module_klass_name.empty?
+						next
+					end
+
+					# 获得参考名
+					relative_entry_descendant_pathname = entry_descendant_pathname.relative_path_from(full_entry_pathname)
+					_i, relative_entry_descendant_pathname = relative_entry_descendant_pathname.split
+					relative_entry_descendant_path = relative_entry_descendant_pathname.to_s
+					module_reference_name = module_reference_name_from_path(relative_entry_descendant_path)
+
+					module_reference_name = [module_type, module_layer_2, module_reference_name].join("/")
+					yield entry_descendant_path, module_type, module_layer_2, module_reference_name, module_klass_name
+				end
+
+			end
+		end
+
+		# 取模块的类名，以便以后调用
+		def get_module_klass_name(full_path)
+			fo = File.open(full_path)
+			fo.each_line do |l|
+				l = Aio::Base::Toolkit::String.safe(l)
+				klass_name = Reg_klass_filter.match(l)
+				if klass_name == nil
+					next
+				end
+				klass_name = klass_name[1].strip
+				return klass_name
+			end
+			return ""
+		end
+
+		def module_reference_name_from_path(path)
+			path.gsub(Reg_ruby_file, '')
+		end
+
+		# 是否是有效的Ruby文件
+		def vaild_ruby_file?(path)
+			if Reg_ruby_file.match(path)
+				return true
+			end
+			return false
+		end
+
+		# 全局加载模块
+		def load_modules(path, opts={})
+			count_by_module_type = {}
+
+			each_module_reference_name(path, opts={}) do |full_module_path, module_type, module_layer_2, module_reference_name, module_klass_name|
+				load_module(full_module_path, 
+										module_type,
+										module_layer_2,
+										module_reference_name,
+										module_klass_name,
+										{
+						#					:recalcuate_by_device_type => recalculate_by_device_type,
+											:count_by_module_type => count_by_module_type,
+										}
+				)
+											
+			end
+			module_manager.notify({
+				:count_by_module_type => count_by_module_type,
+			})
+		end
+			
+		# 真正加载处
+		# 并将模块全部实例化
+		def load_module(full_module_path, module_type,module_layer_2, module_reference_name, module_klass_name, opts={})
+
+			begin
+				require "#{full_module_path}"
+				module_klass = ::Object::const_get(module_klass_name).new
+			rescue Exception
+				puts "[-]	Can not load module: #{full_module_path}"
+				#puts caller
+				return
+			end
+			@module_manager.add_module(full_module_path, module_type, module_layer_2, module_reference_name, module_klass)
+
+			# 计数
+			# count_by_module_type = {"cmd" => { "cisco" => 2 }, "input" => { "style" => 1 } }
+			count_by_module_type = opts[:count_by_module_type]
+			if count_by_module_type
+				if !count_by_module_type[module_type]
+					count_by_module_type[module_type] = {}
+				end
+
+				count_by_module_type[module_type][module_layer_2] ||= 0
+				count_by_module_type[module_type][module_layer_2] += 1
+			end
+		end
+
+	end	#class
+end		#module
+		

data/lib/aio/core/module_loader.rb.bak

@@ -0,0 +1,148 @@
+#这一部分用来加载Module下各模块
+
+module Aio::Module
+	require "find"
+	require "pathname"
+
+	class Loader
+
+	Reg_klass_filter = /class (.*) < Aio::Module::Cmd/
+	Reg_ruby_file = /.rb$/
+
+	attr_accessor :module_manager
+
+		def initialize(module_manager)
+			@module_manager = module_manager
+		end
+			
+		# 分析各个文件夹目录
+		# 目录格式 ~/aio/lib/modules/cisco/show_version.rb
+		def each_module_reference_name(path, opts={})
+			::Dir.foreach(path) do |entry|
+				if entry.downcase == "." or entry.downcase == ".."
+					next
+				end
+
+				full_entry_path = ::File.join(path, entry)
+				device_type = entry
+
+				unless ::File.directory?(full_entry_path) and 
+										module_manager.device_type_enable?(device_type)
+					next
+				end
+
+				# device_type = "cisco"
+				# full_entry_path = "~/aio/lib/modules/cisco"
+
+				full_entry_pathname = Pathname.new(full_entry_path)
+
+				Find.find(full_entry_path) do |entry_descendant_path|
+					if File.directory?(entry_descendant_path)
+						next
+					end
+
+					# 判断是不是.rb 结尾，而不是 .rb.swp 结尾
+					unless vaild_ruby_file?(entry_descendant_path)
+						next
+					end
+
+					# entry_descendant_path 为ruby文件的完整绝对路径
+					entry_descendant_pathname = Pathname.new(entry_descendant_path)
+
+					# 查询是否有效并且取得模块的类名
+					module_klass_name = get_module_klass_name(entry_descendant_pathname)
+					if module_klass_name.empty?
+						next
+					end
+
+					relative_entry_descendant_pathname = entry_descendant_pathname.relative_path_from(full_entry_pathname)
+					relative_entry_descendant_path = relative_entry_descendant_pathname.to_s
+					module_reference_name = module_reference_name_from_path(relative_entry_descendant_path)
+					yield entry_descendant_path, device_type, module_reference_name, module_klass_name
+				end
+
+			end
+		end
+
+		# 取模块的类名，以便以后调用
+		def get_module_klass_name(full_path)
+			fo = File.open(full_path)
+			fo.each_line do |l|
+				l = safe_string(l)
+				klass_name = Reg_klass_filter.match(l)
+				if klass_name == nil
+					next
+				end
+				klass_name = klass_name[1].strip
+				return klass_name
+			end
+			return ""
+		end
+
+		def module_reference_name_from_path(path)
+			path.gsub(Reg_ruby_file, '')
+		end
+
+		def vaild_ruby_file?(path)
+			if Reg_ruby_file.match(path)
+				return true
+			end
+			return false
+		end
+
+		# 遇到了 invalid byte sequence in UTF-8 (ArgumentError) 问题
+		# 解决办法参考 https://stackoverflow.com/questions/29877310/invalid-byte-sequence-in-utf-8-argumenterror
+		def safe_string(str)
+			if ! str.valid_encoding?
+				str = str.encode("UTF-16be", :invalid=>:replace, :replace=>"?").encode('UTF-8')
+			end
+			return str
+		end
+
+		# 全局加载模块
+		def load_modules(path, opts={})
+			count_by_device_type = {}
+
+			each_module_reference_name(path, opts={}) do |full_module_path, device_type, module_reference_name, module_klass_name|
+				load_module(full_module_path, 
+										device_type,
+										module_reference_name,
+										module_klass_name,
+										{
+						#					:recalcuate_by_device_type => recalculate_by_device_type,
+											:count_by_device_type => count_by_device_type,
+										}
+				)
+											
+			end
+			module_manager.notify({
+				:count_by_device_type => count_by_device_type,
+			})
+		end
+			
+		# 真正加载处
+		# 并将模块全部实例化
+		def load_module(full_module_path, device_type, module_reference_name, module_klass_name, opts={})
+
+
+			require "#{full_module_path}"
+			begin
+				module_klass = Object::const_get(module_klass_name).new
+			rescue Exception
+				puts "[-]	Can not load module: #{full_module_path}"
+				puts caller
+				return
+			end
+			@module_manager.add_module(full_module_path, device_type, module_reference_name, module_klass)
+
+			# 计数
+			count_by_device_type = opts[:count_by_device_type]
+			if count_by_device_type
+				count_by_device_type[device_type] ||= 0
+				count_by_device_type[device_type] += 1
+			end
+		end
+
+	end	#class
+end		#module
+		

data/lib/aio/core/module_manager.rb

@@ -0,0 +1,112 @@
+# 管理，保存各个加载的模块
+class Aio::ModuleManager
+	
+	attr_accessor :modules, :modules_count
+
+	include Aio::Ui::Verbose
+
+	# NOTE modules[reference_name] = {
+	# 							:module_type     // 是cmd, input, output
+	#								:module_layer_2  // cisco,huawei , style
+	#	  						:full_path
+  #							  :klass
+	def initialize
+#		self.module_info_by_path = {}
+#		self.module_load_error_by_path = {}
+#		self.module_paths = []
+		self.modules = {}
+		self.modules_count = {}
+	end
+
+	# 加载一个目录下的所有模块
+	# @param [String] path 目录的路径
+	# @param [Hash] options
+	def load_modules(path, options={})
+	end
+
+	# 将实例化后的模块类放到 modules 中
+	def add_module(path, module_type, module_layer_2, reference_name, module_klass)
+
+		modules[reference_name] = {:module_type => module_type,
+															 :module_layer_2 => module_layer_2,
+															 :full_path => path, 
+															 :klass	=> module_klass
+															}
+	end
+
+	# 获得指定模块类型
+	# 返回类型: Hash
+	def get_modules_by_type(type)
+		return_hash = {}
+
+		type = type.to_s
+		modules.each_pair do |n, m|
+			if m[:module_type] != type
+				next
+			end
+			return_hash[n] = m
+		end
+
+		return return_hash
+	end
+
+	# 通过设备类型获得模块类
+	# 返回类型为： 数组
+	def get_modules_by_device_type(device_type)
+		return_array = []
+
+		device_type = device_type.to_s
+		modules.each_pair do |_, m|
+			if m[:module_type] != "cmd"
+				next
+			end
+
+			type = m[:module_layer_2]
+			if device_type == type
+				return_array << m[:klass]
+			end
+		end
+
+		return return_array
+	end
+				
+	# 获得所有加载模块的设备类型的字符串型
+	# 返回类型为： 数组
+	def get_modules_device_type_to_s
+		return_array = []
+		
+		modules_count["cmd"].keys.each do |key|
+			return_array << key.to_s
+		end
+
+		return return_array
+	end
+			
+			
+	# 判断此设备类型的模块是否加载
+	def device_type_enable?(device_type)
+		true
+	end
+
+	def module_type_enable?(module_type)
+		["cmd", "input", "output", "special", "description"].include?(module_type)
+	end
+
+	def notify(opts={})
+		if opts[:count_by_module_type]
+			@modules_count = opts[:count_by_module_type]
+		end
+	end
+
+	# 通过参考名获得模块类
+	def get_module_klass_by_name(reference_name)
+		begin
+			self.modules[reference_name][:klass]
+		rescue Exception
+			print_error "未找到指定模块: #{reference_name}"
+			exit 0
+		end
+	end
+			
+
+end