]> Sergey Matveev's repositories - zk.zsh.git/blobdiff - zk.zsh
Less variable names
[zk.zsh.git] / zk.zsh
diff --git a/zk.zsh b/zk.zsh
index 348a5b521b2e066d581741ded0f5baf0fbf83dcf..d45194d78a663d699cec2976d6c85b82268bcc75 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
 
@@ -6,9 +8,9 @@ usage() {
     cat >&2 <<EOF
 Usage:
   \$ $0:t links PAGE
-    Print the PAGE's links
+    Print PAGE's links
   \$ $0:t backs PAGE
-    Print who backlinks to the PAGE
+    Print PAGE's backlinks
   \$ $0:t htmls DIR
     Generate HTMLs in DIR
 EOF
@@ -17,7 +19,6 @@ EOF
 
 [[ $# -eq 2 ]] || usage
 
-# Collect all pages
 setopt GLOB_STAR_SHORT
 zmodload -F zsh/stat b:zstat
 typeset -A pages
@@ -28,31 +29,28 @@ for p (**(.)) {
 typeset -a cats
 for p (**(/)) cats=($p $cats)
 
-# Determine the links between them
+zmodload zsh/mapfile
 typeset -A links
 typeset -A backs
+typeset -aU ws
 for p (${(k)pages}) {
-    for w (`< $p`) {
-        [[ $w =~ "\[(.*)\]" ]] || continue
+    ws=()
+    for w (${=mapfile[$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]}"
+        ws=($ws $w)
     }
+    [[ $ws ]] && links[$p]=${(j: :)ws}
 }
-
-# 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}
+unset 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
@@ -76,12 +74,12 @@ getrel() {
 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
+    [[ $# -eq 1 ]] && data=${mapfile[$page]} || data=$2
+    data=${data//&/&amp;}
+    data=${data//</&lt;}
+    data=${data//>/&gt;}
+    local _links=(${(oi)=links[$page]})
+    for p ($_links) {
         getrel $page $p
         data="${data//\[${p}\]/<a href=\"${REPLY}.html\">[$p]</a>}"
     }
@@ -89,10 +87,18 @@ genHTML() {
 <html><head><title>$page (${pages[$page]})</title></head><body><pre>
 $data
 </pre>"
+    if [[ $_links ]]; then
+        print "<hr/>Links:<ul>"
+        for p ($_links) {
+            getrel $page $p
+            print "<li><a href=\"${REPLY}.html\">$p</a> <sup>${pages[$p]}</sup></li>"
+        }
+        print "</ul>"
+    fi
     local bs=(${(oi)=${backs[$page]}})
     if [[ $bs ]]; then
-        print "<hr/><ul>"
-        for p (${(oi)=${backs[$page]}}) {
+        print "<hr/>Backlinks:<ul>"
+        for p ($bs) {
             getrel $page $p
             print "<li><a href=\"${REPLY}.html\">$p</a> <sup>${pages[$p]}</sup></li>"
         }
@@ -102,26 +108,32 @@ $data
 }
 
 zmodload -F zsh/datetime b:strftime
-now=$(strftime "%F %T")
+strftime -s now "%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) entries=($entries "[$p] (${pages[$p]})") ;;
+        ($curdepth) _links=($p $_links) ;;
         ( $(( $curdepth + 1 )) ) cats=(${1}${${p#$1}%%/*} $cats) ;;
         (*) continue ;;
         esac
     }
+    for p (${(oi)_links}) entries=($entries "[$p] (${pages[$p]})")
     if [[ $cats ]]; then
-        entries=($entries "------------------------ >8 ------------------------")
-        for p (${(oi)cats}) entries=($entries "[$p/Index]")
+        entries=($entries "\nSubdirectories:\n")
+        for p (${(oi)cats}) {
+            entries=($entries "[$p/Index]")
+            _links=($p/Index $_links)
+        }
     fi
+    links[${1}Index]=${(j: :)_links}
     genHTML ${1}Index ${(F)entries}
 }