function jj-clone --description "Clone a repo into a jj workspace layout with a colocated main"
    if test (count $argv) -eq 0
        echo "Usage: jj-clone <url> [directory]"
        return 1
    end

    set -l url $argv[1]
    set -l dir
    if test (count $argv) -ge 2
        set dir $argv[2]
    else
        set dir (basename "$url" .git)
    end

    mkdir -p "$dir"
    jj git clone "$url" "$dir/main"
    cd "$dir/main"
end