diff --git a/db/file-summary.n1ql b/db/file-summary.n1ql index 157fd49..e66f8ec 100644 --- a/db/file-summary.n1ql +++ b/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 }.* diff --git a/internal/behavior/repostitory.go b/internal/behavior/repostitory.go index 5df6a76..573d0c0 100644 --- a/internal/behavior/repostitory.go +++ b/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