wtcode 0.1.0__py3-none-any.whl

wtcode/__init__.py

@@ -0,0 +1,3 @@
+"""wtcode -- launch your favorite code tool in a git worktree."""
+
+__version__ = "0.1.0"

wtcode/__main__.py

@@ -0,0 +1,15 @@
+"""Thin wrapper to exec wtcode.sh from the installed package."""
+
+import os
+import sys
+from importlib.resources import as_file, files
+
+
+def main():
+    script = files("wtcode").joinpath("wtcode.sh")
+    with as_file(script) as path:
+        os.execvp("bash", ["bash", str(path)] + sys.argv[1:])
+
+
+if __name__ == "__main__":
+    main()

wtcode/wtcode.sh

@@ -0,0 +1,239 @@
+#!/usr/bin/env bash
+# wtcode -- launch a code tool in a git worktree for a given branch
+# https://github.com/netj/wtcode
+# Author: Jaeho Shin <netj@sparcs.org>
+# Created: 2026-02-03
+set -eu
+shopt -s extglob
+${WTCODE_DEBUG:+set -x}
+
+--msg() { echo "wtcode: $*" >&2; }
+
+WTCODE_VERSION=0.1.0
+--version() { echo "wtcode $WTCODE_VERSION"; }
+--help() {
+  cat <<USAGE
+wtcode $WTCODE_VERSION -- launch a code tool in a git worktree
+
+Usage: wtcode [BRANCH] [CMD [CMD-ARGS...]]
+
+  BRANCH     Git branch or worktree name to switch to.
+             If omitted and fzf is available, interactively select one.
+             Prefix with ':' to force creating a new branch
+             (use multiple colons to avoid fzf matching, e.g., :::my-branch).
+
+  CMD        Command to launch in the worktree (default: \$WTCODE_CMD,
+             or first available of: ${WTCODE_CMDS_TO_TRY[*]}, \$SHELL).
+
+Environment variables:
+  WTCODE_CMD             Default tool to launch (e.g., claude, lazygit, vim)
+  WTCODE_DEBUG           Enable debug tracing when set
+  GIT_WORKTREE_ROOT      Override the directory where worktrees are created
+
+Examples:
+  wtcode feature-x                  # select/create worktree, launch default tool
+  wtcode feature-x lazygit          # launch lazygit in the worktree
+  wtcode feature-x claude --resume  # launch claude with --resume
+  wtcode :new-feature               # create new branch and worktree
+  WTCODE_CMD=cursor wtcode feature  # use cursor as the default tool
+USAGE
+}
+
+# commands to try as default, in order of preference
+WTCODE_CMDS_TO_TRY=(
+    ${WTCODE_CMD:-}
+    claude
+    aider
+    codex
+)
+
+###############################################################################
+## --select-git-branch -- determine which branch/worktree to use
+###############################################################################
+--select-git-branch() {
+  if [[ $# -gt 0 ]]; then
+    branch_name=$1; shift
+  elif type fzf &>/dev/null; then
+    # if branch name unspecified, use fzf to select one or enter a new name
+    # format: branch_name<TAB>indicator branch_name date hash [upstream] subject
+    branch_name=$(
+      { set +x; } &>/dev/null
+      # NOTE: git branch --format uses ref-filter syntax which doesn't support
+      # pretty-format's %<(N,trunc) truncation. Using bash to handle truncation,
+      # alignment, and coloring instead.
+      c_reset=$'\e[0m' c_green=$'\e[32m' c_cyan=$'\e[36m' c_blue=$'\e[34m' c_yellow=$'\e[33m' c_red=$'\e[31m' c_dim=$'\e[2m'
+      {
+      git branch --sort=-committerdate \
+          --format=$'x%(refname:lstrip=2)\tx%(HEAD)\tx%(worktreepath)\tx%(committerdate:relative)\tx%(objectname:short)\tx%(upstream:track)\tx%(contents:subject)'
+      # also include remote-tracking branches that don't have a local counterpart
+      local_branches=$(git branch --format='%(refname:lstrip=2)')
+      git branch -r --sort=-committerdate \
+          --format=$'x%(refname:lstrip=2)\tx>\tx\tx%(committerdate:relative)\tx%(objectname:short)\tx\tx%(contents:subject)' |
+      while IFS=$'\t' read -r remote_ref rest; do
+        remote_ref=${remote_ref#x}
+        [[ $remote_ref == */HEAD ]] && continue
+        local_name=${remote_ref#*/}
+        grep -qxF "$local_name" <<< "$local_branches" && continue
+        printf 'x%s\t%s\n' "$remote_ref" "$rest"
+      done
+      } |
+      while IFS=$'\t' read -r branch head worktree date hash upstream subject; do
+        # strip the x prefix from all fields
+        branch=${branch#x} head=${head#x} worktree=${worktree#x} date=${date#x} hash=${hash#x} upstream=${upstream#x} subject=${subject#x}
+        # determine indicator and color: HEAD (*) takes precedence, then worktree (+)
+        if [[ $head == '*' ]]; then ind='*'; c=$c_green
+        elif [[ -n $worktree ]]; then ind='+'; c=$c_cyan
+        elif [[ $head == '>' ]]; then ind=' '; c=$c_dim
+        else ind=' '; c=$c_reset
+        fi
+        # truncate then pad branch name, right-pad date (before adding colors)
+        if (( ${#branch} > 50 )); then name="${branch:0:48}.."; else name=$branch; fi
+        printf -v name '%-50s' "$name"
+        printf -v date '%14s' "$date"
+        # build display fields with colors embedded
+        name_shown="${c}${name}${c_reset}"
+        date_shown="${c_blue}${date}${c_reset}"
+        hash_shown="${c_yellow}${hash}${c_reset}"
+        if [[ -n $upstream ]]; then upstream_shown="${c_red}${upstream}${c_reset} "; else upstream_shown=''; fi
+        printf '%s\t%s %s\t%s  %s %s%s\n' \
+          "$branch" "$ind" "$name_shown" "$date_shown" "$hash_shown" "$upstream_shown" "$subject"
+      done |
+      fzf --ansi --color --tmux --print-query --delimiter=$'\t' --with-nth=2 --nth=1 \
+          --preview 'echo {3}' --preview-window 'down:2:wrap' \
+          --prompt 'wtcode: select worktree/branch (prefix : to create new) > ' |
+      cut -f1 |
+      tail -1
+    )
+  else  # abort if fzf not available
+    branch_name=${1:?Need a worktree/branch name as first argument}; shift
+  fi
+
+  : ${branch_name:?non-empty worktree/branch name required}
+
+  # check if branch name starts with ':' to force new branch creation
+  # supports multiple colons (e.g., :::my-branch) to avoid fzf matching
+  force_new_branch=false
+  if [[ $branch_name == :* ]]; then
+    force_new_branch=true
+    branch_name=${branch_name##+(:)}
+    : ${branch_name:?non-empty branch name required after ':'}
+    # sanitize free-form text into a valid git branch name
+    branch_name=$(printf '%s' "$branch_name" |
+      tr '[:upper:]' '[:lower:]' |   # lowercase
+      sed 's/[^a-z0-9/_-]/-/g' |     # replace non-alnum to hyphens
+      sed 's/--*/-/g' |              # collapse consecutive hyphens
+      sed 's/^-//; s/-$//'           # trim leading/trailing hyphens
+    )
+    : ${branch_name:?branch name is empty after sanitization}
+  fi
+
+  # check if branch_name refers to a remote branch (e.g., origin/feature-x)
+  remote_branch=
+  if ! $force_new_branch; then
+    for remote in $(git remote); do
+      if [[ $branch_name == "$remote/"* ]]; then
+        remote_branch=$branch_name
+        branch_name=${branch_name#"$remote/"}
+        break
+      fi
+    done
+  fi
+
+  # remaining args are the command to launch
+  wtcode_cmd=("$@")
+}
+
+###############################################################################
+## --prepare-git-worktree -- create or switch to the worktree for $branch_name
+###############################################################################
+--prepare-git-worktree() {
+  # determine root of the worktree dirs
+  cd "$(git rev-parse --show-toplevel)"
+  : ${GIT_WORKTREE_ROOT:=$(
+    git_common_dir=$(git rev-parse --git-common-dir)
+    cd "$git_common_dir"
+    cd ..
+    repo_name=$(basename "$PWD")
+    echo "$PWD"/../"$repo_name".worktrees
+  )}
+
+  # ensure worktree based on given branch name
+  worktree_path="$GIT_WORKTREE_ROOT"/"$branch_name"
+  if [[ -e "$worktree_path"/.git ]]; then
+    --msg "using existing worktree: $worktree_path"
+  elif [[ -n ${remote_branch-} ]] && ! git rev-parse --verify "refs/heads/$branch_name" &>/dev/null; then
+    # remote branch: create local tracking branch in a new worktree
+    --msg "creating worktree for remote branch: $remote_branch"
+    git worktree add -b "$branch_name" "$worktree_path" "$remote_branch"
+    git -C "$worktree_path" branch --set-upstream-to="$remote_branch" "$branch_name"
+  elif git rev-parse "$branch_name" &>/dev/null; then
+    # branch exists, just check it out in a new worktree with the same name
+    --msg "creating worktree for branch: $branch_name"
+    git worktree add -B "$branch_name" "$worktree_path" "$(git rev-parse "$branch_name")"
+  else
+    # fork the current HEAD and create the new worktree
+    --msg "creating worktree with new branch: $branch_name"
+    git worktree add -b "$branch_name" "$worktree_path" "$(git rev-parse HEAD)"
+  fi
+  cd "$worktree_path"
+
+  # ensure the branch is checked out on the worktree
+  [[ $(git branch --show-current) = $branch_name ]] ||
+    git checkout "$branch_name" --
+}
+
+###############################################################################
+## --launch-code-tool -- resolve and exec the tool in the worktree
+###############################################################################
+--launch-code-tool() {
+  # resolve the command to launch if not specified by user
+  if [[ ${#wtcode_cmd[@]} -eq 0 ]]; then
+    for _cmd in "${WTCODE_CMDS_TO_TRY[@]}"; do
+      [[ -n "$_cmd" ]] && type "$_cmd" &>/dev/null && wtcode_cmd=("$_cmd") && break
+    done
+    # fall back to an interactive shell
+    wtcode_cmd=("${wtcode_cmd[@]:-${SHELL:-bash}}")
+  fi
+
+  --msg "launching: ${wtcode_cmd[*]}"
+  "${wtcode_cmd[@]}"
+}
+
+###############################################################################
+## Command wrappers -- override specific tools with pre-launch setup.
+## Define a function with the tool's name to add custom behavior.
+###############################################################################
+
+# claude: auto-trust the worktree in Claude Code's config
+claude() {
+  if type jq &>/dev/null && [[ -f ~/.claude.json ]]; then
+    (
+      export worktree_path="$PWD"
+      jq -e '.projects[env.worktree_path]' ~/.claude.json &>/dev/null || {
+        jq '
+          .projects[env.worktree_path] = ({}
+          | .hasTrustDialogAccepted = true
+          )
+        ' ~/.claude.json >~/.claude.json.wtcode.$$
+        mv -f ~/.claude.json.wtcode.$$ ~/.claude.json
+      }
+    )
+  fi
+  exec command claude "$@"
+}
+
+###############################################################################
+## main
+###############################################################################
+
+-h() { --help "$@"; }
+
+--() {
+  --select-git-branch "$@"
+  --prepare-git-worktree
+  --launch-code-tool
+}
+
+# dispatch $1 as a function when it starts with -
+[[ ${1-} == -* ]] || set -- -- "$@"
+"$@"

wtcode-0.1.0.dist-info/METADATA

@@ -0,0 +1,94 @@
+Metadata-Version: 2.4
+Name: wtcode
+Version: 0.1.0
+Summary: Launch your favorite code tool in a git worktree
+Author-email: Jaeho Shin <netj@sparcs.org>
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/netj/wtcode
+Project-URL: Repository, https://github.com/netj/wtcode
+Project-URL: Issues, https://github.com/netj/wtcode/issues
+Keywords: git,worktree,cli,claude,developer-tools
+Classifier: Environment :: Console
+Classifier: Topic :: Software Development
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: license-file
+
+# wtcode
+
+**Launch your favorite code tool in a git worktree.**
+
+*worktree code* / *what-the-code* -- streamlines git worktree selection, creation, and launching tools like [Claude Code](https://claude.ai/code), [lazygit](https://github.com/jesseduffield/lazygit), your editor, or any command.
+
+## Install
+
+### Homebrew
+
+```sh
+brew install netj/tap/wtcode
+```
+
+### PyPI (via [uv](https://docs.astral.sh/uv/))
+
+```sh
+uvx wtcode
+```
+
+Or install globally:
+
+```sh
+uv tool install wtcode
+```
+
+### From source
+
+```sh
+git clone https://github.com/netj/wtcode.git
+ln -s "$PWD/wtcode/wtcode.sh" /usr/local/bin/wtcode
+```
+
+## Usage
+
+```
+wtcode [BRANCH] [CMD [CMD-ARGS...]]
+```
+
+- **`BRANCH`** -- Git branch or worktree name. If omitted and [fzf](https://github.com/junegunn/fzf) is available, interactively select one. Prefix with `:` to create a new branch (use `:::name` to avoid fzf matching the colons).
+- **`CMD`** -- Command to launch in the worktree. Defaults to `$WTCODE_CMD`, or the first available of: `claude`, `aider`, `codex`, `$SHELL`.
+
+### Examples
+
+```sh
+wtcode feature-x                  # launch default tool in feature-x worktree
+wtcode feature-x lazygit          # launch lazygit
+wtcode feature-x claude --resume  # launch claude with flags
+wtcode :new-feature               # create new branch and worktree
+wtcode                            # interactive branch selection via fzf
+WTCODE_CMD=cursor wtcode feature  # use cursor as default tool
+```
+
+## Environment variables
+
+| Variable | Description |
+|---|---|
+| `WTCODE_CMD` | Default tool to launch (e.g., `claude`, `lazygit`, `vim`, `cursor`) |
+| `WTCODE_DEBUG` | Enable debug tracing when set to any value |
+| `GIT_WORKTREE_ROOT` | Override the directory where worktrees are created |
+
+## How it works
+
+1. **Select/specify a branch** -- pass as argument or pick interactively with fzf
+2. **Create or switch to the worktree** -- worktrees are organized under `$GIT_WORKTREE_ROOT` (defaults to `../<repo>.worktrees/`)
+3. **Launch a tool** -- runs the specified command (or smart default) inside the worktree
+
+## Dependencies
+
+- **git** -- required
+- **[fzf](https://github.com/junegunn/fzf)** -- recommended, for interactive branch selection
+- **[jq](https://jqlang.github.io/jq/)** -- optional, for Claude Code auto-trust setup
+
+## License
+
+[MIT](LICENSE)

wtcode-0.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+wtcode/__init__.py,sha256=30WpOukH8wq2FscMYefTmdQXxQBJIcqyBssiXoilFA0,89
+wtcode/__main__.py,sha256=hOvSYANO-q_rf2yy-AQoEOr18KFRSOF8Ezosf8eeFmM,335
+wtcode/wtcode.sh,sha256=ivekpqBoQOFiG_5MfKC5s09kd8liJ7ffKmLF99SU3Lw,9796
+wtcode-0.1.0.dist-info/licenses/LICENSE,sha256=MV9gr1nAVj5R60EM2BQuBhBxCG5V9ulX6-Ep_-LcLqA,1067
+wtcode-0.1.0.dist-info/METADATA,sha256=610ETtcaumMVjhNIXQxbNMW-k7dKjNweTNQiWucS8fs,2947
+wtcode-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+wtcode-0.1.0.dist-info/entry_points.txt,sha256=QkSPHU86FQa-ujQk07fF4sX4DBQgWzDtgusop4AA5bc,48
+wtcode-0.1.0.dist-info/top_level.txt,sha256=oftTSYYjDMqt33wa0c9MC15KuQ3eyRyoTLXy3PdkVfE,7
+wtcode-0.1.0.dist-info/RECORD,,

wtcode-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

wtcode-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+wtcode = wtcode.__main__:main

wtcode-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Jaeho Shin
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

wtcode-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+wtcode