loop: new clauses first_then, for_var, collect_plus.

1 files changed, 36 insertions, 0 deletions

diff --git a/testcases-iter b/testcases-iter
index b9b7d22..629420e 100644
--- a/testcases-iter
+++ b/testcases-iter
@@ -332,3 +332,39 @@ BEGIN {
 }'
 :
 min = -3
+--
+16:
+$cppawk '
+#include <iter.h>
+
+
+function index_from(str, ch, start,
+                    sstr, pos)
+{
+  sstr = substr(str, start)
+  pos = index(sstr, ch)
+  return pos ? pos + start - 1 : 0
+}
+
+function sscl(string, char)
+{
+  loop (first_then    (start, 1, end + 1),
+        for_var       (end, index_from(string, char, start)),
+        collect_plus  (out, end ? substr(string, start, end - start)
+                                : substr(string, start)),
+        while      (end))
+  { }
+  return out
+}
+
+BEGIN {
+  print sexp(sscl(",", ","))
+  print sexp(sscl("a,", ","))
+  print sexp(sscl(",a", ","))
+  print sexp(sscl("a,b,c", ","))
+}'
+:
+(nil nil)
+("a" nil)
+(nil "a")
+("a" "b" "c")