hosum 0.0.1a

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,9 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in hosum.gemspec
+gemspec
+
+group :development do
+  gem 'rb-fsevent'
+  gem 'growl_notify'
+end

data/Guardfile

@@ -0,0 +1,11 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+guard 'rspec', :version => 2 do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})     { "spec" } #{ |m| "spec/lib/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec" }
+  
+  watch(%r{^spec/(.+).(dsl|hosts|dnsmasq)$})    { "spec" }
+end
+

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/VERSION

@@ -0,0 +1 @@
+0.0.1a

data/bin/hosum

@@ -0,0 +1,34 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift File.expand_path("../../lib", __FILE__)
+
+require "hosum"
+require 'optparse'
+
+
+options = {
+  :output_type => :hosts,
+  :context => []
+}
+OptionParser.new do |opts|
+  opts.banner = "Usage: hosum [options] hosum_config_path"
+
+  opts.on("--dnsmasq", "Output dnsmasq config") do |v|
+    options[:output_type] = :dnsmasq
+  end
+  
+  opts.on("--context CONTEXT", "comma separated, ex: --context adsl,bstar") do |values|
+    options[:context] = values.split(",")
+  end
+end.parse!(ARGV)
+
+p options
+p ARGV
+
+
+rs = Hosum.dsl_by_file(ARGV.first)
+if options[:output_type] == :dnsmasq
+  puts rs.to_dnsmasq
+else
+  puts rs.to_hosts
+end

data/examples/my.hosum

@@ -0,0 +1,70 @@
+group "google" do
+  ip "203.208.46.163"
+  
+  subject "core" do
+    hosts "accounts.google.com"
+    hosts "www.google.com"
+    hosts "ssl.gstatic.com"
+    hosts "lh[1-5].googleusercontent.com"
+    hosts "ssl.google-analytics.com"
+    hosts "gg.google.com"
+  end
+  
+  subject "plus" do
+    hosts "plus.google.com"
+  end
+  
+  subject "search" do
+    hosts "encrypted.google.com"
+    hosts "encrypted-tbn[0-3].google.com"
+    hosts "clients1.google.com"
+    hosts "news.google.com"
+    hosts "webcache.googleusercontent.com"
+    hosts "www.google.com.hk"
+  end
+  
+  subject "map" do
+    hosts "maps.google.com"
+    hosts "maps.gstatic.com"
+    hosts "mts[0-1].google.com"
+  end
+  
+  subject "reader" do
+    hosts "s2.googleusercontent.com"
+    hosts "reader.googleusercontent.com"
+    hosts "mw2.google.com"
+    hosts "t0.gstatic.com"
+  end
+  
+  subject "mail" do
+    hosts "www.gmail.com"
+    hosts "mail.google.com"
+    hosts "chatenabled.mail.google.com"
+    hosts "mail-attachment.googleusercontent.com"
+  end
+  
+  subject "docs" do
+    hosts "docs.google.com"
+    hosts "[1-99].docs.google.com", :dnsmasq => ".docs.google.com"
+    hosts "themes.googleusercontent.com"
+  end
+  
+  subject "calendar" do
+    hosts "calendar.google.com"
+  end
+
+  subject "video" do
+    hosts "video.google.com"
+  end
+  
+end
+
+group "itunes" do
+  hosts "a[1-2000].phobos.apple.com", :dnsmasq => ".phobos.apple.com"
+  
+  ip "203.69.138.18"
+end
+
+
+
+

