#!/usr/local/bin/php -q
<?php
########################################################################################
# This file is for splitting binlogs up into their respective parts. To put it more 
# plainly, to seperate the queries for various databases on a particular MySQL server. 
# In the version 4 series (someone correct me if I'm wrong) one may specify which db 
# they want the information for making this script useless. However, with 3.23.xx, 
# this script should come in handy. 
########################################################################################
# Let's first set a timelimit. 
set_time_limit(0);

# Check number of arguments
$args=&check_args();

# Let's also create our file array now
$file_array=array();

# If we are after only one database open a file for it now
if($args['db']!='all')
	{
	# Let's first check to see if the file exists
	if(file_exists($args['file']))
		{
		echo "About to open file\n";
		$f1=fopen($args['db'], 'a');
		if($f1)
			{	
			$file_array[$args['db']]['fp']=$f1; 
			$file_array[$args['db']]['name']=$args['db'];
			$file_array[$args['db']]['count']=0;
			}
		else
			{ echo "Failed to open a file for the db ".$args['db']."\n"; exit; }
		}
	else
		{ echo $args['file']." doesn't exist!\n"; exit; }	
	}

# Now open the sql file into an array
$queries=file($args['file']);

# Don't we need to know which file to write to. 
$current='';
while(list( , $v)=each($queries))
	{
	# This is where the business is taken care of. All data that we care about will 
	# be written to the appropriate file here. Also, if the all option was chosen 
	# earlier, it will be dealt with here by dynamically opening file streams for newly 
	# encountered dbs.
	
	# Let's first check for use statememnts. Also see if we are recording data for the 
	# db in question.
	if( (stristr($v, 'use')) )
		{
		# Do we need to parse the string, or is this a false alarm.
		if( (stristr($v, 'INSERT')) || (stristr($v, 'UPDATE')) || (stristr($v, 'DELETE')) || (stristr($v, 'ALTER')) )
			{
			if(isset($file_array[$current]['fp']))
				{ 
				fwrite($file_array[$current]['fp'], $v); 
				++$file_array[$current]['count']; 
				}
			}
		else
			{	$db=parse_use(&$v); }

		# Is there a file pointer for this db?
		if(isset($file_array[$db]['fp']))
			{	$current=$db; }
		# Are we dumping out all dbs?
		elseif($args['db']=='all')
			{ 
			add_fp(&$file_array, &$db); 
			$current=$db;
			}	
		# If only dumping one db, set current back to nada.		
		else
			{ $current=''; }		
		}
		
	# If we have no use statement, then we are either writing to a file or falling through.	
	elseif($current!='')
		{
		if((stristr($v, 'SET TIMESTAMP')) || (stristr($v, 'SET INSERT_ID')))
			{ /* do nothing and fall through */ }
		else
			{ 
			fwrite($file_array[$current]['fp'], $v); 
			++$file_array[$current]['count'];
			} 
		}
	else
		{ /* do nothing */ }
	}	

# Close all open file pointers and generate report
$temp_count=0;
foreach($file_array as $x)
	{
	echo $x['name']." had a total of ".$x['count']." lines.\n";
	$temp_count=($temp_count+$x['count']);
	fclose($x['fp']);
	}

# The End....

########################################################################################
# Functions list
########################################################################################

# This function will check the arguments 
function check_args()
	{
	# Create the array that will be returned
	$data_arr=array("file"=>'', "db"=>'');
	
	# Let's first check the number of args
	if($_SERVER['argc']!=3)
		{ 
		echo "Incorrect argument count! Use -h for usage.\n";
		exit;
		}
	
	# Assign data to the array
	$data_arr['file']=&$_SERVER['argv'][1];
	$data_arr['db']=&$_SERVER['argv'][2];

	# Return the information
	return $data_arr;
	}


# Parse the 'use db' string
function parse_use($str)
	{
	$new=str_replace("use", "", $str);
	$new=str_replace(";", "", $new);
	$new=str_replace(" ", "", $new);
	$new=trim($new);
	return $new;
	}


# Add another file to be written too to an array of file pointers.
function add_fp($files, $db)
	{
	$fp=fopen($db, 'a');
	if($fp)
		{	
		$files[$db]['fp']=$fp;
		$files[$db]['name']=$db;		
		$files[$db]['count']=0;
		}
	else
		{ echo "Failed to open a file for the db ".$db."\n"; exit; }
	}

?>