Merge pull request #9 from senna-lang/fix/init-hardening

feat(cli): harden and polish loci init command

Author: senna-lang

CHANGELOG.md

@@ -1,5 +1,29 @@
 # Changelog
 
+## [0.2.0] - 2026-04-21
+
+### Added
+
+- `loci init --no-hooks` flag to skip automatic Claude Code hook registration.
+- `EmbedderSetupError` exception for environment-level embedder failures (distinguishes them from per-row errors).
+- SVG banner at the top of the README (`assets/banner.svg`, generated via Freeze).
+- `scripts/generate-banner.sh` to regenerate the banner from the live CLI output.
+
+### Changed
+
+- `loci init` now registers Claude Code hooks automatically at the end of setup — previously required a separate `loci hook install` step. Use `--no-hooks` to opt out.
+- Interactive prompts re-prompt on invalid input instead of silently falling back to a default. The "run distillation now?" prompt now accepts `y`/`n`/`yes`/`no` in addition to `1`/`2`.
+- Custom exchange counts are range-validated (`1..total`) and custom `min_chars` requires `>= 0`.
+- Startup banner uses the pagga half-block font with a blue vertical gradient.
+
+### Fixed
+
+- `loci init` cleans up `.codeatrium/` automatically if the execution phase fails or is interrupted (`KeyboardInterrupt`), so re-running is safe.
+- A single corrupt `.jsonl` no longer aborts the whole indexing loop — it logs a warning and continues.
+- `git_root()` catches `FileNotFoundError` when the `git` binary is missing.
+- `parse_exchanges` returns `[]` for missing files instead of raising.
+- Distillation failures from `sentence_transformers` import issues (e.g. numpy/pyarrow binary mismatch) now print a single friendly message with remediation hints instead of a full traceback followed by per-row error spam.
+
 ## [0.1.0] - 2026-03-31
 
 ### Added

README.ja.md

