#!/bin/sh

set -u

SCRIPT_DIR=$(CDPATH=; cd "$(dirname "$0")" && pwd) || exit 1
PACKAGE_SOURCE=${PI_PACKAGE_SOURCE:-git:git@bitbucket.org:siakitem/my-pi.git}
POWERLEVEL10K_PRESET_FILE=$SCRIPT_DIR/config/p10k.zsh
CODEGRAPH_INSTALL_URL=https://raw.githubusercontent.com/colbymchenry/codegraph/main/install.sh
KITTY_INSTALL_URL=https://sw.kovidgoyal.net/kitty/installer.sh
OH_MY_ZSH_INSTALL_URL=https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh
POWERLEVEL10K_REPOSITORY=https://github.com/romkatv/powerlevel10k.git
ZSH_AUTOSUGGESTIONS_REPOSITORY=https://github.com/zsh-users/zsh-autosuggestions.git
ZSH_SYNTAX_HIGHLIGHTING_REPOSITORY=https://github.com/zsh-users/zsh-syntax-highlighting.git
install_failed=0
zshrc_backup_done=0

say() {
	printf '%s\n' "$*"
}

warn() {
	printf '警告: %s\n' "$*" >&2
}

ask_yes_no() {
	prompt=$1
	printf '%s [y/N] ' "$prompt"
	if ! IFS= read -r answer; then
		answer=n
	fi
	case $answer in
		y|Y|yes|YES|Yes) return 0 ;;
		*) return 1 ;;
	esac
}

ask_to_install() {
	ask_yes_no "未检测到 $1。是否现在安装？"
}

ask_to_enable() {
	ask_yes_no "尚未启用 $1。是否现在安装所需文件并启用？"
}

require_brew() {
	if command -v brew >/dev/null 2>&1; then
		return 0
	fi

	warn "自动安装该依赖需要 Homebrew，但当前 PATH 中没有 brew。"
	warn "请先安装 Homebrew（https://brew.sh/），然后重新运行本脚本。"
	return 1
}

record_install_failure() {
	warn "$1 安装失败；其余依赖仍会继续检查。"
	install_failed=1
}

java_major_version() {
	java_version=$(java -version 2>&1 | awk -F '"' 'NR == 1 { print $2 }')
	case $java_version in
		1.*) java_version=${java_version#1.} ;;
	esac
	java_major=${java_version%%.*}
	case $java_major in
		''|*[!0-9]*) return 1 ;;
		*) printf '%s\n' "$java_major" ;;
	esac
}

install_codegraph() {
	if ! command -v curl >/dev/null 2>&1; then
		warn "安装 CodeGraph 需要 curl。"
		return 1
	fi
	curl -fsSL "$CODEGRAPH_INSTALL_URL" | sh || return 1

	# 官方安装器默认使用 ~/.local/bin；让本次检查立即看到新命令。
	if [ -d "$HOME/.local/bin" ]; then
		PATH="$HOME/.local/bin:$PATH"
		export PATH
	fi
	command -v codegraph >/dev/null 2>&1
}

install_kotlin_lsp() {
	require_brew || return 1
	brew install JetBrains/utils/kotlin-lsp || return 1
	command -v kotlin-lsp >/dev/null 2>&1
}

install_java_21() {
	require_brew || return 1
	brew install openjdk@21 || return 1
	java_home=$(brew --prefix openjdk@21 2>/dev/null) || return 1
	PATH="$java_home/bin:$PATH"
	export PATH
	installed_java_major=$(java_major_version || true)
	if [ -z "$installed_java_major" ] || [ "$installed_java_major" -lt 21 ]; then
		return 1
	fi
	say "已在本次安装流程中使用 $java_home/bin。若新终端仍找不到 Java，请把该目录加入 PATH。"
}

install_jdtls() {
	require_brew || return 1
	brew install jdtls || return 1
	command -v jdtls >/dev/null 2>&1
}

detect_kitty() {
	kitty_bin=
	if command -v kitty >/dev/null 2>&1; then
		kitty_bin=$(command -v kitty)
		return 0
	fi
	for candidate in /Applications/kitty.app/Contents/MacOS/kitty "$HOME/.local/kitty.app/bin/kitty"; do
		if [ -x "$candidate" ]; then
			kitty_bin=$candidate
			return 0
		fi
	done
	return 1
}

install_kitty() {
	case $(uname -s) in
		Darwin)
			require_brew || return 1
			brew install --cask kitty || return 1
			;;
		*)
			if ! command -v curl >/dev/null 2>&1; then
				warn "安装 Kitty 需要 curl。"
				return 1
			fi
			curl -L "$KITTY_INSTALL_URL" | sh /dev/stdin || return 1
			;;
	esac
	detect_kitty
}

