blob: d1256b6fa16ba3082d36b4188df03942a8dae479 [file] [log] [blame]
#!/usr/bin/env perl
# Copyright 2009 Apple Inc. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
# OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
# Helper script to add includes to source files.
use strict;
use warnings;
my $headerPattern = '[\"<][A-Za-z][A-Za-z0-9_/]+(\.h)?[\">]'; # " Make Xcode formatter happy.
my $headerToAdd = shift @ARGV or die;
$headerToAdd =~ /^([A-Za-z][A-Za-z0-9]+)\.h$/ or die "Header to add must be a .h file: $headerToAdd.\n";
sub includesParagraph;
FILE: for my $filename (@ARGV) {
unless ($filename =~ /(\w+)\.cpp$/) { print STDERR "Command line args must be .cpp files: $filename.\n"; next FILE; }
my $base = $1;
my $sawConfig = 0;
my $sawSelfInclude = 0;
my $pastIncludes = 0;
my %includes;
my $beforeIncludes = "";
my $afterIncludes = "";
my $currentCondition = "";
my $entireFileCondition = "";
unless (open INPUT, "<", $filename) { print STDERR "File does not exist: $filename\n"; next FILE; }
while (my $line = <INPUT>) {
if ($line =~ /^\s*#(include|import)\s*($headerPattern)\s*\n/) {
my $include = $2;
if ($pastIncludes) { print STDERR "Saw more includes after include section in $filename, line $.\n"; next FILE; }
if ($include eq "\"config.h\"") {
$sawConfig = 1;
} else {
unless ($sawConfig) { print STDERR "First include must be config.h in $filename, line $.\n"; next FILE; }
if ($include eq "\"$base.h\"") {
$sawSelfInclude = 1;
} else {
unless ($sawSelfInclude) { print STDERR "Second include must be $base.h in $filename, line $.\n"; next FILE; }
$includes{$currentCondition}{$include} = 1;
}
}
} else {
if ($sawConfig && !$pastIncludes) {
if ($line =~ /^\s*#\s*if\s+(.+?)\s*$/) {
my $condition = $1;
if (!$sawSelfInclude) {
$entireFileCondition = $1;
next;
}
unless ($currentCondition eq "") { print STDERR "Nested #if in include section in $filename, line $.\n"; next FILE; }
$currentCondition = $condition;
next;
}
if ($line =~ /^\s*#\s*endif\s*$/) {
unless ($currentCondition ne "") { print STDERR "Extra #endif in include section in $filename, line $.\n"; next FILE; }
$currentCondition = "";
next;
}
}
if (!$sawConfig) {
$beforeIncludes .= $line;
} else {
$pastIncludes = 1 if $line !~ /^\s*$/;
if ($pastIncludes) {
unless ($currentCondition eq "") { print STDERR "Unterminated #if in include section in $filename, line $.\n"; next FILE; }
$afterIncludes .= $line;
}
}
}
}
close INPUT or die;
$includes{""}{"\"$headerToAdd\""} = 1;
$beforeIncludes =~ s/\n+$//;
$afterIncludes =~ s/^\n+//;
my $contents = $beforeIncludes;
$contents .= "\n\n#include \"config.h\"\n";
$contents .= "\n#if $entireFileCondition\n" if $entireFileCondition ne "";
$contents .= "#include \"$base.h\"\n\n";
for my $condition (sort keys %includes) {
$contents .= "#if $condition\n" unless $condition eq "";
$contents .= includesParagraph($includes{$condition});
$contents .= "#endif\n" unless $condition eq "";
$contents .= "\n";
}
$contents .= $afterIncludes;
unless (open OUTPUT, ">", $filename) { print STDERR "Could not open file for writing: $filename\n"; next FILE; };
print OUTPUT $contents;
close OUTPUT or die;
}
sub includesParagraph()
{
my ($includes) = @_;
my $paragraph = "";
for my $include (sort keys %{$includes}) {
$paragraph .= "#include $include\n";
}
return $paragraph;
}