package com.njcn.utils;

import com.njcn.serviceimpl.weeklyreport.WeeklyReportDataServiceImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.util.ArrayList;
import java.util.List;

/**
 * **********************************************
 * @description 计算源代码（src）行数，不计算空行
 *     宗旨：将src下所有文件组装成list,再筛选出文件，对文件进行遍历读取
 * @version 1.0
 ***********************************************
 */
public class LineCounter {
    private static final Logger logger = LoggerFactory.getLogger(LineCounter.class);
    List<File> list = new ArrayList<File>();
    int linenumber = 0;

    FileReader fr = null;
    BufferedReader br = null;

    public void counter(String projectName) {
//        String path = System.getProperty("user.dir");
        String path = LineCounter.class.getResource("/").getPath();  // 同下个path
        path = path.substring(0, path.length() - 24) + projectName + "/src";
        File file = new File(path);
        File files[] = null;
        files = file.listFiles();
        addFile(files);
        isDirectory(files);
        readLinePerFile();
    }

    // 判断是否是目录
    public void isDirectory(File[] files) {
        for (File s : files) {
            if (s.isDirectory()) {
                File file[] = s.listFiles();
                addFile(file);
                isDirectory(file);
                continue;
            }
        }
    }

    //将src下所有文件组织成list
    public void addFile(File file[]) {
        for (int index = 0; index < file.length; index++) {
            list.add(file[index]);
        }
    }

    //读取非空白行
    public void readLinePerFile() {
        try {
            for (File s : list) {
                int yuan = linenumber;
                if (s.isDirectory()) {
                    continue;
                }
                fr = new FileReader(s);
                br = new BufferedReader(fr);
                String i = "";
                while ((i = br.readLine()) != null) {
                    if (isBlankLine(i))
                        linenumber++;
                }
            }
        } catch (Exception e) {
            logger.error(e.getMessage());
        } finally {
            if (br != null) {
                try {
                    br.close();
                } catch (Exception e) {
                }
            }
            if (fr != null) {
                try {
                    fr.close();
                } catch (Exception e) {
                }
            }
        }
    }

    //是否是空行
    public boolean isBlankLine(String i) {
        if (i.trim().length() == 0) {
            return false;
        } else {
            return true;
        }
    }

    public static void main(String args[]) {
        LineCounter lc = new LineCounter();
        String projectName = "/G:/JavaProjectCode/trunk/system/pqs9000";     //这里传入你的项目名称
        lc.counter(projectName);
    }
}
