计算代码行数

//计算代码行数
#import <Foundation/Foundation.h>

//计算单个文件的代码行数

/*
用到的内容:
文件管理类
1,NSFileManger:单例
1)路径是否存在,是否是文件夹:- (BOOL)fileExistsAtPath:(NSString *)path isDirectory:(BOOL *)isDirectory
2)返回数组,元素为指定路径下的文件或者文件夹名称:- (NSArray *)contentsOfDirectoryAtPath:(NSString *)path error:(NSError **)error

2,NSString
1)将通过分隔符分隔字符串返回NSArray:- (NSArray *)componentsSeparatedByString:(NSString *)separator
2)获取文件中的内容:+ (instancetype)stringWithContentsOfURL:(NSURL *)url encoding:(NSStringEncoding)enc error:(NSError **)error
*/

/*
path:文件的全路径,可能是文件夹,也可能是文件
返回值int: 返回代码的行数
*/
int codeLineCount(NSString *path)
{
int count = 0 ;
//1,获得文件管理者,为单例
NSFileManager *mgr = [NSFileManager defaultManager];

//2,判断path是文件夹还是文件路径
BOOL dir = NO;//标记是否问文件夹
//返回值有两个:1)isExist这个路径是否存在,2)如果存在是否是文件夹,是则修改dir的值为YES
BOOL isExist = [mgr fileExistsAtPath:path isDirectory:&dir];

//3,如果路径不存在,直接返回0
if(!isExist)
{
NSLog(@"路径不存在");
return 0;
}

//4,文件路径存在
//如果是文件夹
if(dir)
{
NSLog(@"是一个文件夹");
//装着当前文件夹下面的所有内容(文件夹和文件)
//返回数组为文件名和文件夹名
NSArray *array = [mgr contentsOfDirectoryAtPath:path error:nil];
//NSLog(@"%@",array);
//遍历该文件夹下得所有文件夹和文件
for(NSString *fileName in array)
{
//拼接获得当前文件夹下文件或文件夹的全路径
NSString *fullPath = [NSString stringWithFormat:@"%@%@",path,fileName];
count += codeLineCount(fullPath);
}
return count;
}
else
{

//判断文件的扩展名(忽略大小写,都转为小写)

NSString *extension = [[path pathExtension]lowercaseString];
//如果文件扩展名不是.c或者.m或者.h,则退出
if(![extension isEqualToString:@"h"] &&
![extension isEqualToString:@"m"] &&
![extension isEqualToString:@"c"])
return 0;

NSLog(@"是一个文件");
//加载文件内容,每一行通过'\n'分隔
NSURL *url = [NSURL fileURLWithPath:path];
NSString *contents =  [NSString stringWithContentsOfURL:url encoding:NSUTF8StringEncoding error:nil];
//通过分隔符'\n',将文件切割为一个数组,数组中存放的是每一行的内容
NSArray *array = [contents componentsSeparatedByString:@"\n"];
//遍历每一行数据
for(NSString *str in array)
NSLog(@"%@\n",str);
return array.count;
}
}
int main(int argc, const char * argv[]) {
@autoreleasepool {

int count = codeLineCount(@"/Users/mac/Desktop/11/");
NSLog(@"%d",count);
}
return 0;
}