From: Nicolas Williams <nico@cryptonector.com>
Date: Mon, 2 Dec 2013 18:16:38 +0000 (-0600)
Subject: Add index strings by string; return string indexes
X-Git-Tag: jq-1.4~76^2~4
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=cac14a531de733cbffe50eb4e6c479b5ef577e21;p=jq

Add index strings by string; return string indexes

    % jq '.[","]'
    "a,bc,def,ghij,klmno"
    [1,4,8,13]
    %
---

diff --git a/jv.c b/jv.c
index b66b073..0c0bbb7 100644
--- a/jv.c
+++ b/jv.c
@@ -621,6 +621,26 @@ not_this:
 }
 #endif /* HAVE_MEMMEM */
 
+jv jv_string_indexes(jv j, jv k) {
+  assert(jv_get_kind(j) == JV_KIND_STRING);
+  assert(jv_get_kind(k) == JV_KIND_STRING);
+  const char *jstr = jv_string_value(j);
+  const char *idxstr = jv_string_value(k);
+  const char *p;
+  int jlen = jv_string_length_bytes(jv_copy(j));
+  int idxlen = jv_string_length_bytes(jv_copy(k));
+  jv a = jv_array();
+
+  p = jstr;
+  while ((p = memmem(p, (jstr + jlen) - p, idxstr, idxlen)) != NULL) {
+    a = jv_array_append(a, jv_number(p - jstr));
+    p += idxlen;
+  }
+  jv_free(j);
+  jv_free(k);
+  return a;
+}
+
 jv jv_string_split(jv j, jv sep) {
   assert(jv_get_kind(j) == JV_KIND_STRING);
   assert(jv_get_kind(sep) == JV_KIND_STRING);
diff --git a/jv.h b/jv.h
index 6b06eea..efc6fec 100644
--- a/jv.h
+++ b/jv.h
@@ -83,6 +83,7 @@ int jv_string_length_bytes(jv);
 int jv_string_length_codepoints(jv);
 unsigned long jv_string_hash(jv);
 const char* jv_string_value(jv);
+jv jv_string_indexes(jv j, jv k);
 jv jv_string_slice(jv j, int start, int end);
 jv jv_string_concat(jv, jv);
 jv jv_string_vfmt(const char*, va_list);
diff --git a/jv_aux.c b/jv_aux.c
index f3260c9..76712ae 100644
--- a/jv_aux.c
+++ b/jv_aux.c
@@ -73,6 +73,8 @@ jv jv_get(jv t, jv k) {
       v = jv_invalid_with_msg(jv_string_fmt("Start and end indices of an string slice must be numbers"));
       jv_free(t);
     }
+  } else if (jv_get_kind(t) == JV_KIND_STRING && jv_get_kind(k) == JV_KIND_STRING) {
+    v = jv_string_indexes(t, k);
   } else if (jv_get_kind(t) == JV_KIND_NULL && 
              (jv_get_kind(k) == JV_KIND_STRING || 
               jv_get_kind(k) == JV_KIND_NUMBER ||