include screenshot counts in GET/ summary (#60)

db/file-summary.n1ql

@@ -32,6 +32,7 @@ SELECT
     },
     "pe_meta": f.pe.meta,
     "default_behavior_id": f.default_behavior_id,
+    "screenshots_count": f.screenshots_count,
     "liked": CASE WHEN ARRAY_LENGTH(user_likes) = 0 THEN false
               ELSE ARRAY_BINARY_SEARCH(ARRAY_SORT((user_likes)[0]), f.sha256) >= 0 END
   }.*

internal/behavior/repostitory.go

@@ -163,9 +163,28 @@ func (r repository) CountArtifacts(ctx context.Context, id string) (int, error)
 	var statement string
 	params := make(map[string]interface{}, 1)
 	params["id"] = id
-	statement =
+
+	filters, ok := ctx.Value(filtersKey).(map[string][]string)
+	if ok {
+		statement =
+			"SELECT RAW COUNT(artifacts) AS count FROM `" + r.db.Bucket.Name() + "` d" +
+				" USE KEYS $id UNNEST d.artifacts as artifacts"
+		i := 0
+		for k, v := range filters {
+			if i == 0 {
+				statement += " WHERE"
+			} else {
+				statement += " AND"
+			}
+			i++
+			statement += fmt.Sprintf(" artifacts.%s IN $%s", k, k)
+			params[k] = v
+		}
+	} else {
+		statement =
 			"SELECT RAW ARRAY_LENGTH(d.artifacts) AS count FROM `" + r.db.Bucket.Name() + "` d" +
 				" USE KEYS $id"
+	}
 
 	err := r.db.Count(ctx, statement, params, &count)
 	return count, err