From 2a610da2d14583a08f016d0f985db65203d1f76f Mon Sep 17 00:00:00 2001
From: Stephen Dolan <mu@netsoc.tcd.ie>
Date: Mon, 22 Oct 2012 23:31:07 +0100
Subject: [PATCH] Allow the 'keys' function to take arrays.

---
 builtin.c                        | 6 ++++++
 docs/content/3.manual/manual.yml | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/builtin.c b/builtin.c
index c3d6af1..f184ff7 100644
--- a/builtin.c
+++ b/builtin.c
@@ -228,6 +228,12 @@ static void f_keys(jv input[], jv output[]) {
     }
     free(keys);
     jv_free(input[0]);
+  } else if (jv_get_kind(input[0]) == JV_KIND_ARRAY) {
+    int n = jv_array_length(input[0]);
+    output[0] = jv_array();
+    for (int i=0; i<n; i++){
+      output[0] = jv_array_set(output[0], i, jv_number(i));
+    }
   } else {
     output[0] = jv_invalid_with_msg(jv_string_fmt("'keys' only supports object, not %s",
                                                   jv_kind_name(jv_get_kind(input[0]))));
diff --git a/docs/content/3.manual/manual.yml b/docs/content/3.manual/manual.yml
index 5981ce4..6c10770 100644
--- a/docs/content/3.manual/manual.yml
+++ b/docs/content/3.manual/manual.yml
@@ -385,10 +385,16 @@ sections:
           same for any two objects with the same set of keys,
           regardless of locale settings.
 
+          When `keys` is given an array, it returns the valid indices
+          for that array: the integers from 0 to length-1.
+
         examples:
           - program: 'keys'
             input: '{"abc": 1, "abcd": 2, "Foo": 3}'
             output: ['["Foo", "abc", "abcd"]']
+          - program: 'keys'
+            input: '[42,3,35]'
+            output: ['[0,1,2]']
 
       - title: `select`
         body: |
-- 
2.40.0