gbwd 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in gbwd.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Thomas Brewer
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,53 @@
+# Gbwd
+
+A little Gem to help eliminate website distractions.  It basically routes any requests for the domains that are being blocked to [http://getbacktoworkdummy.com](http://getbacktoworkdummy.com "http://getbacktoworkdummy.com")
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'gbwd'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install gbwd
+
+## Usage
+
+*NOTE: Please back up your /etc/hosts file before running any commands.*
+
+Add a domain to be blocked:
+	
+	sudo gbwd add -d www.youtube.com
+	
+Remove a domain from being blocked:
+
+	sudo gbwd remove -d www.youtube.com
+	
+List all domains being blocked:
+
+	sudo gbwd list
+	
+Stop all domains from being blocked:
+
+	sudo gbwd disable
+	
+Start blocking all domains again:
+
+	sudo gbwd enable
+
+## Todos
+1. Make a copy of /etc/hosts file before modifying it
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/bin/gbwd

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby -w
+
+require 'subcommand'
+require 'gbwd'
+
+include Subcommands
+
+options = {}
+
+command :add do |opts|
+	opts.banner = "Usage: add -d www.youtube.com"
+	opts.description = "Add a domain to the list"
+	opts.on('-d', '-domain DOMAIN', 'Domain name (without the http://)') do |domain|
+      options[:domain] = domain
+    end
+end
+
+command :remove do |opts|
+	opts.banner = "Usage: remove -d www.youtube.com"
+	opts.description = "Removes a domain to the list"
+	opts.on('-d', '-domain DOMAIN', 'Domain name (without the http://)') do |domain|
+      options[:domain] = domain
+    end
+end
+
+command :disable do |opts|
+	opts.banner = "Usage: disable"
+	opts.description = "Disables all domains being blocked."
+end
+
+command :enable do |opts|
+	opts.banner = "Usage: enable"
+	opts.description = "Enables all domains being blocked."
+end
+
+command :list do |opts|
+	opts.banner = "Usage: list"
+	opts.description = "Lists all domain on the list"
+end
+
+puts Gbwd.start(opt_parse(), options)

data/gbwd.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/gbwd/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Thomas Brewer"]
+  gem.email         = ["tom@21purple.com"]
+  gem.description   = %q{Command line tool for the highly performant and scalable GBWD service}
+  gem.summary       = gem.description
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = %w(gbwd)
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "gbwd"
+  gem.require_paths = ["lib"]
+  gem.version       = Gbwd::VERSION
+  gem.add_development_dependency('fakefs')
+  gem.add_development_dependency('rspec')
+  gem.add_dependency('subcommand')
+end

data/lib/gbwd.rb

@@ -0,0 +1,14 @@
+require "gbwd/version"
+require "gbwd/cli"
+require "gbwd/commands"
+
+module Gbwd
+  
+	def self.start(command, options)
+		commands = Commands.new
+		cli = Cli.new(commands)
+		cli.run(command, options)
+	end
+
+end
+

data/lib/gbwd/cli.rb

@@ -0,0 +1,19 @@
+module Gbwd
+  
+	class Cli
+
+		def initialize(commands)
+			@commands = commands
+		end
+
+		def commands
+			@commands
+		end
+
+		def run(command, options)
+			commands.public_send(command.to_sym, options)
+		end
+
+	end
+	  
+end

data/lib/gbwd/commands.rb

@@ -0,0 +1,119 @@
+module Gbwd
+  
+	class Commands
+
+		def initialize
+			@hosts_file_path = '/etc/hosts'
+			@ip_address = '69.55.54.215'
+		end	
+
+		def add(options = {})
+			modify(options) do |domains, domain|
+				domains << [ip_address, domain].join(' ')
+				"You added #{domain}."
+			end
+		end
+
+		def remove(options = {})
+			modify(options) do |domains, domain|
+				domains.select! do |domain_to_check|
+					(Regexp.new(domain) =~ domain_to_check).nil?
+				end
+				"You removed #{domain}."
+			end
+		end
+
+		def enable(options = {})
+			modify(options) do |domains, domain|
+				domains.map! do |line|
+					line[1..-1] if line.start_with?('#')
+				end
+				"All domains have been enabled."
+			end
+		end
+
+		def disable(options = {})
+			modify(options) do |domains, domain|
+				domains.map! do |line|
+					"#" + line
+				end
+				"All domains have been disabled."
+			end
+		end
+
+		def list(options = {})
+			install unless installed?
+			domains = get_domains_from_file
+			if domains.size == 0
+				[
+					"You do not have any blocked domains. You can add one by using the add command.",
+					"  ex. ~ sudo gbwd add -d www.youtube.com"	
+				].join("\n")
+			else
+				domains = domains.map do |line|
+					line.split(' ')[1]
+				end
+				"The following domains are currenlty being blocked:\n" + domains.join("\n")
+			end
+		end
+
+	private
+
+		def modify(options)
+			raise ArgumentError.new('You must pass a block to the modify method.') unless block_given?
+			install unless installed?
+			domain = options.fetch(:domain, nil)
+			domains = get_domains_from_file
+			response = yield(domains, domain)
+			write_to_hosts_file(domains)
+			response
+		end
+
+		def hosts_file_path
+			@hosts_file_path
+		end
+
+		def ip_address
+			@ip_address
+		end
+
+		def get_domains_from_file
+			file = File.open(hosts_file_path, 'r').read
+			matches = file.match(/#-GBWD START(.*)#-GBWD END/m)
+			domains = []
+			unless matches.nil?
+				matches.captures[0].split("\n").each do |domain|
+					unless domain.empty?
+						domains << domain
+					end
+				end
+			end
+			domains
+		end
+
+		def wrap_with_header_and_footer(lines)
+			lines.insert(0, '#-GBWD START')
+			lines << '#-GBWD END'
+		end
+
+		def write_to_hosts_file(lines)
+			updated_content = wrap_with_header_and_footer(lines).join("\n")
+			file = File.open(hosts_file_path, 'r').read
+			updated_file = file.gsub(/#-GBWD START.*#-GBWD END/m, updated_content)
+			File.open(hosts_file_path, 'w').write(updated_file)
+		end
+
+		def installed?
+			file = File.open(hosts_file_path, 'r').read 
+			(file.match(/#-GBWD START/)) ? true : false
+		end
+
+		def install
+			File.open(hosts_file_path, 'a+') do |file|
+				file.puts("#-GBWD START\n#-GBWD END")
+			end
+		end
+
+	end
+
+end

data/lib/gbwd/version.rb

@@ -0,0 +1,3 @@
+module Gbwd
+  VERSION = "0.0.1"
+end

data/spec/gbwd/cli_spec.rb

@@ -0,0 +1,13 @@
+require 'spec_helper'
+
+describe Gbwd::Cli do
+	
+	it "#run should call the appropriate sub command when passed a command" do
+		commands = double("Commands")
+		commands.should_receive(:add)
+		cli = Gbwd::Cli.new(commands)
+		options = double("options")
+		cli.run('add', options)
+	end
+
+end

data/spec/gbwd/commands_spec.rb

@@ -0,0 +1,86 @@
+require 'spec_helper'
+
+describe Gbwd::Commands, fakefs: true do
+	
+	before do
+	    FileUtils.mkdir("/etc")
+	    File.open("/etc/hosts", "w") do |f|
+			f.puts("##")
+			f.puts("# Host Database")
+			f.puts("#")
+			f.puts("# localhost is used to configure the loopback interface")
+			f.puts("# when the system is booting.  Do not change this entry.")
+			f.puts("##")
+			f.puts("127.0.0.1	localhost")
+			f.puts("255.255.255.255	broadcasthost")
+			f.puts("::1             localhost")
+			f.puts("fe80::1%lo0	localhost")
+			f.puts("\n")
+			f.puts("\n")
+			f.puts("# VHX START")
+			f.puts("127.0.0.1 test.dev")
+			f.puts("fe80::1%lo0 test.dev")
+			f.puts("# VHX STOP")
+	    end
+  	end
+	
+	it "#add should add a domain to the hosts file" do
+		commands = Gbwd::Commands.new
+		commands.add({domain: 'www.youtube.com'})
+		file = File.open("/etc/hosts", 'r').read
+		file.match(/www.youtube.com/m).should_not eq(nil)
+	end
+
+	it "#remove should remove a domain to the hosts file" do
+		commands = Gbwd::Commands.new
+		commands.add({domain: 'www.youtube.com'})
+		commands.remove({domain: 'www.youtube.com'})
+		file = File.open("/etc/hosts", 'r').read
+		file.match(/www.youtube.com/m).should eq(nil)
+	end
+
+	it "#list should return a list of all domains" do
+		commands = Gbwd::Commands.new
+		commands.add({domain: 'www.youtube.com'})
+		commands.add({domain: 'www.facebook.com'})
+		commands.list.should eq("The following domains are currenlty being blocked:\nwww.youtube.com\nwww.facebook.com")
+	end
+
+	it "#disable should comment out all lines" do
+		commands = Gbwd::Commands.new
+		commands.add({domain: 'www.youtube.com'})
+		commands.add({domain: 'www.facebook.com'})
+		commands.disable({})
+		file = File.open("/etc/hosts", 'r').read
+		matches = file.match(/#-GBWD START(.*)#-GBWD END/m)
+		unless matches.nil?
+			matches.captures[0].split("\n").each do |domain|
+				unless domain.empty?
+					domain.start_with?('#').should eq(true)
+				end
+			end
+		else
+			raise 'No domains found'
+		end
+	end
+
+	it "#enable should uncomment all lines" do
+		commands = Gbwd::Commands.new
+		commands.add({domain: 'www.youtube.com'})
+		commands.add({domain: 'www.facebook.com'})
+		commands.disable({})
+		commands.enable({})
+		file = File.open("/etc/hosts", 'r').read
+		matches = file.match(/#-GBWD START(.*)#-GBWD END/m)
+		unless matches.nil?
+			matches.captures[0].split("\n").each do |domain|
+				unless domain.empty?
+					domain.start_with?('#').should_not eq(true)
+				end
+			end
+		else
+			raise 'No domains found'
+		end
+	end
+	
+end

data/spec/gbwd_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe Gbwd do
+	
+	
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+require 'rspec'
+require 'gbwd'
+require 'fakefs/spec_helpers'
+
+RSpec.configure do |config|
+  config.mock_with :rspec
+  config.include FakeFS::SpecHelpers, fakefs: true
+end
+

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification
+name: gbwd
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Thomas Brewer
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fakefs
+  requirement: &70343999590760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70343999590760
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70343999590320 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70343999590320
+- !ruby/object:Gem::Dependency
+  name: subcommand
+  requirement: &70343999589720 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70343999589720
+description: Command line tool for the highly performant and scalable GBWD service
+email:
+- tom@21purple.com
+executables:
+- gbwd
+extensions: []
+extra_rdoc_files: []
+files:
+- .DS_Store
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- bin/gbwd
+- gbwd.gemspec
+- lib/.DS_Store
+- lib/gbwd.rb
+- lib/gbwd/.DS_Store
+- lib/gbwd/cli.rb
+- lib/gbwd/commands.rb
+- lib/gbwd/version.rb
+- spec/gbwd/cli_spec.rb
+- spec/gbwd/commands_spec.rb
+- spec/gbwd_spec.rb
+- 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'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Command line tool for the highly performant and scalable GBWD service
+test_files:
+- spec/gbwd/cli_spec.rb
+- spec/gbwd/commands_spec.rb
+- spec/gbwd_spec.rb
+- spec/spec_helper.rb