From 858bd9d2ff3009f4fcff3a13b597ddc8b3ebc89f Mon Sep 17 00:00:00 2001
From: Edward Langley <el-github@elangley.org>
Date: Tue, 29 Mar 2022 20:37:02 -0700
Subject: [PATCH] feat(on): make on more conformant with the Haskell function

---
 lens.lisp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/lens.lisp b/lens.lisp
index 56fac4b..0a1faec 100644
--- a/lens.lisp
+++ b/lens.lisp
@@ -269,10 +269,16 @@
          (apply ,fsym ,@args ,seq)))))
 
 (defun-ct on (fun key)
+  "Transform arguments with KEY and then apply FUN
+
+> (eql (funcall (on 'equal 'car)
+>          '(\"a\" 1 2)
+>          '(\"a\" 2 e))
+>      t)"
   (let ((fun (functionalize fun))
         (key (functionalize key)))
-    (lambda (it)
-      (funcall fun (funcall key it)))))
+    (lambda (&rest its)
+      (funcall fun (mapcar key its)))))
 
 (defun filler (length1 length2 fill-value)
   (if (< length1 length2)