]> Sergey Matveev's repositories - zk.zsh.git/blobdiff - zk.zsh
Do not iterate over all pages during HTML rendering
[zk.zsh.git] / zk.zsh
diff --git a/zk.zsh b/zk.zsh
index 5a46d999208819990a59ed9c152d83c38debec84..eb05aff979a2a5a7796cb6a2698a62c787645e35 100755 (executable)
--- a/zk.zsh
+++ b/zk.zsh
@@ -1,4 +1,6 @@
 #!/usr/bin/env zsh
+# zk.zsh -- zettelkästen-related helper
+# Copyright (C) 2022 Sergey Matveev <stargrave@stargrave.org>
 
 set -e
 
@@ -19,37 +21,37 @@ EOF
 
 # Collect all pages
 setopt GLOB_STAR_SHORT
+zmodload -F zsh/stat b:zstat
 typeset -A pages
-for f (**(.)) pages[$f]=1
+for p (**(.)) {
+    zstat -A reply -F "%F %T" +mtime $p
+    pages[$p]=${reply[1]}
+}
+typeset -a cats
+for p (**(/)) cats=($p $cats)
 
 # Determine the links between them
 typeset -A links
 typeset -A backs
+typeset -aU words
 for p (${(k)pages}) {
-    for w (`cat $p`) {
-        [[ $w =~ "^[([{].*" ]] && w=${MATCH[2,-2]}
-        [[ ${pages[$w]} ]] || continue
-        links[$p]="$w ${links[$p]}"
-    }
-}
-
-# Deduplicate all references
-typeset -A ws
-for p (${(k)links}) {
-    ws=()
-    for w (${=${links[$p]}}) {
-        ws[$w]=1
-        backs[$w]="$p ${backs[$w]}"
+    words=()
+    for w (`< $p`) {
+        [[ $w =~ "\[(.*)\]" ]] || continue
+        w=${match[1]}
+        [[ ${pages[$w]} ]] || {
+            [[ $ZK_SHOW_MISSING ]] && print "missing $w"
+            continue
+        }
+        words=($words $w)
     }
-    links[$p]=${(k)ws}
+    [[ $words ]] && links[$p]=${(j: :)words}
 }
-for p (${(k)backs}) {
-    ws=()
-    for w (${=${backs[$p]}}) ws[$w]=1
-    backs[$p]=${(k)ws}
+for p ws (${(kv)links}) {
+    for w (${=ws}) backs[$w]="$p ${backs[$w]}"
 }
+for p ws (${(kv)backs}) backs[$p]=${(j: :)${(u)=ws}}
 
-autoload -U relative
 getrel() {
     # nearly the copy-paste of Functions/Misc/relative
     local dst=$2:a
@@ -69,45 +71,76 @@ getrel() {
     REPLY=$src${dst:+/$dst}
 }
 
-genhtml() {
+genHTML() {
     local page=$1
-    local data
-    [[ $# -eq 1 ]] && data="`cat $page`" || data=$2
-    local tgt
-    data="${data//&/&amp;}"
-    data="${data//</&lt;}"
-    data="${data//>/&gt;}"
-    for p (${(k)pages}) {
-        [[ $p = ALL ]] && continue
-        [[ $p = index ]] && continue
-        tgt=`getrel $page $p`
-        data="${data//${p}/<a href=\"${tgt}.html\">$p</a>}"
+    local data p
+    typeset -a _links
+    [[ $# -eq 1 ]] && data=`< $page` || data=$2
+    data=${data//&/&amp;}
+    data=${data//</&lt;}
+    data=${data//>/&gt;}
+    [[ ( $page = "Index" ) || ( $page =~ "/Index" ) ]] \
+        && _links=(${(k)pages}) || _links=(${=links[$page]})
+    for p ($_links) {
+        getrel $page $p
+        data="${data//\[${p}\]/<a href=\"${REPLY}.html\">[$p]</a>}"
     }
     print "<\!DOCTYPE html>
-<html><head><title>$page</title></head><body><pre>
+<html><head><title>$page (${pages[$page]})</title></head><body><pre>
 $data
-</pre><hr/><ul>"
-    for p (${(oi)=${backs[$page]}}) {
-        [[ $p = ALL ]] && continue
-        tgt=`getrel $page $p`
-        print "<li><a href=\"${tgt}.html\">$p</a></li>"
+</pre>"
+    local bs=(${(oi)=${backs[$page]}})
+    if [[ $bs ]]; then
+        print "<hr/><ul>"
+        for p (${(oi)=${backs[$page]}}) {
+            getrel $page $p
+            print "<li><a href=\"${REPLY}.html\">$p</a> <sup>${pages[$p]}</sup></li>"
+        }
+        print "</ul>"
+    fi
+    print "</body></html>"
+}
+
+zmodload -F zsh/datetime b:strftime
+now=$(strftime "%F %T")
+
+genIndex() {
+    local p
+    local entries=()
+    typeset -aU cats=()
+    local curdepth=${#${(s:/:)1}}
+    (( curdepth = curdepth + 1 ))
+    for p (${(oi)${(k)pages[(I)$1*]}}) {
+        [[ $p =~ "/Index$" ]] && continue
+        case ${#${(As:/:)p}} in
+        ($curdepth) entries=($entries "[$p] (${pages[$p]})") ;;
+        ( $(( $curdepth + 1 )) ) cats=(${1}${${p#$1}%%/*} $cats) ;;
+        (*) continue ;;
+        esac
     }
-    print "</ul></body></html>"
+    if [[ $cats ]]; then
+        entries=($entries "------------------------ >8 ------------------------")
+        for p (${(oi)cats}) entries=($entries "[$p/Index]")
+    fi
+    genHTML ${1}Index ${(F)entries}
 }
 
 case $1 in
 (links) for w (${(oi)=${links[$2]}}) print $w ;;
 (backs) for w (${(oi)=${backs[$2]}}) print $w ;;
-(html) genhtml $2 ;;
+(html) genHTML $2 ;;
+(html-index) genIndex $2 ;;
 (htmls)
     for p (${(k)pages}) {
-        local subdir=$p:h
-        mkdir -p $2/$subdir
-        genhtml $p > $2/$p.html
+        mkdir -p $2/$p:h
+        genHTML $p > $2/$p.html
+        touch -r $p $2/$p.html
     }
-    local all=""
-    for p (${(Oi)${(k)pages}}) all="$p\n$all"
-    genhtml ALL $all > $2/ALL.html
+    for p ($cats) pages[${p}/Index]=$now
+    pages[Index]=$now
+    for p ($cats) genIndex $p/ > $2/$p/Index.html
+    genIndex "" > $2/Index.html
+    for p ("" $cats) touch -d ${now/ /T} $2/$p/Index.html
     ;;
 (*) usage ;;
 esac