msoucy/git-by-a-truck
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Prepare executors

Browse Source
This commit is contained in:
Matt Soucy 2020-06-20 17:22:38 -04:00
parent 42f9f66793
commit 214fd22b8b
1 changed files with 29 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/main/kotlin/me/msoucy/gbat/Main.kt
View File

@ -2,8 +2,10 @@ package me.msoucy.gbat
import java.io.File import java.io.File
import java.io.IOException import java.io.IOException
import java.util.concurrent.Executors
import java.util.concurrent.TimeUnit import java.util.concurrent.TimeUnit
import kotlin.math.pow import kotlin.math.pow
import kotlin.system.exitProcess
import kotlin.text.Regex import kotlin.text.Regex
import kotlin.text.RegexOption import kotlin.text.RegexOption
import kotlin.text.startsWith import kotlin.text.startsWith
@ -110,12 +112,9 @@ class GitRepo(val project_root : File, val git_exe : String) {
fun main(args: Array<String>) = mainBody { fun main(args: Array<String>) = mainBody {
ArgParser(args).parseInto(::GbatArgs).run { ArgParser(args).parseInto(::GbatArgs).run {
println("Hello world")
println(validateGit(git_exe))
val outDir = File(output) val outDir = File(output)
if(outDir.isDirectory) { if(outDir.isDirectory) {
throw InvalidArgumentException("Output directory already exists") //throw InvalidArgumentException("Output directory already exists")
} }
outDir.mkdirs() outDir.mkdirs()
@ -130,23 +129,38 @@ fun main(args: Array<String>) = mainBody {
} }
val risk_thresh = risk_threshold ?: default_bus_risk.pow(3) val risk_thresh = risk_threshold ?: default_bus_risk.pow(3)
val interesting_res = if (interesting.isEmpty()) DEFAULT_INTERESTING_RES else { val interesting_res = parse_interesting(if (interesting.isEmpty()) DEFAULT_INTERESTING_RES else interesting)
parse_interesting(interesting) val not_interesting_res = if (not_interesting.isEmpty()) listOf() else parse_interesting(not_interesting)
}
val not_interesting_res = if (not_interesting.isEmpty()) listOf() else {
parse_interesting(not_interesting)
}
val project_root_file = File(project_root).also { val project_root_file = File(project_root).also {
if(!it.isDirectory) if(!it.isDirectory)
throw InvalidArgumentException("Provided project root does not exist") throw InvalidArgumentException("Provided project root does not exist")
} }
println(risk_thresh)
println(interesting_res)
println(not_interesting_res)
println(project_root_file)
val repo = GitRepo(project_root_file, validateGit(git_exe)) val repo = GitRepo(project_root_file, validateGit(git_exe))
println(repo.ls())
fun String.isInteresting() : Boolean {
var hasInterest = interesting_res.any { it.containsMatchIn(this) }
if(hasInterest) {
hasInterest = !not_interesting_res.any { it.containsMatchIn(this) }
}
return hasInterest
}
fun GitRepo.interestingNames() = ls().split("\n").filter{ it.isInteresting() }
val fnames = repo.interestingNames()
if (fnames.isEmpty()) {
System.err.println("No interesting files found, exiting.")
exitProcess(1)
}
if (verbose) {
System.err.println("Found ${fnames.size} interesting files")
}
val pool = Executors.newFixedThreadPool(num_analyzer_procs + num_git_procs + 1)
val summ_result = mutableListOf<Int>()
} }
} }