zk.zsh

   1 #!/usr/bin/env zsh
   2 # zk.zsh -- zettelkästen/wiki/static website helper/generator
   3 # Copyright (C) 2022 Sergey Matveev <stargrave@stargrave.org>
   4
   5 set -e
   6 ZK_VERSION=ZKZSH1
   7
   8 usage() {
   9     cat >&2 <<EOF
  10 Usage:
  11   \$ $0:t links PAGE
  12     Print PAGE's links
  13   \$ $0:t backs PAGE
  14     Print PAGE's backlinks
  15   \$ $0:t htmls DIR
  16     Generate HTMLs in DIR
  17 EOF
  18     exit 1
  19 }
  20
  21 [[ $# -eq 2 ]] || usage
  22
  23 setopt GLOB_STAR_SHORT
  24 zmodload -F zsh/stat b:zstat
  25 typeset -A pages
  26 typeset -A sizes
  27 for p (**(.)) {
  28     zstat -A mtime -F "%F %T" +mtime $p
  29     zstat -A size +size $p
  30     pages[$p]=${mtime[1]}
  31     sizes[$p]=${size[1]}
  32 }
  33 typeset -a cats
  34 for p (**(/)) cats=($p $cats)
  35
  36 zmodload zsh/mapfile
  37 zmodload -F zsh/files b:zf_mkdir
  38 typeset -A links
  39 typeset -A backs
  40 typeset -A cached
  41 typeset -aU ws
  42 for p (${(k)pages}) {
  43     [[ $ZK_CACHE ]] && {
  44         zstat -A inode +inode $p
  45         zstat -A ctime +ctime $p
  46         cache=(${(f)mapfile[$ZK_CACHE/$p]})
  47         if [[ ( ${cache[1]} = $ZK_VERSION ) &&
  48               ( ${cache[2]} = ${inode[1]} ) &&
  49               ( ${cache[3]} = ${ctime[1]} ) ]]; then
  50             ws=(${cache[4,-1]})
  51             [[ $ws ]] && links[$p]=${(j: :)ws}
  52             cached[$p]=1
  53             continue
  54         fi
  55     }
  56     ws=()
  57     for w (${=mapfile[$p]}) {
  58         [[ $w =~ "\[([^] ]+)\]" ]] || continue
  59         w=${match[1]}
  60         [[ ${pages[$w]} ]] || {
  61             [[ $ZK_SHOW_MISSING ]] && print "missing $w"
  62             continue
  63         }
  64         ws=($ws $w)
  65     }
  66     [[ $ZK_CACHE ]] && {
  67         zf_mkdir -p $ZK_CACHE/$p:h
  68         ws=($ZK_VERSION ${inode[1]} ${ctime[1]} $ws)
  69         print -l $ws > $ZK_CACHE/$p
  70         ws=(${ws[3,-1]})
  71     }
  72     [[ $ws ]] && links[$p]=${(j: :)ws}
  73 }
  74 unset cache ws
  75 for p ws (${(kv)links}) {
  76     for w (${=ws}) backs[$w]="$p ${backs[$w]}"
  77 }
  78 for p ws (${(kv)backs}) backs[$p]=${(j: :)${(u)=ws}}
  79
  80 getrel() {
  81     # nearly the copy-paste of Functions/Misc/relative
  82     local dst=$2:a
  83     local src=$1:h:a
  84     local -a cur abs
  85     cur=(${(s:/:)src})
  86     abs=(${(s:/:)dst:h} $dst:t)
  87     integer i=1
  88     while [[ i -le $#abs && $abs[i] == $cur[i] ]] ; do
  89         ((++i > $#cur)) && {
  90             REPLY=${(j:/:)abs[i,-1]}
  91             return
  92         }
  93     done
  94     src=${(j:/:)cur[i,-1]/*/..}
  95     dst=${(j:/:)abs[i,-1]}
  96     REPLY=$src${dst:+/$dst}
  97 }
  98
  99 genHTML() {
 100     local page=$1
 101     local data p
 102     [[ $# -eq 1 ]] && data=${mapfile[$page]} || data=$2
 103     local _links=(${(oi)=links[$page]})
 104     if [[ ( ${cached[$page]} ) && ( -s $ZK_CACHE/${page}.html ) ]]; then
 105         cat $ZK_CACHE/${page}.html
 106     else
 107         data=${data//&/&amp;}
 108         data=${data//</&lt;}
 109         data=${data//>/&gt;}
 110         for p ($_links) {
 111             getrel $page $p
 112             data="${data//\[${p}\]/<a href=\"${REPLY}.html\">[$p]</a>}"
 113         }
 114         data="<!DOCTYPE html>
 115 <html><head><title>$page (${pages[$page]})</title></head><body><pre>
 116 $data</pre>"
 117         if [[ $ZK_CACHE ]]; then
 118             print -r "$data" > $ZK_CACHE/${page}.html
 119             cat $ZK_CACHE/${page}.html
 120         else
 121             print -r "$data"
 122         fi
 123     fi
 124     if [[ $_links ]]; then
 125         print "<hr/>Links:<ul>"
 126         for p ($_links) {
 127             getrel $page $p
 128             print "<li><a href=\"${REPLY}.html\">$p</a> <sup>${pages[$p]}</sup></li>"
 129         }
 130         print "</ul>"
 131     fi
 132     local bs=(${(oi)=${backs[$page]}})
 133     if [[ $bs ]]; then
 134         print "<hr/>Backlinks:<ul>"
 135         for p ($bs) {
 136             getrel $page $p
 137             print "<li><a href=\"${REPLY}.html\">$p</a> <sup>${pages[$p]}</sup></li>"
 138         }
 139         print "</ul>"
 140     fi
 141     print "</body></html>"
 142 }
 143
 144 zmodload -F zsh/datetime b:strftime
 145 strftime -s now "%F %T"
 146
 147 genIndex() {
 148     local p
 149     local entries=()
 150     local _links=()
 151     typeset -aU cats=()
 152     local curdepth=${#${(s:/:)1}}
 153     (( curdepth = curdepth + 1 ))
 154     for p (${(oi)${(k)pages[(I)$1*]}}) {
 155         [[ ( $p =~ "/Index$" ) || ( $p = "Index" ) ]] && continue
 156         case ${#${(As:/:)p}} in
 157         ($curdepth) _links=($p $_links) ;;
 158         ( $(( $curdepth + 1 )) ) cats=(${1}${${p#$1}%%/*} $cats) ;;
 159         (*) continue ;;
 160         esac
 161     }
 162     for p (${(oi)_links}) \
 163         entries=($entries "[$p] (${pages[$p]}) (${sizes[$p]} bytes)")
 164     if [[ $cats ]]; then
 165         entries=($entries " " "Subdirectories:" " ")
 166         for p (${(oi)cats}) {
 167             entries=($entries "[$p/Index]")
 168             _links=($p/Index $_links)
 169         }
 170     fi
 171     links[${1}Index]=${(j: :)_links}
 172     genHTML ${1}Index ${(F)entries}
 173 }
 174
 175 case $1 in
 176 (links) for w (${(oi)=${links[$2]}}) print $w ;;
 177 (backs) for w (${(oi)=${backs[$2]}}) print $w ;;
 178 (html) genHTML $2 ;;
 179 (html-index) genIndex $2 ;;
 180 (htmls)
 181     for p (${(k)pages}) {
 182         zf_mkdir -p $2/$p:h
 183         genHTML $p > $2/$p.html
 184         touch -r $p $2/$p.html
 185     }
 186     for p ($cats) pages[${p}/Index]=$now
 187     pages[Index]=$now
 188     for p ($cats) genIndex $p/ > $2/$p/Index.html
 189     genIndex "" > $2/Index.html
 190     for p ("" $cats) touch -d ${now/ /T} $2/$p/Index.html
 191     ;;
 192 (*) usage ;;
 193 esac