qc.pl

#!/usr/bin/perl
use strict;
use File::Basename;
use File::Spec;

# reduce repetitive stuff in c++
#quick C = qc
our $outdir;
our $errorOnChanges;

$errorOnChanges = 0;

sub abspath {
	my ($path) = @_;
	return File::Spec->rel2abs($path);
}

# get an array of all files and sub files of given filetype
sub AllFiles  {
        my ($filetype) = @_;
        my @files;
        my $filtered = [];
        @files = split("\n", `find ./ -type f`);

        foreach(@files){
                if($_ =~ /^\.\/+(.*\.$filetype)$/){
                        push (@$filtered, $1);
                }
        }
        return $filtered;
}

sub equalArrays {
	my ($a, $b) = @_;
	
	if(@$a != @$b){
		return 0;
	}
	my $i;
	for($i = 0; $i < @$a; $i++){
		if($$a[$i] ne $$b[$i]){
			return 0;
		}
	}
	return 1;
}
sub sameFileContents {
	my ($contents, $file) = @_;
	my $fcontents;
	$fcontents = loadFile($file);
	if(!$fcontents) {
		return 0;
	}
	return equalArrays($contents, $fcontents);
}
sub loadFile {
	my ($file) = @_;
	my ($fh);
	if(!(open($fh, "<$file"))){
		print("could not open $file\n");
		return 0;
	}
	
	my $lines = [];
	while(<$fh>){
		chomp $_;
		push @$lines, $_;
	}
	close($fh);
	return $lines;
}

sub saveFile {
	my ($lines, $file) = @_;
	my $i;
	my $fh;
	my $r;
	$r = open $fh, ">$file";
	if(!$r) { 
		print "could not save $file\n"; 
		return 0; 
	}
	for($i = 0; $i < @$lines; $i++){
		print $fh $$lines[$i];
		print $fh "\n";
	}
	close $fh;
	print "Saved $file ...\n";
}
sub OKToRemake {
	my ($file) = @_;
	my $fh;
	if(!(open($fh, "<$file"))){
		print("could not open $file\n");
		return 0;
	}
	
	my ($line_start, $line_end);
	my ($line_num);
	$line_start = 0;
	$line_end = 0;
	
	$line_num =0;
	while(<$fh>){
		if($_ =~ m/^\#pragma\s+qc start\s*/){
			$line_start = $line_num;
		}
		if($_ =~ m/^\#pragma\s+qc end\s*/){
			$line_end = $line_num;
			close($fh);
			return [$line_start, $line_end];
		}
		$line_num++;

	}
	close($fh);
	return 0;
}

# return true when it is a pure header (All qc generated)
sub isQCAll {
    my ($file) = @_;
    my $lines;
	$lines = loadFile($file);
	if(!$lines) {
		return 0;
	}
    if($$lines[0] =~ m/#pragma\s+qc main/){
        return 1;
    }
    return 0;
}