resolve_kitty_config_dir() {
	kitty_config_dir=${KITTY_CONFIG_DIRECTORY:-${XDG_CONFIG_HOME:-$HOME/.config}/kitty}
}

kitty_solarized_dark_configured() {
	resolve_kitty_config_dir
	[ -f "$kitty_config_dir/current-theme.conf" ] || return 1
	[ -f "$kitty_config_dir/kitty.conf" ] || return 1
	grep -F "Solarized Dark" "$kitty_config_dir/current-theme.conf" >/dev/null 2>&1 || return 1
	grep -E '^[[:space:]]*include[[:space:]]+([^#[:space:]]*/)?current-theme\.conf([[:space:]]*(#.*)?)?$' \
		"$kitty_config_dir/kitty.conf" >/dev/null 2>&1
}

configure_kitty_solarized_dark() {
	"$kitty_bin" +kitten themes --reload-in=none "Solarized Dark"
}

resolve_zsh_paths() {
	oh_my_zsh_dir=${ZSH:-$HOME/.oh-my-zsh}
	zsh_custom_dir=${ZSH_CUSTOM:-$oh_my_zsh_dir/custom}
	zshrc_file=${ZDOTDIR:-$HOME}/.zshrc
}

oh_my_zsh_installed() {
	resolve_zsh_paths
	[ -f "$oh_my_zsh_dir/oh-my-zsh.sh" ]
}

install_oh_my_zsh() {
	if ! command -v curl >/dev/null 2>&1; then
		warn "安装 Oh My Zsh 需要 curl。"
		return 1
	fi
	if ! command -v git >/dev/null 2>&1; then
		warn "安装 Oh My Zsh 需要 git。"
		return 1
	fi
	resolve_zsh_paths
	oh_my_zsh_installer=$(curl -fsSL "$OH_MY_ZSH_INSTALL_URL") || return 1
	RUNZSH=no CHSH=no KEEP_ZSHRC=yes ZSH="$oh_my_zsh_dir" \
		sh -c "$oh_my_zsh_installer" "" --unattended --keep-zshrc || return 1
	oh_my_zsh_installed
}

prepare_zshrc() {
	resolve_zsh_paths
	zshrc_dir=${zshrc_file%/*}
	[ "$zshrc_dir" = "$zshrc_file" ] && zshrc_dir=.
	mkdir -p "$zshrc_dir" || return 1
	if [ ! -f "$zshrc_file" ]; then
		: > "$zshrc_file" || return 1
	fi
	if [ "$zshrc_backup_done" -eq 0 ]; then
		zshrc_backup="$zshrc_file.my-pi-backup-$(date +%Y%m%d%H%M%S)"
		cp "$zshrc_file" "$zshrc_backup" || return 1
		zshrc_backup_done=1
		say "已备份现有 Zsh 配置到 $zshrc_backup"
	fi
}

zshrc_has_oh_my_zsh_loader() {
	[ -f "$zshrc_file" ] || return 1
	grep -E '^[[:space:]]*(source|\.)[[:space:]].*oh-my-zsh\.sh' "$zshrc_file" >/dev/null 2>&1
}

ensure_oh_my_zsh_loader() {
	prepare_zshrc || return 1
	if zshrc_has_oh_my_zsh_loader; then
		return 0
	fi
	{
		printf '\n# Oh My Zsh (managed by my-pi install.sh)\n'
		printf 'export ZSH="%s"\n' "$oh_my_zsh_dir"
		printf '%s\n' "source \"\$ZSH/oh-my-zsh.sh\""
	} >> "$zshrc_file"
}

set_powerlevel10k_theme() {
	prepare_zshrc || return 1
	tmp_file=$(mktemp "${TMPDIR:-/tmp}/my-pi-zshrc.XXXXXX") || return 1
	if ! awk '
		BEGIN { inserted = 0 }
		/^[[:space:]]*ZSH_THEME[[:space:]]*=/ { next }
		!inserted && /^[[:space:]]*(source|\.)[[:space:]].*oh-my-zsh\.sh/ {
			print "ZSH_THEME=\"powerlevel10k/powerlevel10k\""
			inserted = 1
		}
		{ print }
		END {
			if (!inserted) print "ZSH_THEME=\"powerlevel10k/powerlevel10k\""
		}
	' "$zshrc_file" > "$tmp_file"; then
		rm -f "$tmp_file"
		return 1
	fi
	cp "$tmp_file" "$zshrc_file" || {
		rm -f "$tmp_file"
		return 1
	}
	rm -f "$tmp_file"
}

p10k_config_sourced() {
	[ -f "$zshrc_file" ] || return 1
	grep -E '^[[:space:]]*\[\[[[:space:]].*\.p10k\.zsh.*source[[:space:]].*\.p10k\.zsh' \
		"$zshrc_file" >/dev/null 2>&1
}

ensure_p10k_config_source() {
	if p10k_config_sourced; then
		return 0
	fi
	printf '\n# Powerlevel10k preset (managed by my-pi install.sh)\n[[ ! -f ~/.p10k.zsh ]] || source ~/.p10k.zsh\n' \
		>> "$zshrc_file"
}

configure_powerlevel10k() {
	resolve_zsh_paths
	powerlevel10k_dir=$zsh_custom_dir/themes/powerlevel10k
	if [ ! -f "$powerlevel10k_dir/powerlevel10k.zsh-theme" ]; then
		if ! command -v git >/dev/null 2>&1; then
			warn "安装 Powerlevel10k 需要 git。"
			return 1
		fi
		mkdir -p "$zsh_custom_dir/themes" || return 1
		git clone --depth=1 "$POWERLEVEL10K_REPOSITORY" "$powerlevel10k_dir" || return 1
	fi

	if [ ! -f "$HOME/.p10k.zsh" ]; then
		if [ ! -f "$POWERLEVEL10K_PRESET_FILE" ]; then
			warn "找不到仓库内置的 Powerlevel10k 配置：$POWERLEVEL10K_PRESET_FILE"
			return 1
		fi
		cp "$POWERLEVEL10K_PRESET_FILE" "$HOME/.p10k.zsh" || return 1
		say "已写入仓库内置的 Powerlevel10k Rainbow/ASCII 配置，首次启动不会进入配置向导。"
	fi
	set_powerlevel10k_theme || return 1
	ensure_oh_my_zsh_loader || return 1
	ensure_p10k_config_source
}

zsh_plugin_enabled() {
	plugin=$1
	[ -f "$zshrc_file" ] || return 1
	awk -v target="$plugin" '
		{
			line = $0
			sub(/[[:space:]]*#.*/, "", line)
			if (line ~ /^[[:space:]]*plugins[[:space:]]*[+]?=/) active = 1
			if (active) {
				gsub(/[()=+]/, " ", line)
				count = split(line, fields, /[[:space:]]+/)
				for (i = 1; i <= count; i++) if (fields[i] == target) found = 1
				if ($0 ~ /\)/) active = 0
			}
		}
		END { exit(found ? 0 : 1) }
	' "$zshrc_file"
}

