
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;
;; MODULE      : lp-build.scm
;; DESCRIPTION : building programs from literate source files
;; COPYRIGHT   : (C) 2015  Joris van der Hoeven
;;
;; This software falls under the GNU general public license version 3 or later.
;; It comes WITHOUT ANY WARRANTY WHATSOEVER. For details, see the file LICENSE
;; in the root directory or <http://www.gnu.org/licenses/gpl-3.0.html>.
;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(texmacs-module (utils literate lp-build)
  (:use (utils literate lp-edit)))

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Building the initial table (without substitutions)
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define (extract-lines t)
  (cond ((tm-atomic? t) (list t))
        ((tm-func? t 'document)
         (append-map extract-lines (tm-children t)))
        ((or (tm-func? t 'folded-newline-before 1)
             (tm-func? t 'unfolded-newline-before 1))
         (cons "" (extract-lines (tm-ref t 0))))
        ((list-or (map (cut tm-func? <> 'document) (tm-children t)))
         (with l (list-filter (tm-children t) (cut tm-func? <> 'document))
           (append-map extract-lines l)))
        (else (list t))))

(define (build-table t)
  (let* ((l (search-chunks t))
         (t (make-ahash-table)))
    (for (c l)
      (let* ((name (tm->string (tm-ref c 0)))
             (body (tm-ref c 3)))
        (if (not (tm-func? body 'document))
            (set! body `(document ,body)))
        (let* ((old (tm-children (or (ahash-ref t name) `(document))))
               (new (append old (extract-lines body))))
          (ahash-set! t name `(document ,@new)))))
    t))

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Performing the necessary substitutions
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define (ahash-table-split t pred?)
  (let* ((t1 (make-ahash-table))
         (t2 (make-ahash-table)))
    (for (key (map car (ahash-table->list t)))
      (with im (ahash-ref t key)
        (if (pred? (ahash-ref t key))
            (ahash-set! t1 key im)
            (ahash-set! t2 key im))))
    (list t1 t2)))

(define (should-expand? t)
  (cond ((tm-atomic? t) #f)
        ((tm-func? t 'chunk-ref 1) #t)
        (else (list-or (map should-expand? (tm-children t))))))

(define (expand-names x t)
  (cond ((tm-atomic? x) x)
        ((tm-func? x 'document)
         (let* ((l1 (map (cut expand-names <> t) (tm-children x)))
                (l2 (map (lambda (y) (if (tm-func? y 'document)
                                         (tm-children y) (list y))) l1)))
           `(document ,@(apply append l2))))
        ((tm-func? x 'chunk-ref 1)
         (if (and (tm-atomic? (tm-ref x 0))
                  (ahash-ref t (tm->string (tm-ref x 0))))
             (ahash-ref t (tm->string (tm-ref x 0)))
             x))
        ((tm-func? x 'concat)
         (if (and (tm-func? (tm-ref x :last) 'chunk-ref 1)
                  (tm-atomic? (tm-ref x :last 0))
                  (ahash-ref t (tm->string (tm-ref x :last 0))))
             (let* ((count (lambda (x) (if (tm-atomic? x)
                                           (tmstring-length (tm->string x))
                                           0)))
                    (i (apply + (map count (cDr (tm-children x)))))
                    (pre (apply string-append (map (lambda (y) " ") (.. 0 i))))
                    (doc (ahash-ref t (tm->string (tm-ref x :last 0))))
                    (lines (tm-children doc)))
               `(document (concat ,@(cDr (tm-children x)) ,(car lines))
                          ,@(map (lambda (x) (tmconcat pre x))
                                 (cdr lines))))
             x))
        (else x)))

(define (expand-table t)
  (with (todo done) (ahash-table-split t should-expand?)
    (with ok? #f
      (for (key (map car (ahash-table->list todo)))
        (with subst (expand-names (ahash-ref todo key) done)
          (when (not (should-expand? subst))
            (ahash-set! done key subst)
            (set! ok? #t))))
      (cond ((== (ahash-size todo) 0)
             done)
            (ok?
             (for (key (map car (ahash-table->list todo)))
               (when (not (ahash-ref done key))
                 (ahash-set! done key (ahash-ref todo key))))
             (expand-table done))
            (else
              (for (key (map car (ahash-table->list todo)))
                (display* "TeXmacs] Problematic chunk: " key "\n"))
              (set-message "Error: cyclic or missing chunks detected"
                           "build-all")
              done)))))

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Debugging subroutines
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define (display-table t)
  (for (key (map car (ahash-table->list t)))
    (display* key "\n----------------------------------\n")
    (with im (tm-children (ahash-ref t key))
      (for (l im)
        (display* "  " l "\n")))
    (display* "----------------------------------\n\n")))

(define (display-table* t*)
  (with t (verbatim-table t*)
    (for (key (map car (ahash-table->list t)))
      (display* key "\n----------------------------------\n")
      (display* (ahash-ref t key))
      (display* "----------------------------------\n\n"))))

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Memorize which files have been built
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define lp-db (url->url "$TEXMACS_HOME_PATH/system/database/lp-master.tmdb"))

(define (lp-set-time-stamp src dir)
  (tmdb-keep-history lp-db #f)
  (let* ((src* (url->system src))
         (dir* (url->system dir))
         (id (string-append src* "-" dir*))
         (time (number->string (url-last-modified src))))
    (tmdb-set-field lp-db id "time-stamp" (list time) (current-time))))

(define (lp-get-time-stamp src dir)
  (tmdb-keep-history lp-db #f)
  (let* ((src* (url->system src))
         (dir* (url->system dir))
         (id (string-append src* "-" dir*)))
    (with l (tmdb-get-field lp-db id "time-stamp" (current-time))
      (and l (nnull? l) (car l)))))

(define (lp-build-conditional src dir fun)
  (with stamp (lp-get-time-stamp src dir)
    (when (or (not stamp)
              (not (url-exists? src))
              (not (url-exists? dir))
              (< (string->number stamp) (url-last-modified src)))
      (fun src dir)
      (when (and (url-exists? src)
                 (url-exists? dir))
        (lp-set-time-stamp src dir)))))

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Main build process
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define (verbatim-table t)
  (with r (make-ahash-table)
    (for (key (map car (ahash-table->list t)))
      (let* ((l (tm-children (tm->stree (ahash-ref t key))))
             (c (map (lambda (x)
                       (convert x "texmacs-stree" "verbatim-document")) l))
             (i (map (cut string-append <> "\n") c)))
        (ahash-set! r key (apply string-append i))))
    r))

(define (write-table t dir)
  (for (key (map car (ahash-table->list t)))
    (when (string-contains? key ".")
      (with target (url-append dir key)
        (when (not (url-exists? (url-head target)))
          (system-mkdir (url-head target)))
        (display* "TeXmacs] Building " (url->system target) "\n")
        (string-save (ahash-ref t key) target)))))

(define (lp-build* file dir)
  (with doc (if (buffer-exists? file)
                (buffer-get-body file)
                (tmfile-extract (tree-import file "texmacs") 'body))
    (with t (build-table doc)
      (with x (expand-table t)
        (with v (verbatim-table x)
          (write-table v dir))))))

(define (lp-build file dir)
  (if (buffer-exists? file)
      (lp-build* file dir)
      (lp-build-conditional file dir lp-build*)))

(tm-define (lp-build-buffer)
  (update-all-chunk-states)
  (lp-build (current-buffer) (url-head (current-buffer))))

(tm-define (lp-build-buffer-in dir)
  (:argument dir "Build directory")
  (update-all-chunk-states)
  (lp-build (current-buffer) dir))

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Building all files in a directory
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define (lp-build-directory source target)
  (let* ((f (url-append (url-append source (url-any)) (url-wildcard "*.tm")))
         (l (url->list (url-expand (url-complete f "r")))))
    (for (src l)
      (let* ((rel (url-delta (url-append source "dummy") src))
             (obj (url-append target rel)))
        (lp-build src (url-head obj))))))

(tm-define (lp-interactive-build-directory)
  (:interactive #t)
  (user-url "Directory" "directory" 
    (lambda (src) (lp-build-directory src src))))

(tm-define (lp-interactive-build-directory-in)
  (:interactive #t)
  (user-url "Source directory" "directory" 
    (lambda (src) (user-url "Destination directory" "directory"
      (lambda (dest) (lp-build-directory src dest))))))
