vmopt 0.0.2

data/bin/vmnotepad

@@ -0,0 +1,144 @@
+#!/usr/bin/env ruby
+#encoding: utf-8
+require "vmopt/notepad"
+require "optparse"
+require 'json'
+
+option = {}
+option_parser = OptionParser.new do | opts |
+  opts.banner = %q/
+  Help: 
+    Description:
+      vmnotepad tool, use for notepade management.
+    function:
+      1.use notepad to create a txt file 
+      2.query an existed txt file msg
+      3.query an opened notepad.exe msg
+
+    Example:
+      vmnotepad --create --path    filepath  --msg msg 
+      vmnotepad --query  --existed  --path filepath 
+      vmnotepad --query  --opened   --path filepath 
+      vmnotepad --close  --titile  txt_windows_titile
+      vmnotepad --unpack   --path    filepath
+      vmnotepad --delete  --path   filepath
+  /
+  #删除文档
+  opts.on('-d', '--delete', 'delete a txt file') do
+    option[:cmd] = "delete"
+  end
+  
+  #创建txt文档
+  opts.on('-c', '--create', 'create a txt file') do
+    option[:cmd] = "create"
+  end
+  #查询txt
+  opts.on('-q', '--query', 'query txt file content or opened notepad window') do
+    option[:cmd] = "query"
+  end
+  #txt文档路径
+  opts.on('-p filepath', '--path filepath', 'specify a txt file with full path') do|value|
+    option[:filepath] = value
+  end
+  #文档信息
+  opts.on('-m msg', '--msg msg', 'fill a txt file with content') do|value|
+    option[:msg] = value
+  end
+  #已经存在但是没有打开的文档
+    option[:existed] = false
+  opts.on('-e', '--existed', 'specify a existed txt file') do
+    option[:existed] = true
+  end
+  #已经打开着的txt文档
+    option[:opened] = false
+  opts.on('-o', '--opened', 'specify a opened txt file') do
+    option[:opened] = true
+  end
+  #关闭打开着的txt文档
+  opts.on('-k', '--close', 'close a opened txt file') do
+    option[:cmd] = "close"
+  end
+  opts.on('-t txt_titile', '--titile txt_titile', 'specify a opened txt window titile') do|value|
+    option[:titile] = value
+  end
+  #关闭打开着的txt文档
+  opts.on('-u', '--unpack', 'open a txt file and hold on') do
+    option[:cmd] = "unpack"
+  end
+
+end.parse!
+
+$result_json = {}
+
+def param_error
+  $result_json = {:status => "-1",:data=>"Please check the params..."}
+end
+
+begin
+   case option[:cmd] 
+   when "delete"
+     if !option[:filepath].nil?
+       if FileTest::exist?(option[:filepath])
+         File.delete(option[:filepath]) 
+	 $result_json = {:status => "0",:data=>"delete file #{option[:filepath]} ok"}
+       else
+         param_error
+       end
+     else
+       param_error
+     end
+     
+   when "create"
+     if !option[:filepath].nil? and !option[:msg].nil?
+       File.delete(option[:filepath]) if FileTest::exist?(option[:filepath])
+       notepad = Vmopt::NotePad.new({:open_window => true})
+       notepad.set_text(option[:msg])
+       notepad.save(option[:filepath])
+       notepad.close
+       $result_json = {:status => "0",:data=>"create file #{option[:filepath].to_utf8} success"}
+     else
+       param_error
+     end
+
+   when "query"
+     if !option[:filepath].nil? and option[:existed]
+       notepad = Vmopt::NotePad.new({:open_window => true, :txt_path => option[:filepath]})
+       msg = notepad.read_text
+       notepad.close
+       $result_json[:status]="0";
+       $result_json[:data]=msg
+     elsif !option[:filepath].nil? and option[:opened]
+       notepad = Vmopt::NotePad.new({:open_window => false, :txt_path => option[:filepath]})
+       msg = notepad.read_text
+       $result_json[:status]="0";
+       $result_json[:data]=msg
+     else 
+       param_error
+     end
+   
+   when "close"
+     if !option[:titile].nil?
+       notepad = Vmopt::NotePad.new({:open_window => false, :txt_path => option[:titile]})
+       notepad.close
+       $result_json = {:status => "0", :data=>"close txt #{option[:titile].to_utf8} success"}
+     else
+       param_error
+     end
+   
+   when "unpack"
+     if !option[:filepath].nil?
+       Vmopt::NotePad.new({:open_window => true, :txt_path => option[:filepath]})
+       $result_json = {:status => "0", :data=>"open txt #{option[:filepath].to_utf8} success"}
+     else
+       param_error 
+     end
+
+   else
+   	  system("vmnotepade -help");
+   end
+rescue Exception => e
+  $result_json = {:status => "-1", :data=>"Catch exception #{e}"}
+end
+
+puts JSON.generate($result_json) unless option[:cmd].nil?
+

