统计同一个项目在不同时期的代码新增情况_Ruby_编程开发_程序员俱乐部

# encoding: utf-8
module ProjectCodeStatistics
   module FetchFile
      def recursion_fetch(dir=nil)
         dir = @base_dir if dir == nil
         Dir.foreach(dir).each do |file_dir|
           dir_path = File.join(dir,file_dir)
           if  File.directory?(dir_path)  then
               recursion_fetch(dir_path) unless file_dir =~ /^\./
           else
              if [/\.java$/,/\.jsp$/,/\.js$/,/\.xml$/].any?{|reg| reg=~file_dir}
                @all_file_with_row_number[dir_path.gsub(@base_dir,"")] = File.new(dir_path).readlines.size
              end
           end
         end
      end
   end

   class Calculate
      include FetchFile
      attr_reader :all_file_with_row_number

      def initialize(dir)
         @base_dir, @all_file_with_row_number = dir, {}
         recursion_fetch
      end
         # 分类统计所有文件数量
      def file_total_numbers()
          row_numbers(:xml,:js,:jsp,:java,@all_file_with_row_number)
      end

      # 分类统计新加入的文件代码行总量的
      def add_file_row_numbers(other)
        add_file_keys = @all_file_with_row_number.keys - other.all_file_with_row_number.keys
        add_file_hash = @all_file_with_row_number.find_all{|key,value| add_file_keys.include? key}
        row_numbers(:xml,:js,:jsp,:java,add_file_hash)
      end

      # 分类统计新加入的各种类型文件总量
      def compare_add_file_count(other)
         add_file_keys = @all_file_with_row_number.keys - other.all_file_with_row_number.keys
         file_numbers(:xml,:js,:jsp,:java,add_file_keys)
      end

      # 分类统计修改过的文件
      def compare_each_modify_file(other)
         file_hash = @all_file_with_row_number
         calculated_hash =other.all_file_with_row_number.find_all.inject({}) do |hash,item|
             hash[item[0]] = (file_hash[item[0]] - item[1]) if file_hash.include?(item[0]) && file_hash[item[0]] != item[1]
             hash
         end
         row_numbers(:xml,:js,:jsp,:java,calculated_hash)
      end

      def file_numbers(*parameter)
         last = parameter.reverse!.shift
         parameter.reverse.inject([]) do |total,type|
            regexp  =  eval('/' + type.to_s + '$/')
            total.push(last.find_all{|key,value| regexp =~ key}.size)
         end
      end

      def row_numbers(*parameter)
         last = parameter.reverse!.shift
         parameter.reverse.inject([]) do |total,type|
            regexp  =  eval('/' + type.to_s + '$/')
            total.push last.find_all{|key,value| regexp =~ key}.inject(0){|sum,item| sum + item[1] }
         end
      end
   end

   class Statistics
      def initialize(new_project,old_project,new_desc="新工程",old_desc="老工程")
         @new_project = Calculate.new(new_project)
         @old_project = Calculate.new(old_project)
         @new_desc,@old_desc = new_desc,old_desc
      end

      def print
         old_total = @old_project.file_total_numbers
         new_total = @new_project.file_total_numbers
         puts '系统代码统计'
         puts "#{@old_desc}代码总量统计，总计：#{old_total.reduce(:+)}行， java：#{old_total[3]}行,jsp: #{old_total[2]}行,js: #{old_total[1]}行,sql: #{old_total[0]}行 "
         puts "#{@new_desc}代码总量统计，总计：#{new_total.reduce(:+)}行，java：#{new_total[3]}行,jsp: #{new_total[2]}行,js: #{new_total[1]}行,sql: #{new_total[0]}行 "
         puts "新增总量 #{new_total.reduce(:+) - old_total.reduce(:+)}行"

         add_files = @new_project.compare_add_file_count(@old_project)
         new_rows = @new_project.add_file_row_numbers(@old_project)
         modify_rows = @new_project.compare_each_modify_file(@old_project)
         total_change = (new_rows.zip  modify_rows).inject([]) {|n,a|n<<a.reduce(:+)}
         puts
         puts  "新增加的文件数量，java：#{add_files[3]}个，jsp: #{add_files[2]}个，js: #{add_files[1]}个， sql: #{add_files[0]}个 "
         puts
         puts  "新增加的文件代码行，java：#{new_rows[3]}行，jsp: #{new_rows[2]}行，js: #{new_rows[1]}行， sql: #{new_rows[0]}行 "
         puts  "新增加的代码行，    java：#{modify_rows[3]}行，jsp: #{modify_rows[2]}行，js: #{modify_rows[1]}行， sql: #{modify_rows[0]}行 "
         puts  "新增加的代码行总计，java：#{total_change[3]}行，jsp: #{total_change[2]}行，js: #{total_change[1]}行， sql: #{total_change[0]}行 "
         puts  "新增总量 #{new_rows.reduce(:+) + modify_rows.reduce(:+)}行"
      end
   end
end

s = ProjectCodeStatistics::Statistics.new('D:\boss\Code\udrm\src','D:\boss\Code\digital20110726\src')
s.print