about summary refs log tree commit diff
path: root/kaakaa/utils.scm
blob: a922ef231495938ebfe3e5bac25530d56f494012 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
;;; kaakaa --- Tiny, security-focused AI agent in Guile
;;; Copyright © 2026 Arun Isaac <arunisaac@systemreboot.net>
;;;
;;; This file is part of kaakaa.
;;;
;;; kaakaa is free software: you can redistribute it and/or modify it
;;; under the terms of the GNU General Public License as published by
;;; the Free Software Foundation, either version 3 of the License, or
;;; (at your option) any later version.
;;;
;;; kaakaa is distributed in the hope that it will be useful, but
;;; WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
;;; General Public License for more details.
;;;
;;; You should have received a copy of the GNU General Public License
;;; along with kaakaa.  If not, see <https://www.gnu.org/licenses/>.

(define-module (kaakaa utils)
  #:use-module (srfi srfi-1)
  #:use-module (srfi srfi-26)
  #:use-module (ice-9 match)
  #:use-module (ice-9 popen)
  #:export (->
            foldn
            alist->plist
            call-with-input-pipe))

(define (->-helper x . procs)
  "Thread @var{x} through @var{procs}."
  (match procs
    (() x)
    ((head tail ...)
     (apply ->-helper (head x) tail))))

(define-syntax-rule (-> x (proc ...) ...)
  "Thread @var{x} through @var{procs}.

For example:
(-> 1
    (1+ <>)
    (* 2 <>)
    (expt <> 2))
=> 16"
  (->-helper x (cut proc ...) ...))

(define (foldn proc lst . inits)
  "Apply @var{proc} to the elements of @var{lst} to build a result, and return
that result. @var{proc} may return multiple values, in which case, an equal
number of values are returned. Each @var{proc} call is @code{(proc element
previous ...)} where @code{element} is an element of @var{lst}, and
@code{(previous ...)} is the return from the previous call to @var{proc} or the
given @var{inits} for the first call. For example,

(foldn (lambda (n sum sum-of-squares)
         (values (+ sum n)
                 (+ sum-of-squares (expt n 2))))
       (iota 10)
       0 0)
=> 45
=> 285"
  (apply values
         (fold (lambda (element results)
                 (call-with-values (cut apply proc element results) list))
               inits
               lst)))

(define (alist->plist alist)
  "Convert association list @var{alist} to a property list. Keys in
@var{alist} are converted to keywords."
  (append-map (match-lambda
                ((key . value)
                 (list (symbol->keyword (string->symbol key))
                       value)))
              alist))

(define (call-with-input-pipe command proc)
  "Call @var{proc} with input pipe to @var{command}. @var{command} is a
list of program arguments."
  (match command
    ((prog args ...)
     (let ((port #f))
       (dynamic-wind
         (lambda ()
           (set! port (apply open-pipe* OPEN_READ prog args)))
         (cut proc port)
         (lambda ()
           (unless (zero? (close-pipe port))
             (error "Command invocation failed" command))))))))