noprocrast/lib/noprocrast.rb

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