add_zsh_plugin() {
	plugin=$1
	prepare_zshrc || return 1
	ensure_oh_my_zsh_loader || return 1
	tmp_file=$(mktemp "${TMPDIR:-/tmp}/my-pi-zshrc.XXXXXX") || return 1
	if ! awk -v plugin="$plugin" '
		BEGIN { added = 0 }
		!added && /^[[:space:]]*(source|\.)[[:space:]].*oh-my-zsh\.sh/ {
			print "plugins+=(" plugin ")"
			added = 1
		}
		{ print }
		END { if (!added) print "plugins+=(" plugin ")" }
	' "$zshrc_file" > "$tmp_file"; then
		rm -f "$tmp_file"
		return 1
	fi
	cp "$tmp_file" "$zshrc_file" || {
		rm -f "$tmp_file"
		return 1
	}
	rm -f "$tmp_file"
}

install_and_enable_zsh_plugin() {
	plugin=$1
	repository=${2:-}
	if [ -n "$repository" ] && [ ! -d "$zsh_custom_dir/plugins/$plugin" ]; then
		if ! command -v git >/dev/null 2>&1; then
			warn "安装 $plugin 需要 git。"
			return 1
		fi
		mkdir -p "$zsh_custom_dir/plugins" || return 1
		git clone --depth=1 "$repository" "$zsh_custom_dir/plugins/$plugin" || return 1
	fi
	add_zsh_plugin "$plugin"
}

check_zsh_plugin() {
	plugin=$1
	description=$2
	repository=${3:-}
	if zsh_plugin_enabled "$plugin"; then
		say "✓ $plugin 已启用"
	elif ask_to_enable "$description"; then
		install_and_enable_zsh_plugin "$plugin" "$repository" || record_install_failure "$plugin"
	else
		say "- 已跳过 ${plugin}。"
	fi
}

if ! command -v pi >/dev/null 2>&1; then
	warn "找不到 pi，无法安装组合包。请先安装 Pi 并确保 pi 位于 PATH。"
	exit 127
fi

say "正在安装 Pi 组合包：$PACKAGE_SOURCE"
if ! pi install "$PACKAGE_SOURCE"; then
	warn "pi install 执行失败，已停止后续依赖检查。"
	exit 1
fi

