get-shit-done.php

#!/usr/bin/env php
<?php

if ( 1 == $argc ) {
	exitWithError("usage: " . $argv[0] . " [work | play]");
}

$whoami = trim(`whoami`);

if ( 'root' != strtolower($whoami) ) {
	exitWithError("Please run script as root.");
}

$homedir = trim(`cd ~ && pwd`);
$iniFile = $homedir.'/.get-shit-done.ini';

if (file_exists($iniFile)) {
	$iniContents = parse_ini_file($iniFile);
	$siteList = $iniContents['sites'];
} else {
	$siteList = array(
		'reddit.com', 'forums.somethingawful.com', 'somethingawful.com',
		'digg.com', 'break.com', 'news.ycombinator.com',
		'infoq.com', 'bebo.com', 'twitter.com',
		'facebook.com', 'blip.com', 'youtube.com',
		'vimeo.com', 'delicious.com', 'flickr.com',
		'friendster.com', 'hi5.com', 'linkedin.com',
		'livejournal.com', 'meetup.com', 'myspace.com',
		'plurk.com', 'stickam.com', 'stumbleupon.com',
		'yelp.com', 'slashdot.org'
	);
}

$uname = trim(`uname`);

if ( $uname == 'Linux' ) {
    $restartNetworkingCommand = '/etc/init.d/networking restart';
} elseif ( $uname == 'Darwin' ) {
    $restartNetworkingCommand = 'dscacheutil -flushcache';
} else {
    $message = '"Please contribute DNS cache flush command on GitHub."';
    $restartNetworkingCommand = "echo $message";
    # Intention isn't to exit, as it still works, but just requires some
    # intervention on the user's part.
}

$hostsFile = '/etc/hosts';
$startToken = '## start-gsd';
$endToken = '## end-gsd';

$action = $argv[1];

switch ( $action ) {
	case 'work': {
		$contents = file_get_contents($hostsFile);
		if($contents && strpos($contents, $startToken) !== false && strpos($contents, $endToken) !== false) {
			exitWithError("Work mode already set.");        
		}
		
		$fh = fopen($hostsFile, 'a');
		if ( false === $fh ) {
			exitWithError("Failed to open the hosts file.");
		}
		
		fwrite($fh, $startToken . PHP_EOL);
		foreach ( $siteList as $site ) {
			fwrite($fh, "127.0.0.1\t{$site}" . PHP_EOL);
			fwrite($fh, "127.0.0.1\twww.{$site}" . PHP_EOL);
		}
		fwrite($fh, $endToken . PHP_EOL);
		
		fclose($fh);
		
		shell_exec($restartNetworkingCommand);
		
		break;
	}
	
	case 'play': {
		$hostContents = file($hostsFile);
		if ( false === $hostContents ) {
			exitWithError("Failed to open the hosts file.");
		}
		
		$startIndex = -1;
		for ( $i=0; $i<count($hostContents); $i++ ) {
			if ( trim($hostContents[$i]) == $startToken ) {
				$startIndex = $i;
			}
		}
		
		if ( $startIndex > -1 ) {
			$hostContents = array_slice($hostContents, 0, $startIndex);
			file_put_contents($hostsFile, $hostContents);
			shell_exec($restartNetworkingCommand);
		}
		
		break;
	}
}

function exitWithError($error) {
	fwrite(STDERR, $error . PHP_EOL);
	exit(1);
}