diff --git a/fun/wcl/default.nix b/fun/wcl/default.nix new file mode 100644 index 000000000..17367f3d8 --- /dev/null +++ b/fun/wcl/default.nix @@ -0,0 +1,13 @@ +{ pkgs, ... }: + +pkgs.nix.buildLisp.program { + name = "wc"; + + srcs = [ + ./wc.lisp + ]; + + deps = with pkgs.third_party.lisp; [ + iterate + ]; +} diff --git a/fun/wcl/wc.lisp b/fun/wcl/wc.lisp new file mode 100644 index 000000000..f443125a8 --- /dev/null +++ b/fun/wcl/wc.lisp @@ -0,0 +1,33 @@ +(defpackage wc + (:use #:cl #:iterate) + (:export :main)) +(in-package :wc) +(declaim (optimize (speed 3) (safety 0))) + +(defun main () + (let ((filename (cadr sb-ext:*posix-argv*)) + (space (char-code #\Space)) + (newline (char-code #\Newline))) + (with-open-file (file-stream filename :element-type '(unsigned-byte 8)) + (iter + (for byte in-stream file-stream using #'read-byte) + (for previous-byte previous byte) + (for is-newline = (eq newline byte)) + + ;; Count each byte + (sum 1 into bytes) + + ;; Count every newline + (counting is-newline into newlines) + + ;; Count every "word", unless the preceding character already + ;; was a space. + (when (or (eq space previous-byte) + (eq newline previous-byte)) + (next-iteration)) + + (counting (or is-newline (eq space byte)) + into words) + + (declare (fixnum bytes newlines words)) + (finally (format t " ~A ~A ~A ~A~%" newlines words bytes filename))))))