#!/usr/bin/perl

$version="1.0";

use Getopt::Long;

$opt_help=$opt_Information=0;
$opt_regexp=".*";
$opt_start_row=1; $opt_rows=9999999999;

GetOptions("Information","help","regexp=s","start_row=i","rows=i")
  || usage();
usage() if ($opt_help || $opt_Information);

$query=$search=$database=$set=""; $eoq=0;
while (<>)
{
  next if (length($query) == 0 && /^\#/); # Skipp comments
  $query.=search($_);
  if ($eoq)
  {
    if ($query =~ /^use / || $query =~ /^SET / || $query =~ /$opt_regexp/)
    {
      if ($opt_start_row <= 1)
      {
	if ($database)
	{
	  print $database, $set;
	  $database=$set="";
	}
	print $query;
	last if (--$opt_rows == 0);
      }
      else
      {
	$opt_start_row--;
	if ($query =~ /^use /)
	{
	  $database=$query;
	  $set="";
	}
	elsif ($query =~ /^SET/)
	{
	  $set=$query;
	}
	else
	{
	  $set="";
	}
      }
    }
    $query=""; $search=""; $eoq=0;
  }
}

exit 0;

sub search
{
  my($row)=shift;
  my($i);

  for ($i=0 ; $i <= $#row ; $i++)
  {
    if ($search)
    {
      if (length($search) > 1)
      {				# Comment
	next if (mid($row,$i,length($search)) ne $search);
	$i+=length($search)-1;
	$search="";
      }
      elsif (mid($row,$i,1) eq '\\') # Escaped char in string
      {
	$i++;
      }
      elsif (mid($row,$i,1) eq $search)
      {
	if (mid($row,$i+1,1) eq $search)	# Double " or '
	{
	  $i++;
	}
	else
	{
	  $search="";
	}
      }
      next;	
    }
    if (mid($row,$i,2) eq '/*')	# Comment
    {
      $search="*/";
      $i++;
    }
    elsif ($row[$i] eq '\'' || $row[$i] eq '"')
    {
      $search=$row[$i];
      $i++;
      next;
    }
  }
  $eoq=1 if (!$search && $row =~ /;\s*$/);
  return $row;
}


sub usage
{
    print <<EOF;
$0  Ver $version

TCX Datakonsult AB, by Monty.
This software comes with NO WARRANTY: see the file PUBLIC for details.

Prints all SQL queries that matches a regexp or contains a 'use
database' or 'set ..' command to stdout.  A SQL query may contain
newlines.  This is useful to find things in a MySQL update log.

$0 takes the following options:

--help or --Information
  Shows this help

--regexp=#
  Print queries that matches this.

--start_row=#
  Start output from this row (first row = 1)

--rows=#
  Quit after this many rows.

EOF
  exit(0);
}