# the new thing to do which generates entire header for you
sub qcAll {
	my ($cppfile, $hfile) = @_;

    my $lines;
    my $gen = [];
	
	my $function;
    my $gl;
	my $list = [];
	my $i;
    my $changed;
	$changed = 0;

    my $purecopy;
    $lines = loadFile($cppfile);
    if(!$lines){
        print "Could not open $cppfile";
    }
    print "Processing $cppfile ...\n";
    $purecopy = 0;
	
	push @$gen, "#pragma qc main";
	push @$gen, "#pragma once";
	
    for($i = 0; $i < @$lines; $i++){
		my $l;
        $l = $$lines[$i];
        if($purecopy > 0){
            if($l =~ m/^#if/){
                $purecopy++;
            } elsif($l =~ m/^#endif/){
                $purecopy--;
            }
            if($purecopy > 0){
                push @$gen, $l;
            }
            next;
        }
		if($l =~ m/^(.*)\scc\s(.*)$/){
			$function = "$1 $2";
			if($function =~ m/(.*?)\s*\{\s*/){
				$function = $1;
			}
			if($function =~ m/^#/){
				next;
			}
			
			# add to our list
			push @$gen, "$function;";
		} elsif($l =~ m/^\s*cc\s(.*)$/){
			$function = "$1";
			if($function =~ m/(.*?)\s*\{\s*/){
				$function = $1;
			}
			if($function =~ m/^#/){
					next;
			}
			
			# add to our list
			push @$gen, "$function;";
		}
		
		elsif($l =~ m/^#pragma\s+qc\s+class\s+(\S+)\s*\:\s*(\S+)\s*$/){
			# auto put public :)
			$gl = "class $1 : public $2 {\npublic:\ntypedef $2 super;";
			my @a;
			@a = split(/\n/, $gl);
			push @$gen, @a;
		}elsif ($l =~ m/^#pragma\s+qc\s+class\s+(.*)$/){
			$gl = $1;
			$gl = "class $gl {\npublic:";
			my @a;
			@a = split(/\n/, $gl);
			push @$gen, @a;
		} elsif ($l =~ m/^#pragma\s+qc\s+endc/){
			$gl = "};";
			push @$gen, $gl;
		} elsif ($l =~ m/^#ifdef QC_PURE/){
			$purecopy = 1;
		} elsif ($l =~ m/^#ifndef QC_PURE/){ # so it can be in both places
			$purecopy = 1;
		} 
		#elsif ($l =~ m/^(#include.*$)/){
		#	push @$gen, $1;
		#} elsif ($l =~ m/^(#import.*$)/){
		#	push @$gen, $1;
		#}
            
        
	}
	
	if(!sameFileContents($gen, $hfile)){
		print ("$hfile changed\n");
		saveFile($gen, $hfile);
		if(!sameFileContents($gen, $hfile)){
			print "error::!!!! saved file but its different??";
		}
		$changed = 1;
	} else {
		print "Error: !!! previous thing failed\n";
	}
	return $changed;
}

sub dofile {
	my ($cppfile, $hfile) = @_;
	my $lines;
	my $changed;
	$changed = 0;
	print "Processing $cppfile/$hfile\n";
	$lines = OKToRemake($hfile);
	my $cfh;
	open $cfh, "<$cppfile" or die "Could not open $cppfile";
	
	my $function;
	my $list = [];
	while(<$cfh>){
		chomp $_;
		if($_ =~ m/^(.*)\scc\s(.*)$/){
			$function = "$1 $2";
			if($function =~ m/(.*?)\s*\{\s*/){
				$function = $1;
			}
			if($function =~ m/^#/){
				next;
			}
			
			# add to our list
			push @$list, $function;
		}
	}
	
	close($cfh);
	my $hfile_lines;
	$hfile_lines = loadFile($hfile);

	my($i);

	
	my $output = [];
	
	for($i = 0; $i <= $$lines[0]; $i++){
		push @$output, $$hfile_lines[$i];
		#print $hfh $$hfile_lines[$i] . "\n";

	}
	for($i =0; $i < @$list; $i++){
		push @$output, $$list[$i] . ";";
		#print $hfh $$list[$i] . ";\n";
	}
	for($i = $$lines[1]; $i < @$hfile_lines; $i++){
		push @$output, $$hfile_lines[$i];
		#print $hfh $$hfile_lines[$i] . "\n";
	}
	if(!sameFileContents($output, $hfile)){
		saveFile($output, $hfile);
		$changed = 1;
	}
	return $changed;
}

sub getQCDir {
	my ($startdir) = @_;
	$startdir = abspath($startdir);
	while ($startdir && length($startdir) > 0){	
		if(-d "$startdir/qcout"){
			return "$startdir/qcout";
		}
		$startdir = dirname($startdir);
		if($startdir eq "/"){
			last;
		}
	}
	return 0;
}
sub qcFileParse {
	my ($file) = @_;
	return fileparse($file, "\\.[^\\/]+\$");
}
sub doqc {
	my ($file, $outdir) = @_;
	my ($cpp, $h);
	my ($name, $dir, $ext) = qcFileParse($file);
	my $hcode;
	my $cppcode;
	my $code;
	my $changed;
	
	$changed = 0;
	$code = loadFile($file);
	
	$hcode = compileH($code, $file);
	$cppcode = compileCPP($code, $file);
	if(!sameFileContents($hcode, "$outdir/$name.h")){
		saveFile($hcode, "$outdir/$name.h");
		$changed = 1;
	}
	if(!sameFileContents($cppcode, "$outdir/$name.cpp")){
		saveFile($cppcode, "$outdir/$name.cpp");
		$changed = 1;
	}
	
	return $changed;
}

sub PreFile {
	my ($file) = @_;
	
	my ($dir, $name, $cpp, $hfile);
	my $lines;
	my $type;
	my $pass;
	
	$pass = 1;
	if($file =~ m/^(.*)\/([^\.]*)\.(.+)$/){
		$dir = $1;
		$name = $2;
		$type = $3;
		$cpp = "$name.$type";
		$hfile = "$name.h";
	} elsif ($file =~ m/^([^\/\.]+)\.(.+)$/){
		$dir = ".";
		$name = $1;
		$type = $2;
		$cpp = "$name.$type";
		$hfile = "$name.h";
	} else {
		print "Could not match file $file\n";
		$pass = 0;
	}
	return 0 if(!$pass);
	if($type eq "qc" or $type eq "qc.h" or $type eq "qc.cpp"){
		print "Preprocessing qc file $dir/$cpp ...\n";
		$outdir = getQCDir($dir);
		if(!$outdir){
			print "Could not find suitable dir for qc output files";
			return -1;
		}
		return doqc("$dir/$cpp", $outdir);
	}
	$lines = OKToRemake("$dir/$hfile");
	if($lines){
		return dofile("$dir/$cpp", "$dir/$hfile");
	}
	$lines = isQCAll("$dir/$hfile");
	if($lines){
		return qcAll("$dir/$cpp", "$dir/$hfile");
	}
	return 0;
	
}

sub replacements {
	my ($string, $search, $replace) = @_;
	
	my $i;
	for($i = @$search-1; $i >= 0; $i--){
		my $s;
		my $r;
		$s = $$search[$i];
		$r = $$replace[$i];
		my $e;
		$e = "s/$s/$r/g;";
		$e = '$string =~ ' . $e;
		eval $e;
		#$string =~ s/$s/$r/g;
	}
	return $string;
}
sub compileCPP {
	my ($code, $file) = @_;
	my $i;
	my $gen = [];
	my $function_prepend="";
	my $braceLevel;
	my $inClass;
	my $inEnum;
	my $lineNumber;
	my $genSize;
	my $fileName;
	my $dir;
	my $ext;
	my $put;
	
	my $replaceStrings = [];
	my $searchStrings = [];
	my $lineBuffer = [];
	
	($fileName, $dir, $ext) = qcFileParse($file);
	
	$braceLevel = 0;
	$inClass = 0;
	$inEnum = 0;
	$lineNumber = 0;
	push @$gen, "#include \"$fileName" . ".h\"";
	
	my $fullFileName;
	$fullFileName = abspath($file);
	push @$gen, "#" . "line 1 \"$fullFileName\"";
	$genSize = 0;
	for($i = 0; $i < @$code || @$lineBuffer > 0; $i++){
		my $line;
		if(@$lineBuffer > 0){
			$lineNumber = $i;
			$i--;
			push @$gen, "#" . "line $lineNumber \"$fullFileName\"";
			$line = pop @$lineBuffer;
		} else {
			$line = $$code[$i];
			$lineNumber = $i+1;
			$genSize++;
			if(@$gen != $genSize){
				push @$gen, "#" . "line $lineNumber \"$fullFileName\"";
				$genSize = @$gen;
			}
		}
		if(!($line =~ /^\s*\$replace\s+/)){
			$line = replacements($line, $searchStrings, $replaceStrings);
		}
		if($line =~ m/\n/){
			my @array;
			@array = split(/\n/, $line);
			@array = reverse(@array);
			push @$lineBuffer, @array;
			next;
		}
		if(!$inEnum && $line =~ m/^(\s*)if\s(.*)$/){
			$put = $1 . "if ($2) {";
			push @$gen, $put;
		} elsif ($line =~ m/^(\s*)while\s(.*)$/){
			$put = $1 . "while ($2) {";
			push @$gen, $put;
		} elsif ($line =~ m/^(\s*)for\s(.*)$/){
			$put = $1 . "for ($2) {";
			push @$gen, $put;
		} elsif($line =~ m/^(\s*)switch\s+(.*)\s*$/){
			$put = $1 . "switch ($2) {";
			push @$gen, $put;
		} elsif($line =~ m/^(\s*)case\s+(.*):\s*$/ or $line =~ m/^(\s*)case\s+(.*)\s*$/){
			$put = $1 . "case $2:";
			push @$gen, $put;
		} elsif($line =~ m/^(\s*)default\s*$/ or $line =~ m/(\s*)default:\s*$/){
			push @$gen, $1 . "default:";
		} elsif($line =~ m/^(\s*)end\s*$/){
			if($inEnum) {
				$inEnum = 0;
			} else {
				push @$gen, "$1}";
			}
		} elsif(!$inEnum && $line =~ m/^(\s*)else(\s*)$/){
			push @$gen, "$1 } else {"; 
		} elsif(!$inEnum && $line =~ m/^([^#].+)\s+(\S+\s*\(.*\).*?)\{\s*$/){
			# its a function
			my $put;
			$put = "$1 $function_prepend" . "$2 {";
			push @$gen, $put;
			$braceLevel++;
		} elsif($line =~ m/^([^#].+)\s+([^\(\)\s]+)\s*\{\s*$/) {
			# its a function with ()
			my $put;
			$put = "$1 $function_prepend" . "$2 () {";
			push @$gen, $put;
			$braceLevel++;
		} elsif($line =~ m/^\s*\{\s*$/){
			push @$gen, $line;
			$braceLevel++;
		} elsif($line =~ m/^(\s*)\}\s*$/){
			push @$gen, $1 . "}";
			$braceLevel--;
		}  elsif($line =~ m/^\s*class\s+(\S+)\s*;\s*$/){ # class aoeu;
			push @$gen, "";
		} elsif($line =~ m/^\s*class\s+(\S+)\s*:\s*(\S+)$/){
			$function_prepend = $1 . "::";
			$inClass = 1;
		} elsif($line =~ m/^\s*class\s+(\S+)\s*$/){
			$function_prepend = $1 . "::";
			$inClass = 1;
		} elsif($line =~ m/^\s*endclass/){
			# do nothing
			$function_prepend = "";
			$inClass = 0;
		} elsif($line =~ m/^\s*enum\s*$/){
			$inEnum = 1;
		} elsif($line =~ m/^\s*include\s*(".*")\s*$/){
			# for includeing files
			push @$gen, ""
		} elsif($line =~ m/^\s*include\s*(<.*>)\s*$/){
			# for includeing files
			push @$gen, ""
		} elsif($line =~ /^\s*require\s*(".*")\s*$/){
			# include file thats only in the source but not header
			push @$gen, "#include $1";
		} elsif($line =~ /^\$replace\s+(.*?)\s*=>\s*(.*?)\s*$/){
			push @$replaceStrings, $2;
			push @$searchStrings, $1;
		} elsif($line =~ /^#/){
			# header macro
		} elsif($line =~ /^\$(.*)$/){
			# allow pasthrough custom macros
			push @$gen, "#" . $1;
		} elsif($line =~ /^%(.*)$/){
			# only in header
		}
		else {
			if(($line =~ m/^\s*$/))
			{
				push @$gen, "$line";
			} else {
				if(!$inClass){
					if(!$inEnum){
						push @$gen, "$line" . ";"; # add a semicolor :)
					}
				} elsif($braceLevel > 0){
					push @$gen, "$line" . ";";
				}
			}
		}
	}
	return $gen;
}
sub compileH {
	my ($code, $file) = @_;
	my $i;
	my $gen = [];
	my $function_prepend="";
	my $braceLevel;
	my $inClass;
	my $inEnum;
	my $put;
	my $searchStrings=[];
	my $replaceStrings=[];
	my $lineBuffer = [];
	
	my $lineNumber;
	my $genSize;
	
	$braceLevel = 0;
	$inClass = 0;
	$inEnum = 0;
	push @$gen, "#pragma once";
	my $fullFileName;
	$fullFileName = abspath($file);

	$lineNumber = 0;
	$genSize = -1;
	for($i = 0; $i < @$code || @$lineBuffer > 0; $i++){
		my $line;

		if(@$lineBuffer > 0){
			$lineNumber = $i;
			$i--;
			push @$gen, "#" . "line $lineNumber \"$fullFileName\"";
			$line = pop @$lineBuffer;
		} else {
			$line = $$code[$i];
			$lineNumber = $i+1;
			$genSize++;
			if(@$gen != $genSize){
				push @$gen, "#" . "line $lineNumber \"$fullFileName\"";
				$genSize = @$gen;
			}
		}
		
		if(!($line =~ /^\s*\$replace\s+/)){
			$line = replacements($line, $searchStrings, $replaceStrings);
		}
		if($line =~ m/\n/){
			my @array;
			@array = split(/\n/, $line);
			@array = reverse(@array);
			push @$lineBuffer, @array;
			next;
		}
		
		if($line =~ m/^\s*end\s*$/){
			if($inEnum){
				push @$gen, "// enum ended";

				push @$gen, "};";
				$inEnum = 0;
			}
		} elsif($line =~ m/(.*)\s+(\S+\s*\(.*\).*?)\{\s*$/){
			# its a function
			$put = "$1 " . "$2;";
			push @$gen, $put;
			$braceLevel++;
		} elsif($line =~ m/^([^#]*)\s+([^\(\)\s]+)\s*\{\s*$/) {
			# its a function with ()
			$put = "$1 $function_prepend" . "$2 ();";
			push @$gen, $put;
			$braceLevel++;
		} elsif($line =~ m/^\s*class\s+(\S+)\s*;\s*$/){
			$put = "class $1;";
			push @$gen, $put;
		} elsif($line =~ m/^\s*class\s+(\S+)\s*:\s*(\S+)$/){
			$put = "class $1 : public $2 {public: typedef $2 super;";
			push @$gen, $put;
			$inClass = 1;
		} elsif($line =~ m/^\s*class\s+(\S+)\s*$/){
			$put = "class $1 {public:";
			push @$gen, $put;
			$inClass = 1;
		} elsif($line =~ m/^\s*include\s*(".*")\s*$/){
			# for includeing files
			push @$gen, "#include $1"
		} elsif($line =~ m/^\s*include\s*(<.*>)\s*$/){
			# for includeing files
			push @$gen, "#include $1"
		} elsif($line =~ m/^\s*endclass/){
			$function_prepend = "";
			$inClass = 0;
			push @$gen, "};";
		} elsif($line =~ m/^\s*\}\s*$/){
			$braceLevel--;
		} elsif($line =~ m/^\s*enum\s*/){
			$inEnum = 1;
			push @$gen, "enum {";
		} elsif($line =~ /^\$replace\s+(.*?)\s*=>\s*(.*?)\s*$/){
			push @$replaceStrings, $2;
			push @$searchStrings, $1;
		} elsif($line =~ /^#/){
			# allow pasthrough custom macros
			# dont print them in header
			push @$gen, $line;
		} elsif($line =~ /^\$(.*)$/){
			# only in cpp macro
		} elsif($line =~ /^%(.*)$/){
			# only in header
			push @$gen, $1 . ";";
		}
		
		else {
			# nothing to do here in header file
			if($line =~ m/^\s*$/){
			} elsif($inEnum){
				$put = "$line,";
				push @$gen, $put;
			} elsif($inClass && $braceLevel == 0){
				$put = "$line" . ";";
				push @$gen, $put;
			}
		}
	}
	return $gen;
}

my $file;
my $i;
my $filetypes = [];
my $doall=0;
push @$filetypes, "cpp";
push @$filetypes, "qc";
push @$filetypes, "qc.h";
push @$filetypes, "qc.cpp";

for($i = 0; $i < @ARGV; $i++){
	if($ARGV[$i] eq "-f"){
		push @$filetypes, $ARGV[$i+1];
		$i++;
	}
	if($ARGV[$i] eq "--all" or $ARGV[$i] eq "-a"){
		$doall = 1;
	}
	if($ARGV[$i] eq "-e"){
		$errorOnChanges = 1;
	}

}
$file = $ARGV[0];
my $changed;
$changed = 0;
if($doall){
	# do all of em
	my $files;
	my $type_n;
	for($type_n = 0; $type_n < @$filetypes; $type_n++){
		$files = AllFiles($$filetypes[$type_n]);
		my $i;
		for($i = 0; $i < @$files; $i++){
			if(PreFile($$files[$i])){
				print "change is 1 " . $$files[$i] . "\n";
				$changed = 1;
			}
		}
	}
} else {
	$changed = PreFile($file);
}

if($changed && $errorOnChanges){
	print "Files have been modified; giving fake error\n";
	exit(1);
}