Scala-抓取项目代码中全部代码注释-蒲公英云

Scala-抓取项目代码中全部代码注释

Myth丶恋晨 2022-01-31 20:47 282阅读 0赞

日前某站代码泄露, 大佬们贴出很多代码中有趣的注释, 于是突发奇想, 写了一个没有用的小程序…

(ps: 某站代码我没看, 你们可, 别瞎说…)

import scala.io.Source
import java.util.regex.Pattern
import java.io.{File, PrintWriter}
object Bi {
  def getFile(file:File): Array[File] ={
    val files = file.listFiles()
                    .filter(! _.isDirectory)
                    // 指定要读取什么文件
                    .filter(t => t.toString.endsWith(".go"))
    files ++ file.listFiles()
                  .filter(f => f.isDirectory
                    // 指定排除掉什么文件夹
                    && f.getName != "vendor")
                  .flatMap(getFile)
  }
  def check(s: String): Boolean = {
    val pattern = "[\u4e00-\u9fa5]+"
    val p = Pattern.compile(pattern)
    val result = p.matcher(s)
    if (result.find()) true else false
  }
  def main(args: Array[String]): Unit = {
    val path = new File("目标文件夹路径")
    val writer = new PrintWriter(new File("输出文件路径"))
    getFile(path).foreach{ file =>
      var flag = false
      val text=Source.fromFile(file)
      for(line <- text.getLines)
      {
        if (line.trim.startsWith("//") && check(line)) {
          flag = true
          writer.println(s"-----$line")
        }
      }
      if (flag) {
        // 打印以上注释出自哪个文件
        writer.println(file.getPath)
      }
      text.close
    }
    writer.close()
  }
}