]> Sergey Matveev's repositories - zk.zsh.git/blobdiff - zk.zsh
Various use-cases
[zk.zsh.git] / zk.zsh
diff --git a/zk.zsh b/zk.zsh
index f4b7a240ba5cd651bcc5d5ce4b10e4f074c46272..38455b6f0b185a1cb47c334f8ab973ae66269a9b 100755 (executable)
--- a/zk.zsh
+++ b/zk.zsh
@@ -1,4 +1,6 @@
 #!/usr/bin/env zsh
+# zk.zsh -- zettelkästen/wiki/static website helper/generator
+# Copyright (C) 2022 Sergey Matveev <stargrave@stargrave.org>
 
 set -e
 
@@ -25,32 +27,30 @@ 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}) {
+    words=()
     for w (`< $p`) {
         [[ $w =~ "\[(.*)\]" ]] || continue
         w=${match[1]}
         [[ ${pages[$w]} ]] || {
-            [[ $ZK_SHOW_MISSING ]] && print "Missing $w"
+            [[ $ZK_SHOW_MISSING ]] && print "missing $w"
             continue
         }
-        links[$p]="$w ${links[$p]}"
+        words=($words $w)
     }
+    [[ $words ]] && links[$p]=${(j: :)words}
 }
-
-# Deduplicate all references
-for p w (${(kv)links}) {
-    local ws=(${(u)=w})
-    links[$p]=${(j: :)ws}
-    for w ($ws) backs[$w]="$p ${backs[$w]}"
-}
-for p w (${(kv)backs}) {
-    local ws=(${(u)=w})
-    backs[$p]=${(j: :)ws}
+for p ws (${(kv)links}) {
+    for w (${=ws}) backs[$w]="$p ${backs[$w]}"
 }
+for p ws (${(kv)backs}) backs[$p]=${(j: :)${(u)=ws}}
 
 getrel() {
     # nearly the copy-paste of Functions/Misc/relative
@@ -71,44 +71,79 @@ getrel() {
     REPLY=$src${dst:+/$dst}
 }
 
-genhtml() {
+genHTML() {
     local page=$1
     local data p
     [[ $# -eq 1 ]] && data=`< $page` || data=$2
-    data="${data//&/&amp;}"
-    data="${data//</&lt;}"
-    data="${data//>/&gt;}"
-    for p (${(k)pages}) {
-        [[ $p = index ]] && continue
+    data=${data//&/&amp;}
+    data=${data//</&lt;}
+    data=${data//>/&gt;}
+    for p (${=links[$page]}) {
         getrel $page $p
-        data="${data//${p}/<a href=\"${REPLY}.html\">$p</a>}"
+        data="${data//\[${p}\]/<a href=\"${REPLY}.html\">[$p]</a>}"
     }
     print "<\!DOCTYPE html>
 <html><head><title>$page (${pages[$page]})</title></head><body><pre>
 $data
-</pre><hr/><ul>"
-    for p (${(oi)=${backs[$page]}}) {
-        getrel $page $p
-        print "<li><a href=\"${REPLY}.html\">$p</a> <sup>${pages[$p]}</sup></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=()
+    local _links=()
+    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) _links=($p $_links) ;;
+        ( $(( $curdepth + 1 )) ) cats=(${1}${${p#$1}%%/*} $cats) ;;
+        (*) continue ;;
+        esac
     }
-    print "</ul></body></html>"
+    for p (${(oi)_links}) entries=($entries "[$p] (${pages[$p]})")
+    if [[ $cats ]]; then
+        entries=($entries "\nBacklinks:\n")
+        for p (${(oi)cats}) {
+            entries=($entries "[$p/Index]")
+            _links=($p/Index $_links)
+        }
+    fi
+    links[${1}Index]=${(j: :)_links}
+    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 (${pages[$p]})" $all)
-    zmodload -F zsh/datetime b:strftime
-    pages[ALL]=$(strftime "%F %T")
-    genhtml ALL ${(j:\n:)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