[OC Foundation框架

注意：

1.变量名和函数名不要混淆调用

2.不要对文件夹进行文件的操作，没有权限

3.递归调用注意初始化变量

 

//
//  main.m
//  CodeLineCount
//
//  Created by hellovoidworld on 14-11-18.
//  Copyright (c) 2014年 com.hellovoidworld. All rights reserved.
//

#import <Foundation/Foundation.h>

// 计算文件夹或文件内所有代码行数
NSInteger codeLineCount(NSString *path)
{
    NSError *error;
   
    // 单例模式创建 NSFileManager
    NSFileManager *fileManager = [NSFileManager defaultManager];
   
    BOOL isDirectory = NO;
    BOOL isFileExist = [fileManager fileExistsAtPath:path isDirectory:&isDirectory];
   
    // 文件是否存在
    if (!isFileExist)
    {
        NSLog(@"%@-->文件或文件夹不存在!", path);
        return 0;
    }
   
    // 如果是文件夹，进行递归调用
    if (isDirectory)
    {
//        NSLog(@"文件夹: %@", path);
       
        NSInteger lineCount = 0; // 代码行数
       
        // 获取文件夹下的所有内容，包括子文件夹和文件
        NSArray *subPaths = [fileManager contentsOfDirectoryAtPath:path error:&error];
       
        if (error != nil)
        {
            NSLog(@"Fail to read the directory, the error is %@", error);
        }
       
        for (NSString *subPath in subPaths)
        {
            NSString *fullSubPath; // 全路径的文件名
            fullSubPath = [NSString stringWithFormat:@"%@/%@", path, subPath]; // 取出来的文件名不带路径
            lineCount += codeLineCount(fullSubPath);
        }
       
        return lineCount;
    }
    else
    {
        NSInteger lineCount = 0; // 代码行数
       
        // 取得文件扩展名
        NSString *fileExtension = [[path pathExtension] lowercaseString];
       
        // 过滤非代码文件
        if (![fileExtension isEqualToString:@"h"]
            && ![fileExtension isEqualToString:@"m"]
            && ![fileExtension isEqualToString:@"c"])
        {
            return 0;
        }
       
        NSString *fileContent = [NSString stringWithContentsOfFile:path encoding:NSUTF8StringEncoding error:&error];
       
        if (error != nil)
        {
            NSLog(@"Read fail, error is %@", error);
        }
       
        NSArray *codeLinesArray = [fileContent componentsSeparatedByString:@"
"];
        lineCount = [codeLinesArray count];
       
        NSLog(@"%@的代码行数是%ld", path, lineCount);
        return lineCount;
    }
}

int main(int argc, const char * argv[]) {
    @autoreleasepool {
       
        NSInteger lineCount = codeLineCount(@"/Users/hellovoidworld/Study");
        NSInteger lineCount2 = codeLineCount(@"/Users/hellovoidworld/Desktop/oc");
       
        NSLog(@"所有源码文件的总行数是%ld", lineCount + lineCount2);
       
    }
   
    return 0;
}