data/bin/vmopt

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+#encoding: utf-8
+banner = %q/
+  Help: 
+    Description:
+      vmopt is a tool box, use for windows system management.
+    
+    Tools:
+      1.vmdrive         
+      2.vmdvd
+      3.vmnetwork
+      4.vmnotepad
+      5.vmpower
+      6.vmserial
+      7.vmsysres
+
+    Example:
+      vmdrive -help   
+  /
+
+  puts banner
+  

data/bin/vmpower

@@ -0,0 +1,73 @@
+#!/usr/bin/env ruby
+#encoding: utf-8
+require "vmopt/power_operation"
+require "optparse"
+require "json"
+
+option = {}
+option_parser = OptionParser.new do | opts |
+  opts.banner = %q/
+  Help: 
+    Description:
+      vmpower tool, use for system power management.
+    function:
+      1.shutdown        
+      2.reboot
+      3.logoff
+      4.lock_user
+      5.sleep
+
+    Example:
+      vmpower --do  shutdown    
+      vmpower --do  reboot  
+      vmpower --do  logoff
+      vmpower --do  lock_user
+      vmpower --do  sleep
+  /
+  opts.on('-d popt', '--do  popt', 'System  power operation') do |value|
+    option[:popt] = value
+    option[:cmd] = "do"
+  end
+
+end.parse!
+  
+  resultok = {"status"=>"0"}
+  resultfail = {"status"=>"-1"}
+  data_value={}
+  $pow = Vmopt::PowerOperation.new()
+
+   if option[:cmd] == "do"
+      if option[:popt] == "shutdown"
+        ret = $pow.shutdown
+        data_value ="system is going to shutdown"
+      elsif option[:popt] == "reboot"
+        ret = $pow.reboot
+        data_value = "system is going to reboot"
+      elsif option[:popt] == "logoff"
+        ret = $pow.logoff
+        data_value = "system is going to logoff"
+      elsif option[:popt] == "lock_user"
+        ret = $pow.lock_user
+        data_value = "system is going to lock_user"
+      elsif option[:popt] == "sleep"
+      	ret = $pow.sleep
+        data_value  = "system is going to sleep"
+      else 
+        system("vmpower -help");
+        option.clear
+      end
+   else
+   	  system("vmpower -help");
+      option.clear
+   end
+
+   if ret==true
+     resultok["data"]=data_value
+     retjson = JSON.generate resultok
+   else
+     resultfail["data"]="power operation fail"
+     retjson = JSON.generate resultfail  
+   end
+   puts retjson unless option[:cmd].nil?
+
+

data/bin/vmserial

@@ -0,0 +1,92 @@
+#!/usr/bin/env ruby
+#encoding: utf-8
+require "vmopt/serialport_operation"
+require "optparse"
+require "json"
+
+option = {}
+option_parser = OptionParser.new do | opts |
+  opts.banner = %q/
+  Help: 
+    Description:
+      vmserial tool, use for vm serialport operation,including query info, write data, and read data.
+    function:
+      1.get_serial_port        
+      2.write  串口号，写入的字符串
+      3.read   串口号
+
+    Example:
+      vmserial --query  serialport    
+      vmserial --write  --serialnumber com1  --msg  helloworld
+      vmserial --read   --serialnumber com2  
+  /
+
+  opts.on('-q serialport', '--query  serialport', 'Query serialport information') do |value|
+    option[:serialport] = value
+    option[:cmd] = "query"
+  end
+  
+  opts.on('-w', '--write', 'write serialport information') do |value|
+    option[:cmd] = "write"
+  end
+
+  opts.on('-r', '--read', 'Read serialport information') do |value|
+    option[:cmd] = "read"
+  end
+  
+  opts.on('-s com', '--serialnumber com ', 'serialnumber') do |value|
+    option[:serialnumber] = value
+  end
+
+  opts.on('-m msgstr', '--msg msgstr', 'serialport data to trans') do |value|
+    option[:msg] = value
+  end
+
+end.parse!
+   
+   resultok = {"status"=>"0"}
+   resultfail = {"status"=>"-1"}
+   data_value={}
+   $sys = Vmopt::SerialPortOperation.new
+   case option[:cmd] 
+   when "query"
+     if option[:serialport]  == 'serialport'
+        data_value = $sys.get_serial_port
+     else 
+        system("vmserial -help") 
+        option.clear
+     end
+
+   when "write"
+     if !option[:serialnumber].nil? and !option[:msg].nil?
+        ret = $sys.write(option[:serialnumber],option[:msg])
+        data_value="write serialport success"
+     else 
+        system("vmserial -help") 
+        option.clear
+     end
+
+   when "read"
+     unless option[:serialnumber].nil?
+       ret = $sys.read(option[:serialnumber])
+       data_value=ret
+     else 
+        system("vmserial -help") 
+        option.clear
+     end
+   else
+    system("vmserial -help") 
+    option.clear
+   end
+
+
+   if ret==false
+     resultfail["data"]="serialport operation fail"
+     retjson = JSON.generate resultfail
+   else
+     resultok["data"]=data_value
+     retjson = JSON.generate resultok 
+   end
+
+   puts retjson unless option[:cmd].nil?
+

