shrinker-ai 0.3.2 → 0.4.0

package/integrations/macos/install.sh

@@ -13,7 +13,9 @@ SKIP_PROFILE=0
 SKIP_AGENT_RULES=0
 COPILOT_ONLY=0
 CLAUDE_ONLY=0
+TRACK_UNCOVERED=""
 PROFILE_PATH="${PROFILE_PATH:-$HOME/.zshrc}"
+CONFIG_PATH="${SHRINKER_CONFIG_PATH:-$HOME/.shrinker/config}"
 STEP="${SHRINKER_STEP_OFFSET:-0}"
 
 print_message() {
@@ -34,6 +36,8 @@ while [[ $# -gt 0 ]]; do
     --skip-link) SKIP_LINK=1 ;;
     --enable-profile-routing) ENABLE_PROFILE_ROUTING=1 ;;
     --skip-profile) SKIP_PROFILE=1 ;;
+    --enable-uncovered-tracking) TRACK_UNCOVERED=1 ;;
+    --disable-uncovered-tracking) TRACK_UNCOVERED=0 ;;
     --skip-agent-rules) SKIP_AGENT_RULES=1 ;;
     --copilot-only) COPILOT_ONLY=1 ;;
     --claude-only) CLAUDE_ONLY=1 ;;
@@ -46,6 +50,39 @@ done
 if (( ENABLE_PROFILE_ROUTING == 1 && SKIP_PROFILE == 1 )); then print_message "❌" "Use either --enable-profile-routing or --skip-profile, not both." >&2; exit 1; fi
 if (( COPILOT_ONLY == 1 && CLAUDE_ONLY == 1 )); then print_message "❌" "Use either --copilot-only or --claude-only, not both." >&2; exit 1; fi
 
+# Prompt only on a TTY; non-interactive installs keep tracking on and leave the shell profile alone.
+prompt_yes_no() {
+  local question="$1"
+  local default_answer="$2"
+  local reply
+  local hint="[y/N]"
+  [[ "$default_answer" == "1" ]] && hint="[Y/n]"
+  printf '   %s %s ' "$question" "$hint" > /dev/tty
+  IFS= read -r reply < /dev/tty || reply=""
+  case "$(printf '%s' "$reply" | tr '[:upper:]' '[:lower:]')" in
+    y|yes) echo 1 ;;
+    n|no) echo 0 ;;
+    *) echo "$default_answer" ;;
+  esac
+}
+
+interactive=0
+[[ -t 0 && -r /dev/tty ]] && interactive=1
+
+if [[ -z "$TRACK_UNCOVERED" ]]; then
+  if (( interactive )); then
+    print_message "❓" "Track uncovered commands? Records which unfiltered commands cost the most tokens."
+    TRACK_UNCOVERED="$(prompt_yes_no "Enable uncovered-command tracking?" 1)"
+  else
+    TRACK_UNCOVERED=1
+  fi
+fi
+
+if (( interactive && SKIP_PROFILE == 0 && ENABLE_PROFILE_ROUTING == 0 )); then
+  print_message "❓" "Route wrapped commands through shrinker automatically? Adds a source line to $PROFILE_PATH."
+  ENABLE_PROFILE_ROUTING="$(prompt_yes_no "Enable automatic shell routing?" 0)"
+fi
+
 if (( LOCAL == 0 )); then
   package_spec="$PACKAGE_NAME"
   [[ -z "$VERSION" ]] || package_spec="$PACKAGE_NAME@$VERSION"
@@ -60,6 +97,7 @@ if (( LOCAL == 0 )); then
     $( (( SKIP_AGENT_RULES )) && echo --skip-agent-rules ) \
     $( (( COPILOT_ONLY )) && echo --copilot-only ) \
     $( (( CLAUDE_ONLY )) && echo --claude-only ) \
