[ { "path": ".cargo/audit.toml", "content": "[advisories]\nignore = [\n # This is a vuln on RSA. RSA is in our lockfile, but not in cargo-tree. \n # It is a issue with sqlx/cargo, and does not affect Atuin.\n # See:\n # - https://github.com/launchbadge/sqlx/issues/3211\n # - https://github.com/rust-lang/cargo/issues/10801\n \"RUSTSEC-2023-0071\"\n]\n" }, { "path": ".codespellrc", "content": "[codespell]\n# Ref: https://github.com/codespell-project/codespell#using-a-config-file\nskip = .git*,*.lock,.codespellrc,CODE_OF_CONDUCT.md,CONTRIBUTORS\ncheck-hidden = true\n# ignore-regex = \nignore-words-list = crate,ratatui,inbetween,iterm,fo,brunch\n\n" }, { "path": ".dockerignore", "content": "./target\nDockerfile\n" }, { "path": ".gitattributes", "content": "*.sh eol=lf\n*.nix eol=lf\n*.zsh eol=lf\n\n*.sql eol=lf\n" }, { "path": ".github/DISCUSSION_TEMPLATE/support.yml", "content": "body:\n - type: input\n attributes:\n label: Operating System\n description: What operating system are you using?\n placeholder: \"Example: macOS Big Sur\"\n validations:\n required: true\n\n - type: input\n attributes:\n label: Shell\n description: What shell are you using?\n placeholder: \"Example: zsh 5.8.1\"\n validations:\n required: true\n\n - type: dropdown\n attributes:\n label: Version\n description: What version of atuin are you running?\n multiple: false\n options: # how often will I forget to update this? a lot.\n - v17.0.0 (Default)\n - v16.0.0\n - v15.0.0\n - v14.0.1\n - v14.0.0\n - v13.0.1\n - v13.0.0\n - v12.0.0\n - v11.0.0\n - v0.10.0\n - v0.9.1\n - v0.9.0\n - v0.8.1\n - v0.8.0\n - v0.7.2\n - v0.7.1\n - v0.7.0\n - v0.6.4\n - v0.6.3\n default: 0\n validations:\n required: true\n\n - type: checkboxes\n attributes:\n label: Self hosted\n description: Are you self hosting atuin server?\n options:\n - label: I am self hosting atuin server\n\n - type: checkboxes\n attributes:\n label: Search the issues\n description: Did you search the issues and discussions for your problem?\n options:\n - label: I checked that someone hasn't already asked about the same issue\n required: true\n\n - type: textarea\n attributes:\n label: Behaviour\n description: \"Please describe the issue - what you expected to happen, what actually happened\"\n \n - type: textarea\n attributes:\n label: Logs\n description: \"If possible, please include logs from atuin, especially if you self host the server - ATUIN_LOG=debug\"\n\n - type: textarea\n attributes:\n label: Extra information\n description: \"Anything else you'd like to add?\"\n\n - type: checkboxes\n attributes:\n label: Code of Conduct\n description: The Code of Conduct helps create a safe space for everyone. We require\n that everyone agrees to it.\n options:\n - label: I agree to follow this project's [Code of Conduct](https://github.com/atuinsh/atuin/blob/main/CODE_OF_CONDUCT.md)\n required: true\n" }, { "path": ".github/FUNDING.yml", "content": "# These are supported funding model platforms\n\ngithub: [atuinsh]\npatreon: # Replace with a single Patreon username\nopen_collective: # Replace with a single Open Collective username\nko_fi: # Replace with a single Ko-fi username\ntidelift: # Replace with a single Tidelift platform-name/package-name e.g., npm/babel\ncommunity_bridge: # Replace with a single Community Bridge project-name e.g., cloud-foundry\nliberapay: # Replace with a single Liberapay username\nissuehunt: # Replace with a single IssueHunt username\notechie: # Replace with a single Otechie username\nlfx_crowdfunding: # Replace with a single LFX Crowdfunding project-name e.g., cloud-foundry\ncustom: # Replace with up to 4 custom sponsorship URLs e.g., ['link1', 'link2']\n" }, { "path": ".github/ISSUE_TEMPLATE/bug.yaml", "content": "name: Bug Report\ndescription: File a bug report\ntitle: \"[Bug]: \"\nlabels: [\"bug\", \"triage\"]\nbody:\n - type: markdown\n attributes:\n value: |\n Thanks for taking the time to fill out this bug report!\n - type: textarea\n id: what-expected\n attributes:\n label: What did you expect to happen?\n placeholder: Tell us what you expected to see!\n validations:\n required: true\n - type: textarea\n id: what-happened\n attributes:\n label: What happened?\n placeholder: Tell us what you see!\n validations:\n required: true\n - type: textarea\n id: doctor\n validations:\n required: true\n attributes:\n label: Atuin doctor output\n description: Please run 'atuin doctor' and share the output. If it fails to run, share any errors. This requires Atuin >=v18.1.0\n render: yaml\n - type: checkboxes\n id: terms\n attributes:\n label: Code of Conduct\n description: By submitting this issue, you agree to follow our [Code of Conduct](https://github.com/atuinsh/atuin/blob/main/CODE_OF_CONDUCT.md)\n options:\n - label: I agree to follow this project's Code of Conduct\n required: true\n" }, { "path": ".github/dependabot.yml", "content": "# To get started with Dependabot version updates, you'll need to specify which\n# package ecosystems to update and where the package manifests are located.\n# Please see the documentation for all configuration options:\n# https://help.github.com/github/administering-a-repository/configuration-options-for-dependency-updates\n\nversion: 2\nupdates:\n - package-ecosystem: \"cargo\" # See documentation for possible values\n directory: \"/\" # Location of package manifests\n schedule:\n interval: \"weekly\"\n - package-ecosystem: \"docker\" # See documentation for possible values\n directory: \"/\" # Location of package manifests\n schedule:\n interval: \"weekly\"\n - package-ecosystem: \"github-actions\"\n directory: \"/\"\n schedule:\n interval: \"weekly\"\n" }, { "path": ".github/pull_request_template.md", "content": "\n\n## Checks\n- [ ] I am happy for maintainers to push small adjustments to this PR, to speed up the review cycle\n- [ ] I have checked that there are no existing pull requests for the same thing\n" }, { "path": ".github/workflows/codespell.yml", "content": "# Codespell configuration is within .codespellrc\n---\nname: Codespell\n\non:\n push:\n branches: [main]\n pull_request:\n branches: [main]\n\npermissions:\n contents: read\n\njobs:\n codespell:\n name: Check for spelling errors\n runs-on: depot-ubuntu-24.04\n\n steps:\n - name: Checkout\n uses: actions/checkout@v6\n - name: Codespell\n uses: codespell-project/actions-codespell@v2\n with:\n # This is regenerated from commit history\n # we cannot rewrite commit history, and I'd rather not correct it\n # every time\n exclude_file: CHANGELOG.md\n" }, { "path": ".github/workflows/docker.yaml", "content": "name: build-docker\n\non:\n push:\n branches: [main]\n\njobs:\n publish:\n concurrency:\n group: ${{ github.ref }}-docker\n cancel-in-progress: true\n permissions:\n packages: write\n contents: read\n id-token: write\n\n runs-on: depot-ubuntu-24.04\n steps:\n - uses: actions/checkout@v6\n\n - name: Get Repo Owner\n id: get_repo_owner\n run: echo \"REPO_OWNER=$(echo ${{ github.repository_owner }} | tr '[:upper:]' '[:lower:]')\" > $GITHUB_ENV\n\n - uses: depot/setup-action@v1\n\n - name: Login to container Registry\n uses: docker/login-action@v3\n with:\n username: ${{ github.repository_owner }}\n password: ${{ secrets.GITHUB_TOKEN }}\n registry: ghcr.io\n\n - name: Get short sha\n id: shortsha\n run: echo \"short_sha=$(git rev-parse --short HEAD)\" >> $GITHUB_OUTPUT\n\n - name: Build and push\n uses: depot/build-push-action@v1\n with:\n push: true\n platforms: linux/amd64,linux/arm64\n file: ./Dockerfile\n context: .\n provenance: false\n build-args: |\n Version=dev\n GitCommit=${{ steps.shortsha.outputs.short_sha }}\n tags: |\n ghcr.io/${{ env.REPO_OWNER }}/atuin:${{ steps.shortsha.outputs.short_sha }}\n" }, { "path": ".github/workflows/installer.yml", "content": "name: Install\n\non:\n push:\n branches: [main]\n pull_request:\n paths: .github/workflows/installer.yml\n\nenv:\n CARGO_TERM_COLOR: always\n\njobs:\n install:\n strategy:\n matrix:\n os: [depot-ubuntu-24.04, macos-14]\n runs-on: ${{ matrix.os }}\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install zsh for ubuntu\n if: matrix.os == 'depot-ubuntu-24.04' \n run: | \n sudo apt install zsh\n\n - name: Test install script on bash\n run: | \n /bin/bash -c \"$(curl --proto '=https' --tlsv1.2 -sSf https://setup.atuin.sh)\"\n [ -d \"$HOME/.atuin\" ] && source $HOME/.atuin/bin/env\n atuin --help\n\n - name: Test install script on zsh\n shell: zsh {0}\n run: | \n /bin/bash -c \"$(curl --proto '=https' --tlsv1.2 -sSf https://setup.atuin.sh)\"\n [ -d \"$HOME/.atuin\" ] && source $HOME/.atuin/bin/env\n atuin --help\n" }, { "path": ".github/workflows/nix.yml", "content": "# Verify the Nix build is working\n# Failures will usually occur due to an out of date Rust version\n# That can be updated to the latest version in nixpkgs-unstable with `nix flake update`\nname: Nix\non:\n push:\n branches: [ main ]\n paths-ignore:\n - 'ui/**'\n pull_request:\n branches: [ main ]\n paths-ignore:\n - 'ui/**'\n\njobs:\n check:\n runs-on: depot-ubuntu-24.04\n\n steps:\n - uses: actions/checkout@v6\n - uses: cachix/install-nix-action@v31\n\n - name: Run nix flake check\n run: nix flake check --print-build-logs\n\n build-test:\n runs-on: depot-ubuntu-24.04\n\n steps:\n - uses: actions/checkout@v6\n - uses: cachix/install-nix-action@v31\n\n - name: Run nix build\n run: nix build --print-build-logs\n" }, { "path": ".github/workflows/release.yml", "content": "# This file was autogenerated by dist: https://axodotdev.github.io/cargo-dist\n#\n# Copyright 2022-2024, axodotdev\n# SPDX-License-Identifier: MIT or Apache-2.0\n#\n# CI that:\n#\n# * checks for a Git Tag that looks like a release\n# * builds artifacts with dist (archives, installers, hashes)\n# * uploads those artifacts to temporary workflow zip\n# * on success, uploads the artifacts to a GitHub Release\n#\n# Note that the GitHub Release will be created with a generated\n# title/body based on your changelogs.\n\nname: Release\npermissions:\n \"contents\": \"write\"\n\n# This task will run whenever you push a git tag that looks like a version\n# like \"1.0.0\", \"v0.1.0-prerelease.1\", \"my-app/0.1.0\", \"releases/v1.0.0\", etc.\n# Various formats will be parsed into a VERSION and an optional PACKAGE_NAME, where\n# PACKAGE_NAME must be the name of a Cargo package in your workspace, and VERSION\n# must be a Cargo-style SemVer Version (must have at least major.minor.patch).\n#\n# If PACKAGE_NAME is specified, then the announcement will be for that\n# package (erroring out if it doesn't have the given version or isn't dist-able).\n#\n# If PACKAGE_NAME isn't specified, then the announcement will be for all\n# (dist-able) packages in the workspace with that version (this mode is\n# intended for workspaces with only one dist-able package, or with all dist-able\n# packages versioned/released in lockstep).\n#\n# If you push multiple tags at once, separate instances of this workflow will\n# spin up, creating an independent announcement for each one. However, GitHub\n# will hard limit this to 3 tags per commit, as it will assume more tags is a\n# mistake.\n#\n# If there's a prerelease-style suffix to the version, then the release(s)\n# will be marked as a prerelease.\non:\n pull_request:\n push:\n tags:\n - '**[0-9]+.[0-9]+.[0-9]+*'\n\njobs:\n # Run 'dist plan' (or host) to determine what tasks we need to do\n plan:\n runs-on: \"ubuntu-22.04\"\n outputs:\n val: ${{ steps.plan.outputs.manifest }}\n tag: ${{ !github.event.pull_request && github.ref_name || '' }}\n tag-flag: ${{ !github.event.pull_request && format('--tag={0}', github.ref_name) || '' }}\n publishing: ${{ !github.event.pull_request }}\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n steps:\n - uses: actions/checkout@v6\n with:\n persist-credentials: false\n submodules: recursive\n - name: Install dist\n # we specify bash to get pipefail; it guards against the `curl` command\n # failing. otherwise `sh` won't catch that `curl` returned non-0\n shell: bash\n run: \"curl --proto '=https' --tlsv1.2 -LsSf https://github.com/axodotdev/cargo-dist/releases/download/v0.31.0/cargo-dist-installer.sh | sh\"\n - name: Cache dist\n uses: actions/upload-artifact@v6\n with:\n name: cargo-dist-cache\n path: ~/.cargo/bin/dist\n # sure would be cool if github gave us proper conditionals...\n # so here's a doubly-nested ternary-via-truthiness to try to provide the best possible\n # functionality based on whether this is a pull_request, and whether it's from a fork.\n # (PRs run on the *source* but secrets are usually on the *target* -- that's *good*\n # but also really annoying to build CI around when it needs secrets to work right.)\n - id: plan\n run: |\n dist ${{ (!github.event.pull_request && format('host --steps=create --tag={0}', github.ref_name)) || 'plan' }} --output-format=json > plan-dist-manifest.json\n echo \"dist ran successfully\"\n cat plan-dist-manifest.json\n echo \"manifest=$(jq -c \".\" plan-dist-manifest.json)\" >> \"$GITHUB_OUTPUT\"\n - name: \"Upload dist-manifest.json\"\n uses: actions/upload-artifact@v6\n with:\n name: artifacts-plan-dist-manifest\n path: plan-dist-manifest.json\n\n # Build and packages all the platform-specific things\n build-local-artifacts:\n name: build-local-artifacts (${{ join(matrix.targets, ', ') }})\n # Let the initial task tell us to not run (currently very blunt)\n needs:\n - plan\n if: ${{ fromJson(needs.plan.outputs.val).ci.github.artifacts_matrix.include != null && (needs.plan.outputs.publishing == 'true' || fromJson(needs.plan.outputs.val).ci.github.pr_run_mode == 'upload') }}\n strategy:\n fail-fast: false\n # Target platforms/runners are computed by dist in create-release.\n # Each member of the matrix has the following arguments:\n #\n # - runner: the github runner\n # - dist-args: cli flags to pass to dist\n # - install-dist: expression to run to install dist on the runner\n #\n # Typically there will be:\n # - 1 \"global\" task that builds universal installers\n # - N \"local\" tasks that build each platform's binaries and platform-specific installers\n matrix: ${{ fromJson(needs.plan.outputs.val).ci.github.artifacts_matrix }}\n runs-on: ${{ matrix.runner }}\n container: ${{ matrix.container && matrix.container.image || null }}\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n BUILD_MANIFEST_NAME: target/distrib/${{ join(matrix.targets, '-') }}-dist-manifest.json\n permissions:\n \"attestations\": \"write\"\n \"contents\": \"read\"\n \"id-token\": \"write\"\n steps:\n - name: enable windows longpaths\n run: |\n git config --global core.longpaths true\n - uses: actions/checkout@v6\n with:\n persist-credentials: false\n submodules: recursive\n - name: Install Rust non-interactively if not already installed\n if: ${{ matrix.container }}\n run: |\n if ! command -v cargo > /dev/null 2>&1; then\n curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y\n echo \"$HOME/.cargo/bin\" >> $GITHUB_PATH\n fi\n - name: Install dist\n run: ${{ matrix.install_dist.run }}\n # Get the dist-manifest\n - name: Fetch local artifacts\n uses: actions/download-artifact@v7\n with:\n pattern: artifacts-*\n path: target/distrib/\n merge-multiple: true\n - name: Install dependencies\n run: |\n ${{ matrix.packages_install }}\n - name: Build artifacts\n run: |\n # Actually do builds and make zips and whatnot\n dist build ${{ needs.plan.outputs.tag-flag }} --print=linkage --output-format=json ${{ matrix.dist_args }} > dist-manifest.json\n echo \"dist ran successfully\"\n - name: Attest\n uses: actions/attest-build-provenance@v3\n with:\n subject-path: \"target/distrib/*${{ join(matrix.targets, ', ') }}*\"\n - id: cargo-dist\n name: Post-build\n # We force bash here just because github makes it really hard to get values up\n # to \"real\" actions without writing to env-vars, and writing to env-vars has\n # inconsistent syntax between shell and powershell.\n shell: bash\n run: |\n # Parse out what we just built and upload it to scratch storage\n echo \"paths<> \"$GITHUB_OUTPUT\"\n dist print-upload-files-from-manifest --manifest dist-manifest.json >> \"$GITHUB_OUTPUT\"\n echo \"EOF\" >> \"$GITHUB_OUTPUT\"\n\n cp dist-manifest.json \"$BUILD_MANIFEST_NAME\"\n - name: \"Upload artifacts\"\n uses: actions/upload-artifact@v6\n with:\n name: artifacts-build-local-${{ join(matrix.targets, '_') }}\n path: |\n ${{ steps.cargo-dist.outputs.paths }}\n ${{ env.BUILD_MANIFEST_NAME }}\n\n # Build and package all the platform-agnostic(ish) things\n build-global-artifacts:\n needs:\n - plan\n - build-local-artifacts\n runs-on: \"ubuntu-22.04\"\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n BUILD_MANIFEST_NAME: target/distrib/global-dist-manifest.json\n steps:\n - uses: actions/checkout@v6\n with:\n persist-credentials: false\n submodules: recursive\n - name: Install cached dist\n uses: actions/download-artifact@v7\n with:\n name: cargo-dist-cache\n path: ~/.cargo/bin/\n - run: chmod +x ~/.cargo/bin/dist\n # Get all the local artifacts for the global tasks to use (for e.g. checksums)\n - name: Fetch local artifacts\n uses: actions/download-artifact@v7\n with:\n pattern: artifacts-*\n path: target/distrib/\n merge-multiple: true\n - id: cargo-dist\n shell: bash\n run: |\n dist build ${{ needs.plan.outputs.tag-flag }} --output-format=json \"--artifacts=global\" > dist-manifest.json\n echo \"dist ran successfully\"\n\n # Parse out what we just built and upload it to scratch storage\n echo \"paths<> \"$GITHUB_OUTPUT\"\n jq --raw-output \".upload_files[]\" dist-manifest.json >> \"$GITHUB_OUTPUT\"\n echo \"EOF\" >> \"$GITHUB_OUTPUT\"\n\n cp dist-manifest.json \"$BUILD_MANIFEST_NAME\"\n - name: \"Upload artifacts\"\n uses: actions/upload-artifact@v6\n with:\n name: artifacts-build-global\n path: |\n ${{ steps.cargo-dist.outputs.paths }}\n ${{ env.BUILD_MANIFEST_NAME }}\n # Determines if we should publish/announce\n host:\n needs:\n - plan\n - build-local-artifacts\n - build-global-artifacts\n # Only run if we're \"publishing\", and only if plan, local and global didn't fail (skipped is fine)\n if: ${{ always() && needs.plan.result == 'success' && needs.plan.outputs.publishing == 'true' && (needs.build-global-artifacts.result == 'skipped' || needs.build-global-artifacts.result == 'success') && (needs.build-local-artifacts.result == 'skipped' || needs.build-local-artifacts.result == 'success') }}\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n runs-on: \"ubuntu-22.04\"\n outputs:\n val: ${{ steps.host.outputs.manifest }}\n steps:\n - uses: actions/checkout@v6\n with:\n persist-credentials: false\n submodules: recursive\n - name: Install cached dist\n uses: actions/download-artifact@v7\n with:\n name: cargo-dist-cache\n path: ~/.cargo/bin/\n - run: chmod +x ~/.cargo/bin/dist\n # Fetch artifacts from scratch-storage\n - name: Fetch artifacts\n uses: actions/download-artifact@v7\n with:\n pattern: artifacts-*\n path: target/distrib/\n merge-multiple: true\n - id: host\n shell: bash\n run: |\n dist host ${{ needs.plan.outputs.tag-flag }} --steps=upload --steps=release --output-format=json > dist-manifest.json\n echo \"artifacts uploaded and released successfully\"\n cat dist-manifest.json\n echo \"manifest=$(jq -c \".\" dist-manifest.json)\" >> \"$GITHUB_OUTPUT\"\n - name: \"Upload dist-manifest.json\"\n uses: actions/upload-artifact@v6\n with:\n # Overwrite the previous copy\n name: artifacts-dist-manifest\n path: dist-manifest.json\n # Create a GitHub Release while uploading all files to it\n - name: \"Download GitHub Artifacts\"\n uses: actions/download-artifact@v7\n with:\n pattern: artifacts-*\n path: artifacts\n merge-multiple: true\n - name: Cleanup\n run: |\n # Remove the granular manifests\n rm -f artifacts/*-dist-manifest.json\n - name: Create GitHub Release\n env:\n PRERELEASE_FLAG: \"${{ fromJson(steps.host.outputs.manifest).announcement_is_prerelease && '--prerelease' || '' }}\"\n ANNOUNCEMENT_TITLE: \"${{ fromJson(steps.host.outputs.manifest).announcement_title }}\"\n ANNOUNCEMENT_BODY: \"${{ fromJson(steps.host.outputs.manifest).announcement_github_body }}\"\n RELEASE_COMMIT: \"${{ github.sha }}\"\n run: |\n # Write and read notes from a file to avoid quoting breaking things\n echo \"$ANNOUNCEMENT_BODY\" > $RUNNER_TEMP/notes.txt\n\n gh release create \"${{ needs.plan.outputs.tag }}\" --target \"$RELEASE_COMMIT\" $PRERELEASE_FLAG --title \"$ANNOUNCEMENT_TITLE\" --notes-file \"$RUNNER_TEMP/notes.txt\" artifacts/*\n\n announce:\n needs:\n - plan\n - host\n # use \"always() && ...\" to allow us to wait for all publish jobs while\n # still allowing individual publish jobs to skip themselves (for prereleases).\n # \"host\" however must run to completion, no skipping allowed!\n if: ${{ always() && needs.host.result == 'success' }}\n runs-on: \"ubuntu-22.04\"\n env:\n GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n steps:\n - uses: actions/checkout@v6\n with:\n persist-credentials: false\n submodules: recursive\n" }, { "path": ".github/workflows/rust.yml", "content": "name: Rust\n\non:\n push:\n branches: [main]\n paths-ignore:\n - \"ui/**\"\n pull_request:\n branches: [main]\n paths-ignore:\n - \"ui/**\"\n\nenv:\n CARGO_TERM_COLOR: always\n\njobs:\n build:\n strategy:\n matrix:\n os: [depot-ubuntu-24.04, macos-14, windows-latest]\n runs-on: ${{ matrix.os }}\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install rust\n uses: dtolnay/rust-toolchain@master\n with:\n toolchain: 1.94.0\n\n - uses: actions/cache@v5\n with:\n path: |\n ~/.cargo/registry\n ~/.cargo/git\n target\n key: ${{ runner.os }}-cargo-release-${{ hashFiles('**/Cargo.lock') }}\n\n - name: Run cargo build common\n run: cargo build -p atuin-common --locked --release\n\n - name: Run cargo build client\n run: cargo build -p atuin-client --locked --release\n\n - name: Run cargo build server\n run: cargo build -p atuin-server --locked --release\n\n - name: Run cargo build main\n run: cargo build --all --locked --release\n\n cross-compile:\n strategy:\n matrix:\n # There was an attempt to make cross-compiles also work on FreeBSD, but that failed with:\n #\n # warning: libelf.so.2, needed by <...>/libkvm.so, not found (try using -rpath or -rpath-link)\n target: [x86_64-unknown-illumos]\n runs-on: depot-ubuntu-24.04\n steps:\n - uses: actions/checkout@v6\n\n - name: Install cross\n uses: taiki-e/install-action@v2\n with:\n tool: cross\n\n - uses: actions/cache@v5\n with:\n path: |\n ~/.cargo/registry\n ~/.cargo/git\n target\n key: ${{ matrix.target }}-cross-compile-${{ hashFiles('**/Cargo.lock') }}\n\n - name: Run cross build common\n run: cross build -p atuin-common --locked --target ${{ matrix.target }}\n\n - name: Run cross build client\n run: cross build -p atuin-client --locked --target ${{ matrix.target }}\n\n - name: Run cross build server\n run: cross build -p atuin-server --locked --target ${{ matrix.target }}\n\n - name: Run cross build main\n run: |\n cross build --all --locked --target ${{ matrix.target }}\n\n unit-test:\n strategy:\n matrix:\n os: [depot-ubuntu-24.04, macos-14, windows-latest]\n runs-on: ${{ matrix.os }}\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install rust\n uses: dtolnay/rust-toolchain@master\n with:\n toolchain: 1.94.0\n\n - uses: taiki-e/install-action@v2\n name: Install nextest\n with:\n tool: cargo-nextest\n\n - uses: actions/cache@v5\n with:\n path: |\n ~/.cargo/registry\n ~/.cargo/git\n target\n key: ${{ runner.os }}-cargo-debug-${{ hashFiles('**/Cargo.lock') }}\n\n - name: Run cargo test\n run: cargo nextest run --lib --bins\n\n check:\n strategy:\n matrix:\n os: [depot-ubuntu-24.04, macos-14, windows-latest]\n runs-on: ${{ matrix.os }}\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install rust\n uses: dtolnay/rust-toolchain@master\n with:\n toolchain: 1.94.0\n\n - uses: actions/cache@v5\n with:\n path: |\n ~/.cargo/registry\n ~/.cargo/git\n target\n key: ${{ runner.os }}-cargo-debug-${{ hashFiles('**/Cargo.lock') }}\n\n - name: Run cargo check (all features)\n run: cargo check --all-features --workspace\n\n - name: Run cargo check (no features)\n run: cargo check --no-default-features --workspace\n\n - name: Run cargo check (sync)\n run: cargo check --no-default-features --features sync --workspace\n\n - name: Run cargo check (server)\n run: cargo check -p atuin-server\n\n - name: Run cargo check (client only)\n run: cargo check --no-default-features --features client --workspace\n\n integration-test:\n runs-on: depot-ubuntu-24.04\n\n services:\n postgres:\n image: postgres\n env:\n POSTGRES_USER: atuin\n POSTGRES_PASSWORD: pass\n POSTGRES_DB: atuin\n ports:\n - 5432:5432\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install rust\n uses: dtolnay/rust-toolchain@master\n with:\n toolchain: 1.94.0\n\n - uses: taiki-e/install-action@v2\n name: Install nextest\n with:\n tool: cargo-nextest\n\n - uses: actions/cache@v5\n with:\n path: |\n ~/.cargo/registry\n ~/.cargo/git\n target\n key: ${{ runner.os }}-cargo-debug-${{ hashFiles('**/Cargo.lock') }}\n\n - name: Run cargo test\n run: cargo nextest run --test '*'\n env:\n ATUIN_DB_URI: postgres://atuin:pass@localhost:5432/atuin\n\n clippy:\n runs-on: depot-ubuntu-24.04\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install latest rust\n uses: dtolnay/rust-toolchain@master\n with:\n toolchain: 1.94.0\n components: clippy\n\n - uses: actions/cache@v5\n with:\n path: |\n ~/.cargo/registry\n ~/.cargo/git\n target\n key: ${{ runner.os }}-cargo-debug-${{ hashFiles('**/Cargo.lock') }}\n\n - name: Run clippy\n run: cargo clippy -- -D warnings -D clippy::redundant_clone\n\n format:\n runs-on: depot-ubuntu-24.04\n\n steps:\n - uses: actions/checkout@v6\n\n - name: Install latest rust\n uses: dtolnay/rust-toolchain@master\n with:\n toolchain: 1.94.0\n components: rustfmt\n\n - name: Format\n run: cargo fmt -- --check\n" }, { "path": ".github/workflows/shellcheck.yml", "content": "name: Shellcheck\n\non:\n push:\n branches: [ main ]\n pull_request:\n branches: [ main ]\n\njobs:\n shellcheck:\n runs-on: depot-ubuntu-24.04\n\n steps:\n - uses: actions/checkout@v6\n - name: Run shellcheck\n uses: ludeeus/action-shellcheck@master\n env:\n SHELLCHECK_OPTS: \"-e SC2148\"\n" }, { "path": ".github/workflows/update-nix-deps.yml", "content": "name: Update Nix Deps\non:\n workflow_dispatch: # allows manual triggering\n schedule:\n - cron: '0 0 1 * *' # runs monthly on the first day of the month at 00:00\n\njobs:\n lockfile:\n runs-on: depot-ubuntu-24.04\n if: github.repository == 'atuinsh/atuin'\n steps:\n - name: Checkout repository\n uses: actions/checkout@v6\n - name: Install Nix\n uses: DeterminateSystems/nix-installer-action@main\n - name: Update flake.lock\n uses: DeterminateSystems/update-flake-lock@main\n with:\n pr-title: \"chore(deps): update flake.lock\"\n pr-labels: |\n dependencies\n" }, { "path": ".gitignore", "content": ".DS_Store\n/target\n*/target\n.env\n.idea/\n.vscode/\nresult\npublish.sh\n.envrc\n.planning/\n\nui/backend/target\nui/backend/gen\n\nsqlite-server.db*\n" }, { "path": ".mailmap", "content": "networkException \nViolet Shreve \nChris Rose \nConrad Ludgate \nCristian Le \nDennis Trautwein \nEllie Huxtable \nEllie Huxtable \nFrank Hamand \nJakob Schrettenbrunner \nNemo157 \nRichard de Boer \nSandro \nTymanWasTaken \n" }, { "path": ".rustfmt.toml", "content": "reorder_imports = true\n# uncomment once stable\n#imports_granularity = \"crate\"\n#group_imports = \"StdExternalCrate\"\n" }, { "path": "AGENTS.md", "content": "# Atuin\n\nShell history tool. Replaces your shell's built-in history with a SQLite database, adds context (cwd, exit code, duration, hostname), and optionally syncs across machines with end-to-end encryption.\n\n## Workspace crates\n\n```\natuin CLI binary + TUI (clap, ratatui, crossterm)\natuin-client Client library: local DB, encryption, sync, settings\natuin-common Shared types, API models, utils\natuin-daemon Background gRPC daemon (tonic) for shell hooks\natuin-dotfiles Alias/var sync via record store\natuin-history Sorting algorithms, stats\natuin-kv Key-value store (synced)\natuin-scripts Script management (minijinja)\natuin-server HTTP sync server (axum) - lib + standalone binary\natuin-server-database Database trait for server\natuin-server-postgres Postgres implementation (sqlx)\natuin-server-sqlite SQLite implementation (sqlx)\n```\n\n## Two sync protocols\n\n- **V1 (legacy)**: Syncs history entries directly. Being phased out. Toggleable via `sync_v1_enabled`.\n- **V2 (current)**: Record store abstraction. All data types (history, KV, aliases, vars, scripts) share the same sync infrastructure using tagged records. Envelope-encrypted with PASETO V4 and per-record CEKs.\n\n## Encryption\n\n- **V1**: XSalsa20Poly1305 (secretbox). Key at `~/.local/share/atuin/key`.\n- **V2**: PASETO V4 Local (XChaCha20-Poly1305 + Blake2b). Envelope encryption: each record gets a random CEK wrapped with the master key. Record metadata (id, idx, version, tag, host) is authenticated as implicit assertions.\n\n## Databases\n\n- **Client**: SQLite everywhere. Separate DBs for history, record store, KV, scripts. All use sqlx + WAL mode.\n- **Server**: Postgres (primary) or SQLite. Auto-detected from URI prefix.\n- Migrations live alongside each crate. Never modify existing migrations, only add new ones.\n\n## Hot paths\n\n`history start`, `history end`, and `init` skip database initialization for latency. Don't add DB calls to these without good reason.\n\n## Conventions\n\n- Rust 2024 edition, toolchain 1.93.1.\n- Errors: `eyre::Result` in binaries, `thiserror` for typed errors in libraries.\n- Async: tokio. Client uses `current_thread`; server uses `multi_thread`.\n- `#![deny(unsafe_code)]` on client/common, `#![forbid(unsafe_code)]` on server.\n- Clippy: `pedantic` + `nursery` on main crate. CI enforces `-D warnings -D clippy::redundant_clone`.\n- Format: `cargo fmt`. Only non-default: `reorder_imports = true`.\n- IDs: UUIDv7 (time-ordered), newtype wrappers (`HistoryId`, `RecordId`, `HostId`).\n- Serialization: MessagePack for encrypted payloads, JSON for API, TOML for config.\n- Storage traits: `Database` (client), `Store` (record store), `Database` (server) -- all `async_trait`.\n- History builders: `HistoryImported`, `HistoryCaptured`, `HistoryFromDb` with compile-time field validation.\n- Feature flags: `client`, `sync`, `daemon`, `clipboard`, `check-update`.\n\n## Testing\n\n- Unit tests inline with `#[cfg(test)]`, async via `#[tokio::test]`.\n- Integration tests in `crates/atuin/tests/` need Postgres (`ATUIN_DB_URI` env var).\n- Use `\":memory:\"` SQLite for unit tests needing a database.\n- Runner: `cargo nextest`.\n- Benchmarks: `divan` in `atuin-history`.\n\n## Build and check\n\n```sh\ncargo build\ncargo test\ncargo clippy -- -D warnings\ncargo fmt --check\n```\n" }, { "path": "CHANGELOG.md", "content": "# Changelog\n\nAll notable changes to this project will be documented in this file.\n\n## [unreleased]\n\n### Bug Fixes\n\n- Nushell 0.111; future Nushell 0.112 support ([#3266](https://github.com/atuinsh/atuin/issues/3266))\n\n\n### Features\n\n- Call atuin setup from install script ([#3265](https://github.com/atuinsh/atuin/issues/3265))\n- Allow headless account ops against Hub server ([#3280](https://github.com/atuinsh/atuin/issues/3280))\n- Add custom filtering and scoring mechanisms\n\n\n### Miscellaneous Tasks\n\n- *(ci)* Migrate to depot runners ([#3279](https://github.com/atuinsh/atuin/issues/3279))\n- *(ci)* Use depot to build docker images too ([#3281](https://github.com/atuinsh/atuin/issues/3281))\n- Update changelog\n- Update permissions in Docker workflow ([#3283](https://github.com/atuinsh/atuin/issues/3283))\n- Change CHANGELOG format to be easier to parse\n- Symlink changelog so dist can pick it up\n- Vendor nucleo-ext + fork, so we can depend on our changes properly ([#3284](https://github.com/atuinsh/atuin/issues/3284))\n\n\n## 18.13.2\n\n### Miscellaneous Tasks\n\n- *(release)* Building windows aarch64 was overly optimistic\n- Update changelog\n\n\n## 18.13.1\n\n### Miscellaneous Tasks\n\n- *(release)* Update dist, remove custom runners\n- Update changelog\n\n\n## 18.13.0\n\n### Bug Fixes\n\n- *(deps)* Add use-dev-tty to crossterm in atuin-ai ([#3185](https://github.com/atuinsh/atuin/issues/3185))\n- *(docs)* Update Postgres volume path in Docker as required by pg18 ([#3174](https://github.com/atuinsh/atuin/issues/3174))\n- Systemd Exec for separate server binary ([#3176](https://github.com/atuinsh/atuin/issues/3176))\n- Multiline commands with fish ([#3179](https://github.com/atuinsh/atuin/issues/3179))\n- Silent DB failures e.g. when disk is full ([#3183](https://github.com/atuinsh/atuin/issues/3183))\n- Forward $PATH to tmux popup in zsh ([#3198](https://github.com/atuinsh/atuin/issues/3198))\n- Dramatically decrease daemon memory usage ([#3211](https://github.com/atuinsh/atuin/issues/3211))\n- Regen cargo dist\n- Clear script database before rebuild to prevent unique constraint violation ([#3232](https://github.com/atuinsh/atuin/issues/3232))\n- Support Nushell 0.111 ([#3249](https://github.com/atuinsh/atuin/issues/3249))\n- Ctrl-c not exiting ai ([#3256](https://github.com/atuinsh/atuin/issues/3256))\n\n\n### Documentation\n\n- Update config.md to remove NuShell support note ([#3190](https://github.com/atuinsh/atuin/issues/3190))\n- Document `search.filters` ([#3195](https://github.com/atuinsh/atuin/issues/3195))\n- Clean up doc references for sqlite-based self-hosting ([#3216](https://github.com/atuinsh/atuin/issues/3216))\n- Document daemon-fuzzy search mode ([#3254](https://github.com/atuinsh/atuin/issues/3254))\n\n\n### Features\n\n- *(docs)* Add Shell Integration and Interoperability docs ([#3163](https://github.com/atuinsh/atuin/issues/3163))\n- `switch-context` ([#3149](https://github.com/atuinsh/atuin/issues/3149))\n- Add Hub authentication for future sync + extra features ([#3010](https://github.com/atuinsh/atuin/issues/3010))\n- Add Atuin AI inline CLI MVP ([#3178](https://github.com/atuinsh/atuin/issues/3178))\n- Add autostart and pid management to daemon ([#3180](https://github.com/atuinsh/atuin/issues/3180))\n- Generate commands or ask questions with `atuin ai` ([#3199](https://github.com/atuinsh/atuin/issues/3199))\n- Add history author/intent metadata and v1 record version ([#3205](https://github.com/atuinsh/atuin/issues/3205))\n- In-memory search index with atuin daemon ([#3201](https://github.com/atuinsh/atuin/issues/3201))\n- Update script for smoother setup ([#3230](https://github.com/atuinsh/atuin/issues/3230))\n- Initial draft of atuin-shell ([#3206](https://github.com/atuinsh/atuin/issues/3206))\n- Allow setting multipliers for frequency, recency, and frecency scores ([#3235](https://github.com/atuinsh/atuin/issues/3235))\n- Allow running `atuin search -i` as subcommand ([#3208](https://github.com/atuinsh/atuin/issues/3208))\n- Use pty proxy for rendering tui popups without clearing the terminal ([#3234](https://github.com/atuinsh/atuin/issues/3234))\n- Allow authenticating with Atuin Hub ([#3237](https://github.com/atuinsh/atuin/issues/3237))\n- Initialize Atuin AI by default with `atuin init` ([#3255](https://github.com/atuinsh/atuin/issues/3255))\n- Add `atuin setup` ([#3257](https://github.com/atuinsh/atuin/issues/3257))\n\n\n### Miscellaneous Tasks\n\n- Update changelog\n- Update changelog\n- Update changelog\n- Use workspace versions ([#3210](https://github.com/atuinsh/atuin/issues/3210))\n- Move atuin ai subcommand into core binary ([#3212](https://github.com/atuinsh/atuin/issues/3212))\n- Update changelog\n- Update to Rust 1.94 ([#3247](https://github.com/atuinsh/atuin/issues/3247))\n- Strip symbols in dist profile to reduce binary size\n- Upgrade thiserror 1.x to 2.x to deduplicate dependency\n- Upgrade axum 0.7 to 0.8 to deduplicate with tonic's axum\n- Update changelog\n- Update changelog\n- Update changelog\n- Update changelog\n\n\n## 18.12.1\n\n### Bug Fixes\n\n- *(shell)* Fix ATUIN_SESSION errors in tmux popup ([#3170](https://github.com/atuinsh/atuin/issues/3170))\n- *(tui)* Enter in vim normal mode, shift-tab keybind ([#3158](https://github.com/atuinsh/atuin/issues/3158))\n- Server start commands for Docker. ([#3160](https://github.com/atuinsh/atuin/issues/3160))\n\n\n### Features\n\n- Expand keybinding system with vim motions, media keys, and inspector improvements ([#3161](https://github.com/atuinsh/atuin/issues/3161))\n- Add original-input-empty keybind condition ([#3171](https://github.com/atuinsh/atuin/issues/3171))\n\n\n### Miscellaneous Tasks\n\n- Update changelog\n\n\n## 18.12.0\n\n### Bug Fixes\n\n- *(powershell)* Preserve `$LASTEXITCODE` ([#3120](https://github.com/atuinsh/atuin/issues/3120))\n- *(powershell)* Display search stderr ([#3146](https://github.com/atuinsh/atuin/issues/3146))\n- *(search)* Allow hyphen-prefixed query args like `---` ([#3129](https://github.com/atuinsh/atuin/issues/3129))\n- *(tui)* Space and F1-F24 keys not handled properly by new keybind system ([#3138](https://github.com/atuinsh/atuin/issues/3138))\n- *(ui)* Don't draw a leading space for command\n- *(ui)* Time column can take up to 9 cells\n- *(ui)* Align cursor with the expand column (usually the command)\n- *(ui)* Align cursor when expand column is in the middle ([#3103](https://github.com/atuinsh/atuin/issues/3103))\n- Zsh import multiline issue ([#2799](https://github.com/atuinsh/atuin/issues/2799))\n- Do not hit sync v1 endpoints for status\n- Do not hit sync v1 endpoints for status ([#3102](https://github.com/atuinsh/atuin/issues/3102))\n- Do not set ATUIN_SESSION if it is already set ([#3107](https://github.com/atuinsh/atuin/issues/3107))\n- Custom data dir test on windows ([#3109](https://github.com/atuinsh/atuin/issues/3109))\n- New session on shlvl change ([#3111](https://github.com/atuinsh/atuin/issues/3111))\n- Larger exit column width on Windows ([#3119](https://github.com/atuinsh/atuin/issues/3119))\n- Halt sync loop if server returns an empty page ([#3122](https://github.com/atuinsh/atuin/issues/3122))\n- Use directories crate for home dir resolution ([#3125](https://github.com/atuinsh/atuin/issues/3125))\n- Tab behaving like enter, eprintln ([#3135](https://github.com/atuinsh/atuin/issues/3135))\n- Issue with shift and modifier keys ([#3143](https://github.com/atuinsh/atuin/issues/3143))\n- Remove invalid IF EXISTS from sqlite drop column migration ([#3145](https://github.com/atuinsh/atuin/issues/3145))\n\n\n### Documentation\n\n- *(CONTRIBUTING)* Update links ([#3117](https://github.com/atuinsh/atuin/issues/3117))\n- *(README)* Update links ([#3116](https://github.com/atuinsh/atuin/issues/3116))\n- *(config)* Clarify scope of directory filter_mode ([#3082](https://github.com/atuinsh/atuin/issues/3082))\n- *(configuration)* Describe new utility \"atuin-bind\" for Bash ([#3064](https://github.com/atuinsh/atuin/issues/3064))\n- *(installation)* Add mise alternative installation method ([#3066](https://github.com/atuinsh/atuin/issues/3066))\n- Various improvements to the `atuin import` docs ([#3062](https://github.com/atuinsh/atuin/issues/3062))\n- Disambiguate 'setup' (noun) vs. 'set up' (verb) ([#3061](https://github.com/atuinsh/atuin/issues/3061))\n- Fix punctuation and grammar in basic usage guide ([#3063](https://github.com/atuinsh/atuin/issues/3063))\n- Expand and clarify usage of the history prune command ([#3084](https://github.com/atuinsh/atuin/issues/3084))\n- Small edit to themes website file ([#3069](https://github.com/atuinsh/atuin/issues/3069))\n- Config/ with initial uid:gid\n- Add PowerShell install instructions\n- Add PowerShell and Windows install instructions ([#3096](https://github.com/atuinsh/atuin/issues/3096))\n- Update the `[keys]` docs ([#3114](https://github.com/atuinsh/atuin/issues/3114))\n- Add history deletion guide ([#3130](https://github.com/atuinsh/atuin/issues/3130))\n- Update how to use Docker to self-host ([#3148](https://github.com/atuinsh/atuin/issues/3148))\n- Add IRC contact information to README\n\n\n### Features\n\n- *(dotfiles)* Add sort and filter options to alias/var list ([#3131](https://github.com/atuinsh/atuin/issues/3131))\n- *(theme)* Note new default theme name and syntax ([#3080](https://github.com/atuinsh/atuin/issues/3080))\n- *(tui)* Add clear-to-start/end actions ([#3141](https://github.com/atuinsh/atuin/issues/3141))\n- *(ui)* Highlight fulltext search as fulltext search instead of fuzzy search\n- *(ui)* Highlight fulltext search as fulltext search instead of fuzzy search ([#3098](https://github.com/atuinsh/atuin/issues/3098))\n- *(ultracompact)* Adds setting for ultracompact mode ([#3079](https://github.com/atuinsh/atuin/issues/3079))\n- Add custom column support ([#3089](https://github.com/atuinsh/atuin/issues/3089))\n- Left arrow/backspace on empty to start edit ([#3090](https://github.com/atuinsh/atuin/issues/3090))\n- Add more vim movement bindings for navigation ([#3041](https://github.com/atuinsh/atuin/issues/3041))\n- Support setting a custom data dir in config ([#3105](https://github.com/atuinsh/atuin/issues/3105))\n- Remove user verification functionality ([#3108](https://github.com/atuinsh/atuin/issues/3108))\n- Add option to use tmux display-popup ([#3058](https://github.com/atuinsh/atuin/issues/3058))\n- Move atuin-server to its own binary ([#3112](https://github.com/atuinsh/atuin/issues/3112))\n- Add a parameter to the sync to specify the download/upload page ([#2408](https://github.com/atuinsh/atuin/issues/2408))\n- Replace several files with a sqlite db ([#3128](https://github.com/atuinsh/atuin/issues/3128))\n- Add new custom keybinding system for search TUI ([#3127](https://github.com/atuinsh/atuin/issues/3127))\n\n\n### Miscellaneous Tasks\n\n- Remove total_history from api index response ([#3094](https://github.com/atuinsh/atuin/issues/3094))\n - **BREAKING**: remove total_history from api index response ([#3094](https://github.com/atuinsh/atuin/issues/3094))\n- Update to rust 1.93\n- Update to rust 1.93 ([#3101](https://github.com/atuinsh/atuin/issues/3101))\n- Update changelog\n- Update agents.md ([#3126](https://github.com/atuinsh/atuin/issues/3126))\n- Update changelog\n- Update changelog\n- Update changelog\n\n\n### Theming\n\n- Explain how to set ANSI codes directly ([#3065](https://github.com/atuinsh/atuin/issues/3065))\n\n\n### Faq\n\n- Add alternative projects ([#3076](https://github.com/atuinsh/atuin/issues/3076))\n\n\n## 18.11.0\n\n### Bug Fixes\n\n- *(bash)* Fix issues with intermediate key sequences in the vi editing mode ([#2977](https://github.com/atuinsh/atuin/issues/2977))\n- *(bash)* Work around a keybinding bug of Bash 5.1 ([#2975](https://github.com/atuinsh/atuin/issues/2975))\n- *(bash/blesh)* Suppress error message for auto-complete source ([#2976](https://github.com/atuinsh/atuin/issues/2976))\n- *(powershell)* Run `atuin history end` in the background ([#3034](https://github.com/atuinsh/atuin/issues/3034))\n- *(powershell)* Add error safety and cleanup ([#3040](https://github.com/atuinsh/atuin/issues/3040))\n- Highlight the correct place when multibyte characters are involved ([#2965](https://github.com/atuinsh/atuin/issues/2965))\n- Prevent interactive search crash when update check fails ([#3016](https://github.com/atuinsh/atuin/issues/3016))\n- Move thorough search through search.filters w/ workspaces ([#2703](https://github.com/atuinsh/atuin/issues/2703))\n\n\n### Documentation\n\n- Migrate docs from separate repo to `docs` subfolder ([#3018](https://github.com/atuinsh/atuin/issues/3018))\n\n\n### Features\n\n- Support additional history filenames in replxx importer ([#3005](https://github.com/atuinsh/atuin/issues/3005))\n- Add colors to --help/-h ([#3000](https://github.com/atuinsh/atuin/issues/3000))\n- Add support for read replicas to postgres ([#3029](https://github.com/atuinsh/atuin/issues/3029))\n- Allow disabling sync v1 ([#3030](https://github.com/atuinsh/atuin/issues/3030))\n- Consider atuin dotfile aliases when calculating atuin wrapped ([#3048](https://github.com/atuinsh/atuin/issues/3048))\n- Add session and uuid column support to history list ([#3049](https://github.com/atuinsh/atuin/issues/3049))\n\n\n### Miscellaneous Tasks\n\n- *(nix)* Prevent deprecation warning on evaluation ([#3006](https://github.com/atuinsh/atuin/issues/3006))\n- Update changelog\n- Adjust update wording ([#2974](https://github.com/atuinsh/atuin/issues/2974))\n- Add Windows builds, second try ([#2966](https://github.com/atuinsh/atuin/issues/2966))\n- Update to rust 1.91 ([#2981](https://github.com/atuinsh/atuin/issues/2981))\n- Add Atuin Desktop information to install script\n- Remove trailing whitespace ([#2985](https://github.com/atuinsh/atuin/issues/2985))\n- Fix typo ([#2994](https://github.com/atuinsh/atuin/issues/2994))\n- Clarify docstring of the enter_accept config key ([#3003](https://github.com/atuinsh/atuin/issues/3003))\n- Fix github action syntax for variables ([#2998](https://github.com/atuinsh/atuin/issues/2998))\n- Add AGENTS.md\n- Update changelog\n- Remove x86_64 mac from build targets ([#3052](https://github.com/atuinsh/atuin/issues/3052))\n\n\n### Build\n\n- *(nix)* Update rust toolchain hash ([#2990](https://github.com/atuinsh/atuin/issues/2990))\n\n\n## 18.10.0\n\n### Bug Fixes\n\n- Stats ngram window size cli parsing ([#2946](https://github.com/atuinsh/atuin/issues/2946))\n\n\n### Features\n\n- *(bash)* Use Readline's accept-line for enter_accept ([#2953](https://github.com/atuinsh/atuin/issues/2953))\n- Add commit to displayed version info ([#2922](https://github.com/atuinsh/atuin/issues/2922))\n- Add import from PowerShell history ([#2864](https://github.com/atuinsh/atuin/issues/2864))\n- Interactive Inspector ([#2319](https://github.com/atuinsh/atuin/issues/2319))\n- Nu ≥ 0.106.0 support commandline accept ([#2957](https://github.com/atuinsh/atuin/issues/2957))\n\n\n### Miscellaneous Tasks\n\n- Update rusty_paseto and rusty_paserk ([#2942](https://github.com/atuinsh/atuin/issues/2942))\n- Update changelog\n\n\n## 18.9.0\n\n### Bug Fixes\n\n- *(dotfiles)* Properly escape spaces/quotes in vars\n- Clippy issues on Windows ([#2856](https://github.com/atuinsh/atuin/issues/2856))\n- Honor timezone in inspector stats ([#2853](https://github.com/atuinsh/atuin/issues/2853))\n- Make status exit 1 if not logged in ([#2843](https://github.com/atuinsh/atuin/issues/2843))\n- Match logic of theme directory with settings directory, so ATUIN_CONFIG_DIR is respected ([#2707](https://github.com/atuinsh/atuin/issues/2707))\n- Expand path for daemon.socket_path ([#2870](https://github.com/atuinsh/atuin/issues/2870))\n- Use fullscreen if `inline_height` is too large ([#2888](https://github.com/atuinsh/atuin/issues/2888))\n- Clean up new rustc and clippy warnings on Rust 1.89\n- `cargo update` and changes needed to accomodate it\n- Run `cargo fmt`\n- Clippy warnings I don't have on my version of clippy\n- Add forgotten `rust-toolchain.toml` to match changes (oops)\n- Update version in Cargo.toml + github workflows\n- Clippy warnings\n- Dissociate command_chaining from enter_accept\n- Remove __atuin_chain_command__ prefix\n- Docker compose link ([#2914](https://github.com/atuinsh/atuin/issues/2914))\n- Fish up binding ([#2902](https://github.com/atuinsh/atuin/issues/2902))\n\n\n### Features\n\n- *(stats)* Add dotnet to default common subcommands\n- *(tui)* Select entries using number in vim-normal mode. closes #2368 ([#2893](https://github.com/atuinsh/atuin/issues/2893))\n- *(tui)* Add show_numeric_shortcuts config to hide 1-9 shortcuts ([#2766](https://github.com/atuinsh/atuin/issues/2766))\n- Highlight matches in interactive search ([#2653](https://github.com/atuinsh/atuin/issues/2653))\n- Add session-preload filter mode to include global history from before session start\n- Add various acceptance keys ([#2928](https://github.com/atuinsh/atuin/issues/2928))\n- More accurately filter secret tokens ([#2932](https://github.com/atuinsh/atuin/issues/2932))\n- Add shell pipelines to command chaining ([#2938](https://github.com/atuinsh/atuin/issues/2938))\n\n\n### Miscellaneous Tasks\n\n- Update changelog\n- Remove legacy Apple SDK frameworks ([#2885](https://github.com/atuinsh/atuin/issues/2885))\n- Update dist workflows\n- Update to Rust 1.90 ([#2916](https://github.com/atuinsh/atuin/issues/2916))\n\n\n### Refactor\n\n- Shell environment variables\n\n\n### Build\n\n- Update flake.nix with new sha256\n\n\n## 18.8.0\n\n### Bug Fixes\n\n- *(build)* Enable sqlite feature for sqlite server ([#2848](https://github.com/atuinsh/atuin/issues/2848))\n- Make login exit 1 if already logged in ([#2832](https://github.com/atuinsh/atuin/issues/2832))\n- Use transaction for idx consistency checking ([#2840](https://github.com/atuinsh/atuin/issues/2840))\n- Ensure the idx cache is cleaned on deletion, only insert if records are inserted ([#2841](https://github.com/atuinsh/atuin/issues/2841))\n\n\n### Features\n\n- Command chaining ([#2834](https://github.com/atuinsh/atuin/issues/2834))\n- Add info for 'official' plugins ([#2835](https://github.com/atuinsh/atuin/issues/2835))\n- Support multi part commands ([#2836](https://github.com/atuinsh/atuin/issues/2836)) ([#2837](https://github.com/atuinsh/atuin/issues/2837))\n- Add inline_height_shell_up_key_binding option ([#2817](https://github.com/atuinsh/atuin/issues/2817))\n- Add IDX_CACHE_ROLLOUT ([#2850](https://github.com/atuinsh/atuin/issues/2850))\n\n\n### Miscellaneous Tasks\n\n- Update to rust 1.88 ([#2815](https://github.com/atuinsh/atuin/issues/2815))\n\n\n### Nushell\n\n- Fix `get -i` deprecation ([#2829](https://github.com/atuinsh/atuin/issues/2829))\n\n\n## 18.7.1\n\n### Bug Fixes\n\n- Add check for postgresql prefix ([#2825](https://github.com/atuinsh/atuin/issues/2825))\n\n\n### Miscellaneous Tasks\n\n- Update changelog\n\n\n## 18.7.0\n\n### Bug Fixes\n\n- *(api)* Allow trailing slashes in sync_address ([#2760](https://github.com/atuinsh/atuin/issues/2760))\n- *(doctor)* Mention the required ble.sh version ([#2774](https://github.com/atuinsh/atuin/issues/2774))\n- *(search)* Prevent panic on malformed format strings ([#2776](https://github.com/atuinsh/atuin/issues/2776)) ([#2777](https://github.com/atuinsh/atuin/issues/2777))\n- Clarify that HISTFILE, if used, must be exported ([#2758](https://github.com/atuinsh/atuin/issues/2758))\n- Don't print errors in `zsh_autosuggest` helper ([#2780](https://github.com/atuinsh/atuin/issues/2780))\n- `atuin.nu` enchancements ([#2778](https://github.com/atuinsh/atuin/issues/2778))\n- Refuse \"--dupkeep 0\" ([#2807](https://github.com/atuinsh/atuin/issues/2807))\n\n\n### Features\n\n- Add sqlite server support for self-hosting ([#2770](https://github.com/atuinsh/atuin/issues/2770))\n\n\n### Miscellaneous Tasks\n\n- *(ci)* Install toolchain that matches rust-toolchain.toml ([#2759](https://github.com/atuinsh/atuin/issues/2759))\n- Allow setting script DB path ([#2750](https://github.com/atuinsh/atuin/issues/2750))\n\n\n## 18.6.1\n\n### Bug Fixes\n\n- Selection vs render issue ([#2706](https://github.com/atuinsh/atuin/issues/2706))\n\n\n### Features\n\n- *(stats)* Add jj to default common subcommands ([#2708](https://github.com/atuinsh/atuin/issues/2708))\n- Delete duplicate history ([#2697](https://github.com/atuinsh/atuin/issues/2697))\n- Sort `atuin store status` output ([#2719](https://github.com/atuinsh/atuin/issues/2719))\n- Implement KV as a write-through cache ([#2732](https://github.com/atuinsh/atuin/issues/2732))\n\n\n### Miscellaneous Tasks\n\n- Use native github arm64 runner ([#2690](https://github.com/atuinsh/atuin/issues/2690))\n- Fix typos ([#2668](https://github.com/atuinsh/atuin/issues/2668))\n\n\n## 18.5.0\n\n### Bug Fixes\n\n- *(1289)* Clear terminal area if inline ([#2600](https://github.com/atuinsh/atuin/issues/2600))\n- *(bash)* Fix preexec of child Bash session started by enter_accept ([#2558](https://github.com/atuinsh/atuin/issues/2558))\n- *(build)* Change atuin-daemon build script .proto paths ([#2638](https://github.com/atuinsh/atuin/issues/2638))\n- *(kv)* Filter deleted keys from `kv list` ([#2665](https://github.com/atuinsh/atuin/issues/2665))\n- *(stats)* Ignore leading environment variables when calculating stats ([#2659](https://github.com/atuinsh/atuin/issues/2659))\n- *(wrapped)* Fix crash when history is empty ([#2508](https://github.com/atuinsh/atuin/issues/2508))\n- *(zsh)* Fix an error introduced earilier with support for bracketed paste mode ([#2651](https://github.com/atuinsh/atuin/issues/2651))\n- *(zsh)* Avoid calling user-defined widgets when searching for history position ([#2670](https://github.com/atuinsh/atuin/issues/2670))\n- Add .histfile as file to look for when doing atuin import zsh ([#2588](https://github.com/atuinsh/atuin/issues/2588))\n- Panic when invoking delete on empty tui ([#2584](https://github.com/atuinsh/atuin/issues/2584))\n- Sql files checksums ([#2601](https://github.com/atuinsh/atuin/issues/2601))\n- Up binding with fish 4.0 ([#2613](https://github.com/atuinsh/atuin/issues/2613)) ([#2616](https://github.com/atuinsh/atuin/issues/2616))\n- Don't save empty commands ([#2605](https://github.com/atuinsh/atuin/issues/2605))\n- Improve broken symlink error handling ([#2589](https://github.com/atuinsh/atuin/issues/2589))\n- Multiline command does not honour max_preview_height ([#2624](https://github.com/atuinsh/atuin/issues/2624))\n- Typeerror in client sync code ([#2647](https://github.com/atuinsh/atuin/issues/2647))\n- Add redundant clones to clippy and cleanup instances of it ([#2654](https://github.com/atuinsh/atuin/issues/2654))\n- Allow -ve values for timezone ([#2609](https://github.com/atuinsh/atuin/issues/2609))\n- Fish up binding bug ([#2677](https://github.com/atuinsh/atuin/issues/2677))\n- Switch to astral cargo-dist ([#2687](https://github.com/atuinsh/atuin/issues/2687))\n\n\n### Documentation\n\n- Update logo and badges in README for zh-CN ([#2392](https://github.com/atuinsh/atuin/issues/2392))\n\n\n### Features\n\n- *(client)* Update AWS secrets env var handling checks ([#2501](https://github.com/atuinsh/atuin/issues/2501))\n- *(health)* Add health check endpoint at `/healthz` ([#2549](https://github.com/atuinsh/atuin/issues/2549))\n- *(kv)* Add support for 'atuin kv delete' ([#2660](https://github.com/atuinsh/atuin/issues/2660))\n- *(wrapped)* Add more pkg managers ([#2503](https://github.com/atuinsh/atuin/issues/2503))\n- *(zsh)* Try to go to the position in zsh's history ([#1469](https://github.com/atuinsh/atuin/issues/1469))\n- *(zsh)* Re-enable bracketed paste ([#2646](https://github.com/atuinsh/atuin/issues/2646))\n- Add the --print0 option to search ([#2562](https://github.com/atuinsh/atuin/issues/2562))\n- Make new arrow key behavior configurable ([#2606](https://github.com/atuinsh/atuin/issues/2606))\n- Use readline binding for ctrl-a when it is not the prefix ([#2626](https://github.com/atuinsh/atuin/issues/2626))\n- Option to include duplicate commands when printing history commands ([#2407](https://github.com/atuinsh/atuin/issues/2407))\n- Binaries as subcommands ([#2661](https://github.com/atuinsh/atuin/issues/2661))\n- Support storing, syncing and executing scripts ([#2644](https://github.com/atuinsh/atuin/issues/2644))\n- Add 'atuin scripts rm' and 'atuin scripts ls' aliases; allow reading from stdin ([#2680](https://github.com/atuinsh/atuin/issues/2680))\n\n\n### Miscellaneous Tasks\n\n- Remove unneeded dependencies ([#2523](https://github.com/atuinsh/atuin/issues/2523))\n- Update rust toolchain to 1.85 ([#2618](https://github.com/atuinsh/atuin/issues/2618))\n- Align daemon and client sync freq ([#2628](https://github.com/atuinsh/atuin/issues/2628))\n- Migrate to rust 2024 ([#2635](https://github.com/atuinsh/atuin/issues/2635))\n- Show host and user in inspector ([#2634](https://github.com/atuinsh/atuin/issues/2634))\n- Update to rust 1.85.1 ([#2642](https://github.com/atuinsh/atuin/issues/2642))\n- Update to rust 1.86 ([#2666](https://github.com/atuinsh/atuin/issues/2666))\n\n\n### Performance\n\n- Cache `SECRET_PATTERNS`'s `RegexSet` ([#2570](https://github.com/atuinsh/atuin/issues/2570))\n\n\n### Styling\n\n- Avoid calling `unwrap()` when we don't have to ([#2519](https://github.com/atuinsh/atuin/issues/2519))\n\n\n### Build\n\n- *(nix)* Bump `flake.lock` ([#2637](https://github.com/atuinsh/atuin/issues/2637))\n\n\n### Flake.lock\n\n- Update ([#2463](https://github.com/atuinsh/atuin/issues/2463))\n\n\n## 18.4.0\n\n### Bug Fixes\n\n- *(crate)* Add missing description ([#2106](https://github.com/atuinsh/atuin/issues/2106))\n- *(crate)* Add description to daemon crate ([#2107](https://github.com/atuinsh/atuin/issues/2107))\n- *(daemon)* Add context to error when unable to connect ([#2394](https://github.com/atuinsh/atuin/issues/2394))\n- *(deps)* Pin tiny_bip to 1.0.0 until breaking change resolved ([#2412](https://github.com/atuinsh/atuin/issues/2412))\n- *(docker)* Update Dockerfile ([#2369](https://github.com/atuinsh/atuin/issues/2369))\n- *(gui)* Update deps ([#2116](https://github.com/atuinsh/atuin/issues/2116))\n- *(gui)* Add support for checking if the cli is installed on windows ([#2162](https://github.com/atuinsh/atuin/issues/2162))\n- *(gui)* WeekInfo call on Edge ([#2252](https://github.com/atuinsh/atuin/issues/2252))\n- *(gui)* Add \\r for windows (shouldn't effect unix bc they should ignore it) ([#2253](https://github.com/atuinsh/atuin/issues/2253))\n- *(gui)* Terminal resize overflow ([#2285](https://github.com/atuinsh/atuin/issues/2285))\n- *(gui)* Kill child on block stop ([#2288](https://github.com/atuinsh/atuin/issues/2288))\n- *(gui)* Do not hardcode db path ([#2309](https://github.com/atuinsh/atuin/issues/2309))\n- *(gui)* Double return on mac/linux ([#2311](https://github.com/atuinsh/atuin/issues/2311))\n- *(gui)* Cursor positioning on new doc creation ([#2310](https://github.com/atuinsh/atuin/issues/2310))\n- *(gui)* Random ts errors ([#2316](https://github.com/atuinsh/atuin/issues/2316))\n- *(history)* Logic for store_failed=false ([#2284](https://github.com/atuinsh/atuin/issues/2284))\n- *(mail)* Incorrect alias and error logs ([#2346](https://github.com/atuinsh/atuin/issues/2346))\n- *(mail)* Enable correct tls features for postmark client ([#2347](https://github.com/atuinsh/atuin/issues/2347))\n- *(theme)* Restore original colours ([#2339](https://github.com/atuinsh/atuin/issues/2339))\n- *(themes)* Restore default theme, refactor ([#2294](https://github.com/atuinsh/atuin/issues/2294))\n- *(tui)* Press ctrl-a twice should jump to beginning of line ([#2246](https://github.com/atuinsh/atuin/issues/2246))\n- *(tui)* Don't panic when search result is empty and up is pressed ([#2395](https://github.com/atuinsh/atuin/issues/2395))\n- Cargo binstall config ([#2112](https://github.com/atuinsh/atuin/issues/2112))\n- Unitless sync_frequence = 0 not parsed by humantime ([#2154](https://github.com/atuinsh/atuin/issues/2154))\n- Some --help comments didn't show properly ([#2176](https://github.com/atuinsh/atuin/issues/2176))\n- Ensure we cleanup all tables when deleting ([#2191](https://github.com/atuinsh/atuin/issues/2191))\n- Add idx cache unique index ([#2226](https://github.com/atuinsh/atuin/issues/2226))\n- Idx cache inconsistency ([#2231](https://github.com/atuinsh/atuin/issues/2231))\n- Ambiguous column name ([#2232](https://github.com/atuinsh/atuin/issues/2232))\n- Atuin-daemon optional dependency ([#2306](https://github.com/atuinsh/atuin/issues/2306))\n- Windows build error ([#2321](https://github.com/atuinsh/atuin/issues/2321))\n- Codespell config still references the ui ([#2330](https://github.com/atuinsh/atuin/issues/2330))\n- Remove dbg! macro ([#2355](https://github.com/atuinsh/atuin/issues/2355))\n- Disable mail by default, resolve #2404 ([#2405](https://github.com/atuinsh/atuin/issues/2405))\n- Time offset display in `atuin status` ([#2433](https://github.com/atuinsh/atuin/issues/2433))\n- Disable the actuated mirror on the x86 docker builder ([#2443](https://github.com/atuinsh/atuin/issues/2443))\n\n\n### Documentation\n\n- *(README)* Fix broken link ([#2206](https://github.com/atuinsh/atuin/issues/2206))\n- *(gui)* Update README ([#2283](https://github.com/atuinsh/atuin/issues/2283))\n- Streamline readme ([#2203](https://github.com/atuinsh/atuin/issues/2203))\n- Update quickstart install command ([#2205](https://github.com/atuinsh/atuin/issues/2205))\n\n\n### Features\n\n- *(bash/blesh)* Hook into BLE_ONLOAD to resolve loading order issue ([#2234](https://github.com/atuinsh/atuin/issues/2234))\n- *(client)* Add filter mode enablement and ordering configuration ([#2430](https://github.com/atuinsh/atuin/issues/2430))\n- *(daemon)* Follow XDG_RUNTIME_DIR if set ([#2171](https://github.com/atuinsh/atuin/issues/2171))\n- *(gui)* Automatically install and setup the cli/shell ([#2139](https://github.com/atuinsh/atuin/issues/2139))\n- *(gui)* Add activity calendar to the homepage ([#2160](https://github.com/atuinsh/atuin/issues/2160))\n- *(gui)* Cache zustand store in localstorage ([#2168](https://github.com/atuinsh/atuin/issues/2168))\n- *(gui)* Toast with prompt for cli install, rather than auto ([#2173](https://github.com/atuinsh/atuin/issues/2173))\n- *(gui)* Runbooks that run ([#2233](https://github.com/atuinsh/atuin/issues/2233))\n- *(gui)* Use fancy new side nav ([#2243](https://github.com/atuinsh/atuin/issues/2243))\n- *(gui)* Add runbook list, ability to create and delete, sql storage ([#2282](https://github.com/atuinsh/atuin/issues/2282))\n- *(gui)* Background terminals and more ([#2303](https://github.com/atuinsh/atuin/issues/2303))\n- *(gui)* Clean up home page, fix a few bugs ([#2304](https://github.com/atuinsh/atuin/issues/2304))\n- *(gui)* Allow interacting with the embedded terminal ([#2312](https://github.com/atuinsh/atuin/issues/2312))\n- *(gui)* Directory block, re-org of some code ([#2314](https://github.com/atuinsh/atuin/issues/2314))\n- *(gui)* Folder select dialogue for directory block ([#2315](https://github.com/atuinsh/atuin/issues/2315))\n- *(history)* Filter out various environment variables containing potential secrets ([#2174](https://github.com/atuinsh/atuin/issues/2174))\n- *(tui)* Configurable prefix character ([#2157](https://github.com/atuinsh/atuin/issues/2157))\n- *(tui)* Customizable Themes ([#2236](https://github.com/atuinsh/atuin/issues/2236))\n- *(tui)* Fixed preview height option ([#2286](https://github.com/atuinsh/atuin/issues/2286))\n- Use cargo-dist installer from our install script ([#2108](https://github.com/atuinsh/atuin/issues/2108))\n- Add user account verification ([#2190](https://github.com/atuinsh/atuin/issues/2190))\n- Add GitLab PAT to secret patterns ([#2196](https://github.com/atuinsh/atuin/issues/2196))\n- Add several other GitHub access token patterns ([#2200](https://github.com/atuinsh/atuin/issues/2200))\n- Add npm, Netlify and Pulumi tokens to secret patterns ([#2210](https://github.com/atuinsh/atuin/issues/2210))\n- Allow advertising a fake version to clients ([#2228](https://github.com/atuinsh/atuin/issues/2228))\n- Monitor idx cache consistency before switching ([#2229](https://github.com/atuinsh/atuin/issues/2229))\n- Ultracompact Mode (search-only) ([#2357](https://github.com/atuinsh/atuin/issues/2357))\n- Right Arrow to modify selected command ([#2453](https://github.com/atuinsh/atuin/issues/2453))\n- Provide additional clarity around key management ([#2467](https://github.com/atuinsh/atuin/issues/2467))\n- Add `atuin wrapped` ([#2493](https://github.com/atuinsh/atuin/issues/2493))\n\n\n### Miscellaneous Tasks\n\n- *(build)* Compile protobufs with protox ([#2122](https://github.com/atuinsh/atuin/issues/2122))\n- *(ci)* Do not run current ci for ui ([#2189](https://github.com/atuinsh/atuin/issues/2189))\n- *(ci)* Codespell again ([#2332](https://github.com/atuinsh/atuin/issues/2332))\n- *(deps-dev)* Bump @tauri-apps/cli in /ui ([#2135](https://github.com/atuinsh/atuin/issues/2135))\n- *(deps-dev)* Bump vite from 5.2.13 to 5.3.1 in /ui ([#2150](https://github.com/atuinsh/atuin/issues/2150))\n- *(deps-dev)* Bump @tauri-apps/cli in /ui ([#2277](https://github.com/atuinsh/atuin/issues/2277))\n- *(deps-dev)* Bump tailwindcss from 3.4.4 to 3.4.6 in /ui ([#2301](https://github.com/atuinsh/atuin/issues/2301))\n- *(install)* Use posix sh, not bash ([#2204](https://github.com/atuinsh/atuin/issues/2204))\n- *(nix)* De-couple atuin nix build from nixpkgs rustc version ([#2123](https://github.com/atuinsh/atuin/issues/2123))\n- Add installer e2e tests ([#2110](https://github.com/atuinsh/atuin/issues/2110))\n- Remove unnecessary proto import ([#2120](https://github.com/atuinsh/atuin/issues/2120))\n- Update to rust 1.78\n- Add audit config, ignore RUSTSEC-2023-0071 ([#2126](https://github.com/atuinsh/atuin/issues/2126))\n- Setup dependabot for the ui ([#2128](https://github.com/atuinsh/atuin/issues/2128))\n- Cargo and pnpm update ([#2127](https://github.com/atuinsh/atuin/issues/2127))\n- Update to rust 1.79 ([#2138](https://github.com/atuinsh/atuin/issues/2138))\n- Update to cargo-dist 0.16, enable attestations ([#2156](https://github.com/atuinsh/atuin/issues/2156))\n- Do not use package managers in installer ([#2201](https://github.com/atuinsh/atuin/issues/2201))\n- Enable record sync by default ([#2255](https://github.com/atuinsh/atuin/issues/2255))\n- Remove ui directory ([#2329](https://github.com/atuinsh/atuin/issues/2329))\n- Update to rust 1.80 ([#2344](https://github.com/atuinsh/atuin/issues/2344))\n- Update rust to `1.80.1` ([#2362](https://github.com/atuinsh/atuin/issues/2362))\n- Enable inline height and compact by default ([#2249](https://github.com/atuinsh/atuin/issues/2249))\n- Update to rust 1.82 ([#2432](https://github.com/atuinsh/atuin/issues/2432))\n- Update cargo-dist ([#2471](https://github.com/atuinsh/atuin/issues/2471))\n\n\n### Performance\n\n- *(search)* Benchmark smart sort ([#2202](https://github.com/atuinsh/atuin/issues/2202))\n- Create idx cache table ([#2140](https://github.com/atuinsh/atuin/issues/2140))\n- Write to the idx cache ([#2225](https://github.com/atuinsh/atuin/issues/2225))\n\n\n### Testing\n\n- Add env ATUIN_TEST_LOCAL_TIMEOUT to control test timeout of SQLite ([#2337](https://github.com/atuinsh/atuin/issues/2337))\n\n\n### Flake.lock\n\n- Update ([#2213](https://github.com/atuinsh/atuin/issues/2213))\n- Update ([#2378](https://github.com/atuinsh/atuin/issues/2378))\n- Update ([#2402](https://github.com/atuinsh/atuin/issues/2402))\n\n\n## 18.3.0\n\n### Bug Fixes\n\n- *(bash)* Fix a workaround for bash-5.2 keybindings ([#2060](https://github.com/atuinsh/atuin/issues/2060))\n- *(ci)* Release workflow ([#1978](https://github.com/atuinsh/atuin/issues/1978))\n- *(client)* Better error reporting on login/registration ([#2076](https://github.com/atuinsh/atuin/issues/2076))\n- *(config)* Add quotes for strategy value in comment ([#1993](https://github.com/atuinsh/atuin/issues/1993))\n- *(daemon)* Do not try to sync if logged out ([#2037](https://github.com/atuinsh/atuin/issues/2037))\n- *(deps)* Replace parse_duration with humantime ([#2074](https://github.com/atuinsh/atuin/issues/2074))\n- *(dotfiles)* Alias import with init output ([#1970](https://github.com/atuinsh/atuin/issues/1970))\n- *(dotfiles)* Fish alias import ([#1972](https://github.com/atuinsh/atuin/issues/1972))\n- *(dotfiles)* More fish alias import ([#1974](https://github.com/atuinsh/atuin/issues/1974))\n- *(dotfiles)* Unquote aliases before quoting ([#1976](https://github.com/atuinsh/atuin/issues/1976))\n- *(dotfiles)* Allow clearing aliases, disable import ([#1995](https://github.com/atuinsh/atuin/issues/1995))\n- *(stats)* Generation for commands starting with a pipe ([#2058](https://github.com/atuinsh/atuin/issues/2058))\n- *(ui)* Handle being logged out gracefully ([#2052](https://github.com/atuinsh/atuin/issues/2052))\n- *(ui)* Fix mistake in last pr ([#2053](https://github.com/atuinsh/atuin/issues/2053))\n- Support not-mac for default shell ([#1960](https://github.com/atuinsh/atuin/issues/1960))\n- Adapt help to `enter_accept` config ([#2001](https://github.com/atuinsh/atuin/issues/2001))\n- Add protobuf compiler to docker image ([#2009](https://github.com/atuinsh/atuin/issues/2009))\n- Add incremental rebuild to daemon loop ([#2010](https://github.com/atuinsh/atuin/issues/2010))\n- Alias enable/enabled in settings ([#2021](https://github.com/atuinsh/atuin/issues/2021))\n- Bogus error message wording ([#1283](https://github.com/atuinsh/atuin/issues/1283))\n- Save sync time in daemon ([#2029](https://github.com/atuinsh/atuin/issues/2029))\n- Redact password in database URI when logging ([#2032](https://github.com/atuinsh/atuin/issues/2032))\n- Save sync time in daemon ([#2051](https://github.com/atuinsh/atuin/issues/2051))\n- Replace serde_yaml::to_string with serde_json::to_string_yaml ([#2087](https://github.com/atuinsh/atuin/issues/2087))\n\n\n### Documentation\n\n- Fix \"From source\" `cd` command ([#1973](https://github.com/atuinsh/atuin/issues/1973))\n- Add docs for store subcommand ([#2097](https://github.com/atuinsh/atuin/issues/2097))\n\n\n### Features\n\n- *(daemon)* Add support for daemon on windows ([#2014](https://github.com/atuinsh/atuin/issues/2014))\n- *(doctor)* Detect active preexec framework ([#1955](https://github.com/atuinsh/atuin/issues/1955))\n- *(doctor)* Report sqlite version ([#2075](https://github.com/atuinsh/atuin/issues/2075))\n- *(dotfiles)* Support syncing shell/env vars ([#1977](https://github.com/atuinsh/atuin/issues/1977))\n- *(gui)* Work on home page, sort state ([#1956](https://github.com/atuinsh/atuin/issues/1956))\n- *(history)* Create atuin-history, add stats to it ([#1990](https://github.com/atuinsh/atuin/issues/1990))\n- *(install)* Add Tuxedo OS ([#2018](https://github.com/atuinsh/atuin/issues/2018))\n- *(server)* Add me endpoint ([#1954](https://github.com/atuinsh/atuin/issues/1954))\n- *(ui)* Scroll history infinitely ([#1999](https://github.com/atuinsh/atuin/issues/1999))\n- *(ui)* Add history explore ([#2022](https://github.com/atuinsh/atuin/issues/2022))\n- *(ui)* Use correct username on welcome screen ([#2050](https://github.com/atuinsh/atuin/issues/2050))\n- *(ui)* Add login/register dialog ([#2056](https://github.com/atuinsh/atuin/issues/2056))\n- *(ui)* Setup single-instance ([#2093](https://github.com/atuinsh/atuin/issues/2093))\n- *(ui/dotfiles)* Add vars ([#1989](https://github.com/atuinsh/atuin/issues/1989))\n- Allow ignoring failed commands ([#1957](https://github.com/atuinsh/atuin/issues/1957))\n- Show preview auto ([#1804](https://github.com/atuinsh/atuin/issues/1804))\n- Add background daemon ([#2006](https://github.com/atuinsh/atuin/issues/2006))\n- Support importing from replxx history files ([#2024](https://github.com/atuinsh/atuin/issues/2024))\n- Support systemd socket activation for daemon ([#2039](https://github.com/atuinsh/atuin/issues/2039))\n\n\n### Miscellaneous Tasks\n\n- *(ci)* Don't run \"Update Nix Deps\" CI on forks ([#2070](https://github.com/atuinsh/atuin/issues/2070))\n- *(codespell)* Ignore CODE_OF_CONDUCT ([#2044](https://github.com/atuinsh/atuin/issues/2044))\n- *(install)* Log cargo and rustc version ([#2068](https://github.com/atuinsh/atuin/issues/2068))\n- *(release)* V18.3.0-prerelease.1 ([#2090](https://github.com/atuinsh/atuin/issues/2090))\n- Move crates into crates/ dir ([#1958](https://github.com/atuinsh/atuin/issues/1958))\n- Fix atuin crate readme ([#1959](https://github.com/atuinsh/atuin/issues/1959))\n- Add some more logging to handlers ([#1971](https://github.com/atuinsh/atuin/issues/1971))\n- Add some more debug logs ([#1979](https://github.com/atuinsh/atuin/issues/1979))\n- Clarify default config file ([#2026](https://github.com/atuinsh/atuin/issues/2026))\n- Handle rate limited responses ([#2057](https://github.com/atuinsh/atuin/issues/2057))\n- Add Systemd config for self-hosted server ([#1879](https://github.com/atuinsh/atuin/issues/1879))\n- Switch to cargo dist for releases ([#2085](https://github.com/atuinsh/atuin/issues/2085))\n- Update email, gitignore, tweak ui ([#2094](https://github.com/atuinsh/atuin/issues/2094))\n- Show scope in changelog ([#2102](https://github.com/atuinsh/atuin/issues/2102))\n\n\n### Performance\n\n- *(nushell)* Use version.(major|minor|patch) if available ([#1963](https://github.com/atuinsh/atuin/issues/1963))\n- Only open the database for commands if strictly required ([#2043](https://github.com/atuinsh/atuin/issues/2043))\n\n\n### Refactor\n\n- Preview_auto to use enum and different option ([#1991](https://github.com/atuinsh/atuin/issues/1991))\n\n\n## 18.2.0\n\n### Bug Fixes\n\n- *(bash)* Do not use \"return\" to cancel initialization ([#1928](https://github.com/atuinsh/atuin/issues/1928))\n- *(crate)* Add missing description ([#1861](https://github.com/atuinsh/atuin/issues/1861))\n- *(doctor)* Detect preexec plugin using env ATUIN_PREEXEC_BACKEND ([#1856](https://github.com/atuinsh/atuin/issues/1856))\n- *(install)* Install script echo ([#1899](https://github.com/atuinsh/atuin/issues/1899))\n- *(nu)* Update atuin.nu to resolve 0.92 deprecation ([#1913](https://github.com/atuinsh/atuin/issues/1913))\n- *(search)* Allow empty search ([#1866](https://github.com/atuinsh/atuin/issues/1866))\n- *(search)* Case insensitive hostname filtering ([#1883](https://github.com/atuinsh/atuin/issues/1883))\n- Pass search query in via env ([#1865](https://github.com/atuinsh/atuin/issues/1865))\n- Pass search query in via env for *Nushell* ([#1874](https://github.com/atuinsh/atuin/issues/1874))\n- Report non-decodable errors correctly ([#1915](https://github.com/atuinsh/atuin/issues/1915))\n- Use spawn_blocking for file access during async context ([#1936](https://github.com/atuinsh/atuin/issues/1936))\n\n\n### Documentation\n\n- *(bash-preexec)* Describe the limitation of missing commands ([#1937](https://github.com/atuinsh/atuin/issues/1937))\n- Add security contact ([#1867](https://github.com/atuinsh/atuin/issues/1867))\n- Add install instructions for cave/exherbo linux in README.md ([#1927](https://github.com/atuinsh/atuin/issues/1927))\n- Add missing cli help text ([#1945](https://github.com/atuinsh/atuin/issues/1945))\n\n\n### Features\n\n- *(bash/blesh)* Use _ble_exec_time_ata for duration even in bash < 5 ([#1940](https://github.com/atuinsh/atuin/issues/1940))\n- *(dotfiles)* Add alias import ([#1938](https://github.com/atuinsh/atuin/issues/1938))\n- *(gui)* Add base structure ([#1935](https://github.com/atuinsh/atuin/issues/1935))\n- *(install)* Update install.sh to support KDE Neon ([#1908](https://github.com/atuinsh/atuin/issues/1908))\n- *(search)* Process [C-h] and [C-?] as representations of backspace ([#1857](https://github.com/atuinsh/atuin/issues/1857))\n- *(search)* Allow specifying search query as an env var ([#1863](https://github.com/atuinsh/atuin/issues/1863))\n- *(search)* Add better search scoring ([#1885](https://github.com/atuinsh/atuin/issues/1885))\n- *(server)* Check PG version before running migrations ([#1868](https://github.com/atuinsh/atuin/issues/1868))\n- Add atuin prefix binding ([#1875](https://github.com/atuinsh/atuin/issues/1875))\n- Sync v2 default for new installs ([#1914](https://github.com/atuinsh/atuin/issues/1914))\n- Add 'ctrl-a a' to jump to beginning of line ([#1917](https://github.com/atuinsh/atuin/issues/1917))\n- Prevents stderr from going to the screen ([#1933](https://github.com/atuinsh/atuin/issues/1933))\n\n\n### Miscellaneous Tasks\n\n- *(ci)* Add codespell support (config, workflow) and make it fix some typos ([#1916](https://github.com/atuinsh/atuin/issues/1916))\n- *(gui)* Cargo update ([#1943](https://github.com/atuinsh/atuin/issues/1943))\n- Add issue form ([#1871](https://github.com/atuinsh/atuin/issues/1871))\n- Require atuin doctor in issue form ([#1872](https://github.com/atuinsh/atuin/issues/1872))\n- Add section to issue form ([#1873](https://github.com/atuinsh/atuin/issues/1873))\n\n\n### Performance\n\n- *(dotfiles)* Cache aliases and read straight from file ([#1918](https://github.com/atuinsh/atuin/issues/1918))\n\n\n## 18.1.0\n\n### Bug Fixes\n\n- *(bash)* Rework #1509 to recover from the preexec failure ([#1729](https://github.com/atuinsh/atuin/issues/1729))\n- *(build)* Make atuin compile on non-win/mac/linux platforms ([#1825](https://github.com/atuinsh/atuin/issues/1825))\n- *(client)* No panic on empty inspector ([#1768](https://github.com/atuinsh/atuin/issues/1768))\n- *(doctor)* Use a different method to detect env vars ([#1819](https://github.com/atuinsh/atuin/issues/1819))\n- *(dotfiles)* Use latest client ([#1859](https://github.com/atuinsh/atuin/issues/1859))\n- *(import/zsh-histdb)* Missing or wrong fields ([#1740](https://github.com/atuinsh/atuin/issues/1740))\n- *(nix)* Set meta.mainProgram in the package ([#1823](https://github.com/atuinsh/atuin/issues/1823))\n- *(nushell)* Readd up-arrow keybinding, now with menu handling ([#1770](https://github.com/atuinsh/atuin/issues/1770))\n- *(regex)* Disable regex error logs ([#1806](https://github.com/atuinsh/atuin/issues/1806))\n- *(stats)* Enable multiple command stats to be shown using unicode_segmentation ([#1739](https://github.com/atuinsh/atuin/issues/1739))\n- *(store-init)* Re-sync after running auto store init ([#1834](https://github.com/atuinsh/atuin/issues/1834))\n- *(sync)* Check store length after sync, not before ([#1805](https://github.com/atuinsh/atuin/issues/1805))\n- *(sync)* Record size limiter ([#1827](https://github.com/atuinsh/atuin/issues/1827))\n- *(tz)* Attempt to fix timezone reading ([#1810](https://github.com/atuinsh/atuin/issues/1810))\n- *(ui)* Don't preserve for empty space ([#1712](https://github.com/atuinsh/atuin/issues/1712))\n- *(xonsh)* Add xonsh to auto import, respect $HISTFILE in xonsh import, and fix issue with up-arrow keybinding in xonsh ([#1711](https://github.com/atuinsh/atuin/issues/1711))\n- Fish init ([#1725](https://github.com/atuinsh/atuin/issues/1725))\n- Typo ([#1741](https://github.com/atuinsh/atuin/issues/1741))\n- Check session file exists for status command ([#1756](https://github.com/atuinsh/atuin/issues/1756))\n- Ensure sync time is saved for sync v2 ([#1758](https://github.com/atuinsh/atuin/issues/1758))\n- Missing characters in preview ([#1803](https://github.com/atuinsh/atuin/issues/1803))\n- Doctor shell wording ([#1858](https://github.com/atuinsh/atuin/issues/1858))\n\n\n### Documentation\n\n- Minor formatting updates to the default config.toml ([#1689](https://github.com/atuinsh/atuin/issues/1689))\n- Update docker compose ([#1818](https://github.com/atuinsh/atuin/issues/1818))\n- Use db name env variable also in uri ([#1840](https://github.com/atuinsh/atuin/issues/1840))\n\n\n### Features\n\n- *(client)* Add config option keys.scroll_exits ([#1744](https://github.com/atuinsh/atuin/issues/1744))\n- *(dotfiles)* Add enable setting to dotfiles, disable by default ([#1829](https://github.com/atuinsh/atuin/issues/1829))\n- *(nix)* Add update action ([#1779](https://github.com/atuinsh/atuin/issues/1779))\n- *(nu)* Return early if history is disabled ([#1807](https://github.com/atuinsh/atuin/issues/1807))\n- *(nushell)* Add nushell completion generation ([#1791](https://github.com/atuinsh/atuin/issues/1791))\n- *(search)* Process Ctrl+m for kitty keyboard protocol ([#1720](https://github.com/atuinsh/atuin/issues/1720))\n- *(stats)* Normalize formatting of default config, suggest nix ([#1764](https://github.com/atuinsh/atuin/issues/1764))\n- *(stats)* Add linux sysadmin commands to common_subcommands ([#1784](https://github.com/atuinsh/atuin/issues/1784))\n- *(ui)* Add config setting for showing tabs ([#1755](https://github.com/atuinsh/atuin/issues/1755))\n- Use ATUIN_TEST_SQLITE_STORE_TIMEOUT to specify test timeout of SQLite store ([#1703](https://github.com/atuinsh/atuin/issues/1703))\n- Add 'a', 'A', 'h', and 'l' bindings to vim-normal mode ([#1697](https://github.com/atuinsh/atuin/issues/1697))\n- Add xonsh history import ([#1678](https://github.com/atuinsh/atuin/issues/1678))\n- Add 'ignored_commands' option to stats ([#1722](https://github.com/atuinsh/atuin/issues/1722))\n- Support syncing aliases ([#1721](https://github.com/atuinsh/atuin/issues/1721))\n- Change fulltext to do multi substring match ([#1660](https://github.com/atuinsh/atuin/issues/1660))\n- Add history prune subcommand ([#1743](https://github.com/atuinsh/atuin/issues/1743))\n- Add alias feedback and list command ([#1747](https://github.com/atuinsh/atuin/issues/1747))\n- Add PHP package manager \"composer\" to list of default common subcommands ([#1757](https://github.com/atuinsh/atuin/issues/1757))\n- Add '/', '?', and 'I' bindings to vim-normal mode ([#1760](https://github.com/atuinsh/atuin/issues/1760))\n- Add `CTRL+[` binding as `` alias ([#1787](https://github.com/atuinsh/atuin/issues/1787))\n- Add atuin doctor ([#1796](https://github.com/atuinsh/atuin/issues/1796))\n- Add checks for common setup issues ([#1799](https://github.com/atuinsh/atuin/issues/1799))\n- Support regex with r/.../ syntax ([#1745](https://github.com/atuinsh/atuin/issues/1745))\n- Guard against ancient versions of bash where this does not work. ([#1794](https://github.com/atuinsh/atuin/issues/1794))\n- Add automatic history store init ([#1831](https://github.com/atuinsh/atuin/issues/1831))\n- Adds info command to show env vars and config files ([#1841](https://github.com/atuinsh/atuin/issues/1841))\n\n\n### Miscellaneous Tasks\n\n- *(ci)* Add cross-compile job for illumos ([#1830](https://github.com/atuinsh/atuin/issues/1830))\n- *(ci)* Setup nextest ([#1848](https://github.com/atuinsh/atuin/issues/1848))\n- Do not show history table stats when using records ([#1835](https://github.com/atuinsh/atuin/issues/1835))\n\n\n### Performance\n\n- Optimize history init-store ([#1691](https://github.com/atuinsh/atuin/issues/1691))\n\n\n### Refactor\n\n- *(alias)* Clarify operation result for working with aliases ([#1748](https://github.com/atuinsh/atuin/issues/1748))\n- *(nushell)* Update `commandline` syntax, closes #1733 ([#1735](https://github.com/atuinsh/atuin/issues/1735))\n- Rename atuin-config to atuin-dotfiles ([#1817](https://github.com/atuinsh/atuin/issues/1817))\n\n\n## 18.0.1\n\n### Bug Fixes\n\n- Reorder the exit of enhanced keyboard mode ([#1694](https://github.com/atuinsh/atuin/issues/1694))\n\n\n## 18.0.0\n\n### Bug Fixes\n\n- *(bash)* Avoid unexpected `atuin history start` for keybindings ([#1509](https://github.com/atuinsh/atuin/issues/1509))\n- *(bash)* Prevent input to be interpreted as options for blesh auto-complete ([#1511](https://github.com/atuinsh/atuin/issues/1511))\n- *(bash)* Work around custom IFS ([#1514](https://github.com/atuinsh/atuin/issues/1514))\n- *(bash)* Fix and improve the keybinding to `up` ([#1515](https://github.com/atuinsh/atuin/issues/1515))\n- *(bash)* Work around bash < 4 and introduce initialization guards ([#1533](https://github.com/atuinsh/atuin/issues/1533))\n- *(bash)* Strip control chars generated by `\\[\\]` in PS1 with bash-preexec ([#1620](https://github.com/atuinsh/atuin/issues/1620))\n- *(bash/preexec)* Erase the prompt last line before Bash renders it\n- *(bash/preexec)* Erase the previous prompt before overwriting\n- *(bash/preexec)* Support termcap names for tput ([#1670](https://github.com/atuinsh/atuin/issues/1670))\n- *(docs)* Update repo url in CONTRIBUTING.md ([#1594](https://github.com/atuinsh/atuin/issues/1594))\n- *(fish)* Integration on older fishes ([#1563](https://github.com/atuinsh/atuin/issues/1563))\n- *(perm)* Set umask 077 ([#1554](https://github.com/atuinsh/atuin/issues/1554))\n- *(search)* Fix invisible tab title ([#1560](https://github.com/atuinsh/atuin/issues/1560))\n- *(shell)* Fix incorrect timing of child shells ([#1510](https://github.com/atuinsh/atuin/issues/1510))\n- *(sync)* Save sync time when it starts, not ends ([#1573](https://github.com/atuinsh/atuin/issues/1573))\n- *(tests)* Add Settings::utc() for utc settings ([#1677](https://github.com/atuinsh/atuin/issues/1677))\n- *(tui)* Dedupe was removing history ([#1610](https://github.com/atuinsh/atuin/issues/1610))\n- *(windows)* Disables unix specific stuff for windows ([#1557](https://github.com/atuinsh/atuin/issues/1557))\n- Prevent input to be interpreted as options for zsh autosuggestions ([#1506](https://github.com/atuinsh/atuin/issues/1506))\n- Disable musl deb building ([#1525](https://github.com/atuinsh/atuin/issues/1525))\n- Shorten text, use ctrl-o for inspector ([#1561](https://github.com/atuinsh/atuin/issues/1561))\n- Print literal control characters to non terminals ([#1586](https://github.com/atuinsh/atuin/issues/1586))\n- Escape control characters in command preview ([#1588](https://github.com/atuinsh/atuin/issues/1588))\n- Use existing db querying for history list ([#1589](https://github.com/atuinsh/atuin/issues/1589))\n- Add acquire timeout to sqlite database connection ([#1590](https://github.com/atuinsh/atuin/issues/1590))\n- Only escape control characters when writing to terminal ([#1593](https://github.com/atuinsh/atuin/issues/1593))\n- Check for format errors when printing history ([#1623](https://github.com/atuinsh/atuin/issues/1623))\n- Skip padding time if it will overflow the allowed prefix length ([#1630](https://github.com/atuinsh/atuin/issues/1630))\n- Never overwrite the key ([#1657](https://github.com/atuinsh/atuin/issues/1657))\n- Set durability for sqlite to recommended settings ([#1667](https://github.com/atuinsh/atuin/issues/1667))\n- Correct download list for incremental builds ([#1672](https://github.com/atuinsh/atuin/issues/1672))\n\n\n### Documentation\n\n- *(README)* Clarify prerequisites for Bash ([#1686](https://github.com/atuinsh/atuin/issues/1686))\n- *(readme)* Add repology badge ([#1494](https://github.com/atuinsh/atuin/issues/1494))\n- Add forum link to contributing ([#1498](https://github.com/atuinsh/atuin/issues/1498))\n- Refer to image with multi-arch support ([#1513](https://github.com/atuinsh/atuin/issues/1513))\n- Remove activity graph\n- Fix `Destination file already exists` in Nushell ([#1530](https://github.com/atuinsh/atuin/issues/1530))\n- Clarify enter/tab usage ([#1538](https://github.com/atuinsh/atuin/issues/1538))\n- Improve style ([#1537](https://github.com/atuinsh/atuin/issues/1537))\n- Remove old docusaurus ([#1581](https://github.com/atuinsh/atuin/issues/1581))\n- Mention environment variables for custom paths ([#1614](https://github.com/atuinsh/atuin/issues/1614))\n- Create pull_request_template.md ([#1632](https://github.com/atuinsh/atuin/issues/1632))\n- Update CONTRIBUTING.md ([#1633](https://github.com/atuinsh/atuin/issues/1633))\n\n\n### Features\n\n- *(bash)* Support high-resolution timing even without ble.sh ([#1534](https://github.com/atuinsh/atuin/issues/1534))\n- *(search)* Introduce keymap-dependent vim-mode ([#1570](https://github.com/atuinsh/atuin/issues/1570))\n- *(search)* Make cursor style configurable ([#1595](https://github.com/atuinsh/atuin/issues/1595))\n- *(shell)* Bind the Atuin search to \"/\" in vi-normal mode ([#1629](https://github.com/atuinsh/atuin/issues/1629))\n - **BREAKING**: bind the Atuin search to \"/\" in vi-normal mode ([#1629](https://github.com/atuinsh/atuin/issues/1629))\n- *(ui)* Add redraw ([#1519](https://github.com/atuinsh/atuin/issues/1519))\n- *(ui)* Vim mode ([#1553](https://github.com/atuinsh/atuin/issues/1553))\n- *(ui)* When in vim-normal mode apply an alternative highlighting to the selected line ([#1574](https://github.com/atuinsh/atuin/issues/1574))\n- *(zsh)* Update widget names ([#1631](https://github.com/atuinsh/atuin/issues/1631))\n- Enable enhanced keyboard mode ([#1505](https://github.com/atuinsh/atuin/issues/1505))\n- Rework record sync for improved reliability ([#1478](https://github.com/atuinsh/atuin/issues/1478))\n- Include atuin login in secret patterns ([#1518](https://github.com/atuinsh/atuin/issues/1518))\n- Make it clear what you are registering for ([#1523](https://github.com/atuinsh/atuin/issues/1523))\n- Add extended help ([#1540](https://github.com/atuinsh/atuin/issues/1540))\n- Add interactive command inspector ([#1296](https://github.com/atuinsh/atuin/issues/1296))\n- Add better error handling for sync ([#1572](https://github.com/atuinsh/atuin/issues/1572))\n- Add history rebuild ([#1575](https://github.com/atuinsh/atuin/issues/1575))\n- Make deleting from the UI work with record store sync ([#1580](https://github.com/atuinsh/atuin/issues/1580))\n- Add metrics counter for records downloaded ([#1584](https://github.com/atuinsh/atuin/issues/1584))\n- Make store init idempotent ([#1609](https://github.com/atuinsh/atuin/issues/1609))\n- Don't stop with invalid key ([#1612](https://github.com/atuinsh/atuin/issues/1612))\n- Add registered and deleted metrics ([#1622](https://github.com/atuinsh/atuin/issues/1622))\n- Make history list format configurable ([#1638](https://github.com/atuinsh/atuin/issues/1638))\n- Add change-password command & support on server ([#1615](https://github.com/atuinsh/atuin/issues/1615))\n- Automatically init history store when record sync is enabled ([#1634](https://github.com/atuinsh/atuin/issues/1634))\n- Add store push ([#1649](https://github.com/atuinsh/atuin/issues/1649))\n- Reencrypt/rekey local store ([#1662](https://github.com/atuinsh/atuin/issues/1662))\n- Add prefers_reduced_motion flag ([#1645](https://github.com/atuinsh/atuin/issues/1645))\n- Add verify command to local store\n- Add store purge command\n- Failure to decrypt history = failure to sync\n- Add `store push --force`\n- Add `store pull`\n- Disable auto record store init ([#1671](https://github.com/atuinsh/atuin/issues/1671))\n- Add progress bars to sync and store init ([#1684](https://github.com/atuinsh/atuin/issues/1684))\n\n\n### Miscellaneous Tasks\n\n- *(ci)* Use github m1 for release builds ([#1658](https://github.com/atuinsh/atuin/issues/1658))\n- *(ci)* Re-enable test cache, add separate check step ([#1663](https://github.com/atuinsh/atuin/issues/1663))\n- *(ci)* Run rust build/test/check on 3 platforms ([#1675](https://github.com/atuinsh/atuin/issues/1675))\n- Remove the teapot response ([#1496](https://github.com/atuinsh/atuin/issues/1496))\n- Schema cleanup ([#1522](https://github.com/atuinsh/atuin/issues/1522))\n- Update funding ([#1543](https://github.com/atuinsh/atuin/issues/1543))\n- Make clipboard dep optional as a feature ([#1558](https://github.com/atuinsh/atuin/issues/1558))\n- Add feature to allow always disable check update ([#1628](https://github.com/atuinsh/atuin/issues/1628))\n- Use resolver 2, update editions + cargo ([#1635](https://github.com/atuinsh/atuin/issues/1635))\n- Disable nix tests ([#1646](https://github.com/atuinsh/atuin/issues/1646))\n- Set ATUIN_ variables for development in devshell ([#1653](https://github.com/atuinsh/atuin/issues/1653))\n\n\n### Refactor\n\n- *(search)* Refactor vim mode ([#1559](https://github.com/atuinsh/atuin/issues/1559))\n- *(search)* Refactor handling of key inputs ([#1606](https://github.com/atuinsh/atuin/issues/1606))\n- *(shell)* Refactor and localize `HISTORY => __atuin_output` ([#1535](https://github.com/atuinsh/atuin/issues/1535))\n- Use enum instead of magic numbers ([#1499](https://github.com/atuinsh/atuin/issues/1499))\n- String -> HistoryId ([#1512](https://github.com/atuinsh/atuin/issues/1512))\n\n\n### Styling\n\n- *(bash)* Use consistent coding style ([#1528](https://github.com/atuinsh/atuin/issues/1528))\n\n\n### Testing\n\n- Add multi-user integration tests ([#1648](https://github.com/atuinsh/atuin/issues/1648))\n\n\n### Stats\n\n- Misc improvements ([#1613](https://github.com/atuinsh/atuin/issues/1613))\n\n\n## 17.2.1\n\n### Bug Fixes\n\n- *(server)* Typo with default config ([#1493](https://github.com/atuinsh/atuin/issues/1493))\n\n\n## 17.2.0\n\n### Bug Fixes\n\n- *(bash)* Fix loss of the last output line with enter_accept ([#1463](https://github.com/atuinsh/atuin/issues/1463))\n- *(bash)* Improve the support for `enter_accept` with `ble.sh` ([#1465](https://github.com/atuinsh/atuin/issues/1465))\n- *(bash)* Fix small issues of `enter_accept` for the plain Bash ([#1467](https://github.com/atuinsh/atuin/issues/1467))\n- *(bash)* Fix error by the use of ${PS1@P} in bash < 4.4 ([#1488](https://github.com/atuinsh/atuin/issues/1488))\n- *(bash,zsh)* Fix quirks on search cancel ([#1483](https://github.com/atuinsh/atuin/issues/1483))\n- *(clippy)* Ignore struct_field_names ([#1466](https://github.com/atuinsh/atuin/issues/1466))\n- *(docs)* Fix typo ([#1439](https://github.com/atuinsh/atuin/issues/1439))\n- *(docs)* Discord link expired\n- *(history)* Disallow deletion if the '--limit' flag is present ([#1436](https://github.com/atuinsh/atuin/issues/1436))\n- *(import/zsh)* Zsh use a special format to escape some characters ([#1490](https://github.com/atuinsh/atuin/issues/1490))\n- *(install)* Discord broken link\n- *(shell)* Respect ZSH's $ZDOTDIR environment variable ([#1441](https://github.com/atuinsh/atuin/issues/1441))\n- *(stats)* Don't require all fields under [stats] ([#1437](https://github.com/atuinsh/atuin/issues/1437))\n- *(stats)* Time now_local not working\n- *(zsh)* Zsh_autosuggest_strategy for no-unset environment ([#1486](https://github.com/atuinsh/atuin/issues/1486))\n\n\n### Documentation\n\n- *(readme)* Add actuated linkback\n- *(readme)* Fix light/dark mode logo\n- *(readme)* Use picture element for logo\n- Add link to forum\n- Align setup links in docs and readme ([#1446](https://github.com/atuinsh/atuin/issues/1446))\n- Add Void Linux install instruction ([#1445](https://github.com/atuinsh/atuin/issues/1445))\n- Add fish install script ([#1447](https://github.com/atuinsh/atuin/issues/1447))\n- Correct link\n- Add docs for zsh-autosuggestion integration ([#1480](https://github.com/atuinsh/atuin/issues/1480))\n- Remove stray character from README\n- Update logo ([#1481](https://github.com/atuinsh/atuin/issues/1481))\n\n\n### Features\n\n- *(bash)* Provide auto-complete source for ble.sh ([#1487](https://github.com/atuinsh/atuin/issues/1487))\n- *(shell)* Support high-resolution duration if available ([#1484](https://github.com/atuinsh/atuin/issues/1484))\n- Add semver checking to client requests ([#1456](https://github.com/atuinsh/atuin/issues/1456))\n- Add TLS to atuin-server ([#1457](https://github.com/atuinsh/atuin/issues/1457))\n- Integrate with zsh-autosuggestions ([#1479](https://github.com/atuinsh/atuin/issues/1479))\n\n\n### Miscellaneous Tasks\n\n- *(repo)* Remove issue config ([#1433](https://github.com/atuinsh/atuin/issues/1433))\n- Remove issue template ([#1444](https://github.com/atuinsh/atuin/issues/1444))\n\n\n### Refactor\n\n- *(bash)* Factorize `__atuin_accept_line` ([#1476](https://github.com/atuinsh/atuin/issues/1476))\n- *(bash)* Refactor and optimize `__atuin_accept_line` ([#1482](https://github.com/atuinsh/atuin/issues/1482))\n\n\n## 17.1.0\n\n### Bug Fixes\n\n- *(fish)* Clean up the fish script options ([#1370](https://github.com/atuinsh/atuin/issues/1370))\n- *(fish)* Use fish builtins for `enter_accept` ([#1373](https://github.com/atuinsh/atuin/issues/1373))\n- *(fish)* Accept multiline commands ([#1418](https://github.com/atuinsh/atuin/issues/1418))\n- *(nix)* Add Appkit to the package build ([#1358](https://github.com/atuinsh/atuin/issues/1358))\n- *(zsh)* Bind in the most popular modes ([#1360](https://github.com/atuinsh/atuin/issues/1360))\n- *(zsh)* Only trigger up-arrow on first line ([#1359](https://github.com/atuinsh/atuin/issues/1359))\n- Initial list of history in workspace mode ([#1356](https://github.com/atuinsh/atuin/issues/1356))\n- Make `atuin account delete` void session + key ([#1393](https://github.com/atuinsh/atuin/issues/1393))\n- New clippy lints ([#1395](https://github.com/atuinsh/atuin/issues/1395))\n- Reenable enter_accept for bash ([#1408](https://github.com/atuinsh/atuin/issues/1408))\n- Respect ZSH's $ZDOTDIR environment variable ([#942](https://github.com/atuinsh/atuin/issues/942))\n\n\n### Documentation\n\n- Update sync.md ([#1409](https://github.com/atuinsh/atuin/issues/1409))\n- Update Arch Linux package URL in advanced-install.md ([#1407](https://github.com/atuinsh/atuin/issues/1407))\n- New stats config ([#1412](https://github.com/atuinsh/atuin/issues/1412))\n\n\n### Features\n\n- *(nix)* Add a nixpkgs overlay ([#1357](https://github.com/atuinsh/atuin/issues/1357))\n- Add metrics server and http metrics ([#1394](https://github.com/atuinsh/atuin/issues/1394))\n- Add some metrics related to Atuin as an app ([#1399](https://github.com/atuinsh/atuin/issues/1399))\n- Allow configuring stats prefix ([#1411](https://github.com/atuinsh/atuin/issues/1411))\n- Allow spaces in stats prefixes ([#1414](https://github.com/atuinsh/atuin/issues/1414))\n\n\n### Miscellaneous Tasks\n\n- *(readme)* Add contributor image to README ([#1430](https://github.com/atuinsh/atuin/issues/1430))\n- Update to sqlx 0.7.3 ([#1416](https://github.com/atuinsh/atuin/issues/1416))\n- `cargo update` ([#1419](https://github.com/atuinsh/atuin/issues/1419))\n- Update rusty_paseto and rusty_paserk ([#1420](https://github.com/atuinsh/atuin/issues/1420))\n- Run dependabot weekly, not daily ([#1423](https://github.com/atuinsh/atuin/issues/1423))\n- Don't group deps ([#1424](https://github.com/atuinsh/atuin/issues/1424))\n- Setup git cliff ([#1431](https://github.com/atuinsh/atuin/issues/1431))\n\n\n## 17.0.1\n\n### Bug Fixes\n\n- *(bash)* Improve output of `enter_accept` ([#1342](https://github.com/atuinsh/atuin/issues/1342))\n- *(enter_accept)* Clear old cmd snippet ([#1350](https://github.com/atuinsh/atuin/issues/1350))\n- *(fish)* Improve output for `enter_accept` ([#1341](https://github.com/atuinsh/atuin/issues/1341))\n\n\n## 17.0.0\n\n### Bug Fixes\n\n- *(1220)* Workspace Filtermode not handled in skim engine ([#1273](https://github.com/atuinsh/atuin/issues/1273))\n- *(nu)* Disable the up-arrow keybinding for Nushell ([#1329](https://github.com/atuinsh/atuin/issues/1329))\n- *(nushell)* Ignore stderr messages ([#1320](https://github.com/atuinsh/atuin/issues/1320))\n- *(ubuntu/arm*)* Detect non amd64 ubuntu and handle ([#1131](https://github.com/atuinsh/atuin/issues/1131))\n\n\n### Documentation\n\n- Update `workspace` config key to `workspaces` ([#1174](https://github.com/atuinsh/atuin/issues/1174))\n- Document the available format options of History list command ([#1234](https://github.com/atuinsh/atuin/issues/1234))\n\n\n### Features\n\n- *(installer)* Try installing via paru for the AUR ([#1262](https://github.com/atuinsh/atuin/issues/1262))\n- *(keyup)* Configure SearchMode for KeyUp invocation #1216 ([#1224](https://github.com/atuinsh/atuin/issues/1224))\n- Mouse selection support ([#1209](https://github.com/atuinsh/atuin/issues/1209))\n- Copy to clipboard ([#1249](https://github.com/atuinsh/atuin/issues/1249))\n\n\n### Refactor\n\n- Duplications reduced in order to align implementations of reading history files ([#1247](https://github.com/atuinsh/atuin/issues/1247))\n\n\n### Config.md\n\n- Invert mode detailed options ([#1225](https://github.com/atuinsh/atuin/issues/1225))\n\n\n## 16.0.0\n\n### Bug Fixes\n\n- *(docs)* List all presently documented commands ([#1140](https://github.com/atuinsh/atuin/issues/1140))\n- *(docs)* Correct command overview paths ([#1145](https://github.com/atuinsh/atuin/issues/1145))\n- *(server)* Teapot is a cup of coffee ([#1137](https://github.com/atuinsh/atuin/issues/1137))\n- Adjust broken link to supported shells ([#1013](https://github.com/atuinsh/atuin/issues/1013))\n- Fixes unix specific impl of shutdown_signal ([#1061](https://github.com/atuinsh/atuin/issues/1061))\n- Nushell empty hooks ([#1138](https://github.com/atuinsh/atuin/issues/1138))\n\n\n### Features\n\n- Do not allow empty passwords durring account creation ([#1029](https://github.com/atuinsh/atuin/issues/1029))\n\n\n### Skim\n\n- Fix filtering aggregates ([#1114](https://github.com/atuinsh/atuin/issues/1114))\n\n\n## 15.0.0\n\n### Documentation\n\n- Fix broken links in README.md ([#920](https://github.com/atuinsh/atuin/issues/920))\n- Fix \"From source\" `cd` command ([#937](https://github.com/atuinsh/atuin/issues/937))\n\n\n### Features\n\n- Add delete account option (attempt 2) ([#980](https://github.com/atuinsh/atuin/issues/980))\n\n\n### Miscellaneous Tasks\n\n- Uuhhhhhh crypto lol ([#805](https://github.com/atuinsh/atuin/issues/805))\n- Fix participle \"be ran\" -> \"be run\" ([#939](https://github.com/atuinsh/atuin/issues/939))\n\n\n### Cwd_filter\n\n- Much like history_filter, only it applies to cwd ([#904](https://github.com/atuinsh/atuin/issues/904))\n\n\n## 14.0.0\n\n### Bug Fixes\n\n- *(client)* Always read session_path from settings ([#757](https://github.com/atuinsh/atuin/issues/757))\n- *(installer)* Use case-insensitive comparison ([#776](https://github.com/atuinsh/atuin/issues/776))\n- Many wins were broken :memo: ([#789](https://github.com/atuinsh/atuin/issues/789))\n- Paste into terminal after switching modes ([#793](https://github.com/atuinsh/atuin/issues/793))\n- Record negative exit codes ([#821](https://github.com/atuinsh/atuin/issues/821))\n- Allow nix package to fetch dependencies from git ([#832](https://github.com/atuinsh/atuin/issues/832))\n\n\n### Documentation\n\n- *(README)* Fix activity graph link ([#753](https://github.com/atuinsh/atuin/issues/753))\n\n\n### Features\n\n- Add common default keybindings ([#719](https://github.com/atuinsh/atuin/issues/719))\n- Add an inline view mode ([#648](https://github.com/atuinsh/atuin/issues/648))\n- Add *Nushell* support ([#788](https://github.com/atuinsh/atuin/issues/788))\n- Add github action to test the nix builds ([#833](https://github.com/atuinsh/atuin/issues/833))\n\n\n### Miscellaneous Tasks\n\n- Remove tui vendoring ([#804](https://github.com/atuinsh/atuin/issues/804))\n- Use fork of skim ([#803](https://github.com/atuinsh/atuin/issues/803))\n\n\n### Nix\n\n- Add flake-compat ([#743](https://github.com/atuinsh/atuin/issues/743))\n\n\n## 13.0.0\n\n### Documentation\n\n- *(README)* Add static activity graph example ([#680](https://github.com/atuinsh/atuin/issues/680))\n- Remove human short flag from docs, duplicate of help -h ([#663](https://github.com/atuinsh/atuin/issues/663))\n- Fix typo in zh-CN/README.md ([#666](https://github.com/atuinsh/atuin/issues/666))\n\n\n### Features\n\n- *(history)* Add new flag to allow custom output format ([#662](https://github.com/atuinsh/atuin/issues/662))\n\n\n### Fish\n\n- Fix `atuin init` for the fish shell ([#699](https://github.com/atuinsh/atuin/issues/699))\n\n\n### Install.sh\n\n- Fallback to using cargo ([#639](https://github.com/atuinsh/atuin/issues/639))\n\n\n## 12.0.0\n\n### Documentation\n\n- Add more details about date parsing in the stats command ([#579](https://github.com/atuinsh/atuin/issues/579))\n\n\n## 0.10.0\n\n### Miscellaneous Tasks\n\n- Allow specifiying the limited of returned entries ([#364](https://github.com/atuinsh/atuin/issues/364))\n\n\n## 0.9.0\n\n### README\n\n- Add MacPorts installation instructions ([#302](https://github.com/atuinsh/atuin/issues/302))\n\n\n## 0.8.1\n\n### Bug Fixes\n\n- Get install.sh working on UbuntuWSL ([#260](https://github.com/atuinsh/atuin/issues/260))\n\n\n## 0.8.0\n\n### Bug Fixes\n\n- Resolve some issues with install.sh ([#188](https://github.com/atuinsh/atuin/issues/188))\n\n\n### Features\n\n- Login/register no longer blocking ([#216](https://github.com/atuinsh/atuin/issues/216))\n\n\n## 0.7.2\n\n### Bug Fixes\n\n- Dockerfile with correct glibc ([#198](https://github.com/atuinsh/atuin/issues/198))\n\n\n### Features\n\n- Allow input of credentials from stdin ([#185](https://github.com/atuinsh/atuin/issues/185))\n\n\n### Miscellaneous Tasks\n\n- Some new linting ([#201](https://github.com/atuinsh/atuin/issues/201))\n- Supply pre-build docker image ([#199](https://github.com/atuinsh/atuin/issues/199))\n- Add more eyre contexts ([#200](https://github.com/atuinsh/atuin/issues/200))\n- Improve build times ([#213](https://github.com/atuinsh/atuin/issues/213))\n\n\n## 0.7.1\n\n### Features\n\n- Build individual crates ([#109](https://github.com/atuinsh/atuin/issues/109))\n\n\n## 0.6.3\n\n### Bug Fixes\n\n- Help text\n\n\n### Features\n\n- Use directories project data dir\n\n\n### Miscellaneous Tasks\n\n- Use structopt wrapper instead of building clap by hand\n\n\n\n" }, { "path": "CODE_OF_CONDUCT.md", "content": "# Contributor Covenant Code of Conduct\n\n## Our Pledge\n\nWe as members, contributors, and leaders pledge to make participation in our\ncommunity a harassment-free experience for everyone, regardless of age, body\nsize, visible or invisible disability, ethnicity, sex characteristics, gender\nidentity and expression, level of experience, education, socio-economic status,\nnationality, personal appearance, race, religion, or sexual identity\nand orientation.\n\nWe pledge to act and interact in ways that contribute to an open, welcoming,\ndiverse, inclusive, and healthy community.\n\n## Our Standards\n\nExamples of behavior that contributes to a positive environment for our\ncommunity include:\n\n* Demonstrating empathy and kindness toward other people\n* Being respectful of differing opinions, viewpoints, and experiences\n* Giving and gracefully accepting constructive feedback\n* Accepting responsibility and apologizing to those affected by our mistakes,\n and learning from the experience\n* Focusing on what is best not just for us as individuals, but for the\n overall community\n\nExamples of unacceptable behavior include:\n\n* The use of sexualized language or imagery, and sexual attention or\n advances of any kind\n* Trolling, insulting or derogatory comments, and personal or political attacks\n* Public or private harassment\n* Publishing others' private information, such as a physical or email\n address, without their explicit permission\n* Other conduct which could reasonably be considered inappropriate in a\n professional setting\n\n## Enforcement Responsibilities\n\nCommunity leaders are responsible for clarifying and enforcing our standards of\nacceptable behavior and will take appropriate and fair corrective action in\nresponse to any behavior that they deem inappropriate, threatening, offensive,\nor harmful.\n\nCommunity leaders have the right and responsibility to remove, edit, or reject\ncomments, commits, code, wiki edits, issues, and other contributions that are\nnot aligned to this Code of Conduct, and will communicate reasons for moderation\ndecisions when appropriate.\n\n## Scope\n\nThis Code of Conduct applies within all community spaces, and also applies when\nan individual is officially representing the community in public spaces.\nExamples of representing our community include using an official e-mail address,\nposting via an official social media account, or acting as an appointed\nrepresentative at an online or offline event.\n\n## Enforcement\n\nInstances of abusive, harassing, or otherwise unacceptable behavior may be\nreported to the community leaders responsible for enforcement at\nellie@elliehuxtable.com.\nAll complaints will be reviewed and investigated promptly and fairly.\n\nAll community leaders are obligated to respect the privacy and security of the\nreporter of any incident.\n\n## Enforcement Guidelines\n\nCommunity leaders will follow these Community Impact Guidelines in determining\nthe consequences for any action they deem in violation of this Code of Conduct:\n\n### 1. Correction\n\n**Community Impact**: Use of inappropriate language or other behavior deemed\nunprofessional or unwelcome in the community.\n\n**Consequence**: A private, written warning from community leaders, providing\nclarity around the nature of the violation and an explanation of why the\nbehavior was inappropriate. A public apology may be requested.\n\n### 2. Warning\n\n**Community Impact**: A violation through a single incident or series\nof actions.\n\n**Consequence**: A warning with consequences for continued behavior. No\ninteraction with the people involved, including unsolicited interaction with\nthose enforcing the Code of Conduct, for a specified period of time. This\nincludes avoiding interactions in community spaces as well as external channels\nlike social media. Violating these terms may lead to a temporary or\npermanent ban.\n\n### 3. Temporary Ban\n\n**Community Impact**: A serious violation of community standards, including\nsustained inappropriate behavior.\n\n**Consequence**: A temporary ban from any sort of interaction or public\ncommunication with the community for a specified period of time. No public or\nprivate interaction with the people involved, including unsolicited interaction\nwith those enforcing the Code of Conduct, is allowed during this period.\nViolating these terms may lead to a permanent ban.\n\n### 4. Permanent Ban\n\n**Community Impact**: Demonstrating a pattern of violation of community\nstandards, including sustained inappropriate behavior, harassment of an\nindividual, or aggression toward or disparagement of classes of individuals.\n\n**Consequence**: A permanent ban from any sort of public interaction within\nthe community.\n\n## Attribution\n\nThis Code of Conduct is adapted from the [Contributor Covenant][homepage],\nversion 2.0, available at\nhttps://www.contributor-covenant.org/version/2/0/code_of_conduct.html.\n\nCommunity Impact Guidelines were inspired by [Mozilla's code of conduct\nenforcement ladder](https://github.com/mozilla/diversity).\n\n[homepage]: https://www.contributor-covenant.org\n\nFor answers to common questions about this code of conduct, see the FAQ at\nhttps://www.contributor-covenant.org/faq. Translations are available at\nhttps://www.contributor-covenant.org/translations.\n" }, { "path": "CONTRIBUTING.md", "content": "# Contributing\n\nThank you so much for considering contributing to Atuin! We really appreciate it <3\n\nDevelopment dependencies\n\n1. A rust toolchain ([rustup](https://rustup.rs) recommended)\n\nWe commit to supporting the latest stable version of Rust - nothing more, nothing less, no nightly.\n\nBefore working on anything, we suggest taking a copy of your Atuin data directory (`~/.local/share/atuin` on most \\*nix platforms). If anything goes wrong, you can always restore it!\n\nWhile data directory backups are always a good idea, you can instruct Atuin to use custom path using the following environment variables:\n\n```shell\nexport ATUIN_RECORD_STORE_PATH=/tmp/atuin_records.db\nexport ATUIN_DB_PATH=/tmp/atuin_dev.db\nexport ATUIN_KV__DB_PATH=/tmp/atuin_kv.db\nexport ATUIN_SCRIPTS__DB_PATH=/tmp/atuin_scripts.db\n```\n\nIt is also recommended to update your `$PATH` so that the pre-exec scripts would use the locally built version:\n\n```shell\nexport PATH=\"./target/release:$PATH\"\n```\n\nIf you'd like to load a different configuration file, set `ATUIN_CONFIG_DIR` to a folder that contains your `config.toml` file:\n\n```shell\nexport ATUIN_CONFIG_DIR=/tmp/atuin-config/\n```\n\nThese variable exports can be added in a local `.envrc` file, read by [direnv](https://direnv.net/).\n\n## PRs\n\nIt can speed up the review cycle if you consent to maintainers pushing to your branch. This will only be in the case of small fixes or adjustments, and not anything large. If you feel OK with this, please check the box on the template!\n\n## What to work on?\n\nAny issues labeled \"bug\" or \"help wanted\" would be fantastic, just drop a comment and feel free to ask for help!\n\nIf there's anything you want to work on that isn't already an issue, either open a feature request or get in touch on the [forum](https://forum.atuin.sh)/Discord.\n\n## Setup\n\n```\ngit clone https://github.com/atuinsh/atuin\ncd atuin\ncargo build\n```\n\n## Running\n\nWhen iterating on a feature, it's useful to use `cargo run`\n\nFor example, if working on a search feature\n\n```\ncargo run -- search --a-new-flag\n```\n\nWhile iterating on the server, I find it helpful to run a new user on my system, with `sync_server` set to be `localhost`.\n\n## Tests\n\nOur test coverage is currently not the best, but we are working on it! Generally tests live in the file next to the functionality they are testing, and are executed just with `cargo test`.\n\n## Logging and Debugging\n\n### Log Files\n\nAtuin writes logs to `~/.atuin/logs` unless configured otherwise. Log files are rotated daily and retained for 4 days by default:\n\n- `search.log.*` - Interactive search session logs\n- `daemon.log.*` - Background daemon logs\n\n### Log Levels\n\nYou can set the `ATUIN_LOG` environment variable to override log verbosity from the config file:\n\n```shell\nATUIN_LOG=debug atuin search # Enable debug logging\nATUIN_LOG=trace atuin search # Enable trace logging (very verbose)\n```\n\n### Span Timing (Performance Profiling)\n\nFor performance analysis, you can capture detailed span timing data as JSON:\n\n```shell\nATUIN_SPAN=spans.json atuin search\n```\n\nThis creates a JSON file with timing information for each instrumented span, including:\n- `time.busy` - Time actively executing code\n- `time.idle` - Time awaiting async operations (I/O, child tasks)\n\nThe `scripts/span-table.ts` script analyzes these logs:\n\n```shell\n# Summary view - shows all spans with timing stats\nbun scripts/span-table.ts spans.json\n\n# Detail view - shows individual calls for a specific span\nbun scripts/span-table.ts spans.json --detail daemon_search\n\n# Filter to specific spans\nbun scripts/span-table.ts spans.json --filter \"search|hydrate\"\n```\n\nThis is useful for comparing performance between different search implementations or identifying bottlenecks.\n\n## Migrations\n\nBe careful creating database migrations - once your database has migrated ahead of current stable, there is no going back\n\n### Stickers\n\nWe try to ship anyone contributing to Atuin a sticker! Only contributors get a shiny one. Fill out [this form](https://noteforms.com/forms/contributors-stickers) if you'd like one.\n" }, { "path": "CONTRIBUTORS", "content": "0x4A6F <0x4A6F@users.noreply.github.com>\nAleks Bunin \nAlex Hamilton <1622250+Aehmlo@users.noreply.github.com>\nAlexandre GV. \nAloxaf \nAlpha Chen \nAmos Bird \nAnderson <141751473+digital-cuttlefish@users.noreply.github.com>\nAndrew Aylett \nAndrew Lee <32912555+candrewlee14@users.noreply.github.com>\nAnish Pallati \nAustin Schey \navinassh <640792+avinassh@users.noreply.github.com>\nAzzam S.A <17734314+azzamsa@users.noreply.github.com>\nb3nj5m1n <47924309+b3nj5m1n@users.noreply.github.com>\nBaptiste <32563450+BapRx@users.noreply.github.com>\nBen J \nBenjamin Vergnaud <9599845+bvergnaud@users.noreply.github.com>\nBenjamin Weinstein-Raun \nBlair Noctis <4474501+nc7s@users.noreply.github.com>\nBrad Robel-Forrest \nBraelyn Boynton \nBrian Kung <2836167+briankung@users.noreply.github.com>\nBruce Huang \nc-14 \nCaleb Maclennan \nCh. (Chanwhi Choi) \nChandra Kiran G \nchitao1234 <1139954766@qq.com>\nChris Rose \nConrad Ludgate \nCosmicHorror \nCristian Le \nCristian Le \nCULT PONY <67918945+cultpony@users.noreply.github.com>\ncyqsimon <28627918+cyqsimon@users.noreply.github.com>\nDagan McGregor \nDaniel \nDaniel Carosone \nDaniPopes <57450786+DaniPopes@users.noreply.github.com>\nDavid \nDavid \nDavid Chocholatý \nDavid Jack Wange Olrik \nDavid Legrand <1110600+davlgd@users.noreply.github.com>\nDennis Trautwein \ndependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>\nDevin Buhl \nDhruv Thakur <13575379+dhth@users.noreply.github.com>\nDiego Carrasco Gubernatis <557703+dacog@users.noreply.github.com>\nDieter Eickstaedt \nDom Rodriguez \nDongxu Wang \nDS/Charlie <82801887+ds-cbo@users.noreply.github.com>\nEd Ive \nEdward Loveall \nEllie Huxtable \nEmanuele Panzeri \nEric Crosson \nEric Hodel \nEric Long \nEric Ripa \nErwin Kroon <123574+ekroon@users.noreply.github.com>\neth3lbert \nEthan Brierley \nEvan McBeth <64177332+AtomicRobotMan0101@users.noreply.github.com>\nEvan Purkhiser \nFarid Zakaria \nFelix Yan \nFrank Hamand \nfrukto \nGokul \nHamza Hamud <53880692+hhamud@users.noreply.github.com>\nHelmut K. C. Tessarek \nHerby Gillot \nHesam Pakdaman <14890379+hesampakdaman@users.noreply.github.com>\nHilmar Wiegand \nHunter Casten <41604962+enchantednatures@users.noreply.github.com>\nIan Manske \nIan Smith \nIan Smith \nIlkin Bayramli <43158991+ibayramli@users.noreply.github.com>\nIvan Toriya <43750521+ivan-toriya@users.noreply.github.com>\nJ. Emiliano Deustua \nJakob Schrettenbrunner \nJakub Jirutka \nJakub Panek \nJames Trew <66286082+jamestrew@users.noreply.github.com>\nJamie Quigley \nJan Larres \nJannik <32144358+mozzieongit@users.noreply.github.com>\nJannik \nJax Young \njean-santos \njean-santos \nJeff Gould \nJeremy Cline \nJeremy Cline \nJerome Ducret \njfmontanaro \nJinn Koriech \nJinna Kiisuo \nJoe Ardent \nJohannes Baiter \nJosef Friedrich \nJT <547158+jntrnr@users.noreply.github.com>\nJulien P \nJustin Su \nJános Illés \nKian-Meng Ang \nKjetil Jørgensen \nKlas Mellbourn \nKoichi Murase \nKorvin Szanto \nKrithic Kumar <30691152+notjedi@users.noreply.github.com>\nKrut Patel \nLaurent le Beau-Martin <1180863+laurentlbm@users.noreply.github.com>\nlchausmann \nLeoniePhiline <22329650+LeoniePhiline@users.noreply.github.com>\nLuca Comellini \nLucas Burns <44355502+lmburns@users.noreply.github.com>\nLucas Trzesniewski \nLucy \nLuke Baker \nLuke Karrys \nMag Mell \nManel Vilar \nMarcin Puc \nMarijan Smetko \nMark Wotton \nMartin Indra \nMartin Junghanns \nMat Jones \nMatheus Martins \nMatt Godbolt \nMatthew Berryman \nMatthias Beyer \nMatthieu LAURENT \nMattias Eriksson \nMaurice Escher \nMaxim Burgerhout \nMaxim Uvarov \nmb6ockatf <104227451+mb6ockatf@users.noreply.github.com>\nmentalisttraceur \nMichael Bianco \nMichael Mior \nMichael Vincent <377567+Vynce@users.noreply.github.com>\nMichele Azzolari \nMichelle Tilley \nMike Pastore \nMike Tsao \nmmx \nmorguldir \nmundry <1453314+mundry@users.noreply.github.com>\nNelyah \nNemo157 \nnetworkException \nNico Kokonas \nNiklas Hambüchen \nnoyez \nOmer Katz \nonkelT2 <126604057+onkelT2@users.noreply.github.com>\nOnè <43485962+c-git@users.noreply.github.com>\nOrhun Parmaksız \nP T Weir \nPatrick \nPatrick Decat \nPatrick Jackson \nPavel Ivanov \nPer Modin \nPeter Brunner \nPeter Holloway \nPhilippe Normand \nPhilippe Normand \nPierluigi <82404704+IoSonoPiero@users.noreply.github.com>\nPlamen Dimitrov \nPoliorcetics \npostmath \nprintfn <1643883+printfn@users.noreply.github.com>\nQiming Xu <33349132+xqm32@users.noreply.github.com>\nRain \nRamses \nRemmy Cat Stock <3317423+remmycat@users.noreply.github.com>\nRemo Senekowitsch \nReverier Xu \nRichard de Boer \nRichard Jones <4550158+RichardDRJ@users.noreply.github.com>\nRichard Turner <63139+zygous@users.noreply.github.com>\nRobin Millette \nrriski \nSam Edwards \nSam Lanning \nSamson \nSandro \nSatyarth Sampath \nsdr135284 <54752759+sdr135284@users.noreply.github.com>\nShroomy \nSimon \nSimon Elsbrock \nslamp \nSteve Kemp \nSteven Xu \nSven-Hendrik Haase \nThomas Buckley-Houston \nTobias Genannt \nTobias Genannt \nTobias Hunger \nTom Cammann \nTom Cammann \nTrygve Aaberge \nTymanWasTaken \nUbiquitous Photon <39134173+UbiquitousPhoton@users.noreply.github.com>\nViolet Shreve \nVlad Stepanov <8uk.8ak@gmail.com>\nVladislav Stepanov <8uk.8ak@gmail.com>\nVuiMuich \nWebmaster At Cosmic DNA <92752640+DanielAtCosmicDNA@users.noreply.github.com>\nWill Fancher \nWind \nWindSoilder \nwinston \nwpbrz <61665187+wpbrz@users.noreply.github.com>\nXavier Vello \nxfzv <78810647+xfzv@users.noreply.github.com>\nYannick Ulrich \nYaroslav Halchenko \nYolo \nYonatan Goldschmidt \nYummyOreo \nYuvi Panda \nZhanibek Adilbekov \nZhiHong Li \nZhizhen He \néclairevoyant <848000+eclairevoyant@users.noreply.github.com>\n依云 \n镜面王子 <153555712@qq.com>\n" }, { "path": "Cargo.toml", "content": "[workspace]\nmembers = [\"crates/*\", \"crates/atuin-nucleo/matcher\", \"crates/atuin-nucleo/bench\"]\n\nresolver = \"2\"\nexclude = [\"ui/backend\", \"crates/atuin-nucleo/matcher/fuzz\"]\n\n[workspace.package]\nversion = \"18.13.3\"\nauthors = [\"Ellie Huxtable \"]\nrust-version = \"1.94.0\"\nlicense = \"MIT\"\nhomepage = \"https://atuin.sh\"\nrepository = \"https://github.com/atuinsh/atuin\"\nreadme = \"README.md\"\n\n[workspace.dependencies]\nasync-trait = \"0.1.58\"\natuin-client = { path = \"crates/atuin-client\", version = \"18.13.3\" }\natuin-common = { path = \"crates/atuin-common\", version = \"18.13.3\" }\natuin-daemon = { path = \"crates/atuin-daemon\", version = \"18.13.3\" }\natuin-dotfiles = { path = \"crates/atuin-dotfiles\", version = \"18.13.3\" }\natuin-history = { path = \"crates/atuin-history\", version = \"18.13.3\" }\natuin-kv = { path = \"crates/atuin-kv\", version = \"18.13.3\" }\natuin-scripts = { path = \"crates/atuin-scripts\", version = \"18.13.3\" }\natuin-server = { path = \"crates/atuin-server\", version = \"18.13.3\" }\natuin-server-database = { path = \"crates/atuin-server-database\", version = \"18.13.3\" }\natuin-server-postgres = { path = \"crates/atuin-server-postgres\", version = \"18.13.3\" }\natuin-server-sqlite = { path = \"crates/atuin-server-sqlite\", version = \"18.13.3\" }\natuin-nucleo = { path = \"crates/atuin-nucleo\", version = \"0.6.0\" }\natuin-nucleo-matcher = { path = \"crates/atuin-nucleo/matcher\", version = \"0.3.1\" }\nbase64 = \"0.22\"\ncrossterm = \"0.29.0\"\nlog = \"0.4\"\ntime = { version = \"0.3.47\", features = [\n \"serde-human-readable\",\n \"macros\",\n \"local-offset\",\n] }\nclap = { version = \"4.5.7\", features = [\"derive\"] }\nconfig = { version = \"0.15.8\", default-features = false, features = [\"toml\"] }\ndirectories = \"6.0.0\"\neyre = \"0.6\"\nfs-err = \"3.1\"\ninterim = { version = \"0.2.0\", features = [\"time_0_3\"] }\nitertools = \"0.14.0\"\nrand = { version = \"0.8.5\", features = [\"std\"] }\nsemver = \"1.0.20\"\nserde = { version = \"1.0.202\", features = [\"derive\"] }\nserde_json = \"1.0.119\"\ntokio = { version = \"1\", features = [\"full\"] }\nuuid = { version = \"1.9\", features = [\"v4\", \"v7\", \"serde\"] }\nwhoami = \"2.1.0\"\ntyped-builder = \"0.18.2\"\npretty_assertions = \"1.3.0\"\nthiserror = \"2\"\nrustix = { version = \"1.1.4\", features = [\"process\", \"fs\"] }\ntower = \"0.5\"\ntracing = \"0.1\"\nratatui = \"0.30.0\"\nsql-builder = \"3\"\ntempfile = { version = \"3.19\" }\nminijinja = \"2.9.0\"\nrustls = { version = \"0.23\", default-features = false, features = [\n \"ring\",\n \"std\",\n \"tls12\",\n] }\n\n[workspace.dependencies.tracing-subscriber]\nversion = \"0.3\"\nfeatures = [\"ansi\", \"fmt\", \"registry\", \"env-filter\", \"json\"]\n\n[workspace.dependencies.reqwest]\nversion = \"0.13\"\nfeatures = [\"json\", \"rustls-no-provider\", \"stream\"]\ndefault-features = false\n\n[workspace.dependencies.sqlx]\nversion = \"0.8\"\nfeatures = [\"runtime-tokio-rustls\", \"time\", \"postgres\", \"uuid\"]\n\n# The profile that 'cargo dist' will build with\n[profile.dist]\ninherits = \"release\"\nlto = \"thin\"\nstrip = \"symbols\"\n" }, { "path": "Dockerfile", "content": "FROM lukemathwalker/cargo-chef:latest-rust-1.94.0-slim-bookworm AS chef\nWORKDIR app\n\nFROM chef AS planner\nCOPY . .\nRUN cargo chef prepare --recipe-path recipe.json\n\nFROM chef AS builder\n\n# Ensure working C compile setup (not installed by default in arm64 images)\nRUN apt update && apt install build-essential -y\n\nCOPY --from=planner /app/recipe.json recipe.json\nRUN cargo chef cook --release --recipe-path recipe.json\n\nCOPY . .\nRUN cargo build --release --bin atuin-server\n\nFROM debian:bookworm-20260202-slim AS runtime\n\nRUN useradd -c 'atuin user' atuin && mkdir /config && chown atuin:atuin /config\n# Install ca-certificates for webhooks to work\nRUN apt update && apt install ca-certificates -y && rm -rf /var/lib/apt/lists/*\nWORKDIR app\n\nUSER atuin\n\nENV TZ=Etc/UTC\nENV RUST_LOG=atuin_server=info\nENV ATUIN_CONFIG_DIR=/config\n\nCOPY --from=builder /app/target/release/atuin-server /usr/local/bin\nENTRYPOINT [\"/usr/local/bin/atuin-server\"]\n" }, { "path": "LICENSE", "content": "MIT License\n\nCopyright (c) 2021 Ellie Huxtable\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n" }, { "path": "README.md", "content": "

\n \n \n \"Text\n\n

\n\n

\nmagical shell history\n

\n\n
\n\n

\n \n \n \n \n \n \n \n

\n\n[English] | [简体中文]\n\nAtuin replaces your existing shell history with a SQLite database, and records\nadditional context for your commands. Additionally, it provides optional and\n_fully encrypted_ synchronisation of your history between machines, via an Atuin\nserver.\n\n

\n \"animated\"\n

\n\n

\nexit code, duration, time and command shown\n

\n\nAs well as the search UI, it can do things like this:\n\n```\n# search for all successful `make` commands, recorded after 3pm yesterday\natuin search --exit 0 --after \"yesterday 3pm\" make\n```\n\nYou may use either the server I host, or host your own! Or just don't use sync\nat all. As all history sync is encrypted, I couldn't access your data even if\nI wanted to. And I **really** don't want to.\n\n## Features\n\n- rebind `ctrl-r` and `up` (configurable) to a full screen history search UI\n- store shell history in a sqlite database\n- back up and sync **encrypted** shell history\n- the same history across terminals, across sessions, and across machines\n- log exit code, cwd, hostname, session, command duration, etc\n- calculate statistics such as \"most used command\"\n- old history file is not replaced\n- quick-jump to previous items with Alt-\\\n- switch filter modes via ctrl-r; search history just from the current session, directory, or globally\n- enter to execute a command, tab to edit\n\n## Documentation\n\n- [Quickstart](#quickstart)\n- [Install](https://docs.atuin.sh/guide/installation/)\n- [Setting up sync](https://docs.atuin.sh/guide/sync/)\n- [Import history](https://docs.atuin.sh/guide/import/)\n- [Basic usage](https://docs.atuin.sh/guide/basic-usage/)\n\n## Supported Shells\n\n- zsh\n- bash\n- fish\n- nushell\n- xonsh\n- powershell (tier 2 support)\n\n## Community\n\n### Forum\n\nAtuin has a community forum, please ask here for help and support: \n\n### IRC\n\nWe're also available via #atuin on libera.chat\n\n### Discord\n\nAtuin also has a community Discord, available [here](https://discord.gg/jR3tfchVvW)\n\n# Quickstart\n\nThis will sign you up for the Atuin Cloud sync server. Everything is end-to-end encrypted, so your secrets are safe!\n\nRead more in the [docs](https://docs.atuin.sh) for an offline setup, self hosted server, and more.\n\n```\ncurl --proto '=https' --tlsv1.2 -LsSf https://setup.atuin.sh | sh\n\natuin register -u -e \natuin import auto\natuin sync\n```\n\nThen restart your shell!\n\n> [!NOTE]\n>\n> **For Bash users**: The above sets up `bash-preexec` for necessary hooks, but\n> `bash-preexec` has limitations. For details, please see the\n> [Bash](https://docs.atuin.sh/guide/installation/#installing-the-shell-plugin)\n> section of the shell plugin documentation.\n\n# Security\n\nIf you find any security issues, we'd appreciate it if you could alert \n\n# Contributors\n\n\n \n\n\nMade with [contrib.rocks](https://contrib.rocks).\n\n[English]: ./README.md\n[简体中文]: ./docs-i18n/zh-CN/README.md\n" }, { "path": "atuin.nix", "content": "# Atuin package definition\n#\n# This file will be similar to the package definition in nixpkgs:\n# https://github.com/NixOS/nixpkgs/blob/master/pkgs/by-name/at/atuin/package.nix\n#\n# Helpful documentation: https://github.com/NixOS/nixpkgs/blob/master/doc/languages-frameworks/rust.section.md\n{\n lib,\n stdenv,\n installShellFiles,\n rustPlatform,\n libiconv,\n}:\nrustPlatform.buildRustPackage {\n name = \"atuin\";\n\n src = lib.cleanSource ./.;\n\n cargoLock = {\n lockFile = ./Cargo.lock;\n # Allow dependencies to be fetched from git and avoid having to set the outputHashes manually\n allowBuiltinFetchGit = true;\n };\n\n nativeBuildInputs = [installShellFiles];\n\n buildInputs = lib.optionals stdenv.isDarwin [libiconv];\n\n postInstall = ''\n installShellCompletion --cmd atuin \\\n --bash <($out/bin/atuin gen-completions -s bash) \\\n --fish <($out/bin/atuin gen-completions -s fish) \\\n --zsh <($out/bin/atuin gen-completions -s zsh)\n '';\n\n doCheck = false;\n\n meta = with lib; {\n description = \"Replacement for a shell history which records additional commands context with optional encrypted synchronization between machines\";\n homepage = \"https://github.com/atuinsh/atuin\";\n license = licenses.mit;\n mainProgram = \"atuin\";\n };\n}\n" }, { "path": "atuin.plugin.zsh", "content": "# shellcheck disable=2148,SC2168,SC1090,SC2125\nlocal FOUND_ATUIN=$+commands[atuin]\n\nif [[ $FOUND_ATUIN -eq 1 ]]; then\n source <(atuin init zsh)\nfi\n" }, { "path": "cliff.toml", "content": "# git-cliff ~ default configuration file\n# https://git-cliff.org/docs/configuration\n#\n# Lines starting with \"#\" are comments.\n# Configuration options are organized into tables and keys.\n# See documentation for more information on available options.\n\n[changelog]\n# changelog header\nheader = \"\"\"\n# Changelog\\n\nAll notable changes to this project will be documented in this file.\\n\n\"\"\"\n# template for the changelog body\n# https://keats.github.io/tera/docs/#introduction\nbody = \"\"\"\n{% if version %}\\\n ## {{ version | trim_start_matches(pat=\"v\") }}\n{% else %}\\\n ## [unreleased]\n{% endif %}\\\n{% for group, commits in commits | group_by(attribute=\"group\") %}\n ### {{ group | upper_first }}\n {% for commit in commits\n | filter(attribute=\"scope\")\n | sort(attribute=\"scope\") %}\n - *({{commit.scope}})* {{ commit.message | upper_first }}\n {%- if commit.breaking %}\n {% raw %} {% endraw %}- **BREAKING**: {{commit.breaking_description}}\n {%- endif -%}\n {%- endfor -%}\n {% raw %}\\n{% endraw %}\\\n {%- for commit in commits %}\n {%- if commit.scope -%}\n {% else -%}\n - {{ commit.message | upper_first }}\n {% if commit.breaking -%}\n {% raw %} {% endraw %}- **BREAKING**: {{commit.breaking_description}}\n {% endif -%}\n {% endif -%}\n {% endfor -%}\n {% raw %}\\n{% endraw %}\\\n{% endfor %}\\n\n\"\"\"\n\n# remove the leading and trailing whitespace from the template\ntrim = true\n# changelog footer\nfooter = \"\"\"\n\n\"\"\"\n# postprocessors\npostprocessors = [\n { pattern = '', replace = \"https://github.com/atuinsh/atuin\" }, # replace repository URL\n]\n[git]\n# parse the commits based on https://www.conventionalcommits.org\nconventional_commits = true\n# filter out the commits that are not conventional\nfilter_unconventional = true\n# process each line of a commit as an individual commit\nsplit_commits = false\n# regex for preprocessing the commit messages\ncommit_preprocessors = [\n { pattern = '\\((\\w+\\s)?#([0-9]+)\\)', replace = \"([#${2}](/issues/${2}))\" }, # replace issue numbers\n]\n# regex for parsing and grouping commits\ncommit_parsers = [\n { message = \"^feat\", group = \"Features\" },\n { message = \"^fix\", group = \"Bug Fixes\" },\n { message = \"^doc\", group = \"Documentation\" },\n { message = \"^perf\", group = \"Performance\" },\n { message = \"^refactor\", group = \"Refactor\" },\n { message = \"^style\", group = \"Styling\" },\n { message = \"^test\", group = \"Testing\" },\n { message = \"^chore\\\\(release\\\\): prepare for\", skip = true },\n { message = \"^chore\\\\(deps\\\\)\", skip = true },\n { message = \"^chore\\\\(pr\\\\)\", skip = true },\n { message = \"^chore\\\\(pull\\\\)\", skip = true },\n { message = \"^chore|ci\", group = \"Miscellaneous Tasks\" },\n { body = \".*security\", group = \"Security\" },\n { message = \"^revert\", group = \"Revert\" },\n]\n# protect breaking changes from being skipped due to matching a skipping commit_parser\nprotect_breaking_commits = false\n# filter out the commits that are not matched by commit parsers\nfilter_commits = false\n# regex for matching git tags\ntag_pattern = \"v[0-9].*\"\n\n# regex for skipping tags\nskip_tags = \"v0.1.0-beta.1\"\n# regex for ignoring tags\nignore_tags = \"prerelease|beta|alpha\"\n# sort the tags topologically\ntopo_order = false\n# sort the commits inside sections by oldest/newest order\nsort_commits = \"oldest\"\n# limit the number of commits included in the changelog.\n# limit_commits = 42\n" }, { "path": "crates/atuin/Cargo.toml", "content": "[package]\nname = \"atuin\"\nedition = \"2024\"\ndescription = \"atuin - magical shell history\"\nreadme = \"./README.md\"\n\nrust-version = { workspace = true }\nversion = { workspace = true }\nauthors = { workspace = true }\nlicense = { workspace = true }\nhomepage = { workspace = true }\nrepository = { workspace = true }\n\n[package.metadata.binstall]\npkg-url = \"{ repo }/releases/download/v{ version }/{ name }-{ target }.tar.gz\"\nbin-dir = \"{ name }-{ target }/{ bin }{ binary-ext }\"\npkg-fmt = \"tgz\"\n\n[package.metadata.deb]\nmaintainer = \"Ellie Huxtable \"\ncopyright = \"2021, Ellie Huxtable \"\nlicense-file = [\"LICENSE\"]\ndepends = \"$auto\"\nsection = \"utility\"\n\n[package.metadata.rpm]\npackage = \"atuin\"\n\n[package.metadata.rpm.cargo]\nbuildflags = [\"--release\"]\n\n[package.metadata.rpm.targets]\natuin = { path = \"/usr/bin/atuin\" }\n\n[features]\ndefault = [\"client\", \"sync\", \"clipboard\", \"check-update\", \"daemon\", \"ai\", \"hex\"]\nclient = [\"atuin-client\"]\nsync = [\"atuin-client/sync\"]\ndaemon = [\"atuin-client/daemon\", \"atuin-daemon\"]\nai = [\"atuin-ai\"]\nhex = [\"atuin-hex\"]\nclipboard = [\"arboard\"]\ncheck-update = [\"atuin-client/check-update\"]\n\n[dependencies]\natuin-ai = { path = \"../atuin-ai\", version = \"18.13.3\", optional = true, default-features = false }\natuin-client = { path = \"../atuin-client\", version = \"18.13.3\", optional = true, default-features = false }\natuin-common = { workspace = true }\natuin-dotfiles = { workspace = true }\natuin-history = { workspace = true }\natuin-daemon = { path = \"../atuin-daemon\", version = \"18.13.3\", optional = true, default-features = false }\natuin-hex = { path = \"../atuin-hex\", version = \"18.13.3\", optional = true, default-features = false }\natuin-scripts = { workspace = true }\natuin-kv = { workspace = true }\n\nlog = { workspace = true }\ntime = { workspace = true }\neyre = { workspace = true }\nindicatif = \"0.18.0\"\nserde = { workspace = true }\nserde_json = { workspace = true }\ncrossterm = { workspace = true, features = [\"use-dev-tty\"] }\nunicode-width = \"0.2\"\nitertools = { workspace = true }\ntokio = { workspace = true }\nasync-trait = { workspace = true }\ninterim = { workspace = true }\nclap = { workspace = true }\nclap_complete = \"4.5.8\"\nclap_complete_nushell = \"4.5.4\"\nfs-err = { workspace = true }\nfs4 = \"0.13.1\"\nrpassword = \"7.0\"\nsemver = { workspace = true }\nrustix = { workspace = true }\nruntime-format = \"0.1.3\"\ntiny-bip39 = \"2\"\nfutures-util = \"0.3\"\nfuzzy-matcher = \"0.3.7\"\ncolored = \"2.0.4\"\nopen = \"5\"\nratatui = { workspace = true }\ntracing = \"0.1\"\ntracing-subscriber = { workspace = true }\ntracing-appender = \"0.2\"\nuuid = { workspace = true }\nsysinfo = \"0.30.7\"\nregex = \"1.10.5\"\nnorm = { version = \"0.1.1\", features = [\"fzf-v2\"] }\natuin-nucleo-matcher = { workspace = true }\ntempfile = { workspace = true }\nshlex = \"1.3.0\"\n\n# settings editor with comment and relative ordering preservation\ntoml_edit = \"0.25.4\"\n\n[target.'cfg(any(target_os = \"windows\", target_os = \"macos\"))'.dependencies]\narboard = { version = \"3.4\", optional = true }\n\n[target.'cfg(target_os = \"linux\")'.dependencies]\narboard = { version = \"3.4\", optional = true, features = [\n \"wayland-data-control\",\n] }\n\n[target.'cfg(unix)'.dependencies]\ndaemonize = \"0.5.0\"\n\n[dev-dependencies]\ntracing-tree = \"0.4\"\n\n# Integration tests in tests/ spin up a test server to verify sync functionality.\n# TODO: Consider moving these tests to atuin-server crate instead (client would become a dev dep there)\natuin-server = { workspace = true }\natuin-server-database = { workspace = true }\natuin-server-postgres = { workspace = true }\n" }, { "path": "crates/atuin/LICENSE", "content": "MIT License\n\nCopyright (c) 2021 Ellie Huxtable\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n" }, { "path": "crates/atuin/build.rs", "content": "use std::process::Command;\nfn main() {\n let output = Command::new(\"git\").args([\"rev-parse\", \"HEAD\"]).output();\n\n let sha = match output {\n Ok(sha) => String::from_utf8(sha.stdout).unwrap(),\n Err(_) => String::from(\"NO_GIT\"),\n };\n\n println!(\"cargo:rustc-env=GIT_HASH={sha}\");\n}\n" }, { "path": "crates/atuin/src/command/client/account/change_password.rs", "content": "use clap::Parser;\nuse eyre::{Result, bail};\n\nuse atuin_client::{\n auth::{self, MutateResponse},\n settings::Settings,\n};\nuse rpassword::prompt_password;\n\n#[derive(Parser, Debug)]\npub struct Cmd {\n #[clap(long, short)]\n pub current_password: Option,\n\n #[clap(long, short)]\n pub new_password: Option,\n\n /// The two-factor authentication code for your account, if any\n #[clap(long, short)]\n pub totp_code: Option,\n}\n\nimpl Cmd {\n pub async fn run(&self, settings: &Settings) -> Result<()> {\n if !settings.logged_in().await? {\n bail!(\"You are not logged in\");\n }\n\n let client = auth::auth_client(settings).await;\n\n let current_password = self.current_password.clone().unwrap_or_else(|| {\n prompt_password(\"Please enter the current password: \")\n .expect(\"Failed to read from input\")\n });\n\n if current_password.is_empty() {\n bail!(\"please provide the current password\");\n }\n\n let new_password = self.new_password.clone().unwrap_or_else(|| {\n prompt_password(\"Please enter the new password: \").expect(\"Failed to read from input\")\n });\n\n if new_password.is_empty() {\n bail!(\"please provide a new password\");\n }\n\n let mut totp_code = self.totp_code.clone();\n\n loop {\n let response = client\n .change_password(¤t_password, &new_password, totp_code.as_deref())\n .await?;\n\n match response {\n MutateResponse::Success => break,\n MutateResponse::TwoFactorRequired => {\n totp_code = Some(super::login::or_user_input(None, \"two-factor code\"));\n }\n }\n }\n\n println!(\"Account password successfully changed!\");\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/account/delete.rs", "content": "use atuin_client::{\n auth::{self, MutateResponse},\n settings::Settings,\n};\nuse clap::Parser;\nuse eyre::{Result, bail};\n\nuse super::login::{or_user_input, read_user_password};\n\n#[derive(Parser, Debug)]\npub struct Cmd {\n #[clap(long, short)]\n pub password: Option,\n\n /// The two-factor authentication code for your account, if any\n #[clap(long, short)]\n pub totp_code: Option,\n}\n\nimpl Cmd {\n pub async fn run(&self, settings: &Settings) -> Result<()> {\n if !settings.logged_in().await? {\n bail!(\"You are not logged in\");\n }\n\n let client = auth::auth_client(settings).await;\n\n let password = self.password.clone().unwrap_or_else(read_user_password);\n\n if password.is_empty() {\n bail!(\"please provide your password\");\n }\n\n let mut totp_code = self.totp_code.clone();\n\n loop {\n let response = client\n .delete_account(&password, totp_code.as_deref())\n .await?;\n\n match response {\n MutateResponse::Success => break,\n MutateResponse::TwoFactorRequired => {\n totp_code = Some(or_user_input(None, \"two-factor code\"));\n }\n }\n }\n\n // Clean up sessions from meta store\n let meta = Settings::meta_store().await?;\n meta.delete_session().await?;\n meta.delete_hub_session().await?;\n\n println!(\"Your account is deleted\");\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/account/link.rs", "content": "use eyre::{Result, bail};\n\nuse atuin_client::settings::Settings;\n\npub async fn run(settings: &Settings) -> Result<()> {\n let meta = Settings::meta_store().await?;\n\n let cli_token = meta.session_token().await?;\n let hub_token = meta.hub_session_token().await?;\n\n let Some(cli_token) = cli_token else {\n bail!(\"No CLI session found. Please log in first with 'atuin login'.\");\n };\n\n let hub_address = settings.active_hub_endpoint().unwrap_or_default();\n\n if hub_token.is_some() {\n println!(\"Found both Hub and CLI sessions. Linking accounts...\");\n } else {\n println!(\"Found CLI session but no Hub session. Logging in to Hub first...\");\n\n let session = atuin_client::hub::HubAuthSession::start(hub_address.as_ref()).await?;\n println!(\"Open this URL to authenticate with Atuin Hub:\");\n println!(\"{}\", session.auth_url);\n\n let token = session\n .wait_for_completion(\n atuin_client::hub::DEFAULT_AUTH_TIMEOUT,\n atuin_client::hub::DEFAULT_POLL_INTERVAL,\n )\n .await?;\n\n atuin_client::hub::save_session(&token).await?;\n println!(\"Hub authentication complete.\");\n }\n\n atuin_client::hub::link_account(hub_address.as_ref(), &cli_token).await?;\n println!(\"Successfully linked CLI account to Hub.\");\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/account/login.rs", "content": "use std::{io, path::PathBuf};\n\nuse clap::Parser;\nuse eyre::{Context, Result, bail};\nuse tokio::{fs::File, io::AsyncWriteExt};\n\nuse atuin_client::{\n auth::{self, AuthResponse},\n encryption::{Key, decode_key, encode_key, load_key},\n record::sqlite_store::SqliteStore,\n record::store::Store,\n settings::Settings,\n};\nuse rpassword::prompt_password;\n\n#[derive(Parser, Debug)]\npub struct Cmd {\n #[clap(long, short)]\n pub username: Option,\n\n #[clap(long, short)]\n pub password: Option,\n\n /// The encryption key for your account\n #[clap(long, short)]\n pub key: Option,\n\n /// The two-factor authentication code for your account, if any\n #[clap(long, short)]\n pub totp_code: Option,\n\n #[clap(long, hide = true)]\n pub from_registration: bool,\n}\n\nfn get_input() -> Result {\n let mut input = String::new();\n io::stdin().read_line(&mut input)?;\n Ok(input.trim_end_matches(&['\\r', '\\n'][..]).to_string())\n}\n\nimpl Cmd {\n pub async fn run(&self, settings: &Settings, store: &SqliteStore) -> Result<()> {\n if settings.logged_in().await? {\n if settings.is_hub_sync() {\n println!(\"You are authenticated with Atuin Hub.\");\n } else {\n println!(\"You are already logged in.\");\n }\n println!(\"Run 'atuin logout' to log out.\");\n return Ok(());\n }\n\n if settings.is_hub_sync() {\n self.run_hub_login(settings, store).await\n } else {\n self.run_legacy_login(settings, store).await\n }\n }\n\n /// Hub login: use the browser OAuth flow unless all three flags\n /// (username, password, key) were provided for headless/CI use.\n async fn run_hub_login(&self, settings: &Settings, store: &SqliteStore) -> Result<()> {\n let endpoint = settings.active_hub_endpoint().unwrap_or_default();\n\n if let Some(username) = &self.username {\n // Headless login via v0 API (for CI / scripting).\n let client = auth::auth_client(settings).await;\n\n self.prompt_and_store_key(settings, store).await?;\n\n let password = self.password.clone().unwrap_or_else(read_user_password);\n let mut totp_code = self.totp_code.clone();\n\n let session = loop {\n let response = client\n .login(username, &password, totp_code.as_deref())\n .await?;\n\n match response {\n AuthResponse::Success { session } => break session,\n AuthResponse::TwoFactorRequired => {\n totp_code = Some(or_user_input(None, \"two-factor code\"));\n }\n }\n };\n\n Settings::meta_store()\n .await?\n .save_hub_session(&session)\n .await?;\n } else {\n // Interactive login via browser OAuth flow.\n if self.from_registration {\n load_key(settings)?;\n } else {\n self.prompt_and_store_key(settings, store).await?;\n }\n\n self.ensure_hub_session(settings, endpoint.as_ref()).await?;\n }\n\n // Silently attempt to link CLI account to Hub if one exists\n if let Ok(cli_token) = settings.session_token().await\n && let Err(e) = atuin_client::hub::link_account(endpoint.as_ref(), &cli_token).await\n {\n tracing::debug!(\"Could not link CLI account to Hub: {}\", e);\n }\n\n println!(\"Successfully authenticated with Atuin Hub.\");\n Ok(())\n }\n\n /// Legacy login: always prompt for username/password interactively\n /// (or accept them via flags).\n async fn run_legacy_login(&self, settings: &Settings, store: &SqliteStore) -> Result<()> {\n let username = or_user_input(self.username.clone(), \"username\");\n let password = self.password.clone().unwrap_or_else(read_user_password);\n\n self.prompt_and_store_key(settings, store).await?;\n\n let client = auth::auth_client(settings).await;\n let response = client.login(&username, &password, None).await?;\n\n match response {\n AuthResponse::Success { session } => {\n Settings::meta_store().await?.save_session(&session).await?;\n }\n AuthResponse::TwoFactorRequired => {\n // Legacy server doesn't support 2FA, so this shouldn't happen.\n bail!(\"unexpected two-factor requirement from legacy server\");\n }\n }\n\n println!(\"Logged in!\");\n Ok(())\n }\n\n async fn ensure_hub_session(&self, _settings: &Settings, hub_address: &str) -> Result<()> {\n tracing::info!(\"Authenticating with Atuin Hub...\");\n\n let session = atuin_client::hub::HubAuthSession::start(hub_address).await?;\n println!(\"Open this URL to continue authenticating with Atuin Hub:\");\n println!(\"{}\", session.auth_url);\n\n let token = session\n .wait_for_completion(\n atuin_client::hub::DEFAULT_AUTH_TIMEOUT,\n atuin_client::hub::DEFAULT_POLL_INTERVAL,\n )\n .await?;\n\n tracing::info!(\"Authentication complete, saving session token\");\n\n atuin_client::hub::save_session(&token).await?;\n\n Ok(())\n }\n\n async fn prompt_and_store_key(&self, settings: &Settings, store: &SqliteStore) -> Result<()> {\n let key_path = settings.key_path.as_str();\n let key_path = PathBuf::from(key_path);\n\n println!(\"IMPORTANT\");\n println!(\n \"If you are already logged in on another machine, you must ensure that the key you use here is the same as the key you used there.\"\n );\n println!(\"You can find your key by running 'atuin key' on the other machine.\");\n println!(\"Do not share this key with anyone.\");\n println!(\"\\nRead more here: https://docs.atuin.sh/guide/sync/#login \\n\");\n\n let key = or_user_input(\n self.key.clone(),\n \"encryption key [blank to use existing key file]\",\n );\n\n // if provided, the key may be EITHER base64, or a bip mnemonic\n // try to normalize on base64\n let key = if key.is_empty() {\n key\n } else {\n // try parse the key as a mnemonic...\n match bip39::Mnemonic::from_phrase(&key, bip39::Language::English) {\n Ok(mnemonic) => encode_key(Key::from_slice(mnemonic.entropy()))?,\n Err(err) => {\n match err {\n // assume they copied in the base64 key\n bip39::ErrorKind::InvalidWord(_) => key,\n bip39::ErrorKind::InvalidChecksum => {\n bail!(\"Key mnemonic is not valid\")\n }\n bip39::ErrorKind::InvalidKeysize(_)\n | bip39::ErrorKind::InvalidWordLength(_)\n | bip39::ErrorKind::InvalidEntropyLength(_, _) => {\n bail!(\"Key is not the correct length\")\n }\n }\n }\n }\n };\n\n if key.is_empty() {\n if key_path.exists() {\n let bytes = fs_err::read_to_string(&key_path).context(format!(\n \"Existing key file at '{}' could not be read\",\n key_path.to_string_lossy()\n ))?;\n if decode_key(bytes).is_err() {\n bail!(format!(\n \"The key in existing key file at '{}' is invalid\",\n key_path.to_string_lossy()\n ));\n }\n } else {\n panic!(\n \"No key provided and no existing key file found. Please use 'atuin key' on your other machine, or recover your key from a backup\"\n )\n }\n } else if !key_path.exists() {\n if decode_key(key.clone()).is_err() {\n bail!(\"The specified key is invalid\");\n }\n\n let mut file = File::create(&key_path).await?;\n file.write_all(key.as_bytes()).await?;\n } else {\n // we now know that the user has logged in specifying a key, AND that the key path\n // exists\n\n // 1. check if the saved key and the provided key match. if so, nothing to do.\n // 2. if not, re-encrypt the local history and overwrite the key\n let current_key: [u8; 32] = load_key(settings)?.into();\n\n let encoded = key.clone(); // gonna want to save it in a bit\n let new_key: [u8; 32] = decode_key(key)\n .context(\"Could not decode provided key; is not valid base64-encoded key\")?\n .into();\n\n if new_key != current_key {\n println!(\"\\nRe-encrypting local store with new key\");\n\n store.re_encrypt(¤t_key, &new_key).await?;\n\n println!(\"Writing new key\");\n let mut file = File::create(&key_path).await?;\n file.write_all(encoded.as_bytes()).await?;\n }\n }\n\n Ok(())\n }\n}\n\npub(super) fn or_user_input(value: Option, name: &'static str) -> String {\n value.unwrap_or_else(|| read_user_input(name))\n}\n\npub(super) fn read_user_password() -> String {\n let password = prompt_password(\"Please enter password: \");\n password.expect(\"Failed to read from input\")\n}\n\nfn read_user_input(name: &'static str) -> String {\n eprint!(\"Please enter {name}: \");\n get_input().expect(\"Failed to read from input\")\n}\n\n#[cfg(test)]\nmod tests {\n use atuin_client::encryption::Key;\n\n #[test]\n fn mnemonic_round_trip() {\n let key = Key::from([\n 3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5, 8, 9, 7, 9, 3, 2, 3, 8, 4, 6, 2, 6, 4, 3, 3, 8, 3, 2,\n 7, 9, 5,\n ]);\n let phrase = bip39::Mnemonic::from_entropy(&key, bip39::Language::English)\n .unwrap()\n .into_phrase();\n let mnemonic = bip39::Mnemonic::from_phrase(&phrase, bip39::Language::English).unwrap();\n assert_eq!(mnemonic.entropy(), key.as_slice());\n assert_eq!(\n phrase,\n \"adapt amused able anxiety mother adapt beef gaze amount else seat alcohol cage lottery avoid scare alcohol cactus school avoid coral adjust catch pink\"\n );\n }\n}\n" }, { "path": "crates/atuin/src/command/client/account/logout.rs", "content": "use eyre::Result;\n\npub async fn run() -> Result<()> {\n atuin_client::logout::logout().await\n}\n" }, { "path": "crates/atuin/src/command/client/account/register.rs", "content": "use clap::Parser;\nuse eyre::{Result, bail};\n\nuse super::login::or_user_input;\nuse atuin_client::{\n auth::{self, AuthResponse},\n record::sqlite_store::SqliteStore,\n settings::Settings,\n};\n\n#[derive(Parser, Debug)]\npub struct Cmd {\n #[clap(long, short)]\n pub username: Option,\n\n #[clap(long, short)]\n pub password: Option,\n\n #[clap(long, short)]\n pub email: Option,\n}\n\nimpl Cmd {\n pub async fn run(&self, settings: &Settings, store: &SqliteStore) -> Result<()> {\n if settings.logged_in().await? {\n if settings.is_hub_sync() {\n println!(\"You are already authenticated with Atuin Hub.\");\n } else {\n println!(\"You are already logged in.\");\n }\n println!(\"Run 'atuin logout' to log out.\");\n return Ok(());\n }\n\n if settings.is_hub_sync() {\n let required_for_headless = 3;\n let provided = [\n self.username.is_some(),\n self.email.is_some(),\n self.password.is_some(),\n ]\n .iter()\n .filter(|&b| *b)\n .count();\n if provided < required_for_headless {\n println!(\n \"Username, password, and email are all required for headless registration. Continuing with interactive registration.\\n\"\n );\n }\n\n if let (Some(username), Some(email), Some(password)) =\n (&self.username, &self.email, &self.password)\n {\n // Headless registration via v0 API (for CI / scripting).\n let client = auth::auth_client(settings).await;\n\n if password.is_empty() {\n bail!(\"please provide a password\");\n }\n\n let response = client.register(username, email, password).await?;\n\n match response {\n AuthResponse::Success { session } => {\n Settings::meta_store()\n .await?\n .save_hub_session(&session)\n .await?;\n }\n AuthResponse::TwoFactorRequired => {\n bail!(\"unexpected two-factor requirement during registration\");\n }\n }\n\n let _key = atuin_client::encryption::load_key(settings)?;\n\n println!(\n \"Registration successful! Please make a note of your key (run 'atuin key') and keep it safe.\"\n );\n println!(\n \"You will need it to log in on other devices, and we cannot help recover it if you lose it.\"\n );\n } else {\n // Interactive registration: delegate to the browser OAuth flow.\n // Registration on Hub happens on the website; the CLI just needs\n // to authenticate afterwards.\n super::login::Cmd {\n username: None,\n password: None,\n key: None,\n totp_code: None,\n from_registration: true,\n }\n .run(settings, store)\n .await?;\n }\n } else {\n // Legacy registration flow\n println!(\"Registering for an Atuin Sync account\");\n\n let username = or_user_input(self.username.clone(), \"username\");\n let email = or_user_input(self.email.clone(), \"email\");\n let password = self\n .password\n .clone()\n .unwrap_or_else(super::login::read_user_password);\n\n if password.is_empty() {\n bail!(\"please provide a password\");\n }\n\n let session = atuin_client::api_client::register(\n settings.sync_address.as_str(),\n &username,\n &email,\n &password,\n )\n .await?;\n\n let meta = Settings::meta_store().await?;\n meta.save_session(&session.session).await?;\n\n let _key = atuin_client::encryption::load_key(settings)?;\n\n println!(\n \"Registration successful! Please make a note of your key (run 'atuin key') and keep it safe.\"\n );\n println!(\n \"You will need it to log in on other devices, and we cannot help recover it if you lose it.\"\n );\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/account.rs", "content": "use clap::{Args, Subcommand};\nuse eyre::Result;\n\nuse atuin_client::record::sqlite_store::SqliteStore;\nuse atuin_client::settings::Settings;\n\npub mod change_password;\npub mod delete;\npub mod link;\npub mod login;\npub mod logout;\npub mod register;\n\n#[derive(Args, Debug)]\npub struct Cmd {\n #[command(subcommand)]\n command: Commands,\n}\n\n#[derive(Subcommand, Debug)]\npub enum Commands {\n /// Login to the configured server\n Login(login::Cmd),\n\n /// Register a new account\n Register(register::Cmd),\n\n /// Log out\n Logout,\n\n /// Delete your account, and all synced data\n Delete(delete::Cmd),\n\n /// Change your password\n ChangePassword(change_password::Cmd),\n\n /// Link your CLI sync account to your Hub account\n Link,\n}\n\nimpl Cmd {\n pub async fn run(self, settings: Settings, store: SqliteStore) -> Result<()> {\n match self.command {\n Commands::Login(l) => l.run(&settings, &store).await,\n Commands::Register(r) => r.run(&settings, &store).await,\n Commands::Logout => logout::run().await,\n Commands::Delete(d) => d.run(&settings).await,\n Commands::ChangePassword(c) => c.run(&settings).await,\n Commands::Link => link::run(&settings).await,\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/client/daemon.rs", "content": "use std::fs::{self, File, OpenOptions};\nuse std::io::{ErrorKind, Write};\n#[cfg(unix)]\nuse std::os::unix::net::UnixStream as StdUnixStream;\nuse std::path::{Path, PathBuf};\nuse std::process::{Command, Stdio};\nuse std::time::{Duration, Instant};\n\nuse atuin_client::{\n database::Sqlite, history::History, record::sqlite_store::SqliteStore, settings::Settings,\n};\nuse atuin_daemon::client::{DaemonClientErrorKind, HistoryClient, classify_error};\nuse clap::Subcommand;\n#[cfg(unix)]\nuse daemonize::Daemonize;\nuse eyre::{Result, WrapErr, bail, eyre};\nuse fs4::fs_std::FileExt;\nuse tokio::time::sleep;\n\n#[derive(clap::Args, Debug)]\npub struct Cmd {\n /// Internal flag for daemonization\n #[arg(long, hide = true)]\n daemonize: bool,\n\n /// Also write daemon logs to the console (useful for debugging)\n #[arg(long)]\n show_logs: bool,\n\n #[command(subcommand)]\n subcmd: Option,\n}\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum SubCmd {\n /// Start the daemon server\n Start {\n #[arg(long, hide = true)]\n daemonize: bool,\n\n /// Also write daemon logs to the console (useful for debugging)\n #[arg(long)]\n show_logs: bool,\n\n /// Force start: kill existing daemon process and reset the socket\n #[arg(long)]\n force: bool,\n },\n\n /// Show the daemon's current status\n Status,\n\n /// Stop the daemon gracefully\n Stop,\n\n /// Restart the daemon (stop, then start in background)\n Restart,\n}\n\nimpl Cmd {\n /// Returns `true` when the process should daemonize before creating the\n /// async runtime or opening any database connections.\n #[cfg(unix)]\n pub fn should_daemonize(&self) -> bool {\n match &self.subcmd {\n Some(SubCmd::Start { daemonize, .. }) => *daemonize,\n None => self.daemonize,\n _ => false,\n }\n }\n\n /// Returns `true` when logs should also be written to the console.\n pub fn show_logs(&self) -> bool {\n match &self.subcmd {\n Some(SubCmd::Start { show_logs, .. }) => *show_logs,\n None => self.show_logs,\n _ => false,\n }\n }\n\n pub async fn run(\n self,\n settings: Settings,\n store: SqliteStore,\n history_db: Sqlite,\n ) -> Result<()> {\n match self.subcmd {\n None => {\n eprintln!(\"Warning: `atuin daemon` is deprecated, use `atuin daemon start`\");\n run(settings, store, history_db, false).await\n }\n Some(SubCmd::Start { force, .. }) => run(settings, store, history_db, force).await,\n Some(SubCmd::Status) => status_cmd(&settings).await,\n Some(SubCmd::Stop) => stop_cmd(&settings).await,\n Some(SubCmd::Restart) => restart_cmd(&settings).await,\n }\n }\n}\n\nconst DAEMON_VERSION: &str = env!(\"CARGO_PKG_VERSION\");\nconst DAEMON_PROTOCOL_VERSION: u32 = 1;\nconst STARTUP_POLL: Duration = Duration::from_millis(40);\nconst LOCK_POLL: Duration = Duration::from_millis(20);\nconst LEGACY_DAEMON_RESTART_MESSAGE: &str = \"legacy daemon detected; restart daemon manually\";\n\nstruct PidfileGuard {\n file: File,\n}\n\nimpl PidfileGuard {\n fn acquire(path: &Path) -> Result {\n let mut file = open_lock_file(path)?;\n\n if !file.try_lock_exclusive()? {\n bail!(\n \"daemon already running (pidfile lock busy at {})\",\n path.display()\n );\n }\n\n file.set_len(0)\n .wrap_err_with(|| format!(\"could not truncate daemon pidfile {}\", path.display()))?;\n writeln!(file, \"{}\", std::process::id())\n .and_then(|()| writeln!(file, \"{DAEMON_VERSION}\"))\n .wrap_err_with(|| format!(\"could not write daemon pidfile {}\", path.display()))?;\n\n Ok(Self { file })\n }\n}\n\nimpl Drop for PidfileGuard {\n fn drop(&mut self) {\n let _ = self.file.unlock();\n }\n}\n\nenum Probe {\n Ready(HistoryClient),\n NeedsRestart(String),\n Unreachable(eyre::Report),\n}\n\nfn daemon_matches_expected(version: &str, protocol: u32) -> bool {\n version == DAEMON_VERSION && protocol == DAEMON_PROTOCOL_VERSION\n}\n\nfn daemon_mismatch_message(version: &str, protocol: u32) -> String {\n if protocol == DAEMON_PROTOCOL_VERSION {\n format!(\"daemon is out of date: expected {DAEMON_VERSION}, got {version}\")\n } else {\n format!(\"daemon protocol mismatch: expected {DAEMON_PROTOCOL_VERSION}, got {protocol}\")\n }\n}\n\nfn is_legacy_daemon_error(err: &eyre::Report) -> bool {\n matches!(classify_error(err), DaemonClientErrorKind::Unimplemented)\n}\n\nfn should_retry_after_error(err: &eyre::Report) -> bool {\n matches!(\n classify_error(err),\n DaemonClientErrorKind::Connect\n | DaemonClientErrorKind::Unavailable\n | DaemonClientErrorKind::Unimplemented\n )\n}\n\nfn daemon_startup_lock_path(pidfile_path: &Path) -> PathBuf {\n let mut os = pidfile_path.as_os_str().to_os_string();\n os.push(\".startup.lock\");\n PathBuf::from(os)\n}\n\nfn open_lock_file(path: &Path) -> Result {\n if let Some(parent) = path.parent() {\n fs::create_dir_all(parent)\n .wrap_err_with(|| format!(\"could not create lock directory {}\", parent.display()))?;\n }\n\n OpenOptions::new()\n .read(true)\n .write(true)\n .create(true)\n .truncate(false)\n .open(path)\n .wrap_err_with(|| format!(\"could not open lock file {}\", path.display()))\n}\n\nasync fn wait_for_lock(path: &Path, timeout: Duration) -> Result {\n let file = open_lock_file(path)?;\n let start = Instant::now();\n\n loop {\n match file.try_lock_exclusive() {\n Ok(true) => return Ok(file),\n Ok(false) => {\n if start.elapsed() >= timeout {\n bail!(\"timed out waiting for lock at {}\", path.display());\n }\n\n sleep(LOCK_POLL).await;\n }\n Err(err) => {\n return Err(eyre!(\"could not lock {}: {err}\", path.display()));\n }\n }\n }\n}\n\nasync fn wait_for_pidfile_available(path: &Path, timeout: Duration) -> Result<()> {\n let file = wait_for_lock(path, timeout).await?;\n file.unlock()\n .wrap_err_with(|| format!(\"failed to unlock {}\", path.display()))?;\n Ok(())\n}\n\nasync fn connect_client(settings: &Settings) -> Result {\n HistoryClient::new(\n #[cfg(not(unix))]\n settings.daemon.tcp_port,\n #[cfg(unix)]\n settings.daemon.socket_path.clone(),\n )\n .await\n}\n\nasync fn probe(settings: &Settings) -> Probe {\n let mut client = match connect_client(settings).await {\n Ok(client) => client,\n Err(err) => return Probe::Unreachable(err),\n };\n\n match client.status().await {\n Ok(status) => {\n if daemon_matches_expected(&status.version, status.protocol) {\n Probe::Ready(client)\n } else {\n Probe::NeedsRestart(daemon_mismatch_message(&status.version, status.protocol))\n }\n }\n Err(err) => Probe::Unreachable(err),\n }\n}\n\nasync fn request_shutdown(settings: &Settings) {\n if let Ok(mut client) = connect_client(settings).await {\n let _ = client.shutdown().await;\n }\n}\n\nfn spawn_daemon_process() -> Result<()> {\n let exe = std::env::current_exe().wrap_err(\"could not locate atuin executable\")?;\n\n let mut cmd = Command::new(exe);\n cmd.arg(\"daemon\")\n .arg(\"start\")\n .stdin(Stdio::null())\n .stdout(Stdio::null())\n .stderr(Stdio::null());\n\n #[cfg(unix)]\n cmd.arg(\"--daemonize\");\n\n cmd.spawn().wrap_err(\"failed to spawn daemon process\")?;\n\n Ok(())\n}\n\nfn startup_timeout(settings: &Settings) -> Duration {\n Duration::from_secs_f64(settings.local_timeout.max(0.5) + 2.0)\n}\n\n#[cfg(unix)]\nfn remove_stale_socket_if_present(settings: &Settings) -> Result<()> {\n if settings.daemon.systemd_socket {\n return Ok(());\n }\n\n let socket_path = Path::new(&settings.daemon.socket_path);\n if !socket_path.exists() {\n return Ok(());\n }\n\n match StdUnixStream::connect(socket_path) {\n Ok(stream) => {\n drop(stream);\n Ok(())\n }\n Err(err) if err.kind() == ErrorKind::ConnectionRefused => {\n fs::remove_file(socket_path).wrap_err_with(|| {\n format!(\n \"failed to remove stale daemon socket {}\",\n socket_path.display()\n )\n })?;\n Ok(())\n }\n Err(err) if err.kind() == ErrorKind::NotFound => Ok(()),\n Err(_) => Ok(()),\n }\n}\n\nasync fn wait_until_ready(settings: &Settings, timeout: Duration) -> Result {\n let start = Instant::now();\n let mut last_error = eyre!(\"daemon did not become ready\");\n\n loop {\n match probe(settings).await {\n Probe::Ready(client) => return Ok(client),\n Probe::NeedsRestart(reason) => {\n last_error = eyre!(reason);\n }\n Probe::Unreachable(err) => {\n if is_legacy_daemon_error(&err) {\n return Err(err.wrap_err(LEGACY_DAEMON_RESTART_MESSAGE));\n }\n last_error = err;\n }\n }\n\n if start.elapsed() >= timeout {\n return Err(last_error.wrap_err(format!(\n \"timed out waiting for daemon startup after {}ms\",\n timeout.as_millis()\n )));\n }\n\n sleep(STARTUP_POLL).await;\n }\n}\n\nfn ensure_autostart_supported(settings: &Settings) -> Result<()> {\n #[cfg(unix)]\n if settings.daemon.systemd_socket {\n bail!(\n \"daemon autostart is incompatible with `daemon.systemd_socket = true`; use systemd to manage the daemon\"\n );\n }\n #[cfg(not(unix))]\n let _ = settings;\n\n Ok(())\n}\n\nasync fn restart_daemon(settings: &Settings) -> Result {\n ensure_autostart_supported(settings)?;\n\n let timeout = startup_timeout(settings);\n let pidfile_path = PathBuf::from(&settings.daemon.pidfile_path);\n let startup_lock_path = daemon_startup_lock_path(&pidfile_path);\n let startup_lock = wait_for_lock(&startup_lock_path, timeout).await?;\n\n match probe(settings).await {\n Probe::Ready(client) => {\n drop(startup_lock);\n return Ok(client);\n }\n Probe::NeedsRestart(_) => {\n request_shutdown(settings).await;\n }\n Probe::Unreachable(err) => {\n if is_legacy_daemon_error(&err) {\n return Err(err.wrap_err(LEGACY_DAEMON_RESTART_MESSAGE));\n }\n }\n }\n\n // This prevents rapid-fire hook invocations from racing daemon restart.\n wait_for_pidfile_available(&pidfile_path, timeout).await?;\n\n #[cfg(unix)]\n remove_stale_socket_if_present(settings)?;\n\n spawn_daemon_process()?;\n let client = wait_until_ready(settings, timeout).await?;\n\n drop(startup_lock);\n Ok(client)\n}\n\nfn ensure_reply_compatible(settings: &Settings, version: &str, protocol: u32) -> Result<()> {\n if daemon_matches_expected(version, protocol) {\n return Ok(());\n }\n\n let message = daemon_mismatch_message(version, protocol);\n if settings.daemon.autostart {\n bail!(\"{message}\");\n }\n\n bail!(\"{message}. Enable `daemon.autostart = true` or restart the daemon manually\");\n}\n\npub async fn start_history(settings: &Settings, history: History) -> Result {\n match async {\n connect_client(settings)\n .await?\n .start_history(history.clone())\n .await\n }\n .await\n {\n Ok(resp) => {\n if daemon_matches_expected(&resp.version, resp.protocol) {\n return Ok(resp.id);\n }\n\n if !settings.daemon.autostart {\n return Err(eyre!(\n \"{}. Enable `daemon.autostart = true` or restart the daemon manually\",\n daemon_mismatch_message(&resp.version, resp.protocol)\n ));\n }\n }\n Err(err) if !settings.daemon.autostart => return Err(err),\n Err(err) if !should_retry_after_error(&err) => return Err(err),\n Err(_) => {}\n }\n\n let resp = restart_daemon(settings)\n .await?\n .start_history(history)\n .await?;\n ensure_reply_compatible(settings, &resp.version, resp.protocol)?;\n Ok(resp.id)\n}\n\npub async fn end_history(settings: &Settings, id: String, duration: u64, exit: i64) -> Result<()> {\n match async {\n connect_client(settings)\n .await?\n .end_history(id.clone(), duration, exit)\n .await\n }\n .await\n {\n Ok(resp) => {\n if daemon_matches_expected(&resp.version, resp.protocol) {\n return Ok(());\n }\n\n if !settings.daemon.autostart {\n return Err(eyre!(\n \"{}. Enable `daemon.autostart = true` or restart the daemon manually\",\n daemon_mismatch_message(&resp.version, resp.protocol)\n ));\n }\n\n // End succeeded on the running daemon, so avoid replaying it.\n // We only restart to make subsequent hook calls target the expected version.\n let _ = restart_daemon(settings).await;\n return Ok(());\n }\n Err(err) if !settings.daemon.autostart => return Err(err),\n Err(err) if !should_retry_after_error(&err) => return Err(err),\n Err(_) => {}\n }\n\n let resp = restart_daemon(settings)\n .await?\n .end_history(id, duration, exit)\n .await?;\n ensure_reply_compatible(settings, &resp.version, resp.protocol)?;\n Ok(())\n}\n\nasync fn status_cmd(settings: &Settings) -> Result<()> {\n match probe(settings).await {\n Probe::Ready(mut client) => {\n let status = client.status().await?;\n println!(\"Daemon running\");\n println!(\" PID: {}\", status.pid);\n println!(\" Version: {}\", status.version);\n println!(\" Protocol: {}\", status.protocol);\n println!(\" Healthy: {}\", status.healthy);\n #[cfg(unix)]\n println!(\" Socket: {}\", settings.daemon.socket_path);\n #[cfg(not(unix))]\n println!(\" Port: {}\", settings.daemon.tcp_port);\n }\n Probe::NeedsRestart(reason) => {\n println!(\"Daemon running (needs restart)\");\n println!(\" Reason: {reason}\");\n }\n Probe::Unreachable(_) => {\n println!(\"Daemon is not running\");\n }\n }\n\n Ok(())\n}\n\nasync fn stop_cmd(settings: &Settings) -> Result<()> {\n let Ok(mut client) = connect_client(settings).await else {\n println!(\"Daemon is not running\");\n return Ok(());\n };\n\n match client.shutdown().await {\n Ok(true) => {\n println!(\"Shutdown requested\");\n\n let pidfile_path = PathBuf::from(&settings.daemon.pidfile_path);\n let timeout = Duration::from_secs(5);\n match wait_for_pidfile_available(&pidfile_path, timeout).await {\n Ok(()) => println!(\"Daemon stopped\"),\n Err(_) => println!(\"Daemon may still be shutting down\"),\n }\n\n Ok(())\n }\n Ok(false) => bail!(\"Daemon rejected shutdown request\"),\n Err(err) => Err(err.wrap_err(\"Failed to send shutdown request\")),\n }\n}\n\nasync fn restart_cmd(settings: &Settings) -> Result<()> {\n // Stop if running\n match probe(settings).await {\n Probe::Ready(_) | Probe::NeedsRestart(_) => {\n request_shutdown(settings).await;\n println!(\"Stopping daemon...\");\n\n let pidfile_path = PathBuf::from(&settings.daemon.pidfile_path);\n let timeout = Duration::from_secs(5);\n wait_for_pidfile_available(&pidfile_path, timeout)\n .await\n .wrap_err(\"Timed out waiting for old daemon to stop\")?;\n }\n Probe::Unreachable(_) => {\n println!(\"No daemon running\");\n }\n }\n\n #[cfg(unix)]\n remove_stale_socket_if_present(settings)?;\n\n spawn_daemon_process()?;\n println!(\"Starting daemon...\");\n\n let timeout = startup_timeout(settings);\n let status = wait_until_ready(settings, timeout).await?.status().await?;\n\n println!(\"Daemon restarted\");\n println!(\" PID: {}\", status.pid);\n println!(\" Version: {}\", status.version);\n\n Ok(())\n}\n\n/// Daemonize the current process. Must be called before creating the tokio\n/// runtime or opening database connections, since `fork()` inside an async\n/// runtime corrupts its internal state.\n#[cfg(unix)]\npub fn daemonize_current_process() -> Result<()> {\n let cwd =\n std::env::current_dir().wrap_err(\"could not determine current directory for daemon\")?;\n\n Daemonize::new()\n .working_directory(cwd)\n .start()\n .wrap_err(\"failed to daemonize process\")?;\n\n Ok(())\n}\n\nasync fn run(\n settings: Settings,\n store: SqliteStore,\n history_db: Sqlite,\n force: bool,\n) -> Result<()> {\n if force {\n force_cleanup(&settings);\n }\n\n let pidfile_path = PathBuf::from(&settings.daemon.pidfile_path);\n let _pidfile_guard = PidfileGuard::acquire(&pidfile_path)?;\n\n atuin_daemon::boot(settings, store, history_db).await?;\n\n Ok(())\n}\n\n/// Force cleanup: kill existing daemon process and remove socket.\nfn force_cleanup(settings: &Settings) {\n let pidfile_path = Path::new(&settings.daemon.pidfile_path);\n\n // Read and kill the existing process if pidfile exists\n if pidfile_path.exists() {\n if let Ok(contents) = fs::read_to_string(pidfile_path)\n && let Some(pid_str) = contents.lines().next()\n && let Ok(pid) = pid_str.parse::()\n {\n kill_process(pid);\n // Give it a moment to release resources\n std::thread::sleep(Duration::from_millis(100));\n }\n\n // Remove the pidfile\n if let Err(e) = fs::remove_file(pidfile_path)\n && e.kind() != ErrorKind::NotFound\n {\n tracing::warn!(\"failed to remove pidfile: {e}\");\n }\n }\n\n // Remove the socket file\n #[cfg(unix)]\n {\n let socket_path = Path::new(&settings.daemon.socket_path);\n if socket_path.exists()\n && let Err(e) = fs::remove_file(socket_path)\n && e.kind() != ErrorKind::NotFound\n {\n tracing::warn!(\"failed to remove socket: {e}\");\n }\n }\n}\n\n/// Kill a process by PID.\n#[cfg(unix)]\nfn kill_process(pid: u32) {\n // Use kill command to send SIGTERM for graceful shutdown\n let _ = Command::new(\"kill\")\n .args([\"-TERM\", &pid.to_string()])\n .stdout(Stdio::null())\n .stderr(Stdio::null())\n .status();\n}\n\n/// Kill a process by PID.\n#[cfg(not(unix))]\nfn kill_process(pid: u32) {\n // On Windows, use taskkill\n let _ = Command::new(\"taskkill\")\n .args([\"/PID\", &pid.to_string(), \"/F\"])\n .stdout(Stdio::null())\n .stderr(Stdio::null())\n .status();\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_version_matches() {\n assert!(daemon_matches_expected(\n DAEMON_VERSION,\n DAEMON_PROTOCOL_VERSION\n ));\n }\n\n #[test]\n fn test_version_mismatch() {\n assert!(!daemon_matches_expected(\"0.0.0\", DAEMON_PROTOCOL_VERSION));\n assert!(!daemon_matches_expected(DAEMON_VERSION, 999));\n assert!(!daemon_matches_expected(\"0.0.0\", 999));\n }\n\n #[test]\n fn test_mismatch_message_version() {\n let msg = daemon_mismatch_message(\"0.0.0\", DAEMON_PROTOCOL_VERSION);\n assert!(msg.contains(\"out of date\"), \"got: {msg}\");\n assert!(msg.contains(\"0.0.0\"));\n assert!(msg.contains(DAEMON_VERSION));\n }\n\n #[test]\n fn test_mismatch_message_protocol() {\n let msg = daemon_mismatch_message(DAEMON_VERSION, 999);\n assert!(msg.contains(\"protocol mismatch\"), \"got: {msg}\");\n }\n\n #[test]\n fn test_startup_lock_path() {\n let pidfile = Path::new(\"/tmp/atuin-daemon.pid\");\n let lock = daemon_startup_lock_path(pidfile);\n assert_eq!(lock, PathBuf::from(\"/tmp/atuin-daemon.pid.startup.lock\"));\n }\n\n #[test]\n fn test_pidfile_guard_acquire_and_drop() {\n let tmp = tempfile::tempdir().unwrap();\n let pidfile = tmp.path().join(\"daemon.pid\");\n\n {\n let _guard = PidfileGuard::acquire(&pidfile).unwrap();\n // Guard holds an exclusive lock — on Windows other handles cannot\n // read the file, so we verify contents after the guard is dropped.\n }\n\n let contents = std::fs::read_to_string(&pidfile).unwrap();\n let lines: Vec<&str> = contents.lines().collect();\n assert_eq!(lines.len(), 2);\n assert_eq!(lines[0], std::process::id().to_string());\n assert_eq!(lines[1], DAEMON_VERSION);\n\n // After guard is dropped, lock should be released — acquiring again must succeed.\n let _guard2 = PidfileGuard::acquire(&pidfile).unwrap();\n }\n\n #[test]\n fn test_pidfile_guard_prevents_double_acquire() {\n let tmp = tempfile::tempdir().unwrap();\n let pidfile = tmp.path().join(\"daemon.pid\");\n\n let _guard = PidfileGuard::acquire(&pidfile).unwrap();\n let result = PidfileGuard::acquire(&pidfile);\n assert!(result.is_err());\n }\n}\n" }, { "path": "crates/atuin/src/command/client/default_config.rs", "content": "use atuin_client::settings::Settings;\n\npub fn run() {\n println!(\"{}\", Settings::example_config());\n}\n" }, { "path": "crates/atuin/src/command/client/doctor.rs", "content": "use std::process::Command;\nuse std::{env, str::FromStr};\n\nuse atuin_client::database::Sqlite;\nuse atuin_client::settings::Settings;\nuse atuin_common::shell::{Shell, shell_name};\nuse atuin_common::utils;\nuse colored::Colorize;\nuse eyre::Result;\nuse serde::Serialize;\n\nuse sysinfo::{Disks, System, get_current_pid};\n\n#[derive(Debug, Serialize)]\nstruct ShellInfo {\n pub name: String,\n\n // best-effort, not supported on all OSes\n pub default: String,\n\n // Detect some shell plugins that the user has installed.\n // I'm just going to start with preexec/blesh\n pub plugins: Vec,\n\n // The preexec framework used in the current session, if Atuin is loaded.\n pub preexec: Option,\n}\n\nimpl ShellInfo {\n // HACK ALERT!\n // Many of the shell vars we need to detect are not exported :(\n // So, we're going to run a interactive session and directly check the\n // variable. There's a chance this won't work, so it should not be fatal.\n //\n // Every shell we support handles `shell -ic 'command'`\n fn shellvar_exists(shell: &str, var: &str) -> bool {\n let cmd = Command::new(shell)\n .args([\n \"-ic\",\n format!(\"[ -z ${var} ] || echo ATUIN_DOCTOR_ENV_FOUND\").as_str(),\n ])\n .output()\n .map_or(String::new(), |v| {\n let out = v.stdout;\n String::from_utf8(out).unwrap_or_default()\n });\n\n cmd.contains(\"ATUIN_DOCTOR_ENV_FOUND\")\n }\n\n fn detect_preexec_framework(shell: &str) -> Option {\n if env::var(\"ATUIN_SESSION\").ok().is_none() {\n None\n } else if shell.starts_with(\"bash\") || shell == \"sh\" {\n env::var(\"ATUIN_PREEXEC_BACKEND\")\n .ok()\n .filter(|value| !value.is_empty())\n .and_then(|atuin_preexec_backend| {\n atuin_preexec_backend.rfind(':').and_then(|pos_colon| {\n u32::from_str(&atuin_preexec_backend[..pos_colon])\n .ok()\n .is_some_and(|preexec_shlvl| {\n env::var(\"SHLVL\")\n .ok()\n .and_then(|shlvl| u32::from_str(&shlvl).ok())\n .is_some_and(|shlvl| shlvl == preexec_shlvl)\n })\n .then(|| atuin_preexec_backend[pos_colon + 1..].to_string())\n })\n })\n } else {\n Some(\"built-in\".to_string())\n }\n }\n\n fn validate_plugin_blesh(\n _shell: &str,\n shell_process: &sysinfo::Process,\n ble_session_id: &str,\n ) -> Option {\n ble_session_id\n .split('/')\n .nth(1)\n .and_then(|field| u32::from_str(field).ok())\n .filter(|&blesh_pid| blesh_pid == shell_process.pid().as_u32())\n .map(|_| \"blesh\".to_string())\n }\n\n pub fn plugins(shell: &str, shell_process: &sysinfo::Process) -> Vec {\n // consider a different detection approach if there are plugins\n // that don't set shell vars\n\n enum PluginShellType {\n Any,\n Bash,\n\n // Note: these are currently unused\n #[allow(dead_code)]\n Zsh,\n #[allow(dead_code)]\n Fish,\n #[allow(dead_code)]\n Nushell,\n #[allow(dead_code)]\n Xonsh,\n }\n\n enum PluginProbeType {\n EnvironmentVariable(&'static str),\n InteractiveShellVariable(&'static str),\n }\n\n type PluginValidator = fn(&str, &sysinfo::Process, &str) -> Option;\n\n let plugin_list: [(\n &str,\n PluginShellType,\n PluginProbeType,\n Option,\n ); 3] = [\n (\n \"atuin\",\n PluginShellType::Any,\n PluginProbeType::EnvironmentVariable(\"ATUIN_SESSION\"),\n None,\n ),\n (\n \"blesh\",\n PluginShellType::Bash,\n PluginProbeType::EnvironmentVariable(\"BLE_SESSION_ID\"),\n Some(Self::validate_plugin_blesh),\n ),\n (\n \"bash-preexec\",\n PluginShellType::Bash,\n PluginProbeType::InteractiveShellVariable(\"bash_preexec_imported\"),\n None,\n ),\n ];\n\n plugin_list\n .into_iter()\n .filter(|(_, shell_type, _, _)| match shell_type {\n PluginShellType::Any => true,\n PluginShellType::Bash => shell.starts_with(\"bash\") || shell == \"sh\",\n PluginShellType::Zsh => shell.starts_with(\"zsh\"),\n PluginShellType::Fish => shell.starts_with(\"fish\"),\n PluginShellType::Nushell => shell.starts_with(\"nu\"),\n PluginShellType::Xonsh => shell.starts_with(\"xonsh\"),\n })\n .filter_map(|(plugin, _, probe_type, validator)| -> Option {\n match probe_type {\n PluginProbeType::EnvironmentVariable(env) => {\n env::var(env).ok().filter(|value| !value.is_empty())\n }\n PluginProbeType::InteractiveShellVariable(shellvar) => {\n ShellInfo::shellvar_exists(shell, shellvar).then_some(String::default())\n }\n }\n .and_then(|value| {\n validator.map_or_else(\n || Some(plugin.to_string()),\n |validator| validator(shell, shell_process, &value),\n )\n })\n })\n .collect()\n }\n\n pub fn new() -> Self {\n // TODO: rework to use atuin_common::Shell\n\n let sys = System::new_all();\n\n let process = sys\n .process(get_current_pid().expect(\"Failed to get current PID\"))\n .expect(\"Process with current pid does not exist\");\n\n let parent = sys\n .process(process.parent().expect(\"Atuin running with no parent!\"))\n .expect(\"Process with parent pid does not exist\");\n\n let name = shell_name(Some(parent));\n\n let plugins = ShellInfo::plugins(name.as_str(), parent);\n\n let default = Shell::default_shell().unwrap_or(Shell::Unknown).to_string();\n\n let preexec = Self::detect_preexec_framework(name.as_str());\n\n Self {\n name,\n default,\n plugins,\n preexec,\n }\n }\n}\n\n#[derive(Debug, Serialize)]\nstruct DiskInfo {\n pub name: String,\n pub filesystem: String,\n}\n\n#[derive(Debug, Serialize)]\nstruct SystemInfo {\n pub os: String,\n\n pub arch: String,\n\n pub version: String,\n pub disks: Vec,\n}\n\nimpl SystemInfo {\n pub fn new() -> Self {\n let disks = Disks::new_with_refreshed_list();\n let disks = disks\n .list()\n .iter()\n .map(|d| DiskInfo {\n name: d.name().to_os_string().into_string().unwrap(),\n filesystem: d.file_system().to_os_string().into_string().unwrap(),\n })\n .collect();\n\n Self {\n os: System::name().unwrap_or_else(|| \"unknown\".to_string()),\n arch: System::cpu_arch().unwrap_or_else(|| \"unknown\".to_string()),\n version: System::os_version().unwrap_or_else(|| \"unknown\".to_string()),\n disks,\n }\n }\n}\n\n#[derive(Debug, Serialize)]\nstruct SyncInfo {\n /// Whether the main Atuin sync server is in use\n /// I'm just calling it Atuin Cloud for lack of a better name atm\n pub cloud: bool,\n pub records: bool,\n pub auto_sync: bool,\n\n pub last_sync: String,\n}\n\nimpl SyncInfo {\n pub async fn new(settings: &Settings) -> Self {\n Self {\n cloud: settings.is_hub_sync(),\n auto_sync: settings.auto_sync,\n records: settings.sync.records,\n last_sync: Settings::last_sync()\n .await\n .map_or_else(|_| \"no last sync\".to_string(), |v| v.to_string()),\n }\n }\n}\n\n#[derive(Debug)]\nstruct SettingPaths {\n db: String,\n record_store: String,\n key: String,\n}\n\nimpl SettingPaths {\n pub fn new(settings: &Settings) -> Self {\n Self {\n db: settings.db_path.clone(),\n record_store: settings.record_store_path.clone(),\n key: settings.key_path.clone(),\n }\n }\n\n pub fn verify(&self) {\n let paths = vec![\n (\"ATUIN_DB_PATH\", &self.db),\n (\"ATUIN_RECORD_STORE\", &self.record_store),\n (\"ATUIN_KEY\", &self.key),\n ];\n\n for (path_env_var, path) in paths {\n if utils::broken_symlink(path) {\n eprintln!(\n \"{path} (${path_env_var}) is a broken symlink. This may cause issues with Atuin.\"\n );\n }\n }\n }\n}\n\n#[derive(Debug, Serialize)]\nstruct AtuinInfo {\n pub version: String,\n pub commit: String,\n\n /// Whether the main Atuin sync server is in use\n /// I'm just calling it Atuin Cloud for lack of a better name atm\n pub sync: Option,\n\n pub sqlite_version: String,\n\n #[serde(skip)] // probably unnecessary to expose this\n pub setting_paths: SettingPaths,\n}\n\nimpl AtuinInfo {\n pub async fn new(settings: &Settings) -> Self {\n let logged_in = settings.logged_in().await.unwrap_or(false);\n\n let sync = if logged_in {\n Some(SyncInfo::new(settings).await)\n } else {\n None\n };\n\n let sqlite_version = match Sqlite::new(\"sqlite::memory:\", 0.1).await {\n Ok(db) => db\n .sqlite_version()\n .await\n .unwrap_or_else(|_| \"unknown\".to_string()),\n Err(_) => \"error\".to_string(),\n };\n\n Self {\n version: crate::VERSION.to_string(),\n commit: crate::SHA.to_string(),\n sync,\n sqlite_version,\n setting_paths: SettingPaths::new(settings),\n }\n }\n}\n\n#[derive(Debug, Serialize)]\nstruct DoctorDump {\n pub atuin: AtuinInfo,\n pub shell: ShellInfo,\n pub system: SystemInfo,\n}\n\nimpl DoctorDump {\n pub async fn new(settings: &Settings) -> Self {\n Self {\n atuin: AtuinInfo::new(settings).await,\n shell: ShellInfo::new(),\n system: SystemInfo::new(),\n }\n }\n}\n\nfn checks(info: &DoctorDump) {\n println!(); // spacing\n //\n let zfs_error = \"[Filesystem] ZFS is known to have some issues with SQLite. Atuin uses SQLite heavily. If you are having poor performance, there are some workarounds here: https://github.com/atuinsh/atuin/issues/952\".bold().red();\n let bash_plugin_error = \"[Shell] If you are using Bash, Atuin requires that either bash-preexec or ble.sh (>= 0.4) be installed. An older ble.sh may not be detected. so ignore this if you have ble.sh >= 0.4 set up! Read more here: https://docs.atuin.sh/guide/installation/#bash\".bold().red();\n let blesh_integration_error = \"[Shell] Atuin and ble.sh seem to be loaded in the session, but the integration does not seem to be working. Please check the setup in .bashrc.\".bold().red();\n\n // ZFS: https://github.com/atuinsh/atuin/issues/952\n if info.system.disks.iter().any(|d| d.filesystem == \"zfs\") {\n println!(\"{zfs_error}\");\n }\n\n info.atuin.setting_paths.verify();\n\n // Shell\n if info.shell.name == \"bash\" {\n if !info\n .shell\n .plugins\n .iter()\n .any(|p| p == \"blesh\" || p == \"bash-preexec\")\n {\n println!(\"{bash_plugin_error}\");\n }\n\n if info.shell.plugins.iter().any(|plugin| plugin == \"atuin\")\n && info.shell.plugins.iter().any(|plugin| plugin == \"blesh\")\n && info.shell.preexec.as_ref().is_some_and(|val| val == \"none\")\n {\n println!(\"{blesh_integration_error}\");\n }\n }\n}\n\npub async fn run(settings: &Settings) -> Result<()> {\n println!(\"{}\", \"Atuin Doctor\".bold());\n println!(\"Checking for diagnostics\");\n let dump = DoctorDump::new(settings).await;\n\n checks(&dump);\n\n let dump = serde_json::to_string_pretty(&dump)?;\n\n println!(\"\\nPlease include the output below with any bug reports or issues\\n\");\n println!(\"{dump}\");\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/dotfiles/alias.rs", "content": "use clap::{Subcommand, ValueEnum};\nuse eyre::{Context, Result, eyre};\n\nuse atuin_client::{encryption, record::sqlite_store::SqliteStore, settings::Settings};\n\nuse atuin_dotfiles::{shell::Alias, store::AliasStore};\n\n#[derive(Clone, Copy, Debug, Default, ValueEnum)]\npub enum SortBy {\n /// Sort by alias name\n #[default]\n Name,\n /// Sort by alias value\n Value,\n}\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Set an alias\n Set { name: String, value: String },\n\n /// Delete an alias\n Delete { name: String },\n\n /// List all aliases\n List {\n /// Sort results by field\n #[arg(long, value_enum, default_value_t = SortBy::Name)]\n sort_by: SortBy,\n\n /// Sort in reverse (descending) order\n #[arg(long, short)]\n reverse: bool,\n\n /// Filter aliases by name (substring match)\n #[arg(long, short)]\n name: Option,\n\n /// Filter aliases by value (substring match)\n #[arg(long, short)]\n value: Option,\n },\n\n /// Delete all aliases\n Clear,\n // There are too many edge cases to parse at the moment. Disable for now.\n // Import,\n}\n\nimpl Cmd {\n async fn set(&self, store: &AliasStore, name: String, value: String) -> Result<()> {\n let illegal_char = regex::Regex::new(\"[ \\t\\n&();<>|\\\\\\\"'`$/]\").unwrap();\n if illegal_char.is_match(name.as_str()) {\n return Err(eyre!(\"Illegal character in alias name\"));\n }\n\n let aliases = store.aliases().await?;\n let found: Vec = aliases.into_iter().filter(|a| a.name == name).collect();\n\n if found.is_empty() {\n println!(\"Aliasing '{name}={value}'.\");\n } else {\n println!(\n \"Overwriting alias '{name}={}' with '{name}={value}'.\",\n found[0].value\n );\n }\n\n store.set(&name, &value).await?;\n\n Ok(())\n }\n\n async fn list(\n &self,\n store: &AliasStore,\n sort_by: SortBy,\n reverse: bool,\n name_filter: Option,\n value_filter: Option,\n ) -> Result<()> {\n let mut aliases = store.aliases().await?;\n\n // Apply filters\n if let Some(ref name_pattern) = name_filter {\n let pattern = name_pattern.to_lowercase();\n aliases.retain(|a| a.name.to_lowercase().contains(&pattern));\n }\n if let Some(ref value_pattern) = value_filter {\n let pattern = value_pattern.to_lowercase();\n aliases.retain(|a| a.value.to_lowercase().contains(&pattern));\n }\n\n // Apply sorting\n match sort_by {\n SortBy::Name => {\n aliases.sort_by(|a, b| a.name.to_lowercase().cmp(&b.name.to_lowercase()));\n }\n SortBy::Value => {\n aliases.sort_by(|a, b| a.value.to_lowercase().cmp(&b.value.to_lowercase()));\n }\n }\n\n // Apply reverse if requested\n if reverse {\n aliases.reverse();\n }\n\n for i in aliases {\n println!(\"{}={}\", i.name, i.value);\n }\n\n Ok(())\n }\n\n async fn clear(&self, store: &AliasStore) -> Result<()> {\n let aliases = store.aliases().await?;\n\n for i in aliases {\n self.delete(store, i.name).await?;\n }\n\n Ok(())\n }\n\n async fn delete(&self, store: &AliasStore, name: String) -> Result<()> {\n let mut aliases = store.aliases().await?.into_iter();\n if let Some(alias) = aliases.find(|alias| alias.name == name) {\n println!(\"Deleting '{name}={}'.\", alias.value);\n store.delete(&name).await?;\n } else {\n eprintln!(\"Cannot delete '{name}': Alias not set.\");\n }\n Ok(())\n }\n\n /*\n async fn import(&self, store: &AliasStore) -> Result<()> {\n let aliases = atuin_dotfiles::shell::import_aliases(store).await?;\n\n for i in aliases {\n println!(\"Importing {}={}\", i.name, i.value);\n }\n\n Ok(())\n }\n */\n\n pub async fn run(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n if !settings.dotfiles.enabled {\n eprintln!(\n \"Dotfiles are not enabled. Add\\n\\n[dotfiles]\\nenabled = true\\n\\nto your configuration file to enable them.\\n\"\n );\n eprintln!(\"The default configuration file is located at ~/.config/atuin/config.toml.\");\n return Ok(());\n }\n\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n let host_id = Settings::host_id().await?;\n\n let alias_store = AliasStore::new(store, host_id, encryption_key);\n\n match self {\n Self::Set { name, value } => self.set(&alias_store, name.clone(), value.clone()).await,\n Self::Delete { name } => self.delete(&alias_store, name.clone()).await,\n Self::List {\n sort_by,\n reverse,\n name,\n value,\n } => {\n self.list(\n &alias_store,\n *sort_by,\n *reverse,\n name.clone(),\n value.clone(),\n )\n .await\n }\n Self::Clear => self.clear(&alias_store).await,\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/client/dotfiles/var.rs", "content": "use clap::{Subcommand, ValueEnum};\nuse eyre::{Context, Result};\n\nuse atuin_client::{encryption, record::sqlite_store::SqliteStore, settings::Settings};\n\nuse atuin_dotfiles::{shell::Var, store::var::VarStore};\n\n#[derive(Clone, Copy, Debug, Default, ValueEnum)]\npub enum SortBy {\n /// Sort by variable name\n #[default]\n Name,\n /// Sort by variable value\n Value,\n}\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Set a variable\n Set {\n name: String,\n value: String,\n\n #[clap(long, short, action)]\n no_export: bool,\n },\n\n /// Delete a variable\n Delete { name: String },\n\n /// List all variables\n List {\n /// Sort results by field\n #[arg(long, value_enum, default_value_t = SortBy::Name)]\n sort_by: SortBy,\n\n /// Sort in reverse (descending) order\n #[arg(long, short)]\n reverse: bool,\n\n /// Filter variables by name (substring match)\n #[arg(long, short)]\n name: Option,\n\n /// Filter variables by value (substring match)\n #[arg(long, short)]\n value: Option,\n\n /// Show only exported variables\n #[arg(long, conflicts_with = \"shell_only\")]\n exports_only: bool,\n\n /// Show only non-exported (shell) variables\n #[arg(long, conflicts_with = \"exports_only\")]\n shell_only: bool,\n },\n}\n\nimpl Cmd {\n async fn set(&self, store: VarStore, name: String, value: String, export: bool) -> Result<()> {\n let vars = store.vars().await?;\n let found: Vec = vars.into_iter().filter(|a| a.name == name).collect();\n let show_export = if export { \"export \" } else { \"\" };\n\n if found.is_empty() {\n println!(\"Setting '{show_export}{name}={value}'.\");\n } else {\n println!(\n \"Overwriting var '{show_export}{name}={}' with '{name}={value}'.\",\n found[0].value\n );\n }\n\n store.set(&name, &value, export).await?;\n\n Ok(())\n }\n\n #[allow(clippy::too_many_arguments)]\n async fn list(\n &self,\n store: VarStore,\n sort_by: SortBy,\n reverse: bool,\n name_filter: Option,\n value_filter: Option,\n exports_only: bool,\n shell_only: bool,\n ) -> Result<()> {\n let mut vars = store.vars().await?;\n\n // Apply export/shell filters\n if exports_only {\n vars.retain(|v| v.export);\n }\n if shell_only {\n vars.retain(|v| !v.export);\n }\n\n // Apply name/value filters\n if let Some(ref name_pattern) = name_filter {\n let pattern = name_pattern.to_lowercase();\n vars.retain(|v| v.name.to_lowercase().contains(&pattern));\n }\n if let Some(ref value_pattern) = value_filter {\n let pattern = value_pattern.to_lowercase();\n vars.retain(|v| v.value.to_lowercase().contains(&pattern));\n }\n\n // Apply sorting\n match sort_by {\n SortBy::Name => {\n vars.sort_by(|a, b| a.name.to_lowercase().cmp(&b.name.to_lowercase()));\n }\n SortBy::Value => {\n vars.sort_by(|a, b| a.value.to_lowercase().cmp(&b.value.to_lowercase()));\n }\n }\n\n // Apply reverse if requested\n if reverse {\n vars.reverse();\n }\n\n for i in vars {\n if i.export {\n println!(\"export {}={}\", i.name, i.value);\n } else {\n println!(\"{}={}\", i.name, i.value);\n }\n }\n\n Ok(())\n }\n\n async fn delete(&self, store: VarStore, name: String) -> Result<()> {\n let mut vars = store.vars().await?.into_iter();\n\n if let Some(var) = vars.find(|var| var.name == name) {\n println!(\"Deleting '{name}={}'.\", var.value);\n store.delete(&name).await?;\n } else {\n eprintln!(\"Cannot delete '{name}': Var not set.\");\n }\n\n Ok(())\n }\n\n pub async fn run(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n if !settings.dotfiles.enabled {\n eprintln!(\n \"Dotfiles are not enabled. Add\\n\\n[dotfiles]\\nenabled = true\\n\\nto your configuration file to enable them.\\n\"\n );\n eprintln!(\"The default configuration file is located at ~/.config/atuin/config.toml.\");\n return Ok(());\n }\n\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n let host_id = Settings::host_id().await?;\n\n let var_store = VarStore::new(store, host_id, encryption_key);\n\n match self {\n Self::Set {\n name,\n value,\n no_export,\n } => {\n self.set(var_store, name.clone(), value.clone(), !no_export)\n .await\n }\n Self::Delete { name } => self.delete(var_store, name.clone()).await,\n Self::List {\n sort_by,\n reverse,\n name,\n value,\n exports_only,\n shell_only,\n } => {\n self.list(\n var_store,\n *sort_by,\n *reverse,\n name.clone(),\n value.clone(),\n *exports_only,\n *shell_only,\n )\n .await\n }\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/client/dotfiles.rs", "content": "use clap::Subcommand;\nuse eyre::Result;\n\nuse atuin_client::{record::sqlite_store::SqliteStore, settings::Settings};\n\nmod alias;\nmod var;\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Manage shell aliases with Atuin\n #[command(subcommand)]\n Alias(alias::Cmd),\n\n /// Manage shell and environment variables with Atuin\n #[command(subcommand)]\n Var(var::Cmd),\n}\n\nimpl Cmd {\n pub async fn run(self, settings: &Settings, store: SqliteStore) -> Result<()> {\n match self {\n Self::Alias(cmd) => cmd.run(settings, store).await,\n Self::Var(cmd) => cmd.run(settings, store).await,\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/client/history.rs", "content": "use std::{\n fmt::{self, Display},\n io::{self, IsTerminal, Write},\n path::PathBuf,\n time::Duration,\n};\n\nuse atuin_common::utils::{self, Escapable as _};\nuse clap::Subcommand;\nuse eyre::{Context, Result};\nuse runtime_format::{FormatKey, FormatKeyError, ParseSegment, ParsedFmt};\n\n#[cfg(feature = \"daemon\")]\nuse atuin_daemon::emit_event;\n\nuse atuin_client::{\n database::{Database, Sqlite, current_context},\n encryption,\n history::{History, store::HistoryStore},\n record::sqlite_store::SqliteStore,\n settings::{\n FilterMode::{Directory, Global, Session},\n Settings, Timezone,\n },\n};\n\n#[cfg(feature = \"sync\")]\nuse atuin_client::{record, sync};\n\nuse log::{debug, warn};\nuse time::{OffsetDateTime, macros::format_description};\n\n#[cfg(feature = \"daemon\")]\nuse super::daemon;\nuse super::search::format_duration_into;\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Begins a new command in the history\n Start {\n /// Collects the command from the `ATUIN_COMMAND_LINE` environment variable,\n /// which does not need escaping and is more compatible between OS and shells\n #[arg(long = \"command-from-env\", hide = true)]\n cmd_env: bool,\n\n /// Author of this command, eg `ellie`, `claude`, or `copilot`\n #[arg(long)]\n author: Option,\n\n /// Optional intent/rationale for running this command\n #[arg(long)]\n intent: Option,\n\n command: Vec,\n },\n\n /// Finishes a new command in the history (adds time, exit code)\n End {\n id: String,\n #[arg(long, short)]\n exit: i64,\n #[arg(long, short)]\n duration: Option,\n },\n\n /// List all items in history\n List {\n #[arg(long, short)]\n cwd: bool,\n\n #[arg(long, short)]\n session: bool,\n\n #[arg(long)]\n human: bool,\n\n /// Show only the text of the command\n #[arg(long)]\n cmd_only: bool,\n\n /// Terminate the output with a null, for better multiline support\n #[arg(long)]\n print0: bool,\n\n #[arg(long, short, default_value = \"true\")]\n // accept no value\n #[arg(num_args(0..=1), default_missing_value(\"true\"))]\n // accept a value\n #[arg(action = clap::ArgAction::Set)]\n reverse: bool,\n\n /// Display the command time in another timezone other than the configured default.\n ///\n /// This option takes one of the following kinds of values:\n /// - the special value \"local\" (or \"l\") which refers to the system time zone\n /// - an offset from UTC (e.g. \"+9\", \"-2:30\")\n #[arg(long, visible_alias = \"tz\")]\n timezone: Option,\n\n /// Available variables: {command}, {directory}, {duration}, {user}, {host}, {author}, {intent}, {exit}, {time}, {session}, and {uuid}\n /// Example: --format \"{time} - [{duration}] - {directory}$\\t{command}\"\n #[arg(long, short)]\n format: Option,\n },\n\n /// Get the last command ran\n Last {\n #[arg(long)]\n human: bool,\n\n /// Show only the text of the command\n #[arg(long)]\n cmd_only: bool,\n\n /// Display the command time in another timezone other than the configured default.\n ///\n /// This option takes one of the following kinds of values:\n /// - the special value \"local\" (or \"l\") which refers to the system time zone\n /// - an offset from UTC (e.g. \"+9\", \"-2:30\")\n #[arg(long, visible_alias = \"tz\")]\n timezone: Option,\n\n /// Available variables: {command}, {directory}, {duration}, {user}, {host}, {author}, {intent}, {time}, {session}, {uuid} and {relativetime}.\n /// Example: --format \"{time} - [{duration}] - {directory}$\\t{command}\"\n #[arg(long, short)]\n format: Option,\n },\n\n InitStore,\n\n /// Delete history entries matching the configured exclusion filters\n Prune {\n /// List matching history lines without performing the actual deletion.\n #[arg(short = 'n', long)]\n dry_run: bool,\n },\n\n /// Delete duplicate history entries (that have the same command, cwd and hostname)\n Dedup {\n /// List matching history lines without performing the actual deletion.\n #[arg(short = 'n', long)]\n dry_run: bool,\n\n /// Only delete results added before this date\n #[arg(long, short)]\n before: String,\n\n /// How many recent duplicates to keep\n #[arg(long)]\n dupkeep: u32,\n },\n}\n\n#[derive(Clone, Copy, Debug)]\npub enum ListMode {\n Human,\n CmdOnly,\n Regular,\n}\n\nimpl ListMode {\n pub const fn from_flags(human: bool, cmd_only: bool) -> Self {\n if human {\n ListMode::Human\n } else if cmd_only {\n ListMode::CmdOnly\n } else {\n ListMode::Regular\n }\n }\n}\n\n#[allow(clippy::cast_sign_loss)]\npub fn print_list(\n h: &[History],\n list_mode: ListMode,\n format: Option<&str>,\n print0: bool,\n reverse: bool,\n tz: Timezone,\n) {\n let w = std::io::stdout();\n let mut w = w.lock();\n\n let fmt_str = match list_mode {\n ListMode::Human => format\n .unwrap_or(\"{time} · {duration}\\t{command}\")\n .replace(\"\\\\t\", \"\\t\"),\n ListMode::Regular => format\n .unwrap_or(\"{time}\\t{command}\\t{duration}\")\n .replace(\"\\\\t\", \"\\t\"),\n // not used\n ListMode::CmdOnly => String::new(),\n };\n\n let parsed_fmt = match list_mode {\n ListMode::Human | ListMode::Regular => parse_fmt(&fmt_str),\n ListMode::CmdOnly => std::iter::once(ParseSegment::Key(\"command\")).collect(),\n };\n\n let iterator = if reverse {\n Box::new(h.iter().rev()) as Box>\n } else {\n Box::new(h.iter()) as Box>\n };\n\n let entry_terminator = if print0 { \"\\0\" } else { \"\\n\" };\n let flush_each_line = print0;\n\n for history in iterator {\n let fh = FmtHistory {\n history,\n cmd_format: CmdFormat::for_output(&w),\n tz: &tz,\n };\n let args = parsed_fmt.with_args(&fh);\n\n // Check for formatting errors before attempting to write\n if let Err(err) = args.status() {\n eprintln!(\"ERROR: history output failed with: {err}\");\n std::process::exit(1);\n }\n\n let write_result = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {\n write!(w, \"{args}{entry_terminator}\")\n }));\n\n match write_result {\n Ok(Ok(())) => {\n // Write succeeded\n }\n Ok(Err(err)) => {\n if err.kind() != io::ErrorKind::BrokenPipe {\n eprintln!(\"ERROR: Failed to write history output: {err}\");\n std::process::exit(1);\n }\n }\n Err(_) => {\n eprintln!(\"ERROR: Format string caused a formatting error.\");\n eprintln!(\n \"This may be due to an unsupported format string containing special characters.\"\n );\n eprintln!(\n \"Please check your format string syntax and ensure literal braces are properly escaped.\"\n );\n std::process::exit(1);\n }\n }\n if flush_each_line {\n check_for_write_errors(w.flush());\n }\n }\n\n if !flush_each_line {\n check_for_write_errors(w.flush());\n }\n}\n\nfn check_for_write_errors(write: Result<(), io::Error>) {\n if let Err(err) = write {\n // Ignore broken pipe (issue #626)\n if err.kind() != io::ErrorKind::BrokenPipe {\n eprintln!(\"ERROR: History output failed with the following error: {err}\");\n std::process::exit(1);\n }\n }\n}\n\n/// Type wrapper around `History` with formatting settings.\n#[derive(Clone, Copy, Debug)]\nstruct FmtHistory<'a> {\n history: &'a History,\n cmd_format: CmdFormat,\n tz: &'a Timezone,\n}\n\n#[derive(Clone, Copy, Debug)]\nenum CmdFormat {\n Literal,\n Escaped,\n}\nimpl CmdFormat {\n fn for_output(out: &O) -> Self {\n if out.is_terminal() {\n Self::Escaped\n } else {\n Self::Literal\n }\n }\n}\n\nstatic TIME_FMT: &[time::format_description::FormatItem<'static>] =\n format_description!(\"[year]-[month]-[day] [hour repr:24]:[minute]:[second]\");\n\n/// defines how to format the history\nimpl FormatKey for FmtHistory<'_> {\n #[allow(clippy::cast_sign_loss)]\n fn fmt(&self, key: &str, f: &mut fmt::Formatter<'_>) -> Result<(), FormatKeyError> {\n match key {\n \"command\" => match self.cmd_format {\n CmdFormat::Literal => f.write_str(self.history.command.trim()),\n CmdFormat::Escaped => f.write_str(&self.history.command.trim().escape_control()),\n }?,\n \"directory\" => f.write_str(self.history.cwd.trim())?,\n \"exit\" => f.write_str(&self.history.exit.to_string())?,\n \"duration\" => {\n let dur = Duration::from_nanos(std::cmp::max(self.history.duration, 0) as u64);\n format_duration_into(dur, f)?;\n }\n \"time\" => {\n self.history\n .timestamp\n .to_offset(self.tz.0)\n .format(TIME_FMT)\n .map_err(|_| fmt::Error)?\n .fmt(f)?;\n }\n \"relativetime\" => {\n let since = OffsetDateTime::now_utc() - self.history.timestamp;\n let d = Duration::try_from(since).unwrap_or_default();\n format_duration_into(d, f)?;\n }\n \"host\" => f.write_str(\n self.history\n .hostname\n .split_once(':')\n .map_or(&self.history.hostname, |(host, _)| host),\n )?,\n \"author\" => f.write_str(&self.history.author)?,\n \"intent\" => f.write_str(self.history.intent.as_deref().unwrap_or_default())?,\n \"user\" => f.write_str(\n self.history\n .hostname\n .split_once(':')\n .map_or(\"\", |(_, user)| user),\n )?,\n \"session\" => f.write_str(&self.history.session)?,\n \"uuid\" => f.write_str(&self.history.id.0)?,\n _ => return Err(FormatKeyError::UnknownKey),\n }\n Ok(())\n }\n}\n\nfn parse_fmt(format: &str) -> ParsedFmt<'_> {\n match ParsedFmt::new(format) {\n Ok(fmt) => fmt,\n Err(err) => {\n eprintln!(\"ERROR: History formatting failed with the following error: {err}\");\n\n if format.contains('\"') && (format.contains(\":{\") || format.contains(\",{\")) {\n eprintln!(\"It looks like you're trying to create JSON output.\");\n eprintln!(\"For JSON, you need to escape literal braces by doubling them:\");\n eprintln!(\"Example: '{{\\\"command\\\":\\\"{{command}}\\\",\\\"time\\\":\\\"{{time}}\\\"}}'\");\n } else {\n eprintln!(\n \"If your formatting string contains literal curly braces, you need to escape them by doubling:\"\n );\n eprintln!(\"Use {{{{ for literal {{ and }}}} for literal }}\");\n }\n std::process::exit(1)\n }\n }\n}\n\nimpl Cmd {\n fn apply_start_metadata(history: &mut History, author: Option<&str>, intent: Option<&str>) {\n if let Some(author) = author.map(str::trim).filter(|author| !author.is_empty()) {\n author.clone_into(&mut history.author);\n }\n\n if let Some(intent) = intent.map(str::trim).filter(|intent| !intent.is_empty()) {\n history.intent = Some(intent.to_owned());\n } else if intent.is_some() {\n history.intent = None;\n }\n }\n\n #[allow(clippy::too_many_lines, clippy::cast_possible_truncation)]\n async fn handle_start(\n db: &impl Database,\n settings: &Settings,\n command: &str,\n author: Option<&str>,\n intent: Option<&str>,\n ) -> Result<()> {\n // It's better for atuin to silently fail here and attempt to\n // store whatever is ran, than to throw an error to the terminal\n let cwd = utils::get_current_dir();\n\n let mut h: History = History::capture()\n .timestamp(OffsetDateTime::now_utc())\n .command(command)\n .cwd(cwd)\n .build()\n .into();\n Self::apply_start_metadata(&mut h, author, intent);\n\n if !h.should_save(settings) {\n return Ok(());\n }\n\n // print the ID\n // we use this as the key for calling end\n println!(\"{}\", h.id);\n\n // Silently ignore database errors to avoid breaking the shell\n // This is important when disk is full or database is locked\n if let Err(e) = db.save(&h).await {\n debug!(\"failed to save history: {e}\");\n }\n\n Ok(())\n }\n\n #[cfg(feature = \"daemon\")]\n async fn handle_daemon_start(\n settings: &Settings,\n command: &str,\n author: Option<&str>,\n intent: Option<&str>,\n ) -> Result<()> {\n // It's better for atuin to silently fail here and attempt to\n // store whatever is ran, than to throw an error to the terminal\n let cwd = utils::get_current_dir();\n\n let mut h: History = History::capture()\n .timestamp(OffsetDateTime::now_utc())\n .command(command)\n .cwd(cwd)\n .build()\n .into();\n Self::apply_start_metadata(&mut h, author, intent);\n\n if !h.should_save(settings) {\n return Ok(());\n }\n\n // Attempt to start history via daemon, but silently ignore errors\n // to avoid breaking the shell when the daemon is unavailable or disk is full\n let resp = match daemon::start_history(settings, h.clone()).await {\n Ok(id) => id,\n Err(e) => {\n debug!(\"failed to start history via daemon: {e}\");\n h.id.0.clone()\n }\n };\n\n // print the ID\n // we use this as the key for calling end\n println!(\"{resp}\");\n\n Ok(())\n }\n\n #[allow(unused_variables)]\n async fn handle_end(\n db: &impl Database,\n store: SqliteStore,\n history_store: HistoryStore,\n settings: &Settings,\n id: &str,\n exit: i64,\n duration: Option,\n ) -> Result<()> {\n if id.trim() == \"\" {\n return Ok(());\n }\n\n let Some(mut h) = db.load(id).await? else {\n warn!(\"history entry is missing\");\n return Ok(());\n };\n\n if h.duration > 0 {\n debug!(\"cannot end history - already has duration\");\n\n // returning OK as this can occur if someone Ctrl-c a prompt\n return Ok(());\n }\n\n if !settings.store_failed && exit > 0 {\n debug!(\"history has non-zero exit code, and store_failed is false\");\n\n // the history has already been inserted half complete. remove it\n db.delete(h).await?;\n\n return Ok(());\n }\n\n h.exit = exit;\n h.duration = match duration {\n Some(value) => i64::try_from(value).context(\"command took over 292 years\")?,\n None => i64::try_from((OffsetDateTime::now_utc() - h.timestamp).whole_nanoseconds())\n .context(\"command took over 292 years\")?,\n };\n\n db.update(&h).await?;\n history_store.push(h).await?;\n\n if settings.should_sync().await? {\n #[cfg(feature = \"sync\")]\n {\n if settings.sync.records {\n let (_, downloaded) = record::sync::sync(settings, &store).await?;\n Settings::save_sync_time().await?;\n\n crate::sync::build(settings, &store, db, Some(&downloaded)).await?;\n } else {\n debug!(\"running periodic background sync\");\n sync::sync(settings, false, db).await?;\n }\n }\n #[cfg(not(feature = \"sync\"))]\n debug!(\"not compiled with sync support\");\n } else {\n debug!(\"sync disabled! not syncing\");\n }\n\n Ok(())\n }\n\n #[cfg(feature = \"daemon\")]\n async fn handle_daemon_end(\n settings: &Settings,\n id: &str,\n exit: i64,\n duration: Option,\n ) -> Result<()> {\n daemon::end_history(settings, id.to_string(), duration.unwrap_or(0), exit).await?;\n\n Ok(())\n }\n\n #[allow(clippy::too_many_arguments)]\n #[allow(clippy::fn_params_excessive_bools)]\n async fn handle_list(\n db: &impl Database,\n settings: &Settings,\n context: atuin_client::database::Context,\n session: bool,\n cwd: bool,\n mode: ListMode,\n format: Option,\n include_deleted: bool,\n print0: bool,\n reverse: bool,\n tz: Timezone,\n ) -> Result<()> {\n let filters = match (session, cwd) {\n (true, true) => [Session, Directory],\n (true, false) => [Session, Global],\n (false, true) => [Global, Directory],\n (false, false) => [\n settings.default_filter_mode(context.git_root.is_some()),\n Global,\n ],\n };\n\n let history = db\n .list(&filters, &context, None, false, include_deleted)\n .await?;\n\n print_list(\n &history,\n mode,\n match format {\n None => Some(settings.history_format.as_str()),\n _ => format.as_deref(),\n },\n print0,\n reverse,\n tz,\n );\n\n Ok(())\n }\n\n async fn handle_prune(\n db: &impl Database,\n settings: &Settings,\n store: SqliteStore,\n context: atuin_client::database::Context,\n dry_run: bool,\n ) -> Result<()> {\n // Grab all executed commands and filter them using History::should_save.\n // We could iterate or paginate here if memory usage becomes an issue.\n let matches: Vec = db\n .list(&[Global], &context, None, false, false)\n .await?\n .into_iter()\n .filter(|h| !h.should_save(settings))\n .collect();\n\n match matches.len() {\n 0 => {\n println!(\"No entries to prune.\");\n return Ok(());\n }\n 1 => println!(\"Found 1 entry to prune.\"),\n n => println!(\"Found {n} entries to prune.\"),\n }\n\n if dry_run {\n print_list(\n &matches,\n ListMode::Human,\n Some(settings.history_format.as_str()),\n false,\n false,\n settings.timezone,\n );\n } else {\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n let host_id = Settings::host_id().await?;\n let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);\n\n for entry in matches {\n eprintln!(\"deleting {}\", entry.id);\n if settings.sync.records {\n let (id, _) = history_store.delete(entry.id.clone()).await?;\n history_store.incremental_build(db, &[id]).await?;\n } else {\n db.delete(entry.clone()).await?;\n }\n }\n\n #[cfg(feature = \"daemon\")]\n let _ = emit_event(atuin_daemon::DaemonEvent::HistoryPruned).await;\n }\n Ok(())\n }\n\n async fn handle_dedup(\n db: &impl Database,\n settings: &Settings,\n store: SqliteStore,\n before: i64,\n dupkeep: u32,\n dry_run: bool,\n ) -> Result<()> {\n if dupkeep == 0 {\n eprintln!(\n \"\\\"--dupkeep 0\\\" would keep 0 copies of duplicate commands and thus delete all of them! Use \\\"atuin search --delete ...\\\" if you really want that.\"\n );\n std::process::exit(1);\n }\n\n let matches: Vec = db.get_dups(before, dupkeep).await?;\n\n match matches.len() {\n 0 => {\n println!(\"No duplicates to delete.\");\n return Ok(());\n }\n 1 => println!(\"Found 1 duplicate to delete.\"),\n n => println!(\"Found {n} duplicates to delete.\"),\n }\n\n if dry_run {\n print_list(\n &matches,\n ListMode::Human,\n Some(settings.history_format.as_str()),\n false,\n false,\n settings.timezone,\n );\n } else {\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n let host_id = Settings::host_id().await?;\n let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);\n\n #[cfg(feature = \"daemon\")]\n let ids = matches.iter().map(|h| h.id.clone()).collect::>();\n\n for entry in matches {\n eprintln!(\"deleting {}\", entry.id);\n if settings.sync.records {\n let (id, _) = history_store.delete(entry.id).await?;\n history_store.incremental_build(db, &[id]).await?;\n } else {\n db.delete(entry).await?;\n }\n }\n\n #[cfg(feature = \"daemon\")]\n let _ = emit_event(atuin_daemon::DaemonEvent::HistoryDeleted { ids }).await;\n }\n Ok(())\n }\n\n pub async fn run(self, settings: &Settings) -> Result<()> {\n let context = current_context().await?;\n\n #[cfg(feature = \"daemon\")]\n // Skip initializing any databases for start/end, if the daemon is enabled\n if settings.daemon.enabled {\n match self {\n Self::Start { .. } => {\n let command = self.get_start_command().unwrap_or_default();\n let (author, intent) = self.get_start_metadata().unwrap_or_default();\n return Self::handle_daemon_start(settings, &command, author, intent).await;\n }\n\n Self::End { id, exit, duration } => {\n return Self::handle_daemon_end(settings, &id, exit, duration).await;\n }\n\n _ => {}\n }\n }\n\n let db_path = PathBuf::from(settings.db_path.as_str());\n let record_store_path = PathBuf::from(settings.record_store_path.as_str());\n\n let db = Sqlite::new(db_path, settings.local_timeout).await?;\n let store = SqliteStore::new(record_store_path, settings.local_timeout).await?;\n\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n\n let host_id = Settings::host_id().await?;\n let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);\n\n match self {\n Self::Start { .. } => {\n let command = self.get_start_command().unwrap_or_default();\n let (author, intent) = self.get_start_metadata().unwrap_or_default();\n Self::handle_start(&db, settings, &command, author, intent).await\n }\n Self::End { id, exit, duration } => {\n Self::handle_end(&db, store, history_store, settings, &id, exit, duration).await\n }\n Self::List {\n session,\n cwd,\n human,\n cmd_only,\n print0,\n reverse,\n timezone,\n format,\n } => {\n let mode = ListMode::from_flags(human, cmd_only);\n let tz = timezone.unwrap_or(settings.timezone);\n Self::handle_list(\n &db, settings, context, session, cwd, mode, format, false, print0, reverse, tz,\n )\n .await\n }\n\n Self::Last {\n human,\n cmd_only,\n timezone,\n format,\n } => {\n let last = db.last().await?;\n let last = last.as_slice();\n let tz = timezone.unwrap_or(settings.timezone);\n print_list(\n last,\n ListMode::from_flags(human, cmd_only),\n match format {\n None => Some(settings.history_format.as_str()),\n _ => format.as_deref(),\n },\n false,\n true,\n tz,\n );\n\n Ok(())\n }\n\n Self::InitStore => history_store.init_store(&db).await,\n\n Self::Prune { dry_run } => {\n Self::handle_prune(&db, settings, store, context, dry_run).await\n }\n\n Self::Dedup {\n dry_run,\n before,\n dupkeep,\n } => {\n let before = i64::try_from(\n interim::parse_date_string(\n before.as_str(),\n OffsetDateTime::now_utc(),\n interim::Dialect::Uk,\n )?\n .unix_timestamp_nanos(),\n )?;\n Self::handle_dedup(&db, settings, store, before, dupkeep, dry_run).await\n }\n }\n }\n\n /// Returns the command line to use for the `Start` variant.\n /// Returns `None` for any other variant.\n fn get_start_command(&self) -> Option {\n match self {\n Self::Start { cmd_env: true, .. } => {\n Some(std::env::var(\"ATUIN_COMMAND_LINE\").unwrap_or_default())\n }\n Self::Start { command, .. } => Some(command.join(\" \")),\n _ => None,\n }\n }\n\n /// Returns `(author, intent)` for the `Start` variant.\n /// Returns `None` for any other variant.\n fn get_start_metadata(&self) -> Option<(Option<&str>, Option<&str>)> {\n match self {\n Self::Start { author, intent, .. } => Some((author.as_deref(), intent.as_deref())),\n _ => None,\n }\n }\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_format_string_no_panic() {\n // Don't panic but provide helpful output (issue #2776)\n let malformed_json = r#\"{\"command\":\"{command}\",\"key\":\"value\"}\"#;\n\n let result = std::panic::catch_unwind(|| parse_fmt(malformed_json));\n\n assert!(result.is_ok());\n }\n\n #[test]\n fn test_valid_formats_still_work() {\n assert!(std::panic::catch_unwind(|| parse_fmt(\"{command}\")).is_ok());\n assert!(std::panic::catch_unwind(|| parse_fmt(\"{time} - {command}\")).is_ok());\n }\n}\n" }, { "path": "crates/atuin/src/command/client/import.rs", "content": "use std::env;\n\nuse async_trait::async_trait;\nuse clap::Parser;\nuse eyre::Result;\nuse indicatif::ProgressBar;\n\nuse atuin_client::{\n database::Database,\n history::History,\n import::{\n Importer, Loader, bash::Bash, fish::Fish, nu::Nu, nu_histdb::NuHistDb,\n powershell::PowerShell, replxx::Replxx, resh::Resh, xonsh::Xonsh,\n xonsh_sqlite::XonshSqlite, zsh::Zsh, zsh_histdb::ZshHistDb,\n },\n};\n\n#[derive(Parser, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Import history for the current shell\n Auto,\n\n /// Import history from the zsh history file\n Zsh,\n /// Import history from the zsh history file\n ZshHistDb,\n /// Import history from the bash history file\n Bash,\n /// Import history from the replxx history file\n Replxx,\n /// Import history from the resh history file\n Resh,\n /// Import history from the fish history file\n Fish,\n /// Import history from the nu history file\n Nu,\n /// Import history from the nu history file\n NuHistDb,\n /// Import history from xonsh json files\n Xonsh,\n /// Import history from xonsh sqlite db\n XonshSqlite,\n /// Import history from the powershell history file\n Powershell,\n}\n\nconst BATCH_SIZE: usize = 100;\n\nimpl Cmd {\n #[allow(clippy::cognitive_complexity)]\n pub async fn run(&self, db: &DB) -> Result<()> {\n println!(\" Atuin \");\n println!(\"======================\");\n println!(\" \\u{1f30d} \");\n println!(\" \\u{1f418}\\u{1f418}\\u{1f418}\\u{1f418} \");\n println!(\" \\u{1f422} \");\n println!(\"======================\");\n println!(\"Importing history...\");\n\n match self {\n Self::Auto => {\n if cfg!(windows) {\n return if env::var(\"PSModulePath\").is_ok() {\n println!(\"Detected PowerShell\");\n import::(db).await\n } else {\n println!(\"Could not detect the current shell.\");\n println!(\"Please run atuin import .\");\n println!(\"To view a list of shells, run atuin import.\");\n Ok(())\n };\n }\n\n // $XONSH_HISTORY_BACKEND isn't always set, but $XONSH_HISTORY_FILE is\n let xonsh_histfile =\n env::var(\"XONSH_HISTORY_FILE\").unwrap_or_else(|_| String::new());\n let shell = env::var(\"SHELL\").unwrap_or_else(|_| String::from(\"NO_SHELL\"));\n\n if xonsh_histfile.to_lowercase().ends_with(\".json\") {\n println!(\"Detected Xonsh\",);\n import::(db).await\n } else if xonsh_histfile.to_lowercase().ends_with(\".sqlite\") {\n println!(\"Detected Xonsh (SQLite backend)\");\n import::(db).await\n } else if shell.ends_with(\"/zsh\") {\n if ZshHistDb::histpath().is_ok() {\n println!(\n \"Detected Zsh-HistDb, using :{}\",\n ZshHistDb::histpath().unwrap().to_str().unwrap()\n );\n import::(db).await\n } else {\n println!(\"Detected ZSH\");\n import::(db).await\n }\n } else if shell.ends_with(\"/fish\") {\n println!(\"Detected Fish\");\n import::(db).await\n } else if shell.ends_with(\"/bash\") {\n println!(\"Detected Bash\");\n import::(db).await\n } else if shell.ends_with(\"/nu\") {\n if NuHistDb::histpath().is_ok() {\n println!(\n \"Detected Nu-HistDb, using :{}\",\n NuHistDb::histpath().unwrap().to_str().unwrap()\n );\n import::(db).await\n } else {\n println!(\"Detected Nushell\");\n import::(db).await\n }\n } else if shell.ends_with(\"/pwsh\") {\n println!(\"Detected PowerShell\");\n import::(db).await\n } else {\n println!(\"cannot import {shell} history\");\n Ok(())\n }\n }\n\n Self::Zsh => import::(db).await,\n Self::ZshHistDb => import::(db).await,\n Self::Bash => import::(db).await,\n Self::Replxx => import::(db).await,\n Self::Resh => import::(db).await,\n Self::Fish => import::(db).await,\n Self::Nu => import::(db).await,\n Self::NuHistDb => import::(db).await,\n Self::Xonsh => import::(db).await,\n Self::XonshSqlite => import::(db).await,\n Self::Powershell => import::(db).await,\n }\n }\n}\n\npub struct HistoryImporter<'db, DB: Database> {\n pb: ProgressBar,\n buf: Vec,\n db: &'db DB,\n}\n\nimpl<'db, DB: Database> HistoryImporter<'db, DB> {\n fn new(db: &'db DB, len: usize) -> Self {\n Self {\n pb: ProgressBar::new(len as u64),\n buf: Vec::with_capacity(BATCH_SIZE),\n db,\n }\n }\n\n async fn flush(self) -> Result<()> {\n if !self.buf.is_empty() {\n self.db.save_bulk(&self.buf).await?;\n }\n self.pb.finish();\n Ok(())\n }\n}\n\n#[async_trait]\nimpl Loader for HistoryImporter<'_, DB> {\n async fn push(&mut self, hist: History) -> Result<()> {\n self.pb.inc(1);\n self.buf.push(hist);\n if self.buf.len() == self.buf.capacity() {\n self.db.save_bulk(&self.buf).await?;\n self.buf.clear();\n }\n Ok(())\n }\n}\n\nasync fn import(db: &DB) -> Result<()> {\n println!(\"Importing history from {}\", I::NAME);\n\n let mut importer = I::new().await?;\n let len = importer.entries().await.unwrap();\n let mut loader = HistoryImporter::new(db, len);\n importer.load(&mut loader).await?;\n loader.flush().await?;\n\n println!(\"Import complete!\");\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/info.rs", "content": "use atuin_client::settings::Settings;\r\n\r\nuse crate::{SHA, VERSION};\r\n\r\npub fn run(settings: &Settings) {\r\n let config = atuin_common::utils::config_dir();\r\n let mut config_file = config.clone();\r\n config_file.push(\"config.toml\");\r\n let mut sever_config = config;\r\n sever_config.push(\"server.toml\");\r\n\r\n let config_paths = format!(\r\n \"Config files:\\nclient config: {:?}\\nserver config: {:?}\\nclient db path: {:?}\\nkey path: {:?}\\nmeta db path: {:?}\",\r\n config_file.to_string_lossy(),\r\n sever_config.to_string_lossy(),\r\n settings.db_path,\r\n settings.key_path,\r\n settings.meta.db_path\r\n );\r\n\r\n let env_vars = format!(\r\n \"Env Vars:\\nATUIN_CONFIG_DIR = {:?}\",\r\n std::env::var(\"ATUIN_CONFIG_DIR\").unwrap_or_else(|_| \"None\".into())\r\n );\r\n\r\n let general_info = format!(\"Version info:\\nversion: {VERSION}\\ncommit: {SHA}\");\r\n\r\n let print_out = format!(\"{config_paths}\\n\\n{env_vars}\\n\\n{general_info}\");\r\n\r\n println!(\"{print_out}\");\r\n}\r\n" }, { "path": "crates/atuin/src/command/client/init/bash.rs", "content": "use atuin_client::settings::Tmux;\nuse atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse eyre::Result;\n\nfn print_tmux_config(tmux: &Tmux) {\n if tmux.enabled {\n println!(\"export ATUIN_TMUX_POPUP_WIDTH='{}'\", tmux.width);\n println!(\"export ATUIN_TMUX_POPUP_HEIGHT='{}'\", tmux.height);\n } else {\n println!(\"export ATUIN_TMUX_POPUP=false\");\n }\n}\n\npub fn init_static(disable_up_arrow: bool, disable_ctrl_r: bool, disable_ai: bool, tmux: &Tmux) {\n let base = include_str!(\"../../../shell/atuin.bash\");\n\n let (bind_ctrl_r, bind_up_arrow) = if std::env::var(\"ATUIN_NOBIND\").is_ok() {\n (false, false)\n } else {\n (!disable_ctrl_r, !disable_up_arrow)\n };\n\n print_tmux_config(tmux);\n println!(\"__atuin_bind_ctrl_r={bind_ctrl_r}\");\n println!(\"__atuin_bind_up_arrow={bind_up_arrow}\");\n println!(\"{base}\");\n\n #[cfg(feature = \"ai\")]\n if !disable_ai {\n let bind_ai = atuin_ai::commands::init::generate_bash_integration();\n println!(\"{bind_ai}\");\n }\n}\n\npub async fn init(\n aliases: AliasStore,\n vars: VarStore,\n disable_up_arrow: bool,\n disable_ctrl_r: bool,\n disable_ai: bool,\n tmux: &Tmux,\n) -> Result<()> {\n init_static(disable_up_arrow, disable_ctrl_r, disable_ai, tmux);\n\n let aliases = atuin_dotfiles::shell::bash::alias_config(&aliases).await;\n let vars = atuin_dotfiles::shell::bash::var_config(&vars).await;\n\n println!(\"{aliases}\");\n println!(\"{vars}\");\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/init/fish.rs", "content": "use atuin_client::settings::Tmux;\nuse atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse eyre::Result;\n\nfn print_tmux_config(tmux: &Tmux) {\n if tmux.enabled {\n println!(\"set -gx ATUIN_TMUX_POPUP_WIDTH '{}'\", tmux.width);\n println!(\"set -gx ATUIN_TMUX_POPUP_HEIGHT '{}'\", tmux.height);\n } else {\n println!(\"set -gx ATUIN_TMUX_POPUP false\");\n }\n}\n\nfn print_bindings(\n indent: &str,\n disable_up_arrow: bool,\n disable_ctrl_r: bool,\n bind_ctrl_r: &str,\n bind_up_arrow: &str,\n bind_ctrl_r_ins: &str,\n bind_up_arrow_ins: &str,\n) {\n if !disable_ctrl_r {\n println!(\"{indent}{bind_ctrl_r}\");\n }\n if !disable_up_arrow {\n println!(\"{indent}{bind_up_arrow}\");\n }\n\n println!(\"{indent}if bind -M insert >/dev/null 2>&1\");\n if !disable_ctrl_r {\n println!(\"{indent}{indent}{bind_ctrl_r_ins}\");\n }\n if !disable_up_arrow {\n println!(\"{indent}{indent}{bind_up_arrow_ins}\");\n }\n println!(\"{indent}end\");\n}\n\npub fn init_static(disable_up_arrow: bool, disable_ctrl_r: bool, disable_ai: bool, tmux: &Tmux) {\n let indent = \" \".repeat(4);\n\n let base = include_str!(\"../../../shell/atuin.fish\");\n\n print_tmux_config(tmux);\n println!(\"{base}\");\n\n if std::env::var(\"ATUIN_NOBIND\").is_err() {\n println!(\"if string match -q '4.*' $version\");\n\n // In fish 4.0 and above the option bind -k doesn't exist anymore,\n // instead we can use key names and modifiers directly.\n print_bindings(\n &indent,\n disable_up_arrow,\n disable_ctrl_r,\n \"bind ctrl-r _atuin_search\",\n \"bind up _atuin_bind_up\",\n \"bind -M insert ctrl-r _atuin_search\",\n \"bind -M insert up _atuin_bind_up\",\n );\n\n println!(\"else\");\n\n // We keep these for compatibility with fish 3.x\n print_bindings(\n &indent,\n disable_up_arrow,\n disable_ctrl_r,\n r\"bind \\cr _atuin_search\",\n &[\n r\"bind -k up _atuin_bind_up\",\n r\"bind \\eOA _atuin_bind_up\",\n r\"bind \\e\\[A _atuin_bind_up\",\n ]\n .join(\"; \"),\n r\"bind -M insert \\cr _atuin_search\",\n &[\n r\"bind -M insert -k up _atuin_bind_up\",\n r\"bind -M insert \\eOA _atuin_bind_up\",\n r\"bind -M insert \\e\\[A _atuin_bind_up\",\n ]\n .join(\"; \"),\n );\n\n println!(\"end\");\n\n #[cfg(feature = \"ai\")]\n if !disable_ai {\n let bind_ai = atuin_ai::commands::init::generate_fish_integration();\n println!(\"{bind_ai}\");\n }\n }\n}\n\npub async fn init(\n aliases: AliasStore,\n vars: VarStore,\n disable_up_arrow: bool,\n disable_ctrl_r: bool,\n disable_ai: bool,\n tmux: &Tmux,\n) -> Result<()> {\n init_static(disable_up_arrow, disable_ctrl_r, disable_ai, tmux);\n\n let aliases = atuin_dotfiles::shell::fish::alias_config(&aliases).await;\n let vars = atuin_dotfiles::shell::fish::var_config(&vars).await;\n\n println!(\"{aliases}\");\n println!(\"{vars}\");\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/init/powershell.rs", "content": "use atuin_client::settings::Tmux;\nuse atuin_dotfiles::store::{AliasStore, var::VarStore};\n\npub fn init_static(disable_up_arrow: bool, disable_ctrl_r: bool, _tmux: &Tmux) {\n let base = include_str!(\"../../../shell/atuin.ps1\");\n\n let (bind_ctrl_r, bind_up_arrow) = if std::env::var(\"ATUIN_NOBIND\").is_ok() {\n (false, false)\n } else {\n (!disable_ctrl_r, !disable_up_arrow)\n };\n\n // TODO: tmux popup for Powershell\n println!(\"{base}\");\n println!(\n \"Enable-AtuinSearchKeys -CtrlR {} -UpArrow {}\",\n ps_bool(bind_ctrl_r),\n ps_bool(bind_up_arrow)\n );\n}\n\npub async fn init(\n aliases: AliasStore,\n vars: VarStore,\n disable_up_arrow: bool,\n disable_ctrl_r: bool,\n tmux: &Tmux,\n) -> eyre::Result<()> {\n init_static(disable_up_arrow, disable_ctrl_r, tmux);\n\n let aliases = atuin_dotfiles::shell::powershell::alias_config(&aliases).await;\n let vars = atuin_dotfiles::shell::powershell::var_config(&vars).await;\n\n println!(\"{aliases}\");\n println!(\"{vars}\");\n\n Ok(())\n}\n\nfn ps_bool(value: bool) -> &'static str {\n if value { \"$true\" } else { \"$false\" }\n}\n" }, { "path": "crates/atuin/src/command/client/init/xonsh.rs", "content": "use atuin_client::settings::Tmux;\nuse atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse eyre::Result;\n\npub fn init_static(disable_up_arrow: bool, disable_ctrl_r: bool, _tmux: &Tmux) {\n let base = include_str!(\"../../../shell/atuin.xsh\");\n\n let (bind_ctrl_r, bind_up_arrow) = if std::env::var(\"ATUIN_NOBIND\").is_ok() {\n (false, false)\n } else {\n (!disable_ctrl_r, !disable_up_arrow)\n };\n\n // TODO: tmux popup for xonsh\n println!(\n \"_ATUIN_BIND_CTRL_R={}\",\n if bind_ctrl_r { \"True\" } else { \"False\" }\n );\n println!(\n \"_ATUIN_BIND_UP_ARROW={}\",\n if bind_up_arrow { \"True\" } else { \"False\" }\n );\n println!(\"{base}\");\n}\n\npub async fn init(\n aliases: AliasStore,\n vars: VarStore,\n disable_up_arrow: bool,\n disable_ctrl_r: bool,\n tmux: &Tmux,\n) -> Result<()> {\n init_static(disable_up_arrow, disable_ctrl_r, tmux);\n\n let aliases = atuin_dotfiles::shell::xonsh::alias_config(&aliases).await;\n let vars = atuin_dotfiles::shell::xonsh::var_config(&vars).await;\n\n println!(\"{aliases}\");\n println!(\"{vars}\");\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/init/zsh.rs", "content": "use atuin_client::settings::Tmux;\nuse atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse eyre::Result;\n\nfn print_tmux_config(tmux: &Tmux) {\n if tmux.enabled {\n println!(\"export ATUIN_TMUX_POPUP_WIDTH='{}'\", tmux.width);\n println!(\"export ATUIN_TMUX_POPUP_HEIGHT='{}'\", tmux.height);\n } else {\n println!(\"export ATUIN_TMUX_POPUP=false\");\n }\n}\n\npub fn init_static(disable_up_arrow: bool, disable_ctrl_r: bool, disable_ai: bool, tmux: &Tmux) {\n let base = include_str!(\"../../../shell/atuin.zsh\");\n\n print_tmux_config(tmux);\n println!(\"{base}\");\n\n if std::env::var(\"ATUIN_NOBIND\").is_err() {\n const BIND_CTRL_R: &str = r\"bindkey -M emacs '^r' atuin-search\nbindkey -M viins '^r' atuin-search-viins\nbindkey -M vicmd '/' atuin-search\";\n\n const BIND_UP_ARROW: &str = r\"bindkey -M emacs '^[[A' atuin-up-search\nbindkey -M vicmd '^[[A' atuin-up-search-vicmd\nbindkey -M viins '^[[A' atuin-up-search-viins\nbindkey -M emacs '^[OA' atuin-up-search\nbindkey -M vicmd '^[OA' atuin-up-search-vicmd\nbindkey -M viins '^[OA' atuin-up-search-viins\nbindkey -M vicmd 'k' atuin-up-search-vicmd\";\n\n if !disable_ctrl_r {\n println!(\"{BIND_CTRL_R}\");\n }\n if !disable_up_arrow {\n println!(\"{BIND_UP_ARROW}\");\n }\n\n #[cfg(feature = \"ai\")]\n if !disable_ai {\n let bind_ai = atuin_ai::commands::init::generate_zsh_integration();\n\n println!(\"{bind_ai}\");\n }\n }\n}\n\npub async fn init(\n aliases: AliasStore,\n vars: VarStore,\n disable_up_arrow: bool,\n disable_ctrl_r: bool,\n disable_ai: bool,\n tmux: &Tmux,\n) -> Result<()> {\n init_static(disable_up_arrow, disable_ctrl_r, disable_ai, tmux);\n\n let aliases = atuin_dotfiles::shell::zsh::alias_config(&aliases).await;\n let vars = atuin_dotfiles::shell::zsh::var_config(&vars).await;\n\n println!(\"{aliases}\");\n println!(\"{vars}\");\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/init.rs", "content": "use std::path::PathBuf;\n\nuse atuin_client::{\n encryption,\n record::sqlite_store::SqliteStore,\n settings::{Settings, Tmux},\n};\nuse atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse clap::{Parser, ValueEnum};\nuse eyre::{Result, WrapErr};\n\nmod bash;\nmod fish;\nmod powershell;\nmod xonsh;\nmod zsh;\n\n#[derive(Parser, Debug)]\npub struct Cmd {\n shell: Shell,\n\n /// Disable the binding of CTRL-R to atuin\n #[clap(long)]\n disable_ctrl_r: bool,\n\n /// Disable the binding of the Up Arrow key to atuin\n #[clap(long)]\n disable_up_arrow: bool,\n\n /// Disable the binding of ? to Atuin AI\n #[clap(long)]\n disable_ai: bool,\n}\n\n#[derive(Clone, Copy, ValueEnum, Debug)]\n#[value(rename_all = \"lower\")]\n#[allow(clippy::enum_variant_names, clippy::doc_markdown)]\npub enum Shell {\n /// Zsh setup\n Zsh,\n /// Bash setup\n Bash,\n /// Fish setup\n Fish,\n /// Nu setup\n Nu,\n /// Xonsh setup\n Xonsh,\n /// PowerShell setup\n PowerShell,\n}\n\nimpl Cmd {\n fn init_nu(&self, _tmux: &Tmux) {\n let full = include_str!(\"../../shell/atuin.nu\");\n\n // TODO: tmux popup for Nu\n println!(\"{full}\");\n\n if std::env::var(\"ATUIN_NOBIND\").is_err() {\n const BIND_CTRL_R: &str = r\"$env.config = (\n $env.config | upsert keybindings (\n $env.config.keybindings\n | append {\n name: atuin\n modifier: control\n keycode: char_r\n mode: [emacs, vi_normal, vi_insert]\n event: { send: executehostcommand cmd: (_atuin_search_cmd) }\n }\n )\n)\";\n const BIND_UP_ARROW: &str = r\"\n$env.config = (\n $env.config | upsert keybindings (\n $env.config.keybindings\n | append {\n name: atuin\n modifier: none\n keycode: up\n mode: [emacs, vi_normal, vi_insert]\n event: {\n until: [\n {send: menuup}\n {send: executehostcommand cmd: (_atuin_search_cmd '--shell-up-key-binding') }\n ]\n }\n }\n )\n)\n\";\n if !self.disable_ctrl_r {\n println!(\"{BIND_CTRL_R}\");\n }\n if !self.disable_up_arrow {\n println!(\"{BIND_UP_ARROW}\");\n }\n }\n }\n\n fn static_init(&self, tmux: &Tmux) {\n match self.shell {\n Shell::Zsh => {\n zsh::init_static(\n self.disable_up_arrow,\n self.disable_ctrl_r,\n self.disable_ai,\n tmux,\n );\n }\n Shell::Bash => {\n bash::init_static(\n self.disable_up_arrow,\n self.disable_ctrl_r,\n self.disable_ai,\n tmux,\n );\n }\n Shell::Fish => {\n fish::init_static(\n self.disable_up_arrow,\n self.disable_ctrl_r,\n self.disable_ai,\n tmux,\n );\n }\n Shell::Nu => {\n self.init_nu(tmux);\n }\n Shell::Xonsh => {\n xonsh::init_static(self.disable_up_arrow, self.disable_ctrl_r, tmux);\n }\n Shell::PowerShell => {\n powershell::init_static(self.disable_up_arrow, self.disable_ctrl_r, tmux);\n }\n }\n }\n\n async fn dotfiles_init(&self, settings: &Settings) -> Result<()> {\n let record_store_path = PathBuf::from(settings.record_store_path.as_str());\n let sqlite_store = SqliteStore::new(record_store_path, settings.local_timeout).await?;\n\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n let host_id = Settings::host_id().await?;\n\n let alias_store = AliasStore::new(sqlite_store.clone(), host_id, encryption_key);\n let var_store = VarStore::new(sqlite_store.clone(), host_id, encryption_key);\n\n match self.shell {\n Shell::Zsh => {\n zsh::init(\n alias_store,\n var_store,\n self.disable_up_arrow,\n self.disable_ctrl_r,\n self.disable_ai,\n &settings.tmux,\n )\n .await?;\n }\n Shell::Bash => {\n bash::init(\n alias_store,\n var_store,\n self.disable_up_arrow,\n self.disable_ctrl_r,\n self.disable_ai,\n &settings.tmux,\n )\n .await?;\n }\n Shell::Fish => {\n fish::init(\n alias_store,\n var_store,\n self.disable_up_arrow,\n self.disable_ctrl_r,\n self.disable_ai,\n &settings.tmux,\n )\n .await?;\n }\n Shell::Nu => self.init_nu(&settings.tmux),\n Shell::Xonsh => {\n xonsh::init(\n alias_store,\n var_store,\n self.disable_up_arrow,\n self.disable_ctrl_r,\n &settings.tmux,\n )\n .await?;\n }\n Shell::PowerShell => {\n powershell::init(\n alias_store,\n var_store,\n self.disable_up_arrow,\n self.disable_ctrl_r,\n &settings.tmux,\n )\n .await?;\n }\n }\n\n Ok(())\n }\n\n pub async fn run(self, settings: &Settings) -> Result<()> {\n if !settings.paths_ok() {\n eprintln!(\n \"Atuin settings paths are broken. Disabling atuin shell hooks. Run `atuin doctor` to diagnose.\"\n );\n return Ok(());\n }\n\n if settings.dotfiles.enabled {\n self.dotfiles_init(settings).await?;\n } else {\n self.static_init(&settings.tmux);\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/kv.rs", "content": "use std::io::{self, IsTerminal, Read};\n\nuse clap::Subcommand;\nuse eyre::{Context, Result, eyre};\n\nuse atuin_client::{encryption, record::sqlite_store::SqliteStore, settings::Settings};\nuse atuin_kv::store::KvStore;\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Set a key-value pair\n Set {\n /// Key to set\n #[arg(long, short)]\n key: String,\n\n /// Value to store (reads from stdin if not provided)\n value: Option,\n\n /// Namespace for the key-value pair\n #[arg(long, short, default_value = \"default\")]\n namespace: String,\n },\n\n /// Delete one or more key-value pairs\n #[command(alias = \"rm\")]\n Delete {\n /// Keys to delete\n #[arg(required = true)]\n keys: Vec,\n\n /// Namespace for the key-value pair\n #[arg(long, short, default_value = \"default\")]\n namespace: String,\n },\n\n /// Retrieve a saved value\n Get {\n /// Key to retrieve\n key: String,\n\n /// Namespace for the key-value pair\n #[arg(long, short, default_value = \"default\")]\n namespace: String,\n },\n\n /// List all keys in a namespace, or in all namespaces\n #[command(alias = \"ls\")]\n List {\n /// Namespace to list keys from\n #[arg(long, short, default_value = \"default\")]\n namespace: String,\n\n /// List all keys in all namespaces\n #[arg(long, short, alias = \"all\")]\n all_namespaces: bool,\n },\n\n /// Rebuild the KV store\n Rebuild,\n}\n\nimpl Cmd {\n pub async fn run(&self, settings: &Settings, store: &SqliteStore) -> Result<()> {\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n\n let host_id = Settings::host_id().await?;\n\n let kv_db = atuin_kv::database::Database::new(settings.kv.db_path.clone(), 1.0).await?;\n let kv_store = KvStore::new(store.clone(), kv_db, host_id, encryption_key);\n\n match self {\n Self::Set {\n key,\n value,\n namespace,\n } => {\n if namespace.is_empty() {\n return Err(eyre!(\"namespace cannot be empty\"));\n }\n\n let value = if let Some(v) = value {\n v.clone()\n } else if !io::stdin().is_terminal() {\n let mut buf = String::new();\n io::stdin()\n .read_to_string(&mut buf)\n .context(\"failed to read value from stdin\")?;\n buf\n } else {\n return Err(eyre!(\n \"no value provided. Pass as an argument or pipe via stdin\"\n ));\n };\n\n kv_store.set(namespace, key, &value).await\n }\n\n Self::Delete { keys, namespace } => kv_store.delete(namespace, keys).await,\n\n Self::Get { key, namespace } => {\n let kv = kv_store.get(namespace, key).await?;\n\n if let Some(val) = kv {\n println!(\"{val}\");\n }\n\n Ok(())\n }\n\n Self::List {\n namespace,\n all_namespaces,\n } => {\n let entries = if *all_namespaces {\n kv_store.list(None).await?\n } else {\n kv_store.list(Some(namespace)).await?\n };\n\n for entry in entries {\n if *all_namespaces {\n println!(\"{}.{}\", entry.namespace, entry.key);\n } else {\n println!(\"{}\", entry.key);\n }\n }\n\n Ok(())\n }\n\n Self::Rebuild {} => kv_store.build().await,\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/client/scripts.rs", "content": "use std::collections::HashMap;\nuse std::collections::HashSet;\nuse std::io::IsTerminal;\nuse std::io::Read;\nuse std::path::PathBuf;\n\nuse atuin_scripts::execution::template_script;\nuse atuin_scripts::{\n execution::{build_executable_script, execute_script_interactive, template_variables},\n store::{ScriptStore, script::Script},\n};\nuse clap::{Parser, Subcommand};\nuse eyre::OptionExt;\nuse eyre::{Result, bail};\nuse tempfile::NamedTempFile;\n\nuse atuin_client::{database::Database, record::sqlite_store::SqliteStore, settings::Settings};\nuse tracing::debug;\n\n#[derive(Parser, Debug)]\npub struct NewScript {\n pub name: String,\n\n #[arg(short, long)]\n pub description: Option,\n\n #[arg(short, long)]\n pub tags: Vec,\n\n #[arg(short, long)]\n pub shebang: Option,\n\n #[arg(long)]\n pub script: Option,\n\n #[allow(clippy::option_option)]\n #[arg(long)]\n /// Use the last command as the script content\n /// Optionally specify a number to use the last N commands\n pub last: Option>,\n\n #[arg(long)]\n /// Skip opening editor when using --last\n pub no_edit: bool,\n}\n\n#[derive(Parser, Debug)]\npub struct Run {\n pub name: String,\n\n /// Specify template variables in the format KEY=VALUE\n /// Example: -v name=John -v greeting=\"Hello there\"\n #[arg(short, long = \"var\")]\n pub var: Vec,\n}\n\n#[derive(Parser, Debug)]\npub struct List {}\n\n#[derive(Parser, Debug)]\npub struct Get {\n pub name: String,\n\n #[arg(short, long)]\n /// Display only the executable script with shebang\n pub script: bool,\n}\n\n#[derive(Parser, Debug)]\npub struct Edit {\n pub name: String,\n\n #[arg(short, long)]\n pub description: Option,\n\n /// Replace all existing tags with these new tags\n #[arg(short, long)]\n pub tags: Vec,\n\n /// Remove all tags from the script\n #[arg(long)]\n pub no_tags: bool,\n\n /// Rename the script\n #[arg(long)]\n pub rename: Option,\n\n #[arg(short, long)]\n pub shebang: Option,\n\n #[arg(long)]\n pub script: Option,\n\n #[allow(clippy::struct_field_names)]\n /// Skip opening editor\n #[arg(long)]\n pub no_edit: bool,\n}\n\n#[derive(Parser, Debug)]\npub struct Delete {\n pub name: String,\n\n #[arg(short, long)]\n pub force: bool,\n}\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n New(NewScript),\n Run(Run),\n #[command(alias = \"ls\")]\n List(List),\n\n Get(Get),\n Edit(Edit),\n #[command(alias = \"rm\")]\n Delete(Delete),\n}\n\nimpl Cmd {\n // Helper function to open an editor with optional initial content\n fn open_editor(initial_content: Option<&str>) -> Result {\n // Create a temporary file\n let temp_file = NamedTempFile::new()?;\n let path = temp_file.into_temp_path();\n\n // Write initial content to the temp file if provided\n if let Some(content) = initial_content {\n std::fs::write(&path, content)?;\n }\n\n // Open the file in the user's preferred editor\n let editor_str = std::env::var(\"EDITOR\").unwrap_or_else(|_| \"vi\".to_string());\n\n // Use shlex to safely split the string into shell-like parts.\n let parts = shlex::split(&editor_str).ok_or_eyre(\"Failed to parse editor command\")?;\n let (command, args) = parts.split_first().ok_or_eyre(\"No editor command found\")?;\n\n let status = std::process::Command::new(command)\n .args(args)\n .arg(&path)\n .status()?;\n if !status.success() {\n bail!(\"failed to open editor\");\n }\n\n // Read back the edited content\n let content = std::fs::read_to_string(&path)?;\n path.close()?;\n\n Ok(content)\n }\n\n // Helper function to execute a script and manage stdin/stdout/stderr\n async fn execute_script(script_content: String, shebang: String) -> Result {\n let mut session = execute_script_interactive(script_content, shebang)\n .await\n .expect(\"failed to execute script\");\n\n // Create a channel to signal when the process exits\n let (exit_tx, mut exit_rx) = tokio::sync::oneshot::channel();\n\n // Set up a task to read from stdin and forward to the script\n let sender = session.stdin_tx.clone();\n let stdin_task = tokio::spawn(async move {\n use tokio::io::AsyncReadExt;\n use tokio::select;\n\n let stdin = tokio::io::stdin();\n let mut reader = tokio::io::BufReader::new(stdin);\n let mut buffer = vec![0u8; 1024]; // Read in chunks for efficiency\n\n loop {\n // Use select to either read from stdin or detect when the process exits\n select! {\n // Check if the script process has exited\n _ = &mut exit_rx => {\n break;\n }\n // Try to read from stdin\n read_result = reader.read(&mut buffer) => {\n match read_result {\n Ok(0) => break, // EOF\n Ok(n) => {\n // Convert the bytes to a string and forward to script\n let input = String::from_utf8_lossy(&buffer[0..n]).to_string();\n if let Err(e) = sender.send(input).await {\n eprintln!(\"Error sending input to script: {e}\");\n break;\n }\n },\n Err(e) => {\n eprintln!(\"Error reading from stdin: {e}\");\n break;\n }\n }\n }\n }\n }\n });\n\n // Wait for the script to complete\n let exit_code = session.wait_for_exit().await;\n\n // Signal the stdin task to stop\n let _ = exit_tx.send(());\n let _ = stdin_task.await;\n\n let code = exit_code.unwrap_or(-1);\n if code != 0 {\n eprintln!(\"Script exited with code {code}\");\n }\n\n Ok(code)\n }\n\n async fn handle_new_script(\n settings: &Settings,\n new_script: NewScript,\n script_store: ScriptStore,\n script_db: atuin_scripts::database::Database,\n history_db: &impl Database,\n ) -> Result<()> {\n let mut stdin = std::io::stdin();\n let script_content = if let Some(count_opt) = new_script.last {\n // Get the last N commands from history, plus 1 to exclude the command that runs this script\n let count = count_opt.unwrap_or(1) + 1; // Add 1 to the count to exclude the current command\n let context = atuin_client::database::current_context().await?;\n\n // Get the last N+1 commands, filtering by the default mode\n let filters = [settings.default_filter_mode(context.git_root.is_some())];\n\n let mut history = history_db\n .list(&filters, &context, Some(count), false, false)\n .await?;\n\n // Reverse to get chronological order\n history.reverse();\n\n // Skip the most recent command (which would be the atuin scripts new command itself)\n if !history.is_empty() {\n history.pop(); // Remove the most recent command\n }\n\n // Format the commands into a script\n let commands: Vec = history.iter().map(|h| h.command.clone()).collect();\n\n if commands.is_empty() {\n bail!(\"No commands found in history\");\n }\n\n let script_text = commands.join(\"\\n\");\n\n // Only open editor if --no-edit is not specified\n if new_script.no_edit {\n Some(script_text)\n } else {\n // Open the editor with the commands pre-loaded\n Some(Self::open_editor(Some(&script_text))?)\n }\n } else if let Some(script_path) = new_script.script {\n let script_content = std::fs::read_to_string(script_path)?;\n Some(script_content)\n } else if !stdin.is_terminal() {\n let mut buffer = String::new();\n stdin.read_to_string(&mut buffer)?;\n Some(buffer)\n } else {\n // Open editor with empty file\n Some(Self::open_editor(None)?)\n };\n\n let script = Script::builder()\n .name(new_script.name)\n .description(new_script.description.unwrap_or_default())\n .shebang(new_script.shebang.unwrap_or_default())\n .tags(new_script.tags)\n .script(script_content.unwrap_or_default())\n .build();\n\n script_store.create(script).await?;\n\n script_store.build(script_db).await?;\n\n Ok(())\n }\n\n async fn handle_run(\n _settings: &Settings,\n run: Run,\n script_db: atuin_scripts::database::Database,\n ) -> Result<()> {\n let script = script_db.get_by_name(&run.name).await?;\n\n if let Some(script) = script {\n // Get variables used in the template\n let variables = template_variables(&script)?;\n\n // Create a hashmap to store variable values\n let mut variable_values: HashMap = HashMap::new();\n\n // Parse variables from command-line arguments first\n for var_str in &run.var {\n if let Some((key, value)) = var_str.split_once('=') {\n // Add to variable values\n variable_values.insert(\n key.to_string(),\n serde_json::Value::String(value.to_string()),\n );\n debug!(\"Using CLI variable: {}={}\", key, value);\n } else {\n eprintln!(\"Warning: Ignoring malformed variable specification: {var_str}\");\n eprintln!(\"Variables should be specified as KEY=VALUE\");\n }\n }\n\n // Collect variables that are still needed (not specified via CLI)\n let remaining_vars: HashSet = variables\n .into_iter()\n .filter(|var| !variable_values.contains_key(var))\n .collect();\n\n // If there are variables in the template that weren't specified on the command line, prompt for them\n if !remaining_vars.is_empty() {\n println!(\"This script contains template variables that need values:\");\n\n let stdin = std::io::stdin();\n let mut input = String::new();\n\n for var in remaining_vars {\n input.clear();\n\n println!(\"Enter value for '{var}': \");\n\n if stdin.read_line(&mut input).is_err() {\n eprintln!(\"Failed to read input for variable '{var}'\");\n // Provide an empty string as fallback\n variable_values.insert(var, serde_json::Value::String(String::new()));\n continue;\n }\n\n let value = input.trim().to_string();\n variable_values.insert(var, serde_json::Value::String(value));\n }\n }\n\n let final_script = if variable_values.is_empty() {\n // No variables to template, just use the original script\n script.script.clone()\n } else {\n // If we have variables, we need to template the script\n debug!(\"Templating script with variables: {:?}\", variable_values);\n template_script(&script, &variable_values)?\n };\n\n // Execute the script (either templated or original)\n Self::execute_script(final_script, script.shebang.clone()).await?;\n } else {\n bail!(\"script not found\");\n }\n Ok(())\n }\n\n async fn handle_list(\n _settings: &Settings,\n _list: List,\n script_db: atuin_scripts::database::Database,\n ) -> Result<()> {\n let scripts = script_db.list().await?;\n\n if scripts.is_empty() {\n println!(\"No scripts found\");\n } else {\n println!(\"Available scripts:\");\n for script in scripts {\n if script.tags.is_empty() {\n println!(\"- {} \", script.name);\n } else {\n println!(\"- {} [tags: {}]\", script.name, script.tags.join(\", \"));\n }\n\n // Print description if it's not empty\n if !script.description.is_empty() {\n println!(\" Description: {}\", script.description);\n }\n }\n }\n\n Ok(())\n }\n\n async fn handle_get(\n _settings: &Settings,\n get: Get,\n script_db: atuin_scripts::database::Database,\n ) -> Result<()> {\n let script = script_db.get_by_name(&get.name).await?;\n\n if let Some(script) = script {\n if get.script {\n // Just print the executable script with shebang\n print!(\n \"{}\",\n build_executable_script(script.script.clone(), script.shebang)\n );\n return Ok(());\n }\n\n // Create a YAML representation of the script\n println!(\"---\");\n println!(\"name: {}\", script.name);\n println!(\"id: {}\", script.id);\n\n if script.description.is_empty() {\n println!(\"description: \\\"\\\"\");\n } else {\n println!(\"description: |\");\n // Indent multiline descriptions properly for YAML\n for line in script.description.lines() {\n println!(\" {line}\");\n }\n }\n\n if script.tags.is_empty() {\n println!(\"tags: []\");\n } else {\n println!(\"tags:\");\n for tag in &script.tags {\n println!(\" - {tag}\");\n }\n }\n\n println!(\"shebang: {}\", script.shebang);\n\n println!(\"script: |\");\n // Indent the script content for proper YAML multiline format\n for line in script.script.lines() {\n println!(\" {line}\");\n }\n\n Ok(())\n } else {\n bail!(\"script '{}' not found\", get.name);\n }\n }\n\n #[allow(clippy::cognitive_complexity)]\n async fn handle_edit(\n _settings: &Settings,\n edit: Edit,\n script_store: ScriptStore,\n script_db: atuin_scripts::database::Database,\n ) -> Result<()> {\n debug!(\"editing script {:?}\", edit);\n // Find the existing script\n let existing_script = script_db.get_by_name(&edit.name).await?;\n debug!(\"existing script {:?}\", existing_script);\n\n if let Some(mut script) = existing_script {\n // Update the script with new values if provided\n if let Some(description) = edit.description {\n script.description = description;\n }\n\n // Handle renaming if requested\n if let Some(new_name) = edit.rename {\n // Check if a script with the new name already exists\n if (script_db.get_by_name(&new_name).await?).is_some() {\n bail!(\"A script named '{}' already exists\", new_name);\n }\n\n // Update the name\n script.name = new_name;\n }\n\n // Handle tag updates with priority:\n // 1. If --no-tags is provided, clear all tags\n // 2. If --tags is provided, replace all tags\n // 3. If neither is provided, tags remain unchanged\n if edit.no_tags {\n // Clear all tags\n script.tags.clear();\n } else if !edit.tags.is_empty() {\n // Replace all tags\n script.tags = edit.tags;\n }\n // If none of the above conditions are met, tags remain unchanged\n\n if let Some(shebang) = edit.shebang {\n script.shebang = shebang;\n }\n\n // Handle script content update\n let script_content = if let Some(script_path) = edit.script {\n // Load script from provided file\n std::fs::read_to_string(script_path)?\n } else if !edit.no_edit {\n // Open the script in editor for interactive editing if --no-edit is not specified\n Self::open_editor(Some(&script.script))?\n } else {\n // If --no-edit is specified, keep the existing script content\n script.script.clone()\n };\n\n // Update the script content\n script.script = script_content;\n\n // Update the script in the store\n script_store.update(script).await?;\n\n // Rebuild the database to apply changes\n script_store.build(script_db).await?;\n\n println!(\"Script '{}' updated successfully!\", edit.name);\n\n Ok(())\n } else {\n bail!(\"script '{}' not found\", edit.name);\n }\n }\n\n async fn handle_delete(\n _settings: &Settings,\n delete: Delete,\n script_store: ScriptStore,\n script_db: atuin_scripts::database::Database,\n ) -> Result<()> {\n // Find the script by name\n let script = script_db.get_by_name(&delete.name).await?;\n\n if let Some(script) = script {\n // If not force, confirm deletion\n if !delete.force {\n println!(\n \"Are you sure you want to delete script '{}'? [y/N]\",\n delete.name\n );\n let mut input = String::new();\n std::io::stdin().read_line(&mut input)?;\n\n let input = input.trim().to_lowercase();\n if input != \"y\" && input != \"yes\" {\n println!(\"Deletion cancelled\");\n return Ok(());\n }\n }\n\n // Delete the script\n script_store.delete(script.id).await?;\n\n // Rebuild the database to apply changes\n script_store.build(script_db).await?;\n\n println!(\"Script '{}' deleted successfully\", delete.name);\n Ok(())\n } else {\n bail!(\"script '{}' not found\", delete.name);\n }\n }\n\n pub async fn run(\n self,\n settings: &Settings,\n store: SqliteStore,\n history_db: &impl Database,\n ) -> Result<()> {\n let host_id = Settings::host_id().await?;\n let encryption_key: [u8; 32] = atuin_client::encryption::load_key(settings)?.into();\n\n let script_store = ScriptStore::new(store, host_id, encryption_key);\n let script_db =\n atuin_scripts::database::Database::new(settings.scripts.db_path.clone(), 1.0).await?;\n\n match self {\n Self::New(new_script) => {\n Self::handle_new_script(settings, new_script, script_store, script_db, history_db)\n .await\n }\n Self::Run(run) => Self::handle_run(settings, run, script_db).await,\n Self::List(list) => Self::handle_list(settings, list, script_db).await,\n Self::Get(get) => Self::handle_get(settings, get, script_db).await,\n Self::Edit(edit) => Self::handle_edit(settings, edit, script_store, script_db).await,\n Self::Delete(delete) => {\n Self::handle_delete(settings, delete, script_store, script_db).await\n }\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/cursor.rs", "content": "use atuin_client::settings::WordJumpMode;\n\npub struct Cursor {\n source: String,\n index: usize,\n}\n\nimpl From for Cursor {\n fn from(source: String) -> Self {\n Self { source, index: 0 }\n }\n}\n\npub struct WordJumper<'a> {\n word_chars: &'a str,\n word_jump_mode: WordJumpMode,\n}\n\nimpl WordJumper<'_> {\n fn is_word_boundary(&self, c: char, next_c: char) -> bool {\n (c.is_whitespace() && !next_c.is_whitespace())\n || (!c.is_whitespace() && next_c.is_whitespace())\n || (self.word_chars.contains(c) && !self.word_chars.contains(next_c))\n || (!self.word_chars.contains(c) && self.word_chars.contains(next_c))\n }\n\n fn emacs_get_next_word_pos(&self, source: &str, index: usize) -> usize {\n let index = (index + 1..source.len().saturating_sub(1))\n .find(|&i| self.word_chars.contains(source.chars().nth(i).unwrap()))\n .unwrap_or(source.len());\n (index + 1..source.len().saturating_sub(1))\n .find(|&i| !self.word_chars.contains(source.chars().nth(i).unwrap()))\n .unwrap_or(source.len())\n }\n\n fn emacs_get_prev_word_pos(&self, source: &str, index: usize) -> usize {\n let index = (1..index)\n .rev()\n .find(|&i| self.word_chars.contains(source.chars().nth(i).unwrap()))\n .unwrap_or(0);\n (1..index)\n .rev()\n .find(|&i| !self.word_chars.contains(source.chars().nth(i).unwrap()))\n .map_or(0, |i| i + 1)\n }\n\n fn subl_get_next_word_pos(&self, source: &str, index: usize) -> usize {\n let index = (index..source.len().saturating_sub(1)).find(|&i| {\n self.is_word_boundary(\n source.chars().nth(i).unwrap(),\n source.chars().nth(i + 1).unwrap(),\n )\n });\n if index.is_none() {\n return source.len();\n }\n (index.unwrap() + 1..source.len())\n .find(|&i| !source.chars().nth(i).unwrap().is_whitespace())\n .unwrap_or(source.len())\n }\n\n fn subl_get_prev_word_pos(&self, source: &str, index: usize) -> usize {\n let index = (1..index)\n .rev()\n .find(|&i| !source.chars().nth(i).unwrap().is_whitespace());\n if index.is_none() {\n return 0;\n }\n (1..index.unwrap())\n .rev()\n .find(|&i| {\n self.is_word_boundary(\n source.chars().nth(i - 1).unwrap(),\n source.chars().nth(i).unwrap(),\n )\n })\n .unwrap_or(0)\n }\n\n fn get_next_word_pos(&self, source: &str, index: usize) -> usize {\n match self.word_jump_mode {\n WordJumpMode::Emacs => self.emacs_get_next_word_pos(source, index),\n WordJumpMode::Subl => self.subl_get_next_word_pos(source, index),\n }\n }\n\n fn get_prev_word_pos(&self, source: &str, index: usize) -> usize {\n match self.word_jump_mode {\n WordJumpMode::Emacs => self.emacs_get_prev_word_pos(source, index),\n WordJumpMode::Subl => self.subl_get_prev_word_pos(source, index),\n }\n }\n}\n\nimpl Cursor {\n pub fn as_str(&self) -> &str {\n self.source.as_str()\n }\n\n pub fn into_inner(self) -> String {\n self.source\n }\n\n /// Returns the string before the cursor\n pub fn substring(&self) -> &str {\n &self.source[..self.index]\n }\n\n /// Returns the currently selected [`char`]\n pub fn char(&self) -> Option {\n self.source[self.index..].chars().next()\n }\n\n pub fn right(&mut self) {\n if self.index < self.source.len() {\n loop {\n self.index += 1;\n if self.source.is_char_boundary(self.index) {\n break;\n }\n }\n }\n }\n\n pub fn left(&mut self) -> bool {\n if self.index > 0 {\n loop {\n self.index -= 1;\n if self.source.is_char_boundary(self.index) {\n break true;\n }\n }\n } else {\n false\n }\n }\n\n pub fn next_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {\n let word_jumper = WordJumper {\n word_chars,\n word_jump_mode,\n };\n self.index = word_jumper.get_next_word_pos(&self.source, self.index);\n }\n\n pub fn prev_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {\n let word_jumper = WordJumper {\n word_chars,\n word_jump_mode,\n };\n self.index = word_jumper.get_prev_word_pos(&self.source, self.index);\n }\n\n /// Move cursor to the end of the current/next word (vim `e` motion).\n ///\n /// If cursor is in the middle of a word, moves to the end of that word.\n /// If cursor is at the end of a word (or on whitespace), moves to the\n /// end of the next word.\n pub fn word_end(&mut self, word_chars: &str) {\n let len = self.source.len();\n if self.index >= len {\n return;\n }\n\n let chars: Vec = self.source.chars().collect();\n let mut char_idx = self.source[..self.index].chars().count();\n\n if char_idx >= chars.len() {\n return;\n }\n\n let current = chars[char_idx];\n\n // Check if we're at a word boundary (end of current word or on whitespace)\n let at_word_boundary = current.is_whitespace() || char_idx + 1 >= chars.len() || {\n let next = chars[char_idx + 1];\n next.is_whitespace() || (word_chars.contains(current) != word_chars.contains(next))\n };\n\n // If at word boundary, advance past it and skip whitespace to find next word\n if at_word_boundary {\n char_idx += 1;\n while char_idx < chars.len() && chars[char_idx].is_whitespace() {\n char_idx += 1;\n }\n }\n\n // If we've gone past end, go to end of string\n if char_idx >= chars.len() {\n self.index = len;\n return;\n }\n\n // Find end of word: advance until next char is whitespace or different word type\n let in_word_chars = word_chars.contains(chars[char_idx]);\n while char_idx < chars.len() {\n let next_idx = char_idx + 1;\n if next_idx >= chars.len() {\n // At last char, move past it\n char_idx = next_idx;\n break;\n }\n let next_c = chars[next_idx];\n if next_c.is_whitespace() || (word_chars.contains(next_c) != in_word_chars) {\n // Next char is start of new word/whitespace, so current char is end\n char_idx = next_idx;\n break;\n }\n char_idx += 1;\n }\n\n // Convert char index back to byte index\n self.index = chars.iter().take(char_idx).map(|c| c.len_utf8()).sum();\n }\n\n pub fn insert(&mut self, c: char) {\n self.source.insert(self.index, c);\n self.index += c.len_utf8();\n }\n\n pub fn remove(&mut self) -> Option {\n if self.index < self.source.len() {\n Some(self.source.remove(self.index))\n } else {\n None\n }\n }\n\n pub fn remove_next_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {\n let word_jumper = WordJumper {\n word_chars,\n word_jump_mode,\n };\n let next_index = word_jumper.get_next_word_pos(&self.source, self.index);\n self.source.replace_range(self.index..next_index, \"\");\n }\n\n pub fn remove_prev_word(&mut self, word_chars: &str, word_jump_mode: WordJumpMode) {\n let word_jumper = WordJumper {\n word_chars,\n word_jump_mode,\n };\n let next_index = word_jumper.get_prev_word_pos(&self.source, self.index);\n self.source.replace_range(next_index..self.index, \"\");\n self.index = next_index;\n }\n\n pub fn back(&mut self) -> Option {\n if self.left() { self.remove() } else { None }\n }\n\n pub fn clear(&mut self) {\n self.source.clear();\n self.index = 0;\n }\n\n pub fn clear_to_start(&mut self) {\n self.source.replace_range(..self.index, \"\");\n self.index = 0;\n }\n\n pub fn clear_to_end(&mut self) {\n self.source.replace_range(self.index.., \"\");\n self.index = self.source.len();\n }\n\n pub fn end(&mut self) {\n self.index = self.source.len();\n }\n\n pub fn start(&mut self) {\n self.index = 0;\n }\n\n pub fn position(&self) -> usize {\n self.index\n }\n}\n\n#[cfg(test)]\nmod cursor_tests {\n use super::Cursor;\n use super::*;\n\n static EMACS_WORD_JUMPER: WordJumper = WordJumper {\n word_chars: \"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789\",\n word_jump_mode: WordJumpMode::Emacs,\n };\n\n static SUBL_WORD_JUMPER: WordJumper = WordJumper {\n word_chars: \"./\\\\()\\\"'-:,.;<>~!@#$%^&*|+=[]{}`~?\",\n word_jump_mode: WordJumpMode::Subl,\n };\n\n #[test]\n fn right() {\n // ö is 2 bytes\n let mut c = Cursor::from(String::from(\"öaöböcödöeöfö\"));\n let indices = [0, 2, 3, 5, 6, 8, 9, 11, 12, 14, 15, 17, 18, 20, 20, 20, 20];\n for i in indices {\n assert_eq!(c.index, i);\n c.right();\n }\n }\n\n #[test]\n fn left() {\n // ö is 2 bytes\n let mut c = Cursor::from(String::from(\"öaöböcödöeöfö\"));\n c.end();\n let indices = [20, 18, 17, 15, 14, 12, 11, 9, 8, 6, 5, 3, 2, 0, 0, 0, 0];\n for i in indices {\n assert_eq!(c.index, i);\n c.left();\n }\n }\n\n #[test]\n fn test_emacs_get_next_word_pos() {\n let s = String::from(\" aaa ((()))bbb ((())) \");\n let indices = [(0, 6), (3, 6), (7, 18), (19, 30)];\n for (i_src, i_dest) in indices {\n assert_eq!(EMACS_WORD_JUMPER.get_next_word_pos(&s, i_src), i_dest);\n }\n assert_eq!(EMACS_WORD_JUMPER.get_next_word_pos(\"\", 0), 0);\n }\n\n #[test]\n fn test_emacs_get_prev_word_pos() {\n let s = String::from(\" aaa ((()))bbb ((())) \");\n let indices = [(30, 15), (29, 15), (15, 3), (3, 0)];\n for (i_src, i_dest) in indices {\n assert_eq!(EMACS_WORD_JUMPER.get_prev_word_pos(&s, i_src), i_dest);\n }\n assert_eq!(EMACS_WORD_JUMPER.get_prev_word_pos(\"\", 0), 0);\n }\n\n #[test]\n fn test_subl_get_next_word_pos() {\n let s = String::from(\" aaa ((()))bbb ((())) \");\n let indices = [(0, 3), (1, 3), (3, 9), (9, 15), (15, 21), (21, 30)];\n for (i_src, i_dest) in indices {\n assert_eq!(SUBL_WORD_JUMPER.get_next_word_pos(&s, i_src), i_dest);\n }\n assert_eq!(SUBL_WORD_JUMPER.get_next_word_pos(\"\", 0), 0);\n }\n\n #[test]\n fn test_subl_get_prev_word_pos() {\n let s = String::from(\" aaa ((()))bbb ((())) \");\n let indices = [(30, 21), (21, 15), (15, 9), (9, 3), (3, 0)];\n for (i_src, i_dest) in indices {\n assert_eq!(SUBL_WORD_JUMPER.get_prev_word_pos(&s, i_src), i_dest);\n }\n assert_eq!(SUBL_WORD_JUMPER.get_prev_word_pos(\"\", 0), 0);\n }\n\n #[test]\n fn pop() {\n let mut s = String::from(\"öaöböcödöeöfö\");\n let mut c = Cursor::from(s.clone());\n c.end();\n while !s.is_empty() {\n let c1 = s.pop();\n let c2 = c.back();\n assert_eq!(c1, c2);\n assert_eq!(s.as_str(), c.substring());\n }\n let c1 = s.pop();\n let c2 = c.back();\n assert_eq!(c1, c2);\n }\n\n #[test]\n fn back() {\n let mut c = Cursor::from(String::from(\"öaöböcödöeöfö\"));\n // move to ^\n for _ in 0..4 {\n c.right();\n }\n assert_eq!(c.substring(), \"öaöb\");\n assert_eq!(c.back(), Some('b'));\n assert_eq!(c.back(), Some('ö'));\n assert_eq!(c.back(), Some('a'));\n assert_eq!(c.back(), Some('ö'));\n assert_eq!(c.back(), None);\n assert_eq!(c.as_str(), \"öcödöeöfö\");\n }\n\n #[test]\n fn insert() {\n let mut c = Cursor::from(String::from(\"öaöböcödöeöfö\"));\n // move to ^\n for _ in 0..4 {\n c.right();\n }\n assert_eq!(c.substring(), \"öaöb\");\n c.insert('ö');\n c.insert('g');\n c.insert('ö');\n c.insert('h');\n assert_eq!(c.substring(), \"öaöbögöh\");\n assert_eq!(c.as_str(), \"öaöbögöhöcödöeöfö\");\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/duration.rs", "content": "use core::fmt;\nuse std::{ops::ControlFlow, time::Duration};\n\n#[allow(clippy::module_name_repetitions)]\npub fn format_duration_into(dur: Duration, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n fn item(unit: &'static str, value: u64) -> ControlFlow<(&'static str, u64)> {\n if value > 0 {\n ControlFlow::Break((unit, value))\n } else {\n ControlFlow::Continue(())\n }\n }\n\n // impl taken and modified from\n // https://github.com/tailhook/humantime/blob/master/src/duration.rs#L295-L331\n // Copyright (c) 2016 The humantime Developers\n fn fmt(f: Duration) -> ControlFlow<(&'static str, u64), ()> {\n let secs = f.as_secs();\n let nanos = f.subsec_nanos();\n\n let years = secs / 31_557_600; // 365.25d\n let year_days = secs % 31_557_600;\n let months = year_days / 2_630_016; // 30.44d\n let month_days = year_days % 2_630_016;\n let days = month_days / 86400;\n let day_secs = month_days % 86400;\n let hours = day_secs / 3600;\n let minutes = day_secs % 3600 / 60;\n let seconds = day_secs % 60;\n\n let millis = nanos / 1_000_000;\n let micros = nanos / 1_000;\n\n // a difference from our impl than the original is that\n // we only care about the most-significant segment of the duration.\n // If the item call returns `Break`, then the `?` will early-return.\n // This allows for a very consise impl\n item(\"y\", years)?;\n item(\"mo\", months)?;\n item(\"d\", days)?;\n item(\"h\", hours)?;\n item(\"m\", minutes)?;\n item(\"s\", seconds)?;\n item(\"ms\", u64::from(millis))?;\n item(\"us\", u64::from(micros))?;\n item(\"ns\", u64::from(nanos))?;\n ControlFlow::Continue(())\n }\n\n match fmt(dur) {\n ControlFlow::Break((unit, value)) => write!(f, \"{value}{unit}\"),\n ControlFlow::Continue(()) => write!(f, \"0s\"),\n }\n}\n\n#[allow(clippy::module_name_repetitions)]\npub fn format_duration(f: Duration) -> String {\n struct F(Duration);\n impl fmt::Display for F {\n fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n format_duration_into(self.0, f)\n }\n }\n F(f).to_string()\n}\n" }, { "path": "crates/atuin/src/command/client/search/engines/daemon.rs", "content": "use async_trait::async_trait;\nuse atuin_client::{\n database::{Database, OptFilters},\n history::History,\n settings::{SearchMode, Settings},\n};\nuse atuin_daemon::client::SearchClient;\nuse atuin_nucleo_matcher::{\n Config, Matcher, Utf32Str,\n pattern::{CaseMatching, Normalization, Pattern},\n};\nuse eyre::Result;\nuse tracing::{Level, debug, instrument, span};\nuse uuid::Uuid;\n\nuse super::{SearchEngine, SearchState};\n\npub struct Search {\n client: Option,\n query_id: u64,\n socket_path: String,\n #[cfg(not(unix))]\n tcp_port: u64,\n}\n\nimpl Search {\n pub fn new(settings: &Settings) -> Self {\n Search {\n client: None,\n query_id: 0,\n socket_path: settings.daemon.socket_path.clone(),\n #[cfg(not(unix))]\n tcp_port: settings.daemon.tcp_port,\n }\n }\n\n #[instrument(skip_all, level = Level::TRACE, name = \"get_daemon_client\")]\n async fn get_client(&mut self) -> Result<&mut SearchClient> {\n if self.client.is_none() {\n #[cfg(unix)]\n let client = SearchClient::new(self.socket_path.clone()).await?;\n\n #[cfg(not(unix))]\n let client = SearchClient::new(self.tcp_port).await?;\n\n self.client = Some(client);\n }\n Ok(self.client.as_mut().unwrap())\n }\n\n fn next_query_id(&mut self) -> u64 {\n self.query_id += 1;\n self.query_id\n }\n\n /// Check if query contains regex pattern (r/.../)\n /// Nucleo doesn't support regex, so we fall back to database search\n fn contains_regex_pattern(query: &str) -> bool {\n query.starts_with(\"r/\") || query.contains(\" r/\")\n }\n\n #[instrument(skip_all, level = Level::TRACE, name = \"daemon_db_fallback\")]\n async fn fallback_to_db_search(\n &self,\n state: &SearchState,\n db: &dyn Database,\n ) -> Result> {\n let results = db\n .search(\n SearchMode::FullText,\n state.filter_mode,\n &state.context,\n state.input.as_str(),\n OptFilters {\n limit: Some(200),\n ..Default::default()\n },\n )\n .await\n .map_or(Vec::new(), |r| r.into_iter().collect());\n Ok(results)\n }\n\n #[instrument(skip_all, level = Level::TRACE, name = \"hydrate_from_db\", fields(count = ids.len()))]\n async fn hydrate_from_db(&self, db: &dyn Database, ids: &[String]) -> Result> {\n let placeholders: Vec = ids.iter().map(|id| format!(\"'{id}'\")).collect();\n let sql_query = format!(\n \"SELECT * FROM history WHERE id IN ({}) ORDER BY timestamp DESC\",\n placeholders.join(\",\")\n );\n Ok(db.query_history(&sql_query).await?)\n }\n}\n\n#[async_trait]\nimpl SearchEngine for Search {\n #[instrument(skip_all, level = Level::TRACE, name = \"daemon_search\", fields(query = %state.input.as_str()))]\n async fn full_query(\n &mut self,\n state: &SearchState,\n db: &mut dyn Database,\n ) -> Result> {\n let query = state.input.as_str().to_string();\n\n // Fall back to database for regex queries (Nucleo doesn't support regex)\n if Self::contains_regex_pattern(&query) {\n debug!(query = %query, \"[daemon-client] regex detected, falling back to db\");\n return self.fallback_to_db_search(state, db).await;\n }\n\n let query_id = self.next_query_id();\n\n let span =\n span!(Level::TRACE, \"daemon_search.req_resp\", query = %query, query_id = query_id);\n\n let client = self.get_client().await?;\n\n let _span = span.enter();\n let mut stream = client\n .search(\n query.clone(),\n query_id,\n state.filter_mode,\n Some(state.context.clone()),\n )\n .await?;\n\n let mut ids = Vec::with_capacity(200);\n span!(Level::TRACE, \"daemon_search.resp\")\n .in_scope(async || {\n while let Ok(Some(response)) = stream.message().await {\n let span2 = span!(\n Level::TRACE,\n \"daemon_search.resp.item\",\n query_id = response.query_id\n );\n let _span2 = span2.enter();\n // Only process if the query_id matches (prevents stale responses)\n if response.query_id == query_id {\n let uuids = response\n .ids\n .iter()\n .map(|id| {\n let bytes: [u8; 16] =\n id.as_slice().try_into().expect(\"id should be 16 bytes\");\n Uuid::from_bytes(bytes).as_simple().to_string()\n })\n .collect::>();\n ids.extend(uuids);\n }\n drop(_span2);\n drop(span2);\n }\n })\n .await;\n drop(_span);\n drop(span);\n\n if ids.is_empty() {\n debug!(query = %query, results = 0, \"[daemon-client] empty results\");\n return Ok(Vec::new());\n }\n\n // // Hydrate from local database\n let results = self.hydrate_from_db(db, &ids).await?;\n\n // // Reorder results to match the order from the daemon (which is ranked by relevance)\n let ordered_results = span!(Level::TRACE, \"reorder_results\").in_scope(|| {\n let mut ordered_results = Vec::with_capacity(results.len());\n for id in &ids {\n if let Some(history) = results.iter().find(|h| h.id.0 == *id) {\n ordered_results.push(history.clone());\n }\n }\n ordered_results\n });\n\n debug!(\n query = %query,\n results = results.len(),\n \"[daemon-client]\"\n );\n\n Ok(ordered_results)\n }\n\n #[instrument(skip_all, level = Level::TRACE, name = \"daemon_highlight\")]\n fn get_highlight_indices(&self, command: &str, search_input: &str) -> Vec {\n // Use fulltext highlighting for regex queries\n if Self::contains_regex_pattern(search_input) {\n return super::db::get_highlight_indices_fulltext(command, search_input);\n }\n\n let mut matcher = Matcher::new(Config::DEFAULT);\n let pattern = Pattern::parse(search_input, CaseMatching::Smart, Normalization::Smart);\n\n let mut indices: Vec = Vec::new();\n let mut haystack_buf = Vec::new();\n\n let haystack = Utf32Str::new(command, &mut haystack_buf);\n pattern.indices(haystack, &mut matcher, &mut indices);\n\n // Convert u32 indices to usize\n indices.into_iter().map(|i| i as usize).collect()\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/engines/db.rs", "content": "use super::{SearchEngine, SearchState};\nuse async_trait::async_trait;\nuse atuin_client::{\n database::Database,\n database::OptFilters,\n database::{QueryToken, QueryTokenizer},\n history::History,\n settings::SearchMode,\n};\nuse eyre::Result;\nuse norm::Metric;\nuse norm::fzf::{FzfParser, FzfV2};\nuse std::ops::Range;\nuse tracing::{Level, instrument};\n\npub struct Search(pub SearchMode);\n\n#[async_trait]\nimpl SearchEngine for Search {\n #[instrument(skip_all, level = Level::TRACE, name = \"db_search\", fields(mode = ?self.0, query = %state.input.as_str()))]\n async fn full_query(\n &mut self,\n state: &SearchState,\n db: &mut dyn Database,\n ) -> Result> {\n let results = db\n .search(\n self.0,\n state.filter_mode,\n &state.context,\n state.input.as_str(),\n OptFilters {\n limit: Some(200),\n ..Default::default()\n },\n )\n .await\n // ignore errors as it may be caused by incomplete regex\n .map_or(Vec::new(), |r| r.into_iter().collect());\n Ok(results)\n }\n\n #[instrument(skip_all, level = Level::TRACE, name = \"db_highlight\")]\n fn get_highlight_indices(&self, command: &str, search_input: &str) -> Vec {\n if self.0 == SearchMode::Prefix {\n return vec![];\n } else if self.0 == SearchMode::FullText {\n return get_highlight_indices_fulltext(command, search_input);\n }\n let mut fzf = FzfV2::new();\n let mut parser = FzfParser::new();\n let query = parser.parse(search_input);\n let mut ranges: Vec> = Vec::new();\n let _ = fzf.distance_and_ranges(query, command, &mut ranges);\n\n // convert ranges to all indices\n ranges.into_iter().flatten().collect()\n }\n}\n\n#[instrument(skip_all, level = Level::TRACE, name = \"db_highlight_fulltext\")]\npub fn get_highlight_indices_fulltext(command: &str, search_input: &str) -> Vec {\n let mut ranges = vec![];\n let lower_command = command.to_ascii_lowercase();\n\n for token in QueryTokenizer::new(search_input) {\n let matchee = if token.has_uppercase() {\n command\n } else {\n &lower_command\n };\n\n if token.is_inverse() {\n continue;\n }\n\n match token {\n QueryToken::Or => {}\n QueryToken::Regex(r) => {\n if let Ok(re) = regex::Regex::new(r) {\n for m in re.find_iter(command) {\n ranges.push(m.range());\n }\n }\n }\n QueryToken::MatchStart(term, _) => {\n if matchee.starts_with(term) {\n ranges.push(0..term.len());\n }\n }\n QueryToken::MatchEnd(term, _) => {\n if matchee.ends_with(term) {\n let l = matchee.len();\n ranges.push((l - term.len())..l);\n }\n }\n QueryToken::Match(term, _) | QueryToken::MatchFull(term, _) => {\n for (idx, m) in matchee.match_indices(term) {\n ranges.push(idx..(idx + m.len()));\n }\n }\n }\n }\n\n let mut ret: Vec<_> = ranges.into_iter().flatten().collect();\n ret.sort_unstable();\n ret.dedup();\n ret\n}\n" }, { "path": "crates/atuin/src/command/client/search/engines/skim.rs", "content": "use std::path::Path;\n\nuse async_trait::async_trait;\nuse atuin_client::{database::Database, history::History, settings::FilterMode};\nuse eyre::Result;\nuse fuzzy_matcher::{FuzzyMatcher, skim::SkimMatcherV2};\nuse itertools::Itertools;\nuse time::OffsetDateTime;\nuse tokio::task::yield_now;\nuse tracing::{Level, instrument, warn};\nuse uuid;\n\nuse super::{SearchEngine, SearchState};\n\npub struct Search {\n all_history: Vec<(History, i32)>,\n engine: SkimMatcherV2,\n}\n\nimpl Search {\n pub fn new() -> Self {\n Search {\n all_history: vec![],\n engine: SkimMatcherV2::default(),\n }\n }\n}\n\n#[async_trait]\nimpl SearchEngine for Search {\n #[instrument(skip_all, level = Level::TRACE, name = \"skim_search\", fields(query = %state.input.as_str()))]\n async fn full_query(\n &mut self,\n state: &SearchState,\n db: &mut dyn Database,\n ) -> Result> {\n if self.all_history.is_empty() {\n self.all_history = load_all_history(db).await;\n }\n\n Ok(fuzzy_search(&self.engine, state, &self.all_history).await)\n }\n\n #[instrument(skip_all, level = Level::TRACE, name = \"skim_highlight\")]\n fn get_highlight_indices(&self, command: &str, search_input: &str) -> Vec {\n let (_, indices) = self\n .engine\n .fuzzy_indices(command, search_input)\n .unwrap_or_default();\n indices\n }\n}\n\n#[instrument(skip_all, level = Level::TRACE, name = \"load_all_history\")]\nasync fn load_all_history(db: &dyn Database) -> Vec<(History, i32)> {\n db.all_with_count().await.unwrap()\n}\n\n#[allow(clippy::too_many_lines)]\n#[instrument(skip_all, level = Level::TRACE, name = \"fuzzy_match\", fields(history_count = all_history.len()))]\nasync fn fuzzy_search(\n engine: &SkimMatcherV2,\n state: &SearchState,\n all_history: &[(History, i32)],\n) -> Vec {\n let mut set = Vec::with_capacity(200);\n let mut ranks = Vec::with_capacity(200);\n let query = state.input.as_str();\n let now = OffsetDateTime::now_utc();\n\n for (i, (history, count)) in all_history.iter().enumerate() {\n if i % 256 == 0 {\n yield_now().await;\n }\n let context = &state.context;\n let git_root = context\n .git_root\n .as_ref()\n .and_then(|git_root| git_root.to_str())\n .unwrap_or(&context.cwd);\n match state.filter_mode {\n FilterMode::Global => {}\n // we aggregate host by ',' separating them\n FilterMode::Host\n if history\n .hostname\n .split(',')\n .contains(&context.hostname.as_str()) => {}\n // we aggregate session by concattenating them.\n // sessions are 32 byte simple uuid formats\n FilterMode::Session\n if history\n .session\n .as_bytes()\n .chunks(32)\n .contains(&context.session.as_bytes()) => {}\n // SessionPreload: include current session + global history from before session start\n FilterMode::SessionPreload => {\n let is_current_session = {\n history\n .session\n .as_bytes()\n .chunks(32)\n .any(|chunk| chunk == context.session.as_bytes())\n };\n\n if !is_current_session {\n let Ok(uuid) = uuid::Uuid::parse_str(&context.session) else {\n warn!(\"failed to parse session id '{}'\", context.session);\n continue;\n };\n let Some(timestamp) = uuid.get_timestamp() else {\n warn!(\n \"failed to get timestamp from uuid '{}'\",\n uuid.as_hyphenated()\n );\n continue;\n };\n let (seconds, nanos) = timestamp.to_unix();\n let Ok(session_start) = time::OffsetDateTime::from_unix_timestamp_nanos(\n i128::from(seconds) * 1_000_000_000 + i128::from(nanos),\n ) else {\n warn!(\n \"failed to create OffsetDateTime from second: {seconds}, nanosecond: {nanos}\"\n );\n continue;\n };\n\n if history.timestamp >= session_start {\n continue;\n }\n }\n }\n // we aggregate directory by ':' separating them\n FilterMode::Directory if history.cwd.split(':').contains(&context.cwd.as_str()) => {}\n FilterMode::Workspace if history.cwd.split(':').contains(&git_root) => {}\n _ => continue,\n }\n #[allow(clippy::cast_lossless, clippy::cast_precision_loss)]\n if let Some((score, indices)) = engine.fuzzy_indices(&history.command, query) {\n let begin = indices.first().copied().unwrap_or_default();\n\n let mut duration = (now - history.timestamp).as_seconds_f64().log2();\n if !duration.is_finite() || duration <= 1.0 {\n duration = 1.0;\n }\n // these + X.0 just make the log result a bit smoother.\n // log is very spiky towards 1-4, but I want a gradual decay.\n // eg:\n // log2(4) = 2, log2(5) = 2.3 (16% increase)\n // log2(8) = 3, log2(9) = 3.16 (5% increase)\n // log2(16) = 4, log2(17) = 4.08 (2% increase)\n let count = (*count as f64 + 8.0).log2();\n let begin = (begin as f64 + 16.0).log2();\n let path = path_dist(history.cwd.as_ref(), state.context.cwd.as_ref());\n let path = (path as f64 + 8.0).log2();\n\n // reduce longer durations, raise higher counts, raise matches close to the start\n let score = (-score as f64) * count / path / duration / begin;\n\n 'insert: {\n // algorithm:\n // 1. find either the position that this command ranks\n // 2. find the same command positioned better than our rank.\n for i in 0..set.len() {\n // do we out score the current position?\n if ranks[i] > score {\n ranks.insert(i, score);\n set.insert(i, history.clone());\n let mut j = i + 1;\n while j < set.len() {\n // remove duplicates that have a worse score\n if set[j].command == history.command {\n ranks.remove(j);\n set.remove(j);\n\n // break this while loop because there won't be any other\n // duplicates.\n break;\n }\n j += 1;\n }\n\n // keep it limited\n if ranks.len() > 200 {\n ranks.pop();\n set.pop();\n }\n\n break 'insert;\n }\n // don't continue if this command has a better score already\n if set[i].command == history.command {\n break 'insert;\n }\n }\n\n if set.len() < 200 {\n ranks.push(score);\n set.push(history.clone());\n }\n }\n }\n }\n\n set\n}\n\nfn path_dist(a: &Path, b: &Path) -> usize {\n let mut a: Vec<_> = a.components().collect();\n let b: Vec<_> = b.components().collect();\n\n let mut dist = 0;\n\n // pop a until there's a common ancestor\n while !b.starts_with(&a) {\n dist += 1;\n a.pop();\n }\n\n b.len() - a.len() + dist\n}\n" }, { "path": "crates/atuin/src/command/client/search/engines.rs", "content": "use async_trait::async_trait;\nuse atuin_client::{\n database::{Context, Database},\n history::{History, HistoryId},\n settings::{FilterMode, SearchMode, Settings},\n};\nuse eyre::Result;\n\nuse super::cursor::Cursor;\n\n#[cfg(feature = \"daemon\")]\npub mod daemon;\npub mod db;\npub mod skim;\n\n#[allow(unused)] // settings is only used if daemon feature is enabled\npub fn engine(search_mode: SearchMode, settings: &Settings) -> Box {\n match search_mode {\n SearchMode::Skim => Box::new(skim::Search::new()) as Box<_>,\n #[cfg(feature = \"daemon\")]\n SearchMode::DaemonFuzzy => Box::new(daemon::Search::new(settings)) as Box<_>,\n #[cfg(not(feature = \"daemon\"))]\n SearchMode::DaemonFuzzy => {\n // Fall back to fuzzy mode if daemon feature is not enabled\n Box::new(db::Search(SearchMode::Fuzzy)) as Box<_>\n }\n mode => Box::new(db::Search(mode)) as Box<_>,\n }\n}\n\npub struct SearchState {\n pub input: Cursor,\n pub filter_mode: FilterMode,\n pub context: Context,\n pub custom_context: Option,\n}\n\nimpl SearchState {\n pub(crate) fn rotate_filter_mode(&mut self, settings: &Settings, offset: isize) {\n let mut i = settings\n .search\n .filters\n .iter()\n .position(|&m| m == self.filter_mode)\n .unwrap_or_default();\n for _ in 0..settings.search.filters.len() {\n i = (i.wrapping_add_signed(offset)) % settings.search.filters.len();\n let mode = settings.search.filters[i];\n if self.filter_mode_available(mode, settings) {\n self.filter_mode = mode;\n break;\n }\n }\n }\n\n fn filter_mode_available(&self, mode: FilterMode, settings: &Settings) -> bool {\n match mode {\n FilterMode::Global | FilterMode::SessionPreload => self.custom_context.is_none(),\n FilterMode::Workspace => settings.workspaces && self.context.git_root.is_some(),\n _ => true,\n }\n }\n}\n\n#[async_trait]\npub trait SearchEngine: Send + Sync + 'static {\n async fn full_query(\n &mut self,\n state: &SearchState,\n db: &mut dyn Database,\n ) -> Result>;\n\n async fn query(&mut self, state: &SearchState, db: &mut dyn Database) -> Result> {\n if state.input.as_str().is_empty() {\n Ok(db\n .list(&[state.filter_mode], &state.context, Some(200), true, false)\n .await?\n .into_iter()\n .collect::>())\n } else {\n self.full_query(state, db).await\n }\n }\n fn get_highlight_indices(&self, command: &str, search_input: &str) -> Vec;\n}\n" }, { "path": "crates/atuin/src/command/client/search/history_list.rs", "content": "use std::time::Duration;\n\nuse super::duration::format_duration;\nuse super::engines::SearchEngine;\nuse atuin_client::{\n history::History,\n settings::{UiColumn, UiColumnType},\n theme::{Meaning, Theme},\n};\nuse atuin_common::utils::Escapable as _;\nuse itertools::Itertools;\nuse ratatui::{\n backend::FromCrossterm,\n buffer::Buffer,\n crossterm::style,\n layout::Rect,\n style::{Modifier, Style},\n widgets::{Block, StatefulWidget, Widget},\n};\nuse time::OffsetDateTime;\n\npub struct HistoryHighlighter<'a> {\n pub engine: &'a dyn SearchEngine,\n pub search_input: &'a str,\n}\n\nimpl HistoryHighlighter<'_> {\n pub fn get_highlight_indices(&self, command: &str) -> Vec {\n self.engine\n .get_highlight_indices(command, self.search_input)\n }\n}\n\npub struct HistoryList<'a> {\n history: &'a [History],\n block: Option>,\n inverted: bool,\n /// Apply an alternative highlighting to the selected row\n alternate_highlight: bool,\n now: &'a dyn Fn() -> OffsetDateTime,\n indicator: &'a str,\n theme: &'a Theme,\n history_highlighter: HistoryHighlighter<'a>,\n show_numeric_shortcuts: bool,\n /// Columns to display (in order, after the indicator)\n columns: &'a [UiColumn],\n}\n\n#[derive(Default)]\npub struct ListState {\n offset: usize,\n selected: usize,\n max_entries: usize,\n}\n\nimpl ListState {\n pub fn selected(&self) -> usize {\n self.selected\n }\n\n pub fn max_entries(&self) -> usize {\n self.max_entries\n }\n\n pub fn offset(&self) -> usize {\n self.offset\n }\n\n pub fn select(&mut self, index: usize) {\n self.selected = index;\n }\n}\n\nimpl StatefulWidget for HistoryList<'_> {\n type State = ListState;\n\n fn render(mut self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {\n let list_area = self.block.take().map_or(area, |b| {\n let inner_area = b.inner(area);\n b.render(area, buf);\n inner_area\n });\n\n if list_area.width < 1 || list_area.height < 1 || self.history.is_empty() {\n return;\n }\n let list_height = list_area.height as usize;\n\n let (start, end) = self.get_items_bounds(state.selected, state.offset, list_height);\n state.offset = start;\n state.max_entries = end - start;\n\n let mut s = DrawState {\n buf,\n list_area,\n x: 0,\n y: 0,\n state,\n inverted: self.inverted,\n alternate_highlight: self.alternate_highlight,\n now: &self.now,\n indicator: self.indicator,\n theme: self.theme,\n history_highlighter: self.history_highlighter,\n show_numeric_shortcuts: self.show_numeric_shortcuts,\n columns: self.columns,\n };\n\n for item in self.history.iter().skip(state.offset).take(end - start) {\n s.render_row(item);\n\n // reset line\n s.y += 1;\n s.x = 0;\n }\n }\n}\n\nimpl<'a> HistoryList<'a> {\n #[allow(clippy::too_many_arguments)]\n pub fn new(\n history: &'a [History],\n inverted: bool,\n alternate_highlight: bool,\n now: &'a dyn Fn() -> OffsetDateTime,\n indicator: &'a str,\n theme: &'a Theme,\n history_highlighter: HistoryHighlighter<'a>,\n show_numeric_shortcuts: bool,\n columns: &'a [UiColumn],\n ) -> Self {\n Self {\n history,\n block: None,\n inverted,\n alternate_highlight,\n now,\n indicator,\n theme,\n history_highlighter,\n show_numeric_shortcuts,\n columns,\n }\n }\n\n pub fn block(mut self, block: Block<'a>) -> Self {\n self.block = Some(block);\n self\n }\n\n fn get_items_bounds(&self, selected: usize, offset: usize, height: usize) -> (usize, usize) {\n let offset = offset.min(self.history.len().saturating_sub(1));\n\n let max_scroll_space = height.min(10).min(self.history.len() - selected);\n if offset + height < selected + max_scroll_space {\n let end = selected + max_scroll_space;\n (end - height, end)\n } else if selected < offset {\n (selected, selected + height)\n } else {\n (offset, offset + height)\n }\n }\n}\n\nstruct DrawState<'a> {\n buf: &'a mut Buffer,\n list_area: Rect,\n x: u16,\n y: u16,\n state: &'a ListState,\n inverted: bool,\n alternate_highlight: bool,\n now: &'a dyn Fn() -> OffsetDateTime,\n indicator: &'a str,\n theme: &'a Theme,\n history_highlighter: HistoryHighlighter<'a>,\n show_numeric_shortcuts: bool,\n columns: &'a [UiColumn],\n}\n\n// these encode the slices of `\" > \"`, `\" {n} \"`, or `\" \"` in a compact form.\n// Yes, this is a hack, but it makes me feel happy\nstatic SLICES: &str = \" > 1 2 3 4 5 6 7 8 9 \";\n\nimpl DrawState<'_> {\n /// Render a complete row for a history item based on configured columns.\n fn render_row(&mut self, h: &History) {\n // Always render the indicator first (width 3)\n self.index();\n\n // Calculate the width for the expanding column\n // Fixed columns use their configured width + 1 (trailing space)\n let indicator_width: u16 = 3;\n let fixed_width: u16 = self\n .columns\n .iter()\n .filter(|c| !c.expand)\n .map(|c| c.width + 1)\n .sum();\n let expand_width = self\n .list_area\n .width\n .saturating_sub(indicator_width + fixed_width);\n\n let style = self.theme.as_style(Meaning::Base);\n // Render each configured column\n for (idx, column) in self.columns.iter().enumerate() {\n if idx != 0 {\n self.draw(\" \", Style::from_crossterm(style));\n }\n let width = if column.expand {\n expand_width\n } else {\n column.width\n };\n match column.column_type {\n UiColumnType::Duration => self.duration(h, width),\n UiColumnType::Time => self.time(h, width),\n UiColumnType::Datetime => self.datetime(h, width),\n UiColumnType::Directory => self.directory(h, width),\n UiColumnType::Host => self.host(h, width),\n UiColumnType::User => self.user(h, width),\n UiColumnType::Exit => self.exit_code(h, width),\n UiColumnType::Command => self.command(h),\n }\n }\n }\n\n fn index(&mut self) {\n if !self.show_numeric_shortcuts {\n let i = self.y as usize + self.state.offset;\n let is_selected = i == self.state.selected();\n let prompt: &str = if is_selected { self.indicator } else { \" \" };\n self.draw(prompt, Style::default());\n return;\n }\n\n // these encode the slices of `\" > \"`, `\" {n} \"`, or `\" \"` in a compact form.\n // Yes, this is a hack, but it makes me feel happy\n\n let i = self.y as usize + self.state.offset;\n let i = i.checked_sub(self.state.selected);\n let i = i.unwrap_or(10).min(10) * 2;\n let prompt: &str = if i == 0 {\n self.indicator\n } else {\n &SLICES[i..i + 3]\n };\n self.draw(prompt, Style::default());\n }\n\n fn duration(&mut self, h: &History, width: u16) {\n let style = self.theme.as_style(if h.success() {\n Meaning::AlertInfo\n } else {\n Meaning::AlertError\n });\n let duration = Duration::from_nanos(u64::try_from(h.duration).unwrap_or(0));\n let formatted = format_duration(duration);\n let w = width as usize;\n // Right-align duration within its column width, plus trailing space\n let display = format!(\"{formatted:>w$}\");\n self.draw(&display, Style::from_crossterm(style));\n }\n\n fn time(&mut self, h: &History, width: u16) {\n let style = self.theme.as_style(Meaning::Guidance);\n\n // Account for the chance that h.timestamp is \"in the future\"\n // This would mean that \"since\" is negative, and the unwrap here\n // would fail.\n // If the timestamp would otherwise be in the future, display\n // the time since as 0.\n let since = (self.now)() - h.timestamp;\n let time = format_duration(since.try_into().unwrap_or_default());\n\n // Format as \"Xs ago\" right-aligned within column width\n let w = width as usize;\n let time_str = format!(\"{time} ago\");\n\n let display = format!(\"{time_str:>w$}\");\n self.draw(&display, Style::from_crossterm(style));\n }\n\n fn command(&mut self, h: &History) {\n let mut style = self.theme.as_style(Meaning::Base);\n let mut row_highlighted = false;\n if !self.alternate_highlight && (self.y as usize + self.state.offset == self.state.selected)\n {\n row_highlighted = true;\n // if not applying alternative highlighting to the whole row, color the command\n style = self.theme.as_style(Meaning::AlertError);\n style.attributes.set(style::Attribute::Bold);\n }\n\n let highlight_indices = self.history_highlighter.get_highlight_indices(\n h.command\n .escape_control()\n .split_ascii_whitespace()\n .join(\" \")\n .as_str(),\n );\n\n let mut pos = 0;\n for section in h.command.escape_control().split_ascii_whitespace() {\n if pos != 0 {\n self.draw(\" \", Style::from_crossterm(style));\n }\n for ch in section.chars() {\n if self.x > self.list_area.width {\n // Avoid attempting to draw a command section beyond the width\n // of the list\n return;\n }\n let mut style = style;\n if highlight_indices.contains(&pos) {\n if row_highlighted {\n // if the row is highlighted bold is not enough as the whole row is bold\n // change the color too\n style = self.theme.as_style(Meaning::AlertWarn);\n }\n style.attributes.set(style::Attribute::Bold);\n }\n let s = ch.to_string();\n self.draw(&s, Style::from_crossterm(style));\n pos += s.len();\n }\n pos += 1;\n }\n }\n\n /// Render the absolute datetime column (e.g., \"2025-01-22 14:35\")\n fn datetime(&mut self, h: &History, width: u16) {\n let style = self.theme.as_style(Meaning::Annotation);\n // Format: YYYY-MM-DD HH:MM\n let formatted = h\n .timestamp\n .format(\n &time::format_description::parse(\"[year]-[month]-[day] [hour]:[minute]\")\n .expect(\"valid format\"),\n )\n .unwrap_or_else(|_| \"????-??-?? ??:??\".to_string());\n let w = width as usize;\n let display = format!(\"{formatted:w$}\");\n self.draw(&display, Style::from_crossterm(style));\n }\n\n /// Render the directory column (working directory, truncated)\n fn directory(&mut self, h: &History, width: u16) {\n let style = self.theme.as_style(Meaning::Annotation);\n let w = width as usize;\n let cwd = &h.cwd;\n let char_count = cwd.chars().count();\n // Truncate from the left with \"...\" if too long, plus trailing space\n // Use character count for comparison and skip for UTF-8 safety\n let display = if char_count > w && w >= 4 {\n let truncated: String = cwd.chars().skip(char_count - (w - 3)).collect();\n format!(\"...{truncated}\")\n } else {\n format!(\"{cwd:w$}\")\n };\n self.draw(&display, Style::from_crossterm(style));\n }\n\n /// Render the host column (just the hostname)\n fn host(&mut self, h: &History, width: u16) {\n let style = self.theme.as_style(Meaning::Annotation);\n let w = width as usize;\n // Database stores hostname as \"hostname:username\"\n let host = h.hostname.split(':').next().unwrap_or(&h.hostname);\n let char_count = host.chars().count();\n // Use character count for comparison and take for UTF-8 safety\n let display = if char_count > w && w >= 4 {\n let truncated: String = host.chars().take(w.saturating_sub(4)).collect();\n format!(\"{truncated}...\")\n } else {\n format!(\"{host:w$}\")\n };\n self.draw(&display, Style::from_crossterm(style));\n }\n\n /// Render the user column\n fn user(&mut self, h: &History, width: u16) {\n let style = self.theme.as_style(Meaning::Annotation);\n let w = width as usize;\n // Database stores hostname as \"hostname:username\"\n let user = h.hostname.split(':').nth(1).unwrap_or(\"\");\n let char_count = user.chars().count();\n // Use character count for comparison and take for UTF-8 safety\n let display = if char_count > w && w >= 4 {\n let truncated: String = user.chars().take(w.saturating_sub(4)).collect();\n format!(\"{truncated}...\")\n } else {\n format!(\"{user:w$}\")\n };\n self.draw(&display, Style::from_crossterm(style));\n }\n\n /// Render the exit code column\n fn exit_code(&mut self, h: &History, width: u16) {\n let style = if h.success() {\n self.theme.as_style(Meaning::AlertInfo)\n } else {\n self.theme.as_style(Meaning::AlertError)\n };\n let w = width as usize;\n let display = format!(\"{:>w$}\", h.exit);\n self.draw(&display, Style::from_crossterm(style));\n }\n\n fn draw(&mut self, s: &str, mut style: Style) {\n let cx = self.list_area.left() + self.x;\n\n let cy = if self.inverted {\n self.list_area.top() + self.y\n } else {\n self.list_area.bottom() - self.y - 1\n };\n\n if self.alternate_highlight && (self.y as usize + self.state.offset == self.state.selected)\n {\n style = style.add_modifier(Modifier::REVERSED);\n }\n\n let w = (self.list_area.width - self.x) as usize;\n self.x += self.buf.set_stringn(cx, cy, s, w, style).0 - cx;\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/inspector.rs", "content": "use std::time::Duration;\nuse time::macros::format_description;\n\nuse atuin_client::{\n history::{History, HistoryStats},\n settings::{Settings, Timezone},\n};\nuse ratatui::{\n Frame,\n backend::FromCrossterm,\n layout::Rect,\n prelude::{Constraint, Direction, Layout},\n style::Style,\n text::{Span, Text},\n widgets::{Bar, BarChart, BarGroup, Block, Borders, Padding, Paragraph, Row, Table},\n};\n\nuse super::duration::format_duration;\n\nuse super::super::theme::{Meaning, Theme};\nuse super::interactive::{Compactness, to_compactness};\n\n#[allow(clippy::cast_sign_loss)]\nfn u64_or_zero(num: i64) -> u64 {\n if num < 0 { 0 } else { num as u64 }\n}\n\npub fn draw_commands(\n f: &mut Frame<'_>,\n parent: Rect,\n history: &History,\n stats: &HistoryStats,\n compact: bool,\n theme: &Theme,\n) {\n let commands = Layout::default()\n .direction(if compact {\n Direction::Vertical\n } else {\n Direction::Horizontal\n })\n .constraints(if compact {\n [\n Constraint::Length(1),\n Constraint::Length(1),\n Constraint::Min(0),\n ]\n } else {\n [\n Constraint::Ratio(1, 4),\n Constraint::Ratio(1, 2),\n Constraint::Ratio(1, 4),\n ]\n })\n .split(parent);\n\n let command = Paragraph::new(Text::from(Span::styled(\n history.command.clone(),\n Style::from_crossterm(theme.as_style(Meaning::Important)),\n )))\n .block(if compact {\n Block::new()\n .borders(Borders::NONE)\n .style(Style::from_crossterm(theme.as_style(Meaning::Base)))\n } else {\n Block::new()\n .borders(Borders::ALL)\n .style(Style::from_crossterm(theme.as_style(Meaning::Base)))\n .title(\"Command\")\n .padding(Padding::horizontal(1))\n });\n\n let previous = Paragraph::new(\n stats\n .previous\n .clone()\n .map_or_else(|| \"[No previous command]\".to_string(), |prev| prev.command),\n )\n .block(if compact {\n Block::new()\n .borders(Borders::NONE)\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation)))\n } else {\n Block::new()\n .borders(Borders::ALL)\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation)))\n .title(\"Previous command\")\n .padding(Padding::horizontal(1))\n });\n\n // Add [] around blank text, as when this is shown in a list\n // compacted, it makes it more obviously control text.\n let next = Paragraph::new(\n stats\n .next\n .clone()\n .map_or_else(|| \"[No next command]\".to_string(), |next| next.command),\n )\n .block(if compact {\n Block::new()\n .borders(Borders::NONE)\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation)))\n } else {\n Block::new()\n .borders(Borders::ALL)\n .title(\"Next command\")\n .padding(Padding::horizontal(1))\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation)))\n });\n\n f.render_widget(previous, commands[0]);\n f.render_widget(command, commands[1]);\n f.render_widget(next, commands[2]);\n}\n\npub fn draw_stats_table(\n f: &mut Frame<'_>,\n parent: Rect,\n history: &History,\n tz: Timezone,\n stats: &HistoryStats,\n theme: &Theme,\n) {\n let duration = Duration::from_nanos(u64_or_zero(history.duration));\n let avg_duration = Duration::from_nanos(stats.average_duration);\n let (host, user) = history.hostname.split_once(':').unwrap_or((\"\", \"\"));\n\n let rows = [\n Row::new(vec![\"Host\".to_string(), host.to_string()]),\n Row::new(vec![\"User\".to_string(), user.to_string()]),\n Row::new(vec![\n \"Time\".to_string(),\n history.timestamp.to_offset(tz.0).to_string(),\n ]),\n Row::new(vec![\"Duration\".to_string(), format_duration(duration)]),\n Row::new(vec![\n \"Avg duration\".to_string(),\n format_duration(avg_duration),\n ]),\n Row::new(vec![\"Exit\".to_string(), history.exit.to_string()]),\n Row::new(vec![\"Directory\".to_string(), history.cwd.clone()]),\n Row::new(vec![\"Session\".to_string(), history.session.clone()]),\n Row::new(vec![\"Total runs\".to_string(), stats.total.to_string()]),\n ];\n\n let widths = [Constraint::Ratio(1, 5), Constraint::Ratio(4, 5)];\n\n let table = Table::new(rows, widths).column_spacing(1).block(\n Block::default()\n .title(\"Command stats\")\n .borders(Borders::ALL)\n .style(Style::from_crossterm(theme.as_style(Meaning::Base)))\n .padding(Padding::vertical(1)),\n );\n\n f.render_widget(table, parent);\n}\n\nfn num_to_day(num: &str) -> String {\n match num {\n \"0\" => \"Sunday\".to_string(),\n \"1\" => \"Monday\".to_string(),\n \"2\" => \"Tuesday\".to_string(),\n \"3\" => \"Wednesday\".to_string(),\n \"4\" => \"Thursday\".to_string(),\n \"5\" => \"Friday\".to_string(),\n \"6\" => \"Saturday\".to_string(),\n _ => \"Invalid day\".to_string(),\n }\n}\n\nfn sort_duration_over_time(durations: &[(String, i64)]) -> Vec<(String, i64)> {\n let format = format_description!(\"[day]-[month]-[year]\");\n let output = format_description!(\"[month]/[year repr:last_two]\");\n\n let mut durations: Vec<(time::Date, i64)> = durations\n .iter()\n .map(|d| {\n (\n time::Date::parse(d.0.as_str(), &format).expect(\"invalid date string from sqlite\"),\n d.1,\n )\n })\n .collect();\n\n durations.sort_by(|a, b| a.0.cmp(&b.0));\n\n durations\n .iter()\n .map(|(date, duration)| {\n (\n date.format(output).expect(\"failed to format sqlite date\"),\n *duration,\n )\n })\n .collect()\n}\n\nfn draw_stats_charts(f: &mut Frame<'_>, parent: Rect, stats: &HistoryStats, theme: &Theme) {\n let exits: Vec = stats\n .exits\n .iter()\n .map(|(exit, count)| {\n Bar::default()\n .label(exit.to_string())\n .value(u64_or_zero(*count))\n })\n .collect();\n\n let exits = BarChart::default()\n .block(\n Block::default()\n .title(\"Exit code distribution\")\n .style(Style::from_crossterm(theme.as_style(Meaning::Base)))\n .borders(Borders::ALL),\n )\n .bar_width(3)\n .bar_gap(1)\n .bar_style(Style::default())\n .value_style(Style::default())\n .label_style(Style::default())\n .data(BarGroup::default().bars(&exits));\n\n let day_of_week: Vec = stats\n .day_of_week\n .iter()\n .map(|(day, count)| {\n Bar::default()\n .label(num_to_day(day.as_str()))\n .value(u64_or_zero(*count))\n })\n .collect();\n\n let day_of_week = BarChart::default()\n .block(\n Block::default()\n .title(\"Runs per day\")\n .style(Style::from_crossterm(theme.as_style(Meaning::Base)))\n .borders(Borders::ALL),\n )\n .bar_width(3)\n .bar_gap(1)\n .bar_style(Style::default())\n .value_style(Style::default())\n .label_style(Style::default())\n .data(BarGroup::default().bars(&day_of_week));\n\n let duration_over_time = sort_duration_over_time(&stats.duration_over_time);\n let duration_over_time: Vec = duration_over_time\n .iter()\n .map(|(date, duration)| {\n let d = Duration::from_nanos(u64_or_zero(*duration));\n Bar::default()\n .label(date.clone())\n .value(u64_or_zero(*duration))\n .text_value(format_duration(d))\n })\n .collect();\n\n let duration_over_time = BarChart::default()\n .block(\n Block::default()\n .title(\"Duration over time\")\n .style(Style::from_crossterm(theme.as_style(Meaning::Base)))\n .borders(Borders::ALL),\n )\n .bar_width(5)\n .bar_gap(1)\n .bar_style(Style::default())\n .value_style(Style::default())\n .label_style(Style::default())\n .data(BarGroup::default().bars(&duration_over_time));\n\n let layout = Layout::default()\n .direction(Direction::Vertical)\n .constraints([\n Constraint::Ratio(1, 3),\n Constraint::Ratio(1, 3),\n Constraint::Ratio(1, 3),\n ])\n .split(parent);\n\n f.render_widget(exits, layout[0]);\n f.render_widget(day_of_week, layout[1]);\n f.render_widget(duration_over_time, layout[2]);\n}\n\npub fn draw(\n f: &mut Frame<'_>,\n chunk: Rect,\n history: &History,\n stats: &HistoryStats,\n settings: &Settings,\n theme: &Theme,\n tz: Timezone,\n) {\n let compactness = to_compactness(f, settings);\n\n match compactness {\n Compactness::Ultracompact => draw_ultracompact(f, chunk, history, stats, theme),\n _ => draw_full(f, chunk, history, stats, theme, tz),\n }\n}\n\npub fn draw_ultracompact(\n f: &mut Frame<'_>,\n chunk: Rect,\n history: &History,\n stats: &HistoryStats,\n theme: &Theme,\n) {\n draw_commands(f, chunk, history, stats, true, theme);\n}\n\npub fn draw_full(\n f: &mut Frame<'_>,\n chunk: Rect,\n history: &History,\n stats: &HistoryStats,\n theme: &Theme,\n tz: Timezone,\n) {\n let vert_layout = Layout::default()\n .direction(Direction::Vertical)\n .constraints([Constraint::Ratio(1, 5), Constraint::Ratio(4, 5)])\n .split(chunk);\n\n let stats_layout = Layout::default()\n .direction(Direction::Horizontal)\n .constraints([Constraint::Ratio(1, 3), Constraint::Ratio(2, 3)])\n .split(vert_layout[1]);\n\n draw_commands(f, vert_layout[0], history, stats, false, theme);\n draw_stats_table(f, stats_layout[0], history, tz, stats, theme);\n draw_stats_charts(f, stats_layout[1], stats, theme);\n}\n\n#[cfg(test)]\nmod tests {\n use super::draw_ultracompact;\n use atuin_client::{\n history::{History, HistoryId, HistoryStats},\n theme::ThemeManager,\n };\n use ratatui::{backend::TestBackend, prelude::*};\n use time::OffsetDateTime;\n\n fn mock_history_stats() -> (History, HistoryStats) {\n let history = History {\n id: HistoryId::from(\"test1\".to_string()),\n timestamp: OffsetDateTime::now_utc(),\n duration: 3,\n exit: 0,\n command: \"/bin/cmd\".to_string(),\n cwd: \"/toot\".to_string(),\n session: \"sesh1\".to_string(),\n hostname: \"hostn\".to_string(),\n author: \"hostn\".to_string(),\n intent: None,\n deleted_at: None,\n };\n let next = History {\n id: HistoryId::from(\"test2\".to_string()),\n timestamp: OffsetDateTime::now_utc(),\n duration: 2,\n exit: 0,\n command: \"/bin/cmd -os\".to_string(),\n cwd: \"/toot\".to_string(),\n session: \"sesh1\".to_string(),\n hostname: \"hostn\".to_string(),\n author: \"hostn\".to_string(),\n intent: None,\n deleted_at: None,\n };\n let prev = History {\n id: HistoryId::from(\"test3\".to_string()),\n timestamp: OffsetDateTime::now_utc(),\n duration: 1,\n exit: 0,\n command: \"/bin/cmd -a\".to_string(),\n cwd: \"/toot\".to_string(),\n session: \"sesh1\".to_string(),\n hostname: \"hostn\".to_string(),\n author: \"hostn\".to_string(),\n intent: None,\n deleted_at: None,\n };\n let stats = HistoryStats {\n next: Some(next.clone()),\n previous: Some(prev.clone()),\n total: 2,\n average_duration: 3,\n exits: Vec::new(),\n day_of_week: Vec::new(),\n duration_over_time: Vec::new(),\n };\n (history, stats)\n }\n\n #[test]\n fn test_output_looks_correct_for_ultracompact() {\n let backend = TestBackend::new(22, 5);\n let mut terminal = Terminal::new(backend).expect(\"Could not create terminal\");\n let chunk = Rect::new(0, 0, 22, 5);\n let (history, stats) = mock_history_stats();\n let prev = stats.previous.clone().unwrap();\n let next = stats.next.clone().unwrap();\n\n let mut manager = ThemeManager::new(Some(true), Some(\"\".to_string()));\n let theme = manager.load_theme(\"(none)\", None);\n let _ = terminal.draw(|f| draw_ultracompact(f, chunk, &history, &stats, &theme));\n let mut lines = [\" \"; 5].map(|l| Line::from(l));\n for (n, entry) in [prev, history, next].iter().enumerate() {\n let mut l = lines[n].to_string();\n l.replace_range(0..entry.command.len(), &entry.command);\n lines[n] = Line::from(l);\n }\n\n terminal.backend().assert_buffer_lines(lines);\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/interactive.rs", "content": "use std::{\n io::{IsTerminal, Write, stdout},\n time::Duration,\n};\n\n#[cfg(unix)]\nuse std::io::Read as _;\n\nuse atuin_common::{shell::Shell, utils::Escapable as _};\nuse eyre::Result;\nuse futures_util::FutureExt;\nuse semver::Version;\nuse time::OffsetDateTime;\nuse unicode_width::UnicodeWidthStr;\n\nuse super::{\n cursor::Cursor,\n engines::{SearchEngine, SearchState},\n history_list::{HistoryList, ListState},\n};\nuse atuin_client::{\n database::{Context, Database, current_context},\n history::{History, HistoryId, HistoryStats, store::HistoryStore},\n settings::{\n CursorStyle, ExitMode, FilterMode, KeymapMode, PreviewStrategy, SearchMode, Settings,\n UiColumn,\n },\n};\n\nuse crate::command::client::search::history_list::HistoryHighlighter;\nuse crate::command::client::search::keybindings::KeymapSet;\nuse crate::command::client::theme::{Meaning, Theme};\nuse crate::{VERSION, command::client::search::engines};\n\nuse ratatui::{\n Frame, Terminal, TerminalOptions, Viewport,\n backend::{CrosstermBackend, FromCrossterm},\n crossterm::{\n cursor::SetCursorStyle,\n event::{self, Event, KeyEvent, MouseEvent},\n execute, queue, terminal,\n },\n layout::{Alignment, Constraint, Direction, Layout},\n prelude::*,\n style::{Modifier, Style},\n text::{Line, Span, Text},\n widgets::{Block, BorderType, Borders, Clear, Padding, Paragraph, Tabs},\n};\n\n#[cfg(not(target_os = \"windows\"))]\nuse ratatui::crossterm::event::{\n KeyboardEnhancementFlags, PopKeyboardEnhancementFlags, PushKeyboardEnhancementFlags,\n};\n\nconst TAB_TITLES: [&str; 2] = [\"Search\", \"Inspect\"];\n\npub enum InputAction {\n Accept(usize),\n AcceptInspecting,\n Copy(usize),\n Delete(usize),\n ReturnOriginal,\n ReturnQuery,\n Continue,\n Redraw,\n SwitchContext(Option),\n}\n\n#[derive(Clone)]\npub struct InspectingState {\n current: Option,\n next: Option,\n previous: Option,\n}\n\nimpl InspectingState {\n pub fn move_to_previous(&mut self) {\n let previous = self.previous.clone();\n self.reset();\n self.current = previous;\n }\n\n pub fn move_to_next(&mut self) {\n let next = self.next.clone();\n self.reset();\n self.current = next;\n }\n\n pub fn reset(&mut self) {\n self.current = None;\n self.next = None;\n self.previous = None;\n }\n}\n\npub fn to_compactness(f: &Frame, settings: &Settings) -> Compactness {\n if match settings.style {\n atuin_client::settings::Style::Auto => f.area().height < 14,\n atuin_client::settings::Style::Compact => true,\n atuin_client::settings::Style::Full => false,\n } {\n if settings.auto_hide_height != 0 && f.area().height <= settings.auto_hide_height {\n Compactness::Ultracompact\n } else {\n Compactness::Compact\n }\n } else {\n Compactness::Full\n }\n}\n\n#[allow(clippy::struct_field_names)]\n#[allow(clippy::struct_excessive_bools)]\npub struct State {\n history_count: i64,\n update_needed: Option,\n results_state: ListState,\n switched_search_mode: bool,\n search_mode: SearchMode,\n results_len: usize,\n accept: bool,\n keymap_mode: KeymapMode,\n prefix: bool,\n current_cursor: Option,\n tab_index: usize,\n pending_vim_key: Option,\n original_input_empty: bool,\n\n pub inspecting_state: InspectingState,\n\n keymaps: KeymapSet,\n search: SearchState,\n engine: Box,\n now: Box OffsetDateTime + Send>,\n}\n\n#[derive(Clone, Copy)]\npub enum Compactness {\n Ultracompact,\n Compact,\n Full,\n}\n\n#[derive(Clone, Copy)]\nstruct StyleState {\n compactness: Compactness,\n invert: bool,\n inner_width: usize,\n}\n\nimpl State {\n async fn query_results(\n &mut self,\n db: &mut dyn Database,\n smart_sort: bool,\n ) -> Result> {\n let results = self.engine.query(&self.search, db).await?;\n\n self.inspecting_state = InspectingState {\n current: None,\n next: None,\n previous: None,\n };\n self.results_state.select(0);\n self.results_len = results.len();\n\n if smart_sort {\n Ok(atuin_history::sort::sort(\n self.search.input.as_str(),\n results,\n ))\n } else {\n Ok(results)\n }\n }\n\n fn handle_input(&mut self, settings: &Settings, input: &Event) -> InputAction {\n match input {\n Event::Key(k) => self.handle_key_input(settings, k),\n Event::Mouse(m) => self.handle_mouse_input(*m),\n Event::Paste(d) => self.handle_paste_input(d),\n _ => InputAction::Continue,\n }\n }\n\n fn handle_mouse_input(&mut self, input: MouseEvent) -> InputAction {\n match input.kind {\n event::MouseEventKind::ScrollDown => {\n self.scroll_down(1);\n }\n event::MouseEventKind::ScrollUp => {\n self.scroll_up(1);\n }\n _ => {}\n }\n InputAction::Continue\n }\n\n fn handle_paste_input(&mut self, input: &str) -> InputAction {\n for i in input.chars() {\n self.search.input.insert(i);\n }\n InputAction::Continue\n }\n\n fn cast_cursor_style(style: CursorStyle) -> SetCursorStyle {\n match style {\n CursorStyle::DefaultUserShape => SetCursorStyle::DefaultUserShape,\n CursorStyle::BlinkingBlock => SetCursorStyle::BlinkingBlock,\n CursorStyle::SteadyBlock => SetCursorStyle::SteadyBlock,\n CursorStyle::BlinkingUnderScore => SetCursorStyle::BlinkingUnderScore,\n CursorStyle::SteadyUnderScore => SetCursorStyle::SteadyUnderScore,\n CursorStyle::BlinkingBar => SetCursorStyle::BlinkingBar,\n CursorStyle::SteadyBar => SetCursorStyle::SteadyBar,\n }\n }\n\n fn set_keymap_cursor(&mut self, settings: &Settings, keymap_name: &str) {\n let cursor_style = if keymap_name == \"__clear__\" {\n None\n } else {\n settings.keymap_cursor.get(keymap_name).copied()\n }\n .or_else(|| self.current_cursor.map(|_| CursorStyle::DefaultUserShape));\n\n if cursor_style != self.current_cursor\n && let Some(style) = cursor_style\n {\n self.current_cursor = cursor_style;\n let _ = execute!(stdout(), Self::cast_cursor_style(style));\n }\n }\n\n pub fn initialize_keymap_cursor(&mut self, settings: &Settings) {\n match self.keymap_mode {\n KeymapMode::Emacs => self.set_keymap_cursor(settings, \"emacs\"),\n KeymapMode::VimNormal => self.set_keymap_cursor(settings, \"vim_normal\"),\n KeymapMode::VimInsert => self.set_keymap_cursor(settings, \"vim_insert\"),\n KeymapMode::Auto => {}\n }\n }\n\n pub fn finalize_keymap_cursor(&mut self, settings: &Settings) {\n match settings.keymap_mode_shell {\n KeymapMode::Emacs => self.set_keymap_cursor(settings, \"emacs\"),\n KeymapMode::VimNormal => self.set_keymap_cursor(settings, \"vim_normal\"),\n KeymapMode::VimInsert => self.set_keymap_cursor(settings, \"vim_insert\"),\n KeymapMode::Auto => self.set_keymap_cursor(settings, \"__clear__\"),\n }\n }\n\n fn handle_key_exit(settings: &Settings) -> InputAction {\n match settings.exit_mode {\n ExitMode::ReturnOriginal => InputAction::ReturnOriginal,\n ExitMode::ReturnQuery => InputAction::ReturnQuery,\n }\n }\n\n /// Select the keymap for the current mode (ignoring prefix).\n fn mode_keymap(&self) -> &super::keybindings::Keymap {\n if self.tab_index == 1 {\n &self.keymaps.inspector\n } else {\n match self.keymap_mode {\n KeymapMode::Emacs | KeymapMode::Auto => &self.keymaps.emacs,\n KeymapMode::VimNormal => &self.keymaps.vim_normal,\n KeymapMode::VimInsert => &self.keymaps.vim_insert,\n }\n }\n }\n\n /// Whether the current mode supports character insertion on unmatched keys.\n fn is_insert_mode(&self) -> bool {\n matches!(\n self.keymap_mode,\n KeymapMode::Emacs | KeymapMode::Auto | KeymapMode::VimInsert\n )\n }\n\n fn handle_key_input(&mut self, settings: &Settings, input: &KeyEvent) -> InputAction {\n use super::keybindings::Action;\n use super::keybindings::EvalContext;\n use super::keybindings::key::{KeyCodeValue, KeyInput, SingleKey};\n\n // Skip release events\n if input.kind == event::KeyEventKind::Release {\n return InputAction::Continue;\n }\n\n // Reset switched_search_mode at start of each key event\n self.switched_search_mode = false;\n\n // Build evaluation context from current state\n let ctx = EvalContext {\n cursor_position: self.search.input.position(),\n input_width: UnicodeWidthStr::width(self.search.input.as_str()),\n input_byte_len: self.search.input.as_str().len(),\n selected_index: self.results_state.selected(),\n results_len: self.results_len,\n original_input_empty: self.original_input_empty,\n has_context: self.search.custom_context.is_some(),\n };\n\n // Convert KeyEvent to SingleKey\n let Some(single) = SingleKey::from_event(input) else {\n return InputAction::Continue;\n };\n\n // --- Phase 1: Resolve (take pending key first, then immutable borrows) ---\n\n // Take pending key before any immutable borrows of self\n let pending = self.pending_vim_key.take();\n\n // If in prefix mode, try prefix keymap first (single keys only)\n let prefix_action = if self.prefix {\n let ki = KeyInput::Single(single.clone());\n self.keymaps.prefix.resolve(&ki, &ctx)\n } else {\n None\n };\n\n // The if-let/else-if chain here is clearer than map_or_else with nested closures.\n #[allow(clippy::option_if_let_else)]\n let (action, new_pending) = if prefix_action.is_some() {\n (prefix_action, None)\n } else {\n // Use mode keymap (handles both single and multi-key sequences)\n let keymap = self.mode_keymap();\n\n if let Some(pending_char) = pending {\n // We have a pending key from a previous press (e.g., first 'g' of 'gg')\n let pending_single = SingleKey {\n code: KeyCodeValue::Char(pending_char),\n ctrl: false,\n alt: false,\n shift: false,\n super_key: false,\n };\n let seq = KeyInput::Sequence(vec![pending_single, single.clone()]);\n let action = keymap\n .resolve(&seq, &ctx)\n .or_else(|| keymap.resolve(&KeyInput::Single(single.clone()), &ctx));\n (action, None)\n } else if keymap.has_sequence_starting_with(&single)\n && matches!(single.code, KeyCodeValue::Char(_))\n && !single.ctrl\n && !single.alt\n {\n // This key starts a multi-key sequence; wait for next key\n let KeyCodeValue::Char(c) = single.code else {\n unreachable!()\n };\n (Some(Action::Noop), Some(c))\n } else {\n (\n keymap.resolve(&KeyInput::Single(single.clone()), &ctx),\n None,\n )\n }\n };\n\n // --- Phase 2: Apply mutations ---\n self.pending_vim_key = new_pending;\n\n // Reset prefix (before execute, so EnterPrefixMode can re-set it)\n self.prefix = false;\n\n if let Some(action) = action {\n self.execute_action(&action, settings)\n } else {\n // No action matched. In insert-capable modes, insert the character.\n if self.is_insert_mode() && !single.ctrl && !single.alt {\n match single.code {\n KeyCodeValue::Char(c) => {\n self.search.input.insert(c);\n }\n KeyCodeValue::Space => {\n self.search.input.insert(' ');\n }\n _ => {}\n }\n }\n InputAction::Continue\n }\n }\n\n fn scroll_down(&mut self, scroll_len: usize) {\n let i = self.results_state.selected().saturating_sub(scroll_len);\n self.inspecting_state.reset();\n self.results_state.select(i);\n }\n\n fn scroll_up(&mut self, scroll_len: usize) {\n let i = self.results_state.selected() + scroll_len;\n self.results_state\n .select(i.min(self.results_len.saturating_sub(1)));\n self.inspecting_state.reset();\n }\n\n /// Execute a resolved action, performing all side effects and returning the\n /// appropriate `InputAction` for the event loop.\n ///\n /// This is the \"do it\" half of the resolve+execute pipeline. The resolver\n /// decides *what* to do (which `Action`), and this function carries it out.\n ///\n /// Invert handling: scroll actions (`SelectNext`, `ScrollPageDown`, etc.) account\n /// for `settings.invert` so that keybindings are always in \"visual\" terms —\n /// users never need to think about invert in their keybinding config.\n #[allow(clippy::too_many_lines)]\n pub(crate) fn execute_action(\n &mut self,\n action: &super::keybindings::Action,\n settings: &Settings,\n ) -> InputAction {\n use crate::command::client::search::keybindings::Action;\n\n match action {\n // -- Cursor movement --\n Action::CursorLeft => {\n self.search.input.left();\n InputAction::Continue\n }\n Action::CursorRight => {\n self.search.input.right();\n InputAction::Continue\n }\n Action::CursorWordLeft => {\n self.search\n .input\n .prev_word(&settings.word_chars, settings.word_jump_mode);\n InputAction::Continue\n }\n Action::CursorWordRight => {\n self.search\n .input\n .next_word(&settings.word_chars, settings.word_jump_mode);\n InputAction::Continue\n }\n Action::CursorWordEnd => {\n self.search.input.word_end(&settings.word_chars);\n InputAction::Continue\n }\n Action::CursorStart => {\n self.search.input.start();\n InputAction::Continue\n }\n Action::CursorEnd => {\n self.search.input.end();\n InputAction::Continue\n }\n\n // -- Editing --\n Action::DeleteCharBefore => {\n self.search.input.back();\n InputAction::Continue\n }\n Action::DeleteCharAfter => {\n self.search.input.remove();\n InputAction::Continue\n }\n Action::DeleteWordBefore => {\n self.search\n .input\n .remove_prev_word(&settings.word_chars, settings.word_jump_mode);\n InputAction::Continue\n }\n Action::DeleteWordAfter => {\n self.search\n .input\n .remove_next_word(&settings.word_chars, settings.word_jump_mode);\n InputAction::Continue\n }\n Action::DeleteToWordBoundary => {\n // ctrl-w: remove trailing whitespace, then delete to word boundary\n while matches!(self.search.input.back(), Some(c) if c.is_whitespace()) {}\n while self.search.input.left() {\n if self.search.input.char().unwrap().is_whitespace() {\n self.search.input.right();\n break;\n }\n self.search.input.remove();\n }\n InputAction::Continue\n }\n Action::ClearLine => {\n self.search.input.clear();\n InputAction::Continue\n }\n Action::ClearToStart => {\n self.search.input.clear_to_start();\n InputAction::Continue\n }\n Action::ClearToEnd => {\n self.search.input.clear_to_end();\n InputAction::Continue\n }\n\n // -- List navigation (invert-aware) --\n Action::SelectNext => {\n if settings.invert {\n self.scroll_up(1);\n } else {\n self.scroll_down(1);\n }\n InputAction::Continue\n }\n Action::SelectPrevious => {\n if settings.invert {\n self.scroll_down(1);\n } else {\n self.scroll_up(1);\n }\n InputAction::Continue\n }\n // -- Page/half-page scroll (invert-aware) --\n Action::ScrollHalfPageUp => {\n let scroll_len = self\n .results_state\n .max_entries()\n .saturating_sub(settings.scroll_context_lines)\n / 2;\n if settings.invert {\n self.scroll_down(scroll_len);\n } else {\n self.scroll_up(scroll_len);\n }\n InputAction::Continue\n }\n Action::ScrollHalfPageDown => {\n let scroll_len = self\n .results_state\n .max_entries()\n .saturating_sub(settings.scroll_context_lines)\n / 2;\n if settings.invert {\n self.scroll_up(scroll_len);\n } else {\n self.scroll_down(scroll_len);\n }\n InputAction::Continue\n }\n Action::ScrollPageUp => {\n let scroll_len = self\n .results_state\n .max_entries()\n .saturating_sub(settings.scroll_context_lines);\n if settings.invert {\n self.scroll_down(scroll_len);\n } else {\n self.scroll_up(scroll_len);\n }\n InputAction::Continue\n }\n Action::ScrollPageDown => {\n let scroll_len = self\n .results_state\n .max_entries()\n .saturating_sub(settings.scroll_context_lines);\n if settings.invert {\n self.scroll_up(scroll_len);\n } else {\n self.scroll_down(scroll_len);\n }\n InputAction::Continue\n }\n\n // -- Absolute jumps (invert-aware) --\n Action::ScrollToTop => {\n // Visual top of history\n if settings.invert {\n self.results_state.select(0);\n } else {\n let last_idx = self.results_len.saturating_sub(1);\n self.results_state.select(last_idx);\n }\n self.inspecting_state.reset();\n InputAction::Continue\n }\n Action::ScrollToBottom => {\n // Visual bottom of history\n if settings.invert {\n let last_idx = self.results_len.saturating_sub(1);\n self.results_state.select(last_idx);\n } else {\n self.results_state.select(0);\n }\n self.inspecting_state.reset();\n InputAction::Continue\n }\n Action::ScrollToScreenTop => {\n // H — jump to top of visible screen\n let top = self.results_state.offset();\n let visible = self.results_state.max_entries().min(self.results_len);\n let bottom = top + visible.saturating_sub(1);\n self.results_state\n .select(bottom.min(self.results_len.saturating_sub(1)));\n self.inspecting_state.reset();\n InputAction::Continue\n }\n Action::ScrollToScreenMiddle => {\n // M — jump to middle of visible screen\n let top = self.results_state.offset();\n let visible = self.results_state.max_entries().min(self.results_len);\n let middle = top + visible / 2;\n self.results_state\n .select(middle.min(self.results_len.saturating_sub(1)));\n self.inspecting_state.reset();\n InputAction::Continue\n }\n Action::ScrollToScreenBottom => {\n // L — jump to bottom of visible screen\n let top_visible = self.results_state.offset();\n self.results_state.select(top_visible);\n self.inspecting_state.reset();\n InputAction::Continue\n }\n\n // -- Commands --\n Action::Accept => {\n if self.tab_index == 1 {\n return InputAction::AcceptInspecting;\n }\n self.accept = true;\n InputAction::Accept(self.results_state.selected())\n }\n Action::AcceptNth(n) => {\n self.accept = true;\n InputAction::Accept(self.results_state.selected() + *n as usize)\n }\n Action::ReturnSelection => {\n if self.tab_index == 1 {\n return InputAction::AcceptInspecting;\n }\n InputAction::Accept(self.results_state.selected())\n }\n Action::ReturnSelectionNth(n) => {\n InputAction::Accept(self.results_state.selected() + *n as usize)\n }\n Action::Copy => InputAction::Copy(self.results_state.selected()),\n Action::Delete => InputAction::Delete(self.results_state.selected()),\n Action::ReturnOriginal => InputAction::ReturnOriginal,\n Action::ReturnQuery => InputAction::ReturnQuery,\n Action::Exit => Self::handle_key_exit(settings),\n Action::Redraw => InputAction::Redraw,\n Action::CycleFilterMode => {\n self.search.rotate_filter_mode(settings, 1);\n InputAction::Continue\n }\n Action::CycleSearchMode => {\n self.switched_search_mode = true;\n self.search_mode = self.search_mode.next(settings);\n self.engine = engines::engine(self.search_mode, settings);\n InputAction::Continue\n }\n Action::SwitchContext => {\n InputAction::SwitchContext(Some(self.results_state.selected()))\n }\n Action::ClearContext => InputAction::SwitchContext(None),\n Action::ToggleTab => {\n self.tab_index = (self.tab_index + 1) % TAB_TITLES.len();\n InputAction::Continue\n }\n\n // -- Mode changes --\n Action::VimEnterNormal => {\n self.set_keymap_cursor(settings, \"vim_normal\");\n self.keymap_mode = KeymapMode::VimNormal;\n InputAction::Continue\n }\n Action::VimEnterInsert => {\n self.set_keymap_cursor(settings, \"vim_insert\");\n self.keymap_mode = KeymapMode::VimInsert;\n InputAction::Continue\n }\n Action::VimEnterInsertAfter => {\n self.search.input.right();\n self.set_keymap_cursor(settings, \"vim_insert\");\n self.keymap_mode = KeymapMode::VimInsert;\n InputAction::Continue\n }\n Action::VimEnterInsertAtStart => {\n self.search.input.start();\n self.set_keymap_cursor(settings, \"vim_insert\");\n self.keymap_mode = KeymapMode::VimInsert;\n InputAction::Continue\n }\n Action::VimEnterInsertAtEnd => {\n self.search.input.end();\n self.set_keymap_cursor(settings, \"vim_insert\");\n self.keymap_mode = KeymapMode::VimInsert;\n InputAction::Continue\n }\n Action::VimSearchInsert => {\n self.search.input.clear();\n self.set_keymap_cursor(settings, \"vim_insert\");\n self.keymap_mode = KeymapMode::VimInsert;\n InputAction::Continue\n }\n Action::VimChangeToEnd => {\n self.search.input.clear_to_end();\n self.set_keymap_cursor(settings, \"vim_insert\");\n self.keymap_mode = KeymapMode::VimInsert;\n InputAction::Continue\n }\n Action::EnterPrefixMode => {\n self.prefix = true;\n InputAction::Continue\n }\n\n // -- Inspector --\n Action::InspectPrevious => {\n self.inspecting_state.move_to_previous();\n InputAction::Redraw\n }\n Action::InspectNext => {\n self.inspecting_state.move_to_next();\n InputAction::Redraw\n }\n\n // -- Special --\n Action::Noop => InputAction::Continue,\n }\n }\n\n #[allow(clippy::cast_possible_truncation)]\n #[allow(clippy::bool_to_int_with_if)]\n fn calc_preview_height(\n settings: &Settings,\n results: &[History],\n selected: usize,\n tab_index: usize,\n compactness: Compactness,\n border_size: u16,\n preview_width: u16,\n ) -> u16 {\n if settings.show_preview\n && settings.preview.strategy == PreviewStrategy::Auto\n && tab_index == 0\n && !results.is_empty()\n {\n let length_current_cmd = results[selected].command.len() as u16;\n // calculate the number of newlines in the command\n let num_newlines = results[selected]\n .command\n .chars()\n .filter(|&c| c == '\\n')\n .count() as u16;\n if num_newlines > 0 {\n std::cmp::min(\n settings.max_preview_height,\n results[selected]\n .command\n .split('\\n')\n .map(|line| {\n (line.len() as u16 + preview_width - 1 - border_size)\n / (preview_width - border_size)\n })\n .sum(),\n ) + border_size * 2\n }\n // The '- 19' takes the characters before the command (duration and time) into account\n else if length_current_cmd > preview_width - 19 {\n std::cmp::min(\n settings.max_preview_height,\n (length_current_cmd + preview_width - 1 - border_size)\n / (preview_width - border_size),\n ) + border_size * 2\n } else {\n 1\n }\n } else if settings.show_preview\n && settings.preview.strategy == PreviewStrategy::Static\n && tab_index == 0\n {\n let longest_command = results\n .iter()\n .max_by(|h1, h2| h1.command.len().cmp(&h2.command.len()));\n longest_command.map_or(0, |v| {\n std::cmp::min(\n settings.max_preview_height,\n v.command\n .split('\\n')\n .map(|line| {\n (line.len() as u16 + preview_width - 1 - border_size)\n / (preview_width - border_size)\n })\n .sum(),\n )\n }) + border_size * 2\n } else if settings.show_preview && settings.preview.strategy == PreviewStrategy::Fixed {\n settings.max_preview_height + border_size * 2\n } else if !matches!(compactness, Compactness::Full) || tab_index == 1 {\n 0\n } else {\n 1\n }\n }\n\n #[allow(clippy::bool_to_int_with_if)]\n #[allow(clippy::too_many_lines)]\n #[allow(clippy::too_many_arguments)]\n fn draw(\n &mut self,\n f: &mut Frame,\n results: &[History],\n stats: Option,\n inspecting: Option<&History>,\n settings: &Settings,\n theme: &Theme,\n popup_mode: bool,\n ) {\n let area = f.area();\n if popup_mode {\n f.render_widget(Clear, area);\n }\n self.draw_inner(f, area, results, stats, inspecting, settings, theme);\n }\n\n #[allow(clippy::too_many_arguments)]\n #[allow(clippy::too_many_lines)]\n #[allow(clippy::bool_to_int_with_if)]\n fn draw_inner(\n &mut self,\n f: &mut Frame,\n area: Rect,\n results: &[History],\n stats: Option,\n inspecting: Option<&History>,\n settings: &Settings,\n theme: &Theme,\n ) {\n let compactness = to_compactness(f, settings);\n let invert = settings.invert;\n let border_size = match compactness {\n Compactness::Full => 1,\n _ => 0,\n };\n let preview_width = area.width.saturating_sub(2);\n let preview_height = Self::calc_preview_height(\n settings,\n results,\n self.results_state.selected(),\n self.tab_index,\n compactness,\n border_size,\n preview_width,\n );\n let show_help =\n settings.show_help && (matches!(compactness, Compactness::Full) || area.height > 1);\n // This is an OR, as it seems more likely for someone to wish to override\n // tabs unexpectedly being missed, than unexpectedly present.\n let show_tabs = settings.show_tabs && !matches!(compactness, Compactness::Ultracompact);\n let chunks = Layout::default()\n .direction(Direction::Vertical)\n .margin(0)\n .horizontal_margin(1)\n .constraints::<&[Constraint]>(\n if invert {\n [\n Constraint::Length(1 + border_size), // input\n Constraint::Min(1), // results list\n Constraint::Length(preview_height), // preview\n Constraint::Length(if show_tabs { 1 } else { 0 }), // tabs\n Constraint::Length(if show_help { 1 } else { 0 }), // header (sic)\n ]\n } else {\n match compactness {\n Compactness::Ultracompact => [\n Constraint::Length(if show_help { 1 } else { 0 }), // header\n Constraint::Length(0), // tabs\n Constraint::Min(1), // results list\n Constraint::Length(0),\n Constraint::Length(0),\n ],\n _ => [\n Constraint::Length(if show_help { 1 } else { 0 }), // header\n Constraint::Length(if show_tabs { 1 } else { 0 }), // tabs\n Constraint::Min(1), // results list\n Constraint::Length(1 + border_size), // input\n Constraint::Length(preview_height), // preview\n ],\n }\n }\n .as_ref(),\n )\n .split(area);\n\n let input_chunk = if invert { chunks[0] } else { chunks[3] };\n let results_list_chunk = if invert { chunks[1] } else { chunks[2] };\n let preview_chunk = if invert { chunks[2] } else { chunks[4] };\n let tabs_chunk = if invert { chunks[3] } else { chunks[1] };\n let header_chunk = if invert { chunks[4] } else { chunks[0] };\n\n // TODO: this should be split so that we have one interactive search container that is\n // EITHER a search box or an inspector. But I'm not doing that now, way too much atm.\n // also allocate less 🙈\n let titles: Vec<_> = TAB_TITLES.iter().copied().map(Line::from).collect();\n\n if show_tabs {\n let tabs = Tabs::new(titles)\n .block(Block::default().borders(Borders::NONE))\n .select(self.tab_index)\n .style(Style::default())\n .highlight_style(Style::from_crossterm(theme.as_style(Meaning::Important)));\n\n f.render_widget(tabs, tabs_chunk);\n }\n\n let style = StyleState {\n compactness,\n invert,\n inner_width: input_chunk.width.into(),\n };\n\n let header_chunks = Layout::default()\n .direction(Direction::Horizontal)\n .constraints::<&[Constraint]>(\n [\n Constraint::Ratio(1, 5),\n Constraint::Ratio(3, 5),\n Constraint::Ratio(1, 5),\n ]\n .as_ref(),\n )\n .split(header_chunk);\n\n let title = self.build_title(theme);\n f.render_widget(title, header_chunks[0]);\n\n let help = self.build_help(settings, theme);\n f.render_widget(help, header_chunks[1]);\n\n let stats_tab = self.build_stats(theme);\n f.render_widget(stats_tab, header_chunks[2]);\n\n let indicator: String = match compactness {\n Compactness::Ultracompact => {\n if self.switched_search_mode {\n format!(\"S{}>\", self.search_mode.as_str().chars().next().unwrap())\n } else if self.search.custom_context.is_some() {\n format!(\n \"C{}>\",\n self.search.filter_mode.as_str().chars().next().unwrap()\n )\n } else {\n format!(\n \"{}> \",\n self.search.filter_mode.as_str().chars().next().unwrap()\n )\n }\n }\n _ => \" > \".to_string(),\n };\n\n match self.tab_index {\n 0 => {\n let history_highlighter = HistoryHighlighter {\n engine: self.engine.as_ref(),\n search_input: self.search.input.as_str(),\n };\n let results_list = Self::build_results_list(\n style,\n results,\n self.keymap_mode,\n &self.now,\n indicator.as_str(),\n theme,\n history_highlighter,\n settings.show_numeric_shortcuts,\n &settings.ui.columns,\n );\n f.render_stateful_widget(results_list, results_list_chunk, &mut self.results_state);\n }\n\n 1 => {\n if results.is_empty() {\n let message = Paragraph::new(\"Nothing to inspect\")\n .block(\n Block::new()\n .title(Line::from(\" Info \".to_string()))\n .title_alignment(Alignment::Center)\n .borders(Borders::ALL)\n .padding(Padding::vertical(2)),\n )\n .alignment(Alignment::Center);\n f.render_widget(message, results_list_chunk);\n } else {\n let inspecting = match inspecting {\n Some(inspecting) => inspecting,\n None => &results[self.results_state.selected()],\n };\n super::inspector::draw(\n f,\n results_list_chunk,\n inspecting,\n &stats.expect(\"Drawing inspector, but no stats\"),\n settings,\n theme,\n settings.timezone,\n );\n }\n\n // HACK: I'm following up with abstracting this into the UI container, with a\n // sub-widget for search + for inspector\n let feedback = Paragraph::new(\n \"The inspector is new - please give feedback (good, or bad) at https://forum.atuin.sh\",\n );\n f.render_widget(feedback, input_chunk);\n\n return;\n }\n\n _ => {\n panic!(\"invalid tab index\");\n }\n }\n\n if !matches!(compactness, Compactness::Ultracompact) {\n let preview_width = match compactness {\n Compactness::Full => preview_width - 2,\n _ => preview_width,\n };\n let preview = self.build_preview(\n results,\n compactness,\n preview_width,\n preview_chunk.width.into(),\n theme,\n );\n #[allow(clippy::cast_possible_truncation)]\n let prefix_width = settings\n .ui\n .columns\n .iter()\n .take_while(|col| !col.expand)\n .map(|col| col.width + 1)\n .sum::()\n + \" > \".len() as u16;\n #[allow(clippy::cast_possible_truncation)]\n let min_prefix_width = \"[ SRCH: FULLTXT ] \".len() as u16;\n self.draw_preview(\n f,\n style,\n input_chunk,\n compactness,\n preview_chunk,\n preview,\n std::cmp::max(prefix_width, min_prefix_width),\n );\n }\n }\n\n #[allow(clippy::cast_possible_truncation, clippy::too_many_arguments)]\n fn draw_preview(\n &self,\n f: &mut Frame,\n style: StyleState,\n input_chunk: Rect,\n compactness: Compactness,\n preview_chunk: Rect,\n preview: Paragraph,\n prefix_width: u16,\n ) {\n let input = self.build_input(style, prefix_width);\n f.render_widget(input, input_chunk);\n\n f.render_widget(preview, preview_chunk);\n\n let extra_width = UnicodeWidthStr::width(self.search.input.substring());\n\n let cursor_offset = match compactness {\n Compactness::Full => 1,\n _ => 0,\n };\n f.set_cursor_position((\n // Put cursor past the end of the input text\n input_chunk.x + extra_width as u16 + prefix_width + cursor_offset,\n input_chunk.y + cursor_offset,\n ));\n }\n\n fn build_title(&self, theme: &Theme) -> Paragraph<'_> {\n let title = if self.update_needed.is_some() {\n let error_style: Style = Style::from_crossterm(theme.get_error());\n Paragraph::new(Text::from(Span::styled(\n format!(\"Atuin v{VERSION} - UPDATE\"),\n error_style.add_modifier(Modifier::BOLD),\n )))\n } else {\n let style: Style = Style::from_crossterm(theme.as_style(Meaning::Base));\n Paragraph::new(Text::from(Span::styled(\n format!(\"Atuin v{VERSION}\"),\n style.add_modifier(Modifier::BOLD),\n )))\n };\n title.alignment(Alignment::Left)\n }\n\n #[allow(clippy::unused_self)]\n fn build_help(&self, settings: &Settings, theme: &Theme) -> Paragraph<'_> {\n match self.tab_index {\n // search\n 0 => Paragraph::new(Text::from(Line::from(vec![\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(\": exit\"),\n Span::raw(\", \"),\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(\": edit\"),\n Span::raw(\", \"),\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(if settings.enter_accept {\n \": run\"\n } else {\n \": edit\"\n }),\n Span::raw(\", \"),\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(\": inspect\"),\n ]))),\n\n 1 => Paragraph::new(Text::from(Line::from(vec![\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(\": exit\"),\n Span::raw(\", \"),\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(\": search\"),\n Span::raw(\", \"),\n Span::styled(\"\", Style::default().add_modifier(Modifier::BOLD)),\n Span::raw(\": delete\"),\n ]))),\n\n _ => unreachable!(\"invalid tab index\"),\n }\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation)))\n .alignment(Alignment::Center)\n }\n\n fn build_stats(&self, theme: &Theme) -> Paragraph<'_> {\n Paragraph::new(Text::from(Span::raw(format!(\n \"history count: {}\",\n self.history_count,\n ))))\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation)))\n .alignment(Alignment::Right)\n }\n\n #[allow(clippy::too_many_arguments)]\n fn build_results_list<'a>(\n style: StyleState,\n results: &'a [History],\n keymap_mode: KeymapMode,\n now: &'a dyn Fn() -> OffsetDateTime,\n indicator: &'a str,\n theme: &'a Theme,\n history_highlighter: HistoryHighlighter<'a>,\n show_numeric_shortcuts: bool,\n columns: &'a [UiColumn],\n ) -> HistoryList<'a> {\n let results_list = HistoryList::new(\n results,\n style.invert,\n keymap_mode == KeymapMode::VimNormal,\n now,\n indicator,\n theme,\n history_highlighter,\n show_numeric_shortcuts,\n columns,\n );\n\n match style.compactness {\n Compactness::Full => {\n if style.invert {\n results_list.block(\n Block::default()\n .borders(Borders::LEFT | Borders::RIGHT)\n .border_type(BorderType::Rounded)\n .title(format!(\"{:─>width$}\", \"\", width = style.inner_width - 2)),\n )\n } else {\n results_list.block(\n Block::default()\n .borders(Borders::TOP | Borders::LEFT | Borders::RIGHT)\n .border_type(BorderType::Rounded),\n )\n }\n }\n _ => results_list,\n }\n }\n\n fn build_input(&self, style: StyleState, prefix_width: u16) -> Paragraph<'_> {\n let (pref, mode) = if self.switched_search_mode {\n (\" SRCH:\", self.search_mode.as_str())\n } else if self.search.custom_context.is_some() {\n (\" CTX:\", self.search.filter_mode.as_str())\n } else {\n (\"\", self.search.filter_mode.as_str())\n };\n // 3: surrounding \"[\" \"] \"\n let mode_width = usize::from(prefix_width) - pref.len() - 3;\n // sanity check to ensure we don't exceed the layout limits\n debug_assert!(mode_width >= mode.len(), \"mode name '{mode}' is too long!\");\n let input = format!(\"[{pref}{mode:^mode_width$}] {}\", self.search.input.as_str(),);\n let input = Paragraph::new(input);\n match style.compactness {\n Compactness::Full => {\n if style.invert {\n input.block(\n Block::default()\n .borders(Borders::LEFT | Borders::RIGHT | Borders::TOP)\n .border_type(BorderType::Rounded),\n )\n } else {\n input.block(\n Block::default()\n .borders(Borders::LEFT | Borders::RIGHT)\n .border_type(BorderType::Rounded)\n .title(format!(\"{:─>width$}\", \"\", width = style.inner_width - 2)),\n )\n }\n }\n _ => input,\n }\n }\n\n fn build_preview(\n &self,\n results: &[History],\n compactness: Compactness,\n preview_width: u16,\n chunk_width: usize,\n theme: &Theme,\n ) -> Paragraph<'_> {\n let selected = self.results_state.selected();\n let command = if results.is_empty() {\n String::new()\n } else {\n use itertools::Itertools as _;\n let s = &results[selected].command;\n s.split('\\n')\n .flat_map(|line| {\n line.char_indices()\n .step_by(preview_width.into())\n .map(|(i, _)| i)\n .chain(Some(line.len()))\n .tuple_windows()\n .map(|(a, b)| (&line[a..b]).escape_control().to_string())\n })\n .join(\"\\n\")\n };\n\n match compactness {\n Compactness::Full => Paragraph::new(command).block(\n Block::default()\n .borders(Borders::BOTTOM | Borders::LEFT | Borders::RIGHT)\n .border_type(BorderType::Rounded)\n .title(format!(\"{:─>width$}\", \"\", width = chunk_width - 2)),\n ),\n _ => Paragraph::new(command)\n .style(Style::from_crossterm(theme.as_style(Meaning::Annotation))),\n }\n }\n}\n\n/// The writer used for terminal output - either stdout or /dev/tty\nenum TerminalWriter {\n Stdout(std::io::Stdout),\n #[cfg(unix)]\n Tty(std::fs::File),\n}\n\nimpl Write for TerminalWriter {\n fn write(&mut self, buf: &[u8]) -> std::io::Result {\n match self {\n TerminalWriter::Stdout(stdout) => stdout.write(buf),\n #[cfg(unix)]\n TerminalWriter::Tty(file) => file.write(buf),\n }\n }\n\n fn flush(&mut self) -> std::io::Result<()> {\n match self {\n TerminalWriter::Stdout(stdout) => stdout.flush(),\n #[cfg(unix)]\n TerminalWriter::Tty(file) => file.flush(),\n }\n }\n}\n\n/// Screen state captured from atuin-hex's screen server.\n#[cfg(unix)]\nstruct SavedScreen {\n #[allow(dead_code)]\n rows: u16,\n #[allow(dead_code)]\n cols: u16,\n cursor_row: u16,\n cursor_col: u16,\n /// Pre-formatted ANSI bytes for each screen row, ready to write to stdout.\n rows_data: Vec>,\n}\n\n/// Connect to atuin-hex's Unix socket and fetch the current screen state.\n///\n/// The wire format is:\n/// ```text\n/// [rows: u16 BE][cols: u16 BE][cursor_row: u16 BE][cursor_col: u16 BE]\n/// [row_0_len: u32 BE][row_0_bytes...]\n/// [row_1_len: u32 BE][row_1_bytes...]\n/// ...\n/// ```\n#[cfg(unix)]\nfn fetch_screen_state(socket_path: &str) -> Option {\n use std::os::unix::net::UnixStream;\n\n let mut stream = UnixStream::connect(socket_path).ok()?;\n stream.set_read_timeout(Some(Duration::from_secs(2))).ok()?;\n\n let mut data = Vec::new();\n stream.read_to_end(&mut data).ok()?;\n\n if data.len() < 8 {\n return None;\n }\n\n let rows = u16::from_be_bytes([data[0], data[1]]);\n let cols = u16::from_be_bytes([data[2], data[3]]);\n let cursor_row = u16::from_be_bytes([data[4], data[5]]);\n let cursor_col = u16::from_be_bytes([data[6], data[7]]);\n\n // Parse length-prefixed rows\n let mut rows_data = Vec::with_capacity(rows as usize);\n let mut offset = 8;\n while offset + 4 <= data.len() {\n let row_len = u32::from_be_bytes([\n data[offset],\n data[offset + 1],\n data[offset + 2],\n data[offset + 3],\n ]) as usize;\n offset += 4;\n if offset + row_len > data.len() {\n break;\n }\n rows_data.push(data[offset..offset + row_len].to_vec());\n offset += row_len;\n }\n\n Some(SavedScreen {\n rows,\n cols,\n cursor_row,\n cursor_col,\n rows_data,\n })\n}\n\n/// Restore the screen area that was covered by the popup.\n///\n/// Writes the pre-formatted per-row ANSI bytes received from atuin-hex\n/// directly to stdout, which correctly handles wide characters, colors, and\n/// all text attributes without needing a client-side vt100 parser.\n#[cfg(unix)]\nfn restore_popup_area(saved: &SavedScreen, popup_rect: Rect, scroll_offset: u16) {\n use ratatui::crossterm::cursor::MoveTo;\n\n let mut stdout = stdout();\n\n for dy in 0..popup_rect.height {\n let target_row = popup_rect.y + dy;\n let source_row = (target_row + scroll_offset) as usize;\n\n // Clear only the popup region. The server-side rows_formatted() skips\n // default cells (spaces with default attributes) using cursor jumps, so\n // any popup content at those positions would remain if not cleared\n // beforehand. We write `popup_rect.width` spaces instead of\n // ClearType::CurrentLine so that only the popup area is cleared, not\n // the entire terminal line.\n let _ = execute!(\n stdout,\n MoveTo(popup_rect.x, target_row),\n ratatui::crossterm::style::SetAttribute(ratatui::crossterm::style::Attribute::Reset),\n );\n let _ = write!(stdout, \"{:width$}\", \"\", width = popup_rect.width as usize);\n let _ = execute!(stdout, MoveTo(popup_rect.x, target_row));\n\n if let Some(row_bytes) = saved.rows_data.get(source_row) {\n let _ = stdout.write_all(row_bytes);\n }\n }\n\n let _ = execute!(\n stdout,\n MoveTo(\n saved.cursor_col,\n saved.cursor_row.saturating_sub(scroll_offset)\n )\n );\n let _ = stdout.flush();\n}\n\nstruct Stdout {\n writer: TerminalWriter,\n inline_mode: bool,\n}\n\nimpl Stdout {\n pub fn new(inline_mode: bool, stdout_is_terminal: bool) -> std::io::Result {\n terminal::enable_raw_mode()?;\n\n // If stdout is not a terminal (e.g., captured by command substitution),\n // fall back to /dev/tty so the TUI can still render.\n // This allows usage like: VAR=$(atuin search -i)\n let mut writer = if stdout_is_terminal {\n TerminalWriter::Stdout(stdout())\n } else {\n #[cfg(unix)]\n {\n TerminalWriter::Tty(\n std::fs::File::options()\n .read(true)\n .write(true)\n .open(\"/dev/tty\")?,\n )\n }\n #[cfg(not(unix))]\n {\n return Err(std::io::Error::new(\n std::io::ErrorKind::Unsupported,\n \"Interactive mode requires a terminal\",\n ));\n }\n };\n\n if !inline_mode {\n execute!(writer, terminal::EnterAlternateScreen)?;\n }\n\n execute!(\n writer,\n event::EnableMouseCapture,\n event::EnableBracketedPaste,\n )?;\n\n #[cfg(not(target_os = \"windows\"))]\n execute!(\n writer,\n PushKeyboardEnhancementFlags(\n KeyboardEnhancementFlags::DISAMBIGUATE_ESCAPE_CODES\n | KeyboardEnhancementFlags::REPORT_ALL_KEYS_AS_ESCAPE_CODES\n | KeyboardEnhancementFlags::REPORT_ALTERNATE_KEYS\n ),\n )?;\n\n Ok(Self {\n writer,\n inline_mode,\n })\n }\n}\n\nimpl Drop for Stdout {\n fn drop(&mut self) {\n #[cfg(not(target_os = \"windows\"))]\n execute!(self.writer, PopKeyboardEnhancementFlags).unwrap();\n\n if !self.inline_mode {\n execute!(self.writer, terminal::LeaveAlternateScreen).unwrap();\n }\n execute!(\n self.writer,\n event::DisableMouseCapture,\n event::DisableBracketedPaste,\n )\n .unwrap();\n\n terminal::disable_raw_mode().unwrap();\n }\n}\n\nimpl Write for Stdout {\n fn write(&mut self, buf: &[u8]) -> std::io::Result {\n self.writer.write(buf)\n }\n\n fn flush(&mut self) -> std::io::Result<()> {\n self.writer.flush()\n }\n}\n\n// this is a big blob of horrible! clean it up!\n/// Compute the popup position and any scroll offset needed to make room.\n///\n/// Given the cursor row, terminal dimensions, and desired popup height,\n/// returns `(popup_rect, scroll_offset)` where `scroll_offset` is the number\n/// of lines the caller should scroll the terminal up before rendering.\n///\n/// This function performs no I/O — it is a pure computation.\nfn compute_popup_placement(\n cursor_row: u16,\n term_rows: u16,\n term_cols: u16,\n inline_height: u16,\n) -> (Rect, u16) {\n let popup_w = term_cols;\n let popup_h = inline_height.min(term_rows);\n let space_below = term_rows.saturating_sub(cursor_row);\n\n let (popup_y, scroll) = if popup_h <= space_below {\n // Fits below cursor\n (cursor_row, 0u16)\n } else if cursor_row >= term_rows / 2 {\n // Bottom half — render above cursor (overlay on existing text)\n (cursor_row.saturating_sub(popup_h), 0u16)\n } else {\n // Top half, not enough space — scroll terminal to make room\n let scroll = popup_h.saturating_sub(space_below);\n let popup_y = cursor_row.saturating_sub(scroll);\n (popup_y, scroll)\n };\n\n (Rect::new(0, popup_y, popup_w, popup_h), scroll)\n}\n\n// for now, it works. But it'd be great if it were more easily readable, and\n// modular. I'd like to add some more stats and stuff at some point\n#[allow(\n clippy::cast_possible_truncation,\n clippy::too_many_lines,\n clippy::cognitive_complexity\n)]\npub async fn history(\n query: &[String],\n settings: &Settings,\n mut db: impl Database,\n history_store: &HistoryStore,\n theme: &Theme,\n) -> Result {\n let inline_height = if settings.shell_up_key_binding {\n settings\n .inline_height_shell_up_key_binding\n .unwrap_or(settings.inline_height)\n } else {\n settings.inline_height\n };\n\n // Check if stdout is a terminal - if not (e.g., command substitution like VAR=$(atuin search -i)),\n // we need to use /dev/tty for the TUI and force fullscreen mode (inline mode requires\n // cursor position queries that don't work when stdout is captured)\n let stdout_is_terminal = stdout().is_terminal();\n\n // Use fullscreen mode if the inline height doesn't fit in the terminal,\n // this will preserve the scroll position upon exit.\n // Also force fullscreen when stdout isn't a terminal (inline mode won't work).\n let inline_height = if !stdout_is_terminal {\n 0\n } else if let Ok(size) = terminal::size()\n && inline_height >= size.1\n {\n 0\n } else {\n inline_height\n };\n\n // Popup mode: if running under atuin-hex and inline mode is requested,\n // fetch the screen state and render as a centered overlay.\n #[cfg(unix)]\n let (saved_screen, popup_rect, popup_scroll_offset) = {\n let socket_path = std::env::var(\"ATUIN_HEX_SOCKET\").ok();\n if let Some(ref path) = socket_path\n && inline_height > 0\n {\n let saved = fetch_screen_state(path);\n if let Some(ref s) = saved {\n let (term_cols, term_rows) = terminal::size().unwrap_or((s.cols, s.rows));\n let (popup_rect, scroll) =\n compute_popup_placement(s.cursor_row, term_rows, term_cols, inline_height);\n\n // Scroll terminal content up to make room if needed\n if scroll > 0 {\n use ratatui::crossterm::cursor::MoveTo;\n let mut stdout = stdout();\n let _ = execute!(stdout, MoveTo(0, term_rows - 1));\n for _ in 0..scroll {\n let _ = writeln!(stdout);\n }\n let _ = stdout.flush();\n }\n\n (saved, popup_rect, scroll)\n } else {\n (None, Rect::default(), 0u16)\n }\n } else {\n (None, Rect::default(), 0u16)\n }\n };\n\n #[cfg(not(unix))]\n let (saved_screen, popup_rect, popup_scroll_offset): (Option<()>, Rect, u16) =\n (None, Rect::default(), 0);\n\n let popup_mode = saved_screen.is_some();\n\n let stdout = Stdout::new(inline_height > 0, stdout_is_terminal)?;\n\n // In popup mode, clear the popup region on the physical terminal before\n // ratatui takes over. Ratatui's diff-based rendering compares against an\n // initially-empty buffer, so cells that remain \"empty\" (spaces with default\n // style) won't be written — leaving underlying terminal text visible.\n // By pre-clearing with spaces, those cells are already correct on screen.\n if popup_mode {\n use ratatui::crossterm::cursor::MoveTo;\n let mut raw_stdout = std::io::stdout();\n // Queue all commands without flushing so the terminal receives them\n // as a single write — no intermediate cursor positions are visible.\n let _ = queue!(\n raw_stdout,\n ratatui::crossterm::style::SetAttribute(ratatui::crossterm::style::Attribute::Reset)\n );\n for row in popup_rect.y..popup_rect.y.saturating_add(popup_rect.height) {\n let _ = queue!(raw_stdout, MoveTo(popup_rect.x, row));\n let _ = write!(\n raw_stdout,\n \"{:width$}\",\n \"\",\n width = popup_rect.width as usize\n );\n }\n let _ = raw_stdout.flush();\n }\n\n let backend = CrosstermBackend::new(stdout);\n let mut terminal = Terminal::with_options(\n backend,\n TerminalOptions {\n viewport: if popup_mode {\n Viewport::Fixed(popup_rect)\n } else if inline_height > 0 {\n Viewport::Inline(inline_height)\n } else {\n Viewport::Fullscreen\n },\n },\n )?;\n\n let original_query = query.join(\" \");\n\n // Check if this is a command chaining scenario\n let is_command_chaining = if settings.command_chaining {\n let trimmed = original_query.trim_end();\n trimmed.ends_with(\"&&\") || trimmed.ends_with('|')\n } else {\n false\n };\n\n // For command chaining, start with empty input to allow searching for new commands\n let search_input = if is_command_chaining {\n String::new()\n } else {\n original_query.clone()\n };\n\n let mut input = Cursor::from(search_input);\n // Put the cursor at the end of the query by default\n input.end();\n\n let settings2 = settings.clone();\n let update_needed = tokio::spawn(async move { settings2.needs_update().await }).fuse();\n tokio::pin!(update_needed);\n\n let initial_context = current_context().await?;\n\n let history_count = db.history_count(false).await?;\n let search_mode = if settings.shell_up_key_binding {\n settings\n .search_mode_shell_up_key_binding\n .unwrap_or(settings.search_mode)\n } else {\n settings.search_mode\n };\n let default_filter_mode = settings\n .filter_mode_shell_up_key_binding\n .filter(|_| settings.shell_up_key_binding)\n .unwrap_or_else(|| settings.default_filter_mode(initial_context.git_root.is_some()));\n let mut app = State {\n history_count,\n results_state: ListState::default(),\n update_needed: None,\n switched_search_mode: false,\n search_mode,\n tab_index: 0,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::from_settings(settings),\n search: SearchState {\n input,\n filter_mode: default_filter_mode,\n context: initial_context.clone(),\n custom_context: None,\n },\n engine: engines::engine(search_mode, settings),\n results_len: 0,\n accept: false,\n keymap_mode: match settings.keymap_mode {\n KeymapMode::Auto => KeymapMode::Emacs,\n value => value,\n },\n current_cursor: None,\n now: if settings.prefers_reduced_motion {\n let now = OffsetDateTime::now_utc();\n Box::new(move || now)\n } else {\n Box::new(OffsetDateTime::now_utc)\n },\n prefix: false,\n pending_vim_key: None,\n original_input_empty: original_query.is_empty(),\n };\n\n app.initialize_keymap_cursor(settings);\n\n let mut results = app.query_results(&mut db, settings.smart_sort).await?;\n\n if inline_height > 0 && !popup_mode {\n terminal.clear()?;\n }\n\n let mut stats: Option = None;\n let mut inspecting: Option = None;\n let accept;\n let result = 'render: loop {\n terminal.draw(|f| {\n app.draw(\n f,\n &results,\n stats.clone(),\n inspecting.as_ref(),\n settings,\n theme,\n popup_mode,\n );\n })?;\n\n let initial_input = app.search.input.as_str().to_owned();\n let initial_filter_mode = app.search.filter_mode;\n let initial_search_mode = app.search_mode;\n let initial_custom_context = app.search.custom_context.clone();\n\n let event_ready = tokio::task::spawn_blocking(|| event::poll(Duration::from_millis(250)));\n\n tokio::select! {\n event_ready = event_ready => {\n if event_ready?? {\n loop {\n match app.handle_input(settings, &event::read()?) {\n InputAction::Continue => {},\n InputAction::Delete(index) => {\n if results.is_empty() {\n break;\n }\n app.results_len -= 1;\n let selected = app.results_state.selected();\n if selected == app.results_len {\n app.inspecting_state.reset();\n app.results_state.select(selected - 1);\n }\n\n let entry = results.remove(index);\n\n if settings.sync.records {\n let (id, _) = history_store.delete(entry.id).await?;\n history_store.incremental_build(&db, &[id]).await?;\n } else {\n db.delete(entry.clone()).await?;\n }\n\n app.tab_index = 0;\n },\n InputAction::SwitchContext(index) => {\n if let Some(index) = index && let Some(entry) = results.get(index) {\n app.search.custom_context = Some(entry.id.clone());\n app.search.context = Context::from_history(entry);\n app.search.filter_mode = FilterMode::Session;\n app.search.input = Cursor::from(String::new());\n app.results_state = ListState::default();\n } else {\n app.search.custom_context = None;\n app.search.context = initial_context.clone();\n app.search.filter_mode = default_filter_mode;\n }\n },\n InputAction::Redraw => {\n if !popup_mode {\n terminal.clear()?;\n }\n terminal.draw(|f| {\n app.draw(f, &results, stats.clone(), inspecting.as_ref(), settings, theme, popup_mode);\n })?;\n },\n r => {\n accept = app.accept;\n break 'render r;\n },\n }\n if !event::poll(Duration::ZERO)? {\n break;\n }\n }\n }\n }\n update_needed = &mut update_needed => {\n // Don't fail interactive search if update check fails\n // The update check is a nice-to-have feature, not critical\n app.update_needed = update_needed.ok().flatten();\n }\n }\n\n if initial_input != app.search.input.as_str()\n || initial_filter_mode != app.search.filter_mode\n || initial_search_mode != app.search_mode\n || initial_custom_context != app.search.custom_context\n {\n results = app.query_results(&mut db, settings.smart_sort).await?;\n }\n\n // In custom context mode, when no filter is applied, highlight the entry which was used\n // to enter the context when changing modes. This helps to find your way around.\n if app.search.custom_context.is_some()\n && app.search.input.as_str().is_empty()\n && (initial_custom_context != app.search.custom_context\n || initial_filter_mode != app.search.filter_mode)\n && let Some(history_id) = app.search.custom_context.clone()\n && let Some(pos) = results.iter().position(|entry| entry.id == history_id)\n {\n app.results_state.select(pos);\n }\n\n let inspecting_id = app.inspecting_state.clone().current;\n // If inspecting ID is not the current inspecting History, update it.\n match inspecting_id {\n Some(inspecting_id) => {\n if inspecting.is_none() || inspecting_id != inspecting.clone().unwrap().id {\n inspecting = db.load(inspecting_id.0.as_str()).await?;\n }\n }\n _ => {\n inspecting = None;\n }\n }\n\n stats = if app.tab_index == 0 {\n None\n } else if !results.is_empty() {\n // If we have stats, then we can indicate next available IDs. This avoids passing\n // around a database object, or a full stats object.\n let selected = match inspecting.clone() {\n Some(insp) => insp,\n None => results[app.results_state.selected()].clone(),\n };\n let stats = db.stats(&selected).await?;\n app.inspecting_state.current = Some(selected.id);\n app.inspecting_state.previous = match stats.previous.clone() {\n Some(p) => Some(p.id),\n _ => None,\n };\n app.inspecting_state.next = match stats.next.clone() {\n Some(p) => Some(p.id),\n _ => None,\n };\n Some(stats)\n } else {\n None\n };\n };\n\n app.finalize_keymap_cursor(settings);\n\n if popup_mode {\n // In popup mode, restore the screen area that was covered by the popup.\n // This must happen before Stdout is dropped (which disables raw mode).\n #[cfg(unix)]\n if let Some(ref saved) = saved_screen {\n restore_popup_area(saved, popup_rect, popup_scroll_offset);\n }\n } else if inline_height > 0 {\n terminal.clear()?;\n }\n\n let accept = accept\n && matches!(\n Shell::from_env(),\n Shell::Zsh | Shell::Fish | Shell::Bash | Shell::Xonsh | Shell::Nu | Shell::Powershell\n );\n\n let accept_prefix = \"__atuin_accept__:\";\n\n match result {\n InputAction::AcceptInspecting => {\n match inspecting {\n Some(result) => {\n let mut command = result.command;\n\n if accept {\n command = String::from(accept_prefix) + &command;\n }\n\n // index is in bounds so we return that entry\n Ok(command)\n }\n None => Ok(String::new()),\n }\n }\n InputAction::Accept(index) if index < results.len() => {\n let mut command = results.swap_remove(index).command;\n\n if is_command_chaining {\n command = format!(\"{} {}\", original_query.trim_end(), command);\n } else if accept {\n command = String::from(accept_prefix) + &command;\n }\n\n // index is in bounds so we return that entry\n Ok(command)\n }\n InputAction::ReturnOriginal => Ok(String::new()),\n InputAction::Copy(index) => {\n let cmd = results.swap_remove(index).command;\n set_clipboard(cmd);\n Ok(String::new())\n }\n InputAction::ReturnQuery | InputAction::Accept(_) => {\n // Either:\n // * index == RETURN_QUERY, in which case we should return the input\n // * out of bounds -> usually implies no selected entry so we return the input\n Ok(app.search.input.into_inner())\n }\n InputAction::Continue\n | InputAction::Redraw\n | InputAction::Delete(_)\n | InputAction::SwitchContext(_) => {\n unreachable!(\"should have been handled!\")\n }\n }\n}\n\n// cli-clipboard only works on Windows, Mac, and Linux.\n\n#[cfg(all(\n feature = \"clipboard\",\n any(target_os = \"windows\", target_os = \"macos\", target_os = \"linux\")\n))]\nfn set_clipboard(s: String) {\n let mut ctx = arboard::Clipboard::new().unwrap();\n ctx.set_text(s).unwrap();\n // Use the clipboard context to make sure it is saved\n ctx.get_text().unwrap();\n}\n\n#[cfg(not(all(\n feature = \"clipboard\",\n any(target_os = \"windows\", target_os = \"macos\", target_os = \"linux\")\n)))]\nfn set_clipboard(_s: String) {}\n\n#[cfg(test)]\nmod tests {\n use atuin_client::database::Context;\n use atuin_client::history::History;\n use atuin_client::settings::{\n FilterMode, KeymapMode, Preview, PreviewStrategy, SearchMode, Settings,\n };\n use time::OffsetDateTime;\n\n use crate::command::client::search::engines::{self, SearchState};\n use crate::command::client::search::history_list::ListState;\n\n use super::{Compactness, InspectingState, KeymapSet, State};\n\n #[test]\n #[allow(clippy::too_many_lines)]\n fn calc_preview_height_test() {\n let settings_preview_auto = Settings {\n preview: Preview {\n strategy: PreviewStrategy::Auto,\n },\n show_preview: true,\n ..Settings::utc()\n };\n\n let settings_preview_auto_h2 = Settings {\n preview: Preview {\n strategy: PreviewStrategy::Auto,\n },\n show_preview: true,\n max_preview_height: 2,\n ..Settings::utc()\n };\n\n let settings_preview_h4 = Settings {\n preview: Preview {\n strategy: PreviewStrategy::Static,\n },\n show_preview: true,\n max_preview_height: 4,\n ..Settings::utc()\n };\n\n let settings_preview_fixed = Settings {\n preview: Preview {\n strategy: PreviewStrategy::Fixed,\n },\n show_preview: true,\n max_preview_height: 15,\n ..Settings::utc()\n };\n\n let cmd_60: History = History::capture()\n .timestamp(time::OffsetDateTime::now_utc())\n .command(\"for i in $(seq -w 10); do echo \\\"item number $i - abcd\\\"; done\")\n .cwd(\"/\")\n .build()\n .into();\n\n let cmd_124: History = History::capture()\n .timestamp(time::OffsetDateTime::now_utc())\n .command(\"echo 'Aurea prima sata est aetas, quae vindice nullo, sponte sua, sine lege fidem rectumque colebat. Poena metusque aberant'\")\n .cwd(\"/\")\n .build()\n .into();\n\n let cmd_200: History = History::capture()\n .timestamp(time::OffsetDateTime::now_utc())\n .command(\"CREATE USER atuin WITH ENCRYPTED PASSWORD 'supersecretpassword'; CREATE DATABASE atuin WITH OWNER = atuin; \\\\c atuin; REVOKE ALL PRIVILEGES ON SCHEMA public FROM PUBLIC; echo 'All done. 200 characters'\")\n .cwd(\"/\")\n .build()\n .into();\n\n let results: Vec = vec![cmd_60, cmd_124, cmd_200];\n\n // the selected command does not require a preview\n let no_preview = State::calc_preview_height(\n &settings_preview_auto,\n &results,\n 0_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 80,\n );\n // the selected command requires 2 lines\n let preview_h2 = State::calc_preview_height(\n &settings_preview_auto,\n &results,\n 1_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 80,\n );\n // the selected command requires 3 lines\n let preview_h3 = State::calc_preview_height(\n &settings_preview_auto,\n &results,\n 2_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 80,\n );\n // the selected command requires a preview of 1 line (happens when the command is between preview_width-19 and preview_width)\n let preview_one_line = State::calc_preview_height(\n &settings_preview_auto,\n &results,\n 0_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 66,\n );\n // the selected command requires 3 lines, but we have a max preview height limit of 2\n let preview_limit_at_2 = State::calc_preview_height(\n &settings_preview_auto_h2,\n &results,\n 2_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 80,\n );\n // the longest command requires 3 lines\n let preview_static_h3 = State::calc_preview_height(\n &settings_preview_h4,\n &results,\n 1_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 80,\n );\n // the longest command requires 10 lines, but we have a max preview height limit of 4\n let preview_static_limit_at_4 = State::calc_preview_height(\n &settings_preview_h4,\n &results,\n 1_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 20,\n );\n // the longest command requires 10 lines, but we have a max preview height of 15 and a fixed preview strategy\n let settings_preview_fixed = State::calc_preview_height(\n &settings_preview_fixed,\n &results,\n 1_usize,\n 0_usize,\n Compactness::Full,\n 1,\n 20,\n );\n\n assert_eq!(no_preview, 1);\n // 1 * 2 is the space for the border\n let border_space = 2;\n assert_eq!(preview_h2, 2 + border_space);\n assert_eq!(preview_h3, 3 + border_space);\n assert_eq!(preview_one_line, 1 + border_space);\n assert_eq!(preview_limit_at_2, 2 + border_space);\n assert_eq!(preview_static_h3, 3 + border_space);\n assert_eq!(preview_static_limit_at_4, 4 + border_space);\n assert_eq!(settings_preview_fixed, 15 + border_space);\n }\n\n // Test when there's no results, scrolling up or down doesn't underflow\n #[test]\n fn state_scroll_up_underflow() {\n let settings = Settings::utc();\n let mut state = State {\n history_count: 0,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 0,\n accept: false,\n keymap_mode: KeymapMode::Auto,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Directory,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n state.scroll_up(1);\n state.scroll_down(1);\n }\n\n #[test]\n fn test_accept_keybindings() {\n use atuin_client::settings::Keys;\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n let mut settings = Settings::utc();\n settings.keys = Keys {\n scroll_exits: true,\n exit_past_line_start: false,\n accept_past_line_end: true,\n accept_past_line_start: false,\n accept_with_backspace: false,\n prefix: \"a\".to_string(),\n };\n\n let mut state = State {\n history_count: 1,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 1,\n accept: false,\n keymap_mode: KeymapMode::Emacs,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n let tab_event = KeyEvent::new(KeyCode::Tab, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &tab_event);\n assert!(\n matches!(result, super::InputAction::Accept(_)),\n \"Tab should always accept\"\n );\n\n // Test left arrow with accept_past_line_start disabled (should continue)\n let left_event = KeyEvent::new(KeyCode::Left, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &left_event);\n assert!(\n matches!(result, super::InputAction::Continue),\n \"Left arrow should continue when disabled\"\n );\n\n // Test left arrow with accept_past_line_start enabled (should accept at start of line)\n settings.keys.accept_past_line_start = true;\n state.keymaps = KeymapSet::defaults(&settings);\n let result = state.handle_key_input(&settings, &left_event);\n assert!(\n matches!(result, super::InputAction::Accept(_)),\n \"Left arrow should accept at start of line when enabled\"\n );\n settings.keys.accept_past_line_start = false;\n state.keymaps = KeymapSet::defaults(&settings);\n\n let backspace_event = KeyEvent::new(KeyCode::Backspace, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &backspace_event);\n assert!(\n matches!(result, super::InputAction::Continue),\n \"Backspace should continue when disabled\"\n );\n\n settings.keys.accept_with_backspace = true;\n state.keymaps = KeymapSet::defaults(&settings);\n let result = state.handle_key_input(&settings, &backspace_event);\n assert!(\n matches!(result, super::InputAction::Accept(_)),\n \"Backspace should accept at start of line when enabled\"\n );\n\n state.search.input.insert('t');\n state.search.input.insert('e');\n state.search.input.insert('s');\n state.search.input.insert('t');\n state.search.input.end();\n\n let right_event = KeyEvent::new(KeyCode::Right, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &right_event);\n assert!(\n matches!(result, super::InputAction::Accept(_)),\n \"Right arrow should accept at end of line when enabled\"\n );\n\n settings.keys.accept_past_line_start = true;\n state.keymaps = KeymapSet::defaults(&settings);\n let left_event = KeyEvent::new(KeyCode::Left, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &left_event);\n assert!(\n matches!(result, super::InputAction::Continue),\n \"Left arrow should continue and end of line, even when enabled\"\n );\n settings.keys.accept_past_line_start = false;\n state.keymaps = KeymapSet::defaults(&settings);\n\n settings.keys.accept_with_backspace = true;\n state.keymaps = KeymapSet::defaults(&settings);\n let backspace_event = KeyEvent::new(KeyCode::Backspace, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &backspace_event);\n assert!(\n matches!(result, super::InputAction::Continue),\n \"Backspace should continue at end of line, even when enabled\"\n );\n settings.keys.accept_with_backspace = false;\n state.keymaps = KeymapSet::defaults(&settings);\n }\n\n #[test]\n fn test_vim_gg_multikey_sequence() {\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n let settings = Settings::utc();\n\n let mut state = State {\n history_count: 100,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 100,\n accept: false,\n keymap_mode: KeymapMode::VimNormal,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n // Start in the middle of the list\n state.results_state.select(50);\n\n // First 'g' should set pending state\n let g_event = KeyEvent::new(KeyCode::Char('g'), KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &g_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.pending_vim_key, Some('g'));\n assert_eq!(state.results_state.selected(), 50); // Position unchanged\n\n // Second 'g' should jump to end (visual top in non-inverted mode)\n let result = state.handle_key_input(&settings, &g_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.pending_vim_key, None);\n assert_eq!(state.results_state.selected(), 99); // Jumped to last index (visual top)\n }\n\n #[test]\n fn test_vim_g_key_clears_on_other_input() {\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n let settings = Settings::utc();\n\n let mut state = State {\n history_count: 100,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 100,\n accept: false,\n keymap_mode: KeymapMode::VimNormal,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n state.results_state.select(50);\n\n // Press 'g' to set pending state\n let g_event = KeyEvent::new(KeyCode::Char('g'), KeyModifiers::NONE);\n state.handle_key_input(&settings, &g_event);\n assert_eq!(state.pending_vim_key, Some('g'));\n\n // Press 'j' - should clear pending state\n let j_event = KeyEvent::new(KeyCode::Char('j'), KeyModifiers::NONE);\n state.handle_key_input(&settings, &j_event);\n assert_eq!(state.pending_vim_key, None);\n }\n\n #[test]\n fn test_vim_big_g_jump_to_bottom() {\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n let settings = Settings::utc();\n\n let mut state = State {\n history_count: 100,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 100,\n accept: false,\n keymap_mode: KeymapMode::VimNormal,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n state.results_state.select(50);\n\n // 'G' should jump to visual bottom (index 0 in non-inverted mode)\n let big_g_event = KeyEvent::new(KeyCode::Char('G'), KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &big_g_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.results_state.selected(), 0);\n }\n\n #[test]\n fn test_vim_ctrl_u_d_half_page_scroll() {\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n let settings = Settings::utc();\n\n let mut state = State {\n history_count: 100,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 100,\n accept: false,\n keymap_mode: KeymapMode::VimNormal,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n state.results_state.select(50);\n\n // Ctrl+d should return Continue and clear pending key\n // (scroll amount depends on max_entries which is 0 in tests)\n state.pending_vim_key = Some('g');\n let ctrl_d_event = KeyEvent::new(KeyCode::Char('d'), KeyModifiers::CONTROL);\n let result = state.handle_key_input(&settings, &ctrl_d_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.pending_vim_key, None);\n\n // Ctrl+u should return Continue and clear pending key\n state.pending_vim_key = Some('g');\n let ctrl_u_event = KeyEvent::new(KeyCode::Char('u'), KeyModifiers::CONTROL);\n let result = state.handle_key_input(&settings, &ctrl_u_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.pending_vim_key, None);\n }\n\n #[test]\n fn test_vim_ctrl_f_b_full_page_scroll() {\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n let settings = Settings::utc();\n\n let mut state = State {\n history_count: 100,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 100,\n accept: false,\n keymap_mode: KeymapMode::VimNormal,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n state.results_state.select(50);\n\n // Ctrl+f should return Continue and clear pending key\n // (scroll amount depends on max_entries which is 0 in tests)\n state.pending_vim_key = Some('g');\n let ctrl_f_event = KeyEvent::new(KeyCode::Char('f'), KeyModifiers::CONTROL);\n let result = state.handle_key_input(&settings, &ctrl_f_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.pending_vim_key, None);\n\n // Ctrl+b should return Continue and clear pending key\n state.pending_vim_key = Some('g');\n let ctrl_b_event = KeyEvent::new(KeyCode::Char('b'), KeyModifiers::CONTROL);\n let result = state.handle_key_input(&settings, &ctrl_b_event);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.pending_vim_key, None);\n }\n\n // -----------------------------------------------------------------------\n // Executor tests (execute_action)\n // -----------------------------------------------------------------------\n\n /// Helper to build a State for executor tests.\n fn make_executor_state(results_len: usize, selected: usize) -> State {\n let settings = Settings::utc();\n let mut state = State {\n history_count: results_len as i64,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len,\n accept: false,\n keymap_mode: KeymapMode::Emacs,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::defaults(&settings),\n search: SearchState {\n input: String::new().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n state.results_state.select(selected);\n state\n }\n\n #[test]\n fn execute_select_next_no_invert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::SelectNext, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Non-inverted: SelectNext = scroll_down = selected - 1\n assert_eq!(state.results_state.selected(), 49);\n }\n\n #[test]\n fn execute_select_next_with_invert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let mut settings = Settings::utc();\n settings.invert = true;\n let result = state.execute_action(&Action::SelectNext, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Inverted: SelectNext = scroll_up = selected + 1\n assert_eq!(state.results_state.selected(), 51);\n }\n\n #[test]\n fn execute_select_previous_no_invert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::SelectPrevious, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Non-inverted: SelectPrevious = scroll_up = selected + 1\n assert_eq!(state.results_state.selected(), 51);\n }\n\n #[test]\n fn execute_vim_enter_normal() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::VimEnterNormal, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.keymap_mode, KeymapMode::VimNormal);\n }\n\n #[test]\n fn execute_vim_enter_insert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n state.keymap_mode = KeymapMode::VimNormal;\n let settings = Settings::utc();\n let result = state.execute_action(&Action::VimEnterInsert, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.keymap_mode, KeymapMode::VimInsert);\n }\n\n #[test]\n fn execute_accept_sets_accept_flag() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 5);\n let mut settings = Settings::utc();\n settings.enter_accept = true;\n let result = state.execute_action(&Action::Accept, &settings);\n assert!(matches!(result, super::InputAction::Accept(5)));\n assert!(state.accept);\n }\n\n #[test]\n fn execute_return_selection_does_not_set_accept() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 5);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::ReturnSelection, &settings);\n assert!(matches!(result, super::InputAction::Accept(5)));\n assert!(!state.accept);\n }\n\n #[test]\n fn execute_accept_nth() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 5);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::AcceptNth(3), &settings);\n assert!(matches!(result, super::InputAction::Accept(8)));\n }\n\n #[test]\n fn execute_scroll_to_top_no_invert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::ScrollToTop, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Non-inverted: visual top = highest index\n assert_eq!(state.results_state.selected(), 99);\n }\n\n #[test]\n fn execute_scroll_to_top_with_invert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let mut settings = Settings::utc();\n settings.invert = true;\n let result = state.execute_action(&Action::ScrollToTop, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Inverted: visual top = index 0\n assert_eq!(state.results_state.selected(), 0);\n }\n\n #[test]\n fn execute_scroll_to_bottom_no_invert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::ScrollToBottom, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Non-inverted: visual bottom = index 0\n assert_eq!(state.results_state.selected(), 0);\n }\n\n #[test]\n fn execute_toggle_tab() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n assert_eq!(state.tab_index, 0);\n state.execute_action(&Action::ToggleTab, &settings);\n assert_eq!(state.tab_index, 1);\n state.execute_action(&Action::ToggleTab, &settings);\n assert_eq!(state.tab_index, 0);\n }\n\n #[test]\n fn execute_enter_prefix_mode() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n assert!(!state.prefix);\n state.execute_action(&Action::EnterPrefixMode, &settings);\n assert!(state.prefix);\n }\n\n #[test]\n fn execute_exit_returns_based_on_exit_mode() {\n use crate::command::client::search::keybindings::Action;\n use atuin_client::settings::ExitMode;\n\n let mut state = make_executor_state(100, 0);\n let mut settings = Settings::utc();\n\n settings.exit_mode = ExitMode::ReturnOriginal;\n let result = state.execute_action(&Action::Exit, &settings);\n assert!(matches!(result, super::InputAction::ReturnOriginal));\n\n settings.exit_mode = ExitMode::ReturnQuery;\n let result = state.execute_action(&Action::Exit, &settings);\n assert!(matches!(result, super::InputAction::ReturnQuery));\n }\n\n #[test]\n fn execute_return_original() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::ReturnOriginal, &settings);\n assert!(matches!(result, super::InputAction::ReturnOriginal));\n }\n\n #[test]\n fn execute_copy() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 7);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::Copy, &settings);\n assert!(matches!(result, super::InputAction::Copy(7)));\n }\n\n #[test]\n fn execute_delete() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 7);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::Delete, &settings);\n assert!(matches!(result, super::InputAction::Delete(7)));\n }\n\n #[test]\n fn execute_switch_context() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 7);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::SwitchContext, &settings);\n assert!(matches!(result, super::InputAction::SwitchContext(Some(7))));\n }\n\n #[test]\n fn execute_clear_context() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 7);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::ClearContext, &settings);\n assert!(matches!(result, super::InputAction::SwitchContext(None)));\n }\n\n #[test]\n fn execute_noop() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 50);\n let settings = Settings::utc();\n let result = state.execute_action(&Action::Noop, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n assert_eq!(state.results_state.selected(), 50);\n }\n\n #[test]\n fn execute_accept_in_inspector_tab() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 5);\n state.tab_index = 1;\n let settings = Settings::utc();\n let result = state.execute_action(&Action::Accept, &settings);\n assert!(matches!(result, super::InputAction::AcceptInspecting));\n }\n\n #[test]\n fn execute_cycle_search_mode() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n let original_mode = state.search_mode;\n let result = state.execute_action(&Action::CycleSearchMode, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n assert!(state.switched_search_mode);\n assert_ne!(state.search_mode, original_mode);\n }\n\n #[test]\n fn execute_vim_search_insert() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n state.search.input.insert('h');\n state.search.input.insert('i');\n state.keymap_mode = KeymapMode::VimNormal;\n let settings = Settings::utc();\n let result = state.execute_action(&Action::VimSearchInsert, &settings);\n assert!(matches!(result, super::InputAction::Continue));\n // Should clear input and switch to insert mode\n assert_eq!(state.search.input.as_str(), \"\");\n assert_eq!(state.keymap_mode, KeymapMode::VimInsert);\n }\n\n #[test]\n fn execute_cursor_movement() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n\n // Insert some text\n state.search.input.insert('h');\n state.search.input.insert('e');\n state.search.input.insert('l');\n state.search.input.insert('l');\n state.search.input.insert('o');\n // cursor is at end (position 5)\n\n // CursorLeft\n state.execute_action(&Action::CursorLeft, &settings);\n assert_eq!(state.search.input.position(), 4);\n\n // CursorStart\n state.execute_action(&Action::CursorStart, &settings);\n assert_eq!(state.search.input.position(), 0);\n\n // CursorEnd\n state.execute_action(&Action::CursorEnd, &settings);\n assert_eq!(state.search.input.position(), 5);\n\n // CursorRight at end does nothing\n state.execute_action(&Action::CursorRight, &settings);\n assert_eq!(state.search.input.position(), 5);\n }\n\n #[test]\n fn execute_editing() {\n use crate::command::client::search::keybindings::Action;\n\n let mut state = make_executor_state(100, 0);\n let settings = Settings::utc();\n\n // Insert \"hello\"\n state.search.input.insert('h');\n state.search.input.insert('e');\n state.search.input.insert('l');\n state.search.input.insert('l');\n state.search.input.insert('o');\n\n // DeleteCharBefore (backspace)\n state.execute_action(&Action::DeleteCharBefore, &settings);\n assert_eq!(state.search.input.as_str(), \"hell\");\n\n // ClearLine\n state.execute_action(&Action::ClearLine, &settings);\n assert_eq!(state.search.input.as_str(), \"\");\n }\n\n #[test]\n fn keymap_config_return_query() {\n use atuin_client::settings::KeyBindingConfig;\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n use std::collections::HashMap;\n\n let mut settings = Settings::utc();\n // Configure tab to return-query\n settings.keymap.emacs = HashMap::from([(\n \"tab\".to_string(),\n KeyBindingConfig::Simple(\"return-query\".to_string()),\n )]);\n\n let mut state = State {\n history_count: 100,\n update_needed: None,\n results_state: ListState::default(),\n switched_search_mode: false,\n search_mode: SearchMode::Fuzzy,\n results_len: 100,\n accept: false,\n keymap_mode: KeymapMode::Emacs,\n prefix: false,\n current_cursor: None,\n tab_index: 0,\n pending_vim_key: None,\n original_input_empty: false,\n inspecting_state: InspectingState {\n current: None,\n next: None,\n previous: None,\n },\n keymaps: KeymapSet::from_settings(&settings),\n search: SearchState {\n input: \"test query\".to_string().into(),\n filter_mode: FilterMode::Global,\n context: Context {\n session: String::new(),\n cwd: String::new(),\n hostname: String::new(),\n host_id: String::new(),\n git_root: None,\n },\n custom_context: None,\n },\n engine: engines::engine(SearchMode::Fuzzy, &settings),\n now: Box::new(OffsetDateTime::now_utc),\n };\n\n let tab_event = KeyEvent::new(KeyCode::Tab, KeyModifiers::NONE);\n let result = state.handle_key_input(&settings, &tab_event);\n assert!(\n matches!(result, super::InputAction::ReturnQuery),\n \"Tab configured as return-query should return InputAction::ReturnQuery\"\n );\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/keybindings/actions.rs", "content": "use std::fmt;\n\nuse serde::{Deserialize, Deserializer, Serialize, Serializer};\n\n/// All possible actions that can be triggered by a keybinding.\n#[derive(Debug, Clone, PartialEq, Eq)]\npub enum Action {\n // Cursor movement\n CursorLeft,\n CursorRight,\n CursorWordLeft,\n CursorWordRight,\n CursorWordEnd,\n CursorStart,\n CursorEnd,\n\n // Editing\n DeleteCharBefore,\n DeleteCharAfter,\n DeleteWordBefore,\n DeleteWordAfter,\n DeleteToWordBoundary,\n ClearLine,\n ClearToStart,\n ClearToEnd,\n\n // List navigation\n SelectNext,\n SelectPrevious,\n ScrollHalfPageUp,\n ScrollHalfPageDown,\n ScrollPageUp,\n ScrollPageDown,\n ScrollToTop,\n ScrollToBottom,\n ScrollToScreenTop,\n ScrollToScreenMiddle,\n ScrollToScreenBottom,\n\n // Commands — accept selection and execute immediately\n Accept,\n AcceptNth(u8),\n // Commands — return selection to command line without executing\n ReturnSelection,\n ReturnSelectionNth(u8),\n // Commands — other\n Copy,\n Delete,\n ReturnOriginal,\n ReturnQuery,\n Exit,\n Redraw,\n CycleFilterMode,\n CycleSearchMode,\n SwitchContext,\n ClearContext,\n ToggleTab,\n\n // Mode changes\n VimEnterNormal,\n VimEnterInsert,\n VimEnterInsertAfter,\n VimEnterInsertAtStart,\n VimEnterInsertAtEnd,\n VimSearchInsert,\n VimChangeToEnd,\n EnterPrefixMode,\n\n // Inspector\n InspectPrevious,\n InspectNext,\n\n // Special\n Noop,\n}\n\nimpl Action {\n /// Convert from a kebab-case string.\n pub fn from_str(s: &str) -> Result {\n // Handle accept-N and return-selection-N patterns\n if let Some(rest) = s.strip_prefix(\"accept-\")\n && let Ok(n) = rest.parse::()\n && (1..=9).contains(&n)\n {\n return Ok(Action::AcceptNth(n));\n }\n if let Some(rest) = s.strip_prefix(\"return-selection-\")\n && let Ok(n) = rest.parse::()\n && (1..=9).contains(&n)\n {\n return Ok(Action::ReturnSelectionNth(n));\n }\n\n match s {\n \"cursor-left\" => Ok(Action::CursorLeft),\n \"cursor-right\" => Ok(Action::CursorRight),\n \"cursor-word-left\" => Ok(Action::CursorWordLeft),\n \"cursor-word-right\" => Ok(Action::CursorWordRight),\n \"cursor-word-end\" => Ok(Action::CursorWordEnd),\n \"cursor-start\" => Ok(Action::CursorStart),\n \"cursor-end\" => Ok(Action::CursorEnd),\n\n \"delete-char-before\" => Ok(Action::DeleteCharBefore),\n \"delete-char-after\" => Ok(Action::DeleteCharAfter),\n \"delete-word-before\" => Ok(Action::DeleteWordBefore),\n \"delete-word-after\" => Ok(Action::DeleteWordAfter),\n \"delete-to-word-boundary\" => Ok(Action::DeleteToWordBoundary),\n \"clear-line\" => Ok(Action::ClearLine),\n \"clear-to-start\" => Ok(Action::ClearToStart),\n \"clear-to-end\" => Ok(Action::ClearToEnd),\n\n \"select-next\" => Ok(Action::SelectNext),\n \"select-previous\" => Ok(Action::SelectPrevious),\n \"scroll-half-page-up\" => Ok(Action::ScrollHalfPageUp),\n \"scroll-half-page-down\" => Ok(Action::ScrollHalfPageDown),\n \"scroll-page-up\" => Ok(Action::ScrollPageUp),\n \"scroll-page-down\" => Ok(Action::ScrollPageDown),\n \"scroll-to-top\" => Ok(Action::ScrollToTop),\n \"scroll-to-bottom\" => Ok(Action::ScrollToBottom),\n \"scroll-to-screen-top\" => Ok(Action::ScrollToScreenTop),\n \"scroll-to-screen-middle\" => Ok(Action::ScrollToScreenMiddle),\n \"scroll-to-screen-bottom\" => Ok(Action::ScrollToScreenBottom),\n\n \"accept\" => Ok(Action::Accept),\n \"return-selection\" => Ok(Action::ReturnSelection),\n \"copy\" => Ok(Action::Copy),\n \"delete\" => Ok(Action::Delete),\n \"return-original\" => Ok(Action::ReturnOriginal),\n \"return-query\" => Ok(Action::ReturnQuery),\n \"exit\" => Ok(Action::Exit),\n \"redraw\" => Ok(Action::Redraw),\n \"cycle-filter-mode\" => Ok(Action::CycleFilterMode),\n \"cycle-search-mode\" => Ok(Action::CycleSearchMode),\n \"switch-context\" => Ok(Action::SwitchContext),\n \"clear-context\" => Ok(Action::ClearContext),\n \"toggle-tab\" => Ok(Action::ToggleTab),\n\n \"vim-enter-normal\" => Ok(Action::VimEnterNormal),\n \"vim-enter-insert\" => Ok(Action::VimEnterInsert),\n \"vim-enter-insert-after\" => Ok(Action::VimEnterInsertAfter),\n \"vim-enter-insert-at-start\" => Ok(Action::VimEnterInsertAtStart),\n \"vim-enter-insert-at-end\" => Ok(Action::VimEnterInsertAtEnd),\n \"vim-search-insert\" => Ok(Action::VimSearchInsert),\n \"vim-change-to-end\" => Ok(Action::VimChangeToEnd),\n \"enter-prefix-mode\" => Ok(Action::EnterPrefixMode),\n\n \"inspect-previous\" => Ok(Action::InspectPrevious),\n \"inspect-next\" => Ok(Action::InspectNext),\n\n \"noop\" => Ok(Action::Noop),\n\n _ => Err(format!(\"unknown action: {s}\")),\n }\n }\n\n /// Convert to a kebab-case string.\n pub fn as_str(&self) -> String {\n match self {\n Action::CursorLeft => \"cursor-left\".to_string(),\n Action::CursorRight => \"cursor-right\".to_string(),\n Action::CursorWordLeft => \"cursor-word-left\".to_string(),\n Action::CursorWordRight => \"cursor-word-right\".to_string(),\n Action::CursorWordEnd => \"cursor-word-end\".to_string(),\n Action::CursorStart => \"cursor-start\".to_string(),\n Action::CursorEnd => \"cursor-end\".to_string(),\n\n Action::DeleteCharBefore => \"delete-char-before\".to_string(),\n Action::DeleteCharAfter => \"delete-char-after\".to_string(),\n Action::DeleteWordBefore => \"delete-word-before\".to_string(),\n Action::DeleteWordAfter => \"delete-word-after\".to_string(),\n Action::DeleteToWordBoundary => \"delete-to-word-boundary\".to_string(),\n Action::ClearLine => \"clear-line\".to_string(),\n Action::ClearToStart => \"clear-to-start\".to_string(),\n Action::ClearToEnd => \"clear-to-end\".to_string(),\n\n Action::SelectNext => \"select-next\".to_string(),\n Action::SelectPrevious => \"select-previous\".to_string(),\n Action::ScrollHalfPageUp => \"scroll-half-page-up\".to_string(),\n Action::ScrollHalfPageDown => \"scroll-half-page-down\".to_string(),\n Action::ScrollPageUp => \"scroll-page-up\".to_string(),\n Action::ScrollPageDown => \"scroll-page-down\".to_string(),\n Action::ScrollToTop => \"scroll-to-top\".to_string(),\n Action::ScrollToBottom => \"scroll-to-bottom\".to_string(),\n Action::ScrollToScreenTop => \"scroll-to-screen-top\".to_string(),\n Action::ScrollToScreenMiddle => \"scroll-to-screen-middle\".to_string(),\n Action::ScrollToScreenBottom => \"scroll-to-screen-bottom\".to_string(),\n\n Action::Accept => \"accept\".to_string(),\n Action::AcceptNth(n) => format!(\"accept-{n}\"),\n Action::ReturnSelection => \"return-selection\".to_string(),\n Action::ReturnSelectionNth(n) => format!(\"return-selection-{n}\"),\n Action::Copy => \"copy\".to_string(),\n Action::Delete => \"delete\".to_string(),\n Action::ReturnOriginal => \"return-original\".to_string(),\n Action::ReturnQuery => \"return-query\".to_string(),\n Action::Exit => \"exit\".to_string(),\n Action::Redraw => \"redraw\".to_string(),\n Action::CycleFilterMode => \"cycle-filter-mode\".to_string(),\n Action::CycleSearchMode => \"cycle-search-mode\".to_string(),\n Action::SwitchContext => \"switch-context\".to_string(),\n Action::ClearContext => \"clear-context\".to_string(),\n Action::ToggleTab => \"toggle-tab\".to_string(),\n\n Action::VimEnterNormal => \"vim-enter-normal\".to_string(),\n Action::VimEnterInsert => \"vim-enter-insert\".to_string(),\n Action::VimEnterInsertAfter => \"vim-enter-insert-after\".to_string(),\n Action::VimEnterInsertAtStart => \"vim-enter-insert-at-start\".to_string(),\n Action::VimEnterInsertAtEnd => \"vim-enter-insert-at-end\".to_string(),\n Action::VimSearchInsert => \"vim-search-insert\".to_string(),\n Action::VimChangeToEnd => \"vim-change-to-end\".to_string(),\n Action::EnterPrefixMode => \"enter-prefix-mode\".to_string(),\n\n Action::InspectPrevious => \"inspect-previous\".to_string(),\n Action::InspectNext => \"inspect-next\".to_string(),\n\n Action::Noop => \"noop\".to_string(),\n }\n }\n}\n\nimpl fmt::Display for Action {\n fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n write!(f, \"{}\", self.as_str())\n }\n}\n\nimpl Serialize for Action {\n fn serialize(&self, serializer: S) -> Result {\n serializer.serialize_str(&self.as_str())\n }\n}\n\nimpl<'de> Deserialize<'de> for Action {\n fn deserialize>(deserializer: D) -> Result {\n let s = String::deserialize(deserializer)?;\n Action::from_str(&s).map_err(serde::de::Error::custom)\n }\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn parse_basic_actions() {\n assert_eq!(Action::from_str(\"cursor-left\").unwrap(), Action::CursorLeft);\n assert_eq!(Action::from_str(\"accept\").unwrap(), Action::Accept);\n assert_eq!(Action::from_str(\"exit\").unwrap(), Action::Exit);\n assert_eq!(Action::from_str(\"noop\").unwrap(), Action::Noop);\n assert_eq!(\n Action::from_str(\"vim-enter-normal\").unwrap(),\n Action::VimEnterNormal\n );\n }\n\n #[test]\n fn parse_accept_nth() {\n assert_eq!(Action::from_str(\"accept-1\").unwrap(), Action::AcceptNth(1));\n assert_eq!(Action::from_str(\"accept-9\").unwrap(), Action::AcceptNth(9));\n }\n\n #[test]\n fn parse_return_selection() {\n assert_eq!(\n Action::from_str(\"return-selection\").unwrap(),\n Action::ReturnSelection\n );\n assert_eq!(\n Action::from_str(\"return-selection-1\").unwrap(),\n Action::ReturnSelectionNth(1)\n );\n assert_eq!(\n Action::from_str(\"return-selection-9\").unwrap(),\n Action::ReturnSelectionNth(9)\n );\n }\n\n #[test]\n fn parse_unknown_action() {\n assert!(Action::from_str(\"unknown-action\").is_err());\n assert!(Action::from_str(\"accept-0\").is_err());\n assert!(Action::from_str(\"accept-10\").is_err());\n assert!(Action::from_str(\"return-selection-0\").is_err());\n assert!(Action::from_str(\"return-selection-10\").is_err());\n }\n\n #[test]\n fn round_trip() {\n let actions = vec![\n Action::CursorLeft,\n Action::Accept,\n Action::AcceptNth(5),\n Action::ReturnSelection,\n Action::ReturnSelectionNth(3),\n Action::VimSearchInsert,\n Action::ScrollToScreenMiddle,\n ];\n for action in actions {\n let s = action.as_str();\n let parsed = Action::from_str(&s).unwrap();\n assert_eq!(action, parsed);\n }\n }\n\n #[test]\n fn serde_round_trip() {\n let action = Action::CursorLeft;\n let json = serde_json::to_string(&action).unwrap();\n assert_eq!(json, \"\\\"cursor-left\\\"\");\n let parsed: Action = serde_json::from_str(&json).unwrap();\n assert_eq!(parsed, Action::CursorLeft);\n\n let action = Action::AcceptNth(3);\n let json = serde_json::to_string(&action).unwrap();\n assert_eq!(json, \"\\\"accept-3\\\"\");\n let parsed: Action = serde_json::from_str(&json).unwrap();\n assert_eq!(parsed, Action::AcceptNth(3));\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/keybindings/conditions.rs", "content": "use std::fmt;\n\nuse serde::{Deserialize, Deserializer, Serialize, Serializer};\n\n/// Atomic (leaf) conditions that can be evaluated against state.\n#[derive(Debug, Clone, PartialEq, Eq)]\npub enum ConditionAtom {\n CursorAtStart,\n CursorAtEnd,\n InputEmpty,\n OriginalInputEmpty,\n ListAtEnd,\n ListAtStart,\n NoResults,\n HasResults,\n HasContext,\n}\n\n/// Boolean expression tree over condition atoms.\n///\n/// Supports negation, conjunction, and disjunction with standard precedence:\n/// `!` binds tightest, then `&&`, then `||`.\n///\n/// Examples of valid expression strings:\n/// - `\"cursor-at-start\"` (bare atom)\n/// - `\"!no-results\"` (negation)\n/// - `\"cursor-at-start && input-empty\"` (conjunction)\n/// - `\"list-at-start || no-results\"` (disjunction)\n/// - `\"(cursor-at-start && !input-empty) || no-results\"` (grouping)\n#[derive(Debug, Clone, PartialEq, Eq)]\npub enum ConditionExpr {\n Atom(ConditionAtom),\n Not(Box),\n And(Box, Box),\n Or(Box, Box),\n}\n\n/// Context needed to evaluate conditions. This is a pure snapshot of state —\n/// no references to mutable data.\npub struct EvalContext {\n /// Current cursor position (unicode width units).\n pub cursor_position: usize,\n /// Width of the input string in unicode width units.\n pub input_width: usize,\n /// Byte length of the input string.\n pub input_byte_len: usize,\n /// Currently selected index in the results list.\n pub selected_index: usize,\n /// Total number of results.\n pub results_len: usize,\n /// Whether the original input (query passed to the TUI) was empty.\n pub original_input_empty: bool,\n /// Whether we use a search context of a command from the history.\n pub has_context: bool,\n}\n\n// ---------------------------------------------------------------------------\n// ConditionAtom\n// ---------------------------------------------------------------------------\n\nimpl ConditionAtom {\n /// Evaluate this atom against the given context.\n pub fn evaluate(&self, ctx: &EvalContext) -> bool {\n match self {\n ConditionAtom::CursorAtStart => ctx.cursor_position == 0,\n ConditionAtom::CursorAtEnd => ctx.cursor_position == ctx.input_width,\n ConditionAtom::InputEmpty => ctx.input_byte_len == 0,\n ConditionAtom::OriginalInputEmpty => ctx.original_input_empty,\n ConditionAtom::ListAtEnd => {\n ctx.results_len == 0 || ctx.selected_index >= ctx.results_len.saturating_sub(1)\n }\n ConditionAtom::ListAtStart => ctx.results_len == 0 || ctx.selected_index == 0,\n ConditionAtom::NoResults => ctx.results_len == 0,\n ConditionAtom::HasResults => ctx.results_len > 0,\n ConditionAtom::HasContext => ctx.has_context,\n }\n }\n\n /// Parse from a kebab-case string.\n pub fn from_str(s: &str) -> Result {\n match s {\n \"cursor-at-start\" => Ok(ConditionAtom::CursorAtStart),\n \"cursor-at-end\" => Ok(ConditionAtom::CursorAtEnd),\n \"input-empty\" => Ok(ConditionAtom::InputEmpty),\n \"original-input-empty\" => Ok(ConditionAtom::OriginalInputEmpty),\n \"list-at-end\" => Ok(ConditionAtom::ListAtEnd),\n \"list-at-start\" => Ok(ConditionAtom::ListAtStart),\n \"no-results\" => Ok(ConditionAtom::NoResults),\n \"has-results\" => Ok(ConditionAtom::HasResults),\n \"has-context\" => Ok(ConditionAtom::HasContext),\n _ => Err(format!(\"unknown condition: {s}\")),\n }\n }\n\n /// Convert to a kebab-case string.\n pub fn as_str(&self) -> &'static str {\n match self {\n ConditionAtom::CursorAtStart => \"cursor-at-start\",\n ConditionAtom::CursorAtEnd => \"cursor-at-end\",\n ConditionAtom::InputEmpty => \"input-empty\",\n ConditionAtom::OriginalInputEmpty => \"original-input-empty\",\n ConditionAtom::ListAtEnd => \"list-at-end\",\n ConditionAtom::ListAtStart => \"list-at-start\",\n ConditionAtom::NoResults => \"no-results\",\n ConditionAtom::HasResults => \"has-results\",\n ConditionAtom::HasContext => \"has-context\",\n }\n }\n}\n\nimpl fmt::Display for ConditionAtom {\n fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n write!(f, \"{}\", self.as_str())\n }\n}\n\n// ---------------------------------------------------------------------------\n// ConditionExpr — evaluation\n// ---------------------------------------------------------------------------\n\nimpl ConditionExpr {\n /// Evaluate this expression against the given context.\n pub fn evaluate(&self, ctx: &EvalContext) -> bool {\n match self {\n ConditionExpr::Atom(atom) => atom.evaluate(ctx),\n ConditionExpr::Not(inner) => !inner.evaluate(ctx),\n ConditionExpr::And(lhs, rhs) => lhs.evaluate(ctx) && rhs.evaluate(ctx),\n ConditionExpr::Or(lhs, rhs) => lhs.evaluate(ctx) || rhs.evaluate(ctx),\n }\n }\n}\n\n// ---------------------------------------------------------------------------\n// ConditionExpr — ergonomic builders\n// ---------------------------------------------------------------------------\n\nimpl From for ConditionExpr {\n fn from(atom: ConditionAtom) -> Self {\n ConditionExpr::Atom(atom)\n }\n}\n\n#[allow(dead_code)]\nimpl ConditionExpr {\n /// Negate this expression: `!self`.\n pub fn not(self) -> Self {\n ConditionExpr::Not(Box::new(self))\n }\n\n /// Conjoin with another expression: `self && other`.\n pub fn and(self, other: ConditionExpr) -> Self {\n ConditionExpr::And(Box::new(self), Box::new(other))\n }\n\n /// Disjoin with another expression: `self || other`.\n pub fn or(self, other: ConditionExpr) -> Self {\n ConditionExpr::Or(Box::new(self), Box::new(other))\n }\n}\n\n// ---------------------------------------------------------------------------\n// ConditionExpr — parser\n// ---------------------------------------------------------------------------\n\n/// Recursive descent parser for boolean condition expressions.\n///\n/// Grammar (standard boolean precedence):\n/// ```text\n/// expr = or_expr\n/// or_expr = and_expr (\"||\" and_expr)*\n/// and_expr = unary (\"&&\" unary)*\n/// unary = \"!\" unary | primary\n/// primary = atom | \"(\" expr \")\"\n/// atom = [a-z][a-z0-9-]*\n/// ```\nstruct ExprParser<'a> {\n input: &'a str,\n pos: usize,\n}\n\nimpl<'a> ExprParser<'a> {\n fn new(input: &'a str) -> Self {\n Self { input, pos: 0 }\n }\n\n fn skip_whitespace(&mut self) {\n while self.pos < self.input.len() && self.input.as_bytes()[self.pos].is_ascii_whitespace() {\n self.pos += 1;\n }\n }\n\n fn starts_with(&mut self, s: &str) -> bool {\n self.skip_whitespace();\n self.input[self.pos..].starts_with(s)\n }\n\n fn consume(&mut self, s: &str) -> bool {\n self.skip_whitespace();\n if self.input[self.pos..].starts_with(s) {\n self.pos += s.len();\n true\n } else {\n false\n }\n }\n\n /// Parse a full expression, expecting to consume all input.\n fn parse(mut self) -> Result {\n let expr = self.parse_or()?;\n self.skip_whitespace();\n if self.pos < self.input.len() {\n return Err(format!(\n \"unexpected input at position {}: {:?}\",\n self.pos,\n &self.input[self.pos..]\n ));\n }\n Ok(expr)\n }\n\n /// `or_expr` = `and_expr` (\"||\" `and_expr`)*\n fn parse_or(&mut self) -> Result {\n let mut left = self.parse_and()?;\n while self.starts_with(\"||\") {\n self.consume(\"||\");\n let right = self.parse_and()?;\n left = ConditionExpr::Or(Box::new(left), Box::new(right));\n }\n Ok(left)\n }\n\n /// `and_expr` = unary (\"&&\" unary)*\n fn parse_and(&mut self) -> Result {\n let mut left = self.parse_unary()?;\n while self.starts_with(\"&&\") {\n self.consume(\"&&\");\n let right = self.parse_unary()?;\n left = ConditionExpr::And(Box::new(left), Box::new(right));\n }\n Ok(left)\n }\n\n /// unary = \"!\" unary | primary\n fn parse_unary(&mut self) -> Result {\n if self.consume(\"!\") {\n let inner = self.parse_unary()?;\n Ok(ConditionExpr::Not(Box::new(inner)))\n } else {\n self.parse_primary()\n }\n }\n\n /// primary = \"(\" expr \")\" | atom\n fn parse_primary(&mut self) -> Result {\n if self.consume(\"(\") {\n let expr = self.parse_or()?;\n if !self.consume(\")\") {\n return Err(format!(\"expected ')' at position {}\", self.pos));\n }\n Ok(expr)\n } else {\n self.parse_atom()\n }\n }\n\n /// atom = [a-z][a-z0-9-]*\n fn parse_atom(&mut self) -> Result {\n self.skip_whitespace();\n let start = self.pos;\n while self.pos < self.input.len() {\n let b = self.input.as_bytes()[self.pos];\n if b.is_ascii_lowercase() || b.is_ascii_digit() || b == b'-' {\n self.pos += 1;\n } else {\n break;\n }\n }\n if self.pos == start {\n return Err(format!(\"expected condition name at position {}\", self.pos));\n }\n let name = &self.input[start..self.pos];\n let atom = ConditionAtom::from_str(name)?;\n Ok(ConditionExpr::Atom(atom))\n }\n}\n\nimpl ConditionExpr {\n /// Parse a condition expression from a string.\n pub fn parse(s: &str) -> Result {\n let parser = ExprParser::new(s);\n parser.parse()\n }\n}\n\n// ---------------------------------------------------------------------------\n// ConditionExpr — Display\n// ---------------------------------------------------------------------------\n\n/// Precedence levels for minimal-parentheses display.\n#[derive(PartialEq, Eq, PartialOrd, Ord, Clone, Copy)]\nenum Prec {\n Or = 0,\n And = 1,\n Not = 2,\n Atom = 3,\n}\n\nimpl ConditionExpr {\n fn prec(&self) -> Prec {\n match self {\n ConditionExpr::Or(..) => Prec::Or,\n ConditionExpr::And(..) => Prec::And,\n ConditionExpr::Not(..) => Prec::Not,\n ConditionExpr::Atom(..) => Prec::Atom,\n }\n }\n\n fn fmt_with_prec(&self, f: &mut fmt::Formatter<'_>, parent_prec: Prec) -> fmt::Result {\n let needs_parens = self.prec() < parent_prec;\n if needs_parens {\n write!(f, \"(\")?;\n }\n match self {\n ConditionExpr::Atom(atom) => write!(f, \"{atom}\")?,\n ConditionExpr::Not(inner) => {\n write!(f, \"!\")?;\n inner.fmt_with_prec(f, Prec::Not)?;\n }\n ConditionExpr::And(lhs, rhs) => {\n lhs.fmt_with_prec(f, Prec::And)?;\n write!(f, \" && \")?;\n rhs.fmt_with_prec(f, Prec::And)?;\n }\n ConditionExpr::Or(lhs, rhs) => {\n lhs.fmt_with_prec(f, Prec::Or)?;\n write!(f, \" || \")?;\n rhs.fmt_with_prec(f, Prec::Or)?;\n }\n }\n if needs_parens {\n write!(f, \")\")?;\n }\n Ok(())\n }\n}\n\nimpl fmt::Display for ConditionExpr {\n fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n self.fmt_with_prec(f, Prec::Or)\n }\n}\n\n// ---------------------------------------------------------------------------\n// Serde\n// ---------------------------------------------------------------------------\n\nimpl Serialize for ConditionExpr {\n fn serialize(&self, serializer: S) -> Result {\n serializer.serialize_str(&self.to_string())\n }\n}\n\nimpl<'de> Deserialize<'de> for ConditionExpr {\n fn deserialize>(deserializer: D) -> Result {\n let s = String::deserialize(deserializer)?;\n ConditionExpr::parse(&s).map_err(serde::de::Error::custom)\n }\n}\n\n// ---------------------------------------------------------------------------\n// Tests\n// ---------------------------------------------------------------------------\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n fn ctx(\n cursor: usize,\n width: usize,\n byte_len: usize,\n selected: usize,\n len: usize,\n ) -> EvalContext {\n ctx_with_original(cursor, width, byte_len, selected, len, false)\n }\n\n fn ctx_with_original(\n cursor: usize,\n width: usize,\n byte_len: usize,\n selected: usize,\n len: usize,\n original_input_empty: bool,\n ) -> EvalContext {\n EvalContext {\n cursor_position: cursor,\n input_width: width,\n input_byte_len: byte_len,\n selected_index: selected,\n results_len: len,\n original_input_empty,\n has_context: false,\n }\n }\n\n // -- Atom evaluation (carried over from Phase 0) --\n\n #[test]\n fn atom_cursor_at_start() {\n assert!(ConditionAtom::CursorAtStart.evaluate(&ctx(0, 5, 5, 0, 10)));\n assert!(!ConditionAtom::CursorAtStart.evaluate(&ctx(3, 5, 5, 0, 10)));\n }\n\n #[test]\n fn atom_cursor_at_end() {\n assert!(ConditionAtom::CursorAtEnd.evaluate(&ctx(5, 5, 5, 0, 10)));\n assert!(!ConditionAtom::CursorAtEnd.evaluate(&ctx(3, 5, 5, 0, 10)));\n assert!(ConditionAtom::CursorAtEnd.evaluate(&ctx(0, 0, 0, 0, 10)));\n }\n\n #[test]\n fn atom_input_empty() {\n assert!(ConditionAtom::InputEmpty.evaluate(&ctx(0, 0, 0, 0, 10)));\n assert!(!ConditionAtom::InputEmpty.evaluate(&ctx(0, 5, 5, 0, 10)));\n }\n\n #[test]\n fn atom_original_input_empty() {\n // original_input_empty = true\n assert!(\n ConditionAtom::OriginalInputEmpty.evaluate(&ctx_with_original(0, 0, 0, 0, 10, true))\n );\n // original_input_empty = false\n assert!(\n !ConditionAtom::OriginalInputEmpty.evaluate(&ctx_with_original(0, 0, 0, 0, 10, false))\n );\n // original_input_empty is independent of current input state\n assert!(\n ConditionAtom::OriginalInputEmpty.evaluate(&ctx_with_original(0, 5, 5, 0, 10, true))\n );\n }\n\n #[test]\n fn atom_list_at_end() {\n assert!(ConditionAtom::ListAtEnd.evaluate(&ctx(0, 0, 0, 99, 100)));\n assert!(!ConditionAtom::ListAtEnd.evaluate(&ctx(0, 0, 0, 50, 100)));\n assert!(ConditionAtom::ListAtEnd.evaluate(&ctx(0, 0, 0, 0, 0)));\n }\n\n #[test]\n fn atom_list_at_start() {\n assert!(ConditionAtom::ListAtStart.evaluate(&ctx(0, 0, 0, 0, 100)));\n assert!(!ConditionAtom::ListAtStart.evaluate(&ctx(0, 0, 0, 50, 100)));\n assert!(ConditionAtom::ListAtStart.evaluate(&ctx(0, 0, 0, 0, 0)));\n }\n\n #[test]\n fn atom_no_results_and_has_results() {\n assert!(ConditionAtom::NoResults.evaluate(&ctx(0, 0, 0, 0, 0)));\n assert!(!ConditionAtom::NoResults.evaluate(&ctx(0, 0, 0, 0, 5)));\n assert!(ConditionAtom::HasResults.evaluate(&ctx(0, 0, 0, 0, 5)));\n assert!(!ConditionAtom::HasResults.evaluate(&ctx(0, 0, 0, 0, 0)));\n }\n\n #[test]\n fn atom_has_context() {\n let mut context = ctx(0, 0, 0, 0, 0);\n assert!(!ConditionAtom::HasContext.evaluate(&context));\n context.has_context = true;\n assert!(ConditionAtom::HasContext.evaluate(&context));\n }\n\n #[test]\n fn atom_parse_round_trip() {\n let conditions = [\n \"cursor-at-start\",\n \"cursor-at-end\",\n \"input-empty\",\n \"original-input-empty\",\n \"list-at-end\",\n \"list-at-start\",\n \"no-results\",\n \"has-results\",\n ];\n for s in conditions {\n let c = ConditionAtom::from_str(s).unwrap();\n assert_eq!(c.as_str(), s);\n }\n }\n\n #[test]\n fn atom_parse_unknown() {\n assert!(ConditionAtom::from_str(\"unknown-condition\").is_err());\n }\n\n // -- Parser tests --\n\n #[test]\n fn parse_bare_atom() {\n let expr = ConditionExpr::parse(\"cursor-at-start\").unwrap();\n assert_eq!(expr, ConditionExpr::Atom(ConditionAtom::CursorAtStart));\n }\n\n #[test]\n fn parse_negation() {\n let expr = ConditionExpr::parse(\"!no-results\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::Not(Box::new(ConditionExpr::Atom(ConditionAtom::NoResults)))\n );\n }\n\n #[test]\n fn parse_double_negation() {\n let expr = ConditionExpr::parse(\"!!no-results\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::Not(Box::new(ConditionExpr::Not(Box::new(ConditionExpr::Atom(\n ConditionAtom::NoResults\n )))))\n );\n }\n\n #[test]\n fn parse_and() {\n let expr = ConditionExpr::parse(\"cursor-at-start && input-empty\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::And(\n Box::new(ConditionExpr::Atom(ConditionAtom::CursorAtStart)),\n Box::new(ConditionExpr::Atom(ConditionAtom::InputEmpty)),\n )\n );\n }\n\n #[test]\n fn parse_or() {\n let expr = ConditionExpr::parse(\"list-at-start || no-results\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::Or(\n Box::new(ConditionExpr::Atom(ConditionAtom::ListAtStart)),\n Box::new(ConditionExpr::Atom(ConditionAtom::NoResults)),\n )\n );\n }\n\n #[test]\n fn parse_precedence_and_binds_tighter_than_or() {\n // \"a || b && c\" should parse as \"a || (b && c)\"\n let expr = ConditionExpr::parse(\"cursor-at-start || input-empty && no-results\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::Or(\n Box::new(ConditionExpr::Atom(ConditionAtom::CursorAtStart)),\n Box::new(ConditionExpr::And(\n Box::new(ConditionExpr::Atom(ConditionAtom::InputEmpty)),\n Box::new(ConditionExpr::Atom(ConditionAtom::NoResults)),\n )),\n )\n );\n }\n\n #[test]\n fn parse_parens_override_precedence() {\n // \"(a || b) && c\"\n let expr = ConditionExpr::parse(\"(cursor-at-start || input-empty) && no-results\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::And(\n Box::new(ConditionExpr::Or(\n Box::new(ConditionExpr::Atom(ConditionAtom::CursorAtStart)),\n Box::new(ConditionExpr::Atom(ConditionAtom::InputEmpty)),\n )),\n Box::new(ConditionExpr::Atom(ConditionAtom::NoResults)),\n )\n );\n }\n\n #[test]\n fn parse_complex_nested() {\n // \"(a && !b) || c\"\n let expr = ConditionExpr::parse(\"(cursor-at-start && !input-empty) || no-results\").unwrap();\n assert_eq!(\n expr,\n ConditionExpr::Or(\n Box::new(ConditionExpr::And(\n Box::new(ConditionExpr::Atom(ConditionAtom::CursorAtStart)),\n Box::new(ConditionExpr::Not(Box::new(ConditionExpr::Atom(\n ConditionAtom::InputEmpty\n )))),\n )),\n Box::new(ConditionExpr::Atom(ConditionAtom::NoResults)),\n )\n );\n }\n\n #[test]\n fn parse_whitespace_tolerance() {\n let a = ConditionExpr::parse(\"cursor-at-start||input-empty\").unwrap();\n let b = ConditionExpr::parse(\"cursor-at-start || input-empty\").unwrap();\n let c = ConditionExpr::parse(\" cursor-at-start || input-empty \").unwrap();\n assert_eq!(a, b);\n assert_eq!(b, c);\n }\n\n #[test]\n fn parse_error_unknown_atom() {\n assert!(ConditionExpr::parse(\"unknown-thing\").is_err());\n }\n\n #[test]\n fn parse_error_trailing_input() {\n assert!(ConditionExpr::parse(\"cursor-at-start blah\").is_err());\n }\n\n #[test]\n fn parse_error_unmatched_paren() {\n assert!(ConditionExpr::parse(\"(cursor-at-start\").is_err());\n }\n\n #[test]\n fn parse_error_empty() {\n assert!(ConditionExpr::parse(\"\").is_err());\n }\n\n // -- Expression evaluation --\n\n #[test]\n fn eval_not() {\n let expr = ConditionExpr::parse(\"!no-results\").unwrap();\n // Has results → !no-results is true\n assert!(expr.evaluate(&ctx(0, 0, 0, 0, 5)));\n // No results → !no-results is false\n assert!(!expr.evaluate(&ctx(0, 0, 0, 0, 0)));\n }\n\n #[test]\n fn eval_and() {\n let expr = ConditionExpr::parse(\"cursor-at-start && input-empty\").unwrap();\n // Both true\n assert!(expr.evaluate(&ctx(0, 0, 0, 0, 10)));\n // First true, second false (non-empty input)\n assert!(!expr.evaluate(&ctx(0, 5, 5, 0, 10)));\n // First false (cursor not at start)\n assert!(!expr.evaluate(&ctx(3, 5, 5, 0, 10)));\n }\n\n #[test]\n fn eval_or() {\n let expr = ConditionExpr::parse(\"list-at-start || no-results\").unwrap();\n // list at bottom (selected=0)\n assert!(expr.evaluate(&ctx(0, 0, 0, 0, 10)));\n // no results\n assert!(expr.evaluate(&ctx(0, 0, 0, 0, 0)));\n // neither\n assert!(!expr.evaluate(&ctx(0, 0, 0, 5, 10)));\n }\n\n #[test]\n fn eval_complex_nested() {\n // (cursor-at-start && !input-empty) || no-results\n let expr = ConditionExpr::parse(\"(cursor-at-start && !input-empty) || no-results\").unwrap();\n\n // cursor at start, input not empty → true (left branch)\n assert!(expr.evaluate(&ctx(0, 5, 5, 0, 10)));\n // no results → true (right branch)\n assert!(expr.evaluate(&ctx(3, 5, 5, 0, 0)));\n // cursor not at start, has results → false\n assert!(!expr.evaluate(&ctx(3, 5, 5, 0, 10)));\n // cursor at start, input empty → false (left: && fails; right: has results)\n assert!(!expr.evaluate(&ctx(0, 0, 0, 0, 10)));\n }\n\n // -- Display --\n\n #[test]\n fn display_atom() {\n let expr = ConditionExpr::Atom(ConditionAtom::CursorAtStart);\n assert_eq!(expr.to_string(), \"cursor-at-start\");\n }\n\n #[test]\n fn display_not() {\n let expr = ConditionExpr::Atom(ConditionAtom::NoResults).not();\n assert_eq!(expr.to_string(), \"!no-results\");\n }\n\n #[test]\n fn display_and() {\n let expr = ConditionExpr::Atom(ConditionAtom::CursorAtStart)\n .and(ConditionExpr::Atom(ConditionAtom::InputEmpty));\n assert_eq!(expr.to_string(), \"cursor-at-start && input-empty\");\n }\n\n #[test]\n fn display_or() {\n let expr = ConditionExpr::Atom(ConditionAtom::ListAtStart)\n .or(ConditionExpr::Atom(ConditionAtom::NoResults));\n assert_eq!(expr.to_string(), \"list-at-start || no-results\");\n }\n\n #[test]\n fn display_parens_when_needed() {\n // (a || b) && c — the Or inside And needs parens\n let expr = ConditionExpr::Atom(ConditionAtom::CursorAtStart)\n .or(ConditionExpr::Atom(ConditionAtom::InputEmpty))\n .and(ConditionExpr::Atom(ConditionAtom::NoResults));\n assert_eq!(\n expr.to_string(),\n \"(cursor-at-start || input-empty) && no-results\"\n );\n }\n\n #[test]\n fn display_no_parens_when_not_needed() {\n // a || b && c — no parens needed (and binds tighter)\n let inner_and = ConditionExpr::Atom(ConditionAtom::InputEmpty)\n .and(ConditionExpr::Atom(ConditionAtom::NoResults));\n let expr = ConditionExpr::Atom(ConditionAtom::CursorAtStart).or(inner_and);\n assert_eq!(\n expr.to_string(),\n \"cursor-at-start || input-empty && no-results\"\n );\n }\n\n // -- Display round-trip --\n\n #[test]\n fn display_round_trip() {\n let cases = [\n \"cursor-at-start\",\n \"!no-results\",\n \"cursor-at-start && input-empty\",\n \"list-at-start || no-results\",\n \"(cursor-at-start || input-empty) && no-results\",\n \"(cursor-at-start && !input-empty) || no-results\",\n ];\n for s in cases {\n let expr = ConditionExpr::parse(s).unwrap();\n let displayed = expr.to_string();\n let reparsed = ConditionExpr::parse(&displayed).unwrap();\n assert_eq!(expr, reparsed, \"round-trip failed for: {s}\");\n }\n }\n\n // -- Serde --\n\n #[test]\n fn serde_simple_atom() {\n let expr = ConditionExpr::Atom(ConditionAtom::CursorAtStart);\n let json = serde_json::to_string(&expr).unwrap();\n assert_eq!(json, \"\\\"cursor-at-start\\\"\");\n let parsed: ConditionExpr = serde_json::from_str(&json).unwrap();\n assert_eq!(parsed, expr);\n }\n\n #[test]\n fn serde_compound_expression() {\n let json = \"\\\"cursor-at-start && !input-empty\\\"\";\n let parsed: ConditionExpr = serde_json::from_str(json).unwrap();\n let expected = ConditionExpr::And(\n Box::new(ConditionExpr::Atom(ConditionAtom::CursorAtStart)),\n Box::new(ConditionExpr::Not(Box::new(ConditionExpr::Atom(\n ConditionAtom::InputEmpty,\n )))),\n );\n assert_eq!(parsed, expected);\n }\n\n #[test]\n fn serde_round_trip() {\n let expr = ConditionExpr::parse(\"(cursor-at-start && !input-empty) || no-results\").unwrap();\n let json = serde_json::to_string(&expr).unwrap();\n let parsed: ConditionExpr = serde_json::from_str(&json).unwrap();\n assert_eq!(expr, parsed);\n }\n\n // -- From --\n\n #[test]\n fn from_atom_into_expr() {\n let expr: ConditionExpr = ConditionAtom::CursorAtStart.into();\n assert_eq!(expr, ConditionExpr::Atom(ConditionAtom::CursorAtStart));\n }\n\n // -- Builder helpers --\n\n #[test]\n fn builder_chain() {\n let expr = ConditionExpr::from(ConditionAtom::CursorAtStart)\n .and(ConditionExpr::from(ConditionAtom::InputEmpty).not())\n .or(ConditionExpr::from(ConditionAtom::NoResults));\n // And binds tighter than Or, so no parens needed around the And\n assert_eq!(\n expr.to_string(),\n \"cursor-at-start && !input-empty || no-results\"\n );\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/keybindings/defaults.rs", "content": "use std::collections::HashMap;\n\nuse atuin_client::settings::{KeyBindingConfig, Settings};\nuse tracing::warn;\n\nuse super::actions::Action;\nuse super::conditions::{ConditionAtom, ConditionExpr};\nuse super::key::KeyInput;\nuse super::keymap::{KeyBinding, KeyRule, Keymap};\n\n/// Helper to bind a scroll key with optional exit behavior.\n///\n/// When `scroll_exits` is true AND the key scrolls toward index 0 (the newest\n/// entry), we add a conditional rule: at `ListAtStart` → `Exit`, otherwise →\n/// the scroll action.\n///\n/// Whether a key scrolls toward index 0 depends on the `invert` setting:\n/// - Non-inverted: \"down\" / \"j\" move toward index 0, \"up\" / \"k\" move away\n/// - Inverted: \"up\" / \"k\" move toward index 0, \"down\" / \"j\" move away\n///\n/// If `toward_index_zero` is false, or `scroll_exits` is false, we just bind\n/// the key to the plain scroll action (no exit).\nfn bind_scroll_key(\n km: &mut Keymap,\n key_str: &str,\n action: Action,\n toward_index_zero: bool,\n scroll_exits: bool,\n) {\n let k = key(key_str);\n if scroll_exits && toward_index_zero {\n km.bind_conditional(\n k,\n vec![\n KeyRule::when(ConditionAtom::ListAtStart, Action::Exit),\n KeyRule::always(action),\n ],\n );\n } else {\n km.bind(k, action);\n }\n}\n\n/// Helper to parse a key string, panicking on invalid keys (these are all\n/// compile-time-known strings).\nfn key(s: &str) -> KeyInput {\n KeyInput::parse(s).unwrap_or_else(|e| panic!(\"invalid default key {s:?}: {e}\"))\n}\n\n/// All five keymaps bundled together.\n#[derive(Debug, Clone)]\npub struct KeymapSet {\n pub emacs: Keymap,\n pub vim_normal: Keymap,\n pub vim_insert: Keymap,\n pub inspector: Keymap,\n pub prefix: Keymap,\n}\n\n// ---------------------------------------------------------------------------\n// Common bindings shared across search-tab keymaps\n// ---------------------------------------------------------------------------\n\n/// Add the bindings that are common to all search-tab keymaps:\n/// ctrl-c, ctrl-g, ctrl-o, and tab.\n///\n/// Note: `esc`/`ctrl-[` are NOT included here because their behavior differs\n/// between emacs (exit), vim-normal (exit), and vim-insert (enter normal mode).\nfn add_common_bindings(km: &mut Keymap) {\n km.bind(key(\"ctrl-c\"), Action::ReturnOriginal);\n km.bind(key(\"ctrl-g\"), Action::ReturnOriginal);\n km.bind(key(\"ctrl-o\"), Action::ToggleTab);\n\n // Tab: always returns selection without executing (unlike Enter which respects enter_accept)\n km.bind(key(\"tab\"), Action::ReturnSelection);\n}\n\n/// Returns `Accept` or `ReturnSelection` based on the `enter_accept` setting.\nfn accept_action(settings: &Settings) -> Action {\n if settings.enter_accept {\n Action::Accept\n } else {\n Action::ReturnSelection\n }\n}\n\n// ---------------------------------------------------------------------------\n// Emacs keymap (also base for vim-insert)\n// ---------------------------------------------------------------------------\n\n/// Build the default emacs keymap. This encodes the behavior from\n/// `handle_key_input` common section + `handle_search_input` shared section.\n///\n/// The `settings` parameter is used for:\n/// - `keys.prefix` — which ctrl-key enters prefix mode\n/// - `keys.scroll_exits`, `invert` — scroll-at-boundary exit behavior\n/// - `keys.accept_past_line_end` — right arrow at end of line accepts\n/// - `keys.exit_past_line_start` — left arrow at start of line exits\n/// - `keys.accept_past_line_start` — left arrow at start accepts (overrides exit)\n/// - `keys.accept_with_backspace` — backspace at start of line accepts\n/// - `ctrl_n_shortcuts` — whether alt or ctrl is used for numeric shortcuts\n// Keymap builder that enumerates every default binding; not worth splitting.\n#[allow(clippy::too_many_lines)]\npub fn default_emacs_keymap(settings: &Settings) -> Keymap {\n let mut km = Keymap::new();\n add_common_bindings(&mut km);\n\n let accept = accept_action(settings);\n\n // esc / ctrl-[ → exit\n km.bind(key(\"esc\"), Action::Exit);\n km.bind(key(\"ctrl-[\"), Action::Exit);\n\n // Prefix key: ctrl- → enter prefix mode\n let prefix_char = settings.keys.prefix.chars().next().unwrap_or('a');\n km.bind(key(&format!(\"ctrl-{prefix_char}\")), Action::EnterPrefixMode);\n\n // --- Accept / navigation edge behaviors (from [keys] settings) ---\n\n // right: behavior at end of line\n if settings.keys.accept_past_line_end {\n km.bind_conditional(\n key(\"right\"),\n vec![\n KeyRule::when(ConditionAtom::CursorAtEnd, Action::ReturnSelection),\n KeyRule::always(Action::CursorRight),\n ],\n );\n } else {\n km.bind(key(\"right\"), Action::CursorRight);\n }\n\n // left: behavior at start of line\n // accept_past_line_start takes precedence over exit_past_line_start\n if settings.keys.accept_past_line_start {\n km.bind_conditional(\n key(\"left\"),\n vec![\n KeyRule::when(ConditionAtom::CursorAtStart, Action::ReturnSelection),\n KeyRule::always(Action::CursorLeft),\n ],\n );\n } else if settings.keys.exit_past_line_start {\n km.bind_conditional(\n key(\"left\"),\n vec![\n KeyRule::when(ConditionAtom::CursorAtStart, Action::Exit),\n KeyRule::always(Action::CursorLeft),\n ],\n );\n } else {\n km.bind(key(\"left\"), Action::CursorLeft);\n }\n\n // down/up: scroll with optional exit at boundary.\n // Non-inverted: down moves toward index 0 (can exit); up moves away (no exit).\n // Inverted: up moves toward index 0 (can exit); down moves away (no exit).\n let scroll_exits = settings.keys.scroll_exits;\n let invert = settings.invert;\n bind_scroll_key(&mut km, \"down\", Action::SelectNext, !invert, scroll_exits);\n bind_scroll_key(&mut km, \"up\", Action::SelectPrevious, invert, scroll_exits);\n\n // backspace: behavior at start of line\n if settings.keys.accept_with_backspace {\n km.bind_conditional(\n key(\"backspace\"),\n vec![\n KeyRule::when(ConditionAtom::CursorAtStart, Action::ReturnSelection),\n KeyRule::always(Action::DeleteCharBefore),\n ],\n );\n } else {\n km.bind(key(\"backspace\"), Action::DeleteCharBefore);\n }\n\n // --- Accept ---\n km.bind(key(\"enter\"), accept.clone());\n km.bind(key(\"ctrl-m\"), accept);\n\n // --- Copy ---\n km.bind(key(\"ctrl-y\"), Action::Copy);\n\n // --- Numeric shortcuts (alt-1..9 by default, ctrl-1..9 if ctrl_n_shortcuts) ---\n // These return the selection without executing, regardless of enter_accept.\n let num_mod = if settings.ctrl_n_shortcuts {\n \"ctrl\"\n } else {\n \"alt\"\n };\n for n in 1..=9u8 {\n km.bind(\n key(&format!(\"{num_mod}-{n}\")),\n Action::ReturnSelectionNth(n),\n );\n }\n\n // --- Cursor movement ---\n km.bind(key(\"ctrl-left\"), Action::CursorWordLeft);\n km.bind(key(\"alt-b\"), Action::CursorWordLeft);\n km.bind(key(\"ctrl-b\"), Action::CursorLeft);\n km.bind(key(\"ctrl-right\"), Action::CursorWordRight);\n km.bind(key(\"alt-f\"), Action::CursorWordRight);\n km.bind(key(\"ctrl-f\"), Action::CursorRight);\n km.bind(key(\"home\"), Action::CursorStart);\n // ctrl-a → CursorStart only if prefix char is NOT 'a'\n // (otherwise ctrl-a is already bound to EnterPrefixMode above)\n if prefix_char != 'a' {\n km.bind(key(\"ctrl-a\"), Action::CursorStart);\n }\n km.bind(key(\"ctrl-e\"), Action::CursorEnd);\n km.bind(key(\"end\"), Action::CursorEnd);\n\n // --- Editing ---\n km.bind(key(\"ctrl-backspace\"), Action::DeleteWordBefore);\n km.bind(key(\"ctrl-h\"), Action::DeleteCharBefore);\n km.bind(key(\"ctrl-?\"), Action::DeleteCharBefore);\n km.bind(key(\"ctrl-delete\"), Action::DeleteWordAfter);\n km.bind(key(\"delete\"), Action::DeleteCharAfter);\n // ctrl-d: if input empty → return original, otherwise delete char\n km.bind_conditional(\n key(\"ctrl-d\"),\n vec![\n KeyRule::when(ConditionAtom::InputEmpty, Action::ReturnOriginal),\n KeyRule::always(Action::DeleteCharAfter),\n ],\n );\n km.bind(key(\"ctrl-w\"), Action::DeleteToWordBoundary);\n km.bind(key(\"ctrl-u\"), Action::ClearLine);\n\n // --- Search mode ---\n km.bind(key(\"ctrl-r\"), Action::CycleFilterMode);\n km.bind(key(\"ctrl-s\"), Action::CycleSearchMode);\n\n // --- Scroll (no exit) ---\n km.bind(key(\"ctrl-n\"), Action::SelectNext);\n km.bind(key(\"ctrl-j\"), Action::SelectNext);\n km.bind(key(\"ctrl-p\"), Action::SelectPrevious);\n km.bind(key(\"ctrl-k\"), Action::SelectPrevious);\n\n // --- Redraw ---\n km.bind(key(\"ctrl-l\"), Action::Redraw);\n\n // --- Page scroll ---\n km.bind(key(\"pagedown\"), Action::ScrollPageDown);\n km.bind(key(\"pageup\"), Action::ScrollPageUp);\n\n km\n}\n\n// ---------------------------------------------------------------------------\n// Vim Normal keymap\n// ---------------------------------------------------------------------------\n\n/// Build the default vim-normal keymap.\npub fn default_vim_normal_keymap(settings: &Settings) -> Keymap {\n let mut km = Keymap::new();\n add_common_bindings(&mut km);\n\n // esc / ctrl-[ → exit (vim-normal exits, unlike vim-insert)\n km.bind(key(\"esc\"), Action::Exit);\n km.bind(key(\"ctrl-[\"), Action::Exit);\n\n // Prefix key\n let prefix_char = settings.keys.prefix.chars().next().unwrap_or('a');\n km.bind(key(&format!(\"ctrl-{prefix_char}\")), Action::EnterPrefixMode);\n\n // --- Vim navigation ---\n // j/k: scroll with optional exit at boundary.\n let scroll_exits = settings.keys.scroll_exits;\n let invert = settings.invert;\n bind_scroll_key(&mut km, \"j\", Action::SelectNext, !invert, scroll_exits);\n bind_scroll_key(&mut km, \"k\", Action::SelectPrevious, invert, scroll_exits);\n km.bind(key(\"h\"), Action::CursorLeft);\n km.bind(key(\"l\"), Action::CursorRight);\n\n // --- Vim cursor movement ---\n km.bind(key(\"0\"), Action::CursorStart);\n km.bind(key(\"$\"), Action::CursorEnd);\n km.bind(key(\"w\"), Action::CursorWordRight);\n km.bind(key(\"b\"), Action::CursorWordLeft);\n km.bind(key(\"e\"), Action::CursorWordEnd);\n\n // --- Vim editing ---\n km.bind(key(\"x\"), Action::DeleteCharAfter);\n km.bind(key(\"d d\"), Action::ClearLine);\n km.bind(key(\"D\"), Action::ClearToEnd);\n km.bind(key(\"C\"), Action::VimChangeToEnd);\n\n // --- Mode switching ---\n km.bind(key(\"?\"), Action::VimSearchInsert);\n km.bind(key(\"/\"), Action::VimSearchInsert);\n km.bind(key(\"a\"), Action::VimEnterInsertAfter);\n km.bind(key(\"A\"), Action::VimEnterInsertAtEnd);\n km.bind(key(\"i\"), Action::VimEnterInsert);\n km.bind(key(\"I\"), Action::VimEnterInsertAtStart);\n\n // --- Numeric shortcuts (return selection without executing) ---\n for n in 1..=9u8 {\n km.bind(key(&n.to_string()), Action::ReturnSelectionNth(n));\n }\n\n // --- Half/full page scroll ---\n km.bind(key(\"ctrl-u\"), Action::ScrollHalfPageUp);\n km.bind(key(\"ctrl-d\"), Action::ScrollHalfPageDown);\n km.bind(key(\"ctrl-b\"), Action::ScrollPageUp);\n km.bind(key(\"ctrl-f\"), Action::ScrollPageDown);\n\n // --- Jump ---\n km.bind(key(\"G\"), Action::ScrollToBottom);\n km.bind(key(\"g g\"), Action::ScrollToTop);\n km.bind(key(\"H\"), Action::ScrollToScreenTop);\n km.bind(key(\"M\"), Action::ScrollToScreenMiddle);\n km.bind(key(\"L\"), Action::ScrollToScreenBottom);\n\n // --- Arrow keys (same as emacs for convenience) ---\n bind_scroll_key(&mut km, \"down\", Action::SelectNext, !invert, scroll_exits);\n bind_scroll_key(&mut km, \"up\", Action::SelectPrevious, invert, scroll_exits);\n\n // --- Page scroll ---\n km.bind(key(\"pagedown\"), Action::ScrollPageDown);\n km.bind(key(\"pageup\"), Action::ScrollPageUp);\n\n // --- Accept ---\n let accept = accept_action(settings);\n km.bind(key(\"enter\"), accept);\n\n km\n}\n\n// ---------------------------------------------------------------------------\n// Vim Insert keymap\n// ---------------------------------------------------------------------------\n\n/// Build the default vim-insert keymap. This clones the emacs keymap and\n/// overlays vim-insert-specific bindings (esc → enter normal mode).\npub fn default_vim_insert_keymap(settings: &Settings) -> Keymap {\n let mut km = default_emacs_keymap(settings);\n\n // Override esc and ctrl-[ to enter normal mode instead of exiting\n km.bind(key(\"esc\"), Action::VimEnterNormal);\n km.bind(key(\"ctrl-[\"), Action::VimEnterNormal);\n\n km\n}\n\n// ---------------------------------------------------------------------------\n// Inspector keymap\n// ---------------------------------------------------------------------------\n\n/// Build the default inspector keymap (tab index 1).\n///\n/// The inspector shows details about the selected history item and has no\n/// text input, so we build a minimal keymap with only inspector-relevant\n/// bindings. We respect the user's `keymap_mode` to provide vim-style j/k\n/// navigation for vim users.\npub fn default_inspector_keymap(settings: &Settings) -> Keymap {\n use atuin_client::settings::KeymapMode;\n\n let mut km = Keymap::new();\n\n // Common bindings (same as search tab)\n km.bind(key(\"ctrl-c\"), Action::ReturnOriginal);\n km.bind(key(\"ctrl-g\"), Action::ReturnOriginal);\n km.bind(key(\"esc\"), Action::Exit);\n km.bind(key(\"ctrl-[\"), Action::Exit);\n km.bind(key(\"tab\"), Action::ReturnSelection);\n km.bind(key(\"ctrl-o\"), Action::ToggleTab);\n\n // Accept behavior respects enter_accept setting\n let accept = if settings.enter_accept {\n Action::Accept\n } else {\n Action::ReturnSelection\n };\n km.bind(key(\"enter\"), accept);\n\n // Inspector-specific: delete history entry\n km.bind(key(\"ctrl-d\"), Action::Delete);\n\n // Inspector navigation\n km.bind(key(\"up\"), Action::InspectPrevious);\n km.bind(key(\"down\"), Action::InspectNext);\n km.bind(key(\"pageup\"), Action::InspectPrevious);\n km.bind(key(\"pagedown\"), Action::InspectNext);\n\n // For vim users, add j/k navigation\n if matches!(\n settings.keymap_mode,\n KeymapMode::VimNormal | KeymapMode::VimInsert\n ) {\n km.bind(key(\"j\"), Action::InspectNext);\n km.bind(key(\"k\"), Action::InspectPrevious);\n }\n\n km\n}\n\n// ---------------------------------------------------------------------------\n// Prefix keymap\n// ---------------------------------------------------------------------------\n\n/// Build the default prefix keymap (active after ctrl-a prefix).\npub fn default_prefix_keymap() -> Keymap {\n let mut km = Keymap::new();\n\n km.bind(key(\"d\"), Action::Delete);\n km.bind(key(\"a\"), Action::CursorStart);\n km.bind_conditional(\n key(\"c\"),\n vec![\n KeyRule::when(ConditionAtom::HasContext, Action::ClearContext),\n KeyRule::always(Action::SwitchContext),\n ],\n );\n\n km\n}\n\n// ---------------------------------------------------------------------------\n// KeymapSet construction\n// ---------------------------------------------------------------------------\n\n// ---------------------------------------------------------------------------\n// Config → Keymap conversion\n// ---------------------------------------------------------------------------\n\n/// Convert a `KeyBindingConfig` (from TOML) into a `KeyBinding`.\n/// Returns `Err` if an action name or condition expression is invalid.\nfn parse_binding_config(config: &KeyBindingConfig) -> Result {\n match config {\n KeyBindingConfig::Simple(action_str) => {\n let action = Action::from_str(action_str)?;\n Ok(KeyBinding::simple(action))\n }\n KeyBindingConfig::Rules(rules) => {\n let mut parsed_rules = Vec::with_capacity(rules.len());\n for rule_cfg in rules {\n let action = Action::from_str(&rule_cfg.action)?;\n let rule = match &rule_cfg.when {\n None => KeyRule::always(action),\n Some(cond_str) => {\n let cond = ConditionExpr::parse(cond_str)?;\n KeyRule::when(cond, action)\n }\n };\n parsed_rules.push(rule);\n }\n Ok(KeyBinding::conditional(parsed_rules))\n }\n }\n}\n\n/// Apply a map of key-string → binding-config overrides to a keymap.\n/// Per-key override replaces the entire rule list for that key.\n/// Invalid keys or action names are logged and skipped.\nfn apply_config_to_keymap(keymap: &mut Keymap, overrides: &HashMap) {\n for (key_str, binding_cfg) in overrides {\n let key = match KeyInput::parse(key_str) {\n Ok(k) => k,\n Err(e) => {\n warn!(\"invalid key in keymap config: {key_str:?}: {e}\");\n continue;\n }\n };\n match parse_binding_config(binding_cfg) {\n Ok(binding) => {\n keymap.bindings.insert(key, binding);\n }\n Err(e) => {\n warn!(\"invalid binding for {key_str:?} in keymap config: {e}\");\n }\n }\n }\n}\n\nimpl KeymapSet {\n /// Build the complete set of default keymaps from settings.\n pub fn defaults(settings: &Settings) -> Self {\n KeymapSet {\n emacs: default_emacs_keymap(settings),\n vim_normal: default_vim_normal_keymap(settings),\n vim_insert: default_vim_insert_keymap(settings),\n inspector: default_inspector_keymap(settings),\n prefix: default_prefix_keymap(),\n }\n }\n\n /// Build keymaps from settings, applying any user `[keymap]` overrides.\n ///\n /// Precedence rules:\n /// - If `[keymap]` has any entries, `[keys]` is **ignored entirely**.\n /// Defaults are built with standard `[keys]` values, then `[keymap]`\n /// overrides are applied per-key.\n /// - If `[keymap]` is empty/absent, `[keys]` customizes the defaults\n /// (current behavior for backward compatibility).\n pub fn from_settings(settings: &Settings) -> Self {\n use atuin_client::settings::Keys;\n\n if settings.keymap.is_empty() {\n // No [keymap] section → use [keys] to customize defaults\n Self::defaults(settings)\n } else {\n // [keymap] present → ignore [keys], use standard defaults as base\n let mut base_settings = settings.clone();\n base_settings.keys = Keys::standard_defaults();\n let mut set = Self::defaults(&base_settings);\n set.apply_config(settings);\n set\n }\n }\n\n /// Apply user keymap config overrides to all modes.\n fn apply_config(&mut self, settings: &Settings) {\n let config = &settings.keymap;\n apply_config_to_keymap(&mut self.emacs, &config.emacs);\n apply_config_to_keymap(&mut self.vim_normal, &config.vim_normal);\n apply_config_to_keymap(&mut self.vim_insert, &config.vim_insert);\n apply_config_to_keymap(&mut self.inspector, &config.inspector);\n apply_config_to_keymap(&mut self.prefix, &config.prefix);\n }\n}\n\n// ---------------------------------------------------------------------------\n// Tests\n// ---------------------------------------------------------------------------\n\n#[cfg(test)]\nmod tests {\n use super::*;\n use crate::command::client::search::keybindings::conditions::EvalContext;\n\n fn make_ctx(cursor: usize, width: usize, selected: usize, len: usize) -> EvalContext {\n EvalContext {\n cursor_position: cursor,\n input_width: width,\n input_byte_len: width,\n selected_index: selected,\n results_len: len,\n original_input_empty: false,\n has_context: false,\n }\n }\n\n fn default_settings() -> Settings {\n Settings::utc()\n }\n\n // -- Emacs keymap tests --\n\n #[test]\n fn emacs_ctrl_c_returns_original() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"ctrl-c\"), &ctx),\n Some(Action::ReturnOriginal)\n );\n }\n\n #[test]\n fn emacs_esc_exits() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"esc\"), &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn emacs_tab_returns_selection() {\n // enter_accept=false in test defaults → ReturnSelection\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"tab\"), &ctx), Some(Action::ReturnSelection));\n }\n\n #[test]\n fn emacs_enter_returns_selection() {\n // enter_accept=false in test defaults → ReturnSelection\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"enter\"), &ctx),\n Some(Action::ReturnSelection)\n );\n }\n\n #[test]\n fn emacs_enter_accept_true_uses_accept() {\n let mut settings = default_settings();\n settings.enter_accept = true;\n let km = default_emacs_keymap(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"enter\"), &ctx), Some(Action::Accept));\n assert_eq!(km.resolve(&key(\"tab\"), &ctx), Some(Action::ReturnSelection));\n }\n\n #[test]\n fn emacs_right_at_end_returns_selection() {\n let km = default_emacs_keymap(&default_settings());\n // cursor at end of \"hello\" (width 5)\n let ctx = make_ctx(5, 5, 0, 10);\n assert_eq!(\n km.resolve(&key(\"right\"), &ctx),\n Some(Action::ReturnSelection)\n );\n }\n\n #[test]\n fn emacs_right_not_at_end_moves() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(2, 5, 0, 10);\n assert_eq!(km.resolve(&key(\"right\"), &ctx), Some(Action::CursorRight));\n }\n\n #[test]\n fn emacs_left_at_start_exits() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(0, 5, 0, 10);\n assert_eq!(km.resolve(&key(\"left\"), &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn emacs_left_not_at_start_moves() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(3, 5, 0, 10);\n assert_eq!(km.resolve(&key(\"left\"), &ctx), Some(Action::CursorLeft));\n }\n\n #[test]\n fn emacs_down_at_start_exits() {\n let km = default_emacs_keymap(&default_settings());\n // selected=0 → ListAtStart → Exit\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"down\"), &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn emacs_down_not_at_start_selects_next() {\n let km = default_emacs_keymap(&default_settings());\n // selected=5 → not at start → SelectNext\n let ctx = make_ctx(0, 0, 5, 10);\n assert_eq!(km.resolve(&key(\"down\"), &ctx), Some(Action::SelectNext));\n }\n\n #[test]\n fn emacs_up_selects_previous() {\n let km = default_emacs_keymap(&default_settings());\n // Non-inverted: up never exits (moves away from index 0)\n let ctx = make_ctx(0, 0, 5, 10);\n assert_eq!(km.resolve(&key(\"up\"), &ctx), Some(Action::SelectPrevious));\n }\n\n #[test]\n fn emacs_ctrl_d_empty_returns_original() {\n let km = default_emacs_keymap(&default_settings());\n // input empty (byte_len = 0)\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"ctrl-d\"), &ctx),\n Some(Action::ReturnOriginal)\n );\n }\n\n #[test]\n fn emacs_ctrl_d_nonempty_deletes() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(2, 5, 0, 10);\n assert_eq!(\n km.resolve(&key(\"ctrl-d\"), &ctx),\n Some(Action::DeleteCharAfter)\n );\n }\n\n #[test]\n fn emacs_ctrl_n_selects_next_no_exit_condition() {\n let km = default_emacs_keymap(&default_settings());\n // at start, but ctrl-n should NOT exit (no exit condition bound)\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"ctrl-n\"), &ctx), Some(Action::SelectNext));\n }\n\n #[test]\n fn emacs_prefix_key_enters_prefix() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"ctrl-a\"), &ctx),\n Some(Action::EnterPrefixMode)\n );\n }\n\n #[test]\n fn emacs_home_cursor_start() {\n let km = default_emacs_keymap(&default_settings());\n let ctx = make_ctx(5, 10, 0, 10);\n assert_eq!(km.resolve(&key(\"home\"), &ctx), Some(Action::CursorStart));\n }\n\n // -- Vim Normal keymap tests --\n\n #[test]\n fn vim_normal_j_at_start_exits() {\n let km = default_vim_normal_keymap(&default_settings());\n // selected=0 → ListAtStart → Exit (non-inverted: j moves toward index 0)\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"j\"), &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn vim_normal_j_not_at_start_selects_next() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 5, 10);\n assert_eq!(km.resolve(&key(\"j\"), &ctx), Some(Action::SelectNext));\n }\n\n #[test]\n fn vim_normal_k_selects_previous() {\n let km = default_vim_normal_keymap(&default_settings());\n // Non-inverted: k never exits (moves away from index 0)\n let ctx = make_ctx(0, 0, 5, 10);\n assert_eq!(km.resolve(&key(\"k\"), &ctx), Some(Action::SelectPrevious));\n }\n\n #[test]\n fn vim_normal_i_enters_insert() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"i\"), &ctx), Some(Action::VimEnterInsert));\n }\n\n #[test]\n fn vim_normal_slash_search_insert() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"/\"), &ctx), Some(Action::VimSearchInsert));\n }\n\n #[test]\n fn vim_normal_gg_scroll_to_top() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 50, 100);\n assert_eq!(km.resolve(&key(\"g g\"), &ctx), Some(Action::ScrollToTop));\n }\n\n #[test]\n fn vim_normal_big_g_scroll_to_bottom() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 50, 100);\n assert_eq!(km.resolve(&key(\"G\"), &ctx), Some(Action::ScrollToBottom));\n }\n\n #[test]\n fn vim_normal_numeric_returns_selection() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"3\"), &ctx),\n Some(Action::ReturnSelectionNth(3))\n );\n }\n\n #[test]\n fn vim_normal_ctrl_u_half_page_up() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 50, 100);\n assert_eq!(\n km.resolve(&key(\"ctrl-u\"), &ctx),\n Some(Action::ScrollHalfPageUp)\n );\n }\n\n #[test]\n fn vim_normal_screen_jumps() {\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 50, 100);\n assert_eq!(km.resolve(&key(\"H\"), &ctx), Some(Action::ScrollToScreenTop));\n assert_eq!(\n km.resolve(&key(\"M\"), &ctx),\n Some(Action::ScrollToScreenMiddle)\n );\n assert_eq!(\n km.resolve(&key(\"L\"), &ctx),\n Some(Action::ScrollToScreenBottom)\n );\n }\n\n #[test]\n fn vim_normal_enter_returns_selection() {\n // enter_accept=false in test defaults → ReturnSelection\n let km = default_vim_normal_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"enter\"), &ctx),\n Some(Action::ReturnSelection)\n );\n }\n\n #[test]\n fn vim_normal_enter_accept_true_uses_accept() {\n let mut settings = default_settings();\n settings.enter_accept = true;\n let km = default_vim_normal_keymap(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"enter\"), &ctx), Some(Action::Accept));\n }\n\n // -- Vim Insert keymap tests --\n\n #[test]\n fn vim_insert_inherits_emacs_enter() {\n let km = default_vim_insert_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n // enter_accept=false → ReturnSelection\n assert_eq!(\n km.resolve(&key(\"enter\"), &ctx),\n Some(Action::ReturnSelection)\n );\n }\n\n #[test]\n fn vim_insert_esc_enters_normal() {\n let km = default_vim_insert_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"esc\"), &ctx), Some(Action::VimEnterNormal));\n }\n\n #[test]\n fn vim_insert_ctrl_bracket_enters_normal() {\n let km = default_vim_insert_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n km.resolve(&key(\"ctrl-[\"), &ctx),\n Some(Action::VimEnterNormal)\n );\n }\n\n #[test]\n fn vim_insert_inherits_emacs_ctrl_d() {\n let km = default_vim_insert_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n // input empty → return original\n assert_eq!(\n km.resolve(&key(\"ctrl-d\"), &ctx),\n Some(Action::ReturnOriginal)\n );\n }\n\n // -- Inspector keymap tests --\n\n #[test]\n fn inspector_ctrl_d_deletes() {\n let km = default_inspector_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"ctrl-d\"), &ctx), Some(Action::Delete));\n }\n\n #[test]\n fn inspector_up_inspects_previous() {\n let km = default_inspector_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"up\"), &ctx), Some(Action::InspectPrevious));\n }\n\n #[test]\n fn inspector_down_inspects_next() {\n let km = default_inspector_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"down\"), &ctx), Some(Action::InspectNext));\n }\n\n #[test]\n fn inspector_esc_exits() {\n let km = default_inspector_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"esc\"), &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn inspector_tab_returns_selection() {\n // enter_accept=false → ReturnSelection\n let km = default_inspector_keymap(&default_settings());\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"tab\"), &ctx), Some(Action::ReturnSelection));\n }\n\n // -- Prefix keymap tests --\n\n #[test]\n fn prefix_d_deletes() {\n let km = default_prefix_keymap();\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"d\"), &ctx), Some(Action::Delete));\n }\n\n #[test]\n fn prefix_a_cursor_start() {\n let km = default_prefix_keymap();\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"a\"), &ctx), Some(Action::CursorStart));\n }\n\n #[test]\n fn prefix_unknown_key_returns_none() {\n let km = default_prefix_keymap();\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(km.resolve(&key(\"x\"), &ctx), None);\n }\n\n // -- KeymapSet tests --\n\n #[test]\n fn keymap_set_defaults_builds() {\n let settings = default_settings();\n let set = KeymapSet::defaults(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n\n // Sanity check each keymap has bindings\n assert!(set.emacs.resolve(&key(\"ctrl-c\"), &ctx).is_some());\n assert!(set.vim_normal.resolve(&key(\"ctrl-c\"), &ctx).is_some());\n assert!(set.vim_insert.resolve(&key(\"ctrl-c\"), &ctx).is_some());\n assert!(set.inspector.resolve(&key(\"ctrl-c\"), &ctx).is_some());\n assert!(set.prefix.resolve(&key(\"d\"), &ctx).is_some());\n }\n\n // -- Settings-dependent behavior --\n\n #[test]\n fn custom_prefix_char() {\n let mut settings = default_settings();\n settings.keys.prefix = \"x\".to_string();\n let km = default_emacs_keymap(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n\n // ctrl-x should be prefix mode\n assert_eq!(\n km.resolve(&key(\"ctrl-x\"), &ctx),\n Some(Action::EnterPrefixMode)\n );\n // ctrl-a should now be CursorStart (not prefix)\n assert_eq!(km.resolve(&key(\"ctrl-a\"), &ctx), Some(Action::CursorStart));\n }\n\n #[test]\n fn ctrl_n_shortcuts_changes_numeric_modifier() {\n let mut settings = default_settings();\n settings.ctrl_n_shortcuts = true;\n let km = default_emacs_keymap(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n\n // ctrl-1 should work\n assert_eq!(\n km.resolve(&key(\"ctrl-1\"), &ctx),\n Some(Action::ReturnSelectionNth(1))\n );\n // alt-1 should NOT be bound\n assert_eq!(km.resolve(&key(\"alt-1\"), &ctx), None);\n }\n\n #[test]\n fn default_alt_numeric_shortcuts() {\n let settings = default_settings();\n let km = default_emacs_keymap(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n\n // alt-1 should work by default\n assert_eq!(\n km.resolve(&key(\"alt-1\"), &ctx),\n Some(Action::ReturnSelectionNth(1))\n );\n }\n\n // -----------------------------------------------------------------------\n // Config parsing and merging tests\n // -----------------------------------------------------------------------\n\n #[test]\n fn parse_simple_binding_config() {\n use atuin_client::settings::KeyBindingConfig;\n let cfg = KeyBindingConfig::Simple(\"accept\".to_string());\n let binding = super::parse_binding_config(&cfg).unwrap();\n assert_eq!(binding.rules.len(), 1);\n assert!(binding.rules[0].condition.is_none());\n assert_eq!(binding.rules[0].action, Action::Accept);\n }\n\n #[test]\n fn parse_conditional_binding_config() {\n use atuin_client::settings::{KeyBindingConfig, KeyRuleConfig};\n let cfg = KeyBindingConfig::Rules(vec![\n KeyRuleConfig {\n when: Some(\"cursor-at-start\".to_string()),\n action: \"exit\".to_string(),\n },\n KeyRuleConfig {\n when: None,\n action: \"cursor-left\".to_string(),\n },\n ]);\n let binding = super::parse_binding_config(&cfg).unwrap();\n assert_eq!(binding.rules.len(), 2);\n assert!(binding.rules[0].condition.is_some());\n assert_eq!(binding.rules[0].action, Action::Exit);\n assert!(binding.rules[1].condition.is_none());\n assert_eq!(binding.rules[1].action, Action::CursorLeft);\n }\n\n #[test]\n fn parse_binding_config_invalid_action() {\n use atuin_client::settings::KeyBindingConfig;\n let cfg = KeyBindingConfig::Simple(\"not-a-real-action\".to_string());\n assert!(super::parse_binding_config(&cfg).is_err());\n }\n\n #[test]\n fn parse_binding_config_invalid_condition() {\n use atuin_client::settings::{KeyBindingConfig, KeyRuleConfig};\n let cfg = KeyBindingConfig::Rules(vec![KeyRuleConfig {\n when: Some(\"not-a-real-condition\".to_string()),\n action: \"exit\".to_string(),\n }]);\n assert!(super::parse_binding_config(&cfg).is_err());\n }\n\n #[test]\n fn config_override_replaces_key() {\n use atuin_client::settings::KeyBindingConfig;\n use std::collections::HashMap;\n\n let mut settings = default_settings();\n let set = KeymapSet::defaults(&settings);\n\n // Default: ctrl-c → ReturnOriginal\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n set.emacs.resolve(&key(\"ctrl-c\"), &ctx),\n Some(Action::ReturnOriginal)\n );\n\n // Override ctrl-c → Exit via config\n settings.keymap.emacs = HashMap::from([(\n \"ctrl-c\".to_string(),\n KeyBindingConfig::Simple(\"exit\".to_string()),\n )]);\n\n let set = KeymapSet::from_settings(&settings);\n assert_eq!(set.emacs.resolve(&key(\"ctrl-c\"), &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn config_override_preserves_unoverridden_keys() {\n use atuin_client::settings::KeyBindingConfig;\n use std::collections::HashMap;\n\n let mut settings = default_settings();\n // Override only ctrl-c; enter should keep its default\n settings.keymap.emacs = HashMap::from([(\n \"ctrl-c\".to_string(),\n KeyBindingConfig::Simple(\"exit\".to_string()),\n )]);\n\n let set = KeymapSet::from_settings(&settings);\n let ctx = make_ctx(0, 0, 0, 10);\n\n // ctrl-c overridden\n assert_eq!(set.emacs.resolve(&key(\"ctrl-c\"), &ctx), Some(Action::Exit));\n // enter still has default (enter_accept=false → ReturnSelection)\n assert_eq!(\n set.emacs.resolve(&key(\"enter\"), &ctx),\n Some(Action::ReturnSelection)\n );\n }\n\n #[test]\n fn config_conditional_override() {\n use atuin_client::settings::{KeyBindingConfig, KeyRuleConfig};\n use std::collections::HashMap;\n\n let mut settings = default_settings();\n // Override \"up\" with a custom conditional\n settings.keymap.emacs = HashMap::from([(\n \"up\".to_string(),\n KeyBindingConfig::Rules(vec![\n KeyRuleConfig {\n when: Some(\"no-results\".to_string()),\n action: \"exit\".to_string(),\n },\n KeyRuleConfig {\n when: None,\n action: \"select-previous\".to_string(),\n },\n ]),\n )]);\n\n let set = KeymapSet::from_settings(&settings);\n\n // With no results → exit\n let ctx = make_ctx(0, 0, 0, 0);\n assert_eq!(set.emacs.resolve(&key(\"up\"), &ctx), Some(Action::Exit));\n\n // With results → select-previous\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(\n set.emacs.resolve(&key(\"up\"), &ctx),\n Some(Action::SelectPrevious)\n );\n }\n\n #[test]\n fn from_settings_with_empty_config_equals_defaults() {\n let settings = default_settings();\n let defaults = KeymapSet::defaults(&settings);\n let from_settings = KeymapSet::from_settings(&settings);\n\n // Verify a sample of keys produce the same results\n let ctx = make_ctx(0, 0, 0, 10);\n let test_keys = [\n \"ctrl-c\", \"enter\", \"esc\", \"tab\", \"up\", \"down\", \"left\", \"right\",\n ];\n for k in &test_keys {\n assert_eq!(\n defaults.emacs.resolve(&key(k), &ctx),\n from_settings.emacs.resolve(&key(k), &ctx),\n \"mismatch for emacs key {k}\"\n );\n }\n }\n\n // -----------------------------------------------------------------------\n // Phase 5: [keys] vs [keymap] backward compatibility\n // -----------------------------------------------------------------------\n\n #[test]\n fn keymap_overrides_ignore_keys_section() {\n use atuin_client::settings::KeyBindingConfig;\n\n // Set up: [keys] disables scroll_exits, but [keymap] is present\n let mut settings = default_settings();\n settings.keys.scroll_exits = false;\n\n // Without [keymap], scroll_exits=false means no exit condition on down\n let set_legacy = KeymapSet::defaults(&settings);\n // At list-at-start (selected=0), down should still be SelectNext (no exit)\n let ctx_at_boundary = make_ctx(0, 0, 0, 10);\n assert_eq!(\n set_legacy.emacs.resolve(&key(\"down\"), &ctx_at_boundary),\n Some(Action::SelectNext),\n \"legacy: down at boundary should be SelectNext with scroll_exits=false\"\n );\n\n // With [keymap] present (even just one override), [keys] is ignored\n // so the standard defaults (scroll_exits=true) apply\n settings.keymap.emacs = HashMap::from([(\n \"ctrl-c\".to_string(),\n KeyBindingConfig::Simple(\"exit\".to_string()),\n )]);\n let set_keymap = KeymapSet::from_settings(&settings);\n\n // Not at boundary (selected=5): should SelectNext normally\n let ctx_not_at_boundary = make_ctx(0, 0, 5, 10);\n assert_eq!(\n set_keymap.emacs.resolve(&key(\"down\"), &ctx_not_at_boundary),\n Some(Action::SelectNext),\n \"keymap: down not at boundary should SelectNext\"\n );\n // At list-at-start (selected=0): should Exit (standard scroll_exits=true)\n assert_eq!(\n set_keymap.emacs.resolve(&key(\"down\"), &ctx_at_boundary),\n Some(Action::Exit),\n \"keymap: down at boundary should Exit (standard defaults restored)\"\n );\n }\n\n #[test]\n fn keymap_present_resets_to_standard_keys_defaults() {\n use atuin_client::settings::KeyBindingConfig;\n\n let mut settings = default_settings();\n // Disable all [keys] behaviors\n settings.keys.exit_past_line_start = false;\n settings.keys.accept_past_line_end = false;\n\n // Without [keymap], left should be plain CursorLeft\n let set_legacy = KeymapSet::defaults(&settings);\n let ctx_at_start = make_ctx(0, 5, 0, 10);\n assert_eq!(\n set_legacy.emacs.resolve(&key(\"left\"), &ctx_at_start),\n Some(Action::CursorLeft),\n \"legacy: left should be plain CursorLeft without exit_past_line_start\"\n );\n\n // Add a [keymap] entry (for a different key)\n settings.keymap.emacs = HashMap::from([(\n \"ctrl-c\".to_string(),\n KeyBindingConfig::Simple(\"exit\".to_string()),\n )]);\n let set_keymap = KeymapSet::from_settings(&settings);\n\n // Now left should use standard defaults (exit_past_line_start=true)\n // At cursor start → Exit\n assert_eq!(\n set_keymap.emacs.resolve(&key(\"left\"), &ctx_at_start),\n Some(Action::Exit),\n \"keymap: left at cursor start should exit (standard defaults)\"\n );\n\n // Right at cursor end should return selection (standard defaults: accept_past_line_end=true, enter_accept=false)\n let ctx_at_end = make_ctx(5, 5, 0, 10);\n assert_eq!(\n set_keymap.emacs.resolve(&key(\"right\"), &ctx_at_end),\n Some(Action::ReturnSelection),\n \"keymap: right at cursor end should return selection (standard defaults)\"\n );\n }\n\n #[test]\n fn keys_has_non_default_values_detection() {\n use atuin_client::settings::Keys;\n\n let standard = Keys::standard_defaults();\n assert!(!standard.has_non_default_values());\n\n let mut modified = Keys::standard_defaults();\n modified.scroll_exits = false;\n assert!(modified.has_non_default_values());\n\n let mut modified = Keys::standard_defaults();\n modified.prefix = \"x\".to_string();\n assert!(modified.has_non_default_values());\n }\n\n #[test]\n fn original_input_empty_condition_in_config() {\n use atuin_client::settings::{KeyBindingConfig, KeyRuleConfig};\n use std::collections::HashMap;\n\n let mut settings = default_settings();\n // Configure esc to: if original-input-empty -> return-query, else return-original\n settings.keymap.emacs = HashMap::from([(\n \"esc\".to_string(),\n KeyBindingConfig::Rules(vec![\n KeyRuleConfig {\n when: Some(\"original-input-empty\".to_string()),\n action: \"return-query\".to_string(),\n },\n KeyRuleConfig {\n when: None,\n action: \"return-original\".to_string(),\n },\n ]),\n )]);\n\n let set = KeymapSet::from_settings(&settings);\n\n // When original input was empty, should return-query\n let ctx_original_empty = EvalContext {\n cursor_position: 0,\n input_width: 5,\n input_byte_len: 5,\n selected_index: 0,\n results_len: 10,\n original_input_empty: true,\n has_context: false,\n };\n assert_eq!(\n set.emacs.resolve(&key(\"esc\"), &ctx_original_empty),\n Some(Action::ReturnQuery),\n \"esc with original_input_empty=true should return-query\"\n );\n\n // When original input was not empty, should return-original\n let ctx_original_not_empty = EvalContext {\n cursor_position: 0,\n input_width: 5,\n input_byte_len: 5,\n selected_index: 0,\n results_len: 10,\n original_input_empty: false,\n has_context: false,\n };\n assert_eq!(\n set.emacs.resolve(&key(\"esc\"), &ctx_original_not_empty),\n Some(Action::ReturnOriginal),\n \"esc with original_input_empty=false should return-original\"\n );\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/keybindings/key.rs", "content": "use std::fmt;\n\nuse ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers, MediaKeyCode};\nuse serde::{Deserialize, Deserializer, Serialize, Serializer};\n\n/// A single key press with modifiers (e.g. `ctrl-c`, `alt-f`, `enter`).\n#[derive(Debug, Clone, PartialEq, Eq, Hash)]\n#[allow(clippy::struct_excessive_bools)]\npub struct SingleKey {\n pub code: KeyCodeValue,\n pub ctrl: bool,\n pub alt: bool,\n pub shift: bool,\n pub super_key: bool,\n}\n\n/// The key code portion of a key press.\n#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]\npub enum KeyCodeValue {\n Char(char),\n Enter,\n Esc,\n Tab,\n Backspace,\n Delete,\n Insert,\n Up,\n Down,\n Left,\n Right,\n Home,\n End,\n PageUp,\n PageDown,\n Space,\n F(u8),\n Media(MediaKeyCode),\n}\n\n/// A key input that may be a single key or a multi-key sequence (e.g. `g g`).\n#[derive(Debug, Clone, PartialEq, Eq, Hash)]\npub enum KeyInput {\n Single(SingleKey),\n Sequence(Vec),\n}\n\nimpl SingleKey {\n /// Convert a crossterm `KeyEvent` into a `SingleKey`.\n pub fn from_event(event: &KeyEvent) -> Option {\n let ctrl = event.modifiers.contains(KeyModifiers::CONTROL);\n let alt = event.modifiers.contains(KeyModifiers::ALT);\n let shift = event.modifiers.contains(KeyModifiers::SHIFT);\n let super_key = event.modifiers.contains(KeyModifiers::SUPER);\n\n let code = match event.code {\n KeyCode::Char(' ') => KeyCodeValue::Space,\n KeyCode::Char(c) => {\n // If shift is the only modifier and it's an uppercase letter,\n // we store the uppercase char directly and clear the shift flag\n // since the case already encodes it.\n if shift && !ctrl && !alt && !super_key && c.is_ascii_uppercase() {\n return Some(SingleKey {\n code: KeyCodeValue::Char(c),\n ctrl: false,\n alt: false,\n shift: false,\n super_key: false,\n });\n }\n KeyCodeValue::Char(c)\n }\n KeyCode::Enter => KeyCodeValue::Enter,\n KeyCode::Esc => KeyCodeValue::Esc,\n KeyCode::Tab => KeyCodeValue::Tab,\n // BackTab is sent by many terminals for Shift+Tab\n KeyCode::BackTab => {\n return Some(SingleKey {\n code: KeyCodeValue::Tab,\n ctrl,\n alt,\n shift: true,\n super_key,\n });\n }\n KeyCode::Backspace => KeyCodeValue::Backspace,\n KeyCode::Delete => KeyCodeValue::Delete,\n KeyCode::Insert => KeyCodeValue::Insert,\n KeyCode::Up => KeyCodeValue::Up,\n KeyCode::Down => KeyCodeValue::Down,\n KeyCode::Left => KeyCodeValue::Left,\n KeyCode::Right => KeyCodeValue::Right,\n KeyCode::Home => KeyCodeValue::Home,\n KeyCode::End => KeyCodeValue::End,\n KeyCode::PageUp => KeyCodeValue::PageUp,\n KeyCode::PageDown => KeyCodeValue::PageDown,\n KeyCode::F(n) => KeyCodeValue::F(n),\n KeyCode::Media(m) => KeyCodeValue::Media(m),\n _ => return None,\n };\n\n Some(SingleKey {\n code,\n ctrl,\n alt,\n shift: if matches!(code, KeyCodeValue::Char(_)) {\n false\n } else {\n shift\n },\n super_key,\n })\n }\n\n /// Parse a key string like `\"ctrl-c\"`, `\"alt-f\"`, `\"enter\"`, `\"G\"`.\n pub fn parse(s: &str) -> Result {\n let s = s.trim();\n let parts: Vec<&str> = s.split('-').collect();\n\n let mut ctrl = false;\n let mut alt = false;\n let mut shift = false;\n let mut super_key = false;\n\n // All parts except the last are modifiers\n for &part in &parts[..parts.len() - 1] {\n match part.to_lowercase().as_str() {\n \"ctrl\" => ctrl = true,\n \"alt\" => alt = true,\n \"shift\" => shift = true,\n \"super\" | \"cmd\" | \"win\" => super_key = true,\n _ => return Err(format!(\"unknown modifier: {part}\")),\n }\n }\n\n let key_part = parts[parts.len() - 1];\n let code = match key_part.to_lowercase().as_str() {\n \"enter\" | \"return\" => KeyCodeValue::Enter,\n \"esc\" | \"escape\" => KeyCodeValue::Esc,\n \"tab\" => KeyCodeValue::Tab,\n \"backspace\" => KeyCodeValue::Backspace,\n \"delete\" | \"del\" => KeyCodeValue::Delete,\n \"insert\" | \"ins\" => KeyCodeValue::Insert,\n \"up\" => KeyCodeValue::Up,\n \"down\" => KeyCodeValue::Down,\n \"left\" => KeyCodeValue::Left,\n \"right\" => KeyCodeValue::Right,\n \"home\" => KeyCodeValue::Home,\n \"end\" => KeyCodeValue::End,\n \"pageup\" => KeyCodeValue::PageUp,\n \"pagedown\" => KeyCodeValue::PageDown,\n \"space\" => KeyCodeValue::Space,\n s if s.starts_with('f') && s.len() > 1 => {\n // Parse function keys like \"f1\", \"f12\"\n if let Ok(n) = s[1..].parse::() {\n if (1..=24).contains(&n) {\n KeyCodeValue::F(n)\n } else {\n return Err(format!(\"function key out of range: {key_part}\"));\n }\n } else {\n return Err(format!(\"unknown key: {key_part}\"));\n }\n }\n \"[\" => KeyCodeValue::Char('['),\n \"]\" => KeyCodeValue::Char(']'),\n \"?\" => KeyCodeValue::Char('?'),\n \"/\" => KeyCodeValue::Char('/'),\n \"$\" => KeyCodeValue::Char('$'),\n // Media keys (no dashes - the parser splits on dash for modifiers)\n \"play\" => KeyCodeValue::Media(MediaKeyCode::Play),\n \"pause\" => KeyCodeValue::Media(MediaKeyCode::Pause),\n \"playpause\" => KeyCodeValue::Media(MediaKeyCode::PlayPause),\n \"stop\" => KeyCodeValue::Media(MediaKeyCode::Stop),\n \"fastforward\" => KeyCodeValue::Media(MediaKeyCode::FastForward),\n \"rewind\" => KeyCodeValue::Media(MediaKeyCode::Rewind),\n \"tracknext\" => KeyCodeValue::Media(MediaKeyCode::TrackNext),\n \"trackprevious\" => KeyCodeValue::Media(MediaKeyCode::TrackPrevious),\n \"record\" => KeyCodeValue::Media(MediaKeyCode::Record),\n \"lowervolume\" => KeyCodeValue::Media(MediaKeyCode::LowerVolume),\n \"raisevolume\" => KeyCodeValue::Media(MediaKeyCode::RaiseVolume),\n \"mutevolume\" | \"mute\" => KeyCodeValue::Media(MediaKeyCode::MuteVolume),\n _ => {\n let chars: Vec = key_part.chars().collect();\n if chars.len() == 1 {\n let c = chars[0];\n // An uppercase letter implies shift (unless shift already specified)\n if c.is_ascii_uppercase() && !ctrl && !alt && !super_key {\n return Ok(SingleKey {\n code: KeyCodeValue::Char(c),\n ctrl: false,\n alt: false,\n shift: false,\n super_key: false,\n });\n }\n KeyCodeValue::Char(c)\n } else {\n return Err(format!(\"unknown key: {key_part}\"));\n }\n }\n };\n\n Ok(SingleKey {\n code,\n ctrl,\n alt,\n shift,\n super_key,\n })\n }\n}\n\nimpl fmt::Display for SingleKey {\n fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n if self.super_key {\n write!(f, \"super-\")?;\n }\n if self.ctrl {\n write!(f, \"ctrl-\")?;\n }\n if self.alt {\n write!(f, \"alt-\")?;\n }\n if self.shift {\n write!(f, \"shift-\")?;\n }\n match &self.code {\n KeyCodeValue::Char(c) => write!(f, \"{c}\"),\n KeyCodeValue::Enter => write!(f, \"enter\"),\n KeyCodeValue::Esc => write!(f, \"esc\"),\n KeyCodeValue::Tab => write!(f, \"tab\"),\n KeyCodeValue::Backspace => write!(f, \"backspace\"),\n KeyCodeValue::Delete => write!(f, \"delete\"),\n KeyCodeValue::Insert => write!(f, \"insert\"),\n KeyCodeValue::Up => write!(f, \"up\"),\n KeyCodeValue::Down => write!(f, \"down\"),\n KeyCodeValue::Left => write!(f, \"left\"),\n KeyCodeValue::Right => write!(f, \"right\"),\n KeyCodeValue::Home => write!(f, \"home\"),\n KeyCodeValue::End => write!(f, \"end\"),\n KeyCodeValue::PageUp => write!(f, \"pageup\"),\n KeyCodeValue::PageDown => write!(f, \"pagedown\"),\n KeyCodeValue::Space => write!(f, \"space\"),\n KeyCodeValue::F(n) => write!(f, \"f{n}\"),\n KeyCodeValue::Media(m) => match m {\n MediaKeyCode::Play => write!(f, \"play\"),\n MediaKeyCode::Pause => write!(f, \"media-pause\"),\n MediaKeyCode::PlayPause => write!(f, \"playpause\"),\n MediaKeyCode::Stop => write!(f, \"stop\"),\n MediaKeyCode::FastForward => write!(f, \"fastforward\"),\n MediaKeyCode::Rewind => write!(f, \"rewind\"),\n MediaKeyCode::TrackNext => write!(f, \"tracknext\"),\n MediaKeyCode::TrackPrevious => write!(f, \"trackprevious\"),\n MediaKeyCode::Record => write!(f, \"record\"),\n MediaKeyCode::LowerVolume => write!(f, \"lowervolume\"),\n MediaKeyCode::RaiseVolume => write!(f, \"raisevolume\"),\n MediaKeyCode::MuteVolume => write!(f, \"mutevolume\"),\n MediaKeyCode::Reverse => write!(f, \"reverse\"),\n },\n }\n }\n}\n\nimpl KeyInput {\n /// Parse a key input string. Supports multi-key sequences separated by spaces\n /// (e.g. `\"g g\"`).\n pub fn parse(s: &str) -> Result {\n let s = s.trim();\n // Check for space-separated multi-key sequences\n // But don't split \"space\" or modifier combos like \"ctrl-a\"\n let parts: Vec<&str> = s.split_whitespace().collect();\n if parts.len() > 1 {\n let keys: Result, String> =\n parts.iter().map(|p| SingleKey::parse(p)).collect();\n Ok(KeyInput::Sequence(keys?))\n } else {\n Ok(KeyInput::Single(SingleKey::parse(s)?))\n }\n }\n}\n\nimpl fmt::Display for KeyInput {\n fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {\n match self {\n KeyInput::Single(k) => write!(f, \"{k}\"),\n KeyInput::Sequence(keys) => {\n for (i, k) in keys.iter().enumerate() {\n if i > 0 {\n write!(f, \" \")?;\n }\n write!(f, \"{k}\")?;\n }\n Ok(())\n }\n }\n }\n}\n\nimpl Serialize for KeyInput {\n fn serialize(&self, serializer: S) -> Result {\n serializer.serialize_str(&self.to_string())\n }\n}\n\nimpl<'de> Deserialize<'de> for KeyInput {\n fn deserialize>(deserializer: D) -> Result {\n let s = String::deserialize(deserializer)?;\n KeyInput::parse(&s).map_err(serde::de::Error::custom)\n }\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\n\n #[test]\n fn parse_simple_keys() {\n let k = SingleKey::parse(\"a\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('a'));\n assert!(!k.ctrl && !k.alt && !k.shift);\n\n let k = SingleKey::parse(\"enter\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Enter);\n\n let k = SingleKey::parse(\"esc\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Esc);\n\n let k = SingleKey::parse(\"tab\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Tab);\n\n let k = SingleKey::parse(\"space\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Space);\n }\n\n #[test]\n fn parse_modifiers() {\n let k = SingleKey::parse(\"ctrl-c\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('c'));\n assert!(k.ctrl);\n assert!(!k.alt);\n\n let k = SingleKey::parse(\"alt-f\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('f'));\n assert!(k.alt);\n assert!(!k.ctrl);\n\n let k = SingleKey::parse(\"ctrl-alt-x\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('x'));\n assert!(k.ctrl && k.alt);\n }\n\n #[test]\n fn parse_uppercase_implies_no_shift_flag() {\n let k = SingleKey::parse(\"G\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('G'));\n assert!(!k.shift);\n assert!(!k.ctrl);\n }\n\n #[test]\n fn parse_special_chars() {\n let k = SingleKey::parse(\"ctrl-[\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('['));\n assert!(k.ctrl);\n\n let k = SingleKey::parse(\"?\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('?'));\n\n let k = SingleKey::parse(\"/\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('/'));\n }\n\n #[test]\n fn parse_multi_key_sequence() {\n let ki = KeyInput::parse(\"g g\").unwrap();\n match ki {\n KeyInput::Sequence(keys) => {\n assert_eq!(keys.len(), 2);\n assert_eq!(keys[0].code, KeyCodeValue::Char('g'));\n assert_eq!(keys[1].code, KeyCodeValue::Char('g'));\n }\n _ => panic!(\"expected sequence\"),\n }\n }\n\n #[test]\n fn display_round_trip() {\n let cases = [\"ctrl-c\", \"alt-f\", \"enter\", \"G\", \"tab\", \"pageup\"];\n for s in cases {\n let k = KeyInput::parse(s).unwrap();\n let display = k.to_string();\n let k2 = KeyInput::parse(&display).unwrap();\n assert_eq!(k, k2, \"round-trip failed for {s}\");\n }\n\n let ki = KeyInput::parse(\"g g\").unwrap();\n assert_eq!(ki.to_string(), \"g g\");\n }\n\n #[test]\n fn from_event_basic() {\n let event = KeyEvent::new(KeyCode::Char('c'), KeyModifiers::CONTROL);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('c'));\n assert!(k.ctrl);\n assert!(!k.alt);\n\n let event = KeyEvent::new(KeyCode::Enter, KeyModifiers::NONE);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Enter);\n }\n\n #[test]\n fn from_event_uppercase() {\n // Crossterm sends uppercase chars with SHIFT modifier\n let event = KeyEvent::new(KeyCode::Char('G'), KeyModifiers::SHIFT);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('G'));\n // shift flag should be cleared since the case encodes it\n assert!(!k.shift);\n }\n\n #[test]\n fn from_event_matches_parsed() {\n // Verify that from_event and parse produce the same SingleKey\n let event = KeyEvent::new(KeyCode::Char('c'), KeyModifiers::CONTROL);\n let from_event = SingleKey::from_event(&event).unwrap();\n let parsed = SingleKey::parse(\"ctrl-c\").unwrap();\n assert_eq!(from_event, parsed);\n\n let event = KeyEvent::new(KeyCode::Char('G'), KeyModifiers::SHIFT);\n let from_event = SingleKey::from_event(&event).unwrap();\n let parsed = SingleKey::parse(\"G\").unwrap();\n assert_eq!(from_event, parsed);\n }\n\n #[test]\n fn parse_super_modifier() {\n let k = SingleKey::parse(\"super-a\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('a'));\n assert!(k.super_key);\n assert!(!k.ctrl && !k.alt && !k.shift);\n\n // \"cmd\" is an alias for \"super\"\n let k2 = SingleKey::parse(\"cmd-a\").unwrap();\n assert_eq!(k, k2);\n\n // \"win\" is an alias for \"super\"\n let k3 = SingleKey::parse(\"win-a\").unwrap();\n assert_eq!(k, k3);\n }\n\n #[test]\n fn parse_super_with_other_modifiers() {\n let k = SingleKey::parse(\"super-ctrl-c\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('c'));\n assert!(k.super_key && k.ctrl);\n assert!(!k.alt && !k.shift);\n }\n\n #[test]\n fn display_super_modifier() {\n let k = SingleKey::parse(\"super-a\").unwrap();\n assert_eq!(k.to_string(), \"super-a\");\n\n let k = SingleKey::parse(\"super-ctrl-x\").unwrap();\n assert_eq!(k.to_string(), \"super-ctrl-x\");\n }\n\n #[test]\n fn display_round_trip_super() {\n let k = KeyInput::parse(\"super-a\").unwrap();\n let display = k.to_string();\n let k2 = KeyInput::parse(&display).unwrap();\n assert_eq!(k, k2, \"round-trip failed for super-a\");\n }\n\n #[test]\n fn from_event_super() {\n let event = KeyEvent::new(KeyCode::Char('a'), KeyModifiers::SUPER);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('a'));\n assert!(k.super_key);\n assert!(!k.ctrl && !k.alt && !k.shift);\n }\n\n #[test]\n fn from_event_super_matches_parsed() {\n let event = KeyEvent::new(KeyCode::Char('a'), KeyModifiers::SUPER);\n let from_event = SingleKey::from_event(&event).unwrap();\n let parsed = SingleKey::parse(\"super-a\").unwrap();\n assert_eq!(from_event, parsed);\n }\n\n #[test]\n fn super_uppercase_preserves_super() {\n // super-G should keep the super flag (unlike bare \"G\" which clears shift)\n let k = SingleKey::parse(\"super-G\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Char('G'));\n assert!(k.super_key);\n }\n\n #[test]\n fn parse_errors() {\n assert!(SingleKey::parse(\"ctrl-alt-shift-xxx\").is_err());\n assert!(SingleKey::parse(\"foobar-a\").is_err());\n }\n\n #[test]\n fn parse_function_keys() {\n let k = SingleKey::parse(\"f1\").unwrap();\n assert_eq!(k.code, KeyCodeValue::F(1));\n assert!(!k.ctrl && !k.alt && !k.shift);\n\n let k = SingleKey::parse(\"F12\").unwrap();\n assert_eq!(k.code, KeyCodeValue::F(12));\n\n let k = SingleKey::parse(\"ctrl-f5\").unwrap();\n assert_eq!(k.code, KeyCodeValue::F(5));\n assert!(k.ctrl);\n\n // F24 is valid (some keyboards have extended function keys)\n let k = SingleKey::parse(\"f24\").unwrap();\n assert_eq!(k.code, KeyCodeValue::F(24));\n\n // F0 and F25+ are invalid\n assert!(SingleKey::parse(\"f0\").is_err());\n assert!(SingleKey::parse(\"f25\").is_err());\n }\n\n #[test]\n fn from_event_function_keys() {\n let event = KeyEvent::new(KeyCode::F(1), KeyModifiers::NONE);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::F(1));\n\n let event = KeyEvent::new(KeyCode::F(12), KeyModifiers::CONTROL);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::F(12));\n assert!(k.ctrl);\n }\n\n #[test]\n fn display_function_keys() {\n let k = SingleKey::parse(\"f1\").unwrap();\n assert_eq!(k.to_string(), \"f1\");\n\n let k = SingleKey::parse(\"ctrl-f12\").unwrap();\n assert_eq!(k.to_string(), \"ctrl-f12\");\n }\n\n #[test]\n fn function_key_round_trip() {\n let cases = [\"f1\", \"f12\", \"ctrl-f5\", \"alt-f10\"];\n for s in cases {\n let k = KeyInput::parse(s).unwrap();\n let display = k.to_string();\n let k2 = KeyInput::parse(&display).unwrap();\n assert_eq!(k, k2, \"round-trip failed for {s}\");\n }\n }\n\n #[test]\n fn from_event_function_key_matches_parsed() {\n let event = KeyEvent::new(KeyCode::F(12), KeyModifiers::NONE);\n let from_event = SingleKey::from_event(&event).unwrap();\n let parsed = SingleKey::parse(\"f12\").unwrap();\n assert_eq!(from_event, parsed);\n }\n\n #[test]\n fn from_event_backtab_becomes_shift_tab() {\n // Many terminals send BackTab for Shift+Tab\n let event = KeyEvent::new(KeyCode::BackTab, KeyModifiers::NONE);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Tab);\n assert!(k.shift);\n assert!(!k.ctrl && !k.alt);\n }\n\n #[test]\n fn from_event_backtab_matches_parsed_shift_tab() {\n let event = KeyEvent::new(KeyCode::BackTab, KeyModifiers::NONE);\n let from_event = SingleKey::from_event(&event).unwrap();\n let parsed = SingleKey::parse(\"shift-tab\").unwrap();\n assert_eq!(from_event, parsed);\n }\n\n #[test]\n fn from_event_backtab_with_ctrl() {\n // BackTab with ctrl modifier\n let event = KeyEvent::new(KeyCode::BackTab, KeyModifiers::CONTROL);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Tab);\n assert!(k.shift);\n assert!(k.ctrl);\n }\n\n #[test]\n fn parse_insert_key() {\n let k = SingleKey::parse(\"insert\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Insert);\n assert!(!k.ctrl && !k.alt && !k.shift);\n\n let k = SingleKey::parse(\"ins\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Insert);\n\n let k = SingleKey::parse(\"ctrl-insert\").unwrap();\n assert_eq!(k.code, KeyCodeValue::Insert);\n assert!(k.ctrl);\n }\n\n #[test]\n fn from_event_insert_key() {\n let event = KeyEvent::new(KeyCode::Insert, KeyModifiers::NONE);\n let k = SingleKey::from_event(&event).unwrap();\n assert_eq!(k.code, KeyCodeValue::Insert);\n }\n\n #[test]\n fn insert_key_round_trip() {\n let k = KeyInput::parse(\"insert\").unwrap();\n let display = k.to_string();\n assert_eq!(display, \"insert\");\n let k2 = KeyInput::parse(&display).unwrap();\n assert_eq!(k, k2);\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/keybindings/keymap.rs", "content": "use std::collections::HashMap;\n\nuse super::actions::Action;\nuse super::conditions::{ConditionExpr, EvalContext};\nuse super::key::{KeyInput, SingleKey};\n\n/// A single rule within a keybinding: an optional condition and an action.\n/// If the condition is `None`, the rule always matches.\n#[derive(Debug, Clone)]\npub struct KeyRule {\n pub condition: Option,\n pub action: Action,\n}\n\n/// A keybinding is an ordered list of rules. The first rule whose condition\n/// matches (or has no condition) wins.\n#[derive(Debug, Clone)]\npub struct KeyBinding {\n pub rules: Vec,\n}\n\n/// A keymap is a collection of keybindings indexed by key input.\n#[derive(Debug, Clone)]\npub struct Keymap {\n pub bindings: HashMap,\n}\n\nimpl KeyRule {\n /// Create an unconditional rule.\n pub fn always(action: Action) -> Self {\n KeyRule {\n condition: None,\n action,\n }\n }\n\n /// Create a conditional rule. Accepts any type convertible to `ConditionExpr`,\n /// including bare `ConditionAtom` values.\n pub fn when(condition: impl Into, action: Action) -> Self {\n KeyRule {\n condition: Some(condition.into()),\n action,\n }\n }\n}\n\nimpl KeyBinding {\n /// Create a simple (unconditional) binding.\n pub fn simple(action: Action) -> Self {\n KeyBinding {\n rules: vec![KeyRule::always(action)],\n }\n }\n\n /// Create a conditional binding from a list of rules.\n pub fn conditional(rules: Vec) -> Self {\n KeyBinding { rules }\n }\n}\n\nimpl Keymap {\n /// Create an empty keymap.\n pub fn new() -> Self {\n Keymap {\n bindings: HashMap::new(),\n }\n }\n\n /// Bind a key input to a simple (unconditional) action.\n pub fn bind(&mut self, key: KeyInput, action: Action) {\n self.bindings.insert(key, KeyBinding::simple(action));\n }\n\n /// Bind a key input to a conditional set of rules.\n pub fn bind_conditional(&mut self, key: KeyInput, rules: Vec) {\n self.bindings.insert(key, KeyBinding::conditional(rules));\n }\n\n /// Resolve a key input to an action given the current evaluation context.\n /// Returns `None` if the key has no binding or no rule's condition matches.\n pub fn resolve(&self, key: &KeyInput, ctx: &EvalContext) -> Option {\n let binding = self.bindings.get(key)?;\n for rule in &binding.rules {\n match &rule.condition {\n None => return Some(rule.action.clone()),\n Some(cond) if cond.evaluate(ctx) => return Some(rule.action.clone()),\n Some(_) => {}\n }\n }\n None\n }\n\n /// Check if any binding starts with the given single key as the first key\n /// of a multi-key sequence. Used to detect pending multi-key sequences.\n pub fn has_sequence_starting_with(&self, prefix: &SingleKey) -> bool {\n self.bindings.keys().any(|ki| match ki {\n KeyInput::Sequence(keys) => keys.first() == Some(prefix),\n KeyInput::Single(_) => false,\n })\n }\n\n /// Merge another keymap into this one. Keys from `other` override keys in `self`.\n #[allow(dead_code)]\n pub fn merge(&mut self, other: &Keymap) {\n for (key, binding) in &other.bindings {\n self.bindings.insert(key.clone(), binding.clone());\n }\n }\n}\n\nimpl Default for Keymap {\n fn default() -> Self {\n Self::new()\n }\n}\n\n#[cfg(test)]\nmod tests {\n use super::super::conditions::ConditionAtom;\n use super::*;\n\n fn make_ctx(cursor: usize, width: usize, selected: usize, len: usize) -> EvalContext {\n EvalContext {\n cursor_position: cursor,\n input_width: width,\n input_byte_len: width,\n selected_index: selected,\n results_len: len,\n original_input_empty: false,\n has_context: false,\n }\n }\n\n #[test]\n fn simple_binding_resolves() {\n let mut keymap = Keymap::new();\n let key = KeyInput::parse(\"ctrl-c\").unwrap();\n keymap.bind(key.clone(), Action::ReturnOriginal);\n\n let ctx = make_ctx(0, 0, 0, 10);\n assert_eq!(keymap.resolve(&key, &ctx), Some(Action::ReturnOriginal));\n }\n\n #[test]\n fn conditional_first_match_wins() {\n let mut keymap = Keymap::new();\n let key = KeyInput::parse(\"left\").unwrap();\n keymap.bind_conditional(\n key.clone(),\n vec![\n KeyRule::when(ConditionAtom::CursorAtStart, Action::Exit),\n KeyRule::always(Action::CursorLeft),\n ],\n );\n\n // Cursor at start → Exit\n let ctx = make_ctx(0, 5, 0, 10);\n assert_eq!(keymap.resolve(&key, &ctx), Some(Action::Exit));\n\n // Cursor not at start → CursorLeft\n let ctx = make_ctx(3, 5, 0, 10);\n assert_eq!(keymap.resolve(&key, &ctx), Some(Action::CursorLeft));\n }\n\n #[test]\n fn no_match_returns_none() {\n let keymap = Keymap::new();\n let key = KeyInput::parse(\"ctrl-c\").unwrap();\n let ctx = make_ctx(0, 0, 0, 0);\n assert_eq!(keymap.resolve(&key, &ctx), None);\n }\n\n #[test]\n fn conditional_no_condition_matches_returns_none() {\n let mut keymap = Keymap::new();\n let key = KeyInput::parse(\"left\").unwrap();\n // Only one rule with a condition that won't match\n keymap.bind_conditional(\n key.clone(),\n vec![KeyRule::when(ConditionAtom::CursorAtStart, Action::Exit)],\n );\n\n // Cursor not at start → no match\n let ctx = make_ctx(3, 5, 0, 10);\n assert_eq!(keymap.resolve(&key, &ctx), None);\n }\n\n #[test]\n fn has_sequence_starting_with() {\n let mut keymap = Keymap::new();\n let seq = KeyInput::parse(\"g g\").unwrap();\n keymap.bind(seq, Action::ScrollToTop);\n\n let g = SingleKey::parse(\"g\").unwrap();\n assert!(keymap.has_sequence_starting_with(&g));\n\n let h = SingleKey::parse(\"h\").unwrap();\n assert!(!keymap.has_sequence_starting_with(&h));\n }\n\n #[test]\n fn merge_overrides() {\n let mut base = Keymap::new();\n let key = KeyInput::parse(\"ctrl-c\").unwrap();\n base.bind(key.clone(), Action::ReturnOriginal);\n\n let mut overlay = Keymap::new();\n overlay.bind(key.clone(), Action::Exit);\n\n base.merge(&overlay);\n\n let ctx = make_ctx(0, 0, 0, 0);\n assert_eq!(base.resolve(&key, &ctx), Some(Action::Exit));\n }\n\n #[test]\n fn merge_preserves_unoverridden() {\n let mut base = Keymap::new();\n let key1 = KeyInput::parse(\"ctrl-c\").unwrap();\n let key2 = KeyInput::parse(\"ctrl-d\").unwrap();\n base.bind(key1.clone(), Action::ReturnOriginal);\n base.bind(key2.clone(), Action::DeleteCharAfter);\n\n let mut overlay = Keymap::new();\n overlay.bind(key1.clone(), Action::Exit);\n\n base.merge(&overlay);\n\n let ctx = make_ctx(0, 0, 0, 0);\n assert_eq!(base.resolve(&key1, &ctx), Some(Action::Exit));\n assert_eq!(base.resolve(&key2, &ctx), Some(Action::DeleteCharAfter));\n }\n}\n" }, { "path": "crates/atuin/src/command/client/search/keybindings/mod.rs", "content": "pub mod actions;\npub mod conditions;\npub mod defaults;\npub mod key;\npub mod keymap;\n\npub use actions::Action;\n#[allow(unused_imports)]\npub use conditions::{ConditionAtom, ConditionExpr, EvalContext};\npub use defaults::KeymapSet;\n#[allow(unused_imports)]\npub use key::{KeyCodeValue, KeyInput, SingleKey};\n#[allow(unused_imports)]\npub use keymap::{KeyBinding, KeyRule, Keymap};\n" }, { "path": "crates/atuin/src/command/client/search.rs", "content": "use std::fs::File;\nuse std::io::{IsTerminal as _, Write, stderr, stdout};\n\nuse atuin_common::utils::{self, Escapable as _};\nuse clap::Parser;\nuse eyre::Result;\n\nuse atuin_client::{\n database::Database,\n database::{OptFilters, current_context},\n encryption,\n history::{History, store::HistoryStore},\n record::sqlite_store::SqliteStore,\n settings::{FilterMode, KeymapMode, SearchMode, Settings, Timezone},\n theme::Theme,\n};\n\nuse super::history::ListMode;\n\nmod cursor;\nmod duration;\nmod engines;\nmod history_list;\nmod inspector;\nmod interactive;\npub mod keybindings;\n\npub use duration::format_duration_into;\n\n#[allow(clippy::struct_excessive_bools, clippy::struct_field_names)]\n#[derive(Parser, Debug)]\npub struct Cmd {\n /// Filter search result by directory\n #[arg(long, short)]\n cwd: Option,\n\n /// Exclude directory from results\n #[arg(long = \"exclude-cwd\")]\n exclude_cwd: Option,\n\n /// Filter search result by exit code\n #[arg(long, short)]\n exit: Option,\n\n /// Exclude results with this exit code\n #[arg(long = \"exclude-exit\")]\n exclude_exit: Option,\n\n /// Only include results added before this date\n #[arg(long, short)]\n before: Option,\n\n /// Only include results after this date\n #[arg(long)]\n after: Option,\n\n /// How many entries to return at most\n #[arg(long)]\n limit: Option,\n\n /// Offset from the start of the results\n #[arg(long)]\n offset: Option,\n\n /// Open interactive search UI\n #[arg(long, short)]\n interactive: bool,\n\n /// Allow overriding filter mode over config\n #[arg(long = \"filter-mode\")]\n filter_mode: Option,\n\n /// Allow overriding search mode over config\n #[arg(long = \"search-mode\")]\n search_mode: Option,\n\n /// Marker argument used to inform atuin that it was invoked from a shell up-key binding (hidden from help to avoid confusion)\n #[arg(long = \"shell-up-key-binding\", hide = true)]\n shell_up_key_binding: bool,\n\n /// Notify the keymap at the shell's side\n #[arg(long = \"keymap-mode\", default_value = \"auto\")]\n keymap_mode: KeymapMode,\n\n /// Use human-readable formatting for time\n #[arg(long)]\n human: bool,\n\n #[arg(allow_hyphen_values = true)]\n query: Option>,\n\n /// Show only the text of the command\n #[arg(long)]\n cmd_only: bool,\n\n /// Terminate the output with a null, for better multiline handling\n #[arg(long)]\n print0: bool,\n\n /// Delete anything matching this query. Will not print out the match\n #[arg(long)]\n delete: bool,\n\n /// Delete EVERYTHING!\n #[arg(long)]\n delete_it_all: bool,\n\n /// Reverse the order of results, oldest first\n #[arg(long, short)]\n reverse: bool,\n\n /// Display the command time in another timezone other than the configured default.\n ///\n /// This option takes one of the following kinds of values:\n /// - the special value \"local\" (or \"l\") which refers to the system time zone\n /// - an offset from UTC (e.g. \"+9\", \"-2:30\")\n #[arg(long, visible_alias = \"tz\")]\n #[arg(allow_hyphen_values = true)]\n // Clippy warns about `Option>`, but we suppress it because we need\n // this distinction for proper argument handling.\n #[allow(clippy::option_option)]\n timezone: Option>,\n\n /// Available variables: {command}, {directory}, {duration}, {user}, {host}, {time}, {exit} and\n /// {relativetime}.\n /// Example: --format \"{time} - [{duration}] - {directory}$\\t{command}\"\n #[arg(long, short)]\n format: Option,\n\n /// Set the maximum number of lines Atuin's interface should take up.\n #[arg(long = \"inline-height\")]\n inline_height: Option,\n\n /// Include duplicate commands in the output (non-interactive only)\n #[arg(long)]\n include_duplicates: bool,\n\n /// File name to write the result to (hidden from help as this is meant to be used from a script)\n #[arg(long = \"result-file\", hide = true)]\n result_file: Option,\n}\n\nimpl Cmd {\n /// Returns true if this search command will run in interactive (TUI) mode\n pub fn is_interactive(&self) -> bool {\n self.interactive\n }\n\n // clippy: please write this instead\n // clippy: now it has too many lines\n // me: I'll do it later OKAY\n #[allow(clippy::too_many_lines)]\n pub async fn run(\n self,\n db: impl Database,\n settings: &mut Settings,\n store: SqliteStore,\n theme: &Theme,\n ) -> Result<()> {\n let query = self.query.unwrap_or_else(|| {\n std::env::var(\"ATUIN_QUERY\").map_or_else(\n |_| vec![],\n |query| {\n query\n .split(' ')\n .map(std::string::ToString::to_string)\n .collect()\n },\n )\n });\n\n if (self.delete_it_all || self.delete) && self.limit.is_some() {\n // Because of how deletion is implemented, it will always delete all matches\n // and disregard the limit option. It is also not clear what deletion with a\n // limit would even mean. Deleting the LIMIT most recent entries that match\n // the search query would make sense, but that wouldn't match what's displayed\n // when running the equivalent search, but deleting those entries that are\n // displayed with the search would leave any duplicates of those lines which may\n // or may not have been intended to be deleted.\n eprintln!(\"\\\"--limit\\\" is not compatible with deletion.\");\n return Ok(());\n }\n\n if self.delete && query.is_empty() {\n eprintln!(\n \"Please specify a query to match the items you wish to delete. If you wish to delete all history, pass --delete-it-all\"\n );\n return Ok(());\n }\n\n if self.delete_it_all && !query.is_empty() {\n eprintln!(\n \"--delete-it-all will delete ALL of your history! It does not require a query.\"\n );\n return Ok(());\n }\n\n if let Some(search_mode) = self.search_mode {\n settings.search_mode = search_mode;\n }\n if let Some(filter_mode) = self.filter_mode {\n settings.filter_mode = Some(filter_mode);\n }\n if let Some(inline_height) = self.inline_height {\n settings.inline_height = inline_height;\n }\n\n settings.shell_up_key_binding = self.shell_up_key_binding;\n\n // `keymap_mode` specified in config.toml overrides the `--keymap-mode`\n // option specified in the keybindings.\n settings.keymap_mode = match settings.keymap_mode {\n KeymapMode::Auto => self.keymap_mode,\n value => value,\n };\n settings.keymap_mode_shell = self.keymap_mode;\n\n let encryption_key: [u8; 32] = encryption::load_key(settings)?.into();\n\n let host_id = Settings::host_id().await?;\n let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);\n\n if self.interactive {\n let item = interactive::history(&query, settings, db, &history_store, theme).await?;\n\n if let Some(result_file) = self.result_file {\n let mut file = File::create(result_file)?;\n write!(file, \"{item}\")?;\n } else if !stdout().is_terminal() {\n // stdout is not a terminal - likely command substitution like VAR=$(atuin search -i)\n // Write to stdout so it gets captured\n println!(\"{item}\");\n } else if stderr().is_terminal() {\n eprintln!(\"{}\", item.escape_control());\n } else {\n eprintln!(\"{item}\");\n }\n } else {\n let opt_filter = OptFilters {\n exit: self.exit,\n exclude_exit: self.exclude_exit,\n cwd: self.cwd,\n exclude_cwd: self.exclude_cwd,\n before: self.before,\n after: self.after,\n limit: self.limit,\n offset: self.offset,\n reverse: self.reverse,\n include_duplicates: self.include_duplicates,\n };\n\n let mut entries =\n run_non_interactive(settings, opt_filter.clone(), &query, &db).await?;\n\n if entries.is_empty() {\n std::process::exit(1)\n }\n\n // if we aren't deleting, print it all\n if self.delete || self.delete_it_all {\n // delete it\n // it only took me _years_ to add this\n // sorry\n while !entries.is_empty() {\n for entry in &entries {\n eprintln!(\"deleting {}\", entry.id);\n\n if settings.sync.records {\n let (id, _) = history_store.delete(entry.id.clone()).await?;\n history_store.incremental_build(&db, &[id]).await?;\n } else {\n db.delete(entry.clone()).await?;\n }\n }\n\n entries =\n run_non_interactive(settings, opt_filter.clone(), &query, &db).await?;\n }\n } else {\n let format = match self.format {\n None => Some(settings.history_format.as_str()),\n _ => self.format.as_deref(),\n };\n let tz = match self.timezone {\n Some(Some(tz)) => tz, // User provided a value\n Some(None) | None => settings.timezone, // No value was provided\n };\n\n super::history::print_list(\n &entries,\n ListMode::from_flags(self.human, self.cmd_only),\n format,\n self.print0,\n true,\n tz,\n );\n }\n }\n Ok(())\n }\n}\n\n// This is supposed to more-or-less mirror the command line version, so ofc\n// it is going to have a lot of args\n#[allow(clippy::too_many_arguments, clippy::cast_possible_truncation)]\nasync fn run_non_interactive(\n settings: &Settings,\n filter_options: OptFilters,\n query: &[String],\n db: &impl Database,\n) -> Result> {\n let dir = if filter_options.cwd.as_deref() == Some(\".\") {\n Some(utils::get_current_dir())\n } else {\n filter_options.cwd\n };\n\n let context = current_context().await?;\n\n let opt_filter = OptFilters {\n cwd: dir.clone(),\n ..filter_options\n };\n\n let filter_mode = settings.default_filter_mode(context.git_root.is_some());\n\n let results = db\n .search(\n settings.search_mode,\n filter_mode,\n &context,\n query.join(\" \").as_str(),\n opt_filter,\n )\n .await?;\n\n Ok(results)\n}\n\n#[cfg(test)]\nmod tests {\n use super::Cmd;\n use clap::Parser;\n\n #[test]\n fn search_for_triple_dash() {\n // Issue #3028: searching for `---` should not be treated as a CLI flag\n let cmd = Cmd::try_parse_from([\"search\", \"---\"]);\n assert!(cmd.is_ok(), \"Failed to parse '---' as a query: {cmd:?}\");\n let cmd = cmd.unwrap();\n assert_eq!(cmd.query, Some(vec![\"---\".to_string()]));\n }\n\n #[test]\n fn search_for_double_dash_value() {\n // Searching for strings starting with -- should also work\n let cmd = Cmd::try_parse_from([\"search\", \"--\", \"--foo\"]);\n assert!(cmd.is_ok());\n let cmd = cmd.unwrap();\n assert_eq!(cmd.query, Some(vec![\"--foo\".to_string()]));\n }\n}\n" }, { "path": "crates/atuin/src/command/client/setup.rs", "content": "use atuin_client::settings::Settings;\n\nuse colored::Colorize;\nuse eyre::Result;\nuse std::io::{self, Write};\nuse toml_edit::{DocumentMut, value};\n\npub async fn run(_settings: &Settings) -> Result<()> {\n let enable_ai = prompt(\n \"Atuin AI\",\n \"This will enable command generation and other AI features via the question mark key\",\n Some(\n \"By default, Atuin AI only has access to the name and version of your operating system and shell - your shell history is not sent to the AI.\",\n ),\n )?;\n\n let enable_daemon = prompt(\n \"Atuin Daemon\",\n \"This will enable improved search and history sync using a persistent background process\",\n None,\n )?;\n\n let config_file = Settings::get_config_path()?;\n let config_str = tokio::fs::read_to_string(&config_file).await?;\n let mut doc = config_str.parse::()?;\n\n let mut changed = false;\n if enable_ai {\n changed = true;\n if !doc.contains_key(\"ai\") {\n doc[\"ai\"] = toml_edit::table();\n }\n doc[\"ai\"][\"enabled\"] = value(true);\n }\n\n if enable_daemon {\n changed = true;\n if !doc.contains_key(\"daemon\") {\n doc[\"daemon\"] = toml_edit::table();\n }\n doc[\"daemon\"][\"enabled\"] = value(true);\n doc[\"daemon\"][\"autostart\"] = value(true);\n doc[\"search_mode\"] = value(\"daemon-fuzzy\");\n }\n\n if changed {\n tokio::fs::write(config_file, doc.to_string()).await?;\n\n println!(\n \"{check} Settings updated successfully\",\n check = \"✓\".bold().bright_green()\n );\n } else {\n println!(\n \"{check} No settings changed\",\n check = \"✓\".bold().bright_green()\n );\n }\n\n Ok(())\n}\n\npub fn prompt(feature: &str, description: &str, note: Option<&str>) -> Result {\n println!(\n \"> Enable {feature}?\",\n feature = feature.bold().bright_blue()\n );\n if let Some(note) = note {\n println!(\" {description}\");\n print!(\" {note} {q} \", q = \"[Y/n]\".bold());\n } else {\n print!(\" {description} {q} \", q = \"[Y/n]\".bold());\n }\n\n io::stdout().flush().ok();\n\n let mut input = String::new();\n io::stdin().read_line(&mut input)?;\n let answer = input.trim().to_lowercase();\n Ok(answer.is_empty() || answer == \"y\" || answer == \"yes\")\n}\n" }, { "path": "crates/atuin/src/command/client/stats.rs", "content": "use clap::Parser;\nuse eyre::Result;\nuse interim::parse_date_string;\nuse time::{Duration, OffsetDateTime, Time};\n\nuse atuin_client::{\n database::{Database, current_context},\n settings::Settings,\n theme::Theme,\n};\n\nuse atuin_history::stats::{compute, pretty_print};\n\nfn parse_ngram_size(s: &str) -> Result {\n let value = s\n .parse::()\n .map_err(|_| format!(\"'{s}' is not a valid window size\"))?;\n\n if value == 0 {\n return Err(\"ngram window size must be at least 1\".to_string());\n }\n\n Ok(value)\n}\n\n#[derive(Parser, Debug)]\n#[command(infer_subcommands = true)]\npub struct Cmd {\n /// Compute statistics for the specified period, leave blank for statistics since the beginning. See [this](https://docs.atuin.sh/reference/stats/) for more details.\n period: Vec,\n\n /// How many top commands to list\n #[arg(long, short, default_value = \"10\")]\n count: usize,\n\n /// The number of consecutive commands to consider\n #[arg(long, short, default_value = \"1\", value_parser = parse_ngram_size)]\n ngram_size: usize,\n}\n\nimpl Cmd {\n pub async fn run(&self, db: &impl Database, settings: &Settings, theme: &Theme) -> Result<()> {\n let context = current_context().await?;\n let words = if self.period.is_empty() {\n String::from(\"all\")\n } else {\n self.period.join(\" \")\n };\n\n let now = OffsetDateTime::now_utc().to_offset(settings.timezone.0);\n let last_night = now.replace_time(Time::MIDNIGHT);\n\n let history = if words.as_str() == \"all\" {\n db.list(&[], &context, None, false, false).await?\n } else if words.trim() == \"today\" {\n let start = last_night;\n let end = start + Duration::days(1);\n db.range(start, end).await?\n } else if words.trim() == \"month\" {\n let end = last_night;\n let start = end - Duration::days(31);\n db.range(start, end).await?\n } else if words.trim() == \"week\" {\n let end = last_night;\n let start = end - Duration::days(7);\n db.range(start, end).await?\n } else if words.trim() == \"year\" {\n let end = last_night;\n let start = end - Duration::days(365);\n db.range(start, end).await?\n } else {\n let start = parse_date_string(&words, now, settings.dialect.into())?;\n let end = start + Duration::days(1);\n db.range(start, end).await?\n };\n\n let stats = compute(settings, &history, self.count, self.ngram_size);\n\n if let Some(stats) = stats {\n pretty_print(stats, self.ngram_size, theme);\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store/pull.rs", "content": "use clap::Args;\nuse eyre::Result;\n\nuse atuin_client::{\n database::Database,\n record::store::Store,\n record::sync::Operation,\n record::{sqlite_store::SqliteStore, sync},\n settings::Settings,\n};\n\n#[derive(Args, Debug)]\npub struct Pull {\n /// The tag to push (eg, 'history'). Defaults to all tags\n #[arg(long, short)]\n pub tag: Option,\n\n /// Force push records\n /// This will first wipe the local store, and then download all records from the remote\n #[arg(long, default_value = \"false\")]\n pub force: bool,\n\n /// Page Size\n /// How many records to download at once. Defaults to 100\n #[arg(long, default_value = \"100\")]\n pub page: u64,\n}\n\nimpl Pull {\n pub async fn run(\n &self,\n settings: &Settings,\n store: SqliteStore,\n db: &dyn Database,\n ) -> Result<()> {\n if self.force {\n println!(\"Forcing local overwrite!\");\n println!(\"Clearing local store\");\n\n store.delete_all().await?;\n }\n\n // We can actually just use the existing diff/etc to push\n // 1. Diff\n // 2. Get operations\n // 3. Filter operations by\n // a) are they a download op?\n // b) are they for the host/tag we are pushing here?\n let (diff, _) = sync::diff(settings, &store).await?;\n let operations = sync::operations(diff, &store).await?;\n\n let operations = operations\n .into_iter()\n .filter(|op| match op {\n // No noops or downloads thx\n Operation::Noop { .. } | Operation::Upload { .. } => false,\n\n // pull, so yes plz to downloads!\n Operation::Download { tag, .. } => {\n if self.force {\n return true;\n }\n\n if let Some(t) = self.tag.clone()\n && t != *tag\n {\n return false;\n }\n\n true\n }\n })\n .collect();\n\n let (_, downloaded) = sync::sync_remote(operations, &store, settings, self.page).await?;\n\n println!(\"Downloaded {} records\", downloaded.len());\n\n crate::sync::build(settings, &store, db, Some(&downloaded)).await?;\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store/purge.rs", "content": "use clap::Args;\nuse eyre::Result;\n\nuse atuin_client::{\n encryption::load_key,\n record::{sqlite_store::SqliteStore, store::Store},\n settings::Settings,\n};\n\n#[derive(Args, Debug)]\npub struct Purge {}\n\nimpl Purge {\n pub async fn run(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n println!(\"Purging local records that cannot be decrypted\");\n\n let key = load_key(settings)?;\n\n match store.purge(&key.into()).await {\n Ok(()) => println!(\"Local store purge completed OK\"),\n Err(e) => println!(\"Failed to purge local store: {e:?}\"),\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store/push.rs", "content": "use atuin_common::record::HostId;\nuse clap::Args;\nuse eyre::Result;\nuse uuid::Uuid;\n\nuse atuin_client::{\n api_client::Client,\n record::sync::Operation,\n record::{sqlite_store::SqliteStore, sync},\n settings::Settings,\n};\n\n#[derive(Args, Debug)]\npub struct Push {\n /// The tag to push (eg, 'history'). Defaults to all tags\n #[arg(long, short)]\n pub tag: Option,\n\n /// The host to push, in the form of a UUID host ID. Defaults to the current host.\n #[arg(long)]\n pub host: Option,\n\n /// Force push records\n /// This will override both host and tag, to be all hosts and all tags. First clear the remote store, then upload all of the\n /// local store\n #[arg(long, default_value = \"false\")]\n pub force: bool,\n\n /// Page Size\n /// How many records to upload at once. Defaults to 100\n #[arg(long, default_value = \"100\")]\n pub page: u64,\n}\n\nimpl Push {\n pub async fn run(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n let host_id = Settings::host_id().await?;\n\n if self.force {\n println!(\"Forcing remote store overwrite!\");\n println!(\"Clearing remote store\");\n\n let client = Client::new(\n &settings.sync_address,\n settings.sync_auth_token().await?,\n settings.network_connect_timeout,\n settings.network_timeout * 10, // we may be deleting a lot of data... so up the\n // timeout\n )\n .expect(\"failed to create client\");\n\n client.delete_store().await?;\n }\n\n // We can actually just use the existing diff/etc to push\n // 1. Diff\n // 2. Get operations\n // 3. Filter operations by\n // a) are they an upload op?\n // b) are they for the host/tag we are pushing here?\n let (diff, _) = sync::diff(settings, &store).await?;\n let operations = sync::operations(diff, &store).await?;\n\n let operations = operations\n .into_iter()\n .filter(|op| match op {\n // No noops or downloads thx\n Operation::Noop { .. } | Operation::Download { .. } => false,\n\n // push, so yes plz to uploads!\n Operation::Upload { host, tag, .. } => {\n if self.force {\n return true;\n }\n\n if let Some(h) = self.host {\n if HostId(h) != *host {\n return false;\n }\n } else if *host != host_id {\n return false;\n }\n\n if let Some(t) = self.tag.clone()\n && t != *tag\n {\n return false;\n }\n\n true\n }\n })\n .collect();\n\n let (uploaded, _) = sync::sync_remote(operations, &store, settings, self.page).await?;\n\n println!(\"Uploaded {uploaded} records\");\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store/rebuild.rs", "content": "use atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse atuin_scripts::store::ScriptStore;\nuse clap::Args;\nuse eyre::{Result, bail};\n\n#[cfg(feature = \"daemon\")]\nuse atuin_daemon::emit_event;\n\nuse atuin_client::{\n database::Database, encryption, history::store::HistoryStore,\n record::sqlite_store::SqliteStore, settings::Settings,\n};\n\n#[derive(Args, Debug)]\npub struct Rebuild {\n pub tag: String,\n}\n\nimpl Rebuild {\n pub async fn run(\n &self,\n settings: &Settings,\n store: SqliteStore,\n database: &dyn Database,\n ) -> Result<()> {\n // keep it as a string and not an enum atm\n // would be super cool to build this dynamically in the future\n // eg register handles for rebuilding various tags without having to make this part of the\n // binary big\n match self.tag.as_str() {\n \"history\" => {\n self.rebuild_history(settings, store.clone(), database)\n .await?;\n }\n\n \"dotfiles\" => {\n self.rebuild_dotfiles(settings, store.clone()).await?;\n }\n\n \"scripts\" => {\n self.rebuild_scripts(settings, store.clone()).await?;\n }\n\n tag => bail!(\"unknown tag: {tag}\"),\n }\n\n Ok(())\n }\n\n async fn rebuild_history(\n &self,\n settings: &Settings,\n store: SqliteStore,\n database: &dyn Database,\n ) -> Result<()> {\n let encryption_key: [u8; 32] = encryption::load_key(settings)?.into();\n\n let host_id = Settings::host_id().await?;\n let history_store = HistoryStore::new(store, host_id, encryption_key);\n\n history_store.build(database).await?;\n\n #[cfg(feature = \"daemon\")]\n let _ = emit_event(atuin_daemon::DaemonEvent::HistoryRebuilt).await;\n\n Ok(())\n }\n\n async fn rebuild_dotfiles(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n let encryption_key: [u8; 32] = encryption::load_key(settings)?.into();\n\n let host_id = Settings::host_id().await?;\n\n let alias_store = AliasStore::new(store.clone(), host_id, encryption_key);\n let var_store = VarStore::new(store.clone(), host_id, encryption_key);\n\n alias_store.build().await?;\n var_store.build().await?;\n\n Ok(())\n }\n\n async fn rebuild_scripts(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n let encryption_key: [u8; 32] = encryption::load_key(settings)?.into();\n let host_id = Settings::host_id().await?;\n let script_store = ScriptStore::new(store, host_id, encryption_key);\n let database =\n atuin_scripts::database::Database::new(settings.scripts.db_path.clone(), 1.0).await?;\n\n script_store.build(database).await?;\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store/rekey.rs", "content": "use clap::Args;\nuse eyre::{Result, bail};\nuse tokio::{fs::File, io::AsyncWriteExt};\n\nuse atuin_client::{\n encryption::{Key, decode_key, encode_key, generate_encoded_key, load_key},\n record::sqlite_store::SqliteStore,\n record::store::Store,\n settings::Settings,\n};\n\n#[derive(Args, Debug)]\npub struct Rekey {\n /// The new key to use for encryption. Omit for a randomly-generated key\n key: Option,\n}\n\nimpl Rekey {\n pub async fn run(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n let key = if let Some(key) = self.key.clone() {\n println!(\"Re-encrypting store with specified key\");\n\n match bip39::Mnemonic::from_phrase(&key, bip39::Language::English) {\n Ok(mnemonic) => encode_key(Key::from_slice(mnemonic.entropy()))?,\n Err(err) => {\n match err {\n // assume they copied in the base64 key\n bip39::ErrorKind::InvalidWord(_) => key,\n bip39::ErrorKind::InvalidChecksum => {\n bail!(\"key mnemonic was not valid\")\n }\n bip39::ErrorKind::InvalidKeysize(_)\n | bip39::ErrorKind::InvalidWordLength(_)\n | bip39::ErrorKind::InvalidEntropyLength(_, _) => {\n bail!(\"key was not the correct length\")\n }\n }\n }\n }\n } else {\n println!(\"Re-encrypting store with freshly-generated key\");\n let (_, encoded) = generate_encoded_key()?;\n encoded\n };\n\n let current_key: [u8; 32] = load_key(settings)?.into();\n let new_key: [u8; 32] = decode_key(key.clone())?.into();\n\n store.re_encrypt(¤t_key, &new_key).await?;\n\n println!(\"Store rewritten. Saving new key\");\n let mut file = File::create(settings.key_path.clone()).await?;\n file.write_all(key.as_bytes()).await?;\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store/verify.rs", "content": "use clap::Args;\nuse eyre::Result;\n\nuse atuin_client::{\n encryption::load_key,\n record::{sqlite_store::SqliteStore, store::Store},\n settings::Settings,\n};\n\n#[derive(Args, Debug)]\npub struct Verify {}\n\nimpl Verify {\n pub async fn run(&self, settings: &Settings, store: SqliteStore) -> Result<()> {\n println!(\"Verifying local store can be decrypted with the current key\");\n\n let key = load_key(settings)?;\n\n match store.verify(&key.into()).await {\n Ok(()) => println!(\"Local store encryption verified OK\"),\n Err(e) => println!(\"Failed to verify local store encryption: {e:?}\"),\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/store.rs", "content": "use clap::Subcommand;\nuse eyre::Result;\n\nuse atuin_client::{\n database::Database,\n record::{sqlite_store::SqliteStore, store::Store},\n settings::Settings,\n};\nuse itertools::Itertools;\nuse time::{OffsetDateTime, UtcOffset};\n\n#[cfg(feature = \"sync\")]\nmod push;\n\n#[cfg(feature = \"sync\")]\nmod pull;\n\nmod purge;\nmod rebuild;\nmod rekey;\nmod verify;\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Print the current status of the record store\n Status,\n\n /// Rebuild a store (eg atuin store rebuild history)\n Rebuild(rebuild::Rebuild),\n\n /// Re-encrypt the store with a new key (potential for data loss!)\n Rekey(rekey::Rekey),\n\n /// Delete all records in the store that cannot be decrypted with the current key\n Purge(purge::Purge),\n\n /// Verify that all records in the store can be decrypted with the current key\n Verify(verify::Verify),\n\n /// Push all records to the remote sync server (one way sync)\n #[cfg(feature = \"sync\")]\n Push(push::Push),\n\n /// Pull records from the remote sync server (one way sync)\n #[cfg(feature = \"sync\")]\n Pull(pull::Pull),\n}\n\nimpl Cmd {\n pub async fn run(\n &self,\n settings: &Settings,\n database: &dyn Database,\n store: SqliteStore,\n ) -> Result<()> {\n match self {\n Self::Status => self.status(store).await,\n Self::Rebuild(rebuild) => rebuild.run(settings, store, database).await,\n Self::Rekey(rekey) => rekey.run(settings, store).await,\n Self::Verify(verify) => verify.run(settings, store).await,\n Self::Purge(purge) => purge.run(settings, store).await,\n\n #[cfg(feature = \"sync\")]\n Self::Push(push) => push.run(settings, store).await,\n\n #[cfg(feature = \"sync\")]\n Self::Pull(pull) => pull.run(settings, store, database).await,\n }\n }\n\n pub async fn status(&self, store: SqliteStore) -> Result<()> {\n let host_id = Settings::host_id().await?;\n let offset = UtcOffset::current_local_offset().unwrap_or(UtcOffset::UTC);\n\n let status = store.status().await?;\n\n // TODO: should probs build some data structure and then pretty-print it or smth\n for (host, st) in status.hosts.iter().sorted_by_key(|(h, _)| *h) {\n let host_string = if host == &host_id {\n format!(\"host: {} <- CURRENT HOST\", host.0.as_hyphenated())\n } else {\n format!(\"host: {}\", host.0.as_hyphenated())\n };\n\n println!(\"{host_string}\");\n\n for (tag, idx) in st.iter().sorted_by_key(|(tag, _)| *tag) {\n println!(\"\\tstore: {tag}\");\n\n let first = store.first(*host, tag).await?;\n let last = store.last(*host, tag).await?;\n\n println!(\"\\t\\tidx: {idx}\");\n\n if let Some(first) = first {\n println!(\"\\t\\tfirst: {}\", first.id.0.as_hyphenated());\n\n let time =\n OffsetDateTime::from_unix_timestamp_nanos(i128::from(first.timestamp))?\n .to_offset(offset);\n println!(\"\\t\\t\\tcreated: {time}\");\n }\n\n if let Some(last) = last {\n println!(\"\\t\\tlast: {}\", last.id.0.as_hyphenated());\n\n let time =\n OffsetDateTime::from_unix_timestamp_nanos(i128::from(last.timestamp))?\n .to_offset(offset);\n println!(\"\\t\\t\\tcreated: {time}\");\n }\n }\n\n println!();\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/client/sync/status.rs", "content": "use crate::{SHA, VERSION};\nuse atuin_client::{api_client, database::Database, settings::Settings};\nuse colored::Colorize;\nuse eyre::{Result, bail};\n\npub async fn run(settings: &Settings, db: &impl Database) -> Result<()> {\n if !settings.logged_in().await? {\n bail!(\"You are not logged in to a sync server - cannot show sync status\");\n }\n\n let client = api_client::Client::new(\n &settings.sync_address,\n settings.sync_auth_token().await?,\n settings.network_connect_timeout,\n settings.network_timeout,\n )?;\n\n let me = client.me().await?;\n let last_sync = Settings::last_sync().await?;\n\n println!(\"Atuin v{VERSION} - Build rev {SHA}\\n\");\n\n println!(\"{}\", \"[Local]\".green());\n\n if settings.auto_sync {\n println!(\"Sync frequency: {}\", settings.sync_frequency);\n println!(\"Last sync: {}\", last_sync.to_offset(settings.timezone.0));\n }\n\n if !settings.sync.records {\n let local_count = db.history_count(false).await?;\n let deleted_count = db.history_count(true).await? - local_count;\n\n println!(\"History count: {local_count}\");\n println!(\"Deleted history count: {deleted_count}\\n\");\n }\n\n if settings.auto_sync {\n println!(\"{}\", \"[Remote]\".green());\n println!(\"Address: {}\", settings.sync_address);\n println!(\"Username: {}\", me.username);\n }\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/sync.rs", "content": "use clap::Subcommand;\nuse eyre::{Result, WrapErr};\n\nuse atuin_client::{\n database::Database,\n encryption,\n history::store::HistoryStore,\n record::{sqlite_store::SqliteStore, store::Store, sync},\n settings::Settings,\n};\n\nmod status;\n\nuse crate::command::client::account;\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Sync with the configured server\n Sync {\n /// Force re-download everything\n #[arg(long, short)]\n force: bool,\n },\n\n /// Login to the configured server\n Login(account::login::Cmd),\n\n /// Log out\n Logout,\n\n /// Register with the configured server\n Register(account::register::Cmd),\n\n /// Print the encryption key for transfer to another machine\n Key {\n /// Switch to base64 output of the key\n #[arg(long)]\n base64: bool,\n },\n\n /// Display the sync status\n Status,\n}\n\nimpl Cmd {\n pub async fn run(\n self,\n settings: Settings,\n db: &impl Database,\n store: SqliteStore,\n ) -> Result<()> {\n match self {\n Self::Sync { force } => run(&settings, force, db, store).await,\n Self::Login(l) => l.run(&settings, &store).await,\n Self::Logout => account::logout::run().await,\n Self::Register(r) => r.run(&settings, &store).await,\n Self::Status => status::run(&settings, db).await,\n Self::Key { base64 } => {\n use atuin_client::encryption::{encode_key, load_key};\n let key = load_key(&settings).wrap_err(\"could not load encryption key\")?;\n\n if base64 {\n let encode = encode_key(&key).wrap_err(\"could not encode encryption key\")?;\n println!(\"{encode}\");\n } else {\n let mnemonic = bip39::Mnemonic::from_entropy(&key, bip39::Language::English)\n .map_err(|_| eyre::eyre!(\"invalid key\"))?;\n println!(\"{mnemonic}\");\n }\n Ok(())\n }\n }\n }\n}\n\nasync fn run(\n settings: &Settings,\n force: bool,\n db: &impl Database,\n store: SqliteStore,\n) -> Result<()> {\n if settings.sync.records {\n let encryption_key: [u8; 32] = encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n\n let host_id = Settings::host_id().await?;\n let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);\n\n let (uploaded, downloaded) = sync::sync(settings, &store).await?;\n\n crate::sync::build(settings, &store, db, Some(&downloaded)).await?;\n\n println!(\"{uploaded}/{} up/down to record store\", downloaded.len());\n\n let history_length = db.history_count(true).await?;\n let store_history_length = store.len_tag(\"history\").await?;\n\n #[allow(clippy::cast_sign_loss)]\n if history_length as u64 > store_history_length {\n println!(\n \"{history_length} in history index, but {store_history_length} in history store\"\n );\n println!(\"Running automatic history store init...\");\n\n // Internally we use the global filter mode, so this context is ignored.\n // don't recurse or loop here.\n history_store.init_store(db).await?;\n\n println!(\"Re-running sync due to new records locally\");\n\n // we'll want to run sync once more, as there will now be stuff to upload\n let (uploaded, downloaded) = sync::sync(settings, &store).await?;\n\n crate::sync::build(settings, &store, db, Some(&downloaded)).await?;\n\n println!(\"{uploaded}/{} up/down to record store\", downloaded.len());\n }\n } else {\n atuin_client::sync::sync(settings, force, db).await?;\n }\n\n println!(\n \"Sync complete! {} items in history database, force: {}\",\n db.history_count(true).await?,\n force\n );\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client/wrapped.rs", "content": "use crossterm::style::{ResetColor, SetAttribute};\nuse eyre::Result;\nuse std::collections::{HashMap, HashSet};\nuse time::{Date, Duration, Month, OffsetDateTime, Time};\n\nuse atuin_client::{\n database::Database, encryption, record::sqlite_store::SqliteStore, settings::Settings,\n theme::Theme,\n};\nuse atuin_dotfiles::store::AliasStore;\n\nuse atuin_history::stats::{Stats, compute};\n\n#[derive(Debug)]\nstruct WrappedStats {\n nav_commands: usize,\n pkg_commands: usize,\n error_rate: f64,\n first_half_commands: Vec<(String, usize)>,\n second_half_commands: Vec<(String, usize)>,\n git_percentage: f64,\n busiest_hour: Option<(String, usize)>,\n}\n\nimpl WrappedStats {\n #[allow(clippy::too_many_lines, clippy::cast_precision_loss)]\n fn new(\n settings: &Settings,\n stats: &Stats,\n history: &[atuin_client::history::History],\n alias_map: &HashMap,\n ) -> Self {\n // Helper to expand alias to its first command word\n let expand_alias = |cmd: &str| -> String {\n alias_map.get(cmd).map_or_else(\n || cmd.to_string(),\n |expanded| {\n expanded\n .split_whitespace()\n .next()\n .unwrap_or(cmd)\n .to_string()\n },\n )\n };\n\n let nav_commands = stats\n .top\n .iter()\n .filter(|(cmd, _)| {\n let cmd = &cmd[0];\n cmd == \"cd\" || cmd == \"ls\" || cmd == \"pwd\" || cmd == \"pushd\" || cmd == \"popd\"\n })\n .map(|(_, count)| count)\n .sum();\n\n let pkg_managers = [\n \"cargo\",\n \"npm\",\n \"pnpm\",\n \"yarn\",\n \"pip\",\n \"pip3\",\n \"pipenv\",\n \"poetry\",\n \"pipx\",\n \"uv\",\n \"brew\",\n \"apt\",\n \"apt-get\",\n \"apk\",\n \"pacman\",\n \"yay\",\n \"paru\",\n \"yum\",\n \"dnf\",\n \"dnf5\",\n \"rpm\",\n \"rpm-ostree\",\n \"zypper\",\n \"pkg\",\n \"chocolatey\",\n \"choco\",\n \"scoop\",\n \"winget\",\n \"gem\",\n \"bundle\",\n \"shards\",\n \"composer\",\n \"gradle\",\n \"maven\",\n \"mvn\",\n \"go get\",\n \"nuget\",\n \"dotnet\",\n \"mix\",\n \"hex\",\n \"rebar3\",\n \"nix\",\n \"nix-env\",\n \"cabal\",\n \"opam\",\n ];\n\n let pkg_commands = history\n .iter()\n .filter(|h| {\n let cmd = h.command.clone();\n pkg_managers.iter().any(|pm| cmd.starts_with(pm))\n })\n .count();\n\n // Error analysis\n let mut command_errors: HashMap = HashMap::new(); // (total_uses, errors)\n let midyear = history[0].timestamp + Duration::days(182); // Split year in half\n\n let mut first_half_commands: HashMap = HashMap::new();\n let mut second_half_commands: HashMap = HashMap::new();\n let mut hours: HashMap = HashMap::new();\n\n for entry in history {\n let raw_cmd = entry\n .command\n .split_whitespace()\n .next()\n .unwrap_or(\"\")\n .to_string();\n let cmd = expand_alias(&raw_cmd);\n let (total, errors) = command_errors.entry(cmd.clone()).or_insert((0, 0));\n *total += 1;\n if entry.exit != 0 {\n *errors += 1;\n }\n\n // Track command evolution\n if entry.timestamp < midyear {\n *first_half_commands.entry(cmd.clone()).or_default() += 1;\n } else {\n *second_half_commands.entry(cmd).or_default() += 1;\n }\n\n // Track hourly distribution\n let local_time = entry\n .timestamp\n .to_offset(time::UtcOffset::current_local_offset().unwrap_or(settings.timezone.0));\n let hour = format!(\"{:02}:00\", local_time.time().hour());\n *hours.entry(hour).or_default() += 1;\n }\n\n let total_errors: usize = command_errors.values().map(|(_, errors)| errors).sum();\n let total_commands: usize = command_errors.values().map(|(total, _)| total).sum();\n let error_rate = total_errors as f64 / total_commands as f64;\n\n // Process command evolution data\n let mut first_half: Vec<_> = first_half_commands.into_iter().collect();\n let mut second_half: Vec<_> = second_half_commands.into_iter().collect();\n first_half.sort_by_key(|(_, count)| std::cmp::Reverse(*count));\n second_half.sort_by_key(|(_, count)| std::cmp::Reverse(*count));\n first_half.truncate(5);\n second_half.truncate(5);\n\n // Calculate git percentage\n let git_commands: usize = stats\n .top\n .iter()\n .filter(|(cmd, _)| cmd[0].starts_with(\"git\"))\n .map(|(_, count)| count)\n .sum();\n let git_percentage = git_commands as f64 / stats.total_commands as f64;\n\n // Find busiest hour\n let busiest_hour = hours.into_iter().max_by_key(|(_, count)| *count);\n\n Self {\n nav_commands,\n pkg_commands,\n error_rate,\n first_half_commands: first_half,\n second_half_commands: second_half,\n git_percentage,\n busiest_hour,\n }\n }\n}\n\npub fn print_wrapped_header(year: i32) {\n let reset = ResetColor;\n let bold = SetAttribute(crossterm::style::Attribute::Bold);\n\n println!(\"{bold}╭────────────────────────────────────╮{reset}\");\n println!(\"{bold}│ ATUIN WRAPPED {year} │{reset}\");\n println!(\"{bold}│ Your Year in Shell History │{reset}\");\n println!(\"{bold}╰────────────────────────────────────╯{reset}\");\n println!();\n}\n\n#[allow(clippy::cast_precision_loss)]\nfn print_fun_facts(wrapped_stats: &WrappedStats, stats: &Stats, year: i32) {\n let reset = ResetColor;\n let bold = SetAttribute(crossterm::style::Attribute::Bold);\n\n if wrapped_stats.git_percentage > 0.05 {\n println!(\n \"{bold}🌟 You're a Git Power User!{reset} {bold}{:.1}%{reset} of your commands were Git operations\\n\",\n wrapped_stats.git_percentage * 100.0\n );\n }\n // Navigation patterns\n let nav_percentage = wrapped_stats.nav_commands as f64 / stats.total_commands as f64 * 100.0;\n if nav_percentage > 0.05 {\n println!(\n \"{bold}🚀 You're a Navigator!{reset} {bold}{nav_percentage:.1}%{reset} of your time was spent navigating directories\\n\",\n );\n }\n\n // Command vocabulary\n println!(\n \"{bold}📚 Command Vocabulary{reset}: You know {bold}{}{reset} unique commands\\n\",\n stats.unique_commands\n );\n\n // Package management\n println!(\n \"{bold}📦 Package Management{reset}: You ran {bold}{}{reset} package-related commands\\n\",\n wrapped_stats.pkg_commands\n );\n\n // Error patterns\n let error_percentage = wrapped_stats.error_rate * 100.0;\n println!(\n \"{bold}🚨 Error Analysis{reset}: Your commands failed {bold}{error_percentage:.1}%{reset} of the time\\n\",\n );\n\n // Command evolution\n println!(\"🔍 Command Evolution:\");\n\n // print stats for each half and compare\n println!(\" {bold}Top Commands{reset} in the first half of {year}:\");\n for (cmd, count) in wrapped_stats.first_half_commands.iter().take(3) {\n println!(\" {bold}{cmd}{reset} ({count} times)\");\n }\n\n println!(\" {bold}Top Commands{reset} in the second half of {year}:\");\n for (cmd, count) in wrapped_stats.second_half_commands.iter().take(3) {\n println!(\" {bold}{cmd}{reset} ({count} times)\");\n }\n\n // Find new favorite commands (in top 5 of second half but not in first half)\n let first_half_set: HashSet<_> = wrapped_stats\n .first_half_commands\n .iter()\n .map(|(cmd, _)| cmd)\n .collect();\n let new_favorites: Vec<_> = wrapped_stats\n .second_half_commands\n .iter()\n .filter(|(cmd, _)| !first_half_set.contains(cmd))\n .take(2)\n .collect();\n\n if !new_favorites.is_empty() {\n println!(\" {bold}New favorites{reset} in the second half:\");\n for (cmd, count) in new_favorites {\n println!(\" {bold}{cmd}{reset} ({count} times)\");\n }\n }\n\n // Time patterns\n if let Some((hour, count)) = &wrapped_stats.busiest_hour {\n println!(\"\\n🕘 Most Productive Hour: {bold}{hour}{reset} ({count} commands)\",);\n\n // Night owl or early bird\n let hour_num = hour\n .split(':')\n .next()\n .unwrap_or(\"0\")\n .parse::()\n .unwrap_or(0);\n if hour_num >= 22 || hour_num <= 4 {\n println!(\" You're quite the night owl! 🦉\");\n } else if (5..=7).contains(&hour_num) {\n println!(\" Early bird gets the worm! 🐦\");\n }\n }\n\n println!();\n}\n\npub async fn run(\n year: Option,\n db: &impl Database,\n settings: &Settings,\n store: SqliteStore,\n theme: &Theme,\n) -> Result<()> {\n let now = OffsetDateTime::now_utc().to_offset(settings.timezone.0);\n let month = now.month();\n\n // If we're in December, then wrapped is for the current year. If not, it's for the previous year\n let year = year.unwrap_or_else(|| {\n if month == Month::December {\n now.year()\n } else {\n now.year() - 1\n }\n });\n\n let start = OffsetDateTime::new_in_offset(\n Date::from_calendar_date(year, Month::January, 1).unwrap(),\n Time::MIDNIGHT,\n now.offset(),\n );\n let end = OffsetDateTime::new_in_offset(\n Date::from_calendar_date(year, Month::December, 31).unwrap(),\n Time::MIDNIGHT + Duration::days(1) - Duration::nanoseconds(1),\n now.offset(),\n );\n\n let history = db.range(start, end).await?;\n if history.is_empty() {\n println!(\n \"Your history for {year} is empty!\\nMaybe 'atuin import' could help you import your previous history 🪄\"\n );\n return Ok(());\n }\n\n // Load aliases for expansion\n let alias_map: HashMap = if settings.dotfiles.enabled {\n if let Ok(encryption_key) = encryption::load_key(settings) {\n let encryption_key: [u8; 32] = encryption_key.into();\n let host_id = Settings::host_id().await?;\n let alias_store = AliasStore::new(store, host_id, encryption_key);\n\n alias_store\n .aliases()\n .await\n .unwrap_or_default()\n .into_iter()\n .map(|a| (a.name, a.value))\n .collect()\n } else {\n HashMap::new()\n }\n } else {\n HashMap::new()\n };\n\n // Compute overall stats using existing functionality\n let stats = compute(settings, &history, 10, 1).expect(\"Failed to compute stats\");\n let wrapped_stats = WrappedStats::new(settings, &stats, &history, &alias_map);\n\n // Print wrapped format\n print_wrapped_header(year);\n\n println!(\"🎉 In {year}, you typed {} commands!\", stats.total_commands);\n println!(\n \" That's ~{} commands every day\\n\",\n stats.total_commands / 365\n );\n\n println!(\"Your Top Commands:\");\n atuin_history::stats::pretty_print(stats.clone(), 1, theme);\n println!();\n\n print_fun_facts(&wrapped_stats, &stats, year);\n\n Ok(())\n}\n" }, { "path": "crates/atuin/src/command/client.rs", "content": "use std::fs::{self, OpenOptions};\nuse std::path::{Path, PathBuf};\n\nuse clap::Subcommand;\nuse eyre::{Result, WrapErr};\n\nuse atuin_client::{\n database::Sqlite, record::sqlite_store::SqliteStore, settings::Settings, theme,\n};\nuse tracing_appender::rolling::{RollingFileAppender, Rotation};\nuse tracing_subscriber::{\n Layer, filter::EnvFilter, filter::LevelFilter, fmt, fmt::format::FmtSpan, prelude::*,\n};\n\nfn cleanup_old_logs(log_dir: &Path, prefix: &str, retention_days: u64) {\n let cutoff = std::time::SystemTime::now()\n - std::time::Duration::from_secs(retention_days * 24 * 60 * 60);\n\n let Ok(entries) = fs::read_dir(log_dir) else {\n return;\n };\n\n for entry in entries.flatten() {\n let path = entry.path();\n let Some(name) = path.file_name().and_then(|n| n.to_str()) else {\n continue;\n };\n\n // Match files like \"search.log.2024-02-23\" or \"daemon.log.2024-02-23\"\n if !name.starts_with(prefix) || name == prefix {\n continue;\n }\n\n if let Ok(metadata) = entry.metadata()\n && let Ok(modified) = metadata.modified()\n && modified < cutoff\n {\n let _ = fs::remove_file(&path);\n }\n }\n}\n\n#[cfg(feature = \"sync\")]\nmod sync;\n\n#[cfg(feature = \"sync\")]\nmod account;\n\n#[cfg(feature = \"daemon\")]\nmod daemon;\n\nmod default_config;\nmod doctor;\nmod dotfiles;\nmod history;\nmod import;\nmod info;\nmod init;\nmod kv;\nmod scripts;\nmod search;\nmod setup;\nmod stats;\nmod store;\nmod wrapped;\n\n#[derive(Subcommand, Debug)]\n#[command(infer_subcommands = true)]\npub enum Cmd {\n /// Setup Atuin features\n #[command()]\n Setup,\n\n /// Manipulate shell history\n #[command(subcommand)]\n History(history::Cmd),\n\n /// Import shell history from file\n #[command(subcommand)]\n Import(import::Cmd),\n\n /// Calculate statistics for your history\n Stats(stats::Cmd),\n\n /// Interactive history search\n Search(search::Cmd),\n\n #[cfg(feature = \"sync\")]\n #[command(flatten)]\n Sync(sync::Cmd),\n\n /// Manage your sync account\n #[cfg(feature = \"sync\")]\n Account(account::Cmd),\n\n /// Get or set small key-value pairs\n #[command(subcommand)]\n Kv(kv::Cmd),\n\n /// Manage the atuin data store\n #[command(subcommand)]\n Store(store::Cmd),\n\n /// Manage your dotfiles with Atuin\n #[command(subcommand)]\n Dotfiles(dotfiles::Cmd),\n\n /// Manage your scripts with Atuin\n #[command(subcommand)]\n Scripts(scripts::Cmd),\n\n /// Print Atuin's shell init script\n #[command()]\n Init(init::Cmd),\n\n /// Information about dotfiles locations and ENV vars\n #[command()]\n Info,\n\n /// Run the doctor to check for common issues\n #[command()]\n Doctor,\n\n #[command()]\n Wrapped { year: Option },\n\n /// *Experimental* Manage the background daemon\n #[cfg(feature = \"daemon\")]\n #[command()]\n Daemon(daemon::Cmd),\n\n /// Print the default atuin configuration (config.toml)\n #[command()]\n DefaultConfig,\n\n /// Run the AI assistant\n #[cfg(feature = \"ai\")]\n #[command(subcommand)]\n Ai(atuin_ai::commands::Commands),\n}\n\nimpl Cmd {\n pub fn run(self) -> Result<()> {\n // Daemonize before creating the async runtime – fork() inside a live\n // tokio runtime corrupts its internal state.\n #[cfg(all(unix, feature = \"daemon\"))]\n if let Self::Daemon(ref cmd) = self\n && cmd.should_daemonize()\n {\n daemon::daemonize_current_process()?;\n }\n\n let runtime = tokio::runtime::Builder::new_current_thread()\n .enable_all()\n .build()\n .unwrap();\n\n let settings = Settings::new().wrap_err(\"could not load client settings\")?;\n let theme_manager = theme::ThemeManager::new(settings.theme.debug, None);\n let res = runtime.block_on(self.run_inner(settings, theme_manager));\n\n runtime.shutdown_timeout(std::time::Duration::from_millis(50));\n\n res\n }\n\n #[allow(clippy::too_many_lines)]\n async fn run_inner(\n self,\n mut settings: Settings,\n mut theme_manager: theme::ThemeManager,\n ) -> Result<()> {\n // ATUIN_LOG env var overrides config file level settings\n let env_log_set = std::env::var(\"ATUIN_LOG\").is_ok();\n\n // Base filter from env var (or empty if not set)\n let base_filter =\n EnvFilter::from_env(\"ATUIN_LOG\").add_directive(\"sqlx_sqlite::regexp=off\".parse()?);\n\n let is_interactive_search = matches!(&self, Self::Search(cmd) if cmd.is_interactive());\n // Use file-based logging for interactive search (TUI mode)\n let use_search_logging = is_interactive_search && settings.logs.search_enabled();\n\n // Use file-based logging for daemon\n #[cfg(feature = \"daemon\")]\n let use_daemon_logging = matches!(&self, Self::Daemon(_)) && settings.logs.daemon_enabled();\n\n #[cfg(not(feature = \"daemon\"))]\n let use_daemon_logging = false;\n\n // Check if daemon should also log to console\n #[cfg(feature = \"daemon\")]\n let daemon_show_logs = matches!(&self, Self::Daemon(cmd) if cmd.show_logs());\n\n #[cfg(not(feature = \"daemon\"))]\n let daemon_show_logs = false;\n\n // Set up span timing JSON logs if ATUIN_SPAN is set\n let span_path = std::env::var(\"ATUIN_SPAN\").ok().map(|p| {\n if p.is_empty() {\n \"atuin-spans.json\".to_string()\n } else {\n p\n }\n });\n\n // Helper to create span timing layer\n macro_rules! make_span_layer {\n ($path:expr) => {{\n let span_file = OpenOptions::new()\n .create(true)\n .truncate(true)\n .write(true)\n .open($path)?;\n Some(\n fmt::layer()\n .json()\n .with_writer(span_file)\n .with_span_events(FmtSpan::NEW | FmtSpan::CLOSE)\n .with_filter(LevelFilter::TRACE),\n )\n }};\n }\n\n // Build the subscriber with all configured layers\n if use_search_logging {\n let search_filename = settings.logs.search.file.clone();\n let log_dir = PathBuf::from(&settings.logs.dir);\n fs::create_dir_all(&log_dir)?;\n\n // Clean up old log files\n cleanup_old_logs(&log_dir, &search_filename, settings.logs.search_retention());\n\n let file_appender =\n RollingFileAppender::new(Rotation::DAILY, &log_dir, &search_filename);\n\n // Use config level unless ATUIN_LOG is set\n let filter = if env_log_set {\n base_filter\n } else {\n EnvFilter::default()\n .add_directive(settings.logs.search_level().as_directive().parse()?)\n .add_directive(\"sqlx_sqlite::regexp=off\".parse()?)\n };\n\n let base = tracing_subscriber::registry().with(\n fmt::layer()\n .with_writer(file_appender)\n .with_ansi(false)\n .with_filter(filter),\n );\n\n match &span_path {\n Some(sp) => {\n base.with(make_span_layer!(sp)).init();\n }\n None => {\n base.init();\n }\n }\n } else if use_daemon_logging {\n let daemon_filename = settings.logs.daemon.file.clone();\n let log_dir = PathBuf::from(&settings.logs.dir);\n fs::create_dir_all(&log_dir)?;\n\n // Clean up old log files\n cleanup_old_logs(&log_dir, &daemon_filename, settings.logs.daemon_retention());\n\n let file_appender =\n RollingFileAppender::new(Rotation::DAILY, &log_dir, &daemon_filename);\n\n // Use config level unless ATUIN_LOG is set\n let file_filter = if env_log_set {\n base_filter\n } else {\n EnvFilter::default()\n .add_directive(settings.logs.daemon_level().as_directive().parse()?)\n .add_directive(\"sqlx_sqlite::regexp=off\".parse()?)\n };\n\n let file_layer = fmt::layer()\n .with_writer(file_appender)\n .with_ansi(false)\n .with_filter(file_filter);\n\n // Optionally add console layer for --show-logs\n if daemon_show_logs {\n let console_filter = EnvFilter::from_env(\"ATUIN_LOG\")\n .add_directive(\"sqlx_sqlite::regexp=off\".parse()?);\n\n let console_layer = fmt::layer().with_filter(console_filter);\n\n let base = tracing_subscriber::registry()\n .with(file_layer)\n .with(console_layer);\n\n match &span_path {\n Some(sp) => {\n base.with(make_span_layer!(sp)).init();\n }\n None => {\n base.init();\n }\n }\n } else {\n let base = tracing_subscriber::registry().with(file_layer);\n\n match &span_path {\n Some(sp) => {\n base.with(make_span_layer!(sp)).init();\n }\n None => {\n base.init();\n }\n }\n }\n }\n\n tracing::trace!(command = ?self, \"client command\");\n\n // Skip initializing any databases for history\n // This is a pretty hot path, as it runs before and after every single command the user\n // runs\n match self {\n Self::History(history) => return history.run(&settings).await,\n Self::Init(init) => return init.run(&settings).await,\n Self::Doctor => return doctor::run(&settings).await,\n _ => {}\n }\n\n let db_path = PathBuf::from(settings.db_path.as_str());\n let record_store_path = PathBuf::from(settings.record_store_path.as_str());\n\n let db = Sqlite::new(db_path, settings.local_timeout).await?;\n let sqlite_store = SqliteStore::new(record_store_path, settings.local_timeout).await?;\n\n let theme_name = settings.theme.name.clone();\n let theme = theme_manager.load_theme(theme_name.as_str(), settings.theme.max_depth);\n\n match self {\n Self::Setup => setup::run(&settings).await,\n Self::Import(import) => import.run(&db).await,\n Self::Stats(stats) => stats.run(&db, &settings, theme).await,\n Self::Search(search) => search.run(db, &mut settings, sqlite_store, theme).await,\n\n #[cfg(feature = \"sync\")]\n Self::Sync(sync) => sync.run(settings, &db, sqlite_store).await,\n\n #[cfg(feature = \"sync\")]\n Self::Account(account) => account.run(settings, sqlite_store).await,\n\n Self::Kv(kv) => kv.run(&settings, &sqlite_store).await,\n\n Self::Store(store) => store.run(&settings, &db, sqlite_store).await,\n\n Self::Dotfiles(dotfiles) => dotfiles.run(&settings, sqlite_store).await,\n\n Self::Scripts(scripts) => scripts.run(&settings, sqlite_store, &db).await,\n\n Self::Info => {\n info::run(&settings);\n Ok(())\n }\n\n Self::DefaultConfig => {\n default_config::run();\n Ok(())\n }\n\n Self::Wrapped { year } => wrapped::run(year, &db, &settings, sqlite_store, theme).await,\n\n #[cfg(feature = \"daemon\")]\n Self::Daemon(cmd) => cmd.run(settings, sqlite_store, db).await,\n\n Self::History(_) | Self::Init(_) | Self::Doctor => unreachable!(),\n\n #[cfg(feature = \"ai\")]\n Self::Ai(cli) => atuin_ai::commands::run(cli, &settings).await,\n }\n }\n}\n" }, { "path": "crates/atuin/src/command/contributors.rs", "content": "static CONTRIBUTORS: &str = include_str!(\"CONTRIBUTORS\");\n\npub fn run() {\n println!(\"\\n{CONTRIBUTORS}\");\n}\n" }, { "path": "crates/atuin/src/command/external.rs", "content": "use std::fmt::Write as _;\nuse std::process::Command;\nuse std::{io, process};\n\n#[cfg(feature = \"client\")]\nuse atuin_client::plugin::OfficialPluginRegistry;\nuse clap::CommandFactory;\nuse clap::builder::{StyledStr, Styles};\nuse eyre::Result;\n\nuse crate::Atuin;\n\npub fn run(args: &[String]) -> Result<()> {\n let subcommand = &args[0];\n let bin = format!(\"atuin-{subcommand}\");\n let mut cmd = Command::new(&bin);\n cmd.args(&args[1..]);\n\n let spawn_result = match cmd.spawn() {\n Ok(child) => Ok(child),\n Err(e) => match e.kind() {\n io::ErrorKind::NotFound => {\n let output = render_not_found(subcommand, &bin);\n Err(output)\n }\n _ => Err(e.to_string().into()),\n },\n };\n\n match spawn_result {\n Ok(mut child) => {\n let status = child.wait()?;\n if status.success() {\n Ok(())\n } else {\n process::exit(status.code().unwrap_or(1));\n }\n }\n Err(e) => {\n eprintln!(\"{}\", e.ansi());\n process::exit(1);\n }\n }\n}\n\nfn render_not_found(subcommand: &str, bin: &str) -> StyledStr {\n let mut output = StyledStr::new();\n let styles = Styles::styled();\n\n let error = styles.get_error();\n let invalid = styles.get_invalid();\n let literal = styles.get_literal();\n\n #[cfg(feature = \"client\")]\n {\n let registry = OfficialPluginRegistry::new();\n\n // Check if this is an official plugin\n if let Some(install_message) = registry.get_install_message(subcommand) {\n let _ = write!(output, \"{error}error:{error:#} \");\n let _ = write!(\n output,\n \"'{invalid}{subcommand}{invalid:#}' is an official atuin plugin, but it's not installed\"\n );\n let _ = write!(output, \"\\n\\n\");\n let _ = write!(output, \"{install_message}\");\n return output;\n }\n }\n\n let mut atuin_cmd = Atuin::command();\n let usage = atuin_cmd.render_usage();\n\n let _ = write!(output, \"{error}error:{error:#} \");\n let _ = write!(\n output,\n \"unrecognized subcommand '{invalid}{subcommand}{invalid:#}' \"\n );\n let _ = write!(\n output,\n \"and no executable named '{invalid}{bin}{invalid:#}' found in your PATH\"\n );\n let _ = write!(output, \"\\n\\n\");\n let _ = write!(output, \"{usage}\");\n let _ = write!(output, \"\\n\\n\");\n let _ = write!(\n output,\n \"For more information, try '{literal}--help{literal:#}'.\"\n );\n\n output\n}\n" }, { "path": "crates/atuin/src/command/gen_completions.rs", "content": "use clap::{CommandFactory, Parser, ValueEnum};\nuse clap_complete::{Generator, Shell, generate, generate_to};\nuse clap_complete_nushell::Nushell;\nuse eyre::Result;\n\n// clap put nushell completions into a separate package due to the maintainers\n// being a little less committed to support them.\n// This means we have to do a tiny bit of legwork to combine these completions\n// into one command.\n#[derive(Debug, Clone, ValueEnum)]\n#[value(rename_all = \"lower\")]\npub enum GenShell {\n Bash,\n Elvish,\n Fish,\n Nushell,\n PowerShell,\n Zsh,\n}\n\nimpl Generator for GenShell {\n fn file_name(&self, name: &str) -> String {\n match self {\n // clap_complete\n Self::Bash => Shell::Bash.file_name(name),\n Self::Elvish => Shell::Elvish.file_name(name),\n Self::Fish => Shell::Fish.file_name(name),\n Self::PowerShell => Shell::PowerShell.file_name(name),\n Self::Zsh => Shell::Zsh.file_name(name),\n\n // clap_complete_nushell\n Self::Nushell => Nushell.file_name(name),\n }\n }\n\n fn generate(&self, cmd: &clap::Command, buf: &mut dyn std::io::prelude::Write) {\n match self {\n // clap_complete\n Self::Bash => Shell::Bash.generate(cmd, buf),\n Self::Elvish => Shell::Elvish.generate(cmd, buf),\n Self::Fish => Shell::Fish.generate(cmd, buf),\n Self::PowerShell => Shell::PowerShell.generate(cmd, buf),\n Self::Zsh => Shell::Zsh.generate(cmd, buf),\n\n // clap_complete_nushell\n Self::Nushell => Nushell.generate(cmd, buf),\n }\n }\n}\n\n#[derive(Debug, Parser)]\npub struct Cmd {\n /// Set the shell for generating completions\n #[arg(long, short)]\n shell: GenShell,\n\n /// Set the output directory\n #[arg(long, short)]\n out_dir: Option,\n}\n\nimpl Cmd {\n pub fn run(self) -> Result<()> {\n let Cmd { shell, out_dir } = self;\n\n let mut cli = crate::Atuin::command();\n\n match out_dir {\n Some(out_dir) => {\n generate_to(shell, &mut cli, env!(\"CARGO_PKG_NAME\"), &out_dir)?;\n }\n None => {\n generate(\n shell,\n &mut cli,\n env!(\"CARGO_PKG_NAME\"),\n &mut std::io::stdout(),\n );\n }\n }\n\n Ok(())\n }\n}\n" }, { "path": "crates/atuin/src/command/mod.rs", "content": "use clap::Subcommand;\nuse eyre::Result;\n\n#[cfg(not(windows))]\nuse rustix::{fs::Mode, process::umask};\n\n#[cfg(feature = \"client\")]\nmod client;\n\nmod contributors;\n\nmod gen_completions;\n\nmod external;\n\n#[derive(Subcommand)]\n#[command(infer_subcommands = true)]\n#[allow(clippy::large_enum_variant)]\npub enum AtuinCmd {\n #[cfg(feature = \"client\")]\n #[command(flatten)]\n Client(client::Cmd),\n\n /// Terminal emulator for atuin\n #[cfg(feature = \"hex\")]\n Hex {\n #[command(subcommand)]\n cmd: Option,\n },\n\n /// Generate a UUID\n Uuid,\n\n Contributors,\n\n /// Generate shell completions\n GenCompletions(gen_completions::Cmd),\n\n #[command(external_subcommand)]\n External(Vec),\n}\n\nimpl AtuinCmd {\n pub fn run(self) -> Result<()> {\n #[cfg(not(windows))]\n {\n // set umask before we potentially open/create files\n // or in other words, 077. Do not allow any access to any other user\n let mode = Mode::RWXG | Mode::RWXO;\n umask(mode);\n }\n\n match self {\n #[cfg(feature = \"client\")]\n Self::Client(client) => client.run(),\n\n #[cfg(feature = \"hex\")]\n Self::Hex { cmd } => {\n atuin_hex::run(cmd);\n Ok(())\n }\n\n Self::Contributors => {\n contributors::run();\n Ok(())\n }\n Self::Uuid => {\n println!(\"{}\", atuin_common::utils::uuid_v7().as_simple());\n Ok(())\n }\n Self::GenCompletions(gen_completions) => gen_completions.run(),\n Self::External(args) => external::run(&args),\n }\n }\n}\n" }, { "path": "crates/atuin/src/main.rs", "content": "#![warn(clippy::pedantic, clippy::nursery)]\n#![allow(clippy::use_self, clippy::missing_const_for_fn)] // not 100% reliable\n\nuse clap::Parser;\nuse clap::builder::Styles;\nuse clap::builder::styling::{AnsiColor, Effects};\nuse eyre::Result;\n\nuse command::AtuinCmd;\n\nmod command;\n\n#[cfg(feature = \"sync\")]\nmod sync;\n\nconst VERSION: &str = env!(\"CARGO_PKG_VERSION\");\nconst SHA: &str = env!(\"GIT_HASH\");\n\nconst LONG_VERSION: &str = concat!(env!(\"CARGO_PKG_VERSION\"), \" (\", env!(\"GIT_HASH\"), \")\");\n\nstatic HELP_TEMPLATE: &str = \"\\\n{before-help}{name} {version}\n{author}\n{about}\n\n{usage-heading}\n {usage}\n\n{all-args}{after-help}\";\n\nconst STYLES: Styles = Styles::styled()\n .header(AnsiColor::Yellow.on_default().effects(Effects::BOLD))\n .usage(AnsiColor::Green.on_default().effects(Effects::BOLD))\n .literal(AnsiColor::Green.on_default().effects(Effects::BOLD))\n .placeholder(AnsiColor::Green.on_default());\n\n/// Magical shell history\n#[derive(Parser)]\n#[command(\n author = \"Ellie Huxtable \",\n version = VERSION,\n long_version = LONG_VERSION,\n help_template(HELP_TEMPLATE),\n styles = STYLES,\n)]\nstruct Atuin {\n #[command(subcommand)]\n atuin: AtuinCmd,\n}\n\nimpl Atuin {\n fn run(self) -> Result<()> {\n self.atuin.run()\n }\n}\n\nfn main() -> Result<()> {\n Atuin::parse().run()\n}\n" }, { "path": "crates/atuin/src/shell/.gitattributes", "content": "* eol=lf\n" }, { "path": "crates/atuin/src/shell/atuin.bash", "content": "# Include guard\nif [[ ${__atuin_initialized-} == true ]]; then\n false\nelif [[ $- != *i* ]]; then\n # Enable only in interactive shells\n false\nelif ((BASH_VERSINFO[0] < 3 || BASH_VERSINFO[0] == 3 && BASH_VERSINFO[1] < 1)); then\n # Require bash >= 3.1\n [[ -t 2 ]] && printf 'atuin: requires bash >= 3.1 for the integration.\\n' >&2\n false\nelse # (include guard) beginning of main content\n#------------------------------------------------------------------------------\n__atuin_initialized=true\n\nif [[ -z \"${ATUIN_SESSION:-}\" || \"${ATUIN_SHLVL:-}\" != \"$SHLVL\" ]]; then\n ATUIN_SESSION=$(atuin uuid)\n export ATUIN_SESSION\n export ATUIN_SHLVL=$SHLVL\nfi\nATUIN_STTY=$(stty -g)\nATUIN_HISTORY_ID=\"\"\n\nexport ATUIN_PREEXEC_BACKEND=$SHLVL:none\n__atuin_update_preexec_backend() {\n if [[ ${BLE_ATTACHED-} ]]; then\n ATUIN_PREEXEC_BACKEND=$SHLVL:blesh-${BLE_VERSION-}\n elif [[ ${bash_preexec_imported-} ]]; then\n ATUIN_PREEXEC_BACKEND=$SHLVL:bash-preexec\n elif [[ ${__bp_imported-} ]]; then\n ATUIN_PREEXEC_BACKEND=\"$SHLVL:bash-preexec (old)\"\n else\n ATUIN_PREEXEC_BACKEND=$SHLVL:unknown\n fi\n}\n\n__atuin_preexec() {\n # Workaround for old versions of bash-preexec\n if [[ ! ${BLE_ATTACHED-} ]]; then\n # In older versions of bash-preexec, the preexec hook may be called\n # even for the commands run by keybindings. There is no general and\n # robust way to detect the command for keybindings, but at least we\n # want to exclude Atuin's keybindings. When the preexec hook is called\n # for a keybinding, the preexec hook for the user command will not\n # fire, so we instead set a fake ATUIN_HISTORY_ID here to notify\n # __atuin_precmd of this failure.\n if [[ $BASH_COMMAND != \"$1\" ]]; then\n case $BASH_COMMAND in\n '__atuin_history'* | '__atuin_widget_run'* | '__atuin_bash42_dispatch'*)\n ATUIN_HISTORY_ID=__bash_preexec_failure__\n return 0 ;;\n esac\n fi\n fi\n\n # Note: We update ATUIN_PREEXEC_BACKEND on every preexec because blesh's\n # attaching state can dynamically change.\n __atuin_update_preexec_backend\n\n local id\n id=$(atuin history start -- \"$1\" 2>/dev/null)\n export ATUIN_HISTORY_ID=$id\n __atuin_preexec_time=${EPOCHREALTIME-}\n}\n\n__atuin_precmd() {\n local EXIT=$? __atuin_precmd_time=${EPOCHREALTIME-}\n\n [[ ! $ATUIN_HISTORY_ID ]] && return\n\n # If the previous preexec hook failed, we manually call __atuin_preexec\n if [[ $ATUIN_HISTORY_ID == __bash_preexec_failure__ ]]; then\n # This is the command extraction code taken from bash-preexec\n local previous_command\n previous_command=$(\n export LC_ALL=C HISTTIMEFORMAT=''\n builtin history 1 | sed '1 s/^ *[0-9][0-9]*[* ] //'\n )\n __atuin_preexec \"$previous_command\"\n fi\n\n local duration=\"\"\n # shellcheck disable=SC2154,SC2309\n if [[ ${BLE_ATTACHED-} && ${_ble_exec_time_ata-} ]]; then\n # With ble.sh, we utilize the shell variable `_ble_exec_time_ata`\n # recorded by ble.sh. It is more accurate than the measurements by\n # Atuin, which includes the spawn cost of Atuin. ble.sh uses the\n # special shell variable `EPOCHREALTIME` in bash >= 5.0 with the\n # microsecond resolution, or the builtin `time` in bash < 5.0 with the\n # millisecond resolution.\n duration=${_ble_exec_time_ata}000\n elif ((BASH_VERSINFO[0] >= 5)); then\n # We calculate the high-resolution duration based on EPOCHREALTIME\n # (bash >= 5.0) recorded by precmd/preexec, though it might not be as\n # accurate as `_ble_exec_time_ata` provided by ble.sh because it\n # includes the extra time of the precmd/preexec handling. Since Bash\n # does not offer floating-point arithmetic, we remove the non-digit\n # characters and perform the integral arithmetic. The fraction part of\n # EPOCHREALTIME is fixed to have 6 digits in Bash. We remove all the\n # non-digit characters because the decimal point is not necessarily a\n # period depending on the locale.\n duration=$((${__atuin_precmd_time//[!0-9]} - ${__atuin_preexec_time//[!0-9]}))\n if ((duration >= 0)); then\n duration=${duration}000\n else\n duration=\"\" # clear the result on overflow\n fi\n fi\n\n (ATUIN_LOG=error atuin history end --exit \"$EXIT\" ${duration:+\"--duration=$duration\"} -- \"$ATUIN_HISTORY_ID\" &) >/dev/null 2>&1\n export ATUIN_HISTORY_ID=\"\"\n}\n\n__atuin_set_ret_value() {\n return ${1:+\"$1\"}\n}\n\n#------------------------------------------------------------------------------\n# section: __atuin_accept_line\n#\n# The function \"__atuin_accept_line\" is kept for backward compatibility of the\n# direct use of __atuin_history in keybindings by users.\n\n# The shell function `__atuin_evaluate_prompt` evaluates prompt sequences in\n# $PS1. We switch the implementation of the shell function\n# `__atuin_evaluate_prompt` based on the Bash version because the expansion\n# ${PS1@P} is only available in bash >= 4.4.\nif ((BASH_VERSINFO[0] >= 5 || BASH_VERSINFO[0] == 4 && BASH_VERSINFO[1] >= 4)); then\n __atuin_evaluate_prompt() {\n __atuin_set_ret_value \"${__bp_last_ret_value-}\" \"${__bp_last_argument_prev_command-}\"\n __atuin_prompt=${PS1@P}\n \n # Note: Strip the control characters ^A (\\001) and ^B (\\002), which\n # Bash internally uses to enclose the escape sequences. They are\n # produced by '\\[' and '\\]', respectively, in $PS1 and used to tell\n # Bash that the strings inbetween do not contribute to the prompt\n # width. After the prompt width calculation, Bash strips those control\n # characters before outputting it to the terminal. We here strip these\n # characters following Bash's behavior.\n __atuin_prompt=${__atuin_prompt//[$'\\001\\002']}\n\n # Count the number of newlines contained in $__atuin_prompt\n __atuin_prompt_offset=${__atuin_prompt//[!$'\\n']}\n __atuin_prompt_offset=${#__atuin_prompt_offset}\n }\nelse\n __atuin_evaluate_prompt() {\n __atuin_prompt='$ '\n __atuin_prompt_offset=0\n }\nfi\n\n# The shell function `__atuin_clear_prompt N` outputs terminal control\n# sequences to clear the contents of the current and N previous lines. After\n# clearing, the cursor is placed at the beginning of the N-th previous line.\n__atuin_clear_prompt_cache=()\n__atuin_clear_prompt() {\n local offset=$1\n if [[ ! ${__atuin_clear_prompt_cache[offset]+set} ]]; then\n if [[ ! ${__atuin_clear_prompt_cache[0]+set} ]]; then\n __atuin_clear_prompt_cache[0]=$'\\r'$(tput el 2>/dev/null || tput ce 2>/dev/null)\n fi\n if ((offset > 0)); then\n __atuin_clear_prompt_cache[offset]=${__atuin_clear_prompt_cache[0]}$(\n tput cuu \"$offset\" 2>/dev/null || tput UP \"$offset\" 2>/dev/null\n tput dl \"$offset\" 2>/dev/null || tput DL \"$offset\" 2>/dev/null\n tput il \"$offset\" 2>/dev/null || tput AL \"$offset\" 2>/dev/null\n )\n fi\n fi\n printf '%s' \"${__atuin_clear_prompt_cache[offset]}\"\n}\n\n__atuin_accept_line() {\n local __atuin_command=$1\n\n # Reprint the prompt, accounting for multiple lines\n local __atuin_prompt __atuin_prompt_offset\n __atuin_evaluate_prompt\n __atuin_clear_prompt \"$__atuin_prompt_offset\"\n printf '%s\\n' \"$__atuin_prompt$__atuin_command\"\n\n # Add it to the bash history\n history -s \"$__atuin_command\"\n\n # Assuming bash-preexec\n # Invoke every function in the preexec array\n local __atuin_preexec_function\n local __atuin_preexec_function_ret_value\n local __atuin_preexec_ret_value=0\n for __atuin_preexec_function in \"${preexec_functions[@]:-}\"; do\n if type -t \"$__atuin_preexec_function\" 1>/dev/null; then\n __atuin_set_ret_value \"${__bp_last_ret_value:-}\"\n \"$__atuin_preexec_function\" \"$__atuin_command\"\n __atuin_preexec_function_ret_value=$?\n if [[ $__atuin_preexec_function_ret_value != 0 ]]; then\n __atuin_preexec_ret_value=$__atuin_preexec_function_ret_value\n fi\n fi\n done\n\n # If extdebug is turned on and any preexec function returns non-zero\n # exit status, we do not run the user command.\n if ! { shopt -q extdebug && ((__atuin_preexec_ret_value)); }; then\n # Note: When a child Bash session is started by enter_accept, if the\n # environment variable READLINE_POINT is present, bash-preexec in the\n # child session does not fire preexec at all because it considers we\n # are inside Atuin's keybinding of the current session. To avoid\n # propagating the environment variable to the child session, we remove\n # the export attribute of READLINE_LINE and READLINE_POINT.\n export -n READLINE_LINE READLINE_POINT\n\n # Juggle the terminal settings so that the command can be interacted\n # with\n local __atuin_stty_backup\n __atuin_stty_backup=$(stty -g)\n stty \"$ATUIN_STTY\"\n\n # Execute the command. Note: We need to record $? and $_ after the\n # user command within the same call of \"eval\" because $_ is otherwise\n # overwritten by the last argument of \"eval\".\n __atuin_set_ret_value \"${__bp_last_ret_value-}\" \"${__bp_last_argument_prev_command-}\"\n eval -- \"$__atuin_command\"$'\\n__bp_last_ret_value=$? __bp_last_argument_prev_command=$_'\n\n stty \"$__atuin_stty_backup\"\n fi\n\n # Execute preprompt commands\n local __atuin_prompt_command\n for __atuin_prompt_command in \"${PROMPT_COMMAND[@]}\"; do\n __atuin_set_ret_value \"${__bp_last_ret_value-}\" \"${__bp_last_argument_prev_command-}\"\n eval -- \"$__atuin_prompt_command\"\n done\n # Bash will redraw only the line with the prompt after we finish,\n # so to work for a multiline prompt we need to print it ourselves,\n # then go to the beginning of the last line.\n __atuin_evaluate_prompt\n printf '%s' \"$__atuin_prompt\"\n __atuin_clear_prompt 0\n}\n\n#------------------------------------------------------------------------------\n\n# Check if tmux popup is available (tmux >= 3.2)\n__atuin_tmux_popup_check() {\n [[ -n \"${TMUX-}\" ]] || return 1\n [[ \"${ATUIN_TMUX_POPUP:-true}\" != \"false\" ]] || return 1\n\n # https://github.com/tmux/tmux/wiki/FAQ#how-often-is-tmux-released-what-is-the-version-number-scheme\n local tmux_version\n tmux_version=$(tmux -V 2>/dev/null | sed -n 's/^[^0-9]*\\([0-9][0-9]*\\.[0-9][0-9]*\\).*/\\1/p') # Could have used grep...\n [[ -z \"$tmux_version\" ]] && return 1\n\n local m1 m2\n m1=${tmux_version%%.*}\n m2=${tmux_version#*.}\n m2=${m2%%.*}\n [[ \"$m1\" =~ ^[0-9]+$ ]] || return 1\n [[ \"$m2\" =~ ^[0-9]+$ ]] || m2=0\n (( m1 > 3 || (m1 == 3 && m2 >= 2) ))\n}\n\n# Use global variable to fix scope issues with traps\n__atuin_popup_tmpdir=\"\"\n__atuin_tmux_popup_cleanup() {\n [[ -n \"$__atuin_popup_tmpdir\" && -d \"$__atuin_popup_tmpdir\" ]] && command rm -rf \"$__atuin_popup_tmpdir\"\n __atuin_popup_tmpdir=\"\"\n}\n\n__atuin_search_cmd() {\n local -a search_args=(\"$@\")\n\n if __atuin_tmux_popup_check; then\n __atuin_popup_tmpdir=$(mktemp -d) || return 1\n local result_file=\"$__atuin_popup_tmpdir/result\"\n\n trap '__atuin_tmux_popup_cleanup' EXIT HUP INT TERM\n\n local escaped_query escaped_args\n escaped_query=$(printf '%s' \"$READLINE_LINE\" | sed \"s/'/'\\\\\\\\''/g\")\n escaped_args=\"\"\n for arg in \"${search_args[@]}\"; do\n escaped_args+=\" '$(printf '%s' \"$arg\" | sed \"s/'/'\\\\\\\\''/g\")'\"\n done\n\n # In the popup, atuin goes to terminal, stderr goes to file\n local cdir popup_width popup_height\n cdir=$(pwd)\n popup_width=\"${ATUIN_TMUX_POPUP_WIDTH:-80%}\" # Keep default value anyways\n popup_height=\"${ATUIN_TMUX_POPUP_HEIGHT:-60%}\"\n tmux display-popup -d \"$cdir\" -w \"$popup_width\" -h \"$popup_height\" -E -E -- \\\n sh -c \"PATH='$PATH' ATUIN_SESSION='$ATUIN_SESSION' ATUIN_SHELL=bash ATUIN_LOG=error ATUIN_QUERY='$escaped_query' atuin search $escaped_args -i 2>'$result_file'\"\n\n if [[ -f \"$result_file\" ]]; then\n cat \"$result_file\"\n fi\n\n __atuin_tmux_popup_cleanup\n trap - EXIT HUP INT TERM\n else\n ATUIN_SHELL=bash ATUIN_LOG=error ATUIN_QUERY=$READLINE_LINE atuin search \"${search_args[@]}\" -i 3>&1 1>&2 2>&3\n fi\n}\n\n__atuin_history() {\n # Default action of the up key: When this function is called with the first\n # argument `--shell-up-key-binding`, we perform Atuin's history search only\n # when the up key is supposed to cause the history movement in the original\n # binding. We do this only for ble.sh because the up key always invokes\n # the history movement in the plain Bash.\n if [[ ${BLE_ATTACHED-} && ${1-} == --shell-up-key-binding ]]; then\n # When the current cursor position is not in the first line, the up key\n # should move the cursor to the previous line. While the selection is\n # performed, the up key should not start the history search.\n # shellcheck disable=SC2154 # Note: these variables are set by ble.sh\n if [[ ${_ble_edit_str::_ble_edit_ind} == *$'\\n'* || $_ble_edit_mark_active ]]; then\n ble/widget/@nomarked backward-line\n local status=$?\n READLINE_LINE=$_ble_edit_str\n READLINE_POINT=$_ble_edit_ind\n READLINE_MARK=$_ble_edit_mark\n return \"$status\"\n fi\n fi\n\n # READLINE_LINE and READLINE_POINT are only supported by bash >= 4.0 or\n # ble.sh. When it is not supported, we clear them to suppress strange\n # behaviors.\n [[ ${BLE_ATTACHED-} ]] || ((BASH_VERSINFO[0] >= 4)) ||\n READLINE_LINE=\"\" READLINE_POINT=0\n\n local __atuin_output\n __atuin_output=$(__atuin_search_cmd \"$@\")\n\n # We do nothing when the search is canceled.\n [[ $__atuin_output ]] || return 0\n\n if [[ $__atuin_output == __atuin_accept__:* ]]; then\n __atuin_output=${__atuin_output#__atuin_accept__:}\n\n if [[ ${BLE_ATTACHED-} ]]; then\n ble-edit/content/reset-and-check-dirty \"$__atuin_output\"\n ble/widget/accept-line\n READLINE_LINE=\"\"\n elif [[ ${__atuin_macro_chain_keymap-} ]]; then\n READLINE_LINE=$__atuin_output\n bind -m \"$__atuin_macro_chain_keymap\" '\"'\"$__atuin_macro_chain\"'\": '\"$__atuin_macro_accept_line\"\n else\n __atuin_accept_line \"$__atuin_output\"\n READLINE_LINE=\"\"\n fi\n\n READLINE_POINT=${#READLINE_LINE}\n else\n READLINE_LINE=$__atuin_output\n READLINE_POINT=${#READLINE_LINE}\n if [[ ! ${BLE_ATTACHED-} ]] && ((BASH_VERSINFO[0] < 4)) && [[ ${__atuin_macro_chain_keymap-} ]]; then\n bind -m \"$__atuin_macro_chain_keymap\" '\"'\"$__atuin_macro_chain\"'\": '\"$__atuin_macro_insert_line\"\n fi\n fi\n}\n\n__atuin_initialize_blesh() {\n # shellcheck disable=SC2154\n [[ ${BLE_VERSION-} ]] && ((_ble_version >= 400)) || return 0\n\n ble-import contrib/integration/bash-preexec\n\n # Define and register an autosuggestion source for ble.sh's auto-complete.\n # If you'd like to overwrite this, define the same name of shell function\n # after the $(atuin init bash) line in your .bashrc. If you do not need\n # the auto-complete source by Atuin, please add the following code to\n # remove the entry after the $(atuin init bash) line in your .bashrc:\n #\n # ble/util/import/eval-after-load core-complete '\n # ble/array#remove _ble_complete_auto_source atuin-history'\n #\n function ble/complete/auto-complete/source:atuin-history {\n local suggestion\n suggestion=$(ATUIN_QUERY=\"$_ble_edit_str\" atuin search --cmd-only --limit 1 --search-mode prefix 2>/dev/null)\n [[ $suggestion == \"$_ble_edit_str\"?* ]] || return 1\n ble/complete/auto-complete/enter h 0 \"${suggestion:${#_ble_edit_str}}\" '' \"$suggestion\"\n }\n ble/util/import/eval-after-load core-complete '\n ble/array#unshift _ble_complete_auto_source atuin-history'\n\n # @env BLE_SESSION_ID: `atuin doctor` references the environment variable\n # BLE_SESSION_ID. We explicitly export the variable because it was not\n # exported in older versions of ble.sh.\n [[ ${BLE_SESSION_ID-} ]] && export BLE_SESSION_ID\n}\n__atuin_initialize_blesh\nBLE_ONLOAD+=(__atuin_initialize_blesh)\nprecmd_functions+=(__atuin_precmd)\npreexec_functions+=(__atuin_preexec)\n\n#------------------------------------------------------------------------------\n# section: atuin-bind\n\n__atuin_widget=()\n\n__atuin_widget_save() {\n local data=$1\n for REPLY in \"${!__atuin_widget[@]}\"; do\n if [[ ${__atuin_widget[REPLY]} == \"$data\" ]]; then\n return 0\n fi\n done\n # shellcheck disable=SC2154\n REPLY=${#__atuin_widget[*]}\n __atuin_widget[REPLY]=$data\n}\n\n__atuin_widget_run() {\n local data=${__atuin_widget[$1]}\n local keymap=${data%%:*} widget=${data#*:}\n local __atuin_macro_chain_keymap=$keymap\n bind -m \"$keymap\" '\"'\"$__atuin_macro_chain\"'\": \"\"'\n builtin eval -- \"$widget\"\n}\n\n# To realize the enter_accept feature in a robust way, we need to call the\n# readline bindable function `accept-line'. However, there is no way to call\n# `accept-line' from the shell script. To call the bindable function\n# `accept-line', we may utilize string macros of readline. When we bind KEYSEQ\n# to a WIDGET that wants to conditionally call `accept-line' at the end, we\n# perform two-step dispatching:\n#\n# 1. [KEYSEQ -> IKEYSEQ1 IKEYSEQ2]---We first translate KEYSEQ to two\n# intermediate key sequences IKEYSEQ1 and IKEYSEQ2 using string macros. For\n# example, when we bind `__atuin_history` to \\C-r, this step can be set up by\n# `bind '\"\\C-r\": \"IKEYSEQ1IKEYSEQ2\"'`.\n#\n# 2. [IKEYSEQ1 -> WIDGET]---Then, IKEYSEQ1 is bound to the WIDGET, and the\n# binding of IKEYSEQ2 is dynamically determined by WIDGET. For example, when\n# we bind `__atuin_history` to \\C-r, this step can be set up by `bind -x\n# '\"IKEYSEQ1\": WIDGET'`.\n#\n# 3. [IKEYSEQ2 -> accept-line] or [IKEYSEQ2 -> \"\"]---To request the execution\n# of `accept-line', WIDGET can change the binding of IKEYSEQ2 by running\n# `bind '\"IKEYSEQ2\": accept-line''. Otherwise, WIDGET can change the binding\n# of IKEYSEQ2 to no-op by running `bind '\"IKEYSEQ2\": \"\"'`.\n#\n# For the choice of the intermediate key sequences, we want to choose key\n# sequences that are unlikely to conflict with others. In addition, we want to\n# avoid a key sequence containing \\e because keymap \"vi-insert\" stops\n# processing key sequences containing \\e in older versions of Bash. We have\n# used \\e[0;A (a variant of the [up] key with modifier ) in Atuin 3.10.0\n# for intermediate key sequences, but this contains \\e and caused a problem.\n# Instead, we use \\C-x\\C-_A\\a, which starts with \\C-x\\C-_ (an unlikely\n# two-byte combination) and A (represents the initial letter of Atuin),\n# followed by the payload and the terminator \\a (BEL, \\C-g).\n\n__atuin_macro_chain='\\C-x\\C-_A0\\a'\nfor __atuin_keymap in emacs vi-insert vi-command; do\n bind -m \"$__atuin_keymap\" \"\\\"$__atuin_macro_chain\\\": \\\"\\\"\"\ndone\nunset -v __atuin_keymap\n\nif ((BASH_VERSINFO[0] >= 5 || BASH_VERSINFO[0] == 4 && BASH_VERSINFO[1] >= 3)); then\n # In Bash >= 4.3\n\n __atuin_macro_accept_line=accept-line\n\n __atuin_bind_impl() {\n local keymap=$1 keyseq=$2 command=$3\n\n # Note: In Bash <= 5.0, the table for `bind -x` from the keyseq to the\n # command is shared by all the keymaps (emacs, vi-insert, and\n # vi-command), so one cannot safely bind different command strings to\n # the same keyseq in different keymaps. Therefore, the command string\n # and the keyseq need to be globally in one-to-one correspondence in\n # all the keymaps.\n local REPLY\n __atuin_widget_save \"$keymap:$command\"\n local widget=$REPLY\n local ikeyseq1='\\C-x\\C-_A'$((1 + widget))'\\a'\n local ikeyseq2=$__atuin_macro_chain\n\n if ((BASH_VERSINFO[0] == 5 && BASH_VERSINFO[1] == 1)); then\n # Workaround for Bash 5.1: Bash 5.1 has a bug that overwriting an\n # existing \"bind -x\" keybinding breaks other existing \"bind -x\"\n # keybindings [1,2]. To work around the problem, we explicitly\n # unbind an existing keybinding before overwriting it.\n #\n # [1] https://lists.gnu.org/archive/html/bug-bash/2021-04/msg00135.html\n # [2] https://github.com/atuinsh/atuin/issues/962#issuecomment-3451132291\n bind -m \"$keymap\" -r \"$keyseq\"\n fi\n\n bind -m \"$keymap\" \"\\\"$keyseq\\\": \\\"$ikeyseq1$ikeyseq2\\\"\"\n bind -m \"$keymap\" -x \"\\\"$ikeyseq1\\\": __atuin_widget_run $widget\"\n }\n\n __atuin_bind_blesh_onload() {\n # In ble.sh, we need to enable unrecognized CSI sequences like \\e[0;0A,\n # which are discarded by ble.sh by default. Note: In Bash <= 4.2, we\n # do not need to unset \"decode_error_cseq_discard\" because \\e[0;A is\n # used only for the macro chaining (which is unused by ble.sh) in Bash\n # <= 4.2.\n bleopt decode_error_cseq_discard=\n }\n if [[ ${BLE_VERSION-} ]]; then\n __atuin_bind_blesh_onload\n fi\n BLE_ONLOAD+=(__atuin_bind_blesh_onload)\nelse\n # In Bash <= 4.2, \"bind -x\" cannot bind a shell command to a keyseq having\n # more than two bytes, so we need to work with only two-byte sequences.\n #\n # However, the number of available combinations of two-byte sequences is\n # limited. To minimize the number of key sequences used by Atuin, instead\n # of specifying a widget by its own intermediate sequence, we specify a\n # widget by a fixed-length sequence of multiple two-byte sequences. More\n # specifically, instead of IKEYSEQ1, we use IKS1 IKS2 IKS3 [IKS4 IKS5]\n # IKSX, where IKS1..IKS5 just stores its information to a global variable,\n # and IKSX collects all the information and determine and call the actual\n # widget based on the stored information. Each of IKn (n=1..5) is one of\n # the two reserved sequences, $__atuin_bash42_code0 and\n # $__atuin_bash42_code1. IKSX is fixed to be $__atuin_bash42_code2.\n #\n # For the choices of the special key sequences, we consider \\C-xQ, \\C-xR,\n # and \\C-xS. In the emacs editing mode of Bash, \\C-x is used as a prefix\n # key, i.e., it is used for the beginning key of the keybindings with\n # multiple keys, so \\C-x is unlikely to be used for a single-key binding by\n # the user. Also, \\C-x is not used in the vi editing mode by default. The\n # combinations \\C-xQ..\\C-xS are also unlikely be used because we need to\n # switch the modifier keys from Control to Shift to input these sequences,\n # and these are not easy to input.\n __atuin_bash42_code0='\\C-xQ'\n __atuin_bash42_code1='\\C-xR'\n __atuin_bash42_code2='\\C-xS'\n\n __atuin_bash42_encode() {\n REPLY=\n local n=$1 min_width=${2-}\n while\n if ((n % 2 == 0)); then\n REPLY=$__atuin_bash42_code0$REPLY\n else\n REPLY=$__atuin_bash42_code1$REPLY\n fi\n (((n /= 2) || ${#REPLY} / ${#__atuin_bash42_code0} < min_width))\n do :; done\n }\n\n __atuin_bash42_bind() {\n local __atuin_keymap\n for __atuin_keymap in emacs vi-insert vi-command; do\n bind -m \"$__atuin_keymap\" -x '\"'\"$__atuin_bash42_code0\"'\": __atuin_bash42_dispatch_selector+=0'\n bind -m \"$__atuin_keymap\" -x '\"'\"$__atuin_bash42_code1\"'\": __atuin_bash42_dispatch_selector+=1'\n bind -m \"$__atuin_keymap\" -x '\"'\"$__atuin_bash42_code2\"'\": __atuin_bash42_dispatch'\n done\n }\n __atuin_bash42_bind\n # In Bash <= 4.2, there is no way to read users' \"bind -x\" settings, so we\n # need to explicitly perform \"bind -x\" when ble.sh is loaded.\n BLE_ONLOAD+=(__atuin_bash42_bind)\n\n if ((BASH_VERSINFO[0] >= 4)); then\n __atuin_macro_accept_line=accept-line\n else\n # Note: We rewrite the command line and invoke `accept-line'. In\n # bash <= 3.2, there is no way to rewrite the command line from the\n # shell script, so we rewrite it using a macro and\n # `shell-expand-line'.\n #\n # Note: Concerning the key sequences to invoke bindable functions\n # such as \"\\C-x\\C-_A1\\a\", another option is to use\n # \"\\exbegginning-of-line\\r\", etc. to make it consistent with bash\n # >= 5.3. However, an older Bash configuration can still conflict\n # on [M-x]. The conflict is more likely than \\C-x\\C-_A1\\a.\n for __atuin_keymap in emacs vi-insert vi-command; do\n bind -m \"$__atuin_keymap\" '\"\\C-x\\C-_A1\\a\": beginning-of-line'\n bind -m \"$__atuin_keymap\" '\"\\C-x\\C-_A2\\a\": kill-line'\n # shellcheck disable=SC2016\n bind -m \"$__atuin_keymap\" '\"\\C-x\\C-_A3\\a\": \"$READLINE_LINE\"'\n bind -m \"$__atuin_keymap\" '\"\\C-x\\C-_A4\\a\": shell-expand-line'\n bind -m \"$__atuin_keymap\" '\"\\C-x\\C-_A5\\a\": accept-line'\n bind -m \"$__atuin_keymap\" '\"\\C-x\\C-_A6\\a\": end-of-line'\n done\n unset -v __atuin_keymap\n\n bind -m vi-command '\"\\C-x\\C-_A7\\a\": vi-insertion-mode'\n bind -m vi-insert '\"\\C-x\\C-_A7\\a\": vi-movement-mode'\n\n # \"\\C-x\\C-_A10\\a\": Replace the command line with READLINE_LINE. When we are\n # in the vi-command keymap, we go to vi-insert, input\n # \"$READLINE_LINE\", and come back to vi-command.\n bind -m emacs '\"\\C-x\\C-_A10\\a\": \"\\C-x\\C-_A1\\a\\C-x\\C-_A2\\a\\C-x\\C-_A3\\a\\C-x\\C-_A4\\a\"'\n bind -m vi-insert '\"\\C-x\\C-_A10\\a\": \"\\C-x\\C-_A1\\a\\C-x\\C-_A2\\a\\C-x\\C-_A3\\a\\C-x\\C-_A4\\a\"'\n bind -m vi-command '\"\\C-x\\C-_A10\\a\": \"\\C-x\\C-_A1\\a\\C-x\\C-_A2\\a\\C-x\\C-_A7\\a\\C-x\\C-_A3\\a\\C-x\\C-_A7\\a\\C-x\\C-_A4\\a\"'\n\n __atuin_macro_accept_line='\"\\C-x\\C-_A10\\a\\C-x\\C-_A5\\a\"'\n __atuin_macro_insert_line='\"\\C-x\\C-_A10\\a\\C-x\\C-_A6\\a\"'\n fi\n\n __atuin_bash42_dispatch_selector=\n\n __atuin_bash42_dispatch() {\n local s=$__atuin_bash42_dispatch_selector\n __atuin_bash42_dispatch_selector=\n __atuin_widget_run \"$((2#0$s))\"\n }\n\n __atuin_bind_impl() {\n local keymap=$1 keyseq=$2 command=$3\n\n __atuin_widget_save \"$keymap:$command\"\n __atuin_bash42_encode \"$REPLY\"\n local macro=$REPLY$__atuin_bash42_code2$__atuin_macro_chain\n\n bind -m \"$keymap\" \"\\\"$keyseq\\\": \\\"$macro\\\"\"\n }\nfi\n\natuin-bind() {\n local keymap=\n local OPTIND=1 OPTARG=\"\" OPTERR=0 flag\n while getopts ':m:' flag \"$@\"; do\n case $flag in\n m) keymap=$OPTARG ;;\n *)\n printf '%s\\n' \"atuin-bind: unrecognized option '-$flag'\" >&2\n return 2\n ;;\n esac\n done\n shift \"$((OPTIND - 1))\"\n\n if (($# != 2)); then\n printf '%s\\n' 'usage: atuin-bind [-m keymap] keyseq widget' >&2\n return 2\n fi\n\n local keyseq=$1\n [[ $keymap ]] || keymap=$(bind -v | awk '$2 == \"keymap\" { print $3 }')\n case $keymap in\n emacs-meta) keymap=emacs keyseq='\\e'$keyseq ;;\n emacs-ctlx) keymap=emacs keyseq='\\C-x'$keyseq ;;\n emacs*) keymap=emacs ;;\n vi-insert) ;;\n vi*) keymap=vi-command ;;\n *)\n printf '%s\\n' \"atuin-bind: unknown keymap '$keymap'\" >&2\n return 2 ;;\n esac\n\n local command=$2 widget=${2%%[[:blank:]]*}\n case $widget in\n atuin-search) command=${2/#\"$widget\"/__atuin_history} ;;\n atuin-search-emacs) command=${2/#\"$widget\"/__atuin_history --keymap-mode=emacs} ;;\n atuin-search-viins) command=${2/#\"$widget\"/__atuin_history --keymap-mode=vim-insert} ;;\n atuin-search-vicmd) command=${2/#\"$widget\"/__atuin_history --keymap-mode=vim-normal} ;;\n atuin-up-search) command=${2/#\"$widget\"/__atuin_history --shell-up-key-binding} ;;\n atuin-up-search-emacs) command=${2/#\"$widget\"/__atuin_history --shell-up-key-binding --keymap-mode=emacs} ;;\n atuin-up-search-viins) command=${2/#\"$widget\"/__atuin_history --shell-up-key-binding --keymap-mode=vim-insert} ;;\n atuin-up-search-vicmd) command=${2/#\"$widget\"/__atuin_history --shell-up-key-binding --keymap-mode=vim-normal} ;;\n esac\n\n __atuin_bind_impl \"$keymap\" \"$keyseq\" \"$command\"\n}\n\n#------------------------------------------------------------------------------\n\n# shellcheck disable=SC2154\nif [[ $__atuin_bind_ctrl_r == true ]]; then\n # Note: We do not overwrite [C-r] in the vi-command keymap because we do\n # not want to overwrite \"redo\", which is already bound to [C-r] in the\n # vi_nmap keymap in ble.sh.\n atuin-bind -m emacs '\\C-r' atuin-search-emacs\n atuin-bind -m vi-insert '\\C-r' atuin-search-viins\n atuin-bind -m vi-command '/' atuin-search-emacs\nfi\n\n# shellcheck disable=SC2154\nif [[ $__atuin_bind_up_arrow == true ]]; then\n atuin-bind -m emacs '\\e[A' atuin-up-search-emacs\n atuin-bind -m emacs '\\eOA' atuin-up-search-emacs\n atuin-bind -m vi-insert '\\e[A' atuin-up-search-viins\n atuin-bind -m vi-insert '\\eOA' atuin-up-search-viins\n atuin-bind -m vi-command '\\e[A' atuin-up-search-vicmd\n atuin-bind -m vi-command '\\eOA' atuin-up-search-vicmd\n atuin-bind -m vi-command 'k' atuin-up-search-vicmd\nfi\n\n#------------------------------------------------------------------------------\nfi # (include guard) end of main content\n" }, { "path": "crates/atuin/src/shell/atuin.fish", "content": "if not set -q ATUIN_SESSION; or test \"$ATUIN_SHLVL\" != \"$SHLVL\"\n set -gx ATUIN_SESSION (atuin uuid)\n set -gx ATUIN_SHLVL $SHLVL\nend\nset --erase ATUIN_HISTORY_ID\n\nfunction _atuin_preexec --on-event fish_preexec\n if not test -n \"$fish_private_mode\"\n set -g ATUIN_HISTORY_ID (atuin history start -- \"$argv[1]\" 2>/dev/null)\n end\nend\n\nfunction _atuin_postexec --on-event fish_postexec\n set -l s $status\n\n if test -n \"$ATUIN_HISTORY_ID\"\n ATUIN_LOG=error atuin history end --exit $s -- $ATUIN_HISTORY_ID &>/dev/null &\n disown\n end\n\n set --erase ATUIN_HISTORY_ID\nend\n\n# Check if tmux popup is available (tmux >= 3.2)\nfunction _atuin_tmux_popup_check\n if not test -n \"$TMUX\"\n echo 0\n return\n end\n\n if test \"$ATUIN_TMUX_POPUP\" = false\n echo 0\n return\n end\n\n set -l tmux_version (tmux -V 2>/dev/null | string match -r '\\d+\\.\\d+')\n if not test -n \"$tmux_version\"\n echo 0\n return\n end\n\n set -l parts (string split '.' $tmux_version)\n set -l m1 $parts[1]\n set -l m2 0\n if test (count $parts) -ge 2\n set m2 $parts[2]\n end\n\n if not string match -rq '^[0-9]+$' -- \"$m1\"\n echo 0\n return\n end\n\n if not string match -rq '^[0-9]+$' -- \"$m2\"\n set m2 0\n end\n\n if test \"$m1\" -gt 3 2>/dev/null; or begin\n test \"$m1\" -eq 3 2>/dev/null; and test \"$m2\" -ge 2 2>/dev/null\n end\n echo 1\n else\n echo 0\n end\nend\n\nfunction _atuin_search\n set -l keymap_mode\n switch $fish_key_bindings\n case fish_vi_key_bindings\n switch $fish_bind_mode\n case default\n set keymap_mode vim-normal\n case insert\n set keymap_mode vim-insert\n end\n case '*'\n set keymap_mode emacs\n end\n\n set -l use_tmux_popup (_atuin_tmux_popup_check)\n\n set -l ATUIN_H\n if test \"$use_tmux_popup\" -eq 1\n set -l tmpdir (mktemp -d)\n if not test -d \"$tmpdir\"\n # if mktemp got errors\n set ATUIN_H (ATUIN_SHELL=fish ATUIN_LOG=error ATUIN_QUERY=(commandline -b) atuin search --keymap-mode=$keymap_mode $argv -i 3>&1 1>&2 2>&3 | string collect)\n else\n set -l result_file \"$tmpdir/result\"\n\n set -l query (commandline -b | string replace -a \"'\" \"'\\\\''\")\n set -l escaped_args \"\"\n for arg in $argv\n set escaped_args \"$escaped_args '\"(string replace -a \"'\" \"'\\\\''\" -- $arg)\"'\"\n end\n\n # In the popup, atuin goes to terminal, stderr goes to file\n set -l cdir (pwd)\n # Keep default value anyways\n set -l popup_width (test -n \"$ATUIN_TMUX_POPUP_WIDTH\" && echo \"$ATUIN_TMUX_POPUP_WIDTH\" || echo \"80%\")\n set -l popup_height (test -n \"$ATUIN_TMUX_POPUP_HEIGHT\" && echo \"$ATUIN_TMUX_POPUP_HEIGHT\" || echo \"60%\")\n tmux display-popup -d \"$cdir\" -w \"$popup_width\" -h \"$popup_height\" -E -E -- \\\n sh -c \"PATH='$PATH' ATUIN_SESSION='$ATUIN_SESSION' ATUIN_SHELL=fish ATUIN_LOG=error ATUIN_QUERY='$query' atuin search --keymap-mode=$keymap_mode$escaped_args -i 2>'$result_file'\"\n\n if test -f \"$result_file\"\n set ATUIN_H (cat \"$result_file\" | string collect)\n end\n\n command rm -rf \"$tmpdir\"\n end\n else\n # In fish 3.4 and above we can use `\"$(some command)\"` to keep multiple lines separate;\n # but to support fish 3.3 we need to use `(some command | string collect)`.\n # https://fishshell.com/docs/current/relnotes.html#id24 (fish 3.4 \"Notable improvements and fixes\")\n set ATUIN_H (ATUIN_SHELL=fish ATUIN_LOG=error ATUIN_QUERY=(commandline -b) atuin search --keymap-mode=$keymap_mode $argv -i 3>&1 1>&2 2>&3 | string collect)\n end\n\n set ATUIN_H (string trim -- $ATUIN_H | string collect) # trim whitespace\n\n if test -n \"$ATUIN_H\"\n if string match --quiet '__atuin_accept__:*' \"$ATUIN_H\"\n set -l ATUIN_HIST (string replace \"__atuin_accept__:\" \"\" -- \"$ATUIN_H\" | string collect)\n commandline -r \"$ATUIN_HIST\"\n commandline -f repaint\n commandline -f execute\n return\n else\n commandline -r \"$ATUIN_H\"\n end\n end\n\n commandline -f repaint\nend\n\nfunction _atuin_bind_up\n # Fallback to fish's builtin up-or-search if we're in search or paging mode\n if commandline --search-mode; or commandline --paging-mode\n up-or-search\n return\n end\n\n # Only invoke atuin if we're on the top line of the command\n set -l lineno (commandline --line)\n\n switch $lineno\n case 1\n _atuin_search --shell-up-key-binding\n case '*'\n up-or-search\n end\nend\n" }, { "path": "crates/atuin/src/shell/atuin.nu", "content": "# Source this in your ~/.config/nushell/config.nu\n# minimum supported version = 0.93.0\nmodule compat {\n export def --wrapped \"random uuid -v 7\" [...rest] { atuin uuid }\n}\nuse (if not (\n (version).major > 0 or\n (version).minor >= 103\n) { \"compat\" }) *\n\nif 'ATUIN_SESSION' not-in $env or ('ATUIN_SHLVL' not-in $env) or ($env.ATUIN_SHLVL != ($env.SHLVL? | default \"\")) {\n $env.ATUIN_SESSION = (random uuid -v 7 | str replace -a \"-\" \"\")\n $env.ATUIN_SHLVL = ($env.SHLVL? | default \"\")\n}\nhide-env -i ATUIN_HISTORY_ID\n\n# Magic token to make sure we don't record commands run by keybindings\nlet ATUIN_KEYBINDING_TOKEN = $\"# (random uuid)\"\n\nlet _atuin_pre_execution = {||\n if ($nu | get history-enabled?) == false {\n return\n }\n let cmd = (commandline)\n if ($cmd | is-empty) {\n return\n }\n if not ($cmd | str starts-with $ATUIN_KEYBINDING_TOKEN) {\n $env.ATUIN_HISTORY_ID = (atuin history start -- $cmd e>| complete | get stdout | str trim)\n }\n}\n\nlet _atuin_pre_prompt = {||\n let last_exit = $env.LAST_EXIT_CODE\n if 'ATUIN_HISTORY_ID' not-in $env {\n return\n }\n with-env { ATUIN_LOG: error } {\n if (version).minor >= 104 or (version).major > 0 {\n job spawn {\n ^atuin history end $'--exit=($env.LAST_EXIT_CODE)' -- $env.ATUIN_HISTORY_ID | complete\n } | ignore\n } else {\n do { atuin history end $'--exit=($last_exit)' -- $env.ATUIN_HISTORY_ID } | complete\n }\n\n }\n hide-env ATUIN_HISTORY_ID\n}\n\ndef _atuin_search_cmd [...flags: string] {\n if (version).minor >= 106 or (version).major > 0 {\n [\n $ATUIN_KEYBINDING_TOKEN,\n ([\n `with-env { ATUIN_LOG: error, ATUIN_QUERY: (commandline), ATUIN_SHELL: nu } {`,\n ([\n 'let output = (run-external atuin search',\n ($flags | append [--interactive] | each {|e| $'\"($e)\"'}),\n 'e>| str trim)',\n ] | flatten | str join ' '),\n 'if ($output | str starts-with \"__atuin_accept__:\") {',\n 'commandline edit --accept ($output | str replace \"__atuin_accept__:\" \"\")',\n '} else {',\n 'commandline edit $output',\n '}',\n `}`,\n ] | flatten | str join \"\\n\"),\n ]\n } else {\n [\n $ATUIN_KEYBINDING_TOKEN,\n ([\n `with-env { ATUIN_LOG: error, ATUIN_QUERY: (commandline) } {`,\n 'commandline edit',\n '(run-external atuin search',\n ($flags | append [--interactive] | each {|e| $'\"($e)\"'}),\n ' e>| str trim)',\n `}`,\n ] | flatten | str join ' '),\n ]\n } | str join \"\\n\"\n}\n\n$env.config = ($env | default {} config).config\n$env.config = ($env.config | default {} hooks)\n$env.config = (\n $env.config | upsert hooks (\n $env.config.hooks\n | upsert pre_execution (\n $env.config.hooks | get pre_execution? | default [] | append $_atuin_pre_execution)\n | upsert pre_prompt (\n $env.config.hooks | get pre_prompt? | default [] | append $_atuin_pre_prompt)\n )\n)\n\n$env.config = ($env.config | default [] keybindings)\n" }, { "path": "crates/atuin/src/shell/atuin.ps1", "content": "# Atuin PowerShell module\n#\n# This should support PowerShell 5.1 (which is shipped with Windows) and later versions, on Windows and Linux.\n#\n# Usage: atuin init powershell | Out-String | Invoke-Expression\n#\n# Settings:\n# - $env:ATUIN_POWERSHELL_PROMPT_OFFSET - Number of lines to offset the prompt position after exiting search.\n# This is useful when using a multi-line prompt: e.g. set this to -1 when using a 2-line prompt.\n# It is initialized from the current prompt line count if not set when the first Atuin search is performed.\n\nif (Get-Module Atuin -ErrorAction Ignore) {\n if ($PSVersionTable.PSVersion.Major -ge 7) {\n Write-Warning \"The Atuin module is already loaded, replacing it.\"\n Remove-Module Atuin\n } else {\n Write-Warning \"The Atuin module is already loaded, skipping.\"\n return\n }\n}\n\nif (!(Get-Command atuin -ErrorAction Ignore)) {\n Write-Error \"The 'atuin' executable needs to be available in the PATH.\"\n return\n}\n\nif (!(Get-Module PSReadLine -ErrorAction Ignore)) {\n Write-Error \"Atuin requires the PSReadLine module to be installed.\"\n return\n}\n\nNew-Module -Name Atuin -ScriptBlock {\n if (-not $env:ATUIN_SESSION -or $env:ATUIN_PID -ne $PID) {\n $env:ATUIN_SESSION = atuin uuid\n $env:ATUIN_PID = $PID\n }\n\n $script:atuinHistoryId = $null\n $script:previousPSConsoleHostReadLine = $Function:PSConsoleHostReadLine\n\n # The ReadLine overloads changed with breaking changes over time, make sure the one we expect is available.\n $script:hasExpectedReadLineOverload = ([Microsoft.PowerShell.PSConsoleReadLine]::ReadLine).OverloadDefinitions.Contains(\"static string ReadLine(runspace runspace, System.Management.Automation.EngineIntrinsics engineIntrinsics, System.Threading.CancellationToken cancellationToken, System.Nullable[bool] lastRunStatus)\")\n\n function Get-CommandLine {\n $commandLine = \"\"\n [Microsoft.PowerShell.PSConsoleReadLine]::GetBufferState([ref]$commandLine, [ref]$null)\n return $commandLine\n }\n\n function Set-CommandLine {\n param([string]$Text)\n\n $commandLine = Get-CommandLine\n [Microsoft.PowerShell.PSConsoleReadLine]::Replace(0, $commandLine.Length, $Text)\n }\n\n # This function name is called by PSReadLine to read the next command line to execute.\n # We replace it with a custom implementation which adds Atuin support.\n function PSConsoleHostReadLine {\n ## 1. Collect the exit code of the previous command.\n\n # This needs to be done as the first thing because any script run will flush $?.\n $lastRunStatus = $?\n\n # Exit statuses are maintained separately for native and PowerShell commands, this needs to be taken into account.\n $lastNativeExitCode = $global:LASTEXITCODE\n $exitCode = if ($lastRunStatus) { 0 } elseif ($lastNativeExitCode) { $lastNativeExitCode } else { 1 }\n\n ## 2. Report the status of the previous command to Atuin (atuin history end).\n\n if ($script:atuinHistoryId) {\n try {\n # The duration is not recorded in old PowerShell versions, let Atuin handle it. $null arguments are ignored.\n $duration = (Get-History -Count 1).Duration.Ticks * 100\n $durationArg = if ($duration) { \"--duration=$duration\" } else { $null }\n\n # Fire and forget the atuin history end command to avoid blocking the shell during a potential sync.\n $process = New-Object System.Diagnostics.Process\n $process.StartInfo.FileName = \"atuin\"\n $process.StartInfo.Arguments = \"history end --exit=$exitCode $durationArg -- $script:atuinHistoryId\"\n $process.StartInfo.UseShellExecute = $false\n $process.StartInfo.CreateNoWindow = $true\n $process.StartInfo.RedirectStandardInput = $true\n $process.StartInfo.RedirectStandardOutput = $true\n $process.StartInfo.RedirectStandardError = $true\n $process.Start() | Out-Null\n $process.StandardInput.Close()\n $process.BeginOutputReadLine()\n $process.BeginErrorReadLine()\n }\n catch {\n # Ignore errors to avoid breaking the shell.\n # An error would occur if the user removes atuin from the PATH, for instance.\n }\n finally {\n $script:atuinHistoryId = $null\n }\n }\n\n ## 3. Read the next command line to execute.\n\n # PSConsoleHostReadLine implementation from PSReadLine, adjusted to support old versions.\n Microsoft.PowerShell.Core\\Set-StrictMode -Off\n\n $line = if ($script:hasExpectedReadLineOverload) {\n # When the overload we expect is available, we can pass $lastRunStatus to it.\n [Microsoft.PowerShell.PSConsoleReadLine]::ReadLine($Host.Runspace, $ExecutionContext, [System.Threading.CancellationToken]::None, $lastRunStatus)\n } else {\n # Either PSReadLine is older than v2.2.0-beta3, or maybe newer than we expect, so use the function from PSReadLine as-is.\n & $script:previousPSConsoleHostReadLine\n }\n\n ## 4. Report the next command line to Atuin (atuin history start).\n\n # PowerShell doesn't handle double quotes in native command line arguments the same way depending on its version,\n # and the value of $PSNativeCommandArgumentPassing - see the about_Parsing help page which explains the breaking changes.\n # This makes it unreliable, so we go through an environment variable, which should always be consistent across versions.\n try {\n $env:ATUIN_COMMAND_LINE = $line\n $script:atuinHistoryId = atuin history start --command-from-env\n }\n catch {\n # Ignore errors to avoid breaking the shell, see above.\n }\n finally {\n $env:ATUIN_COMMAND_LINE = $null\n }\n\n $global:LASTEXITCODE = $lastNativeExitCode\n return $line\n }\n\n function Invoke-AtuinSearch {\n param([string]$ExtraArgs = \"\")\n\n $previousOutputEncoding = [System.Console]::OutputEncoding\n $resultFile = New-TemporaryFile\n $suggestion = \"\"\n $errorOutput = \"\"\n\n try {\n [System.Console]::OutputEncoding = [System.Text.Encoding]::UTF8\n\n # Start-Process does some crazy stuff, just use the Process class directly to have more control.\n $process = New-Object System.Diagnostics.Process\n $process.StartInfo.FileName = \"atuin\"\n $process.StartInfo.Arguments = \"search -i --result-file \"\"$resultFile\"\" $ExtraArgs\"\n $process.StartInfo.UseShellExecute = $false\n $process.StartInfo.RedirectStandardError = $true\n $process.StartInfo.StandardErrorEncoding = [System.Text.Encoding]::UTF8\n $process.StartInfo.EnvironmentVariables[\"ATUIN_SHELL\"] = \"powershell\"\n $process.StartInfo.EnvironmentVariables[\"ATUIN_QUERY\"] = Get-CommandLine\n\n try {\n $process.Start() | Out-Null\n\n # A single stream is redirected, so we can read it synchronously, but we have to start reading it\n # before waiting for the process to exit, otherwise the buffer could fill up and cause a deadlock.\n $errorOutput = $process.StandardError.ReadToEnd().Trim()\n $process.WaitForExit()\n\n $suggestion = (Get-Content -Raw $resultFile -Encoding UTF8 | Out-String).Trim()\n }\n catch {\n $errorOutput = $_\n }\n\n if ($errorOutput) {\n Write-Host -ForegroundColor Red \"Atuin error:\"\n Write-Host -ForegroundColor DarkRed $errorOutput\n }\n\n # If no shell prompt offset is set, initialize it from the current prompt line count.\n if ($null -eq $env:ATUIN_POWERSHELL_PROMPT_OFFSET) {\n try {\n $promptLines = (& $Function:prompt | Out-String | Measure-Object -Line).Lines\n $env:ATUIN_POWERSHELL_PROMPT_OFFSET = -1 * ($promptLines - 1)\n }\n catch {\n $env:ATUIN_POWERSHELL_PROMPT_OFFSET = 0\n }\n }\n\n # PSReadLine maintains its own cursor position, which will no longer be valid if Atuin scrolls the display in inline mode.\n # Fortunately, InvokePrompt can receive a new Y position and reset the internal state.\n $y = $Host.UI.RawUI.CursorPosition.Y + [int]$env:ATUIN_POWERSHELL_PROMPT_OFFSET\n $y = [System.Math]::Max([System.Math]::Min($y, [System.Console]::BufferHeight - 1), 0)\n [Microsoft.PowerShell.PSConsoleReadLine]::InvokePrompt($null, $y)\n\n if ($suggestion -eq \"\") {\n # The previous input was already rendered by InvokePrompt\n return\n }\n\n $acceptPrefix = \"__atuin_accept__:\"\n\n if ( $suggestion.StartsWith($acceptPrefix)) {\n Set-CommandLine $suggestion.Substring($acceptPrefix.Length)\n [Microsoft.PowerShell.PSConsoleReadLine]::AcceptLine()\n } else {\n Set-CommandLine $suggestion\n }\n }\n finally {\n [System.Console]::OutputEncoding = $previousOutputEncoding\n Remove-Item $resultFile\n }\n }\n\n function Enable-AtuinSearchKeys {\n param([bool]$CtrlR = $true, [bool]$UpArrow = $true)\n\n if ($CtrlR) {\n Set-PSReadLineKeyHandler -Chord \"Ctrl+r\" -BriefDescription \"Runs Atuin search\" -ScriptBlock {\n Invoke-AtuinSearch\n }\n }\n\n if ($UpArrow) {\n Set-PSReadLineKeyHandler -Chord \"UpArrow\" -BriefDescription \"Runs Atuin search\" -ScriptBlock {\n $line = Get-CommandLine\n\n if (!$line.Contains(\"`n\")) {\n Invoke-AtuinSearch -ExtraArgs \"--shell-up-key-binding\"\n } else {\n [Microsoft.PowerShell.PSConsoleReadLine]::PreviousLine()\n }\n }\n }\n }\n\n $ExecutionContext.SessionState.Module.OnRemove += {\n $env:ATUIN_SESSION = $null\n $Function:PSConsoleHostReadLine = $script:previousPSConsoleHostReadLine\n }\n\n Export-ModuleMember -Function @(\"Enable-AtuinSearchKeys\", \"PSConsoleHostReadLine\")\n} | Import-Module -Global\n" }, { "path": "crates/atuin/src/shell/atuin.xsh", "content": "import subprocess\n\nfrom prompt_toolkit.application.current import get_app\nfrom prompt_toolkit.filters import Condition\nfrom prompt_toolkit.keys import Keys\n\n\nif \"ATUIN_SESSION\" not in ${...} or ${...}.get(\"ATUIN_SHLVL\", \"\") != ${...}.get(\"SHLVL\", \"\"):\n $ATUIN_SESSION=$(atuin uuid).rstrip('\\n')\n $ATUIN_SHLVL = ${...}.get(\"SHLVL\", \"\")\n\n@events.on_precommand\ndef _atuin_precommand(cmd: str):\n cmd = cmd.rstrip(\"\\n\")\n try:\n $ATUIN_HISTORY_ID = $(atuin history start -- @(cmd) 2>/dev/null).rstrip(\"\\n\")\n except:\n $ATUIN_HISTORY_ID = \"\"\n\n\n@events.on_postcommand\ndef _atuin_postcommand(cmd: str, rtn: int, out, ts):\n if \"ATUIN_HISTORY_ID\" not in ${...}:\n return\n\n duration = ts[1] - ts[0]\n # Duration is float representing seconds, but atuin expects integer of nanoseconds\n nanos = round(duration * 10 ** 9)\n with ${...}.swap(ATUIN_LOG=\"error\"):\n # This causes the entire .xonshrc to be re-executed, which is incredibly slow\n # This happens when using a subshell and using output redirection at the same time\n # For more details, see https://github.com/xonsh/xonsh/issues/5224\n # (atuin history end --exit @(rtn) -- $ATUIN_HISTORY_ID &) > /dev/null 2>&1\n atuin history end --exit @(rtn) --duration @(nanos) -- $ATUIN_HISTORY_ID > /dev/null 2>&1\n del $ATUIN_HISTORY_ID\n\n\ndef _search(event, extra_args: list[str]):\n buffer = event.current_buffer\n cmd = [\"atuin\", \"search\", \"--interactive\", *extra_args]\n # We need to explicitly pass in xonsh env, in case user has set XDG_HOME or something else that matters\n env = ${...}.detype()\n env[\"ATUIN_SHELL\"] = \"xonsh\"\n env[\"ATUIN_QUERY\"] = buffer.text\n\n p = subprocess.run(cmd, stderr=subprocess.PIPE, encoding=\"utf-8\", env=env)\n result = p.stderr.rstrip(\"\\n\")\n # redraw prompt - necessary if atuin is configured to run inline, rather than fullscreen\n event.cli.renderer.erase()\n\n if not result:\n return\n\n buffer.reset()\n if result.startswith(\"__atuin_accept__:\"):\n buffer.insert_text(result[17:])\n buffer.validate_and_handle()\n else:\n buffer.insert_text(result)\n\n\n@events.on_ptk_create\ndef _custom_keybindings(bindings, **kw):\n if _ATUIN_BIND_CTRL_R:\n @bindings.add(Keys.ControlR)\n def r_search(event):\n _search(event, extra_args=[])\n\n if _ATUIN_BIND_UP_ARROW:\n @Condition\n def should_search():\n buffer = get_app().current_buffer\n # disable keybind when there is an active completion, so\n # that up arrow can be used to navigate completion menu\n if buffer.complete_state is not None:\n return False\n # similarly, disable when buffer text contains multiple lines\n if '\\n' in buffer.text:\n return False\n\n return True\n\n @bindings.add(Keys.Up, filter=should_search)\n def up_search(event):\n _search(event, extra_args=[\"--shell-up-key-binding\"])\n" }, { "path": "crates/atuin/src/shell/atuin.zsh", "content": "# shellcheck disable=SC2034,SC2153,SC2086,SC2155\n\n# Above line is because shellcheck doesn't support zsh, per\n# https://github.com/koalaman/shellcheck/wiki/SC1071, and the ignore: param in\n# ludeeus/action-shellcheck only supports _directories_, not _files_. So\n# instead, we manually add any error the shellcheck step finds in the file to\n# the above line ...\n\n# Source this in your ~/.zshrc\nautoload -U add-zsh-hook\n\nzmodload zsh/datetime 2>/dev/null\n\n# If zsh-autosuggestions is installed, configure it to use Atuin's search. If\n# you'd like to override this, then add your config after the $(atuin init zsh)\n# in your .zshrc\n_zsh_autosuggest_strategy_atuin() {\n # silence errors, since we don't want to spam the terminal prompt while typing.\n suggestion=$(ATUIN_QUERY=\"$1\" atuin search --cmd-only --limit 1 --search-mode prefix 2>/dev/null)\n}\n\nif [ -n \"${ZSH_AUTOSUGGEST_STRATEGY:-}\" ]; then\n ZSH_AUTOSUGGEST_STRATEGY=(\"atuin\" \"${ZSH_AUTOSUGGEST_STRATEGY[@]}\")\nelse\n ZSH_AUTOSUGGEST_STRATEGY=(\"atuin\")\nfi\n\nif [[ -z \"${ATUIN_SESSION:-}\" || \"${ATUIN_SHLVL:-}\" != \"$SHLVL\" ]]; then\n export ATUIN_SESSION=$(atuin uuid)\n export ATUIN_SHLVL=$SHLVL\nfi\nATUIN_HISTORY_ID=\"\"\n\n_atuin_preexec() {\n local id\n id=$(atuin history start -- \"$1\" 2>/dev/null)\n export ATUIN_HISTORY_ID=\"$id\"\n __atuin_preexec_time=${EPOCHREALTIME-}\n}\n\n_atuin_precmd() {\n local EXIT=\"$?\" __atuin_precmd_time=${EPOCHREALTIME-}\n\n [[ -z \"${ATUIN_HISTORY_ID:-}\" ]] && return\n\n local duration=\"\"\n if [[ -n $__atuin_preexec_time && -n $__atuin_precmd_time ]]; then\n printf -v duration %.0f $(((__atuin_precmd_time - __atuin_preexec_time) * 1000000000))\n fi\n\n (ATUIN_LOG=error atuin history end --exit $EXIT ${duration:+--duration=$duration} -- $ATUIN_HISTORY_ID &) >/dev/null 2>&1\n export ATUIN_HISTORY_ID=\"\"\n}\n\n# Check if tmux popup is available (tmux >= 3.2)\n__atuin_tmux_popup_check() {\n [[ -n \"${TMUX-}\" ]] || return 1\n [[ \"${ATUIN_TMUX_POPUP:-true}\" != \"false\" ]] || return 1\n\n # https://github.com/tmux/tmux/wiki/FAQ#how-often-is-tmux-released-what-is-the-version-number-scheme\n local tmux_version\n tmux_version=$(tmux -V 2>/dev/null | sed -n 's/^[^0-9]*\\([0-9][0-9]*\\.[0-9][0-9]*\\).*/\\1/p') # Could have used grep...\n [[ -z \"$tmux_version\" ]] && return 1\n\n local m1 m2\n m1=${tmux_version%%.*}\n m2=${tmux_version#*.}\n m2=${m2%%.*}\n [[ \"$m1\" =~ ^[0-9]+$ ]] || return 1\n [[ \"$m2\" =~ ^[0-9]+$ ]] || m2=0\n (( m1 > 3 || (m1 == 3 && m2 >= 2) ))\n}\n\n# Use global variable to fix scope issues with traps\n__atuin_popup_tmpdir=\"\"\n__atuin_tmux_popup_cleanup() {\n [[ -n \"$__atuin_popup_tmpdir\" && -d \"$__atuin_popup_tmpdir\" ]] && command rm -rf \"$__atuin_popup_tmpdir\"\n __atuin_popup_tmpdir=\"\"\n}\n\n__atuin_search_cmd() {\n local -a search_args=(\"$@\")\n\n if __atuin_tmux_popup_check; then\n __atuin_popup_tmpdir=$(mktemp -d) || return 1\n local result_file=\"$__atuin_popup_tmpdir/result\"\n\n trap '__atuin_tmux_popup_cleanup' EXIT HUP INT TERM\n\n local escaped_query escaped_args\n escaped_query=$(printf '%s' \"$BUFFER\" | sed \"s/'/'\\\\\\\\''/g\")\n escaped_args=\"\"\n for arg in \"${search_args[@]}\"; do\n escaped_args+=\" '$(printf '%s' \"$arg\" | sed \"s/'/'\\\\\\\\''/g\")'\"\n done\n\n # In the popup, atuin goes to terminal, stderr goes to file\n local cdir popup_width popup_height\n cdir=$(pwd)\n popup_width=\"${ATUIN_TMUX_POPUP_WIDTH:-80%}\" # Keep default value anyways\n popup_height=\"${ATUIN_TMUX_POPUP_HEIGHT:-60%}\"\n tmux display-popup -d \"$cdir\" -w \"$popup_width\" -h \"$popup_height\" -E -E -- \\\n sh -c \"PATH='$PATH' ATUIN_SESSION='$ATUIN_SESSION' ATUIN_SHELL=zsh ATUIN_LOG=error ATUIN_QUERY='$escaped_query' atuin search $escaped_args -i 2>'$result_file'\"\n\n if [[ -f \"$result_file\" ]]; then\n cat \"$result_file\"\n fi\n\n __atuin_tmux_popup_cleanup\n trap - EXIT HUP INT TERM\n else\n ATUIN_SHELL=zsh ATUIN_LOG=error ATUIN_QUERY=$BUFFER atuin search \"${search_args[@]}\" -i 3>&1 1>&2 2>&3\n fi\n}\n\n_atuin_search() {\n emulate -L zsh\n zle -I\n\n # swap stderr and stdout, so that the tui stuff works\n # TODO: not this\n local output\n # shellcheck disable=SC2048\n output=$(__atuin_search_cmd $*)\n\n zle reset-prompt\n # re-enable bracketed paste\n # shellcheck disable=SC2154\n echo -n ${zle_bracketed_paste[1]} >/dev/tty\n\n if [[ -n $output ]]; then\n RBUFFER=\"\"\n LBUFFER=$output\n\n if [[ $LBUFFER == __atuin_accept__:* ]]\n then\n LBUFFER=${LBUFFER#__atuin_accept__:}\n zle accept-line\n fi\n fi\n}\n_atuin_search_vicmd() {\n _atuin_search --keymap-mode=vim-normal\n}\n_atuin_search_viins() {\n _atuin_search --keymap-mode=vim-insert\n}\n\n_atuin_up_search() {\n # Only trigger if the buffer is a single line\n if [[ ! $BUFFER == *$'\\n'* ]]; then\n _atuin_search --shell-up-key-binding \"$@\"\n else\n zle up-line\n fi\n}\n_atuin_up_search_vicmd() {\n _atuin_up_search --keymap-mode=vim-normal\n}\n_atuin_up_search_viins() {\n _atuin_up_search --keymap-mode=vim-insert\n}\n\nadd-zsh-hook preexec _atuin_preexec\nadd-zsh-hook precmd _atuin_precmd\n\nzle -N atuin-search _atuin_search\nzle -N atuin-search-vicmd _atuin_search_vicmd\nzle -N atuin-search-viins _atuin_search_viins\nzle -N atuin-up-search _atuin_up_search\nzle -N atuin-up-search-vicmd _atuin_up_search_vicmd\nzle -N atuin-up-search-viins _atuin_up_search_viins\n\n# These are compatibility widget names for \"atuin <= 17.2.1\" users.\nzle -N _atuin_search_widget _atuin_search\nzle -N _atuin_up_search_widget _atuin_up_search\n" }, { "path": "crates/atuin/src/sync.rs", "content": "use atuin_dotfiles::store::{AliasStore, var::VarStore};\nuse atuin_scripts::store::ScriptStore;\nuse eyre::{Context, Result};\n\nuse atuin_client::{\n database::Database, history::store::HistoryStore, record::sqlite_store::SqliteStore,\n settings::Settings,\n};\nuse atuin_common::record::RecordId;\nuse atuin_kv::store::KvStore;\n\n// This is the only crate that ties together all other crates.\n// Therefore, it's the only crate where functions tying together all stores can live\n\n/// Rebuild all stores after a sync\n/// Note: for history, this only does an _incremental_ sync. Hence the need to specify downloaded\n/// records.\npub async fn build(\n settings: &Settings,\n store: &SqliteStore,\n db: &dyn Database,\n downloaded: Option<&[RecordId]>,\n) -> Result<()> {\n let encryption_key: [u8; 32] = atuin_client::encryption::load_key(settings)\n .context(\"could not load encryption key\")?\n .into();\n\n let host_id = Settings::host_id().await?;\n\n let downloaded = downloaded.unwrap_or(&[]);\n\n let kv_db = atuin_kv::database::Database::new(settings.kv.db_path.clone(), 1.0).await?;\n\n let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);\n let alias_store = AliasStore::new(store.clone(), host_id, encryption_key);\n let var_store = VarStore::new(store.clone(), host_id, encryption_key);\n let kv_store = KvStore::new(store.clone(), kv_db, host_id, encryption_key);\n let script_store = ScriptStore::new(store.clone(), host_id, encryption_key);\n\n history_store.incremental_build(db, downloaded).await?;\n\n alias_store.build().await?;\n var_store.build().await?;\n kv_store.build().await?;\n\n let script_db =\n atuin_scripts::database::Database::new(settings.scripts.db_path.clone(), 1.0).await?;\n script_store.build(script_db).await?;\n Ok(())\n}\n" }, { "path": "crates/atuin/tests/common/mod.rs", "content": "use std::{env, time::Duration};\n\nuse atuin_client::api_client;\nuse atuin_common::utils::uuid_v7;\nuse atuin_server::{Settings as ServerSettings, launch_with_tcp_listener};\nuse atuin_server_database::DbSettings;\nuse atuin_server_postgres::Postgres;\nuse futures_util::TryFutureExt;\nuse tokio::{net::TcpListener, sync::oneshot, task::JoinHandle};\nuse tracing::{Dispatch, dispatcher};\nuse tracing_subscriber::{EnvFilter, layer::SubscriberExt};\n\npub async fn start_server(path: &str) -> (String, oneshot::Sender<()>, JoinHandle<()>) {\n let formatting_layer = tracing_tree::HierarchicalLayer::default()\n .with_writer(tracing_subscriber::fmt::TestWriter::new())\n .with_indent_lines(true)\n .with_ansi(true)\n .with_targets(true)\n .with_indent_amount(2);\n\n let dispatch: Dispatch = tracing_subscriber::registry()\n .with(formatting_layer)\n .with(EnvFilter::new(\"atuin_server=debug,atuin_client=debug,info\"))\n .into();\n\n let db_uri = env::var(\"ATUIN_DB_URI\")\n .unwrap_or_else(|_| \"postgres://atuin:pass@localhost:5432/atuin\".to_owned());\n\n let server_settings = ServerSettings {\n host: \"127.0.0.1\".to_owned(),\n port: 0,\n path: path.to_owned(),\n sync_v1_enabled: true,\n open_registration: true,\n max_history_length: 8192,\n max_record_size: 1024 * 1024 * 1024,\n page_size: 1100,\n register_webhook_url: None,\n register_webhook_username: String::new(),\n db_settings: DbSettings {\n db_uri: db_uri,\n read_db_uri: None,\n },\n metrics: atuin_server::settings::Metrics::default(),\n fake_version: None,\n };\n\n let (shutdown_tx, shutdown_rx) = tokio::sync::oneshot::channel();\n let listener = TcpListener::bind(\"127.0.0.1:0\").await.unwrap();\n let addr = listener.local_addr().unwrap();\n let server = tokio::spawn(async move {\n let _tracing_guard = dispatcher::set_default(&dispatch);\n\n if let Err(e) = launch_with_tcp_listener::(\n server_settings,\n listener,\n shutdown_rx.unwrap_or_else(|_| ()),\n )\n .await\n {\n tracing::error!(error=?e, \"server error\");\n panic!(\"error running server: {e:?}\");\n }\n });\n\n // let the server come online\n tokio::time::sleep(Duration::from_millis(200)).await;\n\n (format!(\"http://{addr}{path}\"), shutdown_tx, server)\n}\n\npub async fn register_inner<'a>(\n address: &'a str,\n username: &str,\n password: &str,\n) -> api_client::Client<'a> {\n let email = format!(\"{}@example.com\", uuid_v7().as_simple());\n\n // registration works\n let registration_response = api_client::register(address, username, &email, password)\n .await\n .unwrap();\n\n api_client::Client::new(\n address,\n api_client::AuthToken::Token(registration_response.session),\n 5,\n 30,\n )\n .unwrap()\n}\n\n#[allow(dead_code)]\npub async fn login(address: &str, username: String, password: String) -> api_client::Client<'_> {\n // registration works\n let login_response = api_client::login(\n address,\n atuin_common::api::LoginRequest { username, password },\n )\n .await\n .unwrap();\n\n api_client::Client::new(\n address,\n api_client::AuthToken::Token(login_response.session),\n 5,\n 30,\n )\n .unwrap()\n}\n\n#[allow(dead_code)]\npub async fn register(address: &str) -> api_client::Client<'_> {\n let username = uuid_v7().as_simple().to_string();\n let password = uuid_v7().as_simple().to_string();\n register_inner(address, &username, &password).await\n}\n" }, { "path": "crates/atuin/tests/sync.rs", "content": "use atuin_common::{api::AddHistoryRequest, utils::uuid_v7};\nuse time::OffsetDateTime;\n\nmod common;\n\n#[tokio::test]\nasync fn sync() {\n let path = format!(\"/{}\", uuid_v7().as_simple());\n let (address, shutdown, server) = common::start_server(&path).await;\n\n let client = common::register(&address).await;\n let hostname = uuid_v7().as_simple().to_string();\n let now = OffsetDateTime::now_utc();\n\n let data1 = uuid_v7().as_simple().to_string();\n let data2 = uuid_v7().as_simple().to_string();\n\n client\n .post_history(&[\n AddHistoryRequest {\n id: uuid_v7().as_simple().to_string(),\n timestamp: now,\n data: data1.clone(),\n hostname: hostname.clone(),\n },\n AddHistoryRequest {\n id: uuid_v7().as_simple().to_string(),\n timestamp: now,\n data: data2.clone(),\n hostname: hostname.clone(),\n },\n ])\n .await\n .unwrap();\n\n let history = client\n .get_history(OffsetDateTime::UNIX_EPOCH, OffsetDateTime::UNIX_EPOCH, None)\n .await\n .unwrap();\n\n assert_eq!(history.history, vec![data1, data2]);\n\n shutdown.send(()).unwrap();\n server.await.unwrap();\n}\n" }, { "path": "crates/atuin/tests/users.rs", "content": "use atuin_common::utils::uuid_v7;\n\nmod common;\n\n#[tokio::test]\nasync fn registration() {\n let path = format!(\"/{}\", uuid_v7().as_simple());\n let (address, shutdown, server) = common::start_server(&path).await;\n dbg!(&address);\n\n // -- REGISTRATION --\n\n let username = uuid_v7().as_simple().to_string();\n let password = uuid_v7().as_simple().to_string();\n let client = common::register_inner(&address, &username, &password).await;\n\n // the session token works\n let status = client.status().await.unwrap();\n assert_eq!(status.username, username);\n\n // -- LOGIN --\n\n let client = common::login(&address, username.clone(), password).await;\n\n // the session token works\n let status = client.status().await.unwrap();\n assert_eq!(status.username, username);\n\n shutdown.send(()).unwrap();\n server.await.unwrap();\n}\n\n#[tokio::test]\nasync fn change_password() {\n let path = format!(\"/{}\", uuid_v7().as_simple());\n let (address, shutdown, server) = common::start_server(&path).await;\n\n // -- REGISTRATION --\n\n let username = uuid_v7().as_simple().to_string();\n let password = uuid_v7().as_simple().to_string();\n let client = common::register_inner(&address, &username, &password).await;\n\n // the session token works\n let status = client.status().await.unwrap();\n assert_eq!(status.username, username);\n\n // -- PASSWORD CHANGE --\n\n let current_password = password;\n let new_password = uuid_v7().as_simple().to_string();\n let result = client\n .change_password(current_password, new_password.clone())\n .await;\n\n // the password change request succeeded\n assert!(result.is_ok());\n\n // -- LOGIN --\n\n let client = common::login(&address, username.clone(), new_password).await;\n\n // login with new password yields a working token\n let status = client.status().await.unwrap();\n assert_eq!(status.username, username);\n\n shutdown.send(()).unwrap();\n server.await.unwrap();\n}\n\n#[tokio::test]\nasync fn multi_user_test() {\n let path = format!(\"/{}\", uuid_v7().as_simple());\n let (address, shutdown, server) = common::start_server(&path).await;\n dbg!(&address);\n\n // -- REGISTRATION --\n\n let user_one = uuid_v7().as_simple().to_string();\n let password_one = uuid_v7().as_simple().to_string();\n let client_one = common::register_inner(&address, &user_one, &password_one).await;\n\n // the session token works\n let status = client_one.status().await.unwrap();\n assert_eq!(status.username, user_one);\n\n let user_two = uuid_v7().as_simple().to_string();\n let password_two = uuid_v7().as_simple().to_string();\n let client_two = common::register_inner(&address, &user_two, &password_two).await;\n\n // the session token works\n let status = client_two.status().await.unwrap();\n assert_eq!(status.username, user_two);\n\n // check that we can change user one's password, and _this does not affect user two_\n\n let current_password = password_one;\n let new_password = uuid_v7().as_simple().to_string();\n let result = client_one\n .change_password(current_password, new_password.clone())\n .await;\n\n // the password change request succeeded\n assert!(result.is_ok());\n\n // -- LOGIN --\n\n let client_one = common::login(&address, user_one.clone(), new_password).await;\n let client_two = common::login(&address, user_two.clone(), password_two).await;\n\n // login with new password yields a working token\n let status = client_one.status().await.unwrap();\n assert_eq!(status.username, user_one);\n assert_ne!(status.username, user_two);\n\n let status = client_two.status().await.unwrap();\n assert_eq!(status.username, user_two);\n\n shutdown.send(()).unwrap();\n server.await.unwrap();\n}\n" }, { "path": "crates/atuin-ai/Cargo.toml", "content": "[package]\nname = \"atuin-ai\"\nedition = \"2024\"\ndescription = \"AI integration for Atuin CLI\"\n\nrust-version = { workspace = true }\nversion = { workspace = true }\nauthors = { workspace = true }\nlicense = { workspace = true }\nhomepage = { workspace = true }\nrepository = { workspace = true }\n\n# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html\n\n[dependencies]\natuin-client = { workspace = true }\natuin-common = { workspace = true }\ntokio = { workspace = true }\neyre = { workspace = true }\nclap = { workspace = true, features = [\"derive\", \"env\"] }\ntracing = { workspace = true }\ntracing-subscriber = { workspace = true, features = [\n \"ansi\",\n \"fmt\",\n \"registry\",\n \"env-filter\",\n] }\ndirectories = { workspace = true }\ntracing-appender = \"0.2.4\"\nreqwest = { workspace = true }\nserde = { workspace = true }\nserde_json = { workspace = true }\ncrossterm = { workspace = true, features = [\"use-dev-tty\", \"event-stream\"] }\nratatui = { workspace = true, features = [\"unstable-rendered-line-info\"] }\nfutures = \"0.3\"\neventsource-stream = \"0.2\"\npulldown-cmark = \"0.13.0\"\nasync-stream = \"0.3\"\nuuid = { workspace = true }\ntui-textarea-2 = \"0.9.1\"\nunicode-width = \"0.2\"\n\n[dev-dependencies]\npretty_assertions = { workspace = true }\n" }, { "path": "crates/atuin-ai/render-tests.sh", "content": "#!/bin/bash\n# Render all test cases from test-renders.json\n# Usage: ./render-tests.sh [test_name]\n# With no args: renders all tests\n# With arg: renders only matching test (e.g., ./render-tests.sh 05)\n\nset -e\ncd \"$(dirname \"$0\")\"\n\nJSON_FILE=\"test-renders.json\"\nFILTER=\"${1:-}\"\n\n# Build once\ncargo build -p atuin-ai --quiet\n\n# Count tests\nTOTAL=$(jq length \"$JSON_FILE\")\n\nfor i in $(seq 0 $((TOTAL - 1))); do\n NAME=$(jq -r \".[$i].name\" \"$JSON_FILE\")\n DESC=$(jq -r \".[$i].description\" \"$JSON_FILE\")\n STATE=$(jq -c \".[$i].state\" \"$JSON_FILE\")\n\n # Skip if filter provided and doesn't match\n if [[ -n \"$FILTER\" && ! \"$NAME\" =~ $FILTER ]]; then\n continue\n fi\n\n echo \"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\"\n echo \"[$NAME] $DESC\"\n echo \"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\"\n echo \"$STATE\" | cargo run -p atuin-ai --quiet -- debug-render -f plain\n echo \"\"\ndone\n" }, { "path": "crates/atuin-ai/replay-states.sh", "content": "#!/bin/bash\n# Replay state snapshots from a debug state JSONL file\n# Usage: ./replay-states.sh [entry-number]\n# With no entry: renders all frames in sequence (press Enter to advance)\n# With entry number: renders just that frame\n\nset -e\n# cd \"$(dirname \"$0\")\"\n\nSTATE_FILE=\"${1:-}\"\nENTRY_FILTER=\"${2:-}\"\n\nif [[ -z \"$STATE_FILE\" ]]; then\n echo \"Usage: $0 [entry-number]\"\n echo \"\"\n echo \"Examples:\"\n echo \" $0 /tmp/state.jsonl # Interactive replay of all frames\"\n echo \" $0 /tmp/state.jsonl 15 # Show just entry 15\"\n exit 1\nfi\n\nif [[ ! -f \"$STATE_FILE\" ]]; then\n echo \"Error: File not found: $STATE_FILE\"\n exit 1\nfi\n\n# Build once\ncargo build -p atuin --quiet\n\n# Count entries\nTOTAL=$(wc -l < \"$STATE_FILE\" | tr -d ' ')\n\nif [[ -n \"$ENTRY_FILTER\" ]]; then\n # Show single entry\n LINE=$(sed -n \"${ENTRY_FILTER}p\" \"$STATE_FILE\")\n if [[ -z \"$LINE\" ]]; then\n echo \"Error: Entry $ENTRY_FILTER not found (file has $TOTAL entries)\"\n exit 1\n fi\n\n ENTRY=$(echo \"$LINE\" | jq -r '.entry')\n LABEL=$(echo \"$LINE\" | jq -r '.label')\n STATE=$(echo \"$LINE\" | jq -c '.state')\n\n echo \"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\"\n echo \"[$ENTRY/$TOTAL] $LABEL\"\n echo \"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\"\n echo \"$STATE\" | cargo run -p atuin --quiet -- ai debug-render -f ansi\nelse\n # Interactive replay\n echo \"Replaying $TOTAL frames from $STATE_FILE\"\n echo \"Press Enter to advance, 'q' to quit, or number+Enter to jump\"\n echo \"\"\n\n CURRENT=1\n while [[ $CURRENT -le $TOTAL ]]; do\n LINE=$(sed -n \"${CURRENT}p\" \"$STATE_FILE\")\n ENTRY=$(echo \"$LINE\" | jq -r '.entry')\n LABEL=$(echo \"$LINE\" | jq -r '.label')\n STATE=$(echo \"$LINE\" | jq -c '.state')\n\n clear\n echo \"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\"\n echo \"[$CURRENT/$TOTAL] $LABEL\"\n echo \"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\"\n echo \"$STATE\" | cargo run -p atuin --quiet -- ai debug-render -f ansi\n echo \"\"\n echo \"[Enter: next] [p: prev] [number: jump] [s: show state JSON] [q: quit]\"\n\n read -r INPUT\n case \"$INPUT\" in\n q|Q)\n break\n ;;\n p|P)\n if [[ $CURRENT -gt 1 ]]; then\n CURRENT=$((CURRENT - 1))\n fi\n ;;\n s|S)\n echo \"\"\n echo \"State JSON:\"\n echo \"$STATE\" | jq .\n echo \"\"\n echo \"Press Enter to continue...\"\n read -r\n ;;\n ''|' ')\n CURRENT=$((CURRENT + 1))\n ;;\n *[0-9]*)\n if [[ \"$INPUT\" =~ ^[0-9]+$ ]] && [[ \"$INPUT\" -ge 1 ]] && [[ \"$INPUT\" -le $TOTAL ]]; then\n CURRENT=$INPUT\n else\n echo \"Invalid entry number (1-$TOTAL)\"\n sleep 1\n fi\n ;;\n esac\n done\nfi\n" }, { "path": "crates/atuin-ai/src/commands/debug_render.rs", "content": "//! Debug render command for TUI development\n//!\n//! Takes JSON state as input and outputs a single rendered frame as text.\n//! Useful for debugging view model derivation and rendering without running the full TUI.\n\nuse eyre::{Context, Result};\nuse ratatui::{Terminal, backend::TestBackend};\nuse serde::Deserialize;\nuse std::io::{self, Read};\nuse std::time::Instant;\n\nuse crate::tui::{\n render::{RenderContext, render},\n state::{AppMode, AppState, ConversationEvent, StreamingStatus},\n view_model::Blocks,\n};\n\n/// JSON input format for debug rendering\n#[derive(Debug, Deserialize)]\npub struct DebugInput {\n /// Conversation events in API format\n pub events: Vec,\n /// Current mode: \"Input\", \"Generating\", \"Streaming\", \"Review\", \"Error\"\n #[serde(default = \"default_mode\")]\n pub mode: String,\n /// Text being streamed (for Streaming mode)\n #[serde(default)]\n pub streaming_text: String,\n /// Current input buffer\n #[serde(default)]\n pub input: String,\n /// Cursor position\n #[serde(default)]\n pub cursor_pos: usize,\n /// Spinner frame (0-3)\n #[serde(default)]\n pub spinner_frame: usize,\n /// Error message\n #[serde(default)]\n pub error: Option,\n /// Session ID from server\n #[serde(default)]\n pub session_id: Option,\n /// Streaming status\n #[serde(default)]\n pub streaming_status: Option,\n /// Whether current turn was interrupted\n #[serde(default)]\n pub was_interrupted: bool,\n /// Terminal width for rendering\n #[serde(default = \"default_width\")]\n pub width: u16,\n /// Terminal height for rendering\n #[serde(default = \"default_height\")]\n pub height: u16,\n}\n\nfn default_mode() -> String {\n \"Review\".to_string()\n}\n\nfn default_width() -> u16 {\n 80\n}\n\nfn default_height() -> u16 {\n // Default to a reasonable height; state files include calculated height\n 50\n}\n\n/// Event input matching the API protocol format\n#[derive(Debug, Clone, Deserialize)]\n#[serde(tag = \"type\", rename_all = \"snake_case\")]\npub enum EventInput {\n UserMessage {\n content: String,\n },\n Text {\n content: String,\n },\n ToolCall {\n id: String,\n name: String,\n input: serde_json::Value,\n },\n ToolResult {\n tool_use_id: String,\n content: String,\n #[serde(default)]\n is_error: bool,\n },\n}\n\nimpl From for ConversationEvent {\n fn from(input: EventInput) -> Self {\n match input {\n EventInput::UserMessage { content } => ConversationEvent::UserMessage { content },\n EventInput::Text { content } => ConversationEvent::Text { content },\n EventInput::ToolCall { id, name, input } => {\n ConversationEvent::ToolCall { id, name, input }\n }\n EventInput::ToolResult {\n tool_use_id,\n content,\n is_error,\n } => ConversationEvent::ToolResult {\n tool_use_id,\n content,\n is_error,\n },\n }\n }\n}\n\nimpl DebugInput {\n /// Parse JSON from string\n pub fn from_json(json: &str) -> Result {\n serde_json::from_str(json).context(\"Failed to parse debug input JSON\")\n }\n\n /// Convert to AppState\n pub fn to_state(&self) -> AppState {\n let mode = match self.mode.as_str() {\n \"Input\" => AppMode::Input,\n \"Generating\" => AppMode::Generating,\n \"Streaming\" => AppMode::Streaming,\n \"Review\" => AppMode::Review,\n \"Error\" => AppMode::Error,\n _ => AppMode::Review,\n };\n\n let events: Vec = self.events.iter().cloned().map(Into::into).collect();\n\n let streaming_status = self\n .streaming_status\n .as_ref()\n .map(|s| StreamingStatus::from_status_str(s));\n\n // Create textarea from input and set cursor position\n let mut textarea = tui_textarea::TextArea::from(self.input.lines());\n // Disable underline on cursor line\n textarea.set_cursor_line_style(ratatui::style::Style::default());\n // Enable word wrapping\n textarea.set_wrap_mode(tui_textarea::WrapMode::Word);\n // Note: cursor_pos from old format is character-based; new format has row/col\n // For compatibility, just move to end if we have text\n if !self.input.is_empty() {\n textarea.move_cursor(tui_textarea::CursorMove::End);\n }\n\n AppState {\n mode,\n events,\n streaming_text: self.streaming_text.clone(),\n textarea,\n error: self.error.clone(),\n should_exit: false,\n exit_action: None,\n session_id: self.session_id.clone(),\n streaming_status,\n was_interrupted: self.was_interrupted,\n spinner_frame: self.spinner_frame,\n last_spinner_tick: Instant::now(),\n streaming_started: None,\n confirmation_pending: false,\n }\n }\n}\n\n/// Output format options\n#[derive(Debug, Clone, Copy, Default)]\npub enum OutputFormat {\n /// Raw terminal output (ANSI)\n #[default]\n Ansi,\n /// Plain text (strips ANSI codes)\n Plain,\n /// JSON with blocks structure\n Json,\n}\n\n/// Run the debug render command\npub async fn run(input_file: Option, format: OutputFormat) -> Result<()> {\n // Read input JSON\n let json = if let Some(path) = input_file {\n std::fs::read_to_string(&path).context(format!(\"Failed to read input file: {}\", path))?\n } else {\n let mut buffer = String::new();\n io::stdin()\n .read_to_string(&mut buffer)\n .context(\"Failed to read from stdin\")?;\n buffer\n };\n\n let debug_input = DebugInput::from_json(&json)?;\n let state = debug_input.to_state();\n\n match format {\n OutputFormat::Json => {\n // Output the derived blocks as JSON\n let blocks = Blocks::from_state(&state);\n println!(\n \"{}\",\n serde_json::to_string_pretty(&blocks_to_json(&blocks))?\n );\n }\n OutputFormat::Plain | OutputFormat::Ansi => {\n // Render to a test backend\n let backend = TestBackend::new(debug_input.width, debug_input.height);\n let mut terminal = Terminal::new(backend)?;\n\n // Load default theme\n let settings = atuin_client::settings::Settings::new()?;\n let mut theme_manager = atuin_client::theme::ThemeManager::new(None, None);\n let theme = theme_manager.load_theme(&settings.theme.name, None);\n\n let ctx = RenderContext {\n theme,\n anchor_col: 0,\n textarea: Some(&state.textarea),\n max_height: debug_input.height,\n popup_mode: false,\n render_above: false,\n };\n\n terminal.draw(|frame| {\n render(frame, &state, &ctx);\n })?;\n\n // Get buffer content\n let buffer = terminal.backend().buffer();\n let output = buffer_to_string(buffer, matches!(format, OutputFormat::Plain));\n print!(\"{}\", output);\n }\n }\n\n Ok(())\n}\n\n/// Convert blocks to JSON for debugging\nfn blocks_to_json(blocks: &Blocks) -> serde_json::Value {\n serde_json::json!({\n \"count\": blocks.items.len(),\n \"blocks\": blocks.items.iter().map(|block| {\n serde_json::json!({\n \"separator_above\": block.separator_above,\n \"title\": block.title,\n \"content\": block.content.iter().map(content_to_json).collect::>()\n })\n }).collect::>(),\n \"status_bar\": blocks.status_bar.as_ref().map(|sb| serde_json::json!({\n \"frame\": sb.frame,\n \"text\": sb.text\n }))\n })\n}\n\nfn content_to_json(content: &crate::tui::view_model::Content) -> serde_json::Value {\n use crate::tui::view_model::Content;\n match content {\n Content::Input {\n text,\n active,\n cursor_pos,\n } => serde_json::json!({\n \"type\": \"Input\",\n \"text\": text,\n \"active\": active,\n \"cursor_pos\": cursor_pos\n }),\n Content::Command { text, faded } => serde_json::json!({\n \"type\": \"Command\",\n \"text\": text,\n \"faded\": faded\n }),\n Content::Text { markdown } => serde_json::json!({\n \"type\": \"Text\",\n \"markdown\": markdown\n }),\n Content::Error { message } => serde_json::json!({\n \"type\": \"Error\",\n \"message\": message\n }),\n Content::Warning {\n kind,\n text,\n pending_confirm,\n } => serde_json::json!({\n \"type\": \"Warning\",\n \"kind\": format!(\"{:?}\", kind),\n \"text\": text,\n \"pending_confirm\": pending_confirm\n }),\n Content::Spinner { frame, status_text } => serde_json::json!({\n \"type\": \"Spinner\",\n \"frame\": frame,\n \"status_text\": status_text\n }),\n Content::ToolStatus {\n completed_count,\n current_label,\n frame,\n } => serde_json::json!({\n \"type\": \"ToolStatus\",\n \"completed_count\": completed_count,\n \"current_label\": current_label,\n \"frame\": frame\n }),\n }\n}\n\n/// Convert ratatui buffer to string\nfn buffer_to_string(buffer: &ratatui::buffer::Buffer, strip_ansi: bool) -> String {\n let area = buffer.area;\n let mut output = String::new();\n\n for y in 0..area.height {\n for x in 0..area.width {\n let cell = &buffer[(x, y)];\n if strip_ansi {\n output.push_str(cell.symbol());\n } else {\n // Include ANSI styling\n let fg = cell.fg;\n let bg = cell.bg;\n let mods = cell.modifier;\n\n // Simple ANSI encoding\n if fg != ratatui::style::Color::Reset\n || bg != ratatui::style::Color::Reset\n || !mods.is_empty()\n {\n output.push_str(\"\\x1b[\");\n let mut first = true;\n\n if mods.contains(ratatui::style::Modifier::BOLD) {\n output.push('1');\n first = false;\n }\n if mods.contains(ratatui::style::Modifier::DIM) {\n if !first {\n output.push(';');\n }\n output.push('2');\n first = false;\n }\n if mods.contains(ratatui::style::Modifier::REVERSED) {\n if !first {\n output.push(';');\n }\n output.push('7');\n first = false;\n }\n if mods.contains(ratatui::style::Modifier::UNDERLINED) {\n if !first {\n output.push(';');\n }\n output.push('4');\n first = false;\n }\n\n if let Some(code) = color_to_ansi(fg, true) {\n if !first {\n output.push(';');\n }\n output.push_str(&code);\n first = false;\n }\n\n if let Some(code) = color_to_ansi(bg, false) {\n if !first {\n output.push(';');\n }\n output.push_str(&code);\n }\n\n output.push('m');\n }\n\n output.push_str(cell.symbol());\n\n if fg != ratatui::style::Color::Reset\n || bg != ratatui::style::Color::Reset\n || !mods.is_empty()\n {\n output.push_str(\"\\x1b[0m\");\n }\n }\n }\n output.push('\\n');\n }\n\n output\n}\n\nfn color_to_ansi(color: ratatui::style::Color, foreground: bool) -> Option {\n use ratatui::style::Color;\n let base = if foreground { 30 } else { 40 };\n\n match color {\n Color::Reset => None,\n Color::Black => Some((base).to_string()),\n Color::Red => Some((base + 1).to_string()),\n Color::Green => Some((base + 2).to_string()),\n Color::Yellow => Some((base + 3).to_string()),\n Color::Blue => Some((base + 4).to_string()),\n Color::Magenta => Some((base + 5).to_string()),\n Color::Cyan => Some((base + 6).to_string()),\n Color::Gray | Color::White => Some((base + 7).to_string()),\n Color::DarkGray => Some((base + 60).to_string()),\n Color::LightRed => Some((base + 61).to_string()),\n Color::LightGreen => Some((base + 62).to_string()),\n Color::LightYellow => Some((base + 63).to_string()),\n Color::LightBlue => Some((base + 64).to_string()),\n Color::LightMagenta => Some((base + 65).to_string()),\n Color::LightCyan => Some((base + 66).to_string()),\n Color::Indexed(i) => Some(format!(\"{}8;5;{}\", if foreground { 3 } else { 4 }, i)),\n Color::Rgb(r, g, b) => Some(format!(\n \"{}8;2;{};{};{}\",\n if foreground { 3 } else { 4 },\n r,\n g,\n b\n )),\n }\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_parse_simple_input() {\n let json = r#\"{\n \"events\": [\n {\"type\": \"user_message\", \"content\": \"list files\"},\n {\"type\": \"tool_call\", \"id\": \"123\", \"name\": \"suggest_command\", \"input\": {\"command\": \"ls -la\"}}\n ],\n \"mode\": \"Review\"\n }\"#;\n\n let input = DebugInput::from_json(json).unwrap();\n assert_eq!(input.events.len(), 2);\n assert_eq!(input.mode, \"Review\");\n\n let state = input.to_state();\n assert_eq!(state.events.len(), 2);\n assert_eq!(state.mode, AppMode::Review);\n }\n\n #[test]\n fn test_parse_streaming_state() {\n let json = r#\"{\n \"events\": [\n {\"type\": \"user_message\", \"content\": \"explain flags\"}\n ],\n \"mode\": \"Streaming\",\n \"streaming_text\": \"The -l flag means...\"\n }\"#;\n\n let input = DebugInput::from_json(json).unwrap();\n let state = input.to_state();\n assert_eq!(state.mode, AppMode::Streaming);\n assert_eq!(state.streaming_text, \"The -l flag means...\");\n }\n}\n" }, { "path": "crates/atuin-ai/src/commands/init.rs", "content": "use crate::commands::detect_shell;\n\npub async fn run(shell: String) -> eyre::Result<()> {\n let integration = match shell.as_str() {\n \"zsh\" => generate_zsh_integration(),\n \"bash\" => generate_bash_integration(),\n \"fish\" => generate_fish_integration(),\n \"auto\" => generate_auto_integration()?,\n _ => eyre::bail!(\"Unsupported shell: {}\", shell),\n };\n\n println!(\"{}\", integration);\n Ok(())\n}\n\nfn generate_auto_integration() -> eyre::Result<&'static str> {\n let shell = detect_shell();\n match shell.as_deref() {\n Some(\"zsh\") => Ok(generate_zsh_integration()),\n Some(\"bash\") => Ok(generate_bash_integration()),\n Some(\"fish\") => Ok(generate_fish_integration()),\n Some(s) => eyre::bail!(\"Unsupported shell: {}\", s),\n None => eyre::bail!(\"Could not detect shell\"),\n }\n}\n\n/// Generate the zsh integration function - pure function for easy testing\npub fn generate_zsh_integration() -> &'static str {\n r#\"\n# TUI uses an alternate screen, so no explicit cleanup is needed.\n_atuin_ai_cleanup() {\n true\n}\n\n# Question mark at start of line - natural language mode.\n# Named with 'self-' prefix so bracketed-paste-magic activates it during\n# paste, allowing url-quote-magic to escape ? in pasted URLs via self-insert.\nself-atuin-ai-question-mark() {\n # If buffer is empty or just contains '?', trigger natural language mode\n if [[ -z \"$BUFFER\" || \"$BUFFER\" == \"?\" ]]; then\n BUFFER=\"\"\n local output\n output=$(atuin ai inline --hook 3>&1 1>&2 2>&3)\n\n # Clean up the inline viewport\n _atuin_ai_cleanup\n\n if [[ $output == __atuin_ai_print__:* ]]; then\n zle -I\n echo \"${output#__atuin_ai_print__:}\"\n elif [[ $output == __atuin_ai_cancel__ ]]; then\n zle reset-prompt\n elif [[ $output == __atuin_ai_execute__:* ]]; then\n RBUFFER=\"\"\n LBUFFER=${output#__atuin_ai_execute__:}\n zle reset-prompt\n zle accept-line\n elif [[ $output == __atuin_ai_insert__:* ]]; then\n RBUFFER=\"\"\n LBUFFER=${output#__atuin_ai_insert__:}\n zle reset-prompt\n elif [[ -n $output ]]; then\n RBUFFER=\"\"\n LBUFFER=$output\n zle reset-prompt\n else\n zle reset-prompt\n fi\n else\n zle self-insert\n fi\n}\n\n# Set up keybindings\nzle -N self-atuin-ai-question-mark\nbindkey '?' self-atuin-ai-question-mark # Question mark\n\"#\n .trim()\n}\n\n/// Generate the bash integration function - pure function for easy testing\npub fn generate_bash_integration() -> &'static str {\n r#\"\n# Question mark at start of line - natural language mode\n_atuin_ai_question_mark() {\n # If buffer is empty or just contains '?', trigger natural language mode\n if [[ -z \"$READLINE_LINE\" || \"$READLINE_LINE\" == \"?\" ]]; then\n READLINE_LINE=\"\"\n READLINE_POINT=0\n\n local output\n output=$(atuin ai inline --hook 3>&1 1>&2 2>&3)\n\n if [[ $output == __atuin_ai_print__:* ]]; then\n echo \"${output#__atuin_ai_print__:}\"\n READLINE_LINE=\"\"\n READLINE_POINT=0\n elif [[ $output == __atuin_ai_cancel__ ]]; then\n READLINE_LINE=\"\"\n READLINE_POINT=0\n elif [[ $output == __atuin_ai_execute__:* ]]; then\n # Execute the command immediately\n READLINE_LINE=${output#__atuin_ai_execute__:}\n READLINE_POINT=${#READLINE_LINE}\n # Note: We can't directly execute in bash bind -x, but we can\n # use a workaround by binding to a macro that accepts the line\n bind '\"\\C-x\\C-a\": accept-line'\n bind -x '\"\\C-x\\C-e\": _atuin_ai_question_mark'\n elif [[ $output == __atuin_ai_insert__:* ]]; then\n # Insert the command for editing\n READLINE_LINE=${output#__atuin_ai_insert__:}\n READLINE_POINT=${#READLINE_LINE}\n elif [[ -n $output ]]; then\n # Default: insert for editing\n READLINE_LINE=$output\n READLINE_POINT=${#READLINE_LINE}\n fi\n else\n # Not at empty prompt, just insert the question mark\n READLINE_LINE=\"${READLINE_LINE:0:READLINE_POINT}?${READLINE_LINE:READLINE_POINT}\"\n ((READLINE_POINT++))\n fi\n}\n\n# Set up keybindings\n# Bash requires special handling: we use bind -x for the function,\n# but need a two-step approach for execute mode\n__atuin_ai_accept_line=\"\"\n\n_atuin_ai_question_mark_wrapper() {\n _atuin_ai_question_mark\n if [[ -n \"$__atuin_ai_accept_line\" ]]; then\n __atuin_ai_accept_line=\"\"\n fi\n}\n\nbind -x '\"?\": _atuin_ai_question_mark'\n\"#\n .trim()\n}\n\n/// Generate the fish integration function - pure function for easy testing\npub fn generate_fish_integration() -> &'static str {\n r#\"\n# Question mark at start of line - natural language mode\nfunction _atuin_ai_question_mark\n set -l buf (commandline -b)\n\n # If buffer is empty or just contains '?', trigger natural language mode\n if test -z \"$buf\" -o \"$buf\" = \"?\"\n commandline -r \"\"\n\n # Run atuin ai inline, swapping stdout and stderr\n set -l output (atuin ai inline --hook 3>&1 1>&2 2>&3 | string collect)\n\n if string match --quiet '__atuin_ai_print__:*' \"$output\"\n echo (string replace \"__atuin_ai_print__:\" \"\" -- \"$output\" | string collect)\n commandline -f repaint\n else if test \"$output\" = \"__atuin_ai_cancel__\"\n commandline -f repaint\n else if string match --quiet '__atuin_ai_execute__:*' \"$output\"\n # Execute the command immediately\n set -l cmd (string replace \"__atuin_ai_execute__:\" \"\" -- \"$output\" | string collect)\n commandline -r \"$cmd\"\n commandline -f repaint\n commandline -f execute\n else if string match --quiet '__atuin_ai_insert__:*' \"$output\"\n # Insert the command for editing\n set -l cmd (string replace \"__atuin_ai_insert__:\" \"\" -- \"$output\" | string collect)\n commandline -r \"$cmd\"\n commandline -f repaint\n else if test -n \"$output\"\n # Default: insert for editing\n commandline -r \"$output\"\n commandline -f repaint\n else\n commandline -f repaint\n end\n else\n # Not at empty prompt, just insert the question mark\n commandline -i \"?\"\n end\nend\n\n# Set up keybindings\nbind \"?\" _atuin_ai_question_mark\n\"#\n .trim()\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_generate_zsh_integration() {\n let result = generate_zsh_integration();\n assert!(result.contains(\"self-atuin-ai-question-mark\"));\n assert!(result.contains(\"bindkey\"));\n assert!(result.contains(\"atuin ai inline --hook\"));\n assert!(result.contains(\"__atuin_ai_print__\"));\n assert!(result.contains(\"__atuin_ai_cancel__\"));\n assert!(result.contains(\"__atuin_ai_execute__\"));\n assert!(result.contains(\"__atuin_ai_insert__\"));\n assert!(result.contains(\"zle self-insert\"));\n }\n\n #[test]\n fn test_generate_bash_integration() {\n let result = generate_bash_integration();\n assert!(result.contains(\"_atuin_ai_question_mark\"));\n assert!(result.contains(\"bind\"));\n assert!(result.contains(\"READLINE_LINE\"));\n assert!(result.contains(\"atuin ai inline --hook\"));\n assert!(result.contains(\"__atuin_ai_print__\"));\n assert!(result.contains(\"__atuin_ai_cancel__\"));\n assert!(result.contains(\"__atuin_ai_execute__\"));\n assert!(result.contains(\"__atuin_ai_insert__\"));\n }\n\n #[test]\n fn test_generate_fish_integration() {\n let result = generate_fish_integration();\n assert!(result.contains(\"_atuin_ai_question_mark\"));\n assert!(result.contains(\"bind\"));\n assert!(result.contains(\"commandline\"));\n assert!(result.contains(\"atuin ai inline --hook\"));\n assert!(result.contains(\"__atuin_ai_print__\"));\n assert!(result.contains(\"__atuin_ai_cancel__\"));\n assert!(result.contains(\"__atuin_ai_execute__\"));\n assert!(result.contains(\"__atuin_ai_insert__\"));\n }\n}\n" }, { "path": "crates/atuin-ai/src/commands/inline.rs", "content": "use crate::commands::detect_shell;\nuse crate::tui::render::render;\nuse crate::tui::{\n App, AppEvent, AppMode, ConversationEvent, EventLoop, ExitAction, RenderContext, TerminalGuard,\n calculate_needed_height, install_panic_hook,\n};\nuse atuin_client::distro::detect_linux_distribution;\nuse atuin_client::theme::ThemeManager;\nuse atuin_common::tls::ensure_crypto_provider;\nuse crossterm::{\n event::{self, Event, KeyCode},\n terminal::{disable_raw_mode, enable_raw_mode},\n};\nuse eventsource_stream::Eventsource;\nuse eyre::{Context as _, Result, bail};\nuse futures::StreamExt;\nuse reqwest::Url;\nuse std::io::Write;\nuse tracing::{debug, error, info, trace};\n\npub async fn run(\n initial_command: Option,\n api_endpoint: Option,\n api_token: Option,\n keep_output: bool,\n debug_state_file: Option,\n settings: &atuin_client::settings::Settings,\n output_for_hook: bool,\n) -> Result<()> {\n if !settings.ai.enabled {\n emit_shell_result(\n Action::Print(\n \"Atuin AI is not enabled. Please enable it in your settings or run `atuin setup`.\"\n .to_string(),\n ),\n output_for_hook,\n );\n return Ok(());\n }\n\n // Install panic hook once at entry point to ensure terminal restoration\n install_panic_hook();\n\n // Token and endpoint priority:\n // 1. Command line arguments/environment variables\n // 2. Settings file\n // 3. Default\n let endpoint = api_endpoint.as_deref().unwrap_or(\n settings\n .ai\n .endpoint\n .as_deref()\n .unwrap_or(\"https://hub.atuin.sh\"),\n );\n let api_token = api_token.as_deref().or(settings.ai.api_token.as_deref());\n\n let token = if let Some(token) = &api_token {\n token.to_string()\n } else {\n // ensure_hub_session will authenticate against settings.active_hub_endpoint().unwrap_or_default(),\n // which is the default Hub endpoint if no endpoint is provided\n //\n // TODO[mkt]: Atuin AI and the Hub sync endpoint are too tightly coupled;\n // current setup means that Hub endpoint controls auth while AI endpoint controls AI conversations\n ensure_hub_session(settings).await?\n };\n\n let action = run_inline_tui(\n endpoint.to_string(),\n token,\n initial_command,\n keep_output,\n debug_state_file,\n settings,\n )\n .await?;\n emit_shell_result(action, output_for_hook);\n\n Ok(())\n}\n\nasync fn ensure_hub_session(settings: &atuin_client::settings::Settings) -> Result {\n if let Some(token) = atuin_client::hub::get_session_token().await? {\n debug!(\"Found Hub session, using existing token\");\n return Ok(token);\n }\n\n let hub_address = settings.active_hub_endpoint().unwrap_or_default();\n\n let will_sync = settings.is_hub_sync();\n\n info!(\"No Hub session found, prompting for authentication\");\n\n println!(\"Atuin AI requires authenticating with Atuin Hub.\");\n if will_sync {\n println!(\n \"Once logged in, your shell history will be synchronized via Atuin Hub if auto_sync is enabled or when manually syncing.\"\n )\n }\n println!(\n \"If you have an existing Atuin sync account, you can log in with your existing credentials.\"\n );\n println!(\"Press enter to begin (or esc to cancel).\");\n if !wait_for_login_confirmation()? {\n bail!(\"authentication canceled\");\n }\n\n debug!(\"Starting Atuin Hub authentication...\");\n\n println!(\"Authenticating with Atuin Hub...\");\n let session = atuin_client::hub::HubAuthSession::start(hub_address.as_ref()).await?;\n println!(\"Open this URL to continue:\");\n println!(\"{}\", session.auth_url);\n\n let token = session\n .wait_for_completion(\n atuin_client::hub::DEFAULT_AUTH_TIMEOUT,\n atuin_client::hub::DEFAULT_POLL_INTERVAL,\n )\n .await?;\n\n info!(\"Authentication complete, saving session token\");\n\n atuin_client::hub::save_session(&token).await?;\n\n // Silently attempt to link CLI account to Hub if one exists\n // This enables unified auth - users can use their Hub token for sync\n if let Ok(meta) = atuin_client::settings::Settings::meta_store().await\n && let Ok(Some(cli_token)) = meta.session_token().await\n {\n debug!(\"CLI session found, attempting to link accounts\");\n if let Err(e) = atuin_client::hub::link_account(hub_address.as_ref(), &cli_token).await {\n // Don't fail AI flow if linking fails - it's not critical\n debug!(\"Could not link CLI account to Hub: {}\", e);\n } else {\n info!(\"Successfully linked CLI account to Hub\");\n }\n }\n\n Ok(token)\n}\n\n/// SSE event received from chat endpoint\n#[derive(Debug, Clone)]\nenum ChatStreamEvent {\n /// Text chunk to display\n TextChunk(String),\n /// Tool call event (need to echo back, may contain suggest_command)\n ToolCall {\n id: String,\n name: String,\n input: serde_json::Value,\n },\n /// Tool result from server-side execution\n ToolResult {\n tool_use_id: String,\n content: String,\n is_error: bool,\n },\n /// Status update from server\n Status(String),\n /// Stream complete\n Done { session_id: String },\n /// Error from server\n Error(String),\n}\n\nfn create_chat_stream(\n hub_address: String,\n token: String,\n session_id: Option,\n messages: Vec,\n settings: &atuin_client::settings::Settings,\n) -> std::pin::Pin> + Send>> {\n let send_cwd = settings.ai.send_cwd;\n\n Box::pin(async_stream::stream! {\n ensure_crypto_provider();\n let endpoint = match hub_url(&hub_address, \"/api/cli/chat\") {\n Ok(url) => url,\n Err(e) => {\n yield Err(e);\n return;\n }\n };\n\n debug!(\"Sending SSE request to {endpoint}\");\n\n let os = detect_os();\n let shell = detect_shell();\n\n let mut context = serde_json::json!({\n \"os\": os,\n \"shell\": shell,\n \"pwd\": if send_cwd { std::env::current_dir()\n .ok()\n .map(|path| path.to_string_lossy().into_owned()) } else { None },\n });\n\n if os == \"linux\" {\n context[\"distro\"] = serde_json::json!(detect_linux_distribution());\n }\n\n // Build request body\n let mut request_body = serde_json::json!({\n \"messages\": messages,\n \"context\": context,\n });\n\n // Include session_id only if present (not on first request)\n if let Some(ref sid) = session_id {\n trace!(\"Including session_id in request: {sid}\");\n request_body[\"session_id\"] = serde_json::json!(sid);\n }\n\n\n let client = reqwest::Client::new();\n let response = match client\n .post(endpoint.clone())\n .header(\"Accept\", \"text/event-stream\")\n .bearer_auth(&token)\n .json(&request_body)\n .send()\n .await\n {\n Ok(resp) => resp,\n Err(e) => {\n yield Err(eyre::eyre!(\"Failed to send SSE request: {}\", e));\n return;\n }\n };\n\n let status = response.status();\n if status == reqwest::StatusCode::UNAUTHORIZED {\n // Clear saved session on auth error\n error!(\"SSE request failed with status: {status}, clearing session\");\n let _ = atuin_client::hub::delete_session().await;\n yield Err(eyre::eyre!(\"Hub session expired. Re-run to authenticate again.\"));\n return;\n }\n if !status.is_success() {\n let body = response.text().await.unwrap_or_default();\n error!(\"SSE request failed ({}): {}\", status, body);\n yield Err(eyre::eyre!(\"SSE request failed ({}): {}\", status, body));\n return;\n }\n\n let byte_stream = response.bytes_stream();\n let mut stream = byte_stream.eventsource();\n\n while let Some(event) = stream.next().await {\n match event {\n Ok(sse_event) => {\n let event_type = sse_event.event.as_str();\n let data = sse_event.data.clone();\n\n debug!(event_type = %event_type, \"SSE event received\");\n\n match event_type {\n \"text\" => {\n if let Ok(json) = serde_json::from_str::(&data)\n && let Some(content) = json.get(\"content\").and_then(|v| v.as_str())\n {\n yield Ok(ChatStreamEvent::TextChunk(content.to_string()));\n }\n }\n \"tool_call\" => {\n if let Ok(json) = serde_json::from_str::(&data) {\n let id = json.get(\"id\").and_then(|v| v.as_str()).unwrap_or(\"\").to_string();\n let name = json.get(\"name\").and_then(|v| v.as_str()).unwrap_or(\"\").to_string();\n let input = json.get(\"input\").cloned().unwrap_or(serde_json::json!({}));\n yield Ok(ChatStreamEvent::ToolCall { id, name, input });\n }\n }\n \"tool_result\" => {\n if let Ok(json) = serde_json::from_str::(&data) {\n let tool_use_id = json.get(\"tool_use_id\").and_then(|v| v.as_str()).unwrap_or(\"\").to_string();\n let content = json.get(\"content\").and_then(|v| v.as_str()).unwrap_or(\"\").to_string();\n let is_error = json.get(\"is_error\").and_then(|v| v.as_bool()).unwrap_or(false);\n yield Ok(ChatStreamEvent::ToolResult { tool_use_id, content, is_error });\n }\n }\n \"status\" => {\n if let Ok(json) = serde_json::from_str::(&data)\n && let Some(state) = json.get(\"state\").and_then(|v| v.as_str())\n {\n yield Ok(ChatStreamEvent::Status(state.to_string()));\n }\n }\n \"done\" => {\n if let Ok(json) = serde_json::from_str::(&data) {\n let session_id = json.get(\"session_id\")\n .and_then(|v| v.as_str())\n .unwrap_or(\"\")\n .to_string();\n yield Ok(ChatStreamEvent::Done { session_id });\n } else {\n yield Ok(ChatStreamEvent::Done { session_id: String::new() });\n }\n break;\n }\n \"error\" => {\n if let Ok(json) = serde_json::from_str::(&data) {\n let message = json.get(\"message\").and_then(|v| v.as_str()).unwrap_or(\"Unknown error\").to_string();\n error!(\"SSE error: {}\", message);\n yield Ok(ChatStreamEvent::Error(message));\n } else {\n error!(\"SSE error: {}\", data);\n yield Ok(ChatStreamEvent::Error(data));\n }\n break;\n }\n _ => {\n // Unknown event type, ignore\n }\n }\n }\n Err(e) => {\n yield Err(eyre::eyre!(\"SSE error: {}\", e));\n break;\n }\n }\n }\n })\n}\n\nfn hub_url(base: &str, path: &str) -> Result {\n let base_with_slash = if base.ends_with('/') {\n base.to_string()\n } else {\n format!(\"{base}/\")\n };\n let stripped = path.strip_prefix('/').unwrap_or(path);\n Url::parse(&base_with_slash)?\n .join(stripped)\n .context(\"failed to build hub URL\")\n}\n\nfn detect_os() -> String {\n match std::env::consts::OS {\n \"macos\" => \"macos\".to_string(),\n \"linux\" => \"linux\".to_string(),\n \"windows\" => \"windows\".to_string(),\n other => format!(\"Other: {other}\"),\n }\n}\n\n#[derive(Clone)]\nenum Action {\n Execute(String),\n Insert(String),\n Print(String),\n Cancel,\n}\n\n/// Serialize AppState to JSON for debug logging\nfn state_to_json(state: &crate::tui::AppState) -> serde_json::Value {\n let events: Vec = state.events.iter().map(|e| e.to_json()).collect();\n\n let mode = match state.mode {\n AppMode::Input => \"Input\",\n AppMode::Generating => \"Generating\",\n AppMode::Streaming => \"Streaming\",\n AppMode::Review => \"Review\",\n AppMode::Error => \"Error\",\n };\n\n // Get input and cursor from textarea\n let input = state.input();\n let cursor = state.textarea.cursor();\n\n let mut json = serde_json::json!({\n \"events\": events,\n \"mode\": mode,\n \"input\": input,\n \"cursor_row\": cursor.0,\n \"cursor_col\": cursor.1,\n \"spinner_frame\": state.spinner_frame,\n \"confirmation_pending\": state.confirmation_pending,\n });\n\n // Add streaming fields if in streaming mode\n if !state.streaming_text.is_empty() {\n json[\"streaming_text\"] = serde_json::json!(state.streaming_text);\n }\n if let Some(ref status) = state.streaming_status {\n json[\"streaming_status\"] = serde_json::json!(status.display_text());\n }\n if let Some(ref err) = state.error {\n json[\"error\"] = serde_json::json!(err);\n }\n\n json\n}\n\n/// Debug logger that writes state changes to a file\nstruct DebugStateLogger {\n file: std::fs::File,\n entry_count: usize,\n width: u16,\n}\n\nimpl DebugStateLogger {\n fn new(path: &str) -> Result {\n let file = std::fs::File::create(path)\n .with_context(|| format!(\"Failed to create debug state file: {}\", path))?;\n // Get terminal width, default to 80\n let (width, _) = crossterm::terminal::size().unwrap_or((80, 24));\n Ok(Self {\n file,\n entry_count: 0,\n width,\n })\n }\n\n fn log(&mut self, label: &str, state: &crate::tui::AppState) {\n use crate::tui::calculate_needed_height;\n\n self.entry_count += 1;\n let timestamp_ms = std::time::SystemTime::now()\n .duration_since(std::time::UNIX_EPOCH)\n .map(|d| d.as_millis())\n .unwrap_or(0);\n\n // Calculate the actual content height needed for this state\n let content_height = calculate_needed_height(state, 0);\n\n let mut state_json = state_to_json(state);\n // Add dimensions for accurate replay\n state_json[\"width\"] = serde_json::json!(self.width);\n state_json[\"height\"] = serde_json::json!(content_height);\n\n let entry = serde_json::json!({\n \"entry\": self.entry_count,\n \"label\": label,\n \"timestamp_ms\": timestamp_ms,\n \"state\": state_json,\n });\n\n // Write as JSONL (one JSON object per line)\n if let Err(e) = writeln!(self.file, \"{}\", entry) {\n tracing::warn!(\"Failed to write debug state: {}\", e);\n }\n let _ = self.file.flush();\n }\n}\n\nasync fn run_inline_tui(\n endpoint: String,\n token: String,\n initial_prompt: Option,\n keep_output: bool,\n debug_state_file: Option,\n settings: &atuin_client::settings::Settings,\n) -> Result {\n // Detect popup mode (only on Unix where atuin-hex socket is available)\n #[cfg(unix)]\n let mut popup_state = crate::tui::popup::try_setup_popup();\n #[cfg(not(unix))]\n let mut popup_state: Option<()> = None;\n\n let popup_mode = popup_state.is_some();\n\n // Initialize terminal guard: popup mode uses Fixed viewport, inline uses Inline\n #[cfg(unix)]\n let mut guard = if let Some(ref ps) = popup_state {\n TerminalGuard::new_popup(ps.current_rect, ps.saved_screen.cursor_col)?\n } else {\n TerminalGuard::new(keep_output)?\n };\n #[cfg(not(unix))]\n let mut guard = TerminalGuard::new(keep_output)?;\n let mut app = App::new();\n if let Some(prompt) = initial_prompt {\n // Set initial text in textarea\n let mut textarea = tui_textarea::TextArea::from(prompt.lines());\n // Disable underline on cursor line\n textarea.set_cursor_line_style(ratatui::style::Style::default());\n // Enable word wrapping\n textarea.set_wrap_mode(tui_textarea::WrapMode::Word);\n // Move cursor to end\n textarea.move_cursor(tui_textarea::CursorMove::End);\n app.state.textarea = textarea;\n }\n\n // Initialize debug state logger if requested\n let mut debug_logger = debug_state_file\n .map(|path| DebugStateLogger::new(&path))\n .transpose()?;\n\n // Helper macro to log state changes\n macro_rules! log_state {\n ($label:expr) => {\n if let Some(ref mut logger) = debug_logger {\n logger.log($label, &app.state);\n }\n };\n }\n\n // Log initial state\n log_state!(\"init\");\n\n // Load theme\n let mut theme_manager = ThemeManager::new(None, None);\n let theme = theme_manager.load_theme(&settings.theme.name, None);\n\n // Initialize event loop\n let mut event_loop = EventLoop::new();\n\n // Track chat stream\n let mut chat_stream: Option<\n std::pin::Pin> + Send>>,\n > = None;\n\n loop {\n // Ensure viewport is large enough for current content (capped at terminal height)\n // In popup mode, use the actual popup width for accurate height calculation\n let card_width = if popup_mode {\n #[cfg(unix)]\n {\n popup_state\n .as_ref()\n .map(|ps| {\n ps.current_rect\n .width\n .saturating_sub(crate::tui::popup::POPUP_MARGIN * 2)\n })\n .unwrap_or(0)\n }\n #[cfg(not(unix))]\n {\n 0\n }\n } else {\n 0\n };\n let needed_height = calculate_needed_height(&app.state, card_width);\n\n // Grow popup dynamically as content arrives\n #[cfg(unix)]\n if let Some(ref mut ps) = popup_state {\n // Add vertical margin for visual separation from terminal content\n let popup_height = needed_height.saturating_add(crate::tui::popup::POPUP_MARGIN * 2);\n if let Some(new_rect) = ps.fit_to(popup_height) {\n guard.resize_popup(new_rect)?;\n }\n }\n\n let actual_height = guard.ensure_height(needed_height)?;\n\n // Render current state\n let anchor_col = guard.anchor_col();\n #[cfg(unix)]\n let render_above = popup_state.as_ref().is_some_and(|ps| ps.render_above);\n #[cfg(not(unix))]\n let render_above = false;\n\n let ctx = RenderContext {\n theme,\n anchor_col,\n textarea: Some(&app.state.textarea),\n max_height: actual_height,\n popup_mode,\n render_above,\n };\n // Handle draw errors gracefully - cursor position reads can fail during resize\n if let Err(e) = guard.terminal().draw(|frame| {\n render(frame, &app.state, &ctx);\n }) {\n let err_msg = e.to_string();\n if err_msg.contains(\"cursor position\") {\n // Cursor position read failed (common during terminal resize)\n // Skip this frame and continue - next frame will likely succeed\n tracing::debug!(\n \"Skipping frame due to cursor position read error: {}\",\n err_msg\n );\n continue;\n }\n return Err(e.into());\n }\n\n // Get next event\n let event = event_loop.run().await?;\n\n // Handle event based on app mode\n match event {\n AppEvent::Key(key) => {\n app.handle_key(key);\n log_state!(\"key\");\n }\n AppEvent::Tick => {\n app.state.tick();\n\n // Poll chat stream if active - keep polling until done regardless of mode\n // (mode may change to Review before we receive the done event with session_id)\n if let Some(stream) = &mut chat_stream {\n let mut cx = std::task::Context::from_waker(futures::task::noop_waker_ref());\n match stream.as_mut().poll_next(&mut cx) {\n std::task::Poll::Ready(Some(Ok(event))) => match event {\n ChatStreamEvent::TextChunk(text) => {\n trace!(text = %text, \"Processing TextChunk\");\n app.state.append_streaming_text(&text);\n log_state!(\"text_chunk\");\n }\n ChatStreamEvent::ToolCall { id, name, input } => {\n trace!(id = %id, name = %name, \"Processing ToolCall\");\n app.state.add_tool_call(id, name, input);\n log_state!(\"tool_call\");\n }\n ChatStreamEvent::ToolResult {\n tool_use_id,\n content,\n is_error,\n } => {\n trace!(tool_use_id = %tool_use_id, \"Processing ToolResult\");\n app.state.add_tool_result(tool_use_id, content, is_error);\n log_state!(\"tool_result\");\n }\n ChatStreamEvent::Status(status) => {\n trace!(status = %status, \"Processing Status\");\n app.state.update_streaming_status(&status);\n log_state!(\"status\");\n }\n ChatStreamEvent::Done { session_id } => {\n trace!(session_id = %session_id, \"Processing Done\");\n chat_stream = None;\n if !session_id.is_empty() {\n app.state.store_session_id(session_id);\n }\n app.state.finalize_streaming();\n log_state!(\"done\");\n }\n ChatStreamEvent::Error(msg) => {\n trace!(error = %msg, \"Processing Error\");\n chat_stream = None;\n app.state.streaming_error(msg);\n log_state!(\"error\");\n }\n },\n std::task::Poll::Ready(Some(Err(e))) => {\n chat_stream = None;\n app.state.streaming_error(e.to_string());\n log_state!(\"stream_error\");\n }\n std::task::Poll::Ready(None) => {\n chat_stream = None;\n app.state.finalize_streaming();\n log_state!(\"stream_end\");\n }\n std::task::Poll::Pending => {}\n }\n }\n }\n _ => {}\n }\n\n // Handle user cancellation (Esc during streaming) - drop the stream\n if app.state.was_interrupted && chat_stream.is_some() {\n debug!(\"User cancelled streaming, dropping chat stream\");\n chat_stream = None;\n app.state.was_interrupted = false; // Reset the flag\n }\n\n // Check exit condition (includes Ctrl+C / SIGINT from event loop)\n if app.state.should_exit || event_loop.is_shutdown() {\n break;\n }\n\n // Handle generation trigger - unified path for all turns\n if app.state.mode == AppMode::Generating && chat_stream.is_none() {\n // Get the last user message from events\n let last_user_content = app.state.events.iter().rev().find_map(|e| {\n if let ConversationEvent::UserMessage { content } = e {\n Some(content.clone())\n } else {\n None\n }\n });\n\n if last_user_content.is_some() {\n // Build messages in Claude API format\n let messages = app.state.events_to_messages();\n\n // Transition to streaming mode\n app.state.start_streaming();\n log_state!(\"start_streaming\");\n\n // Start the chat stream\n chat_stream = Some(create_chat_stream(\n endpoint.clone(),\n token.clone(),\n app.state.session_id.clone(),\n messages,\n settings,\n ));\n }\n }\n }\n\n // Restore popup area before guard drops (guard skips cleanup in popup mode)\n #[cfg(unix)]\n if let Some(ref ps) = popup_state {\n crate::tui::popup::restore(ps);\n }\n\n // Map exit action to return value\n let result = match app.state.exit_action {\n Some(ExitAction::Execute(cmd)) => Action::Execute(cmd),\n Some(ExitAction::Insert(cmd)) => Action::Insert(cmd),\n _ => Action::Cancel,\n };\n\n Ok(result)\n}\n\nstruct RawModeGuard;\n\nimpl Drop for RawModeGuard {\n fn drop(&mut self) {\n let _ = disable_raw_mode();\n }\n}\n\nfn emit_shell_result(action: Action, output_for_hook: bool) {\n if output_for_hook {\n match action {\n Action::Execute(output) => eprintln!(\"__atuin_ai_execute__:{output}\"),\n Action::Insert(output) => eprintln!(\"__atuin_ai_insert__:{output}\"),\n Action::Print(output) => eprintln!(\"__atuin_ai_print__:{output}\"),\n Action::Cancel => eprintln!(\"__atuin_ai_cancel__\"),\n }\n } else {\n match action {\n Action::Execute(output) => eprintln!(\"{output}\"),\n Action::Insert(output) => eprintln!(\"{output}\"),\n Action::Print(output) => eprintln!(\"{output}\"),\n Action::Cancel => eprintln!(),\n }\n }\n}\n\nfn wait_for_login_confirmation() -> Result {\n enable_raw_mode().context(\"failed enabling raw mode for login prompt\")?;\n let _guard = RawModeGuard;\n\n loop {\n let ev = event::read().context(\"failed to read login confirmation key\")?;\n if let Event::Key(key) = ev {\n match key.code {\n KeyCode::Enter => return Ok(true),\n KeyCode::Esc => return Ok(false),\n _ => {}\n }\n }\n }\n}\n" }, { "path": "crates/atuin-ai/src/commands.rs", "content": "use std::{\n fs,\n path::{Path, PathBuf},\n};\n\nuse atuin_common::shell::Shell;\nuse clap::{Args, Subcommand};\nuse eyre::Result;\nuse tracing_appender::rolling::{RollingFileAppender, Rotation};\nuse tracing_subscriber::{EnvFilter, Layer, fmt, layer::SubscriberExt, util::SubscriberInitExt};\n#[cfg(debug_assertions)]\npub mod debug_render;\n\npub mod init;\npub mod inline;\n\n#[derive(Args, Debug)]\npub struct AiArgs {\n /// Enable verbose logging\n #[arg(short, long, global = true)]\n verbose: bool,\n\n /// Custom API endpoint; defaults to reading from the `ai.endpoint` setting.\n #[arg(long, global = true)]\n api_endpoint: Option,\n\n /// Custom API token; defaults to reading from the `ai.api_token` setting.\n #[arg(long, global = true)]\n api_token: Option,\n}\n\n#[derive(Subcommand, Debug)]\npub enum Commands {\n /// Initialize shell integration\n Init {\n /// Shell to generate integration for; defaults to \"auto\"\n #[arg(value_name = \"SHELL\", default_value = \"auto\")]\n shell: String,\n },\n\n /// Inline completion mode with small TUI overlay\n Inline {\n #[command(flatten)]\n args: AiArgs,\n\n /// Current command line to complete\n #[arg(value_name = \"COMMAND\")]\n command: Option,\n\n /// Keep TUI output visible after exit (default: erase)\n #[arg(long)]\n keep: bool,\n\n /// Use the hook mode\n #[arg(long, hide = true)]\n hook: bool,\n\n /// Log state changes to file for debugging (dev tool)\n #[arg(long, value_name = \"FILE\", hide = true)]\n debug_state: Option,\n },\n\n /// Debug render: output a single frame from JSON state (dev tool)\n #[cfg(debug_assertions)]\n DebugRender {\n /// Input file (reads from stdin if not provided)\n #[arg(short, long)]\n input: Option,\n\n /// Output format: ansi (default), plain, json\n #[arg(short, long, default_value = \"ansi\")]\n format: String,\n },\n}\n\npub async fn run(\n command: Commands,\n settings: &atuin_client::settings::Settings,\n) -> eyre::Result<()> {\n match command {\n Commands::Init { shell } => init::run(shell).await,\n Commands::Inline {\n command,\n keep,\n debug_state,\n hook,\n args,\n ..\n } => {\n if settings.logs.ai_enabled() {\n init_logging(settings, args.verbose)?;\n }\n\n inline::run(\n command,\n args.api_endpoint,\n args.api_token,\n keep,\n debug_state,\n settings,\n hook,\n )\n .await\n }\n #[cfg(debug_assertions)]\n Commands::DebugRender { input, format } => {\n let output_format = match format.as_str() {\n \"plain\" => debug_render::OutputFormat::Plain,\n \"json\" => debug_render::OutputFormat::Json,\n _ => debug_render::OutputFormat::Ansi,\n };\n debug_render::run(input, output_format).await\n }\n }\n}\n\npub fn detect_shell() -> Option {\n Some(Shell::current().to_string())\n}\n\n/// Initializes logging for the AI commands.\nfn init_logging(settings: &atuin_client::settings::Settings, verbose: bool) -> Result<()> {\n // ATUIN_LOG env var overrides config file level settings\n let env_log_set = std::env::var(\"ATUIN_LOG\").is_ok();\n\n // Base filter from env var (or empty if not set)\n let base_filter =\n EnvFilter::from_env(\"ATUIN_LOG\").add_directive(\"sqlx_sqlite::regexp=off\".parse()?);\n\n // Use config level unless ATUIN_LOG is set\n let filter = if env_log_set {\n base_filter\n } else {\n EnvFilter::default()\n .add_directive(settings.logs.ai_level().as_directive().parse()?)\n .add_directive(\"sqlx_sqlite::regexp=off\".parse()?)\n };\n\n let log_dir = PathBuf::from(&settings.logs.dir);\n let ai_log_filename = settings.logs.ai.file.clone();\n\n // Clean up old log files\n cleanup_old_logs(&log_dir, &ai_log_filename, settings.logs.ai_retention());\n\n let console_layer = if verbose {\n Some(\n fmt::layer()\n .with_writer(std::io::stderr)\n .with_ansi(true)\n .with_target(false)\n .with_filter(filter.clone()),\n )\n } else {\n None\n };\n\n let file_appender = RollingFileAppender::new(Rotation::DAILY, &log_dir, &ai_log_filename);\n\n let base = tracing_subscriber::registry().with(\n fmt::layer()\n .with_writer(file_appender)\n .with_ansi(false)\n .with_filter(filter),\n );\n\n if let Some(console_layer) = console_layer {\n base.with(console_layer).init();\n } else {\n base.init();\n };\n\n Ok(())\n}\n\nfn cleanup_old_logs(log_dir: &Path, prefix: &str, retention_days: u64) {\n let cutoff = std::time::SystemTime::now()\n - std::time::Duration::from_secs(retention_days * 24 * 60 * 60);\n\n let Ok(entries) = fs::read_dir(log_dir) else {\n return;\n };\n\n for entry in entries.flatten() {\n let path = entry.path();\n let Some(name) = path.file_name().and_then(|n| n.to_str()) else {\n continue;\n };\n\n // Match files like \"search.log.2024-02-23\" or \"daemon.log.2024-02-23\"\n if !name.starts_with(prefix) || name == prefix {\n continue;\n }\n\n if let Ok(metadata) = entry.metadata()\n && let Ok(modified) = metadata.modified()\n && modified < cutoff\n {\n let _ = fs::remove_file(&path);\n }\n }\n}\n" }, { "path": "crates/atuin-ai/src/lib.rs", "content": "pub mod commands;\npub mod tui;\n" }, { "path": "crates/atuin-ai/src/tui/app.rs", "content": "use super::state::{AppMode, AppState, ExitAction};\nuse crossterm::event::{KeyCode, KeyEvent, KeyModifiers};\nuse tui_textarea::{Input, Key};\n\n/// Thin wrapper around AppState for compatibility\n/// All state lives in AppState, this just provides the handle_key interface\npub struct App {\n pub state: AppState,\n}\n\nimpl App {\n pub fn new() -> Self {\n Self {\n state: AppState::new(),\n }\n }\n\n /// Handle a key event. Returns true if render is needed.\n pub fn handle_key(&mut self, key: KeyEvent) -> bool {\n match self.state.mode {\n AppMode::Input => self.handle_input_key(key),\n AppMode::Generating => self.handle_generating_key(key),\n AppMode::Streaming => self.handle_streaming_key(key),\n AppMode::Review => self.handle_review_key(key),\n AppMode::Error => self.handle_error_key(key),\n }\n }\n\n fn handle_input_key(&mut self, key: KeyEvent) -> bool {\n // Handle special keys ourselves\n match key.code {\n KeyCode::Esc => {\n self.state.exit(ExitAction::Cancel);\n return true;\n }\n KeyCode::Enter => {\n if self.state.input_is_empty() {\n self.state.exit(ExitAction::Cancel);\n } else {\n self.state.start_generating();\n }\n return true;\n }\n _ => {}\n }\n\n // Delegate all other keys to textarea\n // Manually convert crossterm KeyEvent to tui-textarea Input\n // (needed due to crossterm version mismatch)\n let tui_key = match key.code {\n KeyCode::Char(c) => Key::Char(c),\n KeyCode::Backspace => Key::Backspace,\n KeyCode::Delete => Key::Delete,\n KeyCode::Left => Key::Left,\n KeyCode::Right => Key::Right,\n KeyCode::Up => Key::Up,\n KeyCode::Down => Key::Down,\n KeyCode::Home => Key::Home,\n KeyCode::End => Key::End,\n KeyCode::PageUp => Key::PageUp,\n KeyCode::PageDown => Key::PageDown,\n KeyCode::Tab => Key::Tab,\n _ => Key::Null,\n };\n\n if tui_key != Key::Null {\n let input = Input {\n key: tui_key,\n ctrl: key.modifiers.contains(KeyModifiers::CONTROL),\n alt: key.modifiers.contains(KeyModifiers::ALT),\n shift: key.modifiers.contains(KeyModifiers::SHIFT),\n };\n self.state.textarea.input(input);\n }\n true\n }\n\n fn handle_generating_key(&mut self, key: KeyEvent) -> bool {\n match key.code {\n KeyCode::Esc => {\n self.state.cancel_generation();\n true\n }\n _ => false, // Discard other keys during generation\n }\n }\n\n fn handle_streaming_key(&mut self, key: KeyEvent) -> bool {\n match key.code {\n KeyCode::Esc => {\n self.state.cancel_streaming();\n true\n }\n _ => false, // Ignore other keys during streaming\n }\n }\n\n fn handle_review_key(&mut self, key: KeyEvent) -> bool {\n match key.code {\n KeyCode::Esc => {\n self.state.confirmation_pending = false; // Clear confirmation state\n self.state.exit(ExitAction::Cancel);\n true\n }\n KeyCode::Enter => {\n let cmd = self.state.current_command().map(|c| c.to_string());\n if let Some(cmd) = cmd {\n if self.state.is_current_command_dangerous() && !self.state.confirmation_pending\n {\n // First Enter on dangerous command: enter confirmation mode\n self.state.confirmation_pending = true;\n } else {\n // Second Enter (confirmation), or non-dangerous command: execute\n self.state.confirmation_pending = false;\n self.state.exit(ExitAction::Execute(cmd));\n }\n }\n true\n }\n KeyCode::Tab => {\n let cmd = self.state.current_command().map(|c| c.to_string());\n if let Some(cmd) = cmd {\n self.state.confirmation_pending = false; // Clear on Tab too\n self.state.exit(ExitAction::Insert(cmd));\n }\n true\n }\n KeyCode::Char('f') => {\n // Changed from 'e' to 'f' for follow-up mode\n self.state.confirmation_pending = false; // Clear on follow-up\n self.state.start_edit_mode();\n true\n }\n _ => false,\n }\n }\n\n fn handle_error_key(&mut self, key: KeyEvent) -> bool {\n match key.code {\n KeyCode::Esc => {\n self.state.exit(ExitAction::Cancel);\n true\n }\n KeyCode::Enter | KeyCode::Char('r') => {\n self.state.retry();\n true\n }\n _ => false,\n }\n }\n}\n\nimpl Default for App {\n fn default() -> Self {\n Self::new()\n }\n}\n" }, { "path": "crates/atuin-ai/src/tui/component.rs", "content": "//! Component-oriented rendering primitives for the TUI.\n//!\n//! Defines the `Component` trait and container types (`VStack`, `SymbolRow`, etc.)\n//! that enable declarative, composable UI layout.\n\nuse atuin_client::theme::{Meaning, Theme};\nuse ratatui::{\n Frame, backend::FromCrossterm, layout::Rect, style::Style, text::Span, widgets::Paragraph,\n};\nuse tui_textarea::TextArea;\n\n/// Context passed through the component tree during rendering.\npub struct RenderContext<'a> {\n pub theme: &'a Theme,\n pub anchor_col: u16,\n pub textarea: Option<&'a TextArea<'static>>,\n /// Maximum viewport height (for scroll calculations)\n pub max_height: u16,\n /// When true, the viewport is a fixed rect already positioned for the card.\n /// The card fills the entire viewport instead of positioning via anchor_col.\n pub popup_mode: bool,\n /// When true, blocks are rendered in reverse order so that the input field\n /// appears at the bottom of the card (close to the prompt when the popup\n /// is above the cursor).\n pub render_above: bool,\n}\n\n/// A renderable component with intrinsic sizing.\npub trait Component {\n /// Calculate the intrinsic height at the given width.\n fn height(&self, width: u16) -> u16;\n\n /// Render into the given area.\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext);\n}\n\n/// Vertical stack of components.\n///\n/// Children are laid out top-to-bottom with optional spacing between them.\n/// When `scroll_offset > 0`, content is scrolled so that only the visible\n/// portion is rendered.\npub struct VStack {\n pub children: Vec>,\n pub spacing: u16,\n pub scroll_offset: u16,\n}\n\nimpl VStack {\n pub fn new(children: Vec>) -> Self {\n Self {\n children,\n spacing: 0,\n scroll_offset: 0,\n }\n }\n}\n\nimpl Component for VStack {\n fn height(&self, width: u16) -> u16 {\n if self.children.is_empty() {\n return 0;\n }\n let content: u16 = self.children.iter().map(|c| c.height(width)).sum();\n let gaps = (self.children.len() as u16 - 1) * self.spacing;\n content + gaps\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n if self.children.is_empty() {\n return;\n }\n\n let heights: Vec = self.children.iter().map(|c| c.height(area.width)).collect();\n\n let viewport_start = self.scroll_offset;\n let viewport_end = self.scroll_offset + area.height;\n\n let mut cum: u16 = 0;\n for (i, (child, &h)) in self.children.iter().zip(heights.iter()).enumerate() {\n let child_start = cum;\n let child_end = cum + h;\n\n // Render if any part of the child is within the viewport\n if child_end > viewport_start && child_start < viewport_end {\n let visible_start = child_start.max(viewport_start);\n let visible_end = child_end.min(viewport_end);\n\n let child_area = Rect {\n x: area.x,\n y: area.y + visible_start - viewport_start,\n width: area.width,\n height: visible_end - visible_start,\n };\n\n child.render(frame, child_area, ctx);\n }\n\n cum = child_end;\n if i < self.children.len() - 1 {\n cum += self.spacing;\n }\n }\n }\n}\n\n/// Fixed-height empty space.\npub struct Spacer(pub u16);\n\nimpl Component for Spacer {\n fn height(&self, _width: u16) -> u16 {\n self.0\n }\n\n fn render(&self, _frame: &mut Frame, _area: Rect, _ctx: &RenderContext) {}\n}\n\n/// A row with a symbol in column 0 and content in columns 2+.\n///\n/// This is the horizontal layout primitive used by all content types that\n/// display a prefix symbol (>, $, !, ?, etc.) followed by text.\npub struct SymbolRow {\n pub symbol: String,\n pub symbol_meaning: Meaning,\n pub inner: Box,\n}\n\nimpl Component for SymbolRow {\n fn height(&self, width: u16) -> u16 {\n self.inner.height(width.saturating_sub(2))\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n // Render symbol at column 0, first row only\n let style = Style::from_crossterm(ctx.theme.as_style(self.symbol_meaning));\n let symbol_area = Rect {\n x: area.x,\n y: area.y,\n width: 1,\n height: 1,\n };\n frame.render_widget(\n Paragraph::new(self.symbol.as_str()).style(style),\n symbol_area,\n );\n\n // Render inner content at column 2+\n let content_area = Rect {\n x: area.x.saturating_add(2),\n y: area.y,\n width: area.width.saturating_sub(2),\n height: area.height,\n };\n self.inner.render(frame, content_area, ctx);\n }\n}\n\n/// Horizontal separator spanning the full card width (├───┤).\n///\n/// Extends beyond its content area to overlap the card's left and right borders.\npub struct Separator {\n pub card_width: u16,\n}\n\nimpl Component for Separator {\n fn height(&self, _width: u16) -> u16 {\n 1\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Base));\n let inner_width = self.card_width.saturating_sub(2) as usize;\n let separator = format!(\n \"\\u{251c}{}\\u{2524}\", // ├ ... ┤\n \"\\u{2500}\".repeat(inner_width) // ─\n );\n\n // Extend left to overlap the card border (content area is inset by border + padding)\n let sep_area = Rect {\n x: area.x.saturating_sub(2),\n y: area.y,\n width: self.card_width,\n height: 1,\n };\n frame.render_widget(Paragraph::new(Span::styled(separator, style)), sep_area);\n }\n}\n" }, { "path": "crates/atuin-ai/src/tui/components.rs", "content": "//! Leaf components for each content type and factory functions for building\n//! the component tree from the view model.\n\nuse atuin_client::theme::{Meaning, Theme};\nuse ratatui::{\n Frame,\n backend::FromCrossterm,\n layout::Rect,\n style::{Modifier, Style},\n text::{Line, Span},\n widgets::{Paragraph, Wrap},\n};\n\nuse super::component::{Component, RenderContext, Separator, Spacer, SymbolRow, VStack};\nuse super::spinner::active_frame;\nuse super::view_model::{Block, Content, WarningKind};\n\n// ---------------------------------------------------------------------------\n// Text measurement utilities\n// ---------------------------------------------------------------------------\n\n/// Count lines when text is wrapped at given width.\n/// Uses ratatui's Paragraph::line_count for accurate wrapping calculation.\npub(crate) fn line_count_wrapped(text: &str, width: usize) -> u16 {\n if width == 0 {\n return 1;\n }\n let paragraph = Paragraph::new(text).wrap(Wrap { trim: false });\n paragraph.line_count(width as u16).max(1) as u16\n}\n\n/// Count lines using word-wrap algorithm (matches TextArea's WrapMode::Word).\n/// Words won't be broken mid-word, so this may produce more lines than character wrapping.\n/// Returns (line_count, last_line_width) so caller can determine if cursor needs extra space.\npub(crate) fn word_wrap_line_count_with_last_width(text: &str, width: usize) -> (u16, usize) {\n if width == 0 || text.is_empty() {\n return (1, 0);\n }\n\n let mut line_count = 0u16;\n let mut current_line_width = 0usize;\n\n for line in text.lines() {\n if line.is_empty() {\n line_count += 1;\n current_line_width = 0;\n continue;\n }\n\n let mut line_started = false;\n\n for word in line.split_whitespace() {\n let word_width = unicode_width::UnicodeWidthStr::width(word);\n\n if !line_started {\n if word_width > width {\n line_count += word_width.div_ceil(width) as u16;\n current_line_width = word_width % width;\n if current_line_width == 0 {\n current_line_width = 0;\n line_started = false;\n } else {\n line_started = true;\n }\n } else {\n current_line_width = word_width;\n line_started = true;\n }\n } else {\n let needed = current_line_width + 1 + word_width;\n if needed > width {\n line_count += 1;\n if word_width > width {\n line_count += word_width.div_ceil(width) as u16;\n current_line_width = word_width % width;\n if current_line_width == 0 {\n line_started = false;\n }\n } else {\n current_line_width = word_width;\n }\n } else {\n current_line_width = needed;\n }\n }\n }\n\n if line_started {\n line_count += 1;\n }\n }\n\n if line_count == 0 {\n line_count = 1;\n current_line_width = 0;\n }\n\n (line_count, current_line_width)\n}\n\n// ---------------------------------------------------------------------------\n// Inline markdown formatting\n// ---------------------------------------------------------------------------\n\n/// Parse inline markdown formatting (**bold** and `code`) into styled spans.\n/// Preserves all other text — list prefixes, indentation, and line structure\n/// are left exactly as-is.\nfn style_inline_markdown(text: &str, theme: &Theme) -> Vec> {\n let base_style = Style::from_crossterm(theme.as_style(Meaning::Base));\n let code_style = Style::from_crossterm(theme.as_style(Meaning::Guidance));\n let bold_style = base_style.add_modifier(Modifier::BOLD);\n\n text.lines()\n .map(|line| {\n Line::from(parse_inline_formatting(\n line, base_style, bold_style, code_style,\n ))\n })\n .collect()\n}\n\n/// Parse a single line for `code` and **bold** markers, returning styled spans.\nfn parse_inline_formatting(\n line: &str,\n base: Style,\n bold: Style,\n code: Style,\n) -> Vec> {\n let mut spans = Vec::new();\n let mut current = String::new();\n let mut chars = line.chars().peekable();\n\n while let Some(ch) = chars.next() {\n if ch == '`' {\n // Flush accumulated plain text\n if !current.is_empty() {\n spans.push(Span::styled(std::mem::take(&mut current), base));\n }\n // Collect until closing backtick\n let mut code_text = String::new();\n let mut closed = false;\n for next in chars.by_ref() {\n if next == '`' {\n closed = true;\n break;\n }\n code_text.push(next);\n }\n if closed {\n spans.push(Span::styled(code_text, code));\n } else {\n // Unclosed backtick — render as-is\n current.push('`');\n current.push_str(&code_text);\n }\n } else if ch == '*' && chars.peek() == Some(&'*') {\n chars.next(); // consume second *\n // Flush accumulated plain text\n if !current.is_empty() {\n spans.push(Span::styled(std::mem::take(&mut current), base));\n }\n // Collect until closing **\n let mut bold_text = String::new();\n let mut closed = false;\n while let Some(next) = chars.next() {\n if next == '*' && chars.peek() == Some(&'*') {\n chars.next();\n closed = true;\n break;\n }\n bold_text.push(next);\n }\n if closed {\n spans.push(Span::styled(bold_text, bold));\n } else {\n // Unclosed ** — render as-is\n current.push_str(\"**\");\n current.push_str(&bold_text);\n }\n } else {\n current.push(ch);\n }\n }\n\n if !current.is_empty() {\n spans.push(Span::styled(current, base));\n }\n\n spans\n}\n\n// ---------------------------------------------------------------------------\n// Leaf components\n// ---------------------------------------------------------------------------\n\n/// User input display (active textarea or static text).\npub struct InputContent {\n pub text: String,\n pub active: bool,\n}\n\nimpl Component for InputContent {\n fn height(&self, width: u16) -> u16 {\n let w = width as usize;\n if self.active {\n let (lines, last_width) = word_wrap_line_count_with_last_width(&self.text, w);\n if last_width >= w {\n lines.saturating_add(1)\n } else {\n lines\n }\n } else {\n line_count_wrapped(&self.text, w)\n }\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n if self.active {\n if let Some(textarea) = ctx.textarea {\n frame.render_widget(textarea, area);\n }\n } else {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Base));\n frame.render_widget(\n Paragraph::new(self.text.as_str())\n .style(style)\n .wrap(Wrap { trim: false }),\n area,\n );\n }\n }\n}\n\n/// Command suggestion ($ prefix).\npub struct CommandContent {\n pub text: String,\n pub faded: bool,\n}\n\nimpl Component for CommandContent {\n fn height(&self, width: u16) -> u16 {\n line_count_wrapped(&self.text, width as usize)\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let mut style = Style::from_crossterm(ctx.theme.as_style(Meaning::Base));\n if self.faded {\n style = style.add_modifier(Modifier::DIM);\n }\n frame.render_widget(\n Paragraph::new(self.text.as_str())\n .style(style)\n .wrap(Wrap { trim: false }),\n area,\n );\n }\n}\n\n/// Markdown text content (indented, no symbol).\npub struct TextContent {\n pub markdown: String,\n}\n\nimpl Component for TextContent {\n fn height(&self, width: u16) -> u16 {\n // Height uses raw text — slightly overestimates since markdown syntax\n // characters (**, `) are stripped in rendering, but this is harmless\n // (allocates equal or more space than needed, never less).\n line_count_wrapped(&self.markdown, width as usize)\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let lines = style_inline_markdown(&self.markdown, ctx.theme);\n let paragraph = Paragraph::new(lines).wrap(Wrap { trim: false });\n frame.render_widget(paragraph, area);\n }\n}\n\n/// Error message (! prefix).\npub struct ErrorContent {\n pub message: String,\n}\n\nimpl Component for ErrorContent {\n fn height(&self, width: u16) -> u16 {\n line_count_wrapped(&self.message, width as usize)\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Base));\n frame.render_widget(\n Paragraph::new(self.message.as_str())\n .style(style)\n .wrap(Wrap { trim: false }),\n area,\n );\n }\n}\n\n/// Warning for dangerous or low-confidence commands.\npub struct WarningContent {\n pub kind: WarningKind,\n pub text: String,\n pub pending_confirm: bool,\n}\n\nimpl Component for WarningContent {\n fn height(&self, width: u16) -> u16 {\n let display_text = if self.pending_confirm {\n \"Press Enter again to run this dangerous command\"\n } else {\n self.text.as_str()\n };\n line_count_wrapped(display_text, width as usize)\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Base));\n let display_text = if self.pending_confirm {\n \"Press Enter again to run this dangerous command\"\n } else {\n self.text.as_str()\n };\n frame.render_widget(\n Paragraph::new(display_text)\n .style(style)\n .wrap(Wrap { trim: false }),\n area,\n );\n }\n}\n\n/// Animated spinner with status text.\npub struct SpinnerContent {\n pub status_text: String,\n}\n\nimpl Component for SpinnerContent {\n fn height(&self, _width: u16) -> u16 {\n 1\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Annotation));\n frame.render_widget(Paragraph::new(self.status_text.as_str()).style(style), area);\n }\n}\n\n/// Tool call progress (in-flight spinner or completed checkmark).\npub struct ToolStatusContent {\n pub completed_count: usize,\n pub current_label: Option,\n pub frame: usize,\n}\n\nimpl Component for ToolStatusContent {\n fn height(&self, _width: u16) -> u16 {\n 1\n }\n\n fn render(&self, frame: &mut Frame, area: Rect, ctx: &RenderContext) {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Annotation));\n let text = if let Some(ref label) = self.current_label {\n if self.completed_count > 0 {\n format!(\n \"{} (used {} tool{})\",\n label,\n self.completed_count,\n if self.completed_count == 1 { \"\" } else { \"s\" }\n )\n } else {\n label.clone()\n }\n } else {\n format!(\n \"Used {} tool{}\",\n self.completed_count,\n if self.completed_count == 1 { \"\" } else { \"s\" }\n )\n };\n frame.render_widget(Paragraph::new(text).style(style), area);\n }\n}\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\n/// Convert a view model `Content` item into a `SymbolRow`-wrapped component.\nfn content_to_component(content: &Content) -> Box {\n match content {\n Content::Input { text, active, .. } => Box::new(SymbolRow {\n symbol: \">\".to_string(),\n symbol_meaning: Meaning::Guidance,\n inner: Box::new(InputContent {\n text: text.clone(),\n active: *active,\n }),\n }),\n\n Content::Command { text, faded } => Box::new(SymbolRow {\n symbol: \"$\".to_string(),\n symbol_meaning: Meaning::Important,\n inner: Box::new(CommandContent {\n text: text.clone(),\n faded: *faded,\n }),\n }),\n\n Content::Text { markdown } => Box::new(SymbolRow {\n symbol: \" \".to_string(),\n symbol_meaning: Meaning::Base,\n inner: Box::new(TextContent {\n markdown: markdown.clone(),\n }),\n }),\n\n Content::Error { message } => Box::new(SymbolRow {\n symbol: \"!\".to_string(),\n symbol_meaning: Meaning::AlertError,\n inner: Box::new(ErrorContent {\n message: message.clone(),\n }),\n }),\n\n Content::Warning {\n kind,\n text,\n pending_confirm,\n } => {\n let (symbol, meaning) = match kind {\n WarningKind::Danger => (\"!\", Meaning::AlertError),\n WarningKind::LowConfidence => (\"?\", Meaning::AlertWarn),\n };\n Box::new(SymbolRow {\n symbol: symbol.to_string(),\n symbol_meaning: meaning,\n inner: Box::new(WarningContent {\n kind: *kind,\n text: text.clone(),\n pending_confirm: *pending_confirm,\n }),\n })\n }\n\n Content::Spinner { frame, status_text } => Box::new(SymbolRow {\n symbol: active_frame(*frame).to_string(),\n symbol_meaning: Meaning::Annotation,\n inner: Box::new(SpinnerContent {\n status_text: status_text.clone(),\n }),\n }),\n\n Content::ToolStatus {\n completed_count,\n current_label,\n frame,\n } => {\n let symbol = if current_label.is_some() {\n active_frame(*frame).to_string()\n } else {\n \"\\u{2713}\".to_string() // ✓\n };\n Box::new(SymbolRow {\n symbol,\n symbol_meaning: Meaning::Annotation,\n inner: Box::new(ToolStatusContent {\n completed_count: *completed_count,\n current_label: current_label.clone(),\n frame: *frame,\n }),\n })\n }\n }\n}\n\n/// Convert a view model `Block` into a `VStack` of content components.\nfn build_block_component(block: &Block) -> Box {\n let mut children: Vec> = Vec::new();\n\n for (idx, content) in block.content.iter().enumerate() {\n if idx > 0 {\n children.push(Box::new(Spacer(1))); // blank line between items\n }\n children.push(content_to_component(content));\n }\n\n // Trailing blank line (padding after content)\n children.push(Box::new(Spacer(1)));\n\n Box::new(VStack::new(children))\n}\n\n/// Build the full component tree from an ordered list of view model blocks.\n///\n/// The tree is a `VStack` with blocks separated by `Separator` + `Spacer` pairs.\n/// The caller sets `scroll_offset` on the returned `VStack` before rendering.\npub fn build_component_tree(items: &[&Block], card_width: u16) -> VStack {\n let mut children: Vec> = Vec::new();\n\n for (idx, block) in items.iter().enumerate() {\n if idx > 0 {\n children.push(Box::new(Separator { card_width }));\n children.push(Box::new(Spacer(1))); // leading blank after separator\n }\n children.push(build_block_component(block));\n }\n\n VStack::new(children)\n}\n" }, { "path": "crates/atuin-ai/src/tui/event.rs", "content": "use crate::tui::App;\nuse crossterm::event::{Event, EventStream, KeyEvent, KeyEventKind};\nuse eyre::{Result, eyre};\nuse futures::StreamExt;\nuse std::time::Duration;\nuse tokio::time;\n\n/// Base tick interval for the event loop (fast for responsive streaming)\nconst BASE_TICK_INTERVAL: Duration = Duration::from_millis(50);\n\n/// Application events that drive the TUI state machine.\n///\n/// # Event Types\n/// - `Key`: Keyboard input (filtered to KeyEventKind::Press only)\n/// - `Tick`: Periodic event for updates (50ms base interval)\n/// - `Resize`: Terminal window resize\n/// - `StreamChunk/StreamDone/StreamError`: Placeholders for Phase 3 streaming\n///\n/// # Design Decisions\n/// - Fast 50ms base tick for responsive streaming; spinner timing handled in AppState\n/// - Stream events are placeholders - will be wired to channels in Phase 3\n/// - Resize handling enables responsive layout adjustments\n#[derive(Debug, Clone)]\npub enum AppEvent {\n /// Keyboard input event (filtered to Press events only)\n Key(KeyEvent),\n\n /// Periodic tick for updates (50ms base interval; spinner timing in AppState)\n Tick,\n\n /// Terminal resize event (width, height)\n Resize(u16, u16),\n\n /// Stream chunk received (Phase 3 placeholder)\n StreamChunk(String),\n\n /// Stream completed successfully (Phase 3 placeholder)\n StreamDone,\n\n /// Stream error occurred (Phase 3 placeholder)\n StreamError(String),\n}\n\n/// Async event loop that drives the TUI with prioritized event handling.\n///\n/// # Priority Model (Biased Select)\n/// 1. **Stream data** - Highest priority (future Phase 3 streaming)\n/// 2. **Keyboard input** - Medium priority (user responsiveness)\n/// 3. **Tick events** - Lowest priority (spinner animation)\n///\n/// This ensures stream data is processed immediately when available,\n/// keyboard input is responsive, and spinner updates don't block higher priority events.\n///\n/// # Graceful Shutdown\n/// - SIGINT (Ctrl+C) sets shutdown flag and breaks the loop\n/// - EventStream close (stdin EOF) triggers shutdown\n/// - Shutdown flag can be checked/set externally for controlled termination\n///\n/// # Example\n/// ```no_run\n/// use atuin_ai::tui::EventLoop;\n///\n/// # async fn example() -> eyre::Result<()> {\n/// let mut event_loop = EventLoop::new();\n/// loop {\n/// let event = event_loop.run().await?;\n/// // Handle event...\n/// # break;\n/// }\n/// # Ok(())\n/// # }\n/// ```\npub struct EventLoop {\n /// Tick interval timer (created lazily on first run)\n tick_timer: Option,\n\n /// Flag indicating a render was requested (future use in Phase 2)\n #[allow(dead_code)]\n render_requested: bool,\n\n /// Shutdown flag - when true, event loop will terminate\n shutdown: bool,\n}\n\nimpl EventLoop {\n /// Create a new EventLoop with default settings.\n ///\n /// # Defaults\n /// - Tick interval: 50ms base rate (spinner timing handled separately in AppState)\n /// - Render requested: false\n /// - Shutdown: false\n pub fn new() -> Self {\n Self {\n tick_timer: None,\n render_requested: false,\n shutdown: false,\n }\n }\n\n /// Run the event loop, returning the next application event.\n ///\n /// # Priority Model\n /// Uses `tokio::select!` with `biased;` mode to enforce priority:\n /// 1. Stream data (placeholder for Phase 3)\n /// 2. Keyboard input with rapid keypress batching\n /// 3. Tick for spinner animation\n ///\n /// # Keyboard Handling\n /// - Filters to KeyEventKind::Press on all platforms for safety\n /// - Batching of rapid keypresses will be implemented in Phase 2\n /// - Currently returns individual key events\n ///\n /// # Graceful Shutdown\n /// - SIGINT (Ctrl+C) triggers shutdown and returns last event\n /// - EventStream close (stdin EOF) triggers shutdown\n /// - Shutdown flag can be checked after this returns\n ///\n /// # Errors\n /// - Returns error if terminal event stream encounters an error\n /// - EventStream close is handled gracefully as shutdown signal\n ///\n /// # Example\n /// ```no_run\n /// # use atuin_ai::tui::EventLoop;\n /// # async fn example() -> eyre::Result<()> {\n /// let mut event_loop = EventLoop::new();\n /// while !event_loop.is_shutdown() {\n /// match event_loop.run().await? {\n /// // Handle events...\n /// # _ => break,\n /// }\n /// }\n /// # Ok(())\n /// # }\n /// ```\n pub async fn run(&mut self) -> Result {\n // Create async event stream for keyboard/terminal events\n let mut reader = EventStream::new();\n\n // Get or create the tick timer (reused across calls to maintain timing)\n // Uses fast base tick for responsive streaming; spinner timing handled in AppState\n let tick_timer = self.tick_timer.get_or_insert_with(|| {\n let mut interval = time::interval(BASE_TICK_INTERVAL);\n // Skip the first immediate tick\n interval.reset();\n interval\n });\n\n loop {\n if self.shutdown {\n break;\n }\n\n // Biased select: prioritize stream > keyboard > tick\n let event = tokio::select! {\n biased;\n\n // Priority 1: Stream data (placeholder for Phase 3)\n // In Phase 3, this will be:\n // Some(chunk) = stream_rx.recv() => { ... }\n\n // Priority 2: Keyboard input\n maybe_event = reader.next() => {\n match maybe_event {\n Some(Ok(Event::Key(key))) => {\n // Filter to Press events only for cross-platform safety\n if key.kind == KeyEventKind::Press {\n // Note: Rapid keypress batching will be implemented in Phase 2\n // when we integrate with the state machine.\n // For now, just return individual key events.\n Some(AppEvent::Key(key))\n } else {\n None\n }\n }\n Some(Ok(Event::Resize(w, h))) => {\n Some(AppEvent::Resize(w, h))\n }\n Some(Err(e)) => {\n return Err(eyre!(\"terminal event error: {}\", e));\n }\n None => {\n // EventStream closed (stdin EOF) - trigger shutdown\n self.shutdown = true;\n None\n }\n _ => {\n // Ignore other event types (mouse, focus, etc.)\n None\n }\n }\n }\n\n // Priority 3: Tick for spinner animation\n _ = tick_timer.tick() => {\n Some(AppEvent::Tick)\n }\n\n // SIGINT handling (Ctrl+C) - cross-platform\n _ = tokio::signal::ctrl_c() => {\n self.shutdown = true;\n // Return one more event to allow graceful shutdown handling\n Some(AppEvent::Tick)\n }\n };\n\n if let Some(app_event) = event {\n return Ok(app_event);\n }\n }\n\n // Loop exited due to shutdown - return final tick to allow cleanup\n Ok(AppEvent::Tick)\n }\n\n /// Check if the event loop has been signaled to shut down.\n ///\n /// This can be used to cleanly exit the main TUI loop after receiving\n /// a shutdown signal (Ctrl+C, stdin close, etc.)\n pub fn is_shutdown(&self) -> bool {\n self.shutdown\n }\n\n /// Signal the event loop to shut down.\n ///\n /// The shutdown will take effect on the next iteration of `run()`.\n pub fn shutdown(&mut self) {\n self.shutdown = true;\n }\n\n /// Poll for next event and apply to app state.\n ///\n /// This is a convenience method that combines `run()` with `App` state updates.\n /// Returns true if app should continue, false if should exit.\n ///\n /// # Example\n /// ```no_run\n /// # use atuin_ai::tui::{EventLoop, App};\n /// # async fn example() -> eyre::Result<()> {\n /// let mut event_loop = EventLoop::new();\n /// let mut app = App::new();\n ///\n /// while event_loop.poll_and_apply(&mut app).await? {\n /// // Render app state...\n /// }\n /// # Ok(())\n /// # }\n /// ```\n pub async fn poll_and_apply(&mut self, app: &mut App) -> Result {\n let event = self.run().await?;\n\n match event {\n AppEvent::Key(key) => {\n app.handle_key(key);\n }\n AppEvent::Tick => {\n app.state.tick();\n }\n AppEvent::Resize(_, _) => {\n // Render will be triggered anyway\n }\n AppEvent::StreamChunk(_) | AppEvent::StreamDone | AppEvent::StreamError(_) => {\n // Placeholder for Phase 3\n }\n }\n\n Ok(!app.state.should_exit)\n }\n}\n\nimpl Default for EventLoop {\n fn default() -> Self {\n Self::new()\n }\n}\n\n#[cfg(test)]\nmod tests {\n use super::*;\n\n #[test]\n fn test_event_loop_creation() {\n let event_loop = EventLoop::new();\n assert!(!event_loop.shutdown);\n }\n\n #[test]\n fn test_shutdown_flag() {\n let mut event_loop = EventLoop::new();\n assert!(!event_loop.is_shutdown());\n\n event_loop.shutdown();\n assert!(event_loop.is_shutdown());\n }\n\n // Note: Cannot easily test run() in unit tests since it requires a TTY.\n // Integration tests should verify:\n // 1. Tick events are generated at 150ms intervals\n // 2. Keyboard events are properly filtered to Press only\n // 3. Rapid keypresses are batched\n // 4. SIGINT triggers graceful shutdown\n // 5. Resize events are propagated correctly\n}\n" }, { "path": "crates/atuin-ai/src/tui/mod.rs", "content": "pub mod app;\npub mod component;\npub mod components;\npub mod event;\n#[cfg(unix)]\npub mod popup;\npub mod render;\npub mod spinner;\npub mod state;\npub mod terminal;\npub mod view_model;\n\npub use app::App;\npub use event::{AppEvent, EventLoop};\npub use render::{RenderContext, calculate_needed_height, markdown_to_spans};\npub use state::{AppMode, AppState, ConversationEvent, ExitAction};\npub use terminal::{TerminalGuard, install_panic_hook};\npub use view_model::{Block, Blocks, Content};\n" }, { "path": "crates/atuin-ai/src/tui/popup.rs", "content": "use ratatui::layout::Rect;\n\n/// Maximum popup height (lines). Keeps context visible around the popup.\nconst MAX_POPUP_HEIGHT: u16 = 24;\n\n/// Minimum usable popup height.\nconst MIN_POPUP_HEIGHT: u16 = 5;\n\n/// Initial popup height — just enough for input + a small response.\nconst INITIAL_POPUP_HEIGHT: u16 = 5;\n\n/// Margin around the card in popup mode.\npub(crate) const POPUP_MARGIN: u16 = 0;\n\n/// Screen state captured from atuin-hex's screen server.\npub struct SavedScreen {\n #[allow(dead_code)]\n pub rows: u16,\n #[allow(dead_code)]\n pub cols: u16,\n pub cursor_row: u16,\n pub cursor_col: u16,\n /// Pre-formatted ANSI bytes for each screen row, ready to write to stdout.\n pub rows_data: Vec>,\n}\n\n/// Popup mode state: saved screen + computed placement.\npub struct PopupState {\n pub saved_screen: SavedScreen,\n /// Maximum rect computed from placement (the ceiling for growth).\n pub max_rect: Rect,\n /// Current rect — starts small, grows as content arrives.\n pub current_rect: Rect,\n pub scroll_offset: u16,\n /// True when the popup renders above the cursor (input at bottom of card).\n pub render_above: bool,\n}\n\nimpl PopupState {\n /// Resize the popup to fit `needed` lines of content.\n ///\n /// Grows or shrinks the popup as needed (clamped to max_rect / INITIAL_POPUP_HEIGHT).\n /// When growing, clears the new rect area. When shrinking, restores freed rows\n /// from the saved screen data.\n ///\n /// Returns `Some(new_rect)` if the size changed (caller must resize terminal),\n /// or `None` if no change is needed.\n pub fn fit_to(&mut self, needed: u16) -> Option {\n let new_height = needed.clamp(INITIAL_POPUP_HEIGHT, self.max_rect.height);\n if new_height == self.current_rect.height {\n return None;\n }\n\n let old_rect = self.current_rect;\n let growing = new_height > old_rect.height;\n\n if self.render_above {\n let new_y = self.max_rect.y + self.max_rect.height - new_height;\n self.current_rect = Rect::new(old_rect.x, new_y, old_rect.width, new_height);\n } else {\n self.current_rect = Rect::new(old_rect.x, old_rect.y, old_rect.width, new_height);\n }\n\n if growing {\n // Clear the entire new rect so the new Terminal doesn't leave\n // ghost content from the old card.\n self.clear_rows(\n self.current_rect.y,\n self.current_rect.y + self.current_rect.height,\n );\n } else {\n // Shrinking: restore freed rows from saved screen data, then\n // clear the new (smaller) rect for the re-rendered card.\n self.restore_rows(&old_rect);\n self.clear_rows(\n self.current_rect.y,\n self.current_rect.y + self.current_rect.height,\n );\n }\n\n Some(self.current_rect)\n }\n\n /// Clear a range of terminal rows within the popup width.\n fn clear_rows(&self, from_row: u16, to_row: u16) {\n use crossterm::cursor::MoveTo;\n use crossterm::execute;\n use crossterm::style::{Attribute, SetAttribute};\n use std::io::{Write, stdout};\n\n let mut out = stdout();\n for row in from_row..to_row {\n let _ = execute!(\n out,\n MoveTo(self.current_rect.x, row),\n SetAttribute(Attribute::Reset)\n );\n let _ = write!(\n out,\n \"{:width$}\",\n \"\",\n width = self.current_rect.width as usize\n );\n }\n let _ = out.flush();\n }\n\n /// Restore rows that were freed by shrinking — the rows in old_rect\n /// that are no longer covered by current_rect.\n fn restore_rows(&self, old_rect: &Rect) {\n use crossterm::cursor::MoveTo;\n use crossterm::execute;\n use crossterm::style::{Attribute, SetAttribute};\n use std::io::{Write, stdout};\n\n let mut out = stdout();\n\n // Determine which rows are freed\n let (freed_start, freed_end) = if self.render_above {\n // Shrinking from above: freed rows are at the old top\n (old_rect.y, self.current_rect.y)\n } else {\n // Shrinking from below: freed rows are at the old bottom\n (\n self.current_rect.y + self.current_rect.height,\n old_rect.y + old_rect.height,\n )\n };\n\n for row in freed_start..freed_end {\n let source_row = (row + self.scroll_offset) as usize;\n\n // Clear the popup region\n let _ = execute!(out, MoveTo(old_rect.x, row), SetAttribute(Attribute::Reset),);\n let _ = write!(out, \"{:width$}\", \"\", width = old_rect.width as usize);\n\n // Write back saved row data from column 0\n let _ = execute!(out, MoveTo(0, row));\n if let Some(row_bytes) = self.saved_screen.rows_data.get(source_row) {\n let _ = out.write_all(row_bytes);\n }\n }\n let _ = out.flush();\n }\n}\n\n/// Try to set up popup overlay mode.\n///\n/// Checks for `ATUIN_HEX_SOCKET`, fetches screen state, computes placement,\n/// and scrolls the terminal if needed. Returns `None` if popup mode is not\n/// available (no socket, fetch failed, etc.), in which case the caller should\n/// fall back to inline mode.\npub fn try_setup_popup() -> Option {\n use std::io::Write;\n\n let socket_path = std::env::var(\"ATUIN_HEX_SOCKET\").ok()?;\n let saved = fetch_screen_state(&socket_path)?;\n\n let (term_cols, term_rows) = crossterm::terminal::size().unwrap_or((saved.cols, saved.rows));\n // Full-width popup with margin for visual separation\n let popup_width = term_cols;\n let (rect, scroll, render_above) = compute_popup_placement(\n saved.cursor_row,\n saved.cursor_col,\n term_rows,\n term_cols,\n popup_width,\n );\n\n // Scroll terminal up if needed to make room for the popup\n if scroll > 0 {\n let mut stdout = std::io::stdout();\n let _ = crossterm::execute!(stdout, crossterm::cursor::MoveTo(0, term_rows - 1));\n for _ in 0..scroll {\n let _ = writeln!(stdout);\n }\n let _ = stdout.flush();\n }\n\n // Start with a small rect that grows as content arrives\n let initial_height = INITIAL_POPUP_HEIGHT.min(rect.height);\n let current_rect = if render_above {\n // Anchor at the bottom of max_rect (near cursor), grow upward\n Rect::new(\n rect.x,\n rect.y + rect.height - initial_height,\n rect.width,\n initial_height,\n )\n } else {\n // Anchor at the top of max_rect (near cursor), grow downward\n Rect::new(rect.x, rect.y, rect.width, initial_height)\n };\n\n Some(PopupState {\n saved_screen: saved,\n max_rect: rect,\n current_rect,\n scroll_offset: scroll,\n render_above,\n })\n}\n\n/// Restore the screen area that was covered by the popup.\n///\n/// Clears the popup region, then writes pre-formatted per-row ANSI bytes from\n/// column 0 to correctly restore wide characters, colors, and all attributes.\npub fn restore(state: &PopupState) {\n use crossterm::cursor::MoveTo;\n use crossterm::execute;\n use crossterm::style::{Attribute, SetAttribute};\n use std::io::{Write, stdout};\n\n let saved = &state.saved_screen;\n let popup_rect = state.current_rect;\n let scroll_offset = state.scroll_offset;\n\n let mut stdout = stdout();\n\n for dy in 0..popup_rect.height {\n let target_row = popup_rect.y + dy;\n let source_row = (target_row + scroll_offset) as usize;\n\n // Clear only the popup region with spaces\n let _ = execute!(\n stdout,\n MoveTo(popup_rect.x, target_row),\n SetAttribute(Attribute::Reset),\n );\n let _ = write!(stdout, \"{:width$}\", \"\", width = popup_rect.width as usize);\n\n // Write back full row ANSI data from column 0\n let _ = execute!(stdout, MoveTo(0, target_row));\n if let Some(row_bytes) = saved.rows_data.get(source_row) {\n let _ = stdout.write_all(row_bytes);\n }\n }\n\n // Restore cursor position (adjusted for any scrolling)\n let _ = execute!(\n stdout,\n MoveTo(\n saved.cursor_col,\n saved.cursor_row.saturating_sub(scroll_offset)\n )\n );\n let _ = stdout.flush();\n}\n\n/// Connect to atuin-hex's Unix socket and fetch the current screen state.\n///\n/// The wire format is:\n/// ```text\n/// [rows: u16 BE][cols: u16 BE][cursor_row: u16 BE][cursor_col: u16 BE]\n/// [row_0_len: u32 BE][row_0_bytes...]\n/// [row_1_len: u32 BE][row_1_bytes...]\n/// ...\n/// ```\nfn fetch_screen_state(socket_path: &str) -> Option {\n use std::io::Read;\n use std::os::unix::net::UnixStream;\n use std::time::Duration;\n\n let mut stream = UnixStream::connect(socket_path).ok()?;\n stream.set_read_timeout(Some(Duration::from_secs(2))).ok()?;\n\n let mut data = Vec::new();\n stream.read_to_end(&mut data).ok()?;\n\n if data.len() < 8 {\n return None;\n }\n\n let rows = u16::from_be_bytes([data[0], data[1]]);\n let cols = u16::from_be_bytes([data[2], data[3]]);\n let cursor_row = u16::from_be_bytes([data[4], data[5]]);\n let cursor_col = u16::from_be_bytes([data[6], data[7]]);\n\n let mut rows_data = Vec::with_capacity(rows as usize);\n let mut offset = 8;\n while offset + 4 <= data.len() {\n let row_len = u32::from_be_bytes([\n data[offset],\n data[offset + 1],\n data[offset + 2],\n data[offset + 3],\n ]) as usize;\n offset += 4;\n if offset + row_len > data.len() {\n break;\n }\n rows_data.push(data[offset..offset + row_len].to_vec());\n offset += row_len;\n }\n\n Some(SavedScreen {\n rows,\n cols,\n cursor_row,\n cursor_col,\n rows_data,\n })\n}\n\n/// Compute popup placement for the AI card.\n///\n/// Positions the popup near the cursor: below if there's room, above otherwise.\n/// Uses a capped height (MAX_POPUP_HEIGHT) so the popup doesn't fill the screen.\n///\n/// Returns `(popup_rect, scroll_offset, render_above)`:\n/// - `render_above`: true when popup is above cursor (input should be at bottom)\n/// - `scroll_offset`: lines the caller should scroll the terminal up\nfn compute_popup_placement(\n cursor_row: u16,\n cursor_col: u16,\n term_rows: u16,\n term_cols: u16,\n card_width: u16,\n) -> (Rect, u16, bool) {\n // Horizontal: anchor card near cursor, clamp to screen\n let popup_w = card_width.min(term_cols);\n let preferred_x = cursor_col.saturating_sub(2);\n let max_x = term_cols.saturating_sub(popup_w);\n let popup_x = preferred_x.min(max_x);\n\n // Vertical: use a reasonable height, not the full terminal\n let max_h = MAX_POPUP_HEIGHT\n .min(term_rows.saturating_sub(2))\n .max(MIN_POPUP_HEIGHT);\n let space_above = cursor_row;\n let space_below = term_rows.saturating_sub(cursor_row);\n\n if max_h <= space_below {\n // Fits below cursor — input at top (close to prompt)\n let popup_y = cursor_row;\n (Rect::new(popup_x, popup_y, popup_w, max_h), 0, false)\n } else if max_h <= space_above {\n // Fits above cursor — input at bottom (close to prompt)\n let popup_y = cursor_row.saturating_sub(max_h);\n (Rect::new(popup_x, popup_y, popup_w, max_h), 0, true)\n } else {\n // Neither side fits fully — use whichever side has more space,\n // scrolling the terminal if needed to reach MIN_POPUP_HEIGHT.\n let render_above = space_above > space_below;\n let available = if render_above {\n space_above\n } else {\n space_below\n };\n let h = available.max(MIN_POPUP_HEIGHT).min(max_h);\n let scroll = h.saturating_sub(available);\n let popup_y = if render_above {\n cursor_row.saturating_sub(h + scroll)\n } else {\n cursor_row.saturating_sub(scroll)\n };\n (\n Rect::new(popup_x, popup_y, popup_w, h),\n scroll,\n render_above,\n )\n }\n}\n" }, { "path": "crates/atuin-ai/src/tui/render.rs", "content": "use atuin_client::theme::{Meaning, Theme};\nuse pulldown_cmark::{Event, Parser, Tag, TagEnd};\nuse ratatui::{\n Frame,\n backend::FromCrossterm,\n layout::{Alignment, Rect},\n style::{Modifier, Style},\n text::{Line, Span},\n widgets::{Block as RatatuiBlock, Borders, Padding},\n};\n\nuse super::component::Component;\npub use super::component::RenderContext;\nuse super::components::build_component_tree;\nuse super::spinner::active_frame;\nuse super::state::AppState;\nuse super::view_model::Blocks;\n\n/// Fixed card width for the TUI\npub(crate) const CARD_WIDTH: u16 = 64;\n\n/// Calculate the height needed to render the current state.\n/// Used to dynamically resize the viewport before rendering.\n/// `card_width` is the outer card width (including borders); pass 0 to use CARD_WIDTH default.\npub fn calculate_needed_height(state: &AppState, card_width: u16) -> u16 {\n let view = Blocks::from_state(state);\n let w = if card_width > 0 {\n card_width\n } else {\n CARD_WIDTH\n };\n let content_width = w.saturating_sub(4).max(1);\n\n let items: Vec<_> = view.items.iter().collect();\n let tree = build_component_tree(&items, w);\n\n // Add borders (2) + top padding (1), minimum 5\n tree.height(content_width).saturating_add(3).max(5)\n}\n\n/// Main render function: derives view model from state, then renders it\npub fn render(frame: &mut Frame, state: &AppState, ctx: &RenderContext) {\n // PURE DERIVATION: view model is always rebuilt from state\n let view = Blocks::from_state(state);\n\n // Render the derived view model\n render_view(frame, &view, ctx);\n}\n\nfn render_view(frame: &mut Frame, view: &Blocks, ctx: &RenderContext) {\n let full_area = frame.area();\n\n // In popup mode, the viewport is already positioned and sized for the card.\n // Clear it to prevent background bleed-through, then inset by margin for the card.\n let (area, card_x, desired_width) = if ctx.popup_mode {\n #[cfg(unix)]\n use super::popup::POPUP_MARGIN;\n #[cfg(not(unix))]\n const POPUP_MARGIN: u16 = 0;\n frame.render_widget(ratatui::widgets::Clear, full_area);\n let inset = full_area.inner(ratatui::layout::Margin {\n horizontal: POPUP_MARGIN,\n vertical: POPUP_MARGIN,\n });\n (inset, inset.x, inset.width)\n } else {\n let dw = CARD_WIDTH.min(full_area.width.saturating_sub(2)).max(32);\n let max_x = full_area.x + full_area.width.saturating_sub(dw);\n let preferred_x = full_area.x + ctx.anchor_col.saturating_sub(2);\n (full_area, preferred_x.min(max_x), dw)\n };\n\n // Build ordered items list — the active content (input/LLM response)\n // should always be closest to the cursor/prompt:\n // - Popup below cursor (render_above=false): reverse so active is at top\n // - Popup above cursor (render_above=true): normal order, active is at bottom\n // - Inline mode: normal order (no reversal)\n let items: Vec<&super::view_model::Block> = if ctx.popup_mode && !ctx.render_above {\n view.items.iter().rev().collect()\n } else {\n view.items.iter().collect()\n };\n\n // Build component tree from view model\n let mut tree = build_component_tree(&items, desired_width);\n let content_width = desired_width.saturating_sub(4).max(1);\n\n let desired_height = tree.height(content_width).saturating_add(3).max(5);\n\n // Cap card height at viewport height to prevent overflow\n let actual_height = desired_height.min(area.height);\n\n // Calculate scroll offset to keep the active content visible when overflowing.\n // When render_above=false (popup below cursor), items are reversed so the active\n // content (input/spinner) is at the top — scroll_offset stays 0 to show the top.\n // Otherwise, scroll to show the bottom where the active content lives.\n tree.scroll_offset = if ctx.popup_mode && !ctx.render_above {\n 0\n } else {\n desired_height.saturating_sub(actual_height)\n };\n\n let card = Rect {\n x: card_x,\n y: area.y,\n width: desired_width,\n height: actual_height,\n };\n\n // Get title from first block in ORIGINAL order (always the input block)\n let title = view\n .items\n .first()\n .and_then(|b| b.title.as_deref())\n .unwrap_or(\"Describe the command you'd like to generate:\");\n\n // Create bordered frame\n // Padding: left=1, right=1, top=1, bottom=0 (blocks have trailing blanks)\n let mut outer_block = RatatuiBlock::default()\n .borders(Borders::ALL)\n .title(title)\n .title_top(Line::from(\"atuin\").alignment(Alignment::Right))\n .title_bottom(Line::from(view.footer).alignment(Alignment::Right))\n .padding(Padding::new(1, 1, 1, 0));\n\n // Status bar: transient status on the bottom border, left-aligned\n if let Some(ref sb) = view.status_bar {\n let style = Style::from_crossterm(ctx.theme.as_style(Meaning::Annotation));\n let spinner = active_frame(sb.frame);\n let status_text = format!(\" {} {} \", spinner, sb.text);\n outer_block = outer_block\n .title_bottom(Line::from(Span::styled(status_text, style)).alignment(Alignment::Left));\n }\n\n let inner_area = outer_block.inner(card);\n frame.render_widget(outer_block, card);\n\n // Render the component tree\n tree.render(frame, inner_area, ctx);\n}\n\n/// Convert markdown to styled spans\npub fn markdown_to_spans<'a>(text: &'a str, theme: &'a Theme) -> Vec> {\n let parser = Parser::new(text);\n let mut lines: Vec>> = vec![Vec::new()];\n let mut current_line = 0;\n\n let base_style = Style::from_crossterm(theme.as_style(Meaning::Base));\n let code_style = Style::from_crossterm(theme.as_style(Meaning::Important));\n let mut style_stack: Vec