markdown-append-chapter

#!/bin/bash

SCRIPT_NAME="$(basename "$(test -L "$0" && readlink "$0" || echo "$0")")"
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
LAST_LINES_TO_SHOW=20
LAST_HEADERS_LINES_TO_SHOW=7
MARKDOWN_EDIT_LAST_FILES_PATH="$HOME/.markdown-edit-last-files"
MAX_HISTORY_LINES=10

function echoerr {
  printf "%s\n" "$*" >&2;
}

function help_exit { EXIT_CODE="$1";
  if [[ -z "$EXIT_CODE" ]]; then
    EXIT_CODE=1
  fi

  echoerr "Usage: ${SCRIPT_NAME} [-c|--clipboard] [TARGET_FILENAME]"
  exit "$EXIT_CODE"
}

function ask_continue {
  local ACTION_NAME=$1
  if [[ -z "${ACTION_NAME}" ]]; then
    ACTION_NAME="continue"
  fi

  read -n 1 -r -p "Press \"y\" to ${ACTION_NAME}: " input
  echo >&2

  if [ "$input" != "Y" ] && [ "$input" != "y" ] && [ "$input" != "д" ] && [ "$input" != "Д" ]; then
    exit
  fi
  echo 1
}

function select_file_via_fzf {
  local TX_DIR="$HOME/tx"
  cd "$TX_DIR" || {
    echoerr "Error: directory $TX_DIR not found"
    exit 100
  }

  # Get all .md and .txt files
  local ALL_FILES=$(find . -type f \( -name "*.md" -o -name "*.txt" \) | sort)

  # Reorder files based on history
  local REORDERED_FILES="$ALL_FILES"
  if [[ -f "$MARKDOWN_EDIT_LAST_FILES_PATH" ]]; then
    while read HISTORY_LINE; do
      if [[ -n "$HISTORY_LINE" ]]; then
        local MATCHING=$(echo "$REORDERED_FILES" | grep -F "$HISTORY_LINE")
        local NON_MATCHING=$(echo "$REORDERED_FILES" | grep -vF "$HISTORY_LINE")
        if [[ -n "$MATCHING" ]]; then
          REORDERED_FILES=$(printf "%s\n%s" "$MATCHING" "$NON_MATCHING")
        fi
      fi
    done < "$MARKDOWN_EDIT_LAST_FILES_PATH"
  fi

  # Show fzf prompt
  local SELECTED=$(echo "$REORDERED_FILES" | fzf --prompt="Select markdown/txt file to append fragment into: ")

  if [[ -z "$SELECTED" ]]; then
    echoerr "No file selected"
    exit 200
  fi

  # Return relative path (remove leading ./)
  echo "${SELECTED#./}"
}

function add_modified_file_to_history {
  local FILEPATH="$1"

  # Create history file if it doesn't exist
  if [[ ! -f "$MARKDOWN_EDIT_LAST_FILES_PATH" ]]; then
    touch "$MARKDOWN_EDIT_LAST_FILES_PATH"
  fi

  # Remove the filepath if it already exists (to avoid duplicates)
  CURRENT_HISTORY_CONTENTS=$(echo "$CURRENT_HISTORY_CONTENTS" | grep -vF "$FILEPATH")

  # Read current contents, keep only last MAX_LINES-1
  local CURRENT_HISTORY_CONTENTS=$(cat "$MARKDOWN_EDIT_LAST_FILES_PATH" | tail -n $((MAX_HISTORY_LINES - 1)))

  # Write back and append new filepath
  echo "$CURRENT_HISTORY_CONTENTS" > "$MARKDOWN_EDIT_LAST_FILES_PATH"
  echo "$FILEPATH" >> "$MARKDOWN_EDIT_LAST_FILES_PATH"
}

CLIPBOARD_MODE=
TARGET_FILENAME=

while [[ $# -gt 0 ]]; do
    OPTION="$1"

    case "$OPTION" in
        -c|--clipboard)
            CLIPBOARD_MODE=1
        ;;
        -h|--help)
            help_exit
        ;;
        *)
          if [[ "${OPTION:0:1}" == "-" ]]; then
            echoerr "Error: unknown option \"$OPTION\""
            help_exit
          else
            break
          fi
        ;;
    esac
    shift
done