+    $( (( TRACK_UNCOVERED )) && echo --enable-uncovered-tracking || echo --disable-uncovered-tracking ) \
     --profile-path "$PROFILE_PATH"
   if [[ -n "${BASH_SOURCE[0]:-}" && "${BASH_SOURCE[0]}" != "$0" ]]; then
     return 0
@@ -123,6 +161,20 @@ add_profile_integration() {
   fi
 }
 
+set_config_value() {
+  local key="$1"
+  local value="$2"
+  mkdir -p "$(dirname "$CONFIG_PATH")"
+  chmod 700 "$(dirname "$CONFIG_PATH")" 2>/dev/null || true
+  touch "$CONFIG_PATH"
+  if grep -Eq "^[[:space:]]*$key[[:space:]]*=" "$CONFIG_PATH"; then
+    grep -Ev "^[[:space:]]*$key[[:space:]]*=" "$CONFIG_PATH" > "$CONFIG_PATH.tmp" || true
+    mv "$CONFIG_PATH.tmp" "$CONFIG_PATH"
+  fi
+  printf '%s=%s\n' "$key" "$value" >> "$CONFIG_PATH"
+  print_message "⚙️" "Set $key=$value in: $CONFIG_PATH"
+}
+
 print_message "📦" "Installing shrinker from: $REPO_ROOT"
 pushd "$REPO_ROOT" >/dev/null
 if (( SKIP_NPM_INSTALL == 0 )); then
@@ -131,7 +183,7 @@ if (( SKIP_NPM_INSTALL == 0 )); then
 fi
 if (( SKIP_BUILD == 0 )); then
   print_message "🏗️" "Building shrinker..."
-  npm run build --silent
+  SHRINKER_BUILD_QUIET=1 npm run build --silent
 fi
 if (( SKIP_LINK == 0 )); then
   print_message "🔗" "Linking shrinker globally..."
@@ -139,6 +191,7 @@ if (( SKIP_LINK == 0 )); then
 fi
 popd >/dev/null
 
+set_config_value "SHRINKER_TRACK_UNCOVERED" "$TRACK_UNCOVERED"
 if (( SKIP_PROFILE == 0 && ENABLE_PROFILE_ROUTING == 1 )); then add_profile_integration "$PROFILE_PATH"; fi
 if (( SKIP_AGENT_RULES == 0 )); then
   [[ -f "$TEMPLATE_PATH" ]] || { print_message "❌" "Agent rules template not found: $TEMPLATE_PATH" >&2; exit 1; }

package/integrations/macos/shrinker-profile.zsh

