[OC Foundation框架 - 20] 统计代码行数
注意：
1.变量名和函数名不要混淆调用
2.不要对文件夹进行文件的操作，没有权限
3.递归调用注意初始化变量
1 // 2 // main.m 3 // CodeLineCount 4 // 5 // Created by hellovoidworld on 14-11-18. 6 // Copyright (c) 2014年 com.hellovoidworld. All rights reserved. 7 // 8 9 #import <Foundation/Foundation.h> 10 11 // 计算文件夹或文件内所有代码行数 12 NSInteger codeLineCount(NSString *path) 13 { 14 NSError *error; 15 16 // 单例模式创建 NSFileManager 17 NSFileManager *fileManager = [NSFileManager defaultManager]; 18 19 BOOL isDirectory = NO; 20 BOOL isFileExist = [fileManager fileExistsAtPath:path isDirectory:&isDirectory]; 21 22 // 文件是否存在 23 if (!isFileExist) 24 { 25 NSLog(@"%@-->文件或文件夹不存在!", path); 26 return 0; 27 } 28 29 // 如果是文件夹，进行递归调用 30 if (isDirectory) 31 { 32 // NSLog(@"文件夹: %@", path); 33 34 NSInteger lineCount = 0; // 代码行数 35 36 // 获取文件夹下的所有内容，包括子文件夹和文件 37 NSArray *subPaths = [fileManager contentsOfDirectoryAtPath:path error:&error]; 38 39 if (error != nil) 40 { 41 NSLog(@"Fail to read the directory, the error is %@", error); 42 } 43 44 for (NSString *subPath in subPaths) 45 { 46 NSString *fullSubPath; // 全路径的文件名 47 fullSubPath = [NSString stringWithFormat:@"%@/%@", path, subPath]; // 取出来的文件名不带路径 48 lineCount += codeLineCount(fullSubPath); 49 } 50 51 return lineCount; 52 } 53 else 54 { 55 NSInteger lineCount = 0; // 代码行数 56 57 // 取得文件扩展名 58 NSString *fileExtension = [[path pathExtension] lowercaseString]; 59 60 // 过滤非代码文件 61 if (![fileExtension isEqualToString:@"h"] 62 && ![fileExtension isEqualToString:@"m"] 63 && ![fileExtension isEqualToString:@"c"]) 64 { 65 return 0; 66 } 67 68 NSString *fileContent = [NSString stringWithContentsOfFile:path encoding:NSUTF8StringEncoding error:&error]; 69 70 if (error != nil) 71 { 72 NSLog(@"Read fail, error is %@", error); 73 } 74 75 NSArray *codeLinesArray = [fileContent componentsSeparatedByString:@"\n"]; 76 lineCount = [codeLinesArray count]; 77 78 NSLog(@"%@的代码行数是%ld", path, lineCount); 79 return lineCount; 80 } 81 } 82 83 int main(int argc, const char * argv[]) { 84 @autoreleasepool { 85 86 NSInteger lineCount = codeLineCount(@"/Users/hellovoidworld/Study"); 87 NSInteger lineCount2 = codeLineCount(@"/Users/hellovoidworld/Desktop/oc"); 88 89 NSLog(@"所有源码文件的总行数是%ld", lineCount + lineCount2); 90 91 } 92 93 return 0; 94 }
