Skip to content

Commit a738ebc

Browse files
committed
feat(kdev): Choose algo
1 parent d174bb9 commit a738ebc

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

cmd/kdev/main.go

+12-3
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ var sortBy = flag.String("sort", "date", "Sort by date or random")
3232
var maxRecords = flag.Int("max", 25, "Maximum number of records to display")
3333
var repoPath = flag.String("repo", ".", "Path to the repository")
3434
var afterS = flag.String("after", "2022-09-01", "Only show records after this date")
35+
var algo = flag.String("algo", "git", "Record extraction algorithm (git, go-git, stat)")
3536

3637
func main() {
3738
var err error
@@ -111,9 +112,17 @@ func processFile(repoPath string, relativePath string, keywords []string, record
111112
return nil
112113
}
113114

114-
recordExtractor := NewStatRecordExtractor(absolutePath)
115-
// recordExtractor = NewGoGitRecordExtractor(head, relativePath)
116-
// recordExtractor := &GitRecordExtractor{repoPath: repoPath}
115+
var recordExtractor RecordExtractor
116+
switch *algo {
117+
case "git":
118+
recordExtractor = &GitRecordExtractor{repoPath: repoPath}
119+
case "go-git":
120+
recordExtractor = NewGoGitRecordExtractor(head, relativePath)
121+
case "stat":
122+
recordExtractor = NewStatRecordExtractor(absolutePath)
123+
default:
124+
kcore.Assert(false, "wrong algo value (git, go-git, stat)")
125+
}
117126
if recordExtractor == nil {
118127
return nil
119128
}

0 commit comments

Comments
 (0)