# Get TARGET_FILENAME from argument or fzf
TARGET_FILENAME="$1"

# Read input fragment
if [[ "$CLIPBOARD_MODE" -eq 1 ]]; then
  FRAGMENT=$(xclip -selection c -o)
else
  if [ ! -t 0 ]; then
    FRAGMENT=$(< /dev/stdin)
  else
    echoerr "Error: pipe input is empty. Use --clipboard flag if you want to get fragment from clipboard"
    help_exit 1
  fi
fi


if [[ -z "$TARGET_FILENAME" ]]; then
  # cd to TX_DIR for fzf selection (relative paths)
  TX_DIR="$HOME/tx"
  cd "$TX_DIR" || {
    echoerr "Error: directory $TX_DIR not found"
    exit 300
  }

  TARGET_FILENAME=$(select_file_via_fzf)
  if [[ -z "$TARGET_FILENAME" ]]; then
    echoerr "Error: no file selected. Exitting"
    exit 350
  fi
  echo "You selected \"$TARGET_FILENAME\""
fi

# Check if file exists
if [[ ! -f "$TARGET_FILENAME" ]]; then
  echoerr "Error: file \"$TARGET_FILENAME\" does not exist"
  exit 400
fi

# Remove empty lines at the beginning and end
FRAGMENT=$(echo "$FRAGMENT" | sed -e :a -e '/^\n*$/{$d;N;ba' -e '}')

# Find FIRST_URL in the fragment
FIRST_URL=$(echo "$FRAGMENT" | grep -oE 'https?://[^ ]+' | head -n 1)

# If fragment doesn't start with '#', process it through markdown-fix-header
if [[ "${FRAGMENT:0:1}" != "#" ]]; then
  # Save to clipboard
  echo "$FRAGMENT" | xclip -selection c

  # Pass through markdown-fix-header
  # FRAGMENT_FIXED_HEADER=$($SCRIPT_DIR/markdown-fix-header -c <<< "$FRAGMENT")
  FRAGMENT_FIXED_HEADER=$(echo -n -e "$FRAGMENT" | $SCRIPT_DIR/markdown-fix-header --force)

  if [[ $? -ne 0 ]]; then
    echoerr "Warning: markdown-fix-header failed"
    # exit 200
  else
    FRAGMENT="$FRAGMENT_FIXED_HEADER"
  fi

  # Re-read from clipboard after markdown-fix-header processing
  # FRAGMENT=$(xclip -selection c -o)

  # Find FIRST_URL again after processing
  FIRST_URL=$(echo "$FRAGMENT" | grep -oE 'https?://[^ ]+' | head -n 1)
fi

# Check if FIRST_URL already exists in target file
if [[ -n "$FIRST_URL" ]]; then
  if grep -qF "$FIRST_URL" "$TARGET_FILENAME"; then
    ask_continue "add fragment with existing URL \"$FIRST_URL\"" > /dev/null
  fi
fi

# Write fragment to file
# Remove trailing empty lines using for loop (1 to 10 iterations)
for TRAILING_CHAR_NUM in $(seq 1 10); do
  FILE_LAST_SYMBOL=$(tail -c 1 "$TARGET_FILENAME")
  if [[ "$FILE_LAST_SYMBOL" != "" ]]; then
    break
  fi
  # Remove last character (empty line)
  truncate -s -1 "$TARGET_FILENAME"
done

# Add new line and fragment
echo -e "\n" >> "$TARGET_FILENAME"
echo -n "$FRAGMENT" >> "$TARGET_FILENAME"

add_modified_file_to_history "$TARGET_FILENAME"

echo -e "\n=== Last $LAST_HEADERS_LINES_TO_SHOW: lines of \"$TARGET_FILENAME\" after fragment add: ==="
echo ". . . . ."

"$SCRIPT_DIR/markdown-headers.sh" -n "$TARGET_FILENAME" \
  | tail -n "$LAST_HEADERS_LINES_TO_SHOW"

echo -e "\n=== Last $LAST_LINES_TO_SHOW: lines of \"$TARGET_FILENAME\" after fragment add: ==="
echo ". . . . ."
tail -n "$LAST_LINES_TO_SHOW" "$TARGET_FILENAME"

"$SCRIPT_DIR/obsidian-open" "$TARGET_FILENAME"