1 files changed, 9 insertions, 1 deletions

diff --git a/stuff.go b/stuff.go
index b1e8283..38a68e8 100644
--- a/stuff.go
+++ b/stuff.go
@@ -18,13 +18,21 @@ import (
 )
 
 // scanDebs finds all .deb files and extracts their metadata.
-func scanDebs(root string) ([]DebInfo, error) {
+func scanDebs(root string, count int) ([]DebInfo, error) {
 	var debs []DebInfo
+	var counter int
 	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
 		if err != nil {
 			return err
 		}
 		if !info.IsDir() && strings.HasSuffix(info.Name(), ".deb") {
+			if count == -1 {
+				return nil
+			}
+			counter += 1
+			if counter > count {
+				return nil
+			}
 			log.Printf("  -> Processing %s", path)
 
 			// Get control info