lib/itools/count_code_line.rb in itools-0.4.4 vs lib/itools/count_code_line.rb in itools-0.4.5

@@ -13,11 +13,11 @@
                 File.read(@file_path).each_line do |line|
                     if line.match(/^\/\/|^$/) == nil   #去掉单行注释和空行
                         @line_number = @line_number + 1  
                     end
                 end
-                puts "\033[32m统计#{counter.file_path}结束，共#{counter.line_number}行\033[0m"
+              
                 return
             end
             if File::directory?(@file_path)
                 Find.find(@file_path) do |file|
                     if File.file?(file) #判断是否是文件
@@ -28,22 +28,22 @@
                                 end
                             end
                         end
                     end
                 end
-                puts "\033[32m统计#{counter.file_path}结束，共#{counter.line_number}行\033[0m"
                 return
             end
             puts "\033[31m找不到指定路径的文件或者文件夹，请重新输入路径\033[0m"
         end
 
-        def self.counter(args)
+        def self.count_line(args)
             file = args[0]
             if file.nil?
                 puts "\033[31m参数异常，请传入一个参数(项目目录/要统计的文件目录/要统计的文件)\033[0m"
                 return
             end
             counter = CodeCouner.new(file)
             counter.calculate_line_number
+            puts "\033[32m统计#{counter.file_path}结束，共#{counter.line_number}行\033[0m"
         end
     end
 end
\ No newline at end of file