data/bin/vmsysres

@@ -0,0 +1,58 @@
+#!/usr/bin/env ruby
+#encoding: utf-8
+require "vmopt/system_resource"
+require "optparse"
+require "json"
+
+option = {}
+option_parser = OptionParser.new do | opts |
+  opts.banner = %q/
+  Help: 
+    Description:
+      vmsysres tool, use for query system resource information, including cpu and memery.
+    function:
+      1.get_cpu        
+      2.get_memory
+
+    Example:
+      vmsysres --query  cpu    
+      vmsysres --query  memory  
+
+  /
+  opts.on('-q queryopt', '--query  queryopt', 'Query system resource information') do |value|
+    option[:queryopt] = value
+    option[:cmd] = "query"
+  end
+
+end.parse!
+
+   resultok = {"status"=>"0"}
+   resultfail = {"status"=>"-1"}
+   data_value={}
+   ret = true
+   $sys = Vmopt::SystemResource.new
+   if option[:cmd] == "query"
+      if option[:queryopt] == 'cpu'
+        data_value = $sys.get_cpu
+      elsif option[:queryopt] == 'memory'
+        data_value = $sys.get_memory
+      else 
+        system("vmsysres -help")
+        option.clear
+      end
+   else
+    system("vmsysres -help")
+    option.clear
+   end
+
+   if ret==false
+     resultfail["data"]="system resource operation fail"
+     retjson = JSON.generate resultfail
+   else
+     resultok["data"]=data_value
+     retjson = JSON.generate resultok
+   end
+   
+   puts retjson unless option[:cmd].nil?
+
+

data/lib/vmopt.rb

@@ -0,0 +1,14 @@
+require "win32ole"
+WIN32OLE.codepage = WIN32OLE::CP_UTF8
+require "Win32API"
+
+require "vmopt/notepad"
+require "vmopt/network"
+require "vmopt/power_operation"
+require "vmopt/serialport_operation"
+require "vmopt/system_resource"
+require "vmopt/disk_operation"
+require "vmopt/dvd_operation"
+
+require "vmopt/version"
+

data/lib/vmopt/disk_operation.rb

