view/script/addExtraIncludeGuard.rb

#!/usr/bin/ruby

# 冗長インクルードガードの適用スクリプト

require "fileutils"

# 引数にファイル指定がなければ終了
(print "usage:\n\t" + __FILE__ + " <header files>\n\n"; exit) if ARGV.size == 0


# 冗長インクルードガードの追加
def addIncludeGuard(fname, pre_line)

  # 大文字の箇所で _ を追加する
  macro_name = ''

  # !!! いずれ、それっぽいイテレータに置き換える
  # !!! pack() とかを使うべきか？
  n = fname.size
  for i in 0 .. n-1
    ch = fname[i].chr
    if ch =~ /\./
      macro_name += '_'
    elsif ch =~ /[A-Z]/ && !macro_name.empty?
      macro_name += '_' + ch
    else
      # !!! 大文字にする方法を調べて置き換える
      if ch =~ /[A-Z0-9_]/
        macro_name += ch
      else
        macro_name += (fname[i] + 'A'[0] - 'a'[0]).chr
      end
    end
  end

  # 内容の更新
  if pre_line.match(macro_name)
    return false
  end
  lines = ''
  lines += "#ifndef " + macro_name + "\n"
  lines += "#include \"" + fname + "\"\n"
  lines += "#endif\n"
end

# ファイル毎に冗長インクルードガードの適用を行う
ARGV.each { |file|
  print file + ' ... '

  # 「#include "(.+?)"」に対して、前後に「#ifndef」「#endif」を追加する
  updated_lines = ''            #!< 変更後のファイル内容
  is_updated = false            #!< 変更の必要があれば true

  open (file) { |io|
    pre_line = ''
    while line = io.gets
      case (line.chomp())
      when /\#include "(.+?)"/
        header_file = $1
        ret = addIncludeGuard(header_file, pre_line)
        if ret != false
          updated_lines += ret
          is_updated = true
        else
          updated_lines += line
        end
      else
        updated_lines += line
      end
      pre_line = line
    end
  }
  if is_updated
    # ファイルの置き換え
    replace_file = file + '.bak'
    FileUtils.mv(file, replace_file)
    File.open(file, "a") { |io|
      io.write(updated_lines)
    }
    print "update!\n"

  else
    print "O.K.\n"
  end
}