blackhole.rb

blackhole.rb

#! /usr/bin/ruby

require 'fileutils'

path = "/etc/hosts"

sites = %w(
  news.ycombinator.com
  reddit.com
  www.reddit.com
  facebook.com
  www.facebook.com
  ign.com
  www.ign.com
  nytimes.com
  www.nytimes.com
)

blacholed = File.open(path).grep(/Blackhole/)

if blacholed.empty?
  File.open(path, "a") do |f|
    f.puts "\n"
    sites.each do |s|
      f.puts("127.0.0.1 #{s} # Blackhole")
      f.puts("fe80::1%lo0 #{s} # Blackhole")
    end
  end
else
  tmp_path = path + ".tmp"
  FileUtils.cp path, tmp_path

  File.open(path, 'w') do |output|
    IO.foreach(tmp_path) do |line|
      output.write(line) unless line.include? "Blackhole"
    end
  end

  FileUtils.rm tmp_path
end