@@ -1,98 +1,165 @@
-typeset -gA SHRINK_RULES
-SHRINK_RULES=(
-  git "status diff log show reflog branch tag stash"
-  npm "test t install i ci ls list"
-  docker "ps logs images compose"
-  kubectl "get describe logs"
-  gh "pr issue run"
-  rg "*"
-  find "*"
-  tail "*"
-  cat "*"
-  ls "*"
-)
-
-typeset -gA SHRINK_OPTION_VALUE_FLAGS
-SHRINK_OPTION_VALUE_FLAGS=(
-  git "-C -c --git-dir --work-tree --namespace"
-  npm "--prefix --cache --registry --workspace --userconfig -w -C"
-  docker "-H --host --context --config"
-  kubectl "-n --namespace -o --output --context --kubeconfig --cluster --user"
-  gh "-R --repo"
-)
-
-_shrink_get_subcommand() {
-  local cmd="$1"
-  shift
-
-  local -a args
-  args=("$@")
-
-  local -a value_flags
-  value_flags=(${=SHRINK_OPTION_VALUE_FLAGS[$cmd]})
-
-  local i part
-  i=1
-  while (( i <= ${#args[@]} )); do
-    part="${args[$i]}"
-    if [[ -z "$part" ]]; then
-      (( i++ ))
-      continue
-    fi
-
-    if (( ${value_flags[(Ie)$part]} > 0 )); then
-      (( i += 2 ))
-      continue
-    fi
-
-    if [[ "$part" == -* ]]; then
-      (( i++ ))
-      continue
-    fi
-
-    print -r -- "${part:l}"
-    return 0
-  done
-
-  print -r -- ""
-}
-
-_shrink_should_route() {
-  local cmd="$1"
-  shift
-
-  local rules="${SHRINK_RULES[$cmd]}"
-  [[ -z "$rules" ]] && return 1
-  [[ "$rules" == *"*"* ]] && return 0
-
-  local subcommand
-  subcommand="$(_shrink_get_subcommand "$cmd" "$@")"
-  [[ -z "$subcommand" ]] && return 1
-
-  local -a allowlist
-  allowlist=(${=rules})
-  (( ${allowlist[(Ie)$subcommand]} > 0 ))
-}
-
-_shrink_invoke_or_native() {
-  local cmd="$1"
-  shift
-
-  if _shrink_should_route "$cmd" "$@" && command -v shrinker >/dev/null 2>&1; then
-    command shrinker "$cmd" "$@"
-    return $?
-  fi
-
-  command "$cmd" "$@"
-}
-
-git() { _shrink_invoke_or_native git "$@"; }
-npm() { _shrink_invoke_or_native npm "$@"; }
-docker() { _shrink_invoke_or_native docker "$@"; }
-kubectl() { _shrink_invoke_or_native kubectl "$@"; }
-gh() { _shrink_invoke_or_native gh "$@"; }
-rg() { _shrink_invoke_or_native rg "$@"; }
-find() { _shrink_invoke_or_native find "$@"; }
-tail() { _shrink_invoke_or_native tail "$@"; }
-cat() { _shrink_invoke_or_native cat "$@"; }
-ls() { _shrink_invoke_or_native ls "$@"; }
+typeset -gA SHRINK_RULES
+SHRINK_RULES=(
+  git "status diff log show reflog branch tag stash"
+  npm "test t install i ci ls list"
+  docker "ps logs images compose"
+  kubectl "get describe logs"
+  gh "pr issue run"
+  rg "*"
+  find "*"
+  tail "*"
+  cat "*"
+  ls "*"
+)
+
+typeset -gA SHRINK_OPTION_VALUE_FLAGS
+SHRINK_OPTION_VALUE_FLAGS=(
+  git "-C -c --git-dir --work-tree --namespace"
+  npm "--prefix --cache --registry --workspace --userconfig -w -C"
+  docker "-H --host --context --config"
+  kubectl "-n --namespace -o --output --context --kubeconfig --cluster --user"
+  gh "-R --repo"
+)
+
+# Resolved once at load so wrapped commands never pay for reading the config file.
+typeset -g SHRINK_TRACK_UNCOVERED_DEFAULT=""
+_shrink_load_config() {
+  local config="${SHRINKER_CONFIG_PATH:-$HOME/.shrinker/config}"
+  [[ -r "$config" ]] || return 0
+  local value
+  value="$(sed -n 's/^[[:space:]]*SHRINKER_TRACK_UNCOVERED[[:space:]]*=[[:space:]]*\([^#[:space:]]*\).*/\1/p' "$config" | tail -n 1)"
+  SHRINK_TRACK_UNCOVERED_DEFAULT="$value"
+}
+_shrink_load_config
+
+_shrink_uncovered_enabled() {
+  local value="${SHRINKER_TRACK_UNCOVERED:-$SHRINK_TRACK_UNCOVERED_DEFAULT}"
+  case "${value:l}" in
+    1|true|yes) return 0 ;;
+    *) return 1 ;;
+  esac
+}
+
+_shrink_get_subcommand() {
+  local cmd="$1"
+  shift
+
+  local -a args
+  args=("$@")
+
+  local -a value_flags
+  value_flags=(${=SHRINK_OPTION_VALUE_FLAGS[$cmd]})
+
+  local i part
+  i=1
+  while (( i <= ${#args[@]} )); do
+    part="${args[$i]}"
+    if [[ -z "$part" ]]; then
+      (( i++ ))
+      continue
+    fi
+
+    if (( ${value_flags[(Ie)$part]} > 0 )); then
+      (( i += 2 ))
+      continue
+    fi
+
+    if [[ "$part" == -* ]]; then
+      (( i++ ))
+      continue
+    fi
+
+    print -r -- "${part:l}"
+    return 0
+  done
+
+  print -r -- ""
+}
+
+_shrink_should_route() {
+  local cmd="$1"
+  shift
+
+  local rules="${SHRINK_RULES[$cmd]}"
+  [[ -z "$rules" ]] && return 1
+  [[ "$rules" == *"*"* ]] && return 0
+
+  local subcommand
+  subcommand="$(_shrink_get_subcommand "$cmd" "$@")"
+  [[ -z "$subcommand" ]] && return 1
+
+  local -a allowlist
+  allowlist=(${=rules})
+  (( ${allowlist[(Ie)$subcommand]} > 0 ))
+}
+
+_shrink_track_uncovered() {
+  local cmd="$1"
+  local bytes="$2"
+  local status_code="$3"
+  shift 3
+
+  _shrink_uncovered_enabled || return 0
+  command -v shrinker >/dev/null 2>&1 || return 0
+
+  local subcommand
+  subcommand="$(_shrink_get_subcommand "$cmd" "$@")"
+
+  local -a track_args
+  track_args=(track --executable "$cmd" --bytes "$bytes" --exit-code "$status_code")
+  [[ -n "$subcommand" ]] && track_args+=(--subcommand "$subcommand")
+
+  (command shrinker "${track_args[@]}" >/dev/null 2>&1 &)
+}
+
+_shrink_run_native_tracked() {
+  local cmd="$1"
+  shift
+
+  if ! _shrink_uncovered_enabled; then
+    command "$cmd" "$@"
+    return $?
+  fi
+
+  # Only measure output volume when stdout is redirected (the agent case). Interactive
+  # terminals keep the native command untouched so pagers and prompts still work.
+  if [[ -t 1 ]]; then
+    command "$cmd" "$@"
+    local status_code=$?
+    _shrink_track_uncovered "$cmd" 0 "$status_code" "$@"
+    return $status_code
+  fi
+
+  local capture
+  capture="$(mktemp -t shrinker-track)" || { command "$cmd" "$@"; return $?; }
+  command "$cmd" "$@" | tee "$capture"
+  local status_code=${pipestatus[1]}
+  local bytes
+  bytes="$(wc -c < "$capture" | tr -d ' ')"
+  rm -f "$capture"
+  _shrink_track_uncovered "$cmd" "${bytes:-0}" "$status_code" "$@"
+  return $status_code
+}
+
+_shrink_invoke_or_native() {
+  local cmd="$1"
+  shift
+
+  if _shrink_should_route "$cmd" "$@" && command -v shrinker >/dev/null 2>&1; then
+    command shrinker "$cmd" "$@"
+    return $?
+  fi
+
+  _shrink_run_native_tracked "$cmd" "$@"
+}
+
+git() { _shrink_invoke_or_native git "$@"; }
+npm() { _shrink_invoke_or_native npm "$@"; }
+docker() { _shrink_invoke_or_native docker "$@"; }
+kubectl() { _shrink_invoke_or_native kubectl "$@"; }
+gh() { _shrink_invoke_or_native gh "$@"; }
+rg() { _shrink_invoke_or_native rg "$@"; }
+find() { _shrink_invoke_or_native find "$@"; }
+tail() { _shrink_invoke_or_native tail "$@"; }
+cat() { _shrink_invoke_or_native cat "$@"; }
+ls() { _shrink_invoke_or_native ls "$@"; }

package/integrations/macos/uninstall.sh

@@ -7,7 +7,11 @@ SKIP_UNLINK=0
 SKIP_AGENT_RULES=0
 COPILOT_ONLY=0
 CLAUDE_ONLY=0
+PURGE_DATA=0
+DASHBOARD_PORT=4317
 PROFILE_PATH="${PROFILE_PATH:-$HOME/.zshrc}"
+CONFIG_PATH="${SHRINKER_CONFIG_PATH:-$HOME/.shrinker/config}"
+DATA_DIR="$(dirname "$CONFIG_PATH")"
 UNINSTALL_STEP=0
 
 print_step() {
@@ -25,6 +29,8 @@ while [[ $# -gt 0 ]]; do
     --profile-path) PROFILE_PATH="${2:-}"; shift ;;
     --install-dir) INSTALL_DIR="${2:-}"; shift ;;
     --remove-install-dir) REMOVE_INSTALL_DIR=1 ;;
+    --purge-data) PURGE_DATA=1 ;;
+    --port) DASHBOARD_PORT="${2:-}"; shift ;;
     *) echo "Unknown option: $1" >&2; exit 1 ;;
   esac
   shift
@@ -48,6 +54,38 @@ remove_profile_integration() {
   mv "$profile_file.tmp" "$profile_file"
 }
 
+# The dashboard runs as a detached daemon, so unlinking the package never stops it.
+stop_dashboard_server() {
+  local url="http://127.0.0.1:$DASHBOARD_PORT"
+  if command -v curl >/dev/null 2>&1; then
+    curl -fsS -m 3 -X POST "$url/__shrinker_shutdown" -o /dev/null 2>/dev/null || true
+    sleep 1
+  fi
+
+  local pid
+  pid="$(lsof -nP -iTCP:"$DASHBOARD_PORT" -sTCP:LISTEN -t 2>/dev/null | head -n 1 || true)"
+  if [[ -n "$pid" ]]; then
+    kill "$pid" 2>/dev/null || true
+    sleep 1
+    pid="$(lsof -nP -iTCP:"$DASHBOARD_PORT" -sTCP:LISTEN -t 2>/dev/null | head -n 1 || true)"
+    [[ -n "$pid" ]] && kill -9 "$pid" 2>/dev/null || true
+  fi
+}
+
+# Only drop installer-managed keys so hand-written settings survive.
+remove_managed_config() {
+  [[ -f "$CONFIG_PATH" ]] || return 0
+  grep -Ev '^[[:space:]]*SHRINKER_TRACK_UNCOVERED[[:space:]]*=' "$CONFIG_PATH" > "$CONFIG_PATH.tmp" || true
+  if [[ -s "$CONFIG_PATH.tmp" ]]; then
+    mv "$CONFIG_PATH.tmp" "$CONFIG_PATH"
+  else
+    rm -f "$CONFIG_PATH.tmp" "$CONFIG_PATH"
+  fi
+}
+
+print_step "🛑" "Stopping the dashboard server on port $DASHBOARD_PORT..."
+stop_dashboard_server
+
 if (( SKIP_UNLINK == 0 )); then
   print_step "🔗" "Unlinking shrinker globally..."
   npm unlink --silent --global shrinker-ai
@@ -62,6 +100,18 @@ if (( SKIP_AGENT_RULES == 0 )); then
 else
   print_step "⏭️" "Skipped managed agent rules."
 fi
+
+print_step "🔧" "Removing managed settings..."
+remove_managed_config
+rm -f "$DATA_DIR/dashboard.html"
+
+if (( PURGE_DATA == 1 )); then
+  rm -rf "$DATA_DIR"
+  print_step "🧹" "Removed local data: $DATA_DIR"
+elif [[ -d "$DATA_DIR" ]]; then
+  print_step "💾" "Kept saved stats in $DATA_DIR (use --purge-data to delete)."
+fi
+
 print_step "✅" "Uninstall complete."
 
 if (( REMOVE_INSTALL_DIR == 1 )) && [[ -d "$INSTALL_DIR" ]]; then