70 lines
1.8 KiB
Ruby
70 lines
1.8 KiB
Ruby
require 'fileutils'
|
|
|
|
class Noprocrast
|
|
class << self
|
|
def default_hosts
|
|
['news.ycombinator.com', 'twitter.com', 'facebook.com', 'reddit.com']
|
|
end
|
|
|
|
def deny_file_path
|
|
File.expand_path("~/.noprocast")
|
|
end
|
|
|
|
def current_hosts
|
|
setup_deny_file_if_required!
|
|
hosts = File.read(deny_file_path).split(/\n/).select { |line| line.match(/[a-zA-Z0-9]/) }.map(&:strip)
|
|
wwwhosts = hosts.map { |h| "www." + h.to_s unless h =~ /^www/ }
|
|
(hosts + wwwhosts).sort
|
|
end
|
|
|
|
def hosts_file_content
|
|
File.read("/etc/hosts")
|
|
end
|
|
|
|
def activate!
|
|
backup_hosts_file_if_required!
|
|
deactivate! # ensure that /etc/hosts is clean
|
|
File.open("/etc/hosts", 'a') do |file|
|
|
file << "\n\n# noprocrast start\n#{current_hosts.map { |host| "127.0.0.1 #{host}" }.join("\n")}\n# noprocrast end"
|
|
end
|
|
system "dscacheutil -flushcache" # only for OSX >= 10.5: flush the DNS cache
|
|
end
|
|
|
|
def deactivate!
|
|
clean_hosts = hosts_file_content.gsub(/(\n\n)?\# noprocrast start.*\# noprocrast end/m, '')
|
|
File.open("/etc/hosts", 'w') do |file|
|
|
file << clean_hosts
|
|
end
|
|
end
|
|
|
|
def active?
|
|
hosts_file_content.match(/\# noprocrast start/)
|
|
end
|
|
|
|
def status_message
|
|
active? ? "noprocrast enabled for #{current_hosts.size} hosts" : "noprocrast disabled"
|
|
end
|
|
|
|
def backup_hosts_file_if_required!
|
|
unless File.exists?("/etc/.hosts.noprocrastbackup")
|
|
FileUtils.cp("/etc/hosts", "/etc/.hosts.noprocrastbackup")
|
|
end
|
|
end
|
|
|
|
def setup_deny_file_if_required!
|
|
unless File.exists?(deny_file_path)
|
|
File.open(deny_file_path, 'w') do |file|
|
|
file << default_hosts.join("\n")
|
|
end
|
|
end
|
|
end
|
|
|
|
def edit!
|
|
setup_deny_file_if_required!
|
|
editor = ENV['EDITOR'] || 'vi'
|
|
system "#{editor} #{deny_file_path}"
|
|
end
|
|
end
|
|
end
|
|
|