Class: Itools::CodeCouner

Inherits:
Object
  • Object
show all
Defined in:
lib/itools/count_code_line.rb

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(path) ⇒ CodeCouner

Returns a new instance of CodeCouner.



5
6
7
8
 
# File 'lib/itools/count_code_line.rb', line 5

def initialize(path)
    @file_path = path
    @line_number = 0
end

Instance Attribute Details

#file_path ⇒ Object

Returns the value of attribute file_path.



4
5
6
 
# File 'lib/itools/count_code_line.rb', line 4

def file_path
  @file_path
end

#line_number ⇒ Object

Returns the value of attribute line_number.



4
5
6
 
# File 'lib/itools/count_code_line.rb', line 4

def line_number
  @line_number
end

Class Method Details

.count_line(args) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
 
# File 'lib/itools/count_code_line.rb', line 38

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

Instance Method Details

#calculate_line_number ⇒ Object

统计行数



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/itools/count_code_line.rb', line 10

def calculate_line_number    
    puts "\033[32m正在统计，请稍后...\033[0m"
    if File.file?(@file_path)
        File.read(@file_path).each_line do |line|
            if line.match(/^\/\/|^$/) == nil   #去掉单行注释和空行
                @line_number = @line_number + 1  
            end
        end
      
        return
    end
    if File::directory?(@file_path)
        Find.find(@file_path) do |file|
            if File.file?(file) #判断是否是文件
                if File.extname(file).match(/^.[hm]m?$|.cpp/)  #只统计.h/.m/.mm/.cpp几个文件
                    File.read(file).each_line do |line|
                        if line.match(/^\/\/|^$/) == nil   #去掉单行注释和空行 
                            @line_number = @line_number + 1  
                        end
                    end
                end
            end
        end
        return
    end
    puts "\033[31m找不到指定路径的文件或者文件夹，请重新输入路径\033[0m"
end