#!/bin/sh
# Zaivern Code ワンライナーインストーラ
#   curl -fsSL https://raw.githubusercontent.com/tacyan/zaivern-code/main/install.sh | sh
#
# やること:
#   1. OS/CPU を判定し、GitHub Releases のビルド済みバイナリを取得
#   2. リリースの checksums.txt と SHA-256 を突き合わせてから ~/.local/bin へ配置
#      (**検証できなければ展開も実行もせずに中止する = fail-closed**)
#   3. ビルド済みが無い環境や取得失敗時はソースからビルド
#      (Rust が無ければ rustup ごと非対話でセットアップ)
#
# 2回目以降の実行は「更新」として動作する:
#   最新版を取得して上書きし、PATH 上の別の場所(~/.cargo/bin 等)に残った
#   古い zai も同じバイナリで揃える(古い方が先に見つかって起動するのを防ぐ)
#
# 環境変数:
#   ZAI_INSTALL_DIR    ビルド済みバイナリの配置先 (既定: ~/.local/bin)
#   ZAI_FROM_SOURCE=1  常にソースビルドする
set -eu

REPO="tacyan/zaivern-code"
REPO_URL="https://github.com/$REPO"
REQUIRED_MINOR=88
INSTALL_DIR="${ZAI_INSTALL_DIR:-$HOME/.local/bin}"

say() { printf '\033[1;36m[zaivern-code]\033[0m %s\n' "$1"; }
err() { printf '\033[1;31m[zaivern-code]\033[0m %s\n' "$1" >&2; }

# --- 配布物の検証 (fail-closed) ----------------------------------------------
# 取得したアーカイブは **展開する前に** リリースの checksums.txt と突き合わせる。
# 展開してから確かめても、tar が中身を書き出した後では手遅れ (パス横断を含む)。
#
# 検証できない理由が何であれ (取得失敗・行が無い・道具が無い・不一致)、
# ここは必ず中止する。「確かめられなかったので、とりあえず入れた」は
# チェックサムを作っていないのと同じ意味しか持たない。
abort_unverified() {
    err ""
    err "⛔ 配布物を検証できなかったため中止しました: $1"
    err "   ダウンロードしたものは展開も実行もしていません。"
    err "   ネットワーク経路 (プロキシ・社内ミラー) を確認するか、"
    err "   ソースからビルドしてください: ZAI_FROM_SOURCE=1 を付けて再実行"
    exit 1
}

# SHA-256 を計算する。sha256sum は macOS に無く、shasum は一部の最小 Linux に
# 無いので 3 通り試す。1 つも無ければ「計算できない」として失敗する
# (握りつぶすと検証したふりになる)。
sha256_of() {
    if command -v sha256sum >/dev/null 2>&1; then
        sha256sum "$1" | cut -d' ' -f1
    elif command -v shasum >/dev/null 2>&1; then
        shasum -a 256 "$1" | cut -d' ' -f1
    elif command -v openssl >/dev/null 2>&1; then
        openssl dgst -sha256 "$1" | sed -En 's/^.*[= ]([0-9a-fA-F]{64})$/\1/p'
    else
        return 1
    fi
}

# checksums.txt は release.yml が sha256sum(1) の標準形
# "<64桁hex><SP><SP><ファイル名>" で書き出す。$2 がファイル名になる。
verify_checksum() {
    _file="$1"; _base="$2"; _sums="$3"
    _want=$(awk -v f="$_base" '$2 == f { print $1; exit }' "$_sums" 2>/dev/null) || _want=""
    if [ "${#_want}" -ne 64 ] || [ -n "$(printf '%s' "$_want" | tr -d '0-9a-f')" ]; then
        abort_unverified "checksums.txt に $_base の行がありません"
    fi
    _got=$(sha256_of "$_file" | tr 'ABCDEF' 'abcdef') || _got=""
    if [ "${#_got}" -ne 64 ]; then
        abort_unverified "SHA-256 を計算できません (sha256sum / shasum / openssl のいずれかが必要です)"
    fi
    if [ "$_want" != "$_got" ]; then
        err "   期待値: $_want"
        err "   実際  : $_got"
        abort_unverified "SHA-256 が一致しません ($_base)"
    fi
    say "✅ SHA-256 一致: $_base"
}

path_hint() {
    case ":$PATH:" in
        *":$1:"*) ;;
        *) say "⚠ $1 が PATH にありません。シェルの rc に以下を追記してください:"
           say "   export PATH=\"$1:\$PATH\"" ;;
    esac
}

# 既知のインストール先に残った古い zai を新バイナリで揃える
# (PATH 順によっては古い方が起動してしまい「更新されない」ように見えるため)
sync_stale() {
    new_bin="$1"; skip_dir="$2"
    for d in "$HOME/.local/bin" "$HOME/.cargo/bin"; do
        [ "$d" = "$skip_dir" ] && continue
        if [ -x "$d/zai" ]; then
            say "旧バイナリを更新します: $d/zai"
            install -m 755 "$new_bin" "$d/zai" || true
        fi
    done
}

