@pithosai/pithosai 1.0.7 → 1.0.32

package/dist/pithosai-nl-fallback.sh

@@ -0,0 +1,456 @@
+# pithosai-nl-fallback.sh
+# Source from zsh or bash 4+ (do not execute):
+#   source /path/to/scripts/pithosai-nl-fallback.sh
+#
+# Unknown commands that look like natural language are classified by Node
+# (`nl-fallback-classify`: lookpath + tinyld) and forwarded to:
+#   pithosai --no-interactive --model krouter-rd "$input"
+#
+# The command_not_found path never rewrites the typed line. Interactive zsh
+# (ZLE) and bash 4+ (readline `bind -x`, on Ctrl-J) intercept before parse so
+# `for these files: …` can reach pithosai instead of a keyword parse error; the
+# bash handler rewrites READLINE_LINE in place and lets accept-line run it.
+# Real PATH commands skip that intercept.
+#
+# Env:
+#   PITHOSAI_NL_FALLBACK=0        disable all NL interception
+#   PITHOSAI_NL_ZLE=0             disable zsh/bash Enter intercept only
+#   PITHOSAI_NL_MODEL             default krouter-rd
+#   PITHOSAI_NL_CLASSIFY_LLM=0    skip the LLM step when shell-score < 50
+#   PITHOSAI_NL_DEBUG_LOG=1       print classify/LLM debug on stderr
+#   PITHOSAI_BIN                  default pithosai
+#   PITHOSAI_LANG                 override locale for messages and --lang
+#   PITHOSAI_NODE                 node binary (default: node)
+
+# Executed instead of sourced: print help and exit. `return` works only when sourced.
+if [ -n "${ZSH_VERSION:-}" ]; then
+  case "${ZSH_EVAL_CONTEXT:-}" in
+    *:file*) ;;
+    *)
+      echo "Source this file from bash or zsh (do not execute it): source scripts/pithosai-nl-fallback.sh" >&2
+      return 1 2>/dev/null || exit 1
+      ;;
+  esac
+  # zsh NOMATCH aborts on unmatched globs (`you?`, `*.xyz`) before command_not_found_handler.
+  setopt NO_NOMATCH
+  setopt NO_BAD_PATTERN
+elif [ -n "${BASH_VERSION:-}" ]; then
+  if [ "${BASH_SOURCE[0]}" = "$0" ]; then
+    echo "Source this file from bash or zsh (do not execute it): source scripts/pithosai-nl-fallback.sh" >&2
+    exit 1
+  fi
+else
+  echo "Source this file from bash or zsh (do not execute it): source scripts/pithosai-nl-fallback.sh" >&2
+  return 1 2>/dev/null || exit 1
+fi
+
+if [ -n "${_PITHOSAI_NL_FALLBACK_LOADED:-}" ]; then
+  :
+else
+  # Copy a pre-existing command-not-found handler so skips still use it.
+  _pithosai_copy_shell_function() {
+    local src="$1"
+    local dst="$2"
+    if [ -n "${ZSH_VERSION:-}" ]; then
+      if typeset -f "$src" >/dev/null 2>&1; then
+        functions -c "$src" "$dst"
+      fi
+    elif typeset -f "$src" >/dev/null 2>&1; then
+      eval "$(typeset -f "$src" | sed "1s/^${src}/${dst}/")"
+    fi
+  }
+
+  if [ -n "${ZSH_VERSION:-}" ]; then
+    if typeset -f command_not_found_handler >/dev/null 2>&1; then
+      _pithosai_copy_shell_function command_not_found_handler _pithosai_orig_cnf
+    fi
+  elif typeset -f command_not_found_handle >/dev/null 2>&1; then
+    _pithosai_copy_shell_function command_not_found_handle _pithosai_orig_cnf
+  fi
+
+  _PITHOSAI_NL_FALLBACK_LOADED=1
+fi
+
+# Directory of this sourced script (zsh %x / bash BASH_SOURCE).
+pithosai_nl_fallback_script_dir() {
+  if [ -n "${ZSH_VERSION:-}" ]; then
+    local file="${(%):-%x}"
+    printf '%s' "${file:A:h}"
+    return 0
+  fi
+  (cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
+}
+
+# Path to the Node classifier (dist bundle, or src when sourcing from the repo).
+pithosai_nl_fallback_classify_js() {
+  local dir
+  dir="$(pithosai_nl_fallback_script_dir)"
+  if [ -f "$dir/nl-fallback-classify.mjs" ]; then
+    printf '%s' "$dir/nl-fallback-classify.mjs"
+    return 0
+  fi
+  if [ -f "$dir/../src/cli/shared/nl-fallback/nl-fallback-classify.js" ]; then
+    printf '%s' "$dir/../src/cli/shared/nl-fallback/nl-fallback-classify.js"
+    return 0
+  fi
+  return 1
+}
+
+# Map locale env to a pithosai --lang code (en, zh, zh-Hant, …).
+# Used when Node is unavailable. LANG=en_US.UTF-8 → en; zh_CN → zh; zh_TW → zh-Hant.
+pithosai_nl_fallback_lang() {
+  local raw="${PITHOSAI_LANG:-${LC_ALL:-${LC_MESSAGES:-${LANG:-}}}}"
+  raw="${raw%%.*}"
+  case "$raw" in
+    zh_TW|zh-TW|zh_HK|zh-HK|zh-Hant|zh_Hant|zh-hant|zh_hant)
+      echo "zh-Hant"
+      return 0
+      ;;
+  esac
+  raw="${raw%%_*}"
+  raw="${raw%%-*}"
+  case "$raw" in
+    zh) echo "zh" ;;
+    ja) echo "ja" ;;
+    ko) echo "ko" ;;
+    ar) echo "ar" ;;
+    de) echo "de" ;;
+    es) echo "es" ;;
+    fr) echo "fr" ;;
+    it) echo "it" ;;
+    nl) echo "nl" ;;
+    pt) echo "pt" ;;
+    en) echo "en" ;;
+    *) echo "en" ;;
+  esac
+}
+
+# Print a localized hook message (pithosaiMissing, classifyUnavailable, or sourceOnly).
+pithosai_nl_fallback_msg() {
+  local key="$1"
+  local input="${2:-}"
+  local lang
+  lang="$(pithosai_nl_fallback_lang)"
+  local msg=""
+  case "$key" in
+    pithosaiMissing)
+      case "$lang" in
+        zh) msg="未找到 pithosai。请安装 pithosai 或设置 PITHOSAI_BIN。" ;;
+        zh-Hant) msg="找不到 pithosai。請安裝 pithosai 或設定 PITHOSAI_BIN。" ;;
+        ja) msg="pithosai が見つかりません。インストールするか PITHOSAI_BIN を設定してください。" ;;
+        ko) msg="pithosai를 찾을 수 없습니다. pithosai를 설치하거나 PITHOSAI_BIN을 설정하세요." ;;
+        de) msg="pithosai nicht gefunden. Installieren Sie pithosai oder setzen Sie PITHOSAI_BIN." ;;
+        fr) msg="pithosai introuvable. Installez pithosai ou définissez PITHOSAI_BIN." ;;
+        es) msg="No se encontró pithosai. Instale pithosai o defina PITHOSAI_BIN." ;;
+        it) msg="pithosai non trovato. Installa pithosai o imposta PITHOSAI_BIN." ;;
+        nl) msg="pithosai niet gevonden. Installeer pithosai of zet PITHOSAI_BIN." ;;
+        pt) msg="pithosai não encontrado. Instale pithosai ou defina PITHOSAI_BIN." ;;
+        ar) msg="لم يتم العثور على pithosai. ثبّت pithosai أو عيّن PITHOSAI_BIN." ;;
+        *) msg="pithosai not found. Install pithosai or set PITHOSAI_BIN." ;;
+      esac
+      ;;
+    classifyUnavailable)
+      case "$lang" in
+        zh) msg="pithosai nl-fallback：无法运行分类器（需要 node 和 nl-fallback-classify）。" ;;
+        zh-Hant) msg="pithosai nl-fallback：無法執行分類器（需要 node 和 nl-fallback-classify）。" ;;
+        ja) msg="pithosai nl-fallback: 分類器を実行できません（node と nl-fallback-classify が必要です）。" ;;
+        ko) msg="pithosai nl-fallback: 분류기를 실행할 수 없습니다(node 및 nl-fallback-classify 필요)." ;;
+        de) msg="pithosai nl-fallback: Klassifizierer nicht verfügbar (node und nl-fallback-classify erforderlich)." ;;
+        fr) msg="pithosai nl-fallback : classifieur indisponible (node et nl-fallback-classify requis)." ;;
+        es) msg="pithosai nl-fallback: clasificador no disponible (se necesita node y nl-fallback-classify)." ;;
+        it) msg="pithosai nl-fallback: classificatore non disponibile (servono node e nl-fallback-classify)." ;;
+        nl) msg="pithosai nl-fallback: classifier niet beschikbaar (node en nl-fallback-classify vereist)." ;;
+        pt) msg="pithosai nl-fallback: classificador indisponível (é preciso node e nl-fallback-classify)." ;;
+        ar) msg="pithosai nl-fallback: المصنّف غير متاح (يلزم node و nl-fallback-classify)." ;;
+        *) msg="pithosai nl-fallback: classifier unavailable (need node + nl-fallback-classify)." ;;
+      esac
+      ;;
+    sourceOnly)
+      case "$lang" in
+        zh) msg="请在 bash 或 zsh 中 source 此文件（不要直接执行）：source scripts/pithosai-nl-fallback.sh" ;;
+        zh-Hant) msg="請在 bash 或 zsh 中 source 此檔案（不要直接執行）：source scripts/pithosai-nl-fallback.sh" ;;
+        ja) msg="bash または zsh から source してください（直接実行しないでください）: source scripts/pithosai-nl-fallback.sh" ;;
+        ko) msg="bash 또는 zsh에서 이 파일을 source 하세요(직접 실행하지 마세요): source scripts/pithosai-nl-fallback.sh" ;;
+        de) msg="Diese Datei in bash oder zsh sourcen (nicht ausführen): source scripts/pithosai-nl-fallback.sh" ;;
+        fr) msg="Sourcez ce fichier depuis bash ou zsh (ne l'exécutez pas) : source scripts/pithosai-nl-fallback.sh" ;;
+        es) msg="Haga source de este archivo en bash o zsh (no lo ejecute): source scripts/pithosai-nl-fallback.sh" ;;
+        it) msg="Esegui source di questo file da bash o zsh (non eseguirlo): source scripts/pithosai-nl-fallback.sh" ;;
+        nl) msg="Source dit bestand in bash of zsh (niet uitvoeren): source scripts/pithosai-nl-fallback.sh" ;;
+        pt) msg="Faça source deste ficheiro no bash ou zsh (não o execute): source scripts/pithosai-nl-fallback.sh" ;;
+        ar) msg="قم بعمل source لهذا الملف من bash أو zsh (لا تشغّله): source scripts/pithosai-nl-fallback.sh" ;;
+        *) msg="Source this file from bash or zsh (do not execute it): source scripts/pithosai-nl-fallback.sh" ;;
+      esac
+      ;;
+  esac
+  printf '%s\n' "${msg//\{\{input\}\}/$input}"
+}
+
+# Return 0 when NL fallback env is enabled.
+pithosai_nl_fallback_is_enabled() {
+  local disabled
+  disabled="$(printf '%s' "${PITHOSAI_NL_FALLBACK:-}" | tr '[:upper:]' '[:lower:]')"
+  case "$disabled" in
+    0|false|no|off) return 1 ;;
+  esac
+  return 0
+}
+
+# Print a debug line on stderr when PITHOSAI_NL_DEBUG_LOG=1.
+pithosai_nl_debug() {
+  local flag
+  flag="$(printf '%s' "${PITHOSAI_NL_DEBUG_LOG:-}" | tr '[:upper:]' '[:lower:]')"
+  case "$flag" in
+    1|true|yes|on) printf '[pithosai-nl] %s\n' "$*" >&2 ;;
+  esac
+}
+
+# Run pithosai --no-interactive with the reconstructed natural-language input.
+# $1 = input line, $2 = optional --lang code from the Node classifier.
+pithosai_nl_fallback_run() {
+  local input="$1"
+  local lang="${2:-}"
+  local bin="${PITHOSAI_BIN:-pithosai}"
+  local model="${PITHOSAI_NL_MODEL:-krouter-rd}"
+  if [ -z "$lang" ]; then
+    lang="$(pithosai_nl_fallback_lang)"
+  fi
+  if [ -x "$bin" ] || command -v "$bin" >/dev/null 2>&1; then
+    "$bin" --no-interactive --model "$model" --lang "$lang" "$input"
+    return $?
+  fi
+  pithosai_nl_fallback_msg pithosaiMissing >&2
+  echo "command not found: ${bin}" >&2
+  return 127
+}
+
+# Ask Node whether this line is NL. Prints lang on stdout when it should run.
+# Exit 0 = run, 1 = skip, 2 = classifier error.
+pithosai_nl_fallback_classify() {
+  local input="$1"
+  local js nodebin
+  js="$(pithosai_nl_fallback_classify_js)" || return 2
+  nodebin="${PITHOSAI_NODE:-node}"
+  if ! command -v "$nodebin" >/dev/null 2>&1; then
+    return 2
+  fi
+  "$nodebin" "$js" "$input"
+}
+
+# Shared bash/zsh command-not-found body.
+pithosai_handle_command_not_found() {
+  local cmd="$1"
+  local input="$*"
+  local lang code
+  pithosai_nl_debug "command_not_found cmd=${cmd} input=${input}"
+  if ! pithosai_nl_fallback_is_enabled; then
+    pithosai_nl_debug "fallback disabled"
+    pithosai_nl_fallback_not_found "$@"
+    return $?
+  fi
+  lang="$(pithosai_nl_fallback_classify "$input")"
+  code=$?
+  pithosai_nl_debug "classify exit=${code} lang=${lang}"
+  if [ "$code" -eq 0 ]; then
+    pithosai_nl_debug "run pithosai"
+    pithosai_nl_fallback_run "$input" "$lang"
+    return $?
+  fi
+  if [ "$code" -eq 2 ]; then
+    pithosai_nl_fallback_msg classifyUnavailable >&2
+  fi
+  pithosai_nl_debug "skip as command-not-found"
+  pithosai_nl_fallback_not_found "$@"
+}
+
+# Delegate to a previous handler, or print the usual command-not-found line.
+pithosai_nl_fallback_not_found() {
+  local cmd="$1"
+  if typeset -f _pithosai_orig_cnf >/dev/null 2>&1; then
+    _pithosai_orig_cnf "$@"
+    return $?
+  fi
+  if [ -n "${ZSH_VERSION:-}" ]; then
+    echo "zsh: command not found: $cmd" >&2
+  else
+    echo "bash: $cmd: command not found" >&2
+  fi
+  return 127
+}
+
+command_not_found_handler() {
+  pithosai_handle_command_not_found "$@"
+}
+
+command_not_found_handle() {
+  pithosai_handle_command_not_found "$@"
+}
+
+# Return 0 when the line is empty or only whitespace (no Node classify).
+pithosai_nl_line_is_blank() {
+  case "$1" in
+    "") return 0 ;;
+  esac
+  [ -z "${1##*[![:space:]]*}" ] && return 1
+  return 0
+}
+
+# First whitespace-separated token of a line.
+pithosai_nl_first_word() {
+  printf '%s\n' "$1" | awk '{ print $1; exit }'
+}
+
+# Return 0 for shell keywords that users also type as English (`for these files:`).
+pithosai_nl_is_reserved_word() {
+  local w
+  w="$(printf '%s' "$1" | tr '[:upper:]' '[:lower:]')"
+  w="${w%%[^a-z]*}"
+  case "$w" in
+    for|do|if|while|until|case|then|else|elif|fi|done|in|select) return 0 ;;
+  esac
+  return 1
+}
+
+# Return 0 when Enter should pass the line through to the shell (no classify).
+pithosai_nl_should_passthrough_line() {
+  local line="$1"
+  local first
+  if pithosai_nl_line_is_blank "$line"; then
+    return 0
+  fi
+  first="$(pithosai_nl_first_word "$line")"
+  if [ -n "$first" ] && ! pithosai_nl_is_reserved_word "$first"; then
+    if command -v -- "$first" >/dev/null 2>&1; then
+      return 0
+    fi
+  fi
+  return 1
+}
+
+# Return 0 when the zsh/bash Enter intercept is enabled.
+pithosai_nl_zle_is_enabled() {
+  local zle_off
+  zle_off="$(printf '%s' "${PITHOSAI_NL_ZLE:-}" | tr '[:upper:]' '[:lower:]')"
+  case "$zle_off" in
+    0|false|no|off) return 1 ;;
+  esac
+  pithosai_nl_fallback_is_enabled
+}
+
+# Classify and run pithosai when the line is NL. Return 0 if pithosai ran.
+pithosai_nl_try_run_line() {
+  local line="$1"
+  local lang code
+  pithosai_nl_debug "try_run_line input=${line}"
+  lang="$(pithosai_nl_fallback_classify "$line")"
+  code=$?
+  pithosai_nl_debug "try_run_line classify exit=${code} lang=${lang}"
+  if [ "$code" -eq 2 ]; then
+    pithosai_nl_fallback_msg classifyUnavailable >&2
+  fi
+  if [ "$code" -eq 0 ]; then
+    pithosai_nl_fallback_run "$line" "$lang"
+    return 0
+  fi
+  return 1
+}
+
+# zsh: intercept Enter before parse. Never assign BUFFER to a pithosai CLI string.
+if [ -n "${ZSH_VERSION:-}" ]; then
+  pithosai_nl_accept_line() {
+    local line="${BUFFER}"
+    pithosai_nl_debug "zle accept-line input=${line}"
+    if ! pithosai_nl_zle_is_enabled; then
+      pithosai_nl_debug "zle intercept disabled"
+      zle .accept-line
+      return
+    fi
+    if pithosai_nl_should_passthrough_line "$line"; then
+      pithosai_nl_debug "zle passthrough"
+      zle .accept-line
+      return
+    fi
+    # Unfreeze the line editor so classify/LLM stderr can paint (otherwise Enter looks hung).
+    zle -I
+    print
+    if pithosai_nl_try_run_line "$line"; then
+      pithosai_nl_debug "zle run pithosai"
+      print -s -- "$line"
+      BUFFER=""
+      zle .reset-prompt
+      return 0
+    fi
+    zle .accept-line
+  }
+  if [[ -o interactive ]]; then
+    zle -N pithosai_nl_accept_line 2>/dev/null || true
+    zle -N accept-line pithosai_nl_accept_line 2>/dev/null || true
+  fi
+fi
+
+# Build the quoted pithosai CLI invocation for an NL line so the native
+# accept-line (not the readline handler) runs it. $1 = line, $2 = lang code.
+pithosai_nl_bash_cli_command() {
+  local input="$1"
+  local lang="${2:-}"
+  local bin="${PITHOSAI_BIN:-pithosai}"
+  local model="${PITHOSAI_NL_MODEL:-krouter-rd}"
+  if [ -z "$lang" ]; then
+    lang="$(pithosai_nl_fallback_lang)"
+  fi
+  printf '%q --no-interactive --model %q --lang %q %q' \
+    "$bin" "$model" "$lang" "$input"
+}
+
+# bash 4+: intercept a key via readline bind -x and REWRITE the line in place.
+#
+# A bind -x handler runs while readline does not own the terminal, so it must
+# never eval a command, spawn a child that reads stdin, or printf a newline:
+# doing so desyncs readline and breaks TAB completion afterwards. Instead the
+# handler only rewrites READLINE_LINE and returns.
+#
+# bind -x ALONE does NOT submit the line — it replaces the default accept-line
+# binding. To submit the rewritten line in one keypress we use a two-key macro:
+#   bind -x '"\C-x\C-p": handler'       (shell command)
+#   bind '"\C-j": "\C-x\C-p\C-m"'    (macro: handler + accept-line)
+# Pressing Ctrl-J dispatches \C-x\C-p (runs handler, rewrites), then \C-m
+# (accept-line executes the rewritten line) — all with readline in control.
+#
+# Note: bash cannot intercept plain Enter (\C-m is consumed by accept-line and
+# never reaches a bind -x macro). Lines that fail a shell parse error
+# (`for these files: ...`) never reach the command-not-found path either;
+# only \C-j or the command-not-found handle can recover them.
+if [ -n "${BASH_VERSION:-}" ]; then
+  pithosai_nl_bash_accept_line() {
+    local line="${READLINE_LINE}"
+    local lang code cmd
+    pithosai_nl_debug "bash accept-line input=${line}"
+    if pithosai_nl_zle_is_enabled && ! pithosai_nl_should_passthrough_line "$line"; then
+      lang="$(pithosai_nl_fallback_classify "$line")"
+      code=$?
+      pithosai_nl_debug "bash classify exit=${code} lang=${lang}"
+      if [ "$code" -eq 0 ]; then
+        cmd="$(pithosai_nl_bash_cli_command "$line" "$lang")"
+        pithosai_nl_debug "bash rewrite to pithosai"
+        READLINE_LINE="$cmd"
+        READLINE_POINT=${#cmd}
+        return 0
+      fi
+      if [ "$code" -eq 2 ]; then
+        pithosai_nl_fallback_msg classifyUnavailable >&2
+      fi
+      pithosai_nl_debug "bash passthrough (classify miss)"
+    fi
+    # No rewrite happened: restore the buffer exactly as readline gave it, so a
+    # skipped line (PATH command, valid for/if one-liner, empty line) runs
+    # unchanged. Set both vars explicitly because a previous handler call may
+    # have left a rewritten buffer in place.
+    READLINE_LINE="$line"
+    READLINE_POINT=${#line}
+    return 0
+  }
+
+  if [[ $- == *i* ]] && [ "${BASH_VERSINFO[0]}" -ge 4 ]; then
+    bind -x '"\C-x\C-p": pithosai_nl_bash_accept_line' 2>/dev/null || true
+    bind '"\C-j": "\C-x\C-p\C-m"' 2>/dev/null || true
+  fi
+fi