vmopt 0.0.2

data/lib/vmopt/dvd_operation.rb

@@ -0,0 +1,120 @@
+#encoding: utf-8
+require "vmopt/utils/wmi"
+require "au3"
+require "Win32API"
+module Vmopt
+class DVDOperation
+=begin
+参数：无
+功能：获取光驱的基本信息
+返回值：默认
+=end
+	def get_dvd_information()
+		data_value={}
+		colItems = WMI.execquery ("select * from Win32_CDROMDrive")
+		for colItem in colItems do	
+			str={"介质类型"=>colItem.MediaType,
+			     "盘符"=> colItem.Drive,
+			     "描述信息"=>colItem.Description
+			 }
+			 data_value["#{colItem.Drive}"]=str
+		end
+		return data_value
+	end
+
+=begin
+参数：无
+功能：弹出光驱
+返回值：默认	
+=end
+	def eject_dvd(str1="set cdauto door open",str2="",num1=0,num2=0)
+		mci = Win32API.new("winmm.dll","mciSendString",%w(p p i i),"i")
+		mci.call(str1,str2,num1,num2)
+	end
+
+=begin
+参数：无
+功能：关闭光驱与弹出光驱对应
+返回值：默认
+=end
+	def close_dvd(str1="set cdauto door closed",str2="",num1=0,num2=0)
+		mci = Win32API.new("winmm.dll","mciSendString",%w(p p i i),"i")
+		mci.call(str1,str2,num1,num2)		
+	end
+
+=begin 
+参数：无
+功能：禁用系统所有的光驱设备
+返回值：默认
+=end
+	def disable_all_dvd
+		system("devcon.exe disable *DVD* >null 2>&1")
+	end
+
+=begin
+参数：盘符号例如D:
+功能：禁用指定盘符的光驱设备
+返回值：默认
+=end
+	def disable_index_dvd(str)
+		colItems = WMI.execquery ("select * from Win32_CDROMDrive where Drive='#{str}'")	
+		for colItem in colItems do	
+			strID = colItem.DeviceId
+			system("devcon.exe disable @#{strID}  >null 2>&1")
+		end			
+	end
+
+=begin 
+参数：无
+功能：启用系统所有的光驱设备
+返回值：默认
+=end
+	def enable_all_dvd
+		system("devcon.exe enable *DVD*  >null 2>&1 ")
+	end
+
+
+=begin 
+参数：无
+功能：删除或卸载统所有的光驱设备
+返回值：默认
+=end
+	def remove_all_dvd
+		system("devcon.exe remove *DVD*  >null 2>&1")
+	end
+
+=begin
+参数：盘符号例如D:
+功能：删除或卸载指定盘符的光驱设备
+返回值：默认
+=end
+	def remove_index_dvd(str)
+		colItems = WMI.execquery ("select * from Win32_CDROMDrive where Drive='#{str}'")	
+			for colItem in colItems do	
+				strID = colItem.DeviceId
+				system("devcon.exe remove @#{strID}  >null 2>&1")
+			end			
+	end
+=begin
+参数：无
+功能：自动添加或者挂载所有光驱设备
+返回值：默认
+=end
+	def mount_all_dvd
+		system("devcon.exe rescan  >null 2>&1")
+	end
+
+=begin
+参数：盘符号例如D:
+功能：打开光盘里的内容
+返回值：默认
+=end
+	def read_dvd(str)
+		#AutoItX3.run("explorer.exe #{str}")
+		system("dir  #{str}  >null 2>&1")
+	end
+end
+
+end #module vmopt
+
+

data/lib/vmopt/ext/string_ext.rb