# --- ビルド済みバイナリのインストール ----------------------------------------
install_prebuilt() {
    case "$(uname -s)" in
        Darwin) os=macos ;;
        Linux)  os=linux ;;
        *) return 1 ;;
    esac
    case "$(uname -m)" in
        arm64|aarch64) arch=arm64 ;;
        x86_64|amd64)  arch=x86_64 ;;
        *) return 1 ;;
    esac
    # Rosetta 配下のシェルは uname -m が x86_64 になるため実 CPU で補正
    if [ "$os" = "macos" ] && [ "$arch" = "x86_64" ] \
        && [ "$(sysctl -n hw.optional.arm64 2>/dev/null)" = "1" ]; then
        arch=arm64
    fi
    tag=$(curl -fsSL "https://api.github.com/repos/$REPO/releases/latest" 2>/dev/null \
        | sed -En 's/.*"tag_name": *"([^"]+)".*/\1/p' | head -n1) || return 1
    [ -n "$tag" ] || return 1
    name="zai-$tag-$os-$arch"
    base="$name.tar.gz"
    url="$REPO_URL/releases/download/$tag/$base"
    sums_url="$REPO_URL/releases/download/$tag/checksums.txt"
    tmp=$(mktemp -d) || return 1
    trap 'rm -rf "$tmp"' EXIT
    say "ビルド済みバイナリを取得します: $url"
    curl -fsSL "$url" -o "$tmp/$base" || return 1
    # ここから先は fail-closed。checksums.txt が取れない時点で中止する
    # (「古いリリースには無いかもしれない」を理由に素通ししない)。
    say "チェックサムを確認します: $sums_url"
    curl -fsSL "$sums_url" -o "$tmp/checksums.txt" \
        || abort_unverified "checksums.txt を取得できませんでした"
    verify_checksum "$tmp/$base" "$base" "$tmp/checksums.txt"
    tar xzf "$tmp/$base" -C "$tmp" || return 1
    mkdir -p "$INSTALL_DIR" || return 1
    verb="インストール"
    [ -x "$INSTALL_DIR/zai" ] && verb="更新"
    install -m 755 "$tmp/$name/zai" "$INSTALL_DIR/zai" || return 1
    sync_stale "$tmp/$name/zai" "$INSTALL_DIR"
    # OS のアプリ一覧 (Launchpad / アプリメニュー) へも登録する。失敗しても続行。
    "$INSTALL_DIR/zai" app install || true
    say ""
    say "✅ ${verb}完了: $INSTALL_DIR/zai ($tag)"
    say "   起動: プロジェクトのフォルダで zai . (または zai [ワークスペースのパス])"
    say "   OS のアプリ一覧の「Zaivern Code」からも起動できます (解除: zai app uninstall)"
    path_hint "$INSTALL_DIR"
    return 0
}

if [ "${ZAI_FROM_SOURCE:-0}" != "1" ] && install_prebuilt; then
    exit 0
fi

# --- ソースビルド (フォールバック) -------------------------------------------
say "ソースからビルド・インストールします..."

# 1. Rust ツールチェーンの確認
if ! command -v cargo >/dev/null 2>&1; then
    # rustup 導入直後で PATH が未反映のケースを拾う
    if [ -f "$HOME/.cargo/env" ]; then
        . "$HOME/.cargo/env"
    fi
fi
if ! command -v cargo >/dev/null 2>&1; then
    say "Rust (cargo) が見つかりません。rustup をインストールします..."
    curl --proto '=https' --tlsv1.2 -fsSL https://sh.rustup.rs | sh -s -- -y --default-toolchain stable
    . "$HOME/.cargo/env"
fi

# 2. rustc 1.88+ の確認
minor=$(rustc --version 2>/dev/null | sed -En 's/^rustc 1\.([0-9]+).*/\1/p')
if [ -z "${minor:-}" ] || [ "$minor" -lt "$REQUIRED_MINOR" ]; then
    say "rustc 1.$REQUIRED_MINOR+ が必要です(現在: $(rustc --version 2>/dev/null || echo '不明'))。stable を更新します..."
    rustup update stable
fi

# 3. Linux の場合はビルド依存のヒントを出す
if [ "$(uname -s)" = "Linux" ] && command -v apt-get >/dev/null 2>&1; then
    if ! dpkg -s libgtk-3-dev >/dev/null 2>&1; then
        say "ヒント: ビルドに失敗する場合は次を実行してください:"
        say "  sudo apt-get install -y build-essential libgtk-3-dev libxcb-shape0-dev libxcb-xfixes0-dev libxkbcommon-dev"
    fi
fi

# 4. GitHub から直接ビルド & インストール
#    --force: 同一バージョンがインストール済みでも再ビルドして上書き(=再実行で更新)
say "GitHub からビルド・インストールします(初回は数分かかります)..."
cargo install --git "$REPO_URL" --locked --force zaivern-code

sync_stale "$HOME/.cargo/bin/zai" "$HOME/.cargo/bin"
bin_path=$(command -v zai 2>/dev/null || echo "$HOME/.cargo/bin/zai")
# OS のアプリ一覧 (Launchpad / アプリメニュー) へも登録する。失敗しても続行。
"$bin_path" app install || true
say ""
say "✅ インストール完了: $bin_path"
say "   起動: プロジェクトのフォルダで zai . (または zai [ワークスペースのパス])"
say "   OS のアプリ一覧の「Zaivern Code」からも起動できます (解除: zai app uninstall)"
path_hint "$HOME/.cargo/bin"