#!r6rs
;;; -*- mode: scheme -*-
;;; run64 --- SRFI-64 test runner
;;; Copyright © 2022 Arun Isaac <arunisaac@systemreboot.net>
;;;
;;; This file is part of run64.
;;;
;;; run64 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.
;;;
;;; run64 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 run64.  If not, see <https://www.gnu.org/licenses/>.

(import (rnrs programs)
        (srfi srfi-64))

(define (color code str)
  (string-append (string #\esc)
                 "["
                 (number->string code)
                 "m"
                 str
                 (string #\esc)
                 "[0m"))

(define (red str)
  (color 31 str))

(define (green str)
  (color 32 str))

(define (magenta str)
  (color 35 str))

(define (make-runner)
  (let ((runner (test-runner-null)))
    (test-runner-on-group-begin! runner
      (lambda (runner suite-name count)
        (display (magenta (string-append "%%%% " suite-name)))
        (newline)))
    (test-runner-on-group-end! runner
      (lambda _
        (newline)))
    (test-runner-on-test-end! runner
      (lambda (runner)
        (let ((name (test-runner-test-name runner))
              (result (string-upcase
                       (symbol->string (test-result-kind runner))))
              (result-alist (test-result-alist runner)))
          (display (case (test-result-kind runner)
                     ((pass) (green result))
                     (else (red result))))
          (display " ")
          (display name)
          (newline)
          ;; If test did not pass, print details.
          (unless (eq? (test-result-kind runner)
                       'pass)
            (display (assq-ref result-alist 'source-file))
            (display (assq-ref result-alist 'source-line))
            (newline)
            (display "expected: ")
            (display (assq-ref result-alist 'expected-value))
            (newline)
            (display "actual: ")
            (display (assq-ref result-alist 'actual-value))
            (newline)))))
    runner))

(define (main args)
  (let ((runner (make-runner)))
    (test-with-runner runner
      (for-each load (cdr args))
      (display (magenta "SUMMARY"))
      (newline)
      (display "PASS: ")
      (display (test-runner-pass-count runner))
      (newline)
      (display "FAIL: ")
      (display (test-runner-fail-count runner))
      (newline)
      (display "XPASS: ")
      (display (test-runner-xpass-count runner))
      (newline)
      (display "XFAIL: ")
      (display (test-runner-xfail-count runner))
      (newline)
      (display "SKIP: ")
      (display (test-runner-skip-count runner))
      (newline)
      (exit (zero? (test-runner-fail-count runner))))))

(main (command-line))