@@ -0,0 +1,104 @@
+# encoding: utf-8
+require 'iconv' if RUBY_VERSION.to_f < 1.9
+class String # :nodoc: all
+  #判断是否为utf8
+  def utf8?
+    if String.method_defined?(:encode)
+      return true if self.force_encoding("UTF-8").valid_encoding?
+      return false
+    else
+      unpack('U*') rescue return false  
+      return true
+    end      
+  end
+
+  #TODO：参数功能目前没有用  
+  def to_utf8(enc='GBK')
+    if self.respond_to?(:encode)
+       return self if utf8?
+       self.encode('UTF-8',enc)
+    else
+      require 'iconv' if RUBY_VERSION.to_f < 1.9
+      return self if utf8?
+      begin 
+        Iconv.iconv('UTF-8',enc,self).at(0).to_s
+      rescue
+         self
+      end
+    end
+  end
+  
+  #TODO：参数功能目前没有用
+  def to_gbk(enc='UTF-8')
+    if self.respond_to?(:encode)
+      return self unless utf8?
+      self.encode('GBK', enc)
+    else
+      return self unless utf8?
+      require 'iconv' if RUBY_VERSION.to_f < 1.9
+      begin
+        Iconv.iconv('GBK',enc,self).at(0).to_s
+      rescue
+        self
+      end
+    end
+  end
+  
+  def to_gb2312(enc = 'UTF-8')
+    # require 'iconv'
+    # Iconv.iconv('GB2312','UTF-8',self).at(0).to_s
+    if self.respond_to?(:encode)
+      self.encode('GB2312', enc)
+    else
+      begin
+        self.unpack("U*")
+        require 'iconv' if RUBY_VERSION.to_f < 1.9
+        Iconv.iconv('GB2312',enc,self).at(0).to_s
+      rescue ::ArgumentError
+        self
+      end
+    end    
+  end
+  def to_arr()
+    self.split('&') #暂时未考虑&在\之后的问题,fix this bug
+  end
+
+  def __camelize(first_letter_in_uppercase=true)
+      if first_letter_in_uppercase
+        self.to_s.gsub(/\/(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase }
+      else
+        self.to_s[0].chr.downcase + camelize(self)[1..-1]
+      end
+  end
+  alias camelize __camelize  
+  
+  def __underscore
+    word = self.to_s.dup
+    word.gsub!(/::/, '/')
+    word.gsub!(/([A-Z]+)([A-Z][a-z])/,'\1_\2')
+    word.gsub!(/([a-z\d])([A-Z])/,'\1_\2')
+    word.tr!("-", "_")
+    word.downcase!
+    word
+  end
+  alias underscore __underscore
+
+end
+
+class << nil
+  def to_utf8
+    ""
+  end
+
+end
+
+
+if $0 == __FILE__
+  #  require 'string'
+  puts "中文123中文".to_gbk.dump
+  puts "中文123中文".to_gb2312.dump
+  puts File.join("中文abc/abc","中文").to_gbk.dump
+  puts File.join("中文abc/abc","中文").to_gb2312.dump
+  puts 'ab_c'.camelize
+end
+

data/lib/vmopt/network.rb

@@ -0,0 +1,128 @@
+#encoding: utf-8
+#$:.unshift File.join(__FILE__,"..","..")
+require "vmopt/windows/win_net"
+require "vmopt/ext/string_ext"
+
+module Vmopt
+	class NetWork
+     
+    #功能：显示可用网络接口
+    #返回：返回接口名称数组
+    def self.show_netinterface()
+      WinNet.interfaces
+    end
+    
+    #功能：检查网络接口的连接状态
+    #返回：返回状态字串
+	  def self.show_interface_stat(interface)
+      adapter(interface.to_utf8);      
+	    WinNet.netconnstatus[interface.to_utf8]
+	  end
+    
+    #功能：检查接口是否存在，返回接口名对应的适配器
+    def self.adapter(interface)
+      interface = interface.to_utf8
+      raise WinNetError::NoInterfaceError "Give me no NetWorkinterface" if interface.nil? or interface.empty? 
+      adapter =WinNet.network_adapter.select do |inf|
+        inf.NetConnectionId == interface ;
+      end
+      raise WinNetError::NoInterfaceError "Find no interface #{interface}" if adapter.empty?
+      adapter[0]
+    end
+    
+    #功能：执行devcon命令 
+    def self.devcon_exec(cmd)
+        status = system("#{cmd} 1>&2");
+        status
+    end
+
+    #功能：给定一个网卡接口名，卸载网卡
+    #例子：adapter_opt("本地连接 2"，"禁用")
+    def self.adapter_opt(interface, opt)
+      pci = adapter(interface).PNPDeviceID;
+      optt=case opt.to_utf8
+           when "卸载" then "remove"
+           when "禁用" then "disable"
+           when "启用" then "enable"
+           else raise WinNetError::InvalidParamError "Unkown option for adapter.-#{opt}"
+           end
+      
+      cmd = "devcon.exe /r #{optt} \"@#{pci}\" "
+      devcon_exec(cmd)
+    end
+    
+    # #功能：安装一个网卡,暂时不做了，用户需要获取网卡的描述
+    # def self.adapter_install()
+    #    devcon /r install C:\windows\system32\drivers\rtl8139.sys 硬件ID
+    # end
+
+    #功能：扫描硬件改动
+    #
+    def self.adapter_rescan()
+      cmd = "devcon.exe rescan"
+      devcon_exec(cmd)
+    end
+    
+    #功能：返回网卡描述信息，用来判断网卡型号
+    def self.adapter_discript()
+      description=[]
+      WinNet.network_adapter_configurations.each do |interface| 
+        description << interface.description 
+      end
+      description
+    end
+   
+    #功能：设置网卡的配置模式为dhcp
+    def self.adapter_set_dhcp(interface)
+      adapter(interface);
+      interface = "\""+interface.to_gbk+"\""
+      ipset_cmd = "netsh interface ip set address name=#{interface} source=dhcp".to_gbk
+      dnsset_cmd = "netsh interface ip set dns name=#{interface} source=dhcp".to_gbk
+      ipset_status = system("#{ipset_cmd} >null 2>&1");
+      dnsset_status = system("#{dnsset_cmd} >null 2>&1");
+      
+      ipset_status & dnsset_status
+
+    end
+    
+    #功能：设置网卡为配置的参数
+    def self.adapter_set_static(interface, opt={})
+      adapter(interface);
+      interface = "\""+interface.to_gbk+"\""
+      ipset_cmd = "netsh interface ip set address name=#{interface} source=static addr=#{opt[:addr]} mask=#{opt[:mask]}".to_gbk
+      gwset_cmd = "netsh interface ip set address name=#{interface} gateway=#{opt[:gateway]} gwmetric=0".to_gbk
+      dnsset_cmd = "netsh interface ip set dns name=#{interface} source=static addr=202.106.128.86 register=primary".to_gbk
+      dns2set_cmd = "netsh interface ip add dns name=#{interface} addr=8.8.8.8 index=2".to_gbk
+      
+      ipset_status = system(ipset_cmd) 
+      gwset_status = system(gwset_cmd)
+      dns_status = system(dnsset_cmd)
+      dns2_staus = system(dns2set_cmd)
+
+      ipset_status & dns_status & dns2_staus & gwset_status
+
+    end
+
+  end
+end
+
+if __FILE__ == $0
+ 
+  Vmopt::NetWork.show_netinterface.each { |e| puts (e) }
+  Vmopt::NetWork.show_netinterface.each do |inf|
+    puts  "#{inf} :" + Vmopt::NetWork.show_interface_stat(inf)
+  end
+  Vmopt::NetWork.adapter_opt("本地连接 4", "禁用")   
+  Vmopt::NetWork.adapter_opt("本地连接 4", "启用")
+  #Vmopt::NetWork.adapter_opt("本地连接 3", "卸载")
+  # Vmopt::NetWork.adapter_rescan
+  # sleep 10
+  Vmopt::NetWork.adapter_discript.each{|d| puts (d) }
+   
+  Vmopt::NetWork.adapter_set_static("本地连接 4", {:addr => "192.168.12.10", :mask => "255.255.252.0", :gateway => "192.168.12.1"})
+  Vmopt::NetWork.adapter_set_dhcp("本地连接 4")
+  
+end
+
+
+

data/lib/vmopt/notepad.rb

@@ -0,0 +1,145 @@
+# encoding: utf-8
+#$:.unshift File.join(__FILE__,"..","..")
+require "vmopt/windows/win_winutils"
+
+module Vmopt
+
+  class NotePad
+    
+    # 功能：
+    # 1.初始化一个notepad程序，如果给定路径，则打开特定路径下的文件，同时激活该窗口
+    # 2.如果不给定路径，则打开notepad，同时激活该窗口。
+    # 参数：
+    # txt_path: 文件路径
+    # open_window: 表示需不需要创建该notepad窗口，默认创建,不设置表示获取该窗口
+    # 
+    
+    def initialize(opt)
+  
+      if opt[:txt_path].nil?
+        @title = "记事本"
+      else  
+        basename = File.basename(opt[:txt_path]).to_utf8
+        @title = "#{basename} - 记事本"
+        @path = opt[:txt_path]
+      end
+
+      if opt[:open_window] 
+        winexe = Win32API.new('kernel32', 'WinExec', 'PI', 'I')
+	      opt[:txt_path].nil? ? winexe.call("notepad.exe", 1) : winexe.call("notepad.exe #{opt[:txt_path]}", 1)
+        sleep 1
+      end
+
+      @window = WinUtils::find_window(/#{@title}/) 
+      RAutomation::Window.wait_timeout=40
+      
+    end
+    
+    # 功能：
+    # 1.对一个激活的窗口设置文本内容
+    # 参数：
+    # msg：表示即将填充的信息
+    def set_text(msg)
+    	@window.text_field(:class => "Edit", :id => 15).set(msg)
+    end
+    
+    # 功能：读取一个已经打开的notepad的里面的内容
+
+    def read_text
+      @window.text_field(:class => "Edit", :id => 15).value
+    end
+     
+    #功能：点击菜单中的保存按钮来保存文件
+    #参数：
+    # filepath：
+    #   如果指定了filepath 就表示要另外保存在filepath所在文件
+    #   如果没有指定filepath 而且 初始化时指定了path，那就只按保存键
+    #   如果初始化时没有指定path，要保存时又不提供路径，报异常
+
+    def save(filepath=nil)
+      if !filepath.nil?
+        @window.WinMenuSelectItem("[CLASS:Notepad]", "", "文件", "另存为")
+
+	savewindow = WinUtils::find_window(/另存为/)
+	savewindow.text_field(:class => "Edit").set(filepath)       
+       #窗口关闭，遍历button会导致卡死，
+       # savewindow.buttons.each do|button| 
+       #   if button.value == "保存(&S)";
+       #    button.click; 
+       #   end
+       # end
+       
+      savewindow.send_keys("{ENTER}")    
+      sleep 1
+      elsif !@path.nil? and filepath.nil?
+        @window.WinMenuSelectItem("[CLASS:Notepad]", "", "文件", "保存")      
+      elsif @path.nil? and filepath.nil?
+        raise WinUtils::NoSavePathError,"Not found the save path ."
+      end
+    end #end of save
+    
+    #功能：关闭记事本
+    def close
+      @window.WinClose("[CLASS:Notepad]", "")
+    end #end of close
+
+  end #end of class NotePad
+
+end #end of module vmopt
+
+if __FILE__ == $0
+
+  #File.delete("C:\\1.txt") if FileTest::exist?("C:\\1.txt")
+  notepad = Vmopt::NotePad.new({:open_window => true, :txt_path => "C:\\1.txt"})
+  #notepad.set_text("hello world")
+  puts notepad.read_text
+  #notepad.save("C:\\1.txt")
+  #notepad.close
+  
+end
+
+
+#------使用 au3 的实现----先放着后续可能复用---
+#
+# module Vmopt
+#   class NotePad
+#     def initialize()
+#       AutoItX3.run("notepad.exe")
+#       AutoItX3::Window.wait("无标题 - 记事本")
+#       @window = AutoItX3::Window.new("无标题 - 记事本")
+#     end
+    
+#     def close
+#         @window.close
+#       end
+
+#     def set_text(msg)
+#       @window.activate
+#       edit = @window.focused_control
+#       edit.text = msg
+#     end
+    
+#     def read_text
+#       @window.activate 
+#       edit = @window.focused_control
+#       edit.text
+#     end  
+
+#       def save_to(filepath)
+#         @window.activate
+#         @window.select_menu_item("文件", "另存为")
+#         window_save = AutoItX3::Window.new("另存为")
+#         window_save.activate
+#         window_save.focused_control.text = filepath
+#         AutoItX3::send_keys("{enter}")
+
+#       end
+
+#   end
+# end
+
+# notepad = Vmopt::NotePad.new()
+# notepad.set_text("adfhj")
+# puts notepad.read_text
+# notepad.save_to("C:\\1.txt")
+# notepad.close