data/hosum.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name        = "hosum"
+  s.version     = File.read(File.expand_path("../VERSION", __FILE__)).chomp
+  s.authors     = ["Sunteya"]
+  s.email       = ["Sunteya@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{host file generator}
+  s.description = %q{host file generator}
+
+  s.rubyforge_project = "hosum"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_runtime_dependency "activesupport"
+  s.add_runtime_dependency "i18n"
+  
+  s.add_development_dependency "rspec"
+  s.add_development_dependency "guard-rspec"
+  s.add_development_dependency "rake"
+end

data/lib/hosum.rb

@@ -0,0 +1,24 @@
+require "active_support/all"
+require "hosum/dsl"
+
+module Hosum
+  VERSION = File.read(File.expand_path("../../VERSION", __FILE__)).chomp
+  
+  autoload :Container,  "hosum/container"
+  autoload :Subject, "hosum/subject"
+  autoload :Hosts,   "hosum/hosts"
+  autoload :Dsl,     "hosum/dsl"
+  autoload :Group,   "hosum/group"
+  autoload :Ips,     "hosum/ips"
+  
+  def self.dsl(options = {}, &block)
+    Dsl.new(options = {}, &block)
+  end
+  
+  def self.dsl_by_file(path)
+    self.dsl do
+      instance_eval(File.read(path), path, 1)
+    end
+  end
+  
+end

data/lib/hosum/.rb

@@ -0,0 +1,5 @@
+require "hosum//version"
+
+module Hosum/
+  # Your code goes here...
+end

data/lib/hosum/container.rb

@@ -0,0 +1,24 @@
+module Hosum
+  class Container
+    attr_accessor :parent
+    attr_accessor :children
+    attr_accessor :options
+    
+    attr_accessor :ips_list
+    
+    def initialize
+      self.children = []
+      self.ips_list = []
+      self.options = {}
+    end
+    
+    def find_first_ip
+      if self.ips_list.empty?
+        self.parent.find_first_ip if self.parent
+      else
+        self.ips_list.first.addresses.first
+      end
+    end
+    
+  end
+end

data/lib/hosum/dsl.rb

@@ -0,0 +1,93 @@
+require "ostruct"
+
+module Hosum
+  class Dsl
+    attr_accessor :containers, :ips_list_scopes
+    attr_accessor :groups
+    
+    def initialize(options = {}, &block)
+      self.groups = []
+      self.containers = []
+      self.ips_list_scopes = []
+      
+      instance_eval(&block) if block
+    end
+    
+    def containers_eval(object, &block)
+      self.containers.push(object)
+      instance_eval(&block)
+      self.containers.pop
+    end
+    
+    def group(name, &block)
+      group = Group.new
+      group.name = name
+      self.groups << group
+      
+      containers_eval(group, &block) if block
+    end
+    
+    def ip(*args, &block)
+      options = args.extract_options!
+      
+      ips = Ips.new
+      ips.options = options
+      ips.addresses = [ args ].flatten
+      
+      if block
+        self.ips_list_scopes << [ ips ]
+        instance_eval(&block)
+        self.ips_list_scopes.pop
+      else
+        self.containers.last.ips_list << ips
+      end
+    end
+    
+    def subject(name, &block)
+      subject = Subject.new
+      subject.name = name
+      
+      parent = self.containers.last
+      parent.children << subject
+      subject.parent = parent
+      
+      containers_eval(subject, &block) if block
+    end
+    
+    def hosts(*args, &block)
+      options = args.extract_options!
+      
+      container = self.containers.last
+      hosts = Hosts.new
+      hosts.options = options
+      hosts.domains = [ args ].flatten
+      hosts.parent = container
+      container.children << hosts
+      
+      if !self.ips_list_scopes.empty?
+        ips_list = self.ips_list_scopes.last
+        hosts.ips_list = ips_list
+      end
+      
+      containers_eval(hosts, &block) if block
+    end
+    
+    def to_hosts(options = {})
+      self.groups.map do |group|
+        group.to_hosts(options)
+      end.join("\n\n")
+    end
+    
+    def to_dnsmasq(options = {})
+      self.groups.map do |group|
+        group.to_dnsmasq(options)
+      end.join("\n\n")
+    end
+    
+  end
+  
+  class DslException < Exception
+    
+  end
+  
+end

data/lib/hosum/group.rb

@@ -0,0 +1,15 @@
+module Hosum
+  
+  class Group < Container
+    attr_accessor :name
+    
+    def to_hosts(options = {})
+      "# #{self.name}\n" + children.map{ |c| c.to_hosts(options) }.join
+    end
+    
+    def to_dnsmasq(options = {})
+      "# #{self.name}\n" + children.map{ |c| c.to_dnsmasq(options) }.join
+    end
+  end
+  
+end

data/lib/hosum/hosts.rb

@@ -0,0 +1,59 @@
+module Hosum
+  class Hosts < Container
+    attr_accessor :domains
+
+    def initialize
+      super
+      self.domains = []
+    end
+    
+    def expand_domains
+      self.domains.map do |domain|
+        holders = []
+        
+        regex = /\[([^\[\]]+)-([^\[\]]+)\]/
+        domain = domain.gsub(regex) do |str|
+          m = str.match(regex)
+          holders << Range.new(m[1], m[2]).to_a
+          "%s"
+        end
+        
+        expand_list = []
+        holders.each do |holder|
+          if expand_list.empty?
+            expand_list = holder.map {|v| [v] }
+            next
+          end
+
+          tmp = []
+          expand_list.each do |expand|
+            holder.each do |h|
+              tmp << expand + [ h ]
+            end
+          end
+          expand_list = tmp
+        end
+        
+        if holders.empty?
+          domain
+        else
+          expand_list.map { |e| domain % e}
+        end
+        
+      end.flatten
+    end
+
+    def to_hosts(options = {})
+      self.expand_domains.map{ |domain| "#{find_first_ip}	#{domain}\n" }.join
+    end
+    
+    def to_dnsmasq(options = {})
+      domains = self.expand_domains
+      if self.options[:dnsmasq]
+        domains = [ self.options[:dnsmasq] ].flatten
+      end
+      
+      domains.map{ |domain| "address=/#{domain}/#{find_first_ip}\n" }.join
+    end
+  end
+end

data/lib/hosum/ips.rb

@@ -0,0 +1,6 @@
+module Hosum
+  class Ips
+    attr_accessor :options
+    attr_accessor :addresses
+  end
+end

data/lib/hosum/subject.rb

@@ -0,0 +1,15 @@
+module Hosum
+  
+  class Subject < Container
+    attr_accessor :name
+    
+    def to_hosts(options = {})
+      "\n# #{self.parent.name}.#{self.name}\n" + children.map{ |c| c.to_hosts(options) }.join
+    end
+    
+    def to_dnsmasq(options = {})
+      "\n# #{self.parent.name}.#{self.name}\n" + children.map{ |c| c.to_dnsmasq(options) }.join
+    end
+  end
+  
+end

data/spec/lib/dnsmasq.dnsmasq

@@ -0,0 +1,2 @@
+# itunes
+address=/.phobos.apple.com/203.69.138.18

data/spec/lib/dnsmasq.dsl

@@ -0,0 +1,5 @@
+group "itunes" do
+  hosts "a[1-2000].phobos.apple.com", :dnsmasq => ".phobos.apple.com"
+  
+  ip "203.69.138.18", :any => :bstar
+end

data/spec/lib/hosum/hosts_spec.rb

@@ -0,0 +1,17 @@
+require "spec_helper"
+
+describe Hosum::Hosts do
+  
+  it "expand hosts" do
+    hosts = Hosum::Hosts.new
+    hosts.domains << "a[7-10].apple.com"
+    hosts.expand_domains.should == [ "a7.apple.com", "a8.apple.com", "a9.apple.com", "a10.apple.com" ]
+  end
+  
+  it "don't expand hosts" do
+    hosts = Hosum::Hosts.new
+    hosts.domains << "www.apple.com"
+    hosts.expand_domains.should == [ "www.apple.com" ]
+  end
+  
+end

data/spec/lib/hosum_spec.rb

@@ -0,0 +1,19 @@
+require "spec_helper"
+
+describe Hosum do
+  
+  it "dsl" do
+    paths = File.expand_path("../*.dsl", __FILE__)
+    Dir.glob(paths).each do |dsl_path|
+      rs = Hosum.dsl_by_file(dsl_path)
+      
+      hosts_file = dsl_path.sub(".dsl", ".hosts")
+      rs.to_hosts.should == File.read(hosts_file) if File.exist?(hosts_file)
+      
+      dnsmasq_file = dsl_path.sub(".dsl", ".dnsmasq")
+      rs.to_dnsmasq.should == File.read(dnsmasq_file) if File.exist?(dnsmasq_file)
+      
+    end
+  end
+  
+end

data/spec/lib/simple.dsl

@@ -0,0 +1,12 @@
+group "itunes" do
+  ip "1.1.1.1"
+  hosts "www1.apple.com"
+  
+  hosts "www2.apple.com" do
+    ip "2.2.2.2"
+  end
+  
+  ip "3.3.3.3" do
+    hosts "www3.apple.com"
+  end
+end

data/spec/lib/simple.hosts

@@ -0,0 +1,4 @@
+# itunes
+1.1.1.1	www1.apple.com
+2.2.2.2	www2.apple.com
+3.3.3.3	www3.apple.com

data/spec/lib/subject.dsl

@@ -0,0 +1,14 @@
+group "google" do
+  
+  ip "1.1.1.1"
+  
+  subject "core" do
+    hosts "www.google.com"
+  end
+  
+  subject "search" do
+    ip "2.2.2.2"
+    hosts "encrypted.google.com"
+  end
+  
+end

data/spec/lib/subject.hosts

@@ -0,0 +1,7 @@
+# google
+
+# google.core
+1.1.1.1	www.google.com
+
+# google.search
+2.2.2.2	encrypted.google.com

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'rspec'
+Dir[File.expand_path(File.join(File.dirname(__FILE__),'support','**','*.rb'))].each {|f| require f}
+
+require "hosum"
+
+RSpec.configure do |c|
+end

metadata

@@ -0,0 +1,139 @@
+--- !ruby/object:Gem::Specification
+name: hosum
+version: !ruby/object:Gem::Version
+  version: 0.0.1a
+  prerelease: 5
+platform: ruby
+authors:
+- Sunteya
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-09-21 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: &70341398285800 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70341398285800
+- !ruby/object:Gem::Dependency
+  name: i18n
+  requirement: &70341398285320 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70341398285320
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70341398284780 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70341398284780
+- !ruby/object:Gem::Dependency
+  name: guard-rspec
+  requirement: &70341398284060 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70341398284060
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &70341398274020 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70341398274020
+description: host file generator
+email:
+- Sunteya@gmail.com
+executables:
+- hosum
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Guardfile
+- README
+- Rakefile
+- VERSION
+- bin/hosum
+- examples/my.hosum
+- hosum.gemspec
+- lib/hosum.rb
+- lib/hosum/.rb
+- lib/hosum/container.rb
+- lib/hosum/dsl.rb
+- lib/hosum/group.rb
+- lib/hosum/hosts.rb
+- lib/hosum/ips.rb
+- lib/hosum/subject.rb
+- spec/lib/dnsmasq.dnsmasq
+- spec/lib/dnsmasq.dsl
+- spec/lib/hosum/hosts_spec.rb
+- spec/lib/hosum_spec.rb
+- spec/lib/simple.dsl
+- spec/lib/simple.hosts
+- spec/lib/subject.dsl
+- spec/lib/subject.hosts
+- spec/spec_helper.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -3945610133677893382
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>'
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubyforge_project: hosum
+rubygems_version: 1.8.7
+signing_key: 
+specification_version: 3
+summary: host file generator
+test_files:
+- spec/lib/dnsmasq.dnsmasq
+- spec/lib/dnsmasq.dsl
+- spec/lib/hosum/hosts_spec.rb
+- spec/lib/hosum_spec.rb
+- spec/lib/simple.dsl
+- spec/lib/simple.hosts
+- spec/lib/subject.dsl
+- spec/lib/subject.hosts
+- spec/spec_helper.rb