hostor 1.0.0-mswin32

data/bin/hostor

@@ -0,0 +1,241 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'optparse'
+require 'etc'
+require 'ftools'
+require 'tempfile'
+require 'rbconfig'
+
+class File
+  def self.prepend(path, string)
+    Tempfile.open File.basename(path) do |tempfile|
+      # prepend data to tempfile
+      tempfile << string
+
+      File.open(path, 'r+') do |file|
+        # append original data to tempfile
+        tempfile << file.read
+        # reset file positions
+        file.pos = tempfile.pos = 0
+        # copy all data back to original file
+        file << tempfile.read
+      end
+    end
+  end
+end
+
+
+module Hostor
+  def os
+    host_os = RbConfig::CONFIG['host_os']
+    case host_os
+    when /mswin|msys|mingw32/
+      :windows
+    when /darwin|mac os/
+      :macosx
+    when /linux/
+      :linux
+    when /solaris|bsd/
+      :unix
+    else
+      raise Error::WebDriverError, "unknown os: #{host_os.inspect}"
+    end
+  end
+
+  def is_windows 
+    os.eql? :windows
+  end
+
+  def is_mac 
+    os.eql? :macosx
+  end
+
+  def is_linux  
+    os.eql? :linux
+  end
+
+  #todo: 根据不同平台获取当前目录
+  def hostor_dir
+    user = Etc.getlogin
+    is_windows ?  "#{ENV['USERPROFILE']}\\hostor" : "#{ENV['HOME']}/.hostor"
+  end
+
+  def default_host
+    "#{hostor_dir}/default.host"
+  end
+
+  def sys_host_file
+    is_windows ? "#{ENV['SystemDrive']}/windows/system32/drivers/etc/hosts" : '/etc/hosts'
+  end
+
+  def get_host_name(path)
+    unknown = "未定名称的HOST文件"
+    ln = nil
+    File.open(path, 'r').each_line do |line| 
+      ln = line
+      break
+    end
+    return unknown if ln.nil?
+
+    if (ln =~ /#==\[.*?\]==/).eql? 0
+      m = ln.match /#==\[(.*)?\.host\]==\B/
+      m[1] || unknown
+    else
+      unknown
+    end
+  end
+
+  def init
+    if File.exists?(hostor_dir)
+      puts "已经初始化过了!"
+    else
+      Dir.mkdir(hostor_dir)
+      puts "创建hostor配置目录#{hostor_dir}"
+
+      File.copy(sys_host_file, default_host)
+      puts "复制#{sys_host_file}到#{default_host}"
+
+      File.prepend(default_host, "#==[default.host]==\n")
+
+      puts "初始化完成"
+    end
+  end
+
+  def get_path_by_name(name)
+    p1 = "#{hostor_dir}/#{name}"
+    return p1 if File.exists? p1
+
+    p2 = "#{hostor_dir}/#{name}.host"
+    return p2 if File.exists? p2
+
+    raise "名为#{name}.host的文件不存在"
+  end
+
+  def switch(file)
+    File.copy(get_path_by_name(file), sys_host_file)
+  end
+
+  def current
+    get_host_name(sys_host_file)
+  end
+
+  def view(file)
+    dest = file ? get_path_by_name(file) : sys_host_file
+    File.open(dest, 'r').each_line{|ln| puts ln }
+  end
+
+  def remove(file)
+    File.delete(get_path_by_name(file))
+  end
+
+  def delete(keyword)
+  end
+
+  def list
+    files = []
+    Dir.foreach(hostor_dir) do |file|
+      if not File.directory? file and File.extname(file).eql? '.host'
+        files << file
+      end
+    end
+    files
+  end
+
+  def add(file)
+    if File.exists? file
+      file_bak = file.to_s + "" #for_win
+      basename = File.basename(file)
+      ext = File.extname(file)
+      if ext.empty?
+        basename += '.host'
+      elsif not ext.eql? '.host'
+        basename.gsub!(/\.[^.]*?$/, '.host')
+      end
+      dest = hostor_dir + "/" + basename
+      if File.exists? dest
+        raise "已经添加了名为#{basename}文件，请重命名后添加或者使用#$0 --remove #{basename} 删除已有文件"
+      else
+        File.copy(file_bak, dest)
+        File.prepend(dest, "#==[#{basename}]==\n")
+      end
+    else
+      raise "文件#{file}不存在"
+    end
+  end
+
+  def main(args)
+    opts = OptionParser.new do |opts|
+      opts.banner = <<-EOF
+      Usage: 
+      #$0  查看系统当前host文件
+      #$0 <hostname> 设定指定的文件为系统host文件
+      EOF
+
+      opts.on('--init', "初始化hostor") do
+        init()
+      end
+
+      opts.on('-a', '--add <file>', "将指定文件加入hostor管理") do |file|
+        add(file)
+      end
+
+      opts.on('-e', '--edit [file]', "编辑指定的HOST文件") do |file|
+        f = file || current
+        dest = get_path_by_name(f)
+        if is_mac
+          `open -a macvim #{dest}`
+        elsif is_linux
+          `xterm -e vim #{dest}`
+        elsif is_windows
+          `notepad #{dest}`
+        end
+        if f.gsub(/\..*/, '') == current
+          switch( f )
+        end
+      end
+
+      opts.on('-r', '--remove <file>', "从hostor文件夹移除指定的HOST文件") do |file|
+        remove(file)
+      end
+
+      #适用于 -a -d 
+      #opts.on('-t', '--target <file>', "指定操作的HOST文件") do |name|
+      #end
+
+      opts.on('-l', '--list', "列举当前用户的所有HOST文件") do
+        list().each{|l| puts l}
+      end
+
+      opts.on('-v', '--view [file]', "查看指定的HOST文件") do |name|
+        view(name)
+      end
+
+      opts.on('-c', '--curr', "列举当前使用HOST文件") do 
+        puts current()
+      end
+
+      opts.on('-s', '--switch <file>', "将指定HOST文件设为系统当前HOST文件") do |file|
+        switch(file)
+      end
+    end
+
+    begin
+      if args.length.eql? 0
+        view(nil)
+      elsif args.length.eql? 1 and args[0] =~ /^[^-]/
+        switch(args[0])
+      else
+        opts.parse!(args)
+      end
+    rescue OptionParser::InvalidOption => e
+      puts "参数非法, 查看帮助请使用: #$0 --help "
+    rescue Exception => e
+      puts e.to_s
+    end
+  end
+end
+
+include Hostor
+main ARGV
+
+

data/bin/hostor_gui

@@ -0,0 +1,111 @@
+#!/usr/bin/env jruby
+require 'java'
+module Hostor
+  module SysPopupMenu
+    def self.run
+      if (Java::JavaAwt::SystemTray.isSupported()) 
+        tray = Java::JavaAwt::SystemTray.getSystemTray();
+        image = Java::JavaAwt::Toolkit.getDefaultToolkit().getImage(File.expand_path(File.dirname(__FILE__))+"/hostor.png")
+
+        popup = Java::JavaAwt::PopupMenu.new
+        self.refresh(popup)
+
+        icon = Java::JavaAwt::TrayIcon.new(image, "hostor", popup)
+        icon.setImageAutoSize true
+
+        begin 
+          tray.add(icon)
+        rescue Java::JavaAwt::AWTException => e
+          java::lang::System.err.println("TrayIcon could not be added.")
+        end
+      else 
+        java::lang::System.err.println("System tray is currently not supported.")
+      end
+    end
+
+    def self.refresh(popup)
+      popup.removeAll()
+
+      curr = get_curr_host()
+      get_host_list.each do |name|
+        popup.add(make_host_item(name, curr))
+      end
+
+      popup.add(Java::JavaAwt::MenuItem.new("--------"))
+      popup.add(make_refresh_item())
+      popup.add(make_exit_item())
+    end
+
+    class << self
+      private
+      def get_host_list()
+        return (`hostor -l`).split("\n").collect {|x| x.sub /.host/, ''}
+      end
+
+      def get_curr_host()
+        return (`hostor -c`).delete("\n")
+      end
+
+      def get_prefix(name, curr)
+        name.eql?(curr) ? "*" : " "
+      end
+
+      def get_host_item_label(name, curr)
+        get_prefix(name, curr) + name
+      end
+
+      def make_host_item(name, curr)
+        item = Java::JavaAwt::MenuItem.new(get_host_item_label(name, curr))
+        item.setActionCommand(name)
+        listener = java::awt::event::ActionListener.new
+        def listener.actionPerformed(e)
+          item = e.getSource()
+          curr_host = item.getActionCommand()
+
+          old_host = (`hostor -c`).delete("\n")
+          system("hostor #{curr_host}")
+
+          if curr_host != old_host
+            item.setLabel("*"+curr_host)
+            popup = item.getParent()
+
+            n = popup.getItemCount()
+            n.times do |i|
+              o = popup.getItem(i)
+              name = o.getActionCommand()
+              if(name == old_host)
+                o.setLabel(' '+ name)
+                break
+              end
+            end
+          end
+        end
+
+        item.addActionListener(listener)
+        return item
+      end
+
+      def make_exit_item()
+        item = Java::JavaAwt::MenuItem.new("exit")
+        listener = java::awt::event::ActionListener.new
+        def listener.actionPerformed(e) 
+          java::lang::System.exit(0)
+        end
+        item.addActionListener(listener)
+        return item
+      end
+
+      def make_refresh_item()
+        item = Java::JavaAwt::MenuItem.new("reload")
+        listener = java::awt::event::ActionListener.new
+        def listener.actionPerformed(e) 
+          Hostor::SysPopupMenu.refresh(e.getSource().getParent())
+        end
+        item.addActionListener(listener)
+        return item
+      end
+    end
+  end
+end
+
+Hostor::SysPopupMenu.run

data/bin/hostor_gui.bat

@@ -0,0 +1,2 @@
+@ECHO OFF
+@start jrubyw hostor_gui

metadata

@@ -0,0 +1,67 @@
+--- !ruby/object:Gem::Specification 
+name: hostor
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+    - 1
+    - 0
+    - 0
+  version: 1.0.0
+platform: mswin32
+authors: 
+  - Zhong Xingdou
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-12-24 00:00:00 +08:00
+default_executable: 
+dependencies: []
+
+description: "    system host file manager\n"
+email: zhongxingdou@gmail.com
+executables: 
+  - hostor
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+  - bin/hostor_gui.bat
+  - bin/hostor
+  - bin/hostor_gui
+  - bin/hostor.png
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+  - lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+          - 0
+        version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+          - 0
+        version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: system host file manager
+test_files: []
+