@@ -0,0 +1,188 @@
+#encoding: utf-8
+require "vmopt/utils/wmi"
+require "json"
+module Vmopt
+class DiskOperation
+
+=begin
+参数：无
+作用：查找物理磁盘的基本信息，并打印出来
+返回值：默认
+=end
+	def get_disk_information
+		data_value={}
+		colItems = WMI.execquery ("select * from Win32_DiskDrive")
+        for colItem in colItems do
+        	size=colItem.size
+			size=size.to_i/1048576
+        	str={ "磁盘ID" => colItem.DeviceID,
+        		  "磁盘索引" => colItem.Index,
+        		  "接口类型" => colItem.InterfaceType,
+        		  "磁盘容量" => size
+        		}
+        	data_value["diks#{colItem.Index}"] = str
+        end
+        return data_value
+	end
+=begin
+参数：无
+作用：查找磁盘分区的基本信息，并打印出来
+返回值：默认
+=end
+	def get_partition_information
+		data_value={}
+		colItems = WMI.execquery ("select * from Win32_LogicalDisk where DriveType=3")
+        for colItem in colItems do
+			size=colItem.size
+			size=size.to_i/1048576
+			freeSpace=colItem.FreeSpace
+			freeSpace=freeSpace.to_i/1048576
+
+        	str={"逻辑驱动器号" => colItem.DeviceID,
+        	     "文件系统" => colItem.FileSystem ,
+        	     "分区容量" => size,  
+        	     "剩余容量" => freeSpace
+        	    }
+    
+        	data_value["partition#{colItem.DeviceID}"] = str
+        end
+   		return data_value
+	end
+		
+=begin
+参数：无
+功能：判断哪些磁盘没有格式化
+返回值：没有格式化的磁盘索引号
+=end
+	def unformat_disk
+		data_value = []
+		colItems = WMI.execquery ("select * from Win32_DiskDrive")
+        for colItem in colItems do
+        	if colItem.Partitions==0
+        		#str="#{colItem.Index}"
+        		#data_value["index#{colItem.Index}"]= str
+        		data_value << colItem.Index
+        	end      	
+        end
+
+        return {"diskIndex" => data_value}
+	end
+=begin rdoc
+参数：无
+功能：格式化磁盘,供其他方法调用
+返回值：默认
+=end
+	def format_disk(volumename)
+		ret1= system("diskpart /s c:/1.txt > null 2>&1 ")
+		ret2 = system("format /FS:NTFS /force /Q  #{volumename}  >null 2>&1" )
+		return ret1 && ret2
+	end
+=begin	
+参数：无
+功能：扫描整个系统磁盘，若存在没有格式化的磁盘则进行格式化
+返回值：默认	
+=end
+	def chk_format_disk
+		colItemindexs = WMI.execquery ("select * from Win32_DiskDrive")
+		index = []
+        for colItemindex in colItemindexs do
+        	if colItemindex.Partitions==0
+        		index.push(colItemindex.Index)
+        	end      	
+        end
+        if index.empty?
+        	return false
+        end
+		for i in index do
+			colItems = WMI.execquery ("select * from Win32_DiskDrive where index=#{i}")
+			for colItem in colItems do
+				size=colItem.size
+				size=(size.to_i/1048576).to_s
+			end
+			
+			logicalcolItems = WMI.execquery ("select * from Win32_LogicalDisk")
+			str=[]
+	        for colItem in logicalcolItems do
+	        	str.push(colItem.DeviceID)
+	        	volumename=str.sort!.last.next
+	        end
+
+			File.open("c:/1.txt","w")do|file|
+			file.puts "select disk=#{i}"
+			file.puts "create partition primary size=#{size}"
+			file.puts "assign letter=#{volumename}"
+		    end
+			unless format_disk(volumename)
+			 	return false
+			end 
+		File.delete("c:/1.txt")
+		end
+		return true
+	end
+=begin	
+参数：磁盘索引号
+功能：根据指定的磁盘索引号
+返回值：指定的磁盘已经被格式化返回false,成功格式化返回true
+=end
+	def format_disk_by_index(index)
+
+		indexEffectives = WMI.execquery("select * from Win32_DiskDrive ")
+			 indexarr=[]
+			 for indexeff in indexEffectives do
+			 	indexarr.push(indexeff.index)
+			 end
+		if index<0 or index > indexarr.sort!.last 
+		   return false 
+		end
+
+		
+		colItems = WMI.execquery ("select * from Win32_DiskDrive where index=#{index}")
+		for colItem in colItems do
+        	if colItem.Partitions==0
+        		size=colItem.size
+				size=(size.to_i/1048576).to_s
+        	else
+        		return false
+        	end
+        end
+        logicalcolItems = WMI.execquery ("select * from Win32_LogicalDisk")
+		    str=[]
+        for colItem in logicalcolItems do
+        	str.push(colItem.DeviceID)
+        	volumename=str.sort!.last.next
+        end
+
+		File.open("c:/1.txt","w") do|file|
+		file.puts "select disk=#{index}"
+		file.puts "create partition primary size=#{size}"
+		file.puts "assign letter=#{volumename}"
+		end
+		return false unless format_disk(volumename)
+		#File.open("c:/1.txt","w+")
+		File.delete("c:/1.txt")
+		return true
+	end 
+	def delete_partition(index)
+		indexEffectives = WMI.execquery("select * from Win32_DiskDrive ")
+			 indexarr=[]
+			 for indexeff in indexEffectives do
+			 	indexarr.push(indexeff.index)
+			 end
+		if index<0 or index > indexarr.sort!.last 
+		   return false 
+		end
+
+		File.open("c:/2.txt","w") do|file|
+		file.puts "select disk=#{index}"
+		file.puts "select partition 1"
+		file.puts "delete partition override"
+		end
+		ret = system("diskpart /s c:/2.txt > null 2>&1 ")
+		File.delete("c:/2.txt")
+		if ret == false
+			return false
+		end
+		return true
+	end
+end
+end #module vmopt