add GetEntriesByDirectory method to the index (#136)

JunNishimura · Jun 23, 2023 · e3bc6a5 · e3bc6a5
1 parent c3bf589
commit e3bc6a5
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/internal/store/index.go b/internal/store/index.go
@@ -90,6 +90,19 @@ func (idx *Index) GetEntry(path []byte) (int, *Entry, bool) {
 	return newEntryFlag, nil, false
 }
 
+func (idx *Index) GetEntriesByDirectory(dirName string) []*Entry {
+	var entries []*Entry
+
+	dirRegexp := regexp.MustCompile(fmt.Sprintf(`%s\/.+`, dirName))
+	for _, entry := range idx.Entries {
+		if dirRegexp.Match(entry.Path) {
+			entries = append(entries, entry)
+		}
+	}
+
+	return entries
+}
+
 func (idx *Index) IsRegisteredAsDirectory(dirName string) bool {
 	if idx.EntryNum == 0 {
 		return false