fix

CERT-Polska · Oct 1, 2024 · a0f6021 · a0f6021
1 parent 2d3ff4b
commit a0f6021
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/libursa/Query.cpp b/libursa/Query.cpp
@@ -80,10 +80,16 @@ const QString &Query::as_value() const {
     if (type != QueryType::PRIMITIVE) {
         throw std::runtime_error("This query doesn\'t have any value.");
     }
-
     return value;
 }
 
+PrimitiveQuery Query::as_ngram() const {
+    if (type != QueryType::PRIMITIVE) {
+        throw std::runtime_error("This query doesn\'t contain a ngram.");
+    }
+    return ngram;
+}
+
 std::string Query::as_string_repr() const {
     std::string out = "";
     if (value.empty()) {

diff --git a/libursa/Query.h b/libursa/Query.h
@@ -54,6 +54,7 @@ class Query {
     std::vector<Query> &as_queries();
     const QString &as_value() const;
     uint32_t as_count() const;
+    PrimitiveQuery as_ngram() const;
     std::string as_string_repr() const;
     const QueryType &get_type() const;
     bool operator==(const Query &other) const;