say "Pi 组合包安装完成，开始检查终端环境和机器级 CLI 依赖。"

kitty_ok=0
if detect_kitty; then
	kitty_ok=1
	say "✓ Kitty 已安装（${kitty_bin}）"
elif ask_to_install "Kitty 终端"; then
	if install_kitty; then
		kitty_ok=1
		say "✓ Kitty 安装完成"
	else
		record_install_failure "Kitty"
	fi
else
	say "- 已跳过 Kitty；不会检查 Solarized Dark 主题。"
fi

if [ "$kitty_ok" -eq 1 ]; then
	if kitty_solarized_dark_configured; then
		say "✓ Kitty 已配置 Solarized Dark"
	elif ask_yes_no "Kitty 尚未配置 Solarized Dark。是否现在配置？"; then
		configure_kitty_solarized_dark || record_install_failure "Kitty Solarized Dark 主题"
	else
		say "- 已跳过 Kitty Solarized Dark 主题。"
	fi
fi

oh_my_zsh_ok=0
if oh_my_zsh_installed; then
	oh_my_zsh_ok=1
	say "✓ Oh My Zsh 已安装（${oh_my_zsh_dir}）"
elif ask_to_install "Oh My Zsh"; then
	if install_oh_my_zsh; then
		oh_my_zsh_ok=1
		say "✓ Oh My Zsh 安装完成"
	else
		record_install_failure "Oh My Zsh"
	fi
else
	say "- 已跳过 Oh My Zsh；不会检查 Powerlevel10k 和 Zsh 插件。"
fi

if [ "$oh_my_zsh_ok" -eq 1 ]; then
	resolve_zsh_paths
	powerlevel10k_dir=$zsh_custom_dir/themes/powerlevel10k
	if [ -f "$powerlevel10k_dir/powerlevel10k.zsh-theme" ] && \
		grep -E "^[[:space:]]*ZSH_THEME[[:space:]]*=[[:space:]]*[\"']?powerlevel10k/powerlevel10k[\"']?" \
		"$zshrc_file" >/dev/null 2>&1 && [ -f "$HOME/.p10k.zsh" ] && p10k_config_sourced; then
		say "✓ Powerlevel10k 已安装并配置"
	elif ask_yes_no "Powerlevel10k 尚未完整配置。是否安装并应用仓库内置的 Rainbow/ASCII 配置？"; then
		configure_powerlevel10k || record_install_failure "Powerlevel10k"
	else
		say "- 已跳过 Powerlevel10k。"
	fi

	check_zsh_plugin "git" "git 插件" ""
	check_zsh_plugin "zsh-autosuggestions" "zsh-autosuggestions（命令自动建议）" \
		"$ZSH_AUTOSUGGESTIONS_REPOSITORY"
	check_zsh_plugin "zsh-syntax-highlighting" "zsh-syntax-highlighting（语法高亮）" \
		"$ZSH_SYNTAX_HIGHLIGHTING_REPOSITORY"
fi

if command -v codegraph >/dev/null 2>&1; then
	say "✓ codegraph 已安装"
elif ask_to_install "CodeGraph（提供 codegraph_explore）"; then
	install_codegraph || record_install_failure "CodeGraph"
else
	say "- 已跳过 CodeGraph；codegraph_explore 将不可用。"
fi

if command -v kotlin-lsp >/dev/null 2>&1; then
	say "✓ kotlin-lsp 已安装"
elif ask_to_install "kotlin-lsp（Kotlin LSP）"; then
	install_kotlin_lsp || record_install_failure "kotlin-lsp"
else
	say "- 已跳过 kotlin-lsp；Kotlin LSP 功能将不可用。"
fi

java_ok=0
if command -v java >/dev/null 2>&1; then
	java_major=$(java_major_version || true)
	if [ -n "$java_major" ] && [ "$java_major" -ge 21 ]; then
		java_ok=1
		say "✓ Java $java_major 已安装（满足 Java 21+）"
	fi
fi

if [ "$java_ok" -eq 0 ]; then
	if ask_to_install "Java 21+（JDT LS 运行时）"; then
		if install_java_21; then
			java_ok=1
		else
			record_install_failure "Java 21+"
		fi
	else
		say "- 已跳过 Java 21+；JDT LS 可能无法启动。"
	fi
fi

if command -v jdtls >/dev/null 2>&1; then
	say "✓ jdtls 已安装"
elif ask_to_install "jdtls（Java LSP）"; then
	install_jdtls || record_install_failure "jdtls"
else
	say "- 已跳过 jdtls；Java LSP 功能将不可用。"
fi

say "终端环境与依赖检查完成。RTK 命令改写默认关闭，因此未检查可选的 rtk CLI。"
exit "$install_failed"