@@ -1,3 +1,7 @@
+<p align="center">
+  <img src="assets/banner.svg" alt="codeatrium — memory palace for AI coding agents" width="560">
+</p>
+
 # Codeatrium
 
 [![CI](https://github.com/senna-lang/Codeatrium/actions/workflows/ci.yml/badge.svg)](https://github.com/senna-lang/Codeatrium/actions/workflows/ci.yml)
@@ -47,13 +51,12 @@ Python 3.11 以上が必要です。
 ## クイックスタート
 
 ```bash
-# プロジェクトルートで初期化
+# プロジェクトルートで初期化（Claude Code フックも自動登録）
 loci init
-
-# 自動インデックスのフックをインストール
-loci hook install
 ```
 
+`loci init` は DB 初期化・既存セッション検出・Claude Code フック登録をすべて一度に済ませます。`--no-hooks` を付けるとフック登録をスキップします。途中で失敗した場合は `.codeatrium/` が自動で掃除されるので、再実行して問題ありません。
+
 `loci init` を実行すると、過去のセッションログが検出された場合に以下の質問が表示されます:
 
 > [!IMPORTANT]
@@ -65,7 +68,9 @@ loci hook install
    - Distill last 50（直近の履歴のみ）
    - Distill all（全件、トークン消費あり）
    - Custom（件数を指定）
-3. **蒸留を今すぐ実行するか** — No を選ぶと次回セッション開始時に自動実行されます
+3. **蒸留を今すぐ実行するか** — `1`/`2`/`y`/`n`/`yes`/`no` を受理。No を選ぶと次回セッション開始時に自動実行されます。
+
+各プロンプトで無効な入力をした場合、サイレントにデフォルトへ倒れず再入力を求めます。
 
 ## エージェント向けインストラクション
 
@@ -78,19 +83,19 @@ loci hook install
 
 | コマンド | 説明 |
 |---------|------|
-| `loci init` | プロジェクトルートに `.codeatrium/` を初期化 |
+| `loci init` | `.codeatrium/` を初期化し Claude Code フックを登録（`--no-hooks` で省略可） |
 | `loci index` | 新しいセッションログをインデックス |
 | `loci distill [--limit N]` | 未蒸留の exchange を LLM で蒸留 |
 | `loci search "クエリ" --json` | セマンティック検索（エージェント向け） |
 | `loci context --symbol "名前" --json` | コードシンボル → 過去の会話 |
 | `loci show "<ref>" --json` | 会話原文を取得 |
 | `loci status` | インデックス状態を表示 |
 | `loci server start/stop/status` | 埋め込みサーバー管理 |
-| `loci hook install` | Claude Code の設定にフックを登録 |
+| `loci hook install` | フックを再登録（通常は `loci init` で済む） |
 
 ## 自動化（Claude Code フック）
 
-`loci hook install` 後、すべて自動で動作します:
+`loci init`（または `loci hook install`）後、すべて自動で動作します:
 
 | フック | トリガー | コマンド |
 |--------|---------|---------|

README.md

@@ -1,3 +1,7 @@
+<p align="center">
+  <img src="assets/banner.svg" alt="codeatrium — memory palace for AI coding agents" width="560">
+</p>
+
 # Codeatrium
 
 [![CI](https://github.com/senna-lang/Codeatrium/actions/workflows/ci.yml/badge.svg)](https://github.com/senna-lang/Codeatrium/actions/workflows/ci.yml)
@@ -47,13 +51,12 @@ Requires Python 3.11+.
 ## Quick Start
 
 ```bash
-# Initialize in project root
+# Initialize in project root (also registers Claude Code hooks)
 loci init
-
-# Install hooks for automatic indexing
-loci hook install
 ```
 
+`loci init` now handles everything in one step — database setup, existing session detection, and Claude Code hook registration. Pass `--no-hooks` to skip hook registration. If init fails partway through, `.codeatrium/` is cleaned up automatically so re-running is safe.
+
 When running `loci init`, if past session logs are detected, you'll be prompted with:
 
 > [!IMPORTANT]
@@ -65,7 +68,9 @@ When running `loci init`, if past session logs are detected, you'll be prompted
    - Distill last 50 (recent history only)
    - Distill all (everything — high token cost)
    - Custom (specify a number)
-3. **Run distillation now?** — Choose No to defer to the next session start
+3. **Run distillation now?** — Accepts `1`/`2`/`y`/`n`/`yes`/`no`. Choose No to defer to the next session start.
+
+Invalid input on any prompt re-prompts instead of silently falling back to a default.
 
 ## Agent Instructions
 
@@ -78,19 +83,19 @@ Agent instructions are injected automatically — no manual setup required:
 
 | Command | Description |
 |---------|-------------|
-| `loci init` | Initialize `.codeatrium/` in project root |
+| `loci init` | Initialize `.codeatrium/` and register Claude Code hooks (`--no-hooks` to skip) |
 | `loci index` | Index new session logs |
 | `loci distill [--limit N]` | Distill undistilled exchanges via LLM |
 | `loci search "query" --json` | Semantic search (agent-facing) |
 | `loci context --symbol "name" --json` | Code symbol → past conversations |
 | `loci show "<ref>" --json` | Retrieve verbatim conversation |
 | `loci status` | Show index state |
 | `loci server start/stop/status` | Embedding server management |
-| `loci hook install` | Register hooks in Claude Code settings |
+| `loci hook install` | Re-register hooks (normally already done by `loci init`) |
 
 ## Automation (Claude Code Hooks)
 
-After `loci hook install`, everything runs automatically:
+After `loci init` (or `loci hook install`), everything runs automatically:
 
 | Hook | Trigger | Command |
 |------|---------|---------|

assets/banner.svg

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "codeatrium"
-version = "0.1.0"
+version = "0.2.0"
 description = "Memory palace for AI coding agents — index sessions, recall code context in <0.2s"
 readme = "README.md"
 license = { text = "MIT" }

pyproject.toml

@@ -0,0 +1,25 @@
+#!/usr/bin/env bash
+# Regenerate assets/banner.svg from the live loci banner using Freeze.
+# Requires: brew install charmbracelet/tap/freeze
+#
+# Freeze embeds font glyphs as <path> elements so the rendering is
+# identical across GitHub, local browsers, and other SVG consumers.
+
+set -euo pipefail
+
+cd "$(dirname "$0")/.."
+
+FORCE_COLOR=1 python -c "
+from codeatrium.cli import _print_banner
+_print_banner()
+" | freeze \
+    --output assets/banner.svg \
+    --padding 24,28,24,28 \
+    --margin 0 \
+    --background '#00000000' \
+    --border.radius 8 \
+    --font.family 'JetBrains Mono' \
+    --font.size 14 \
+    --line-height 1.2
+
+echo "wrote: assets/banner.svg"

scripts/generate-banner.sh

@@ -1,3 +1,3 @@
 """codeatrium — Memory palace for AI coding agents"""
 
-__version__ = "0.1.0"
+__version__ = "0.2.0"