Ensure old tab state is cleared between iterations of run-benchmark
[WebKit-https.git] / Tools / Scripts / add-include
1 #!/usr/bin/env perl
2
3 # Copyright 2009 Apple Inc. All rights reserved.
4 #
5 # Redistribution and use in source and binary forms, with or without
6 # modification, are permitted provided that the following conditions
7 # are met:
8 # 1. Redistributions of source code must retain the above copyright
9 #    notice, this list of conditions and the following disclaimer.
10 # 2. Redistributions in binary form must reproduce the above copyright
11 #    notice, this list of conditions and the following disclaimer in the
12 #    documentation and/or other materials provided with the distribution.
13 #
14 # THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
15 # EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17 # PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
18 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19 # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20 # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21 # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22 # OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
25
26 # Helper script to add includes to source files.
27
28 use strict;
29 use warnings;
30
31 my $headerPattern = '[\"<][A-Za-z][A-Za-z0-9_/]+(\.h)?[\">]'; # " Make Xcode formatter happy.
32
33 my $headerToAdd = shift @ARGV or die;
34 $headerToAdd =~ /^([A-Za-z][A-Za-z0-9]+)\.h$/ or die "Header to add must be a .h file: $headerToAdd.\n";
35
36 sub includesParagraph;
37
38 FILE: for my $filename (@ARGV) {
39     unless ($filename =~ /(\w+)\.cpp$/) { print STDERR "Command line args must be .cpp files: $filename.\n"; next FILE; }
40
41     my $base = $1;
42
43     my $sawConfig = 0;
44     my $sawSelfInclude = 0;
45
46     my $pastIncludes = 0;
47     my %includes;
48
49     my $beforeIncludes = "";
50     my $afterIncludes = "";
51
52     my $currentCondition = "";
53
54     my $entireFileCondition = "";
55
56     unless (open INPUT, "<", $filename) { print STDERR "File does not exist: $filename\n"; next FILE; }
57     while (my $line = <INPUT>) {
58         if ($line =~ /^\s*#(include|import)\s*($headerPattern)\s*\n/) {
59             my $include = $2;
60             if ($pastIncludes) { print STDERR "Saw more includes after include section in $filename, line $.\n"; next FILE; }
61             if ($include eq "\"config.h\"") {
62                 $sawConfig = 1;
63             } else {
64                 unless ($sawConfig) { print STDERR "First include must be config.h in $filename, line $.\n"; next FILE; }
65                 if ($include eq "\"$base.h\"") {
66                     $sawSelfInclude = 1;
67                 } else {
68                     unless ($sawSelfInclude) { print STDERR "Second include must be $base.h in $filename, line $.\n"; next FILE; }
69                     $includes{$currentCondition}{$include} = 1;
70                 }
71             }
72         } else {
73             if ($sawConfig && !$pastIncludes) {
74                 if ($line =~ /^\s*#\s*if\s+(.+?)\s*$/) {
75                     my $condition = $1;
76                     if (!$sawSelfInclude) {
77                         $entireFileCondition = $1;
78                         next;
79                     }
80                     unless ($currentCondition eq "") { print STDERR "Nested #if in include section in $filename, line $.\n"; next FILE; }
81                     $currentCondition = $condition;
82                     next;
83                 }
84                 if ($line =~ /^\s*#\s*endif\s*$/) {
85                     unless ($currentCondition ne "") { print STDERR "Extra #endif in include section in $filename, line $.\n"; next FILE; }
86                     $currentCondition = "";
87                     next;
88                 }
89             }
90             if (!$sawConfig) {
91                 $beforeIncludes .= $line;
92             } else {
93                 $pastIncludes = 1 if $line !~ /^\s*$/;
94                 if ($pastIncludes) {
95                     unless ($currentCondition eq "") { print STDERR "Unterminated #if in include section in $filename, line $.\n"; next FILE; }
96                     $afterIncludes .= $line;
97                 }
98             }
99         }
100     }
101     close INPUT or die;
102
103     $includes{""}{"\"$headerToAdd\""} = 1;
104
105     $beforeIncludes =~ s/\n+$//;
106     $afterIncludes =~ s/^\n+//;
107
108     my $contents = $beforeIncludes;
109     $contents .= "\n\n#include \"config.h\"\n";
110     $contents .= "\n#if $entireFileCondition\n" if $entireFileCondition ne "";
111     $contents .= "#include \"$base.h\"\n\n";
112     for my $condition (sort keys %includes) {
113         $contents .= "#if $condition\n" unless $condition eq "";
114         $contents .= includesParagraph($includes{$condition});
115         $contents .= "#endif\n" unless $condition eq "";
116         $contents .= "\n";
117     }
118     $contents .= $afterIncludes;
119
120     unless (open OUTPUT, ">", $filename) { print STDERR "Could not open file for writing: $filename\n"; next FILE; };
121     print OUTPUT $contents;
122     close OUTPUT or die;
123 }
124
125 sub includesParagraph()
126 {
127     my ($includes) = @_;
128
129     my $paragraph = "";
130
131     for my $include (sort keys %{$includes}) {
132         $paragraph .= "#include $include\n";
133     }
134
135     return $paragraph;
136 }