#!/usr/bin/env bash

set -o pipefail

IFS= read -rd '' USAGE <<EOF

Zettel helper script.

Usage:

    zet create|new|c|n TITLE...   : Create a new zettel
    zet edit|e QUERY...           : Search for a zettel and edit it
    zet help|h                    : Show this help screen
    zet id QUERY...               : Search for a zettel and display its ID
    zet latest                    : Edit the latest zettel
    zet print|p QUERY...          : Search for a zettel and print it
    zet view|v QUERY...           : Search for a zettel and view it
    zet QUERY...                  : Print IDs and titles of zettels matching QUERY

EOF

set -o errexit

SELECTED_ZET=''
ZET_DIR=${ZET_DIR:-.}
ZET_LIST=()

cmd_create() {
  create_zettel "${1^}"
}

cmd_edit() {
  QUERY="$*"
  [[ -d "$QUERY" ]] && edit_zet "$QUERY"

  parse_zet_list < <(search_zettel "$QUERY")
  select_zet
  edit_zet "$SELECTED_ZET"
}

cmd_id() {
  parse_zet_list < <(search_zettel "$@")
  select_zet

  echo "$SELECTED_ZET"
}

cmd_links() {
  QUERY="$1"

  generate_links "$(cmd_search "$QUERY")"
}

cmd_search() {
  parse_zet_list < <(search_zettel "$@")
  printf "%s\n" "${ZET_LIST[@]}"
}

cmd_view() {
  QUERY="$*"
  if [[ -d "$QUERY" ]]; then
    view_zettel "$QUERY"
    exit
  fi

  parse_zet_list < <(search_zettel "$QUERY")
  select_zet
  view_zettel "$SELECTED_ZET"
}

commit_zettel() {
  ZID="$1"
  MESSAGE="$2"

  if [[ -z "$MESSAGE" ]]; then
    get_title "$ZID"
  fi

  git add "$ZID"
  git commit -m "$MESSAGE"
  git push
}

create_zettel() {
  TITLE="$1"
  ZID=$(new_zid)

  mkdir -p "$ZID"
  echo "= $TITLE" > "$ZID/index.adoc"
  edit_file "$ZID/index.adoc"
  on_save "$ZID"
}

delete_zettel() {
  [[ -d "$ZID" ]] && rm -fr "$ZID"
  commit_zettel "$ZID"
}

edit_file() {
  "$EDITOR" "$1"
}

edit_zet() {
  edit_file "$1/index.adoc"
  on_save "$1"
}

generate_links() {
  echo "$1" | while IFS= read -r line; do
    id="${line%% *}"
    title="${line#* }"
    echo "* link:../${id}/index.adoc[${title}]"
  done
}

get_latest_zettel() {
  find . -maxdepth 1 -type d -name '[0-9]*' -printf '%f\n' | sort -nr | head -n 1
}

get_title() {
  get_title_from_file "$1/index.adoc"
}

get_title_from_file() {
  head -n 1 "$1" | sed -e 's/^[#=] //'
}

main() {
  case "$1" in
    create | new | c | n)
      shift 1
      cmd_create "$@"
      ;;

    edit | e)
      shift 1
      cmd_edit "$@"
      ;;

    help | h)
      show_usage
      ;;

    id)
      shift 1
      cmd_id "$@"
      ;;

    latest)
      ZID="$(get_latest_zettel)"
      edit_zet "$ZID"
      ;;

    links | l)
      shift 1
      cmd_links "$@"
      ;;

    view | v)
      shift 1
      cmd_view "$@"
      ;;

    *)
      cmd_search "$@"
      ;;
  esac
}

new_zid() {
  EXISTING_ZETTELS=$(find . -maxdepth 1 -type d -regex './[0-9]+' | wc -l)

  echo $((EXISTING_ZETTELS + 1))
}

on_save() {
  ZID="$1"

  if [[ -s "$ZID/index.adoc" ]]; then
    TITLE=$(get_title "$ZID")
    commit_zettel "$ZID" "$TITLE"
  else
    echo "Deleting empty zettel: $ZID"
    delete_zettel "$ZID"
  fi
}

parse_zet_list() {
  ZET_LIST=()

  while IFS= read -r ZID; do
    TITLE=$(get_title "$ZID")
    ZET_LIST+=("$ZID $TITLE")
  done
}

search_zettel() {
  if [[ "$*" == "latest" ]] || [[ "$*" == "l" ]]; then
    get_latest_zettel
    return
  fi

  QUERY="$*"

  grep_args=("--extended-regexp")
  [[ "$QUERY" != "" ]] && grep_args+=("--word-regex")

  git grep -i --name-only "${grep_args[@]}" "$QUERY" | grep -o -E '[0-9]+' | sort -un
}

select_zet() {
  if [[ "${#ZET_LIST[@]}" == 0 ]]; then
    echo "No zettels to select"
    exit 1
  fi

  if [[ "${#ZET_LIST[@]}" == 1 ]]; then
    SELECTED_ZET=$(awk '{ print $1 }' <<<"${ZET_LIST[0]}")
    return
  fi

  selector

  if [[ -z "$SELECTED_ZET" ]]; then
    echo "No zet selected"
    exit 1
  fi
}

selector() {
  ITEM=$(printf "%s\n" "${ZET_LIST[@]}" | fzf --prompt="Select a zet: ")

  SELECTED_ZET=$(awk '{ print $1 }' <<< "$ITEM")
}

show_usage() {
  echo "$USAGE"
}

view_zettel() {
  cat "$1/index.adoc"
}

(cd "$ZET_DIR" && main "$@")