utils/UpdateCMakeLists.pl

   1 #!/usr/bin/env perl
   2 use strict;
   3 use File::Find;
   4 use File::Copy;
   5 use Digest::MD5;
   6
   7 my @fileTypes = ("cpp", "c");
   8 my %dirFiles;
   9 my %dirCMake;
  10
  11 sub GetFiles {
  12   my $dir = shift;
  13   my $x = $dirFiles{$dir};
  14   if (!defined $x) {
  15     $x = [];
  16     $dirFiles{$dir} = $x;
  17   }
  18   return $x;
  19 }
  20
  21 sub ProcessFile {
  22   my $file = $_;
  23   my $dir = $File::Find::dir;
  24   # Record if a CMake file was found.
  25   if ($file eq "CMakeLists.txt") {
  26     $dirCMake{$dir} = $File::Find::name;
  27     return 0;
  28   }
  29   # Grab the extension of the file.
  30   $file =~ /\.([^.]+)$/;
  31   my $ext = $1;
  32   my $files;
  33   foreach my $x (@fileTypes) {
  34     if ($ext eq $x) {
  35       if (!defined $files) {
  36         $files = GetFiles($dir);
  37       }
  38       push @$files, $file;
  39       return 0;
  40     }
  41   }
  42   return 0;
  43 }
  44
  45 sub EmitCMakeList {
  46   my $dir = shift;
  47   my $files = $dirFiles{$dir};
  48
  49   if (!defined $files) {
  50     return;
  51   }
  52
  53   foreach my $file (sort @$files) {
  54     print OUT "  ";
  55     print OUT $file;
  56     print OUT "\n";
  57   }
  58 }
  59
  60 sub UpdateCMake {
  61   my $cmakeList = shift;
  62   my $dir = shift;
  63   my $cmakeListNew = $cmakeList . ".new";
  64   open(IN, $cmakeList);
  65   open(OUT, ">", $cmakeListNew);
  66   my $foundLibrary = 0;
  67
  68   while(<IN>) {
  69     if (!$foundLibrary) {
  70       print OUT $_;
  71       if (/^add_clang_library\(/ || /^add_llvm_library\(/ || /^add_llvm_target\(/
  72           || /^add_executable\(/) {
  73         $foundLibrary = 1;
  74         EmitCMakeList($dir);
  75       }
  76     }
  77     else {
  78       if (/\)/) {
  79         print OUT $_;
  80         $foundLibrary = 0;
  81       }
  82     }
  83   }
  84
  85   close(IN);
  86   close(OUT);
  87
  88   open(FILE, $cmakeList) or
  89     die("Cannot open $cmakeList when computing digest\n");
  90   binmode FILE;
  91   my $digestA = Digest::MD5->new->addfile(*FILE)->hexdigest;
  92   close(FILE);
  93
  94   open(FILE, $cmakeListNew) or
  95     die("Cannot open $cmakeListNew when computing digest\n");
  96   binmode FILE;
  97   my $digestB = Digest::MD5->new->addfile(*FILE)->hexdigest;
  98   close(FILE);
  99
 100   if ($digestA ne $digestB) {
 101     move($cmakeListNew, $cmakeList);
 102     return 1;
 103   }
 104
 105   unlink($cmakeListNew);
 106   return 0;
 107 }
 108
 109 sub UpdateCMakeFiles {
 110   foreach my $dir (sort keys %dirCMake) {
 111     if (UpdateCMake($dirCMake{$dir}, $dir)) {
 112       print "Updated: $dir\n";
 113     }
 114   }
 115 }
 116
 117 find({ wanted => \&ProcessFile, follow => 1 }, '.');
 118 UpdateCMakeFiles();
 119