mirror of
https://github.com/Gu1llaum-3/sshm.git
synced 2026-01-27 03:04:21 +01:00
Compare commits
19 Commits
v1.6.0
...
v1.10.0-be
| Author | SHA1 | Date | |
|---|---|---|---|
| 167e4c0a09 | |||
| 825c534ebe | |||
| c1457af73a | |||
| 12d97270f0 | |||
| 6ba82b1c97 | |||
| 42e87b6827 | |||
| d686d97f8c | |||
| 8d5f59fab2 | |||
| 049998c235 | |||
|
|
5986659048 | ||
|
|
abbda54125 | ||
|
|
986017a552 | ||
|
|
120cd6c009 | ||
| 3d746ec49a | |||
| f31fe9dacf | |||
| 7b15db1f34 | |||
| 55f3359287 | |||
| 4efec57a8a | |||
| 0975ae2fe2 |
136
.github/workflows/build.yml
vendored
136
.github/workflows/build.yml
vendored
@@ -1,136 +0,0 @@
|
|||||||
name: Build Binaries
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
tags:
|
|
||||||
- '*'
|
|
||||||
release:
|
|
||||||
types: [created]
|
|
||||||
workflow_dispatch:
|
|
||||||
|
|
||||||
permissions:
|
|
||||||
contents: write
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
build:
|
|
||||||
name: Build
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
strategy:
|
|
||||||
matrix:
|
|
||||||
include:
|
|
||||||
# Linux AMD64
|
|
||||||
- goos: linux
|
|
||||||
goarch: amd64
|
|
||||||
suffix: linux-amd64
|
|
||||||
# Linux ARM64
|
|
||||||
- goos: linux
|
|
||||||
goarch: arm64
|
|
||||||
suffix: linux-arm64
|
|
||||||
# macOS AMD64 (Intel)
|
|
||||||
- goos: darwin
|
|
||||||
goarch: amd64
|
|
||||||
suffix: darwin-amd64
|
|
||||||
# macOS ARM64 (Apple Silicon)
|
|
||||||
- goos: darwin
|
|
||||||
goarch: arm64
|
|
||||||
suffix: darwin-arm64
|
|
||||||
# Windows AMD64
|
|
||||||
- goos: windows
|
|
||||||
goarch: amd64
|
|
||||||
suffix: windows-amd64
|
|
||||||
# Windows ARM64
|
|
||||||
- goos: windows
|
|
||||||
goarch: arm64
|
|
||||||
suffix: windows-arm64
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- name: Checkout code
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
|
|
||||||
- name: Set up Go
|
|
||||||
uses: actions/setup-go@v5
|
|
||||||
with:
|
|
||||||
go-version: '1.23'
|
|
||||||
|
|
||||||
- name: Cache Go modules
|
|
||||||
uses: actions/cache@v4
|
|
||||||
with:
|
|
||||||
path: ~/go/pkg/mod
|
|
||||||
key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
|
|
||||||
restore-keys: |
|
|
||||||
${{ runner.os }}-go-
|
|
||||||
|
|
||||||
- name: Build binary
|
|
||||||
env:
|
|
||||||
GOOS: ${{ matrix.goos }}
|
|
||||||
GOARCH: ${{ matrix.goarch }}
|
|
||||||
CGO_ENABLED: 0
|
|
||||||
run: |
|
|
||||||
mkdir -p dist
|
|
||||||
VERSION=${GITHUB_REF#refs/tags/}
|
|
||||||
# Remove 'v' prefix if present for version injection
|
|
||||||
VERSION_CLEAN=${VERSION#v}
|
|
||||||
if [ "${{ matrix.goos }}" = "windows" ]; then
|
|
||||||
go build -ldflags="-s -w -X github.com/Gu1llaum-3/sshm/cmd.AppVersion=${VERSION_CLEAN}" -o dist/sshm-${{ matrix.suffix }}.exe .
|
|
||||||
else
|
|
||||||
go build -ldflags="-s -w -X github.com/Gu1llaum-3/sshm/cmd.AppVersion=${VERSION_CLEAN}" -o dist/sshm-${{ matrix.suffix }} .
|
|
||||||
fi
|
|
||||||
|
|
||||||
- name: Create archive
|
|
||||||
run: |
|
|
||||||
cd dist
|
|
||||||
if [ "${{ matrix.goos }}" = "windows" ]; then
|
|
||||||
zip sshm-${{ matrix.suffix }}.zip sshm-${{ matrix.suffix }}.exe
|
|
||||||
rm sshm-${{ matrix.suffix }}.exe
|
|
||||||
else
|
|
||||||
tar -czf sshm-${{ matrix.suffix }}.tar.gz sshm-${{ matrix.suffix }}
|
|
||||||
rm sshm-${{ matrix.suffix }}
|
|
||||||
fi
|
|
||||||
|
|
||||||
- name: Upload artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: sshm-${{ matrix.suffix }}
|
|
||||||
path: |
|
|
||||||
dist/sshm-${{ matrix.suffix }}.tar.gz
|
|
||||||
dist/sshm-${{ matrix.suffix }}.zip
|
|
||||||
|
|
||||||
release:
|
|
||||||
name: Create Release
|
|
||||||
needs: build
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
if: startsWith(github.ref, 'refs/tags/')
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- name: Download all artifacts
|
|
||||||
uses: actions/download-artifact@v4
|
|
||||||
with:
|
|
||||||
path: ./artifacts
|
|
||||||
merge-multiple: true
|
|
||||||
|
|
||||||
- name: Prepare release assets
|
|
||||||
run: |
|
|
||||||
mkdir -p release
|
|
||||||
find ./artifacts -name "*.tar.gz" -exec cp {} ./release/ \;
|
|
||||||
find ./artifacts -name "*.zip" -exec cp {} ./release/ \;
|
|
||||||
ls -la ./release/
|
|
||||||
|
|
||||||
- name: Check if pre-release
|
|
||||||
id: check_prerelease
|
|
||||||
run: |
|
|
||||||
if [[ "${GITHUB_REF#refs/tags/}" == *"-beta"* ]] || [[ "${GITHUB_REF#refs/tags/}" == *"-alpha"* ]] || [[ "${GITHUB_REF#refs/tags/}" == *"-rc"* ]] || [[ "${GITHUB_REF#refs/tags/}" == *"-dev"* ]]; then
|
|
||||||
echo "is_prerelease=true" >> $GITHUB_OUTPUT
|
|
||||||
else
|
|
||||||
echo "is_prerelease=false" >> $GITHUB_OUTPUT
|
|
||||||
fi
|
|
||||||
|
|
||||||
- name: Create Release
|
|
||||||
uses: softprops/action-gh-release@v2
|
|
||||||
with:
|
|
||||||
files: ./release/*
|
|
||||||
draft: false
|
|
||||||
prerelease: ${{ steps.check_prerelease.outputs.is_prerelease }}
|
|
||||||
generate_release_notes: true
|
|
||||||
name: ${{ github.ref_name }}${{ steps.check_prerelease.outputs.is_prerelease == 'true' && ' (Pre-release)' || '' }}
|
|
||||||
env:
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
39
.github/workflows/release.yml
vendored
Normal file
39
.github/workflows/release.yml
vendored
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
name: Release with GoReleaser
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- '*'
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
contents: write
|
||||||
|
# Required for Homebrew tap updates
|
||||||
|
issues: write
|
||||||
|
pull-requests: write
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
goreleaser:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
# Fetch full history for changelog generation
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Set up Go
|
||||||
|
uses: actions/setup-go@v5
|
||||||
|
with:
|
||||||
|
go-version: '1.23'
|
||||||
|
cache: true
|
||||||
|
|
||||||
|
- name: Run GoReleaser
|
||||||
|
uses: goreleaser/goreleaser-action@v6
|
||||||
|
with:
|
||||||
|
distribution: goreleaser
|
||||||
|
version: '~> v2'
|
||||||
|
args: release --clean
|
||||||
|
env:
|
||||||
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
# Token for updating Homebrew tap (create this secret in your repo settings)
|
||||||
|
HOMEBREW_TAP_GITHUB_TOKEN: ${{ secrets.HOMEBREW_TAP_GITHUB_TOKEN }}
|
||||||
155
.goreleaser.yaml
Normal file
155
.goreleaser.yaml
Normal file
@@ -0,0 +1,155 @@
|
|||||||
|
version: 2
|
||||||
|
project_name: sshm
|
||||||
|
|
||||||
|
before:
|
||||||
|
hooks:
|
||||||
|
- go mod tidy
|
||||||
|
- go test ./...
|
||||||
|
|
||||||
|
builds:
|
||||||
|
- id: sshm
|
||||||
|
main: ./main.go
|
||||||
|
binary: sshm
|
||||||
|
goos:
|
||||||
|
- linux
|
||||||
|
- windows
|
||||||
|
- darwin
|
||||||
|
goarch:
|
||||||
|
- amd64
|
||||||
|
- arm64
|
||||||
|
- "386"
|
||||||
|
- arm
|
||||||
|
goarm:
|
||||||
|
- "6"
|
||||||
|
- "7"
|
||||||
|
ignore:
|
||||||
|
# Skip ARM for Windows (not commonly used)
|
||||||
|
- goos: windows
|
||||||
|
goarch: arm
|
||||||
|
- goos: windows
|
||||||
|
goarch: arm64
|
||||||
|
env:
|
||||||
|
- CGO_ENABLED=0
|
||||||
|
ldflags:
|
||||||
|
- -s -w
|
||||||
|
- -X github.com/Gu1llaum-3/sshm/cmd.AppVersion={{.Version}}
|
||||||
|
flags:
|
||||||
|
- -trimpath
|
||||||
|
|
||||||
|
archives:
|
||||||
|
- id: sshm
|
||||||
|
formats: [ "tar.gz" ]
|
||||||
|
# Use zip for Windows
|
||||||
|
format_overrides:
|
||||||
|
- goos: windows
|
||||||
|
formats: [ "zip" ]
|
||||||
|
# Template for archive name
|
||||||
|
name_template: >-
|
||||||
|
{{ .ProjectName }}_ {{- title .Os }}_ {{- if eq .Arch "amd64" }}x86_64 {{- else if eq .Arch "386" }}i386 {{- else }}{{ .Arch }}{{ end }} {{- if .Arm }}v{{ .Arm }}{{ end }}
|
||||||
|
files:
|
||||||
|
- LICENSE
|
||||||
|
- README.md
|
||||||
|
|
||||||
|
checksum:
|
||||||
|
name_template: "checksums.txt"
|
||||||
|
algorithm: sha256
|
||||||
|
|
||||||
|
changelog:
|
||||||
|
use: github
|
||||||
|
sort: asc
|
||||||
|
filters:
|
||||||
|
exclude:
|
||||||
|
- "^docs:"
|
||||||
|
- "^test:"
|
||||||
|
- "^ci:"
|
||||||
|
- "^chore:"
|
||||||
|
- "^build:"
|
||||||
|
groups:
|
||||||
|
- title: Features
|
||||||
|
regexp: '^.*?feat(\([[:word:]]+\))??!?:.+$'
|
||||||
|
order: 0
|
||||||
|
- title: Bug fixes
|
||||||
|
regexp: '^.*?fix(\([[:word:]]+\))??!?:.+$'
|
||||||
|
order: 1
|
||||||
|
- title: Others
|
||||||
|
order: 999
|
||||||
|
|
||||||
|
# Homebrew tap configuration (Formula pour CLI)
|
||||||
|
brews:
|
||||||
|
- name: sshm
|
||||||
|
repository:
|
||||||
|
owner: Gu1llaum-3
|
||||||
|
name: homebrew-sshm
|
||||||
|
# Token with repo permissions for your homebrew-sshm repo
|
||||||
|
token: "{{ .Env.HOMEBREW_TAP_GITHUB_TOKEN }}"
|
||||||
|
commit_author:
|
||||||
|
name: goreleaserbot
|
||||||
|
email: bot@goreleaser.com
|
||||||
|
commit_msg_template: "Brew formula update for {{ .ProjectName }} version {{ .Tag }}"
|
||||||
|
homepage: "https://github.com/Gu1llaum-3/sshm"
|
||||||
|
description: "A modern SSH connection manager for your terminal"
|
||||||
|
license: MIT
|
||||||
|
skip_upload: auto
|
||||||
|
# Test command to verify installation
|
||||||
|
test: |
|
||||||
|
system "#{bin}/sshm --version"
|
||||||
|
|
||||||
|
# Release configuration
|
||||||
|
release:
|
||||||
|
github:
|
||||||
|
owner: Gu1llaum-3
|
||||||
|
name: sshm
|
||||||
|
prerelease: auto
|
||||||
|
draft: false
|
||||||
|
replace_existing_draft: true
|
||||||
|
target_commitish: "{{ .Commit }}"
|
||||||
|
name_template: "{{.ProjectName}} {{.Version}}"
|
||||||
|
header: |
|
||||||
|
## SSHM {{.Version}}
|
||||||
|
|
||||||
|
Thank you for downloading SSHM!
|
||||||
|
|
||||||
|
footer: |
|
||||||
|
## Installation
|
||||||
|
|
||||||
|
### Homebrew (macOS/Linux)
|
||||||
|
```bash
|
||||||
|
brew tap Gu1llaum-3/sshm
|
||||||
|
brew install sshm
|
||||||
|
```
|
||||||
|
|
||||||
|
### Installation Script (Recommended)
|
||||||
|
**Unix/Linux/macOS:**
|
||||||
|
```bash
|
||||||
|
curl -sSL https://github.com/Gu1llaum-3/sshm/raw/main/install/unix.sh | bash
|
||||||
|
```
|
||||||
|
|
||||||
|
**Windows (PowerShell):**
|
||||||
|
```powershell
|
||||||
|
iwr -useb https://github.com/Gu1llaum-3/sshm/raw/main/install/windows.ps1 | iex
|
||||||
|
```
|
||||||
|
|
||||||
|
### Manual Installation
|
||||||
|
Download the appropriate binary for your platform from the assets above, extract it, and place it in your PATH.
|
||||||
|
|
||||||
|
## Full Changelog
|
||||||
|
|
||||||
|
See all changes at https://github.com/Gu1llaum-3/sshm/compare/{{.PreviousTag}}...{{.Tag}}
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
📖 **Documentation:** See the updated [README](https://github.com/Gu1llaum-3/sshm/blob/main/README.md)
|
||||||
|
|
||||||
|
🐛 **Issues:** Found a bug? Open an [issue](https://github.com/Gu1llaum-3/sshm/issues)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Released with ❤️ by [GoReleaser](https://github.com/goreleaser/goreleaser)
|
||||||
|
|
||||||
|
# Snapshot builds (for non-tag builds)
|
||||||
|
snapshot:
|
||||||
|
version_template: "{{ .Tag }}-snapshot-{{.ShortCommit}}"
|
||||||
|
|
||||||
|
# Metadata for package managers
|
||||||
|
metadata:
|
||||||
|
mod_timestamp: "{{ .CommitTimestamp }}"
|
||||||
44
Makefile
Normal file
44
Makefile
Normal file
@@ -0,0 +1,44 @@
|
|||||||
|
.PHONY: build build-local test clean release snapshot
|
||||||
|
|
||||||
|
# Version can be overridden via environment variable or command line
|
||||||
|
VERSION ?= dev
|
||||||
|
|
||||||
|
# Go build flags
|
||||||
|
LDFLAGS := -s -w -X github.com/Gu1llaum-3/sshm/cmd.AppVersion=$(VERSION)
|
||||||
|
|
||||||
|
# Build with specific version
|
||||||
|
build:
|
||||||
|
@mkdir -p dist
|
||||||
|
go build -ldflags="$(LDFLAGS)" -o dist/sshm .
|
||||||
|
|
||||||
|
# Build with git version
|
||||||
|
build-local: VERSION := $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
|
||||||
|
build-local: build
|
||||||
|
|
||||||
|
# Run tests
|
||||||
|
test:
|
||||||
|
go test ./...
|
||||||
|
|
||||||
|
# Clean build artifacts
|
||||||
|
clean:
|
||||||
|
rm -rf dist
|
||||||
|
|
||||||
|
# Release with GoReleaser (requires tag)
|
||||||
|
release:
|
||||||
|
@if [ -z "$(shell git tag --points-at HEAD)" ]; then \
|
||||||
|
echo "Error: No git tag found at current commit. Create a tag first with: git tag vX.Y.Z"; \
|
||||||
|
exit 1; \
|
||||||
|
fi
|
||||||
|
goreleaser release --clean
|
||||||
|
|
||||||
|
# Build snapshot (without tag)
|
||||||
|
snapshot:
|
||||||
|
goreleaser release --snapshot --clean
|
||||||
|
|
||||||
|
# Check GoReleaser config
|
||||||
|
release-check:
|
||||||
|
goreleaser check
|
||||||
|
|
||||||
|
# Run GoReleaser in dry-run mode
|
||||||
|
release-dry-run:
|
||||||
|
goreleaser release --snapshot --skip=publish --clean
|
||||||
79
README.md
79
README.md
@@ -34,7 +34,7 @@ SSHM is a beautiful command-line tool that transforms how you manage and connect
|
|||||||
- **🔍 Smart Search** - Find hosts quickly with built-in filtering and search
|
- **🔍 Smart Search** - Find hosts quickly with built-in filtering and search
|
||||||
- **📝 Real-time Status** - Live SSH connectivity indicators with asynchronous ping checks and color-coded status
|
- **📝 Real-time Status** - Live SSH connectivity indicators with asynchronous ping checks and color-coded status
|
||||||
- **🔔 Smart Updates** - Automatic version checking with update notifications
|
- **🔔 Smart Updates** - Automatic version checking with update notifications
|
||||||
- **📈 Connection History** - Track your SSH connections with last login timestamps
|
- **📈 Connection History** - Track both configured and manual SSH connections with timestamps and usage counts
|
||||||
|
|
||||||
### 🛠️ **Technical Features**
|
### 🛠️ **Technical Features**
|
||||||
- **🔒 Secure** - Works directly with your existing `~/.ssh/config` file
|
- **🔒 Secure** - Works directly with your existing `~/.ssh/config` file
|
||||||
@@ -53,6 +53,11 @@ SSHM is a beautiful command-line tool that transforms how you manage and connect
|
|||||||
|
|
||||||
### Installation
|
### Installation
|
||||||
|
|
||||||
|
**Homebrew (Recommended for macOS):**
|
||||||
|
```bash
|
||||||
|
brew install Gu1llaum-3/sshm/sshm
|
||||||
|
```
|
||||||
|
|
||||||
**Unix/Linux/macOS (One-line install):**
|
**Unix/Linux/macOS (One-line install):**
|
||||||
```bash
|
```bash
|
||||||
curl -sSL https://raw.githubusercontent.com/Gu1llaum-3/sshm/main/install/unix.sh | bash
|
curl -sSL https://raw.githubusercontent.com/Gu1llaum-3/sshm/main/install/unix.sh | bash
|
||||||
@@ -101,6 +106,7 @@ sshm
|
|||||||
- `d` - Delete selected host
|
- `d` - Delete selected host
|
||||||
- `m` - Move host to another config file (requires SSH Include directives)
|
- `m` - Move host to another config file (requires SSH Include directives)
|
||||||
- `f` - Port forwarding setup
|
- `f` - Port forwarding setup
|
||||||
|
- `Ctrl+H` - Switch to connection history view
|
||||||
- `q` - Quit
|
- `q` - Quit
|
||||||
- `/` - Search/filter hosts
|
- `/` - Search/filter hosts
|
||||||
|
|
||||||
@@ -280,6 +286,47 @@ sshm web-01
|
|||||||
- **Error handling** - Clear messages if host doesn't exist or configuration issues
|
- **Error handling** - Clear messages if host doesn't exist or configuration issues
|
||||||
- **Config file support** - Works with custom config files using `-c` flag
|
- **Config file support** - Works with custom config files using `-c` flag
|
||||||
|
|
||||||
|
### Connection History
|
||||||
|
|
||||||
|
SSHM automatically tracks all your SSH connections, including both configured hosts and manual connections made outside of SSHM.
|
||||||
|
|
||||||
|
**Access History:**
|
||||||
|
Press `Ctrl+H` from the main interface to switch to the history view. Press `Ctrl+L` to return to the main host list.
|
||||||
|
|
||||||
|
**Features:**
|
||||||
|
- **Automatic tracking** - Records all SSH connections with timestamps and connection counts
|
||||||
|
- **Manual connection detection** - Captures `ssh user@host -p port -i key` commands made in your terminal
|
||||||
|
- **Visual indicators** - Manual connections (not in your SSH config) are marked with ★
|
||||||
|
- **Search & filter** - Find connections quickly using the search bar
|
||||||
|
- **Add to config** - Press `a` on any manual connection (★) to add it to your SSH config
|
||||||
|
- **Persistent storage** - History is saved in `~/.config/sshm/sshm_history.json`
|
||||||
|
|
||||||
|
**Tracked Information:**
|
||||||
|
- Host name or hostname for manual connections
|
||||||
|
- Username and hostname
|
||||||
|
- Port number
|
||||||
|
- Last connection timestamp
|
||||||
|
- Total connection count
|
||||||
|
|
||||||
|
**Use Cases:**
|
||||||
|
- Review your recent SSH activity
|
||||||
|
- Find frequently used manual connections
|
||||||
|
- Promote manual connections to permanent SSH config entries
|
||||||
|
- Track when you last connected to a host
|
||||||
|
|
||||||
|
**Example Workflow:**
|
||||||
|
```bash
|
||||||
|
# Make a manual SSH connection
|
||||||
|
ssh deploy@192.168.1.100 -p 2222 -i ~/.ssh/custom_key
|
||||||
|
|
||||||
|
# Launch SSHM and press Ctrl+H to view history
|
||||||
|
sshm
|
||||||
|
# Press Ctrl+H → see the manual connection with ★ indicator
|
||||||
|
# Press 'a' to add it to your SSH config
|
||||||
|
# Give it a name like "deploy-server" and save
|
||||||
|
# Press Ctrl+L to return to main list → now it's a configured host
|
||||||
|
```
|
||||||
|
|
||||||
### Backup Configuration
|
### Backup Configuration
|
||||||
|
|
||||||
SSHM automatically creates backups of your SSH configuration files before making any changes to ensure your configurations are safe.
|
SSHM automatically creates backups of your SSH configuration files before making any changes to ensure your configurations are safe.
|
||||||
@@ -548,6 +595,34 @@ This will be automatically converted to:
|
|||||||
StrictHostKeyChecking no
|
StrictHostKeyChecking no
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Custom Key Bindings
|
||||||
|
|
||||||
|
SSHM supports customizable key bindings through a configuration file. This is particularly useful for users who want to modify the default quit behavior.
|
||||||
|
|
||||||
|
**Configuration File Location:**
|
||||||
|
- **Linux/macOS**: `~/.config/sshm/config.json`
|
||||||
|
- **Windows**: `%APPDATA%\sshm\config.json`
|
||||||
|
|
||||||
|
**Example Configuration:**
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"key_bindings": {
|
||||||
|
"quit_keys": ["q", "ctrl+c"],
|
||||||
|
"disable_esc_quit": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
**Available Options:**
|
||||||
|
- **quit_keys**: Array of keys that will quit the application. Default: `["q", "ctrl+c"]`
|
||||||
|
- **disable_esc_quit**: Boolean flag to disable ESC key from quitting the application. Default: `false`
|
||||||
|
|
||||||
|
**For Vim Users:**
|
||||||
|
If you frequently press ESC accidentally causing the application to quit, set `disable_esc_quit` to `true`. This will disable ESC as a quit key while preserving all other functionality.
|
||||||
|
|
||||||
|
**Default Configuration:**
|
||||||
|
If no configuration file exists, SSHM will automatically create one with default settings that maintain backward compatibility.
|
||||||
|
|
||||||
## 🛠️ Development
|
## 🛠️ Development
|
||||||
|
|
||||||
### Prerequisites
|
### Prerequisites
|
||||||
@@ -664,6 +739,8 @@ This project is licensed under the MIT License - see the [LICENSE](LICENSE) file
|
|||||||
- [Charm](https://charm.sh/) for the amazing TUI libraries
|
- [Charm](https://charm.sh/) for the amazing TUI libraries
|
||||||
- [Cobra](https://cobra.dev/) for the excellent CLI framework
|
- [Cobra](https://cobra.dev/) for the excellent CLI framework
|
||||||
- [@yimeng](https://github.com/yimeng) for contributing SSH Include directive support
|
- [@yimeng](https://github.com/yimeng) for contributing SSH Include directive support
|
||||||
|
- [@ldreux](https://github.com/ldreux) for contributing multi-word search functionality
|
||||||
|
- [@qingfengzxr](https://github.com/qingfengzxr) for contributing custom key bindings support
|
||||||
- The Go community for building such fantastic tools
|
- The Go community for building such fantastic tools
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|||||||
122
cmd/root.go
122
cmd/root.go
@@ -49,6 +49,7 @@ Hosts are read from your ~/.ssh/config file by default.`,
|
|||||||
}
|
}
|
||||||
|
|
||||||
// If a host name is provided, connect directly
|
// If a host name is provided, connect directly
|
||||||
|
// (manual SSH commands are handled in Execute() before reaching here)
|
||||||
hostName := args[0]
|
hostName := args[0]
|
||||||
connectToHost(hostName)
|
connectToHost(hostName)
|
||||||
return nil
|
return nil
|
||||||
@@ -103,27 +104,18 @@ func runInteractiveMode() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func connectToHost(hostName string) {
|
func connectToHost(hostName string) {
|
||||||
// Parse SSH configurations to verify host exists
|
// Quick check if host exists without full parsing (optimized for connection)
|
||||||
var hosts []config.SSHHost
|
var hostFound bool
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
if configFile != "" {
|
if configFile != "" {
|
||||||
hosts, err = config.ParseSSHConfigFile(configFile)
|
hostFound, err = config.QuickHostExistsInFile(hostName, configFile)
|
||||||
} else {
|
} else {
|
||||||
hosts, err = config.ParseSSHConfig()
|
hostFound, err = config.QuickHostExists(hostName)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("Error reading SSH config file: %v", err)
|
log.Fatalf("Error checking SSH config: %v", err)
|
||||||
}
|
|
||||||
|
|
||||||
// Check if host exists
|
|
||||||
var hostFound bool
|
|
||||||
for _, host := range hosts {
|
|
||||||
if host.Name == hostName {
|
|
||||||
hostFound = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if !hostFound {
|
if !hostFound {
|
||||||
@@ -149,11 +141,84 @@ func connectToHost(hostName string) {
|
|||||||
fmt.Printf("Connecting to %s...\n", hostName)
|
fmt.Printf("Connecting to %s...\n", hostName)
|
||||||
|
|
||||||
var sshCmd *exec.Cmd
|
var sshCmd *exec.Cmd
|
||||||
|
var args []string
|
||||||
|
|
||||||
if configFile != "" {
|
if configFile != "" {
|
||||||
sshCmd = exec.Command("ssh", "-F", configFile, hostName)
|
args = append(args, "-F", configFile)
|
||||||
} else {
|
|
||||||
sshCmd = exec.Command("ssh", hostName)
|
|
||||||
}
|
}
|
||||||
|
args = append(args, hostName)
|
||||||
|
|
||||||
|
// Note: We don't add RemoteCommand here because if it's configured in SSH config,
|
||||||
|
// SSH will handle it automatically. Adding it as a command line argument would conflict.
|
||||||
|
|
||||||
|
sshCmd = exec.Command("ssh", args...)
|
||||||
|
|
||||||
|
// Set up the command to use the same stdin, stdout, and stderr as the parent process
|
||||||
|
sshCmd.Stdin = os.Stdin
|
||||||
|
sshCmd.Stdout = os.Stdout
|
||||||
|
sshCmd.Stderr = os.Stderr
|
||||||
|
|
||||||
|
// Execute the SSH command
|
||||||
|
err = sshCmd.Run()
|
||||||
|
if err != nil {
|
||||||
|
if exitError, ok := err.(*exec.ExitError); ok {
|
||||||
|
// SSH command failed, exit with the same code
|
||||||
|
if status, ok := exitError.Sys().(syscall.WaitStatus); ok {
|
||||||
|
os.Exit(status.ExitStatus())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fmt.Printf("Error executing SSH command: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// connectManualSSH handles manual SSH connections like: sshm -p 2222 user@host
|
||||||
|
func connectManualSSH(args []string) {
|
||||||
|
// Parse the manual connection arguments
|
||||||
|
conn, ok := history.ParseSSHArgs(args)
|
||||||
|
if !ok || conn.Hostname == "" {
|
||||||
|
fmt.Println("Error: Invalid SSH connection arguments")
|
||||||
|
fmt.Println("Usage: sshm [-p port] [-i identity] [user@]hostname")
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Record the manual connection in history
|
||||||
|
historyManager, err := history.NewHistoryManager()
|
||||||
|
if err != nil {
|
||||||
|
// Log the error but don't prevent the connection
|
||||||
|
fmt.Printf("Warning: Could not initialize connection history: %v\n", err)
|
||||||
|
} else {
|
||||||
|
err = historyManager.RecordManualConnection(*conn)
|
||||||
|
if err != nil {
|
||||||
|
// Log the error but don't prevent the connection
|
||||||
|
fmt.Printf("Warning: Could not record connection history: %v\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build and execute the SSH command
|
||||||
|
fmt.Printf("Connecting to %s@%s:%s...\n", conn.User, conn.Hostname, conn.Port)
|
||||||
|
|
||||||
|
// Build SSH arguments
|
||||||
|
var sshArgs []string
|
||||||
|
|
||||||
|
// Add port if not default
|
||||||
|
if conn.Port != "" && conn.Port != "22" {
|
||||||
|
sshArgs = append(sshArgs, "-p", conn.Port)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add identity file if specified
|
||||||
|
if conn.Identity != "" {
|
||||||
|
sshArgs = append(sshArgs, "-i", conn.Identity)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add user@host or just host
|
||||||
|
if conn.User != "" {
|
||||||
|
sshArgs = append(sshArgs, fmt.Sprintf("%s@%s", conn.User, conn.Hostname))
|
||||||
|
} else {
|
||||||
|
sshArgs = append(sshArgs, conn.Hostname)
|
||||||
|
}
|
||||||
|
|
||||||
|
sshCmd := exec.Command("ssh", sshArgs...)
|
||||||
|
|
||||||
// Set up the command to use the same stdin, stdout, and stderr as the parent process
|
// Set up the command to use the same stdin, stdout, and stderr as the parent process
|
||||||
sshCmd.Stdin = os.Stdin
|
sshCmd.Stdin = os.Stdin
|
||||||
@@ -200,6 +265,29 @@ func getVersionWithUpdateCheck() string {
|
|||||||
|
|
||||||
// Execute adds all child commands to the root command and sets flags appropriately.
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
||||||
func Execute() {
|
func Execute() {
|
||||||
|
// Check if this looks like a manual SSH command BEFORE Cobra parses flags
|
||||||
|
// This prevents Cobra from complaining about unknown flags like -p, -i, etc.
|
||||||
|
if len(os.Args) > 1 {
|
||||||
|
// Check if any argument looks like a manual SSH connection
|
||||||
|
args := os.Args[1:]
|
||||||
|
|
||||||
|
// Skip if it's a known subcommand
|
||||||
|
knownCommands := []string{"add", "edit", "search", "move", "help", "completion", "version", "--version", "-v"}
|
||||||
|
isSubcommand := false
|
||||||
|
for _, cmd := range knownCommands {
|
||||||
|
if args[0] == cmd {
|
||||||
|
isSubcommand = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If not a subcommand and looks like manual SSH, handle it directly
|
||||||
|
if !isSubcommand && history.IsManualSSHCommand(args) {
|
||||||
|
connectManualSSH(args)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Custom error handling for unknown commands that might be host names
|
// Custom error handling for unknown commands that might be host names
|
||||||
if err := RootCmd.Execute(); err != nil {
|
if err := RootCmd.Execute(); err != nil {
|
||||||
// Check if this is an "unknown command" error and the argument might be a host name
|
// Check if this is an "unknown command" error and the argument might be a host name
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ USE_SUDO="false"
|
|||||||
OS=""
|
OS=""
|
||||||
ARCH=""
|
ARCH=""
|
||||||
FORCE_INSTALL="${FORCE_INSTALL:-false}"
|
FORCE_INSTALL="${FORCE_INSTALL:-false}"
|
||||||
|
SSHM_VERSION="${SSHM_VERSION:-latest}"
|
||||||
|
|
||||||
RED='\033[0;31m'
|
RED='\033[0;31m'
|
||||||
PURPLE='\033[0;35m'
|
PURPLE='\033[0;35m'
|
||||||
@@ -14,13 +15,27 @@ GREEN='\033[0;32m'
|
|||||||
YELLOW='\033[1;33m'
|
YELLOW='\033[1;33m'
|
||||||
NC='\033[0m'
|
NC='\033[0m'
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
printf "${PURPLE}SSHM Installation Script${NC}\n\n"
|
||||||
|
printf "Usage:\n"
|
||||||
|
printf " Default (latest stable): ${GREEN}bash install.sh${NC}\n"
|
||||||
|
printf " Specific version: ${GREEN}SSHM_VERSION=v1.8.0 bash install.sh${NC}\n"
|
||||||
|
printf " Beta/pre-release: ${GREEN}SSHM_VERSION=v1.8.1-beta bash install.sh${NC}\n"
|
||||||
|
printf " Force install: ${GREEN}FORCE_INSTALL=true bash install.sh${NC}\n"
|
||||||
|
printf " Custom install directory: ${GREEN}INSTALL_DIR=/opt/bin bash install.sh${NC}\n\n"
|
||||||
|
printf "Environment variables:\n"
|
||||||
|
printf " SSHM_VERSION - Version to install (default: latest)\n"
|
||||||
|
printf " FORCE_INSTALL - Skip confirmation prompts (default: false)\n"
|
||||||
|
printf " INSTALL_DIR - Installation directory (default: /usr/local/bin)\n\n"
|
||||||
|
}
|
||||||
|
|
||||||
setSystem() {
|
setSystem() {
|
||||||
ARCH=$(uname -m)
|
ARCH=$(uname -m)
|
||||||
case $ARCH in
|
case $ARCH in
|
||||||
i386|i686) ARCH="amd64" ;;
|
i386|i686) ARCH="amd64" ;;
|
||||||
x86_64) ARCH="amd64";;
|
x86_64) ARCH="amd64";;
|
||||||
armv6*) ARCH="arm64" ;;
|
armv6*) ARCH="armv6" ;;
|
||||||
armv7*) ARCH="arm64" ;;
|
armv7*) ARCH="armv7" ;;
|
||||||
aarch64*) ARCH="arm64" ;;
|
aarch64*) ARCH="arm64" ;;
|
||||||
arm64) ARCH="arm64" ;;
|
arm64) ARCH="arm64" ;;
|
||||||
esac
|
esac
|
||||||
@@ -46,17 +61,48 @@ runAsRoot() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
getLatestVersion() {
|
getLatestVersion() {
|
||||||
printf "${YELLOW}Fetching latest version...${NC}\n"
|
if [ "$SSHM_VERSION" = "latest" ]; then
|
||||||
|
printf "${YELLOW}Fetching latest stable version...${NC}\n"
|
||||||
LATEST_VERSION=$(curl -s https://api.github.com/repos/Gu1llaum-3/sshm/releases/latest | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
|
LATEST_VERSION=$(curl -s https://api.github.com/repos/Gu1llaum-3/sshm/releases/latest | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
|
||||||
if [ -z "$LATEST_VERSION" ]; then
|
if [ -z "$LATEST_VERSION" ]; then
|
||||||
printf "${RED}Failed to fetch latest version${NC}\n"
|
printf "${RED}Failed to fetch latest version${NC}\n"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
printf "${GREEN}Latest version: $LATEST_VERSION${NC}\n"
|
else
|
||||||
|
printf "${YELLOW}Using specified version: $SSHM_VERSION${NC}\n"
|
||||||
|
# Validate that the specified version exists
|
||||||
|
RELEASE_CHECK=$(curl -s "https://api.github.com/repos/Gu1llaum-3/sshm/releases/tags/$SSHM_VERSION" | grep '"tag_name":')
|
||||||
|
if [ -z "$RELEASE_CHECK" ]; then
|
||||||
|
printf "${RED}Version $SSHM_VERSION not found. Available versions:${NC}\n"
|
||||||
|
curl -s https://api.github.com/repos/Gu1llaum-3/sshm/releases | grep '"tag_name":' | head -10 | sed -E 's/.*"([^"]+)".*/ - \1/'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
LATEST_VERSION="$SSHM_VERSION"
|
||||||
|
fi
|
||||||
|
printf "${GREEN}Installing version: $LATEST_VERSION${NC}\n"
|
||||||
}
|
}
|
||||||
|
|
||||||
downloadBinary() {
|
downloadBinary() {
|
||||||
GITHUB_FILE="sshm-${OS}-${ARCH}.tar.gz"
|
# Map OS names to match GoReleaser format
|
||||||
|
local GORELEASER_OS="$OS"
|
||||||
|
case $OS in
|
||||||
|
"darwin") GORELEASER_OS="Darwin" ;;
|
||||||
|
"linux") GORELEASER_OS="Linux" ;;
|
||||||
|
"windows") GORELEASER_OS="Windows" ;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
# Map architecture names to match GoReleaser format
|
||||||
|
local GORELEASER_ARCH="$ARCH"
|
||||||
|
case $ARCH in
|
||||||
|
"amd64") GORELEASER_ARCH="x86_64" ;;
|
||||||
|
"arm64") GORELEASER_ARCH="arm64" ;;
|
||||||
|
"386") GORELEASER_ARCH="i386" ;;
|
||||||
|
"armv6") GORELEASER_ARCH="armv6" ;;
|
||||||
|
"armv7") GORELEASER_ARCH="armv7" ;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
# GoReleaser format: sshm_Linux_armv7.tar.gz
|
||||||
|
GITHUB_FILE="sshm_${GORELEASER_OS}_${GORELEASER_ARCH}.tar.gz"
|
||||||
GITHUB_URL="https://github.com/Gu1llaum-3/sshm/releases/download/$LATEST_VERSION/$GITHUB_FILE"
|
GITHUB_URL="https://github.com/Gu1llaum-3/sshm/releases/download/$LATEST_VERSION/$GITHUB_FILE"
|
||||||
|
|
||||||
printf "${YELLOW}Downloading $GITHUB_FILE...${NC}\n"
|
printf "${YELLOW}Downloading $GITHUB_FILE...${NC}\n"
|
||||||
@@ -74,8 +120,8 @@ downloadBinary() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Check if the expected binary exists (no find needed)
|
# GoReleaser extracts the binary as just "sshm", not with the platform suffix
|
||||||
EXTRACTED_BINARY="./sshm-${OS}-${ARCH}"
|
EXTRACTED_BINARY="./sshm"
|
||||||
if [ ! -f "$EXTRACTED_BINARY" ]; then
|
if [ ! -f "$EXTRACTED_BINARY" ]; then
|
||||||
printf "${RED}Could not find extracted binary: $EXTRACTED_BINARY${NC}\n"
|
printf "${RED}Could not find extracted binary: $EXTRACTED_BINARY${NC}\n"
|
||||||
exit 1
|
exit 1
|
||||||
@@ -158,18 +204,24 @@ checkExisting() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
main() {
|
main() {
|
||||||
printf "${PURPLE}Installing SSHM - SSH Connection Manager${NC}\n\n"
|
# Check for help argument
|
||||||
|
if [ "$1" = "-h" ] || [ "$1" = "--help" ] || [ "$1" = "help" ]; then
|
||||||
|
usage
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
# Check if already installed
|
printf "${PURPLE}Installing SSHM - SSH Connection Manager${NC}\n\n"
|
||||||
checkExisting
|
|
||||||
|
|
||||||
# Set up system detection
|
# Set up system detection
|
||||||
setSystem
|
setSystem
|
||||||
printf "${GREEN}Detected system: $OS ($ARCH)${NC}\n"
|
printf "${GREEN}Detected system: $OS ($ARCH)${NC}\n"
|
||||||
|
|
||||||
# Get latest version
|
# Get and validate version FIRST (this can fail early)
|
||||||
getLatestVersion
|
getLatestVersion
|
||||||
|
|
||||||
|
# Check if already installed (this might prompt user)
|
||||||
|
checkExisting
|
||||||
|
|
||||||
# Download and install
|
# Download and install
|
||||||
downloadBinary
|
downloadBinary
|
||||||
install
|
install
|
||||||
|
|||||||
@@ -80,7 +80,11 @@ if ($LocalBinary -ne "") {
|
|||||||
}
|
}
|
||||||
|
|
||||||
# Download binary
|
# Download binary
|
||||||
$fileName = "sshm-windows-$arch.zip"
|
# Map architecture to match GoReleaser format
|
||||||
|
$goreleaserArch = if ($arch -eq "amd64") { "x86_64" } else { "i386" }
|
||||||
|
|
||||||
|
# GoReleaser format: sshm_Windows_x86_64.zip
|
||||||
|
$fileName = "sshm_Windows_$goreleaserArch.zip"
|
||||||
$downloadUrl = "https://github.com/Gu1llaum-3/sshm/releases/download/$latestVersion/$fileName"
|
$downloadUrl = "https://github.com/Gu1llaum-3/sshm/releases/download/$latestVersion/$fileName"
|
||||||
$tempFile = "$env:TEMP\$fileName"
|
$tempFile = "$env:TEMP\$fileName"
|
||||||
|
|
||||||
@@ -101,7 +105,8 @@ if ($LocalBinary -ne "") {
|
|||||||
Write-Info "Extracting..."
|
Write-Info "Extracting..."
|
||||||
try {
|
try {
|
||||||
Expand-Archive -Path $tempFile -DestinationPath $env:TEMP -Force
|
Expand-Archive -Path $tempFile -DestinationPath $env:TEMP -Force
|
||||||
$extractedBinary = "$env:TEMP\sshm-windows-$arch.exe"
|
# GoReleaser extracts the binary as just "sshm.exe", not with platform suffix
|
||||||
|
$extractedBinary = "$env:TEMP\sshm.exe"
|
||||||
$targetPath = "$InstallDir\sshm.exe"
|
$targetPath = "$InstallDir\sshm.exe"
|
||||||
|
|
||||||
Move-Item -Path $extractedBinary -Destination $targetPath -Force
|
Move-Item -Path $extractedBinary -Destination $targetPath -Force
|
||||||
|
|||||||
146
internal/config/keybindings.go
Normal file
146
internal/config/keybindings.go
Normal file
@@ -0,0 +1,146 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// KeyBindings represents configurable key bindings for the application
|
||||||
|
type KeyBindings struct {
|
||||||
|
// Quit keys - keys that will quit the application
|
||||||
|
QuitKeys []string `json:"quit_keys"`
|
||||||
|
|
||||||
|
// DisableEscQuit - if true, ESC key won't quit the application (useful for vim users)
|
||||||
|
DisableEscQuit bool `json:"disable_esc_quit"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AppConfig represents the main application configuration
|
||||||
|
type AppConfig struct {
|
||||||
|
KeyBindings KeyBindings `json:"key_bindings"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetDefaultKeyBindings returns the default key bindings configuration
|
||||||
|
func GetDefaultKeyBindings() KeyBindings {
|
||||||
|
return KeyBindings{
|
||||||
|
QuitKeys: []string{"q", "ctrl+c"}, // Default keeps current behavior minus ESC
|
||||||
|
DisableEscQuit: false, // Default to false for backward compatibility
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetDefaultAppConfig returns the default application configuration
|
||||||
|
func GetDefaultAppConfig() AppConfig {
|
||||||
|
return AppConfig{
|
||||||
|
KeyBindings: GetDefaultKeyBindings(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetAppConfigPath returns the path to the application config file
|
||||||
|
func GetAppConfigPath() (string, error) {
|
||||||
|
configDir, err := GetSSHMConfigDir()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return filepath.Join(configDir, "config.json"), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadAppConfig loads the application configuration from file
|
||||||
|
// If the file doesn't exist, it returns the default configuration
|
||||||
|
func LoadAppConfig() (*AppConfig, error) {
|
||||||
|
configPath, err := GetAppConfigPath()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// If config file doesn't exist, return default config and create the file
|
||||||
|
if _, err := os.Stat(configPath); os.IsNotExist(err) {
|
||||||
|
defaultConfig := GetDefaultAppConfig()
|
||||||
|
|
||||||
|
// Create config directory if it doesn't exist
|
||||||
|
configDir := filepath.Dir(configPath)
|
||||||
|
if err := os.MkdirAll(configDir, 0755); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save default config to file
|
||||||
|
if err := SaveAppConfig(&defaultConfig); err != nil {
|
||||||
|
// If we can't save, just return the default config without erroring
|
||||||
|
// This allows the app to work even if config file can't be created
|
||||||
|
return &defaultConfig, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return &defaultConfig, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read existing config file
|
||||||
|
data, err := os.ReadFile(configPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var config AppConfig
|
||||||
|
if err := json.Unmarshal(data, &config); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate and fill in missing fields with defaults
|
||||||
|
config = mergeWithDefaults(config)
|
||||||
|
|
||||||
|
return &config, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveAppConfig saves the application configuration to file
|
||||||
|
func SaveAppConfig(config *AppConfig) error {
|
||||||
|
if config == nil {
|
||||||
|
return errors.New("config cannot be nil")
|
||||||
|
}
|
||||||
|
|
||||||
|
configPath, err := GetAppConfigPath()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create config directory if it doesn't exist
|
||||||
|
configDir := filepath.Dir(configPath)
|
||||||
|
if err := os.MkdirAll(configDir, 0755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := json.MarshalIndent(config, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return os.WriteFile(configPath, data, 0644)
|
||||||
|
}
|
||||||
|
|
||||||
|
// mergeWithDefaults ensures all required fields are set with defaults if missing
|
||||||
|
func mergeWithDefaults(config AppConfig) AppConfig {
|
||||||
|
defaults := GetDefaultAppConfig()
|
||||||
|
|
||||||
|
// If QuitKeys is empty, use defaults
|
||||||
|
if len(config.KeyBindings.QuitKeys) == 0 {
|
||||||
|
config.KeyBindings.QuitKeys = defaults.KeyBindings.QuitKeys
|
||||||
|
}
|
||||||
|
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
// ShouldQuitOnKey checks if the given key should trigger quit based on configuration
|
||||||
|
func (kb *KeyBindings) ShouldQuitOnKey(key string) bool {
|
||||||
|
// Special handling for ESC key
|
||||||
|
if key == "esc" {
|
||||||
|
return !kb.DisableEscQuit
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if key is in the quit keys list
|
||||||
|
for _, quitKey := range kb.QuitKeys {
|
||||||
|
if quitKey == key {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
181
internal/config/keybindings_test.go
Normal file
181
internal/config/keybindings_test.go
Normal file
@@ -0,0 +1,181 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestDefaultKeyBindings(t *testing.T) {
|
||||||
|
kb := GetDefaultKeyBindings()
|
||||||
|
|
||||||
|
// Test default configuration
|
||||||
|
if kb.DisableEscQuit {
|
||||||
|
t.Error("Default configuration should allow ESC to quit (backward compatibility)")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test default quit keys
|
||||||
|
expectedQuitKeys := []string{"q", "ctrl+c"}
|
||||||
|
if len(kb.QuitKeys) != len(expectedQuitKeys) {
|
||||||
|
t.Errorf("Expected %d quit keys, got %d", len(expectedQuitKeys), len(kb.QuitKeys))
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, expected := range expectedQuitKeys {
|
||||||
|
if i >= len(kb.QuitKeys) || kb.QuitKeys[i] != expected {
|
||||||
|
t.Errorf("Expected quit key %s, got %s", expected, kb.QuitKeys[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestShouldQuitOnKey(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
keyBindings KeyBindings
|
||||||
|
key string
|
||||||
|
expectedResult bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Default config - ESC should quit",
|
||||||
|
keyBindings: KeyBindings{
|
||||||
|
QuitKeys: []string{"q", "ctrl+c"},
|
||||||
|
DisableEscQuit: false,
|
||||||
|
},
|
||||||
|
key: "esc",
|
||||||
|
expectedResult: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Disabled ESC quit - ESC should not quit",
|
||||||
|
keyBindings: KeyBindings{
|
||||||
|
QuitKeys: []string{"q", "ctrl+c"},
|
||||||
|
DisableEscQuit: true,
|
||||||
|
},
|
||||||
|
key: "esc",
|
||||||
|
expectedResult: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Q key should quit",
|
||||||
|
keyBindings: KeyBindings{
|
||||||
|
QuitKeys: []string{"q", "ctrl+c"},
|
||||||
|
DisableEscQuit: true,
|
||||||
|
},
|
||||||
|
key: "q",
|
||||||
|
expectedResult: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Ctrl+C should quit",
|
||||||
|
keyBindings: KeyBindings{
|
||||||
|
QuitKeys: []string{"q", "ctrl+c"},
|
||||||
|
DisableEscQuit: true,
|
||||||
|
},
|
||||||
|
key: "ctrl+c",
|
||||||
|
expectedResult: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Other keys should not quit",
|
||||||
|
keyBindings: KeyBindings{
|
||||||
|
QuitKeys: []string{"q", "ctrl+c"},
|
||||||
|
DisableEscQuit: true,
|
||||||
|
},
|
||||||
|
key: "enter",
|
||||||
|
expectedResult: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
result := tt.keyBindings.ShouldQuitOnKey(tt.key)
|
||||||
|
if result != tt.expectedResult {
|
||||||
|
t.Errorf("ShouldQuitOnKey(%q) = %v, expected %v", tt.key, result, tt.expectedResult)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAppConfigBasics(t *testing.T) {
|
||||||
|
// Test default config creation
|
||||||
|
defaultConfig := GetDefaultAppConfig()
|
||||||
|
|
||||||
|
if defaultConfig.KeyBindings.DisableEscQuit {
|
||||||
|
t.Error("Default configuration should allow ESC to quit")
|
||||||
|
}
|
||||||
|
|
||||||
|
expectedQuitKeys := []string{"q", "ctrl+c"}
|
||||||
|
if len(defaultConfig.KeyBindings.QuitKeys) != len(expectedQuitKeys) {
|
||||||
|
t.Errorf("Expected %d quit keys, got %d", len(expectedQuitKeys), len(defaultConfig.KeyBindings.QuitKeys))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMergeWithDefaults(t *testing.T) {
|
||||||
|
// Test config with missing QuitKeys
|
||||||
|
incompleteConfig := AppConfig{
|
||||||
|
KeyBindings: KeyBindings{
|
||||||
|
DisableEscQuit: true,
|
||||||
|
// QuitKeys is missing
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
mergedConfig := mergeWithDefaults(incompleteConfig)
|
||||||
|
|
||||||
|
// Should preserve DisableEscQuit
|
||||||
|
if !mergedConfig.KeyBindings.DisableEscQuit {
|
||||||
|
t.Error("Should preserve DisableEscQuit as true")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should fill in default QuitKeys
|
||||||
|
expectedQuitKeys := []string{"q", "ctrl+c"}
|
||||||
|
if len(mergedConfig.KeyBindings.QuitKeys) != len(expectedQuitKeys) {
|
||||||
|
t.Errorf("Expected %d quit keys, got %d", len(expectedQuitKeys), len(mergedConfig.KeyBindings.QuitKeys))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSaveAndLoadAppConfigIntegration(t *testing.T) {
|
||||||
|
// Create a temporary directory for testing
|
||||||
|
tempDir, err := os.MkdirTemp("", "sshm_test")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create temp directory: %v", err)
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(tempDir)
|
||||||
|
|
||||||
|
// Create a custom config file directly in temp directory
|
||||||
|
configPath := filepath.Join(tempDir, "config.json")
|
||||||
|
|
||||||
|
customConfig := AppConfig{
|
||||||
|
KeyBindings: KeyBindings{
|
||||||
|
QuitKeys: []string{"q"},
|
||||||
|
DisableEscQuit: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save config directly to file
|
||||||
|
data, err := json.MarshalIndent(customConfig, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to marshal config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.WriteFile(configPath, data, 0644)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to write config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read and unmarshal config
|
||||||
|
readData, err := os.ReadFile(configPath)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to read config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var loadedConfig AppConfig
|
||||||
|
err = json.Unmarshal(readData, &loadedConfig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to unmarshal config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify the loaded config matches what we saved
|
||||||
|
if !loadedConfig.KeyBindings.DisableEscQuit {
|
||||||
|
t.Error("DisableEscQuit should be true")
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(loadedConfig.KeyBindings.QuitKeys) != 1 || loadedConfig.KeyBindings.QuitKeys[0] != "q" {
|
||||||
|
t.Errorf("Expected quit keys to be ['q'], got %v", loadedConfig.KeyBindings.QuitKeys)
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -987,3 +987,710 @@ func TestMoveHostToFile(t *testing.T) {
|
|||||||
// Test that the component functions work for the move operation
|
// Test that the component functions work for the move operation
|
||||||
t.Log("MoveHostToFile() error handling works correctly")
|
t.Log("MoveHostToFile() error handling works correctly")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestParseSSHConfigWithMultipleHostsOnSameLine(t *testing.T) {
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
|
||||||
|
configFile := filepath.Join(tempDir, "config")
|
||||||
|
configContent := `# Test multiple hosts on same line
|
||||||
|
Host local1 local2
|
||||||
|
HostName ::1
|
||||||
|
User myuser
|
||||||
|
|
||||||
|
Host root-server
|
||||||
|
User root
|
||||||
|
HostName root.example.com
|
||||||
|
|
||||||
|
Host web1 web2 web3
|
||||||
|
HostName ::1
|
||||||
|
User webuser
|
||||||
|
Port 8080
|
||||||
|
|
||||||
|
Host single-host
|
||||||
|
HostName single.example.com
|
||||||
|
User singleuser
|
||||||
|
`
|
||||||
|
|
||||||
|
err := os.WriteFile(configFile, []byte(configContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
hosts, err := ParseSSHConfigFile(configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ParseSSHConfigFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should get 7 hosts: local1, local2, root-server, web1, web2, web3, single-host
|
||||||
|
expectedHosts := map[string]struct{}{
|
||||||
|
"local1": {},
|
||||||
|
"local2": {},
|
||||||
|
"root-server": {},
|
||||||
|
"web1": {},
|
||||||
|
"web2": {},
|
||||||
|
"web3": {},
|
||||||
|
"single-host": {},
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(hosts) != len(expectedHosts) {
|
||||||
|
t.Errorf("Expected %d hosts, got %d", len(expectedHosts), len(hosts))
|
||||||
|
for _, host := range hosts {
|
||||||
|
t.Logf("Found host: %s", host.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
hostMap := make(map[string]SSHHost)
|
||||||
|
for _, host := range hosts {
|
||||||
|
hostMap[host.Name] = host
|
||||||
|
}
|
||||||
|
|
||||||
|
for expectedHostName := range expectedHosts {
|
||||||
|
if _, found := hostMap[expectedHostName]; !found {
|
||||||
|
t.Errorf("Expected host %s not found", expectedHostName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify properties based on host name
|
||||||
|
if host, found := hostMap["local1"]; found {
|
||||||
|
if host.Hostname != "::1" || host.User != "myuser" {
|
||||||
|
t.Errorf("local1 properties incorrect: hostname=%s, user=%s", host.Hostname, host.User)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["local2"]; found {
|
||||||
|
if host.Hostname != "::1" || host.User != "myuser" {
|
||||||
|
t.Errorf("local2 properties incorrect: hostname=%s, user=%s", host.Hostname, host.User)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["web1"]; found {
|
||||||
|
if host.Hostname != "::1" || host.User != "webuser" || host.Port != "8080" {
|
||||||
|
t.Errorf("web1 properties incorrect: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["web2"]; found {
|
||||||
|
if host.Hostname != "::1" || host.User != "webuser" || host.Port != "8080" {
|
||||||
|
t.Errorf("web2 properties incorrect: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["web3"]; found {
|
||||||
|
if host.Hostname != "::1" || host.User != "webuser" || host.Port != "8080" {
|
||||||
|
t.Errorf("web3 properties incorrect: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["root-server"]; found {
|
||||||
|
if host.User != "root" || host.Hostname != "root.example.com" {
|
||||||
|
t.Errorf("root-server properties incorrect: user=%s, hostname=%s", host.User, host.Hostname)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateSSHHostInFileWithMultiHost(t *testing.T) {
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
|
||||||
|
configFile := filepath.Join(tempDir, "config")
|
||||||
|
configContent := `# Test config with multi-host
|
||||||
|
Host web1 web2 web3
|
||||||
|
HostName webserver.example.com
|
||||||
|
User webuser
|
||||||
|
Port 2222
|
||||||
|
|
||||||
|
Host database
|
||||||
|
HostName db.example.com
|
||||||
|
User dbuser
|
||||||
|
`
|
||||||
|
|
||||||
|
err := os.WriteFile(configFile, []byte(configContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update web2 in the multi-host line
|
||||||
|
newHost := SSHHost{
|
||||||
|
Name: "web2-updated",
|
||||||
|
Hostname: "newweb.example.com",
|
||||||
|
User: "newuser",
|
||||||
|
Port: "22",
|
||||||
|
}
|
||||||
|
|
||||||
|
err = UpdateSSHHostInFile("web2", newHost, configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("UpdateSSHHostInFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the updated config
|
||||||
|
hosts, err := ParseSSHConfigFile(configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ParseSSHConfigFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should have: web1, web3, web2-updated, database
|
||||||
|
expectedHosts := []string{"web1", "web3", "web2-updated", "database"}
|
||||||
|
|
||||||
|
hostMap := make(map[string]SSHHost)
|
||||||
|
for _, host := range hosts {
|
||||||
|
hostMap[host.Name] = host
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(hosts) != len(expectedHosts) {
|
||||||
|
t.Errorf("Expected %d hosts, got %d", len(expectedHosts), len(hosts))
|
||||||
|
for _, host := range hosts {
|
||||||
|
t.Logf("Found host: %s", host.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, expectedHostName := range expectedHosts {
|
||||||
|
if _, found := hostMap[expectedHostName]; !found {
|
||||||
|
t.Errorf("Expected host %s not found", expectedHostName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify web1 and web3 still have original properties
|
||||||
|
if host, found := hostMap["web1"]; found {
|
||||||
|
if host.Hostname != "webserver.example.com" || host.User != "webuser" || host.Port != "2222" {
|
||||||
|
t.Errorf("web1 properties changed: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["web3"]; found {
|
||||||
|
if host.Hostname != "webserver.example.com" || host.User != "webuser" || host.Port != "2222" {
|
||||||
|
t.Errorf("web3 properties changed: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify web2-updated has new properties
|
||||||
|
if host, found := hostMap["web2-updated"]; found {
|
||||||
|
if host.Hostname != "newweb.example.com" || host.User != "newuser" || host.Port != "22" {
|
||||||
|
t.Errorf("web2-updated properties incorrect: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify database is unchanged
|
||||||
|
if host, found := hostMap["database"]; found {
|
||||||
|
if host.Hostname != "db.example.com" || host.User != "dbuser" {
|
||||||
|
t.Errorf("database properties changed: hostname=%s, user=%s", host.Hostname, host.User)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIsPartOfMultiHostDeclaration(t *testing.T) {
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
|
||||||
|
configFile := filepath.Join(tempDir, "config")
|
||||||
|
configContent := `Host single
|
||||||
|
HostName single.example.com
|
||||||
|
|
||||||
|
Host multi1 multi2 multi3
|
||||||
|
HostName multi.example.com
|
||||||
|
|
||||||
|
Host another
|
||||||
|
HostName another.example.com
|
||||||
|
`
|
||||||
|
|
||||||
|
err := os.WriteFile(configFile, []byte(configContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
hostName string
|
||||||
|
expectedMulti bool
|
||||||
|
expectedHosts []string
|
||||||
|
}{
|
||||||
|
{"single", false, []string{"single"}},
|
||||||
|
{"multi1", true, []string{"multi1", "multi2", "multi3"}},
|
||||||
|
{"multi2", true, []string{"multi1", "multi2", "multi3"}},
|
||||||
|
{"multi3", true, []string{"multi1", "multi2", "multi3"}},
|
||||||
|
{"another", false, []string{"another"}},
|
||||||
|
{"nonexistent", false, nil},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.hostName, func(t *testing.T) {
|
||||||
|
isMulti, hostNames, err := IsPartOfMultiHostDeclaration(tt.hostName, configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("IsPartOfMultiHostDeclaration() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if isMulti != tt.expectedMulti {
|
||||||
|
t.Errorf("Expected isMulti=%v, got %v", tt.expectedMulti, isMulti)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tt.expectedHosts == nil && hostNames != nil {
|
||||||
|
t.Errorf("Expected hostNames to be nil, got %v", hostNames)
|
||||||
|
} else if tt.expectedHosts != nil {
|
||||||
|
if len(hostNames) != len(tt.expectedHosts) {
|
||||||
|
t.Errorf("Expected %d hostNames, got %d", len(tt.expectedHosts), len(hostNames))
|
||||||
|
} else {
|
||||||
|
for i, expectedHost := range tt.expectedHosts {
|
||||||
|
if i < len(hostNames) && hostNames[i] != expectedHost {
|
||||||
|
t.Errorf("Expected hostNames[%d]=%s, got %s", i, expectedHost, hostNames[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteSSHHostFromFileWithMultiHost(t *testing.T) {
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
|
||||||
|
configFile := filepath.Join(tempDir, "config")
|
||||||
|
configContent := `# Test config with multi-host deletion
|
||||||
|
Host web1 web2 web3
|
||||||
|
HostName webserver.example.com
|
||||||
|
User webuser
|
||||||
|
Port 2222
|
||||||
|
|
||||||
|
Host database
|
||||||
|
HostName db.example.com
|
||||||
|
User dbuser
|
||||||
|
|
||||||
|
# Tags: production, critical
|
||||||
|
Host app1 app2
|
||||||
|
HostName appserver.example.com
|
||||||
|
User appuser
|
||||||
|
`
|
||||||
|
|
||||||
|
err := os.WriteFile(configFile, []byte(configContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test 1: Delete one host from multi-host block (should keep others)
|
||||||
|
err = DeleteSSHHostFromFile("web2", configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("DeleteSSHHostFromFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the updated config
|
||||||
|
hosts, err := ParseSSHConfigFile(configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ParseSSHConfigFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should have: web1, web3, database, app1, app2 (web2 removed)
|
||||||
|
expectedHosts := []string{"web1", "web3", "database", "app1", "app2"}
|
||||||
|
|
||||||
|
hostMap := make(map[string]SSHHost)
|
||||||
|
for _, host := range hosts {
|
||||||
|
hostMap[host.Name] = host
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(hosts) != len(expectedHosts) {
|
||||||
|
t.Errorf("Expected %d hosts, got %d", len(expectedHosts), len(hosts))
|
||||||
|
for _, host := range hosts {
|
||||||
|
t.Logf("Found host: %s", host.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, expectedHostName := range expectedHosts {
|
||||||
|
if _, found := hostMap[expectedHostName]; !found {
|
||||||
|
t.Errorf("Expected host %s not found", expectedHostName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify web2 is not present
|
||||||
|
if _, found := hostMap["web2"]; found {
|
||||||
|
t.Error("web2 should have been deleted")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify web1 and web3 still have original properties
|
||||||
|
if host, found := hostMap["web1"]; found {
|
||||||
|
if host.Hostname != "webserver.example.com" || host.User != "webuser" || host.Port != "2222" {
|
||||||
|
t.Errorf("web1 properties incorrect: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if host, found := hostMap["web3"]; found {
|
||||||
|
if host.Hostname != "webserver.example.com" || host.User != "webuser" || host.Port != "2222" {
|
||||||
|
t.Errorf("web3 properties incorrect: hostname=%s, user=%s, port=%s", host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test 2: Delete one host from multi-host block with tags
|
||||||
|
err = DeleteSSHHostFromFile("app1", configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("DeleteSSHHostFromFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse again
|
||||||
|
hosts, err = ParseSSHConfigFile(configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ParseSSHConfigFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should have: web1, web3, database, app2 (app1 removed)
|
||||||
|
expectedHosts = []string{"web1", "web3", "database", "app2"}
|
||||||
|
|
||||||
|
hostMap = make(map[string]SSHHost)
|
||||||
|
for _, host := range hosts {
|
||||||
|
hostMap[host.Name] = host
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(hosts) != len(expectedHosts) {
|
||||||
|
t.Errorf("Expected %d hosts, got %d", len(expectedHosts), len(hosts))
|
||||||
|
for _, host := range hosts {
|
||||||
|
t.Logf("Found host: %s", host.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify app2 still has tags
|
||||||
|
if host, found := hostMap["app2"]; found {
|
||||||
|
if !contains(host.Tags, "production") || !contains(host.Tags, "critical") {
|
||||||
|
t.Errorf("app2 tags incorrect: %v", host.Tags)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateMultiHostBlock(t *testing.T) {
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
|
||||||
|
configFile := filepath.Join(tempDir, "config")
|
||||||
|
configContent := `# Test config for multi-host block update
|
||||||
|
Host server1 server2 server3
|
||||||
|
HostName cluster.example.com
|
||||||
|
User clusteruser
|
||||||
|
Port 2222
|
||||||
|
|
||||||
|
Host single
|
||||||
|
HostName single.example.com
|
||||||
|
User singleuser
|
||||||
|
`
|
||||||
|
|
||||||
|
err := os.WriteFile(configFile, []byte(configContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update the multi-host block
|
||||||
|
originalHosts := []string{"server1", "server2", "server3"}
|
||||||
|
newHosts := []string{"server1", "server4", "server5"} // Remove server2, server3 and add server4, server5
|
||||||
|
commonProperties := SSHHost{
|
||||||
|
Hostname: "newcluster.example.com",
|
||||||
|
User: "newuser",
|
||||||
|
Port: "22",
|
||||||
|
Tags: []string{"updated", "cluster"},
|
||||||
|
}
|
||||||
|
|
||||||
|
err = UpdateMultiHostBlock(originalHosts, newHosts, commonProperties, configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("UpdateMultiHostBlock() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the updated config
|
||||||
|
hosts, err := ParseSSHConfigFile(configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ParseSSHConfigFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should have: server1, server4, server5, single
|
||||||
|
expectedHosts := []string{"server1", "server4", "server5", "single"}
|
||||||
|
|
||||||
|
hostMap := make(map[string]SSHHost)
|
||||||
|
for _, host := range hosts {
|
||||||
|
hostMap[host.Name] = host
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(hosts) != len(expectedHosts) {
|
||||||
|
t.Errorf("Expected %d hosts, got %d", len(expectedHosts), len(hosts))
|
||||||
|
for _, host := range hosts {
|
||||||
|
t.Logf("Found host: %s", host.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify new hosts have updated properties
|
||||||
|
for _, hostName := range []string{"server1", "server4", "server5"} {
|
||||||
|
if host, found := hostMap[hostName]; found {
|
||||||
|
if host.Hostname != "newcluster.example.com" || host.User != "newuser" || host.Port != "22" {
|
||||||
|
t.Errorf("%s properties incorrect: hostname=%s, user=%s, port=%s",
|
||||||
|
hostName, host.Hostname, host.User, host.Port)
|
||||||
|
}
|
||||||
|
if !contains(host.Tags, "updated") || !contains(host.Tags, "cluster") {
|
||||||
|
t.Errorf("%s tags incorrect: %v", hostName, host.Tags)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
t.Errorf("Expected host %s not found", hostName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify single host is unchanged
|
||||||
|
if host, found := hostMap["single"]; found {
|
||||||
|
if host.Hostname != "single.example.com" || host.User != "singleuser" {
|
||||||
|
t.Errorf("single host properties changed: hostname=%s, user=%s", host.Hostname, host.User)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify old hosts are gone
|
||||||
|
for _, oldHost := range []string{"server2", "server3"} {
|
||||||
|
if _, found := hostMap[oldHost]; found {
|
||||||
|
t.Errorf("Old host %s should have been removed", oldHost)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper function to check if slice contains a string
|
||||||
|
func contains(slice []string, item string) bool {
|
||||||
|
for _, s := range slice {
|
||||||
|
if s == item {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper function to create temporary config files for testing
|
||||||
|
func createTempConfigFile(content string) (string, error) {
|
||||||
|
tempFile, err := os.CreateTemp("", "ssh_config_test_*.conf")
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer tempFile.Close()
|
||||||
|
|
||||||
|
_, err = tempFile.WriteString(content)
|
||||||
|
if err != nil {
|
||||||
|
os.Remove(tempFile.Name())
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return tempFile.Name(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFormatSSHConfigValue(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "simple path without spaces",
|
||||||
|
input: "/home/user/.ssh/id_rsa",
|
||||||
|
expected: "/home/user/.ssh/id_rsa",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "path with spaces",
|
||||||
|
input: "/home/user/My Documents/ssh key",
|
||||||
|
expected: "\"/home/user/My Documents/ssh key\"",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Windows path with spaces",
|
||||||
|
input: `G:\My Drive\7 - Tech\9 - SSH Keys\Server_WF.opk`,
|
||||||
|
expected: `"G:\My Drive\7 - Tech\9 - SSH Keys\Server_WF.opk"`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "path with quotes but no spaces",
|
||||||
|
input: `/home/user/key"with"quotes`,
|
||||||
|
expected: `/home/user/key"with"quotes`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "path with spaces and quotes",
|
||||||
|
input: `/home/user/key "with" quotes`,
|
||||||
|
expected: `"/home/user/key "with" quotes"`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty path",
|
||||||
|
input: "",
|
||||||
|
expected: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "path with single space at end",
|
||||||
|
input: "/home/user/key ",
|
||||||
|
expected: "\"/home/user/key \"",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
result := formatSSHConfigValue(tt.input)
|
||||||
|
if result != tt.expected {
|
||||||
|
t.Errorf("formatSSHConfigValue(%q) = %q, want %q", tt.input, result, tt.expected)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAddSSHHostWithSpacesInPath(t *testing.T) {
|
||||||
|
// Create temporary config file
|
||||||
|
configFile, err := createTempConfigFile(`Host existing
|
||||||
|
HostName existing.com
|
||||||
|
`)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config file: %v", err)
|
||||||
|
}
|
||||||
|
defer os.Remove(configFile)
|
||||||
|
|
||||||
|
// Test adding host with path containing spaces
|
||||||
|
host := SSHHost{
|
||||||
|
Name: "test-spaces",
|
||||||
|
Hostname: "test.com",
|
||||||
|
User: "testuser",
|
||||||
|
Identity: "/path/with spaces/key file",
|
||||||
|
}
|
||||||
|
|
||||||
|
err = AddSSHHostToFile(host, configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("AddSSHHostToFile failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read the file and verify quotes are added
|
||||||
|
content, err := os.ReadFile(configFile)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to read config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
contentStr := string(content)
|
||||||
|
expectedIdentityLine := ` IdentityFile "/path/with spaces/key file"`
|
||||||
|
if !strings.Contains(contentStr, expectedIdentityLine) {
|
||||||
|
t.Errorf("Expected identity file line with quotes not found.\nContent:\n%s\nExpected line: %s", contentStr, expectedIdentityLine)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIsNonSSHConfigFile(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
fileName string
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
// Should be excluded
|
||||||
|
{"README", true},
|
||||||
|
{"README.txt", true},
|
||||||
|
{"README.md", true},
|
||||||
|
{"script.sh", true},
|
||||||
|
{"data.json", true},
|
||||||
|
{"notes.txt", true},
|
||||||
|
{".gitignore", true},
|
||||||
|
{"backup.bak", true},
|
||||||
|
{"old.orig", true},
|
||||||
|
{"log.log", true},
|
||||||
|
{"temp.tmp", true},
|
||||||
|
{"archive.zip", true},
|
||||||
|
{"image.jpg", true},
|
||||||
|
{"python.py", true},
|
||||||
|
{"golang.go", true},
|
||||||
|
{"config.yaml", true},
|
||||||
|
{"config.yml", true},
|
||||||
|
{"config.toml", true},
|
||||||
|
|
||||||
|
// Should NOT be excluded (valid SSH config files)
|
||||||
|
{"config", false},
|
||||||
|
{"servers.conf", false},
|
||||||
|
{"production", false},
|
||||||
|
{"staging", false},
|
||||||
|
{"hosts", false},
|
||||||
|
{"ssh_config", false},
|
||||||
|
{"work-servers", false},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create a temporary file for content testing
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
filePath := filepath.Join(tempDir, test.fileName)
|
||||||
|
|
||||||
|
// Write appropriate content based on expected result
|
||||||
|
var content string
|
||||||
|
if test.expected {
|
||||||
|
// Write non-SSH content for files that should be excluded
|
||||||
|
content = "# This is not an SSH config file\nSome random content"
|
||||||
|
} else {
|
||||||
|
// Write SSH-like content for files that should be included
|
||||||
|
content = "Host example\n HostName example.com\n User testuser"
|
||||||
|
}
|
||||||
|
|
||||||
|
err := os.WriteFile(filePath, []byte(content), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create test file %s: %v", test.fileName, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
result := isNonSSHConfigFile(filePath)
|
||||||
|
if result != test.expected {
|
||||||
|
t.Errorf("isNonSSHConfigFile(%q) = %v, want %v", test.fileName, result, test.expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQuickHostExists(t *testing.T) {
|
||||||
|
// Create temporary directory for test files
|
||||||
|
tempDir := t.TempDir()
|
||||||
|
|
||||||
|
// Create main config file
|
||||||
|
mainConfig := filepath.Join(tempDir, "config")
|
||||||
|
mainConfigContent := `Host main-host
|
||||||
|
HostName example.com
|
||||||
|
|
||||||
|
Include config.d/*
|
||||||
|
|
||||||
|
Host another-host
|
||||||
|
HostName another.example.com
|
||||||
|
`
|
||||||
|
|
||||||
|
err := os.WriteFile(mainConfig, []byte(mainConfigContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create main config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create config.d directory
|
||||||
|
configDir := filepath.Join(tempDir, "config.d")
|
||||||
|
err = os.MkdirAll(configDir, 0700)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create config.d: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create valid SSH config file in config.d
|
||||||
|
validConfig := filepath.Join(configDir, "servers.conf")
|
||||||
|
validConfigContent := `Host included-host
|
||||||
|
HostName included.example.com
|
||||||
|
User includeduser
|
||||||
|
|
||||||
|
Host production-server
|
||||||
|
HostName prod.example.com
|
||||||
|
User produser
|
||||||
|
`
|
||||||
|
|
||||||
|
err = os.WriteFile(validConfig, []byte(validConfigContent), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create valid config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create files that should be excluded (README, etc.)
|
||||||
|
excludedFiles := map[string]string{
|
||||||
|
"README": "# This is a README file\nDocumentation goes here",
|
||||||
|
"README.md": "# SSH Configuration\nThis directory contains...",
|
||||||
|
"script.sh": "#!/bin/bash\necho 'hello world'",
|
||||||
|
"data.json": `{"key": "value"}`,
|
||||||
|
}
|
||||||
|
|
||||||
|
for fileName, content := range excludedFiles {
|
||||||
|
filePath := filepath.Join(configDir, fileName)
|
||||||
|
err = os.WriteFile(filePath, []byte(content), 0600)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create %s: %v", fileName, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test hosts that should be found
|
||||||
|
existingHosts := []string{"main-host", "another-host", "included-host", "production-server"}
|
||||||
|
for _, hostName := range existingHosts {
|
||||||
|
found, err := QuickHostExistsInFile(hostName, mainConfig)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("QuickHostExistsInFile(%q) error = %v", hostName, err)
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
t.Errorf("QuickHostExistsInFile(%q) = false, want true", hostName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test hosts that should NOT be found
|
||||||
|
nonExistingHosts := []string{"nonexistent-host", "fake-server", "unknown"}
|
||||||
|
for _, hostName := range nonExistingHosts {
|
||||||
|
found, err := QuickHostExistsInFile(hostName, mainConfig)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("QuickHostExistsInFile(%q) error = %v", hostName, err)
|
||||||
|
}
|
||||||
|
if found {
|
||||||
|
t.Errorf("QuickHostExistsInFile(%q) = true, want false", hostName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package history
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"sort"
|
"sort"
|
||||||
@@ -306,3 +307,99 @@ func (hm *HistoryManager) GetPortForwardingConfig(hostName string) *PortForwardC
|
|||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ManualConnection represents a manual SSH connection (e.g., ssh user@host -p 2222)
|
||||||
|
type ManualConnection struct {
|
||||||
|
User string
|
||||||
|
Hostname string
|
||||||
|
Port string
|
||||||
|
Identity string
|
||||||
|
}
|
||||||
|
|
||||||
|
// RecordManualConnection records a manual SSH connection (like ssh user@host -p 2222 -i key)
|
||||||
|
// These are stored with a generated host name like "manual:user@host:port"
|
||||||
|
func (hm *HistoryManager) RecordManualConnection(conn ManualConnection) error {
|
||||||
|
// Generate a unique identifier for this manual connection
|
||||||
|
hostID := generateManualHostID(conn)
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
|
||||||
|
if existingConn, exists := hm.history.Connections[hostID]; exists {
|
||||||
|
// Update existing connection
|
||||||
|
existingConn.LastConnect = now
|
||||||
|
existingConn.ConnectCount++
|
||||||
|
hm.history.Connections[hostID] = existingConn
|
||||||
|
} else {
|
||||||
|
// Create new connection record
|
||||||
|
hm.history.Connections[hostID] = ConnectionInfo{
|
||||||
|
HostName: hostID,
|
||||||
|
LastConnect: now,
|
||||||
|
ConnectCount: 1,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return hm.saveHistory()
|
||||||
|
}
|
||||||
|
|
||||||
|
// generateManualHostID generates a unique ID for manual connections
|
||||||
|
func generateManualHostID(conn ManualConnection) string {
|
||||||
|
// Format: manual:user@hostname:port
|
||||||
|
user := conn.User
|
||||||
|
if user == "" {
|
||||||
|
user = "default"
|
||||||
|
}
|
||||||
|
port := conn.Port
|
||||||
|
if port == "" {
|
||||||
|
port = "22"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("manual:%s@%s:%s", user, conn.Hostname, port)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsManualConnection checks if a hostname represents a manual connection
|
||||||
|
func IsManualConnection(hostName string) bool {
|
||||||
|
return len(hostName) > 7 && hostName[:7] == "manual:"
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseManualConnectionID parses a manual connection ID back into its components
|
||||||
|
func ParseManualConnectionID(hostID string) (user, hostname, port string, ok bool) {
|
||||||
|
if !IsManualConnection(hostID) {
|
||||||
|
return "", "", "", false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove "manual:" prefix
|
||||||
|
parts := hostID[7:] // Skip "manual:"
|
||||||
|
|
||||||
|
// Split by last ':'
|
||||||
|
lastColon := -1
|
||||||
|
for i := len(parts) - 1; i >= 0; i-- {
|
||||||
|
if parts[i] == ':' {
|
||||||
|
lastColon = i
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if lastColon == -1 {
|
||||||
|
return "", "", "", false
|
||||||
|
}
|
||||||
|
|
||||||
|
port = parts[lastColon+1:]
|
||||||
|
userHost := parts[:lastColon]
|
||||||
|
|
||||||
|
// Split user@host
|
||||||
|
atSign := -1
|
||||||
|
for i := 0; i < len(userHost); i++ {
|
||||||
|
if userHost[i] == '@' {
|
||||||
|
atSign = i
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if atSign == -1 {
|
||||||
|
return "", "", "", false
|
||||||
|
}
|
||||||
|
|
||||||
|
user = userHost[:atSign]
|
||||||
|
hostname = userHost[atSign+1:]
|
||||||
|
|
||||||
|
return user, hostname, port, true
|
||||||
|
}
|
||||||
|
|||||||
95
internal/history/parser.go
Normal file
95
internal/history/parser.go
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
package history
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os/user"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ParseSSHArgs parses SSH command line arguments and extracts connection details
|
||||||
|
// It handles formats like: user@host, -p port, -i identity, etc.
|
||||||
|
func ParseSSHArgs(args []string) (*ManualConnection, bool) {
|
||||||
|
if len(args) == 0 {
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
|
conn := &ManualConnection{
|
||||||
|
Port: "22", // Default SSH port
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get current user as default
|
||||||
|
currentUser, err := user.Current()
|
||||||
|
if err == nil {
|
||||||
|
conn.User = currentUser.Username
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse arguments
|
||||||
|
for i := 0; i < len(args); i++ {
|
||||||
|
arg := args[i]
|
||||||
|
|
||||||
|
// Handle -p <port> or -p<port>
|
||||||
|
if arg == "-p" {
|
||||||
|
if i+1 < len(args) {
|
||||||
|
conn.Port = args[i+1]
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
} else if strings.HasPrefix(arg, "-p") {
|
||||||
|
conn.Port = arg[2:]
|
||||||
|
} else if arg == "-i" {
|
||||||
|
// Handle -i <identity>
|
||||||
|
if i+1 < len(args) {
|
||||||
|
conn.Identity = args[i+1]
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
} else if arg == "-F" || arg == "-c" || arg == "--config" {
|
||||||
|
// Skip config file arguments - these are handled separately
|
||||||
|
if i+1 < len(args) {
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
return nil, false
|
||||||
|
} else if strings.HasPrefix(arg, "-") {
|
||||||
|
// Skip other SSH options like -v, -A, -X, etc.
|
||||||
|
// If they have a value, skip it too
|
||||||
|
if i+1 < len(args) && !strings.HasPrefix(args[i+1], "-") {
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
} else if strings.Contains(arg, "@") {
|
||||||
|
// Parse user@hostname
|
||||||
|
parts := strings.SplitN(arg, "@", 2)
|
||||||
|
if len(parts) == 2 {
|
||||||
|
conn.User = parts[0]
|
||||||
|
conn.Hostname = parts[1]
|
||||||
|
}
|
||||||
|
} else if conn.Hostname == "" {
|
||||||
|
// If no @, treat as just hostname
|
||||||
|
conn.Hostname = arg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If we got a hostname, this is a valid manual connection
|
||||||
|
if conn.Hostname != "" {
|
||||||
|
return conn, true
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsManualSSHCommand checks if the arguments represent a manual SSH connection
|
||||||
|
// (not a configured host name)
|
||||||
|
func IsManualSSHCommand(args []string) bool {
|
||||||
|
if len(args) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for SSH flags that indicate manual connection
|
||||||
|
for _, arg := range args {
|
||||||
|
if arg == "-p" || strings.HasPrefix(arg, "-p") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if strings.Contains(arg, "@") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
277
internal/history/parser_test.go
Normal file
277
internal/history/parser_test.go
Normal file
@@ -0,0 +1,277 @@
|
|||||||
|
package history
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestParseSSHArgs(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args []string
|
||||||
|
wantConn *ManualConnection
|
||||||
|
wantOk bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "user@host",
|
||||||
|
args: []string{"user@example.com"},
|
||||||
|
wantConn: &ManualConnection{
|
||||||
|
User: "user",
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "22",
|
||||||
|
},
|
||||||
|
wantOk: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "user@host with -p port",
|
||||||
|
args: []string{"-p", "2222", "user@example.com"},
|
||||||
|
wantConn: &ManualConnection{
|
||||||
|
User: "user",
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "2222",
|
||||||
|
},
|
||||||
|
wantOk: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "user@host with -p2222 (no space)",
|
||||||
|
args: []string{"-p2222", "user@example.com"},
|
||||||
|
wantConn: &ManualConnection{
|
||||||
|
User: "user",
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "2222",
|
||||||
|
},
|
||||||
|
wantOk: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "user@host with -i identity",
|
||||||
|
args: []string{"-i", "~/.ssh/id_rsa", "user@example.com"},
|
||||||
|
wantConn: &ManualConnection{
|
||||||
|
User: "user",
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "22",
|
||||||
|
Identity: "~/.ssh/id_rsa",
|
||||||
|
},
|
||||||
|
wantOk: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "complete connection",
|
||||||
|
args: []string{"-p", "2222", "-i", "~/.ssh/id_rsa", "guillaume@127.0.0.1"},
|
||||||
|
wantConn: &ManualConnection{
|
||||||
|
User: "guillaume",
|
||||||
|
Hostname: "127.0.0.1",
|
||||||
|
Port: "2222",
|
||||||
|
Identity: "~/.ssh/id_rsa",
|
||||||
|
},
|
||||||
|
wantOk: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "just hostname (no user)",
|
||||||
|
args: []string{"example.com"},
|
||||||
|
wantConn: &ManualConnection{
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "22",
|
||||||
|
// User will be current system user, so we don't check it
|
||||||
|
},
|
||||||
|
wantOk: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "config file args should return false",
|
||||||
|
args: []string{"-F", "~/.ssh/config", "host"},
|
||||||
|
wantConn: nil,
|
||||||
|
wantOk: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty args",
|
||||||
|
args: []string{},
|
||||||
|
wantConn: nil,
|
||||||
|
wantOk: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
gotConn, gotOk := ParseSSHArgs(tt.args)
|
||||||
|
|
||||||
|
if gotOk != tt.wantOk {
|
||||||
|
t.Errorf("ParseSSHArgs() gotOk = %v, want %v", gotOk, tt.wantOk)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if !tt.wantOk {
|
||||||
|
if gotConn != nil {
|
||||||
|
t.Errorf("ParseSSHArgs() gotConn = %v, want nil", gotConn)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if gotConn == nil {
|
||||||
|
t.Errorf("ParseSSHArgs() gotConn = nil, want non-nil")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if gotConn.User != tt.wantConn.User {
|
||||||
|
// Skip user check if wantConn.User is empty (current user)
|
||||||
|
if tt.wantConn.User != "" {
|
||||||
|
t.Errorf("ParseSSHArgs() User = %v, want %v", gotConn.User, tt.wantConn.User)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if gotConn.Hostname != tt.wantConn.Hostname {
|
||||||
|
t.Errorf("ParseSSHArgs() Hostname = %v, want %v", gotConn.Hostname, tt.wantConn.Hostname)
|
||||||
|
}
|
||||||
|
if gotConn.Port != tt.wantConn.Port {
|
||||||
|
t.Errorf("ParseSSHArgs() Port = %v, want %v", gotConn.Port, tt.wantConn.Port)
|
||||||
|
}
|
||||||
|
if gotConn.Identity != tt.wantConn.Identity {
|
||||||
|
t.Errorf("ParseSSHArgs() Identity = %v, want %v", gotConn.Identity, tt.wantConn.Identity)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIsManualSSHCommand(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args []string
|
||||||
|
want bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "user@host is manual",
|
||||||
|
args: []string{"user@example.com"},
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "with -p flag is manual",
|
||||||
|
args: []string{"-p", "2222", "host"},
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "with -p2222 is manual",
|
||||||
|
args: []string{"-p2222", "host"},
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "just hostname is not manual",
|
||||||
|
args: []string{"myhost"},
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty is not manual",
|
||||||
|
args: []string{},
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := IsManualSSHCommand(tt.args); got != tt.want {
|
||||||
|
t.Errorf("IsManualSSHCommand() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestManualConnectionID(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
conn ManualConnection
|
||||||
|
wantHostID string
|
||||||
|
wantUser string
|
||||||
|
wantHostname string
|
||||||
|
wantPort string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "complete connection",
|
||||||
|
conn: ManualConnection{
|
||||||
|
User: "guillaume",
|
||||||
|
Hostname: "127.0.0.1",
|
||||||
|
Port: "2222",
|
||||||
|
Identity: "~/.ssh/id_rsa",
|
||||||
|
},
|
||||||
|
wantHostID: "manual:guillaume@127.0.0.1:2222",
|
||||||
|
wantUser: "guillaume",
|
||||||
|
wantHostname: "127.0.0.1",
|
||||||
|
wantPort: "2222",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "default port",
|
||||||
|
conn: ManualConnection{
|
||||||
|
User: "user",
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "",
|
||||||
|
},
|
||||||
|
wantHostID: "manual:user@example.com:22",
|
||||||
|
wantUser: "user",
|
||||||
|
wantHostname: "example.com",
|
||||||
|
wantPort: "22",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "no user specified",
|
||||||
|
conn: ManualConnection{
|
||||||
|
Hostname: "example.com",
|
||||||
|
Port: "2222",
|
||||||
|
},
|
||||||
|
wantHostID: "manual:default@example.com:2222",
|
||||||
|
wantUser: "default",
|
||||||
|
wantHostname: "example.com",
|
||||||
|
wantPort: "2222",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
// Test generation
|
||||||
|
gotHostID := generateManualHostID(tt.conn)
|
||||||
|
if gotHostID != tt.wantHostID {
|
||||||
|
t.Errorf("generateManualHostID() = %v, want %v", gotHostID, tt.wantHostID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test IsManualConnection
|
||||||
|
if !IsManualConnection(gotHostID) {
|
||||||
|
t.Errorf("IsManualConnection(%v) = false, want true", gotHostID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test parsing
|
||||||
|
user, hostname, port, ok := ParseManualConnectionID(gotHostID)
|
||||||
|
if !ok {
|
||||||
|
t.Errorf("ParseManualConnectionID() ok = false, want true")
|
||||||
|
}
|
||||||
|
if user != tt.wantUser {
|
||||||
|
t.Errorf("ParseManualConnectionID() user = %v, want %v", user, tt.wantUser)
|
||||||
|
}
|
||||||
|
if hostname != tt.wantHostname {
|
||||||
|
t.Errorf("ParseManualConnectionID() hostname = %v, want %v", hostname, tt.wantHostname)
|
||||||
|
}
|
||||||
|
if port != tt.wantPort {
|
||||||
|
t.Errorf("ParseManualConnectionID() port = %v, want %v", port, tt.wantPort)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseManualConnectionID_Invalid(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
hostID string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "not a manual connection",
|
||||||
|
hostID: "myhost",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "missing components",
|
||||||
|
hostID: "manual:invalid",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "no @ sign",
|
||||||
|
hostID: "manual:hostname:22",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
_, _, _, ok := ParseManualConnectionID(tt.hostID)
|
||||||
|
if ok {
|
||||||
|
t.Errorf("ParseManualConnectionID() ok = true, want false for invalid input")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
package ui
|
package ui
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"os/user"
|
"os/user"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -16,6 +17,7 @@ import (
|
|||||||
type addFormModel struct {
|
type addFormModel struct {
|
||||||
inputs []textinput.Model
|
inputs []textinput.Model
|
||||||
focused int
|
focused int
|
||||||
|
currentTab int // 0 = General, 1 = Advanced
|
||||||
err string
|
err string
|
||||||
styles Styles
|
styles Styles
|
||||||
success bool
|
success bool
|
||||||
@@ -47,7 +49,7 @@ func NewAddForm(hostname string, styles Styles, width, height int, configFile st
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
inputs := make([]textinput.Model, 8)
|
inputs := make([]textinput.Model, 10) // Increased from 9 to 10 for RequestTTY
|
||||||
|
|
||||||
// Name input
|
// Name input
|
||||||
inputs[nameInput] = textinput.New()
|
inputs[nameInput] = textinput.New()
|
||||||
@@ -101,9 +103,22 @@ func NewAddForm(hostname string, styles Styles, width, height int, configFile st
|
|||||||
inputs[tagsInput].CharLimit = 200
|
inputs[tagsInput].CharLimit = 200
|
||||||
inputs[tagsInput].Width = 50
|
inputs[tagsInput].Width = 50
|
||||||
|
|
||||||
|
// Remote Command input
|
||||||
|
inputs[remoteCommandInput] = textinput.New()
|
||||||
|
inputs[remoteCommandInput].Placeholder = "ls -la, htop, bash"
|
||||||
|
inputs[remoteCommandInput].CharLimit = 300
|
||||||
|
inputs[remoteCommandInput].Width = 70
|
||||||
|
|
||||||
|
// RequestTTY input
|
||||||
|
inputs[requestTTYInput] = textinput.New()
|
||||||
|
inputs[requestTTYInput].Placeholder = "yes, no, force, auto"
|
||||||
|
inputs[requestTTYInput].CharLimit = 10
|
||||||
|
inputs[requestTTYInput].Width = 30
|
||||||
|
|
||||||
return &addFormModel{
|
return &addFormModel{
|
||||||
inputs: inputs,
|
inputs: inputs,
|
||||||
focused: nameInput,
|
focused: nameInput,
|
||||||
|
currentTab: tabGeneral, // Start on General tab
|
||||||
styles: styles,
|
styles: styles,
|
||||||
width: width,
|
width: width,
|
||||||
height: height,
|
height: height,
|
||||||
@@ -111,6 +126,11 @@ func NewAddForm(hostname string, styles Styles, width, height int, configFile st
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
tabGeneral = iota
|
||||||
|
tabAdvanced
|
||||||
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
nameInput = iota
|
nameInput = iota
|
||||||
hostnameInput
|
hostnameInput
|
||||||
@@ -118,8 +138,11 @@ const (
|
|||||||
portInput
|
portInput
|
||||||
identityInput
|
identityInput
|
||||||
proxyJumpInput
|
proxyJumpInput
|
||||||
optionsInput
|
|
||||||
tagsInput
|
tagsInput
|
||||||
|
// Advanced tab inputs
|
||||||
|
optionsInput
|
||||||
|
remoteCommandInput
|
||||||
|
requestTTYInput
|
||||||
)
|
)
|
||||||
|
|
||||||
// Messages for communication with parent model
|
// Messages for communication with parent model
|
||||||
@@ -153,36 +176,20 @@ func (m *addFormModel) Update(msg tea.Msg) (*addFormModel, tea.Cmd) {
|
|||||||
// Allow submission from any field with Ctrl+S (Save)
|
// Allow submission from any field with Ctrl+S (Save)
|
||||||
return m, m.submitForm()
|
return m, m.submitForm()
|
||||||
|
|
||||||
|
case "ctrl+j":
|
||||||
|
// Switch to next tab
|
||||||
|
m.currentTab = (m.currentTab + 1) % 2
|
||||||
|
m.focused = m.getFirstInputForTab(m.currentTab)
|
||||||
|
return m, m.updateFocus()
|
||||||
|
|
||||||
|
case "ctrl+k":
|
||||||
|
// Switch to previous tab
|
||||||
|
m.currentTab = (m.currentTab - 1 + 2) % 2
|
||||||
|
m.focused = m.getFirstInputForTab(m.currentTab)
|
||||||
|
return m, m.updateFocus()
|
||||||
|
|
||||||
case "tab", "shift+tab", "enter", "up", "down":
|
case "tab", "shift+tab", "enter", "up", "down":
|
||||||
s := msg.String()
|
return m, m.handleNavigation(msg.String())
|
||||||
|
|
||||||
// Handle form submission
|
|
||||||
if s == "enter" && m.focused == len(m.inputs)-1 {
|
|
||||||
return m, m.submitForm()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Cycle inputs
|
|
||||||
if s == "up" || s == "shift+tab" {
|
|
||||||
m.focused--
|
|
||||||
} else {
|
|
||||||
m.focused++
|
|
||||||
}
|
|
||||||
|
|
||||||
if m.focused > len(m.inputs)-1 {
|
|
||||||
m.focused = 0
|
|
||||||
} else if m.focused < 0 {
|
|
||||||
m.focused = len(m.inputs) - 1
|
|
||||||
}
|
|
||||||
|
|
||||||
for i := range m.inputs {
|
|
||||||
if i == m.focused {
|
|
||||||
cmds = append(cmds, m.inputs[i].Focus())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
m.inputs[i].Blur()
|
|
||||||
}
|
|
||||||
|
|
||||||
return m, tea.Batch(cmds...)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
case addFormSubmitMsg:
|
case addFormSubmitMsg:
|
||||||
@@ -206,32 +213,104 @@ func (m *addFormModel) Update(msg tea.Msg) (*addFormModel, tea.Cmd) {
|
|||||||
return m, tea.Batch(cmds...)
|
return m, tea.Batch(cmds...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getFirstInputForTab returns the first input index for a given tab
|
||||||
|
func (m *addFormModel) getFirstInputForTab(tab int) int {
|
||||||
|
switch tab {
|
||||||
|
case tabGeneral:
|
||||||
|
return nameInput
|
||||||
|
case tabAdvanced:
|
||||||
|
return optionsInput
|
||||||
|
default:
|
||||||
|
return nameInput
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getInputsForCurrentTab returns the input indices for the current tab
|
||||||
|
func (m *addFormModel) getInputsForCurrentTab() []int {
|
||||||
|
switch m.currentTab {
|
||||||
|
case tabGeneral:
|
||||||
|
return []int{nameInput, hostnameInput, userInput, portInput, identityInput, proxyJumpInput, tagsInput}
|
||||||
|
case tabAdvanced:
|
||||||
|
return []int{optionsInput, remoteCommandInput, requestTTYInput}
|
||||||
|
default:
|
||||||
|
return []int{nameInput, hostnameInput, userInput, portInput, identityInput, proxyJumpInput, tagsInput}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// updateFocus updates focus for inputs
|
||||||
|
func (m *addFormModel) updateFocus() tea.Cmd {
|
||||||
|
var cmds []tea.Cmd
|
||||||
|
for i := range m.inputs {
|
||||||
|
if i == m.focused {
|
||||||
|
cmds = append(cmds, m.inputs[i].Focus())
|
||||||
|
} else {
|
||||||
|
m.inputs[i].Blur()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return tea.Batch(cmds...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// handleNavigation handles tab/arrow navigation within the current tab
|
||||||
|
func (m *addFormModel) handleNavigation(key string) tea.Cmd {
|
||||||
|
currentTabInputs := m.getInputsForCurrentTab()
|
||||||
|
|
||||||
|
// Find current position within the tab
|
||||||
|
currentPos := 0
|
||||||
|
for i, input := range currentTabInputs {
|
||||||
|
if input == m.focused {
|
||||||
|
currentPos = i
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle form submission on last field of Advanced tab
|
||||||
|
if key == "enter" && m.currentTab == tabAdvanced && currentPos == len(currentTabInputs)-1 {
|
||||||
|
return m.submitForm()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Navigate within current tab
|
||||||
|
if key == "up" || key == "shift+tab" {
|
||||||
|
currentPos--
|
||||||
|
} else {
|
||||||
|
currentPos++
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wrap around within current tab
|
||||||
|
if currentPos >= len(currentTabInputs) {
|
||||||
|
currentPos = 0
|
||||||
|
} else if currentPos < 0 {
|
||||||
|
currentPos = len(currentTabInputs) - 1
|
||||||
|
}
|
||||||
|
|
||||||
|
m.focused = currentTabInputs[currentPos]
|
||||||
|
return m.updateFocus()
|
||||||
|
}
|
||||||
|
|
||||||
func (m *addFormModel) View() string {
|
func (m *addFormModel) View() string {
|
||||||
if m.success {
|
if m.success {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Check if terminal height is sufficient
|
||||||
|
if !m.isHeightSufficient() {
|
||||||
|
return m.renderHeightWarning()
|
||||||
|
}
|
||||||
|
|
||||||
var b strings.Builder
|
var b strings.Builder
|
||||||
|
|
||||||
b.WriteString(m.styles.FormTitle.Render("Add SSH Host Configuration"))
|
b.WriteString(m.styles.FormTitle.Render("Add SSH Host Configuration"))
|
||||||
b.WriteString("\n\n")
|
b.WriteString("\n\n")
|
||||||
|
|
||||||
fields := []string{
|
// Render tabs
|
||||||
"Host Name *",
|
b.WriteString(m.renderTabs())
|
||||||
"Hostname/IP *",
|
|
||||||
"User",
|
|
||||||
"Port",
|
|
||||||
"Identity File",
|
|
||||||
"ProxyJump",
|
|
||||||
"SSH Options",
|
|
||||||
"Tags (comma-separated)",
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, field := range fields {
|
|
||||||
b.WriteString(m.styles.FormField.Render(field))
|
|
||||||
b.WriteString("\n")
|
|
||||||
b.WriteString(m.inputs[i].View())
|
|
||||||
b.WriteString("\n\n")
|
b.WriteString("\n\n")
|
||||||
|
|
||||||
|
// Render current tab content
|
||||||
|
switch m.currentTab {
|
||||||
|
case tabGeneral:
|
||||||
|
b.WriteString(m.renderGeneralTab())
|
||||||
|
case tabAdvanced:
|
||||||
|
b.WriteString(m.renderAdvancedTab())
|
||||||
}
|
}
|
||||||
|
|
||||||
if m.err != "" {
|
if m.err != "" {
|
||||||
@@ -239,13 +318,133 @@ func (m *addFormModel) View() string {
|
|||||||
b.WriteString("\n\n")
|
b.WriteString("\n\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
b.WriteString(m.styles.FormHelp.Render("Tab/Shift+Tab: navigate • Enter on last field: submit • Ctrl+S: save • Ctrl+C/Esc: cancel"))
|
// Help text
|
||||||
|
b.WriteString(m.styles.FormHelp.Render("Tab/Shift+Tab: navigate • Ctrl+J/K: switch tabs"))
|
||||||
|
b.WriteString("\n")
|
||||||
|
b.WriteString(m.styles.FormHelp.Render("Enter on last field: submit • Ctrl+S: save • Ctrl+C/Esc: cancel"))
|
||||||
b.WriteString("\n")
|
b.WriteString("\n")
|
||||||
b.WriteString(m.styles.FormHelp.Render("* Required fields"))
|
b.WriteString(m.styles.FormHelp.Render("* Required fields"))
|
||||||
|
|
||||||
return b.String()
|
return b.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getMinimumHeight calculates the minimum height needed to display the form
|
||||||
|
func (m *addFormModel) getMinimumHeight() int {
|
||||||
|
// Title: 1 line + 2 newlines = 3
|
||||||
|
titleLines := 3
|
||||||
|
// Tabs: 1 line + 2 newlines = 3
|
||||||
|
tabLines := 3
|
||||||
|
// Fields in current tab
|
||||||
|
var fieldsCount int
|
||||||
|
if m.currentTab == tabGeneral {
|
||||||
|
fieldsCount = 7 // 7 fields in general tab
|
||||||
|
} else {
|
||||||
|
fieldsCount = 3 // 3 fields in advanced tab
|
||||||
|
}
|
||||||
|
// Each field: label (1) + input (1) + spacing (2) = 4 lines per field, but let's be more conservative
|
||||||
|
fieldsLines := fieldsCount * 3 // Reduced from 4 to 3
|
||||||
|
// Help text: 3 lines
|
||||||
|
helpLines := 3
|
||||||
|
// Error message space when needed: 2 lines
|
||||||
|
errorLines := 0 // Only count when there's actually an error
|
||||||
|
if m.err != "" {
|
||||||
|
errorLines = 2
|
||||||
|
}
|
||||||
|
|
||||||
|
return titleLines + tabLines + fieldsLines + helpLines + errorLines + 1 // +1 minimal safety margin
|
||||||
|
}
|
||||||
|
|
||||||
|
// isHeightSufficient checks if the current terminal height is sufficient
|
||||||
|
func (m *addFormModel) isHeightSufficient() bool {
|
||||||
|
return m.height >= m.getMinimumHeight()
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderHeightWarning renders a warning message when height is insufficient
|
||||||
|
func (m *addFormModel) renderHeightWarning() string {
|
||||||
|
required := m.getMinimumHeight()
|
||||||
|
current := m.height
|
||||||
|
|
||||||
|
warning := m.styles.ErrorText.Render("⚠️ Terminal height is too small!")
|
||||||
|
details := m.styles.FormField.Render(fmt.Sprintf("Current: %d lines, Required: %d lines", current, required))
|
||||||
|
instruction := m.styles.FormHelp.Render("Please resize your terminal window and try again.")
|
||||||
|
instruction2 := m.styles.FormHelp.Render("Press Ctrl+C to cancel or resize terminal window.")
|
||||||
|
|
||||||
|
return warning + "\n\n" + details + "\n\n" + instruction + "\n" + instruction2
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderTabs renders the tab headers
|
||||||
|
func (m *addFormModel) renderTabs() string {
|
||||||
|
var generalTab, advancedTab string
|
||||||
|
|
||||||
|
if m.currentTab == tabGeneral {
|
||||||
|
generalTab = m.styles.FocusedLabel.Render("[ General ]")
|
||||||
|
advancedTab = m.styles.FormField.Render(" Advanced ")
|
||||||
|
} else {
|
||||||
|
generalTab = m.styles.FormField.Render(" General ")
|
||||||
|
advancedTab = m.styles.FocusedLabel.Render("[ Advanced ]")
|
||||||
|
}
|
||||||
|
|
||||||
|
return generalTab + " " + advancedTab
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderGeneralTab renders the general tab content
|
||||||
|
func (m *addFormModel) renderGeneralTab() string {
|
||||||
|
var b strings.Builder
|
||||||
|
|
||||||
|
fields := []struct {
|
||||||
|
index int
|
||||||
|
label string
|
||||||
|
}{
|
||||||
|
{nameInput, "Host Name *"},
|
||||||
|
{hostnameInput, "Hostname/IP *"},
|
||||||
|
{userInput, "User"},
|
||||||
|
{portInput, "Port"},
|
||||||
|
{identityInput, "Identity File"},
|
||||||
|
{proxyJumpInput, "ProxyJump"},
|
||||||
|
{tagsInput, "Tags (comma-separated)"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, field := range fields {
|
||||||
|
fieldStyle := m.styles.FormField
|
||||||
|
if m.focused == field.index {
|
||||||
|
fieldStyle = m.styles.FocusedLabel
|
||||||
|
}
|
||||||
|
b.WriteString(fieldStyle.Render(field.label))
|
||||||
|
b.WriteString("\n")
|
||||||
|
b.WriteString(m.inputs[field.index].View())
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderAdvancedTab renders the advanced tab content
|
||||||
|
func (m *addFormModel) renderAdvancedTab() string {
|
||||||
|
var b strings.Builder
|
||||||
|
|
||||||
|
fields := []struct {
|
||||||
|
index int
|
||||||
|
label string
|
||||||
|
}{
|
||||||
|
{optionsInput, "SSH Options"},
|
||||||
|
{remoteCommandInput, "Remote Command"},
|
||||||
|
{requestTTYInput, "Request TTY"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, field := range fields {
|
||||||
|
fieldStyle := m.styles.FormField
|
||||||
|
if m.focused == field.index {
|
||||||
|
fieldStyle = m.styles.FocusedLabel
|
||||||
|
}
|
||||||
|
b.WriteString(fieldStyle.Render(field.label))
|
||||||
|
b.WriteString("\n")
|
||||||
|
b.WriteString(m.inputs[field.index].View())
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
// Standalone wrapper for add form
|
// Standalone wrapper for add form
|
||||||
type standaloneAddForm struct {
|
type standaloneAddForm struct {
|
||||||
*addFormModel
|
*addFormModel
|
||||||
@@ -291,6 +490,8 @@ func (m *addFormModel) submitForm() tea.Cmd {
|
|||||||
identity := strings.TrimSpace(m.inputs[identityInput].Value())
|
identity := strings.TrimSpace(m.inputs[identityInput].Value())
|
||||||
proxyJump := strings.TrimSpace(m.inputs[proxyJumpInput].Value())
|
proxyJump := strings.TrimSpace(m.inputs[proxyJumpInput].Value())
|
||||||
options := strings.TrimSpace(m.inputs[optionsInput].Value())
|
options := strings.TrimSpace(m.inputs[optionsInput].Value())
|
||||||
|
remoteCommand := strings.TrimSpace(m.inputs[remoteCommandInput].Value())
|
||||||
|
requestTTY := strings.TrimSpace(m.inputs[requestTTYInput].Value())
|
||||||
|
|
||||||
// Set defaults
|
// Set defaults
|
||||||
if user == "" {
|
if user == "" {
|
||||||
@@ -326,6 +527,8 @@ func (m *addFormModel) submitForm() tea.Cmd {
|
|||||||
Identity: identity,
|
Identity: identity,
|
||||||
ProxyJump: proxyJump,
|
ProxyJump: proxyJump,
|
||||||
Options: config.ParseSSHOptionsFromCommand(options),
|
Options: config.ParseSSHOptionsFromCommand(options),
|
||||||
|
RemoteCommand: remoteCommand,
|
||||||
|
RequestTTY: requestTTY,
|
||||||
Tags: tags,
|
Tags: tags,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package ui
|
package ui
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/Gu1llaum-3/sshm/internal/config"
|
"github.com/Gu1llaum-3/sshm/internal/config"
|
||||||
@@ -8,23 +9,38 @@ import (
|
|||||||
|
|
||||||
"github.com/charmbracelet/bubbles/textinput"
|
"github.com/charmbracelet/bubbles/textinput"
|
||||||
tea "github.com/charmbracelet/bubbletea"
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
"github.com/charmbracelet/lipgloss"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type editFormModel struct {
|
const (
|
||||||
inputs []textinput.Model
|
focusAreaHosts = iota
|
||||||
focused int
|
focusAreaProperties
|
||||||
err string
|
)
|
||||||
success bool
|
|
||||||
styles Styles
|
type editFormSubmitMsg struct {
|
||||||
originalName string
|
hostname string
|
||||||
host *config.SSHHost // Store the original host with SourceFile
|
err error
|
||||||
width int
|
|
||||||
height int
|
|
||||||
configFile string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewEditForm creates a new edit form model
|
type editFormCancelMsg struct{}
|
||||||
|
|
||||||
|
type editFormModel struct {
|
||||||
|
hostInputs []textinput.Model // Support for multiple hosts
|
||||||
|
inputs []textinput.Model
|
||||||
|
focusArea int // 0=hosts, 1=properties
|
||||||
|
focused int
|
||||||
|
currentTab int // 0=General, 1=Advanced (only applies when focusArea == focusAreaProperties)
|
||||||
|
err string
|
||||||
|
styles Styles
|
||||||
|
originalName string
|
||||||
|
originalHosts []string // Store original host names for multi-host detection
|
||||||
|
host *config.SSHHost // Store the original host with SourceFile
|
||||||
|
configFile string // Configuration file path passed by user
|
||||||
|
actualConfigFile string // Actual config file to use (either configFile or host.SourceFile)
|
||||||
|
width int
|
||||||
|
height int
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewEditForm creates a new edit form model that supports both single and multi-host editing
|
||||||
func NewEditForm(hostName string, styles Styles, width, height int, configFile string) (*editFormModel, error) {
|
func NewEditForm(hostName string, styles Styles, width, height int, configFile string) (*editFormModel, error) {
|
||||||
// Get the existing host configuration
|
// Get the existing host configuration
|
||||||
var host *config.SSHHost
|
var host *config.SSHHost
|
||||||
@@ -40,207 +56,482 @@ func NewEditForm(hostName string, styles Styles, width, height int, configFile s
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
inputs := make([]textinput.Model, 8)
|
// Check if this host is part of a multi-host declaration
|
||||||
|
var actualConfigFile string
|
||||||
|
var hostNames []string
|
||||||
|
var isMulti bool
|
||||||
|
|
||||||
// Name input
|
if configFile != "" {
|
||||||
inputs[nameInput] = textinput.New()
|
actualConfigFile = configFile
|
||||||
inputs[nameInput].Placeholder = "server-name"
|
} else {
|
||||||
inputs[nameInput].Focus()
|
actualConfigFile = host.SourceFile
|
||||||
inputs[nameInput].CharLimit = 50
|
|
||||||
inputs[nameInput].Width = 30
|
|
||||||
inputs[nameInput].SetValue(host.Name)
|
|
||||||
|
|
||||||
// Hostname input
|
|
||||||
inputs[hostnameInput] = textinput.New()
|
|
||||||
inputs[hostnameInput].Placeholder = "192.168.1.100 or example.com"
|
|
||||||
inputs[hostnameInput].CharLimit = 100
|
|
||||||
inputs[hostnameInput].Width = 30
|
|
||||||
inputs[hostnameInput].SetValue(host.Hostname)
|
|
||||||
|
|
||||||
// User input
|
|
||||||
inputs[userInput] = textinput.New()
|
|
||||||
inputs[userInput].Placeholder = "root"
|
|
||||||
inputs[userInput].CharLimit = 50
|
|
||||||
inputs[userInput].Width = 30
|
|
||||||
inputs[userInput].SetValue(host.User)
|
|
||||||
|
|
||||||
// Port input
|
|
||||||
inputs[portInput] = textinput.New()
|
|
||||||
inputs[portInput].Placeholder = "22"
|
|
||||||
inputs[portInput].CharLimit = 5
|
|
||||||
inputs[portInput].Width = 30
|
|
||||||
inputs[portInput].SetValue(host.Port)
|
|
||||||
|
|
||||||
// Identity input
|
|
||||||
inputs[identityInput] = textinput.New()
|
|
||||||
inputs[identityInput].Placeholder = "~/.ssh/id_rsa"
|
|
||||||
inputs[identityInput].CharLimit = 200
|
|
||||||
inputs[identityInput].Width = 50
|
|
||||||
inputs[identityInput].SetValue(host.Identity)
|
|
||||||
|
|
||||||
// ProxyJump input
|
|
||||||
inputs[proxyJumpInput] = textinput.New()
|
|
||||||
inputs[proxyJumpInput].Placeholder = "user@jump-host:port or existing-host-name"
|
|
||||||
inputs[proxyJumpInput].CharLimit = 200
|
|
||||||
inputs[proxyJumpInput].Width = 50
|
|
||||||
inputs[proxyJumpInput].SetValue(host.ProxyJump)
|
|
||||||
|
|
||||||
// SSH Options input
|
|
||||||
inputs[optionsInput] = textinput.New()
|
|
||||||
inputs[optionsInput].Placeholder = "-o Compression=yes -o ServerAliveInterval=60"
|
|
||||||
inputs[optionsInput].CharLimit = 500
|
|
||||||
inputs[optionsInput].Width = 70
|
|
||||||
inputs[optionsInput].SetValue(config.FormatSSHOptionsForCommand(host.Options))
|
|
||||||
|
|
||||||
// Tags input
|
|
||||||
inputs[tagsInput] = textinput.New()
|
|
||||||
inputs[tagsInput].Placeholder = "production, web, database"
|
|
||||||
inputs[tagsInput].CharLimit = 200
|
|
||||||
inputs[tagsInput].Width = 50
|
|
||||||
if len(host.Tags) > 0 {
|
|
||||||
inputs[tagsInput].SetValue(strings.Join(host.Tags, ", "))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if actualConfigFile != "" {
|
||||||
|
isMulti, hostNames, err = config.IsPartOfMultiHostDeclaration(hostName, actualConfigFile)
|
||||||
|
if err != nil {
|
||||||
|
// If we can't determine multi-host status, treat as single host
|
||||||
|
isMulti = false
|
||||||
|
hostNames = []string{hostName}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isMulti {
|
||||||
|
hostNames = []string{hostName}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create host inputs
|
||||||
|
hostInputs := make([]textinput.Model, len(hostNames))
|
||||||
|
for i, name := range hostNames {
|
||||||
|
hostInputs[i] = textinput.New()
|
||||||
|
hostInputs[i].Placeholder = "host-name"
|
||||||
|
hostInputs[i].SetValue(name)
|
||||||
|
if i == 0 {
|
||||||
|
hostInputs[i].Focus()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
inputs := make([]textinput.Model, 9) // Increased from 8 to 9 for RequestTTY
|
||||||
|
|
||||||
|
// Hostname input
|
||||||
|
inputs[0] = textinput.New()
|
||||||
|
inputs[0].Placeholder = "192.168.1.100 or example.com"
|
||||||
|
inputs[0].CharLimit = 100
|
||||||
|
inputs[0].Width = 30
|
||||||
|
inputs[0].SetValue(host.Hostname)
|
||||||
|
|
||||||
|
// User input
|
||||||
|
inputs[1] = textinput.New()
|
||||||
|
inputs[1].Placeholder = "root"
|
||||||
|
inputs[1].CharLimit = 50
|
||||||
|
inputs[1].Width = 30
|
||||||
|
inputs[1].SetValue(host.User)
|
||||||
|
|
||||||
|
// Port input
|
||||||
|
inputs[2] = textinput.New()
|
||||||
|
inputs[2].Placeholder = "22"
|
||||||
|
inputs[2].CharLimit = 5
|
||||||
|
inputs[2].Width = 30
|
||||||
|
inputs[2].SetValue(host.Port)
|
||||||
|
|
||||||
|
// Identity input
|
||||||
|
inputs[3] = textinput.New()
|
||||||
|
inputs[3].Placeholder = "~/.ssh/id_rsa"
|
||||||
|
inputs[3].CharLimit = 200
|
||||||
|
inputs[3].Width = 50
|
||||||
|
inputs[3].SetValue(host.Identity)
|
||||||
|
|
||||||
|
// ProxyJump input
|
||||||
|
inputs[4] = textinput.New()
|
||||||
|
inputs[4].Placeholder = "jump-server"
|
||||||
|
inputs[4].CharLimit = 100
|
||||||
|
inputs[4].Width = 30
|
||||||
|
inputs[4].SetValue(host.ProxyJump)
|
||||||
|
|
||||||
|
// Options input
|
||||||
|
inputs[5] = textinput.New()
|
||||||
|
inputs[5].Placeholder = "-o StrictHostKeyChecking=no"
|
||||||
|
inputs[5].CharLimit = 200
|
||||||
|
inputs[5].Width = 50
|
||||||
|
if host.Options != "" {
|
||||||
|
inputs[5].SetValue(config.FormatSSHOptionsForCommand(host.Options))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tags input
|
||||||
|
inputs[6] = textinput.New()
|
||||||
|
inputs[6].Placeholder = "production, web, database"
|
||||||
|
inputs[6].CharLimit = 200
|
||||||
|
inputs[6].Width = 50
|
||||||
|
if len(host.Tags) > 0 {
|
||||||
|
inputs[6].SetValue(strings.Join(host.Tags, ", "))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remote Command input
|
||||||
|
inputs[7] = textinput.New()
|
||||||
|
inputs[7].Placeholder = "ls -la, htop, bash"
|
||||||
|
inputs[7].CharLimit = 300
|
||||||
|
inputs[7].Width = 70
|
||||||
|
inputs[7].SetValue(host.RemoteCommand)
|
||||||
|
|
||||||
|
// RequestTTY input
|
||||||
|
inputs[8] = textinput.New()
|
||||||
|
inputs[8].Placeholder = "yes, no, force, auto"
|
||||||
|
inputs[8].CharLimit = 10
|
||||||
|
inputs[8].Width = 30
|
||||||
|
inputs[8].SetValue(host.RequestTTY)
|
||||||
|
|
||||||
return &editFormModel{
|
return &editFormModel{
|
||||||
|
hostInputs: hostInputs,
|
||||||
inputs: inputs,
|
inputs: inputs,
|
||||||
focused: nameInput,
|
focusArea: focusAreaHosts, // Start with hosts focused for multi-host editing
|
||||||
|
focused: 0,
|
||||||
|
currentTab: 0, // Start on General tab
|
||||||
originalName: hostName,
|
originalName: hostName,
|
||||||
|
originalHosts: hostNames,
|
||||||
host: host,
|
host: host,
|
||||||
configFile: configFile,
|
configFile: configFile,
|
||||||
|
actualConfigFile: actualConfigFile,
|
||||||
styles: styles,
|
styles: styles,
|
||||||
width: width,
|
width: width,
|
||||||
height: height,
|
height: height,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Messages for communication with parent model
|
|
||||||
type editFormSubmitMsg struct {
|
|
||||||
hostname string
|
|
||||||
err error
|
|
||||||
}
|
|
||||||
|
|
||||||
type editFormCancelMsg struct{}
|
|
||||||
|
|
||||||
func (m *editFormModel) Init() tea.Cmd {
|
func (m *editFormModel) Init() tea.Cmd {
|
||||||
return textinput.Blink
|
return textinput.Blink
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *editFormModel) Update(msg tea.Msg) (*editFormModel, tea.Cmd) {
|
// addHostInput adds a new empty host input
|
||||||
|
func (m *editFormModel) addHostInput() tea.Cmd {
|
||||||
|
newInput := textinput.New()
|
||||||
|
newInput.Placeholder = "host-name"
|
||||||
|
newInput.Focus()
|
||||||
|
|
||||||
|
// Unfocus current input regardless of which area we're in
|
||||||
|
if m.focusArea == focusAreaHosts && m.focused < len(m.hostInputs) {
|
||||||
|
m.hostInputs[m.focused].Blur()
|
||||||
|
} else if m.focusArea == focusAreaProperties && m.focused < len(m.inputs) {
|
||||||
|
m.inputs[m.focused].Blur()
|
||||||
|
}
|
||||||
|
|
||||||
|
m.hostInputs = append(m.hostInputs, newInput)
|
||||||
|
|
||||||
|
// Move focus to the new host input
|
||||||
|
m.focusArea = focusAreaHosts
|
||||||
|
m.focused = len(m.hostInputs) - 1
|
||||||
|
|
||||||
|
return textinput.Blink
|
||||||
|
}
|
||||||
|
|
||||||
|
// deleteHostInput removes the currently focused host input
|
||||||
|
func (m *editFormModel) deleteHostInput() tea.Cmd {
|
||||||
|
if len(m.hostInputs) <= 1 || m.focusArea != focusAreaHosts {
|
||||||
|
return nil // Can't delete if only one host or not in host area
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove the focused host input
|
||||||
|
m.hostInputs = append(m.hostInputs[:m.focused], m.hostInputs[m.focused+1:]...)
|
||||||
|
|
||||||
|
// Adjust focus
|
||||||
|
if m.focused >= len(m.hostInputs) {
|
||||||
|
m.focused = len(m.hostInputs) - 1
|
||||||
|
}
|
||||||
|
|
||||||
|
// Focus the new current input
|
||||||
|
if len(m.hostInputs) > 0 {
|
||||||
|
m.hostInputs[m.focused].Focus()
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// updateFocus updates the focus state based on current area and index
|
||||||
|
func (m *editFormModel) updateFocus() tea.Cmd {
|
||||||
|
// Blur all inputs first
|
||||||
|
for i := range m.hostInputs {
|
||||||
|
m.hostInputs[i].Blur()
|
||||||
|
}
|
||||||
|
for i := range m.inputs {
|
||||||
|
m.inputs[i].Blur()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Focus the appropriate input
|
||||||
|
if m.focusArea == focusAreaHosts {
|
||||||
|
if m.focused < len(m.hostInputs) {
|
||||||
|
m.hostInputs[m.focused].Focus()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if m.focused < len(m.inputs) {
|
||||||
|
m.inputs[m.focused].Focus()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return textinput.Blink
|
||||||
|
}
|
||||||
|
|
||||||
|
// getPropertiesForCurrentTab returns the property input indices for the current tab
|
||||||
|
func (m *editFormModel) getPropertiesForCurrentTab() []int {
|
||||||
|
switch m.currentTab {
|
||||||
|
case 0: // General
|
||||||
|
return []int{0, 1, 2, 3, 4, 6} // hostname, user, port, identity, proxyjump, tags
|
||||||
|
case 1: // Advanced
|
||||||
|
return []int{5, 7, 8} // options, remotecommand, requesttty
|
||||||
|
default:
|
||||||
|
return []int{0, 1, 2, 3, 4, 6}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getFirstPropertyForTab returns the first property index for a given tab
|
||||||
|
func (m *editFormModel) getFirstPropertyForTab(tab int) int {
|
||||||
|
properties := []int{0, 1, 2, 3, 4, 6} // General tab
|
||||||
|
if tab == 1 {
|
||||||
|
properties = []int{5, 7, 8} // Advanced tab
|
||||||
|
}
|
||||||
|
if len(properties) > 0 {
|
||||||
|
return properties[0]
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// handleEditNavigation handles navigation in the edit form with tab support
|
||||||
|
func (m *editFormModel) handleEditNavigation(key string) tea.Cmd {
|
||||||
|
if m.focusArea == focusAreaHosts {
|
||||||
|
// Navigate in hosts area
|
||||||
|
if key == "up" || key == "shift+tab" {
|
||||||
|
m.focused--
|
||||||
|
} else {
|
||||||
|
m.focused++
|
||||||
|
}
|
||||||
|
|
||||||
|
if m.focused >= len(m.hostInputs) {
|
||||||
|
// Move to properties area, keep current tab
|
||||||
|
m.focusArea = focusAreaProperties
|
||||||
|
// Keep the current tab instead of forcing it to 0
|
||||||
|
m.focused = m.getFirstPropertyForTab(m.currentTab)
|
||||||
|
} else if m.focused < 0 {
|
||||||
|
m.focused = len(m.hostInputs) - 1
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Navigate in properties area within current tab
|
||||||
|
currentTabProperties := m.getPropertiesForCurrentTab()
|
||||||
|
|
||||||
|
// Find current position within the tab
|
||||||
|
currentPos := 0
|
||||||
|
for i, prop := range currentTabProperties {
|
||||||
|
if prop == m.focused {
|
||||||
|
currentPos = i
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle form submission on last field of Advanced tab
|
||||||
|
if key == "enter" && m.currentTab == 1 && currentPos == len(currentTabProperties)-1 {
|
||||||
|
return m.submitEditForm()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Navigate within current tab
|
||||||
|
if key == "up" || key == "shift+tab" {
|
||||||
|
currentPos--
|
||||||
|
} else {
|
||||||
|
currentPos++
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle transitions between areas and tabs
|
||||||
|
if currentPos >= len(currentTabProperties) {
|
||||||
|
// Move to next area/tab
|
||||||
|
if m.currentTab == 0 {
|
||||||
|
// Move to advanced tab
|
||||||
|
m.currentTab = 1
|
||||||
|
m.focused = m.getFirstPropertyForTab(1)
|
||||||
|
} else {
|
||||||
|
// Move back to hosts area
|
||||||
|
m.focusArea = focusAreaHosts
|
||||||
|
m.focused = 0
|
||||||
|
}
|
||||||
|
} else if currentPos < 0 {
|
||||||
|
// Move to previous area/tab
|
||||||
|
if m.currentTab == 1 {
|
||||||
|
// Move to general tab
|
||||||
|
m.currentTab = 0
|
||||||
|
properties := m.getPropertiesForCurrentTab()
|
||||||
|
m.focused = properties[len(properties)-1]
|
||||||
|
} else {
|
||||||
|
// Move to hosts area
|
||||||
|
m.focusArea = focusAreaHosts
|
||||||
|
m.focused = len(m.hostInputs) - 1
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
m.focused = currentTabProperties[currentPos]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return m.updateFocus()
|
||||||
|
}
|
||||||
|
|
||||||
|
// getMinimumHeight calculates the minimum height needed to display the edit form
|
||||||
|
func (m *editFormModel) getMinimumHeight() int {
|
||||||
|
// Title: 1 line + 2 newlines = 3
|
||||||
|
titleLines := 3
|
||||||
|
// Config file info: 1 line + 2 newlines = 3
|
||||||
|
configLines := 3
|
||||||
|
// Host Names section: title (1) + spacing (2) = 3
|
||||||
|
hostSectionLines := 3
|
||||||
|
// Host inputs: number of hosts * 3 lines each (reduced from 4)
|
||||||
|
hostLines := len(m.hostInputs) * 3
|
||||||
|
// Properties section: title (1) + spacing (2) = 3
|
||||||
|
propertiesSectionLines := 3
|
||||||
|
// Tabs: 1 line + 2 newlines = 3
|
||||||
|
tabLines := 3
|
||||||
|
// Fields in current tab
|
||||||
|
var fieldsCount int
|
||||||
|
if m.currentTab == 0 {
|
||||||
|
fieldsCount = 6 // 6 fields in general tab
|
||||||
|
} else {
|
||||||
|
fieldsCount = 3 // 3 fields in advanced tab
|
||||||
|
}
|
||||||
|
// Each field: reduced from 4 to 3 lines per field
|
||||||
|
fieldsLines := fieldsCount * 3
|
||||||
|
// Help text: 3 lines
|
||||||
|
helpLines := 3
|
||||||
|
// Error message space when needed: 2 lines
|
||||||
|
errorLines := 0 // Only count when there's actually an error
|
||||||
|
if m.err != "" {
|
||||||
|
errorLines = 2
|
||||||
|
}
|
||||||
|
|
||||||
|
return titleLines + configLines + hostSectionLines + hostLines + propertiesSectionLines + tabLines + fieldsLines + helpLines + errorLines + 1 // +1 minimal safety margin
|
||||||
|
}
|
||||||
|
|
||||||
|
// isHeightSufficient checks if the current terminal height is sufficient
|
||||||
|
func (m *editFormModel) isHeightSufficient() bool {
|
||||||
|
return m.height >= m.getMinimumHeight()
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderHeightWarning renders a warning message when height is insufficient
|
||||||
|
func (m *editFormModel) renderHeightWarning() string {
|
||||||
|
required := m.getMinimumHeight()
|
||||||
|
current := m.height
|
||||||
|
|
||||||
|
warning := m.styles.ErrorText.Render("⚠️ Terminal height is too small!")
|
||||||
|
details := m.styles.FormField.Render(fmt.Sprintf("Current: %d lines, Required: %d lines", current, required))
|
||||||
|
instruction := m.styles.FormHelp.Render("Please resize your terminal window and try again.")
|
||||||
|
instruction2 := m.styles.FormHelp.Render("Press Ctrl+C to cancel or resize terminal window.")
|
||||||
|
|
||||||
|
return warning + "\n\n" + details + "\n\n" + instruction + "\n" + instruction2
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *editFormModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||||
var cmds []tea.Cmd
|
var cmds []tea.Cmd
|
||||||
|
|
||||||
switch msg := msg.(type) {
|
switch msg := msg.(type) {
|
||||||
case tea.WindowSizeMsg:
|
case tea.WindowSizeMsg:
|
||||||
m.width = msg.Width
|
m.width = msg.Width
|
||||||
m.height = msg.Height
|
m.height = msg.Height
|
||||||
m.styles = NewStyles(m.width)
|
|
||||||
return m, nil
|
|
||||||
|
|
||||||
case tea.KeyMsg:
|
case tea.KeyMsg:
|
||||||
switch msg.String() {
|
switch msg.String() {
|
||||||
case "ctrl+c", "esc":
|
case "ctrl+c", "esc":
|
||||||
|
m.err = ""
|
||||||
return m, func() tea.Msg { return editFormCancelMsg{} }
|
return m, func() tea.Msg { return editFormCancelMsg{} }
|
||||||
|
|
||||||
case "ctrl+s":
|
case "ctrl+s":
|
||||||
// Allow submission from any field with Ctrl+S (Save)
|
// Allow submission from any field with Ctrl+S (Save)
|
||||||
return m, m.submitEditForm()
|
return m, m.submitEditForm()
|
||||||
|
|
||||||
|
case "ctrl+j":
|
||||||
|
// Switch to next tab
|
||||||
|
m.currentTab = (m.currentTab + 1) % 2
|
||||||
|
// If we're in hosts area, stay there. If in properties, go to the first field of the new tab
|
||||||
|
if m.focusArea == focusAreaProperties {
|
||||||
|
m.focused = m.getFirstPropertyForTab(m.currentTab)
|
||||||
|
}
|
||||||
|
return m, m.updateFocus()
|
||||||
|
|
||||||
|
case "ctrl+k":
|
||||||
|
// Switch to previous tab
|
||||||
|
m.currentTab = (m.currentTab - 1 + 2) % 2
|
||||||
|
// If we're in hosts area, stay there. If in properties, go to the first field of the new tab
|
||||||
|
if m.focusArea == focusAreaProperties {
|
||||||
|
m.focused = m.getFirstPropertyForTab(m.currentTab)
|
||||||
|
}
|
||||||
|
return m, m.updateFocus()
|
||||||
|
|
||||||
case "tab", "shift+tab", "enter", "up", "down":
|
case "tab", "shift+tab", "enter", "up", "down":
|
||||||
s := msg.String()
|
return m, m.handleEditNavigation(msg.String())
|
||||||
|
|
||||||
// Handle form submission
|
case "ctrl+a":
|
||||||
if s == "enter" && m.focused == len(m.inputs)-1 {
|
// Add a new host input
|
||||||
return m, m.submitEditForm()
|
return m, m.addHostInput()
|
||||||
}
|
|
||||||
|
|
||||||
// Cycle inputs
|
case "ctrl+d":
|
||||||
if s == "up" || s == "shift+tab" {
|
// Delete the currently focused host (if more than one exists)
|
||||||
m.focused--
|
if m.focusArea == focusAreaHosts && len(m.hostInputs) > 1 {
|
||||||
} else {
|
return m, m.deleteHostInput()
|
||||||
m.focused++
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if m.focused > len(m.inputs)-1 {
|
|
||||||
m.focused = 0
|
|
||||||
} else if m.focused < 0 {
|
|
||||||
m.focused = len(m.inputs) - 1
|
|
||||||
}
|
|
||||||
|
|
||||||
for i := range m.inputs {
|
|
||||||
if i == m.focused {
|
|
||||||
cmds = append(cmds, m.inputs[i].Focus())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
m.inputs[i].Blur()
|
|
||||||
}
|
|
||||||
|
|
||||||
return m, tea.Batch(cmds...)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
case editFormSubmitMsg:
|
case editFormSubmitMsg:
|
||||||
if msg.err != nil {
|
if msg.err != nil {
|
||||||
m.err = msg.err.Error()
|
m.err = msg.err.Error()
|
||||||
} else {
|
} else {
|
||||||
m.success = true
|
// Success: let the wrapper handle this
|
||||||
m.err = ""
|
// In TUI mode, this will be handled by the parent
|
||||||
// Don't quit here, let parent handle the success
|
// In standalone mode, the wrapper will quit
|
||||||
}
|
}
|
||||||
return m, nil
|
return m, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Update inputs
|
// Update host inputs
|
||||||
cmd := make([]tea.Cmd, len(m.inputs))
|
hostCmd := make([]tea.Cmd, len(m.hostInputs))
|
||||||
for i := range m.inputs {
|
for i := range m.hostInputs {
|
||||||
m.inputs[i], cmd[i] = m.inputs[i].Update(msg)
|
m.hostInputs[i], hostCmd[i] = m.hostInputs[i].Update(msg)
|
||||||
}
|
}
|
||||||
cmds = append(cmds, cmd...)
|
cmds = append(cmds, hostCmd...)
|
||||||
|
|
||||||
|
// Update property inputs
|
||||||
|
propCmd := make([]tea.Cmd, len(m.inputs))
|
||||||
|
for i := range m.inputs {
|
||||||
|
m.inputs[i], propCmd[i] = m.inputs[i].Update(msg)
|
||||||
|
}
|
||||||
|
cmds = append(cmds, propCmd...)
|
||||||
|
|
||||||
return m, tea.Batch(cmds...)
|
return m, tea.Batch(cmds...)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *editFormModel) View() string {
|
func (m *editFormModel) View() string {
|
||||||
if m.success {
|
// Check if terminal height is sufficient
|
||||||
return ""
|
if !m.isHeightSufficient() {
|
||||||
|
return m.renderHeightWarning()
|
||||||
}
|
}
|
||||||
|
|
||||||
var b strings.Builder
|
var b strings.Builder
|
||||||
|
|
||||||
b.WriteString(m.styles.FormTitle.Render("Edit SSH Host Configuration"))
|
if m.err != "" {
|
||||||
b.WriteString("\n")
|
b.WriteString(m.styles.Error.Render("Error: " + m.err))
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
b.WriteString(m.styles.Header.Render("Edit SSH Host"))
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
|
||||||
// Show source file information
|
|
||||||
if m.host != nil && m.host.SourceFile != "" {
|
if m.host != nil && m.host.SourceFile != "" {
|
||||||
b.WriteString("\n") // Ligne d'espace avant Config file
|
labelStyle := m.styles.FormField
|
||||||
|
pathStyle := m.styles.FormField
|
||||||
// Style for "Config file:" label in primary color
|
|
||||||
labelStyle := lipgloss.NewStyle().
|
|
||||||
Foreground(lipgloss.Color("#00ADD8")). // Primary color
|
|
||||||
Bold(true)
|
|
||||||
|
|
||||||
// Style for the file path in white
|
|
||||||
pathStyle := lipgloss.NewStyle().
|
|
||||||
Foreground(lipgloss.Color("#FFFFFF"))
|
|
||||||
|
|
||||||
configInfo := labelStyle.Render("Config file: ") + pathStyle.Render(formatConfigFile(m.host.SourceFile))
|
configInfo := labelStyle.Render("Config file: ") + pathStyle.Render(formatConfigFile(m.host.SourceFile))
|
||||||
b.WriteString(configInfo)
|
b.WriteString(configInfo)
|
||||||
}
|
}
|
||||||
|
|
||||||
b.WriteString("\n\n")
|
b.WriteString("\n\n")
|
||||||
|
|
||||||
fields := []string{
|
// Host Names Section
|
||||||
"Host Name *",
|
b.WriteString(m.styles.FormTitle.Render("Host Names"))
|
||||||
"Hostname/IP *",
|
b.WriteString("\n\n")
|
||||||
"User",
|
|
||||||
"Port",
|
for i, hostInput := range m.hostInputs {
|
||||||
"Identity File",
|
hostStyle := m.styles.FormField
|
||||||
"ProxyJump",
|
if m.focusArea == focusAreaHosts && m.focused == i {
|
||||||
"SSH Options",
|
hostStyle = m.styles.FocusedLabel
|
||||||
"Tags (comma-separated)",
|
}
|
||||||
|
b.WriteString(hostStyle.Render(fmt.Sprintf("Host Name %d *", i+1)))
|
||||||
|
b.WriteString("\n")
|
||||||
|
b.WriteString(hostInput.View())
|
||||||
|
b.WriteString("\n\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
for i, field := range fields {
|
// Properties Section
|
||||||
b.WriteString(m.styles.FormField.Render(field))
|
b.WriteString(m.styles.FormTitle.Render("Common Properties"))
|
||||||
b.WriteString("\n")
|
|
||||||
b.WriteString(m.inputs[i].View())
|
|
||||||
b.WriteString("\n\n")
|
b.WriteString("\n\n")
|
||||||
|
|
||||||
|
// Render tabs for properties
|
||||||
|
b.WriteString(m.renderEditTabs())
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
|
||||||
|
// Render current tab content
|
||||||
|
switch m.currentTab {
|
||||||
|
case 0: // General
|
||||||
|
b.WriteString(m.renderEditGeneralTab())
|
||||||
|
case 1: // Advanced
|
||||||
|
b.WriteString(m.renderEditAdvancedTab())
|
||||||
}
|
}
|
||||||
|
|
||||||
if m.err != "" {
|
if m.err != "" {
|
||||||
@@ -248,9 +539,87 @@ func (m *editFormModel) View() string {
|
|||||||
b.WriteString("\n\n")
|
b.WriteString("\n\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
b.WriteString(m.styles.FormHelp.Render("Tab/Shift+Tab: navigate • Enter on last field: submit • Ctrl+S: save • Ctrl+C/Esc: cancel"))
|
// Show different help based on number of hosts
|
||||||
|
if len(m.hostInputs) > 1 {
|
||||||
|
b.WriteString(m.styles.FormHelp.Render("Tab/↑↓/Enter: navigate • Ctrl+J/K: switch tabs • Ctrl+A: add host • Ctrl+D: delete host"))
|
||||||
b.WriteString("\n")
|
b.WriteString("\n")
|
||||||
b.WriteString(m.styles.FormHelp.Render("* Required fields"))
|
} else {
|
||||||
|
b.WriteString(m.styles.FormHelp.Render("Tab/↑↓/Enter: navigate • Ctrl+J/K: switch tabs • Ctrl+A: add host"))
|
||||||
|
b.WriteString("\n")
|
||||||
|
}
|
||||||
|
b.WriteString(m.styles.FormHelp.Render("Ctrl+S: save • Ctrl+C/Esc: cancel • * Required fields"))
|
||||||
|
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderEditTabs renders the tab headers for properties
|
||||||
|
func (m *editFormModel) renderEditTabs() string {
|
||||||
|
var generalTab, advancedTab string
|
||||||
|
|
||||||
|
if m.currentTab == 0 {
|
||||||
|
generalTab = m.styles.FocusedLabel.Render("[ General ]")
|
||||||
|
advancedTab = m.styles.FormField.Render(" Advanced ")
|
||||||
|
} else {
|
||||||
|
generalTab = m.styles.FormField.Render(" General ")
|
||||||
|
advancedTab = m.styles.FocusedLabel.Render("[ Advanced ]")
|
||||||
|
}
|
||||||
|
|
||||||
|
return generalTab + " " + advancedTab
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderEditGeneralTab renders the general tab content for properties
|
||||||
|
func (m *editFormModel) renderEditGeneralTab() string {
|
||||||
|
var b strings.Builder
|
||||||
|
|
||||||
|
fields := []struct {
|
||||||
|
index int
|
||||||
|
label string
|
||||||
|
}{
|
||||||
|
{0, "Hostname/IP *"},
|
||||||
|
{1, "User"},
|
||||||
|
{2, "Port"},
|
||||||
|
{3, "Identity File"},
|
||||||
|
{4, "Proxy Jump"},
|
||||||
|
{6, "Tags (comma-separated)"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, field := range fields {
|
||||||
|
fieldStyle := m.styles.FormField
|
||||||
|
if m.focusArea == focusAreaProperties && m.focused == field.index {
|
||||||
|
fieldStyle = m.styles.FocusedLabel
|
||||||
|
}
|
||||||
|
b.WriteString(fieldStyle.Render(field.label))
|
||||||
|
b.WriteString("\n")
|
||||||
|
b.WriteString(m.inputs[field.index].View())
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderEditAdvancedTab renders the advanced tab content for properties
|
||||||
|
func (m *editFormModel) renderEditAdvancedTab() string {
|
||||||
|
var b strings.Builder
|
||||||
|
|
||||||
|
fields := []struct {
|
||||||
|
index int
|
||||||
|
label string
|
||||||
|
}{
|
||||||
|
{5, "SSH Options"},
|
||||||
|
{7, "Remote Command"},
|
||||||
|
{8, "Request TTY"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, field := range fields {
|
||||||
|
fieldStyle := m.styles.FormField
|
||||||
|
if m.focusArea == focusAreaProperties && m.focused == field.index {
|
||||||
|
fieldStyle = m.styles.FocusedLabel
|
||||||
|
}
|
||||||
|
b.WriteString(fieldStyle.Render(field.label))
|
||||||
|
b.WriteString("\n")
|
||||||
|
b.WriteString(m.inputs[field.index].View())
|
||||||
|
b.WriteString("\n\n")
|
||||||
|
}
|
||||||
|
|
||||||
return b.String()
|
return b.String()
|
||||||
}
|
}
|
||||||
@@ -265,29 +634,29 @@ func (m standaloneEditForm) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|||||||
case editFormSubmitMsg:
|
case editFormSubmitMsg:
|
||||||
if msg.err != nil {
|
if msg.err != nil {
|
||||||
m.editFormModel.err = msg.err.Error()
|
m.editFormModel.err = msg.err.Error()
|
||||||
|
return m, nil
|
||||||
} else {
|
} else {
|
||||||
m.editFormModel.success = true
|
// Success: quit the program
|
||||||
return m, tea.Quit
|
return m, tea.Quit
|
||||||
}
|
}
|
||||||
return m, nil
|
|
||||||
case editFormCancelMsg:
|
case editFormCancelMsg:
|
||||||
return m, tea.Quit
|
return m, tea.Quit
|
||||||
}
|
}
|
||||||
|
|
||||||
newForm, cmd := m.editFormModel.Update(msg)
|
newForm, cmd := m.editFormModel.Update(msg)
|
||||||
m.editFormModel = newForm
|
m.editFormModel = newForm.(*editFormModel)
|
||||||
return m, cmd
|
return m, cmd
|
||||||
}
|
}
|
||||||
|
|
||||||
// RunEditForm provides backward compatibility for standalone edit form
|
// RunEditForm runs the edit form as a standalone program
|
||||||
func RunEditForm(hostName string, configFile string) error {
|
func RunEditForm(hostName string, configFile string) error {
|
||||||
styles := NewStyles(80)
|
styles := NewStyles(80) // Default width
|
||||||
editForm, err := NewEditForm(hostName, styles, 80, 24, configFile)
|
editForm, err := NewEditForm(hostName, styles, 80, 24, configFile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
m := standaloneEditForm{editForm}
|
|
||||||
|
|
||||||
|
m := standaloneEditForm{editForm}
|
||||||
p := tea.NewProgram(m, tea.WithAltScreen())
|
p := tea.NewProgram(m, tea.WithAltScreen())
|
||||||
_, err = p.Run()
|
_, err = p.Run()
|
||||||
return err
|
return err
|
||||||
@@ -295,28 +664,48 @@ func RunEditForm(hostName string, configFile string) error {
|
|||||||
|
|
||||||
func (m *editFormModel) submitEditForm() tea.Cmd {
|
func (m *editFormModel) submitEditForm() tea.Cmd {
|
||||||
return func() tea.Msg {
|
return func() tea.Msg {
|
||||||
// Get values
|
// Collect host names
|
||||||
name := strings.TrimSpace(m.inputs[nameInput].Value())
|
var hostNames []string
|
||||||
hostname := strings.TrimSpace(m.inputs[hostnameInput].Value())
|
for _, input := range m.hostInputs {
|
||||||
user := strings.TrimSpace(m.inputs[userInput].Value())
|
name := strings.TrimSpace(input.Value())
|
||||||
port := strings.TrimSpace(m.inputs[portInput].Value())
|
if name != "" {
|
||||||
identity := strings.TrimSpace(m.inputs[identityInput].Value())
|
hostNames = append(hostNames, name)
|
||||||
proxyJump := strings.TrimSpace(m.inputs[proxyJumpInput].Value())
|
}
|
||||||
options := strings.TrimSpace(m.inputs[optionsInput].Value())
|
}
|
||||||
|
|
||||||
|
if len(hostNames) == 0 {
|
||||||
|
return editFormSubmitMsg{err: fmt.Errorf("at least one host name is required")}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get property values using direct indices
|
||||||
|
hostname := strings.TrimSpace(m.inputs[0].Value()) // hostnameInput
|
||||||
|
user := strings.TrimSpace(m.inputs[1].Value()) // userInput
|
||||||
|
port := strings.TrimSpace(m.inputs[2].Value()) // portInput
|
||||||
|
identity := strings.TrimSpace(m.inputs[3].Value()) // identityInput
|
||||||
|
proxyJump := strings.TrimSpace(m.inputs[4].Value()) // proxyJumpInput
|
||||||
|
options := strings.TrimSpace(m.inputs[5].Value()) // optionsInput
|
||||||
|
remoteCommand := strings.TrimSpace(m.inputs[7].Value()) // remoteCommandInput
|
||||||
|
requestTTY := strings.TrimSpace(m.inputs[8].Value()) // requestTTYInput
|
||||||
|
|
||||||
// Set defaults
|
// Set defaults
|
||||||
if port == "" {
|
if port == "" {
|
||||||
port = "22"
|
port = "22"
|
||||||
}
|
}
|
||||||
// Do not auto-fill identity with placeholder if left empty; keep it empty so it's optional
|
|
||||||
|
|
||||||
// Validate all fields
|
// Validate hostname
|
||||||
if err := validation.ValidateHost(name, hostname, port, identity); err != nil {
|
if hostname == "" {
|
||||||
|
return editFormSubmitMsg{err: fmt.Errorf("hostname is required")}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate all host names
|
||||||
|
for _, hostName := range hostNames {
|
||||||
|
if err := validation.ValidateHost(hostName, hostname, port, identity); err != nil {
|
||||||
return editFormSubmitMsg{err: err}
|
return editFormSubmitMsg{err: err}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Parse tags
|
// Parse tags
|
||||||
tagsStr := strings.TrimSpace(m.inputs[tagsInput].Value())
|
tagsStr := strings.TrimSpace(m.inputs[6].Value()) // tagsInput
|
||||||
var tags []string
|
var tags []string
|
||||||
if tagsStr != "" {
|
if tagsStr != "" {
|
||||||
for _, tag := range strings.Split(tagsStr, ",") {
|
for _, tag := range strings.Split(tagsStr, ",") {
|
||||||
@@ -327,25 +716,33 @@ func (m *editFormModel) submitEditForm() tea.Cmd {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create updated host configuration
|
// Create the common host configuration
|
||||||
host := config.SSHHost{
|
commonHost := config.SSHHost{
|
||||||
Name: name,
|
|
||||||
Hostname: hostname,
|
Hostname: hostname,
|
||||||
User: user,
|
User: user,
|
||||||
Port: port,
|
Port: port,
|
||||||
Identity: identity,
|
Identity: identity,
|
||||||
ProxyJump: proxyJump,
|
ProxyJump: proxyJump,
|
||||||
Options: config.ParseSSHOptionsFromCommand(options),
|
Options: options,
|
||||||
|
RemoteCommand: remoteCommand,
|
||||||
|
RequestTTY: requestTTY,
|
||||||
Tags: tags,
|
Tags: tags,
|
||||||
}
|
}
|
||||||
|
|
||||||
// Update the configuration
|
|
||||||
var err error
|
var err error
|
||||||
if m.configFile != "" {
|
if len(hostNames) == 1 && len(m.originalHosts) == 1 {
|
||||||
err = config.UpdateSSHHostInFile(m.originalName, host, m.configFile)
|
// Single host editing
|
||||||
|
commonHost.Name = hostNames[0]
|
||||||
|
if m.actualConfigFile != "" {
|
||||||
|
err = config.UpdateSSHHostInFile(m.originalName, commonHost, m.actualConfigFile)
|
||||||
} else {
|
} else {
|
||||||
err = config.UpdateSSHHost(m.originalName, host)
|
err = config.UpdateSSHHost(m.originalName, commonHost)
|
||||||
}
|
}
|
||||||
return editFormSubmitMsg{hostname: name, err: err}
|
} else {
|
||||||
|
// Multi-host editing or conversion from single to multi
|
||||||
|
err = config.UpdateMultiHostBlock(m.originalHosts, hostNames, commonHost, m.actualConfigFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
return editFormSubmitMsg{hostname: hostNames[0], err: err}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -58,6 +58,9 @@ func (m *helpModel) View() string {
|
|||||||
lipgloss.JoinHorizontal(lipgloss.Left,
|
lipgloss.JoinHorizontal(lipgloss.Left,
|
||||||
m.styles.FocusedLabel.Render("Tab "),
|
m.styles.FocusedLabel.Render("Tab "),
|
||||||
m.styles.HelpText.Render("switch focus")),
|
m.styles.HelpText.Render("switch focus")),
|
||||||
|
lipgloss.JoinHorizontal(lipgloss.Left,
|
||||||
|
m.styles.FocusedLabel.Render("Ctrl+H "),
|
||||||
|
m.styles.HelpText.Render("switch to history view")),
|
||||||
"",
|
"",
|
||||||
m.styles.FocusedLabel.Render("Host Management"),
|
m.styles.FocusedLabel.Render("Host Management"),
|
||||||
"",
|
"",
|
||||||
|
|||||||
530
internal/ui/history_tui.go
Normal file
530
internal/ui/history_tui.go
Normal file
@@ -0,0 +1,530 @@
|
|||||||
|
package ui
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/Gu1llaum-3/sshm/internal/config"
|
||||||
|
"github.com/Gu1llaum-3/sshm/internal/history"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/bubbles/table"
|
||||||
|
"github.com/charmbracelet/bubbles/textinput"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
"github.com/charmbracelet/lipgloss"
|
||||||
|
)
|
||||||
|
|
||||||
|
// HistoryModel represents the TUI model for history view
|
||||||
|
type HistoryModel struct {
|
||||||
|
table table.Model
|
||||||
|
connections []history.ConnectionInfo
|
||||||
|
searchInput textinput.Model
|
||||||
|
searchActive bool
|
||||||
|
filteredConns []history.ConnectionInfo
|
||||||
|
configFile string
|
||||||
|
currentVersion string
|
||||||
|
styles Styles
|
||||||
|
width int
|
||||||
|
height int
|
||||||
|
showAddForm bool
|
||||||
|
addForm *addFormModel
|
||||||
|
selectedConn *history.ConnectionInfo
|
||||||
|
err string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewHistoryModel creates a new history TUI model
|
||||||
|
func NewHistoryModel(connections []history.ConnectionInfo, configFile, currentVersion string) HistoryModel {
|
||||||
|
styles := NewStyles(80)
|
||||||
|
|
||||||
|
// Create search input (different placeholder than main interface)
|
||||||
|
searchInput := textinput.New()
|
||||||
|
searchInput.Placeholder = "Search connections..."
|
||||||
|
searchInput.CharLimit = 50
|
||||||
|
searchInput.Width = 25 // Same width as main interface
|
||||||
|
|
||||||
|
m := HistoryModel{
|
||||||
|
connections: connections,
|
||||||
|
filteredConns: connections,
|
||||||
|
searchInput: searchInput,
|
||||||
|
configFile: configFile,
|
||||||
|
currentVersion: currentVersion,
|
||||||
|
styles: styles,
|
||||||
|
}
|
||||||
|
|
||||||
|
m.updateTable()
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
// Init initializes the history model
|
||||||
|
func (m HistoryModel) Init() tea.Cmd {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update handles messages for the history model
|
||||||
|
func (m HistoryModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||||
|
var cmd tea.Cmd
|
||||||
|
var cmds []tea.Cmd
|
||||||
|
|
||||||
|
// Handle add form if active
|
||||||
|
if m.showAddForm && m.addForm != nil {
|
||||||
|
switch msg := msg.(type) {
|
||||||
|
case addFormSubmitMsg:
|
||||||
|
if msg.err != nil {
|
||||||
|
m.err = msg.err.Error()
|
||||||
|
} else {
|
||||||
|
m.showAddForm = false
|
||||||
|
m.addForm = nil
|
||||||
|
// Return to main list and refresh hosts
|
||||||
|
return m, func() tea.Msg { return refreshHostsMsg{} }
|
||||||
|
}
|
||||||
|
case addFormCancelMsg:
|
||||||
|
m.showAddForm = false
|
||||||
|
m.addForm = nil
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
newForm, cmd := m.addForm.Update(msg)
|
||||||
|
m.addForm = newForm
|
||||||
|
return m, cmd
|
||||||
|
}
|
||||||
|
|
||||||
|
switch msg := msg.(type) {
|
||||||
|
case tea.WindowSizeMsg:
|
||||||
|
m.width = msg.Width
|
||||||
|
m.height = msg.Height
|
||||||
|
m.styles = NewStyles(m.width)
|
||||||
|
m.updateTable()
|
||||||
|
return m, nil
|
||||||
|
|
||||||
|
case tea.KeyMsg:
|
||||||
|
// Handle search mode
|
||||||
|
if m.searchActive {
|
||||||
|
switch msg.String() {
|
||||||
|
case "esc", "ctrl+c":
|
||||||
|
m.searchActive = false
|
||||||
|
m.searchInput.Blur()
|
||||||
|
m.searchInput.SetValue("")
|
||||||
|
m.filteredConns = m.connections
|
||||||
|
m.updateTable()
|
||||||
|
return m, nil
|
||||||
|
case "enter":
|
||||||
|
m.searchActive = false
|
||||||
|
m.searchInput.Blur()
|
||||||
|
return m, nil
|
||||||
|
default:
|
||||||
|
m.searchInput, cmd = m.searchInput.Update(msg)
|
||||||
|
cmds = append(cmds, cmd)
|
||||||
|
m.filterConnections()
|
||||||
|
m.updateTable()
|
||||||
|
return m, tea.Batch(cmds...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Normal mode key handling
|
||||||
|
switch msg.String() {
|
||||||
|
case "ctrl+c", "q", "esc":
|
||||||
|
return m, tea.Quit
|
||||||
|
|
||||||
|
case "ctrl+l":
|
||||||
|
// Return to main list view
|
||||||
|
return m, func() tea.Msg { return returnToListMsg{} }
|
||||||
|
|
||||||
|
case "enter":
|
||||||
|
// Connect to selected host
|
||||||
|
if len(m.filteredConns) > 0 {
|
||||||
|
selectedIdx := m.table.Cursor()
|
||||||
|
if selectedIdx < len(m.filteredConns) {
|
||||||
|
conn := m.filteredConns[selectedIdx]
|
||||||
|
return m, m.connectToHistory(conn)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case "a":
|
||||||
|
// Add manual connection to config
|
||||||
|
if len(m.filteredConns) > 0 {
|
||||||
|
selectedIdx := m.table.Cursor()
|
||||||
|
if selectedIdx < len(m.filteredConns) {
|
||||||
|
conn := m.filteredConns[selectedIdx]
|
||||||
|
// Only allow adding manual connections to config
|
||||||
|
if history.IsManualConnection(conn.HostName) {
|
||||||
|
m.selectedConn = &conn
|
||||||
|
m.showAddForm = true
|
||||||
|
m.addForm = m.createAddFormFromConnection(conn)
|
||||||
|
return m, m.addForm.Init()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case "d":
|
||||||
|
// Delete connection from history
|
||||||
|
if len(m.filteredConns) > 0 {
|
||||||
|
selectedIdx := m.table.Cursor()
|
||||||
|
if selectedIdx < len(m.filteredConns) {
|
||||||
|
conn := m.filteredConns[selectedIdx]
|
||||||
|
return m, m.deleteFromHistory(conn)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case "/":
|
||||||
|
// Activate search
|
||||||
|
m.searchActive = true
|
||||||
|
m.searchInput.Focus()
|
||||||
|
return m, textinput.Blink
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update table
|
||||||
|
m.table, cmd = m.table.Update(msg)
|
||||||
|
cmds = append(cmds, cmd)
|
||||||
|
|
||||||
|
return m, tea.Batch(cmds...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// View renders the history TUI
|
||||||
|
func (m HistoryModel) View() string {
|
||||||
|
if m.showAddForm && m.addForm != nil {
|
||||||
|
return m.addForm.View()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build the interface components (same structure as main view)
|
||||||
|
components := []string{}
|
||||||
|
|
||||||
|
// Add the ASCII title
|
||||||
|
components = append(components, m.styles.Header.Render(asciiTitle))
|
||||||
|
|
||||||
|
// Add error message if there's one to show
|
||||||
|
if m.err != "" {
|
||||||
|
errorStyle := lipgloss.NewStyle().
|
||||||
|
Foreground(lipgloss.Color("9")). // Red color
|
||||||
|
Background(lipgloss.Color("1")). // Dark red background
|
||||||
|
Bold(true).
|
||||||
|
Padding(0, 1).
|
||||||
|
Border(lipgloss.RoundedBorder()).
|
||||||
|
BorderForeground(lipgloss.Color("9")).
|
||||||
|
Align(lipgloss.Center)
|
||||||
|
|
||||||
|
components = append(components, errorStyle.Render("❌ "+m.err))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the search bar with the appropriate style based on focus
|
||||||
|
searchPrompt := "Search (/ to focus): "
|
||||||
|
if m.searchActive {
|
||||||
|
components = append(components, m.styles.SearchFocused.Render(searchPrompt+m.searchInput.View()))
|
||||||
|
} else {
|
||||||
|
components = append(components, m.styles.SearchUnfocused.Render(searchPrompt+m.searchInput.View()))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the table with the appropriate style based on focus
|
||||||
|
if m.searchActive {
|
||||||
|
// The table is not focused, use the unfocused style
|
||||||
|
components = append(components, m.styles.TableUnfocused.Render(m.table.View()))
|
||||||
|
} else {
|
||||||
|
// The table is focused, use the focused style
|
||||||
|
components = append(components, m.styles.TableFocused.Render(m.table.View()))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the help text
|
||||||
|
var helpText string
|
||||||
|
if !m.searchActive {
|
||||||
|
helpText = " ↑/↓: navigate • Enter: connect • Ctrl+L: list • a: add to config (★) • d: delete • q: quit"
|
||||||
|
} else {
|
||||||
|
helpText = " Type to filter • Enter: validate • Tab: switch • ESC: quit"
|
||||||
|
}
|
||||||
|
components = append(components, m.styles.HelpText.Render(helpText))
|
||||||
|
|
||||||
|
// Join all components vertically with appropriate spacing
|
||||||
|
mainView := m.styles.App.Render(
|
||||||
|
lipgloss.JoinVertical(
|
||||||
|
lipgloss.Left,
|
||||||
|
components...,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
return mainView
|
||||||
|
} // updateTable updates the table with current filtered connections
|
||||||
|
func (m *HistoryModel) updateTable() {
|
||||||
|
columns := []table.Column{
|
||||||
|
{Title: "Host", Width: 22}, // Host name with ★ for manual connections
|
||||||
|
{Title: "User", Width: 15},
|
||||||
|
{Title: "Hostname", Width: 25},
|
||||||
|
{Title: "Port", Width: 6},
|
||||||
|
{Title: "Last Connect", Width: 20},
|
||||||
|
{Title: "Count", Width: 6},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load SSH hosts to get details for configured connections
|
||||||
|
var sshHosts []config.SSHHost
|
||||||
|
var err error
|
||||||
|
if m.configFile != "" {
|
||||||
|
sshHosts, err = config.ParseSSHConfigFile(m.configFile)
|
||||||
|
} else {
|
||||||
|
sshHosts, err = config.ParseSSHConfig()
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
sshHosts = []config.SSHHost{}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a map for quick lookup
|
||||||
|
hostsMap := make(map[string]config.SSHHost)
|
||||||
|
for _, host := range sshHosts {
|
||||||
|
hostsMap[host.Name] = host
|
||||||
|
}
|
||||||
|
|
||||||
|
rows := []table.Row{}
|
||||||
|
for _, conn := range m.filteredConns {
|
||||||
|
var hostDisplay, user, hostname, port string
|
||||||
|
|
||||||
|
// Parse manual connections
|
||||||
|
if history.IsManualConnection(conn.HostName) {
|
||||||
|
u, h, p, ok := history.ParseManualConnectionID(conn.HostName)
|
||||||
|
if ok {
|
||||||
|
hostDisplay = "★" // Star indicates this can be added to config
|
||||||
|
user = u
|
||||||
|
hostname = h
|
||||||
|
port = p
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// For configured hosts, show the host name
|
||||||
|
hostDisplay = conn.HostName
|
||||||
|
|
||||||
|
if host, exists := hostsMap[conn.HostName]; exists {
|
||||||
|
user = host.User
|
||||||
|
hostname = host.Hostname
|
||||||
|
port = host.Port
|
||||||
|
if port == "" {
|
||||||
|
port = "22"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
lastConnect := formatTimeSince(conn.LastConnect)
|
||||||
|
|
||||||
|
rows = append(rows, table.Row{
|
||||||
|
hostDisplay,
|
||||||
|
user,
|
||||||
|
hostname,
|
||||||
|
port,
|
||||||
|
lastConnect,
|
||||||
|
fmt.Sprintf("%d", conn.ConnectCount),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Calculate dynamic table height (same logic as main interface)
|
||||||
|
tableHeight := m.calculateTableHeight(len(rows))
|
||||||
|
|
||||||
|
t := table.New(
|
||||||
|
table.WithColumns(columns),
|
||||||
|
table.WithRows(rows),
|
||||||
|
table.WithFocused(true),
|
||||||
|
table.WithHeight(tableHeight),
|
||||||
|
)
|
||||||
|
|
||||||
|
s := table.DefaultStyles()
|
||||||
|
s.Header = s.Header.
|
||||||
|
BorderStyle(lipgloss.NormalBorder()).
|
||||||
|
BorderForeground(lipgloss.Color(PrimaryColor)).
|
||||||
|
BorderBottom(true).
|
||||||
|
Bold(true)
|
||||||
|
s.Selected = s.Selected.
|
||||||
|
Foreground(lipgloss.Color("229")).
|
||||||
|
Background(lipgloss.Color(PrimaryColor)).
|
||||||
|
Bold(false)
|
||||||
|
|
||||||
|
t.SetStyles(s)
|
||||||
|
m.table = t
|
||||||
|
}
|
||||||
|
|
||||||
|
// calculateTableHeight calculates the appropriate height for the table based on terminal size
|
||||||
|
func (m *HistoryModel) calculateTableHeight(rowCount int) int {
|
||||||
|
// Calculate dynamic table height based on terminal size
|
||||||
|
// Layout breakdown (same as main interface):
|
||||||
|
// - ASCII title: 5 lines (1 empty + 4 text lines)
|
||||||
|
// - Search bar: 1 line
|
||||||
|
// - Help text: 1 line
|
||||||
|
// - App margins/spacing: 3 lines
|
||||||
|
// - Safety margin: 3 lines
|
||||||
|
// Total reserved: 13 lines
|
||||||
|
reservedHeight := 13
|
||||||
|
availableHeight := m.height - reservedHeight
|
||||||
|
|
||||||
|
// Add 1 if there's an error message showing
|
||||||
|
if m.err != "" {
|
||||||
|
availableHeight -= 3 // Error box takes about 3 lines
|
||||||
|
}
|
||||||
|
|
||||||
|
// Minimum height should be at least 3 rows for basic usability
|
||||||
|
minTableHeight := 4 // 1 header + 3 data rows minimum
|
||||||
|
maxTableHeight := availableHeight
|
||||||
|
if maxTableHeight < minTableHeight {
|
||||||
|
maxTableHeight = minTableHeight
|
||||||
|
}
|
||||||
|
|
||||||
|
tableHeight := 1 // header
|
||||||
|
dataRowsNeeded := rowCount
|
||||||
|
maxDataRows := maxTableHeight - 1 // subtract 1 for header
|
||||||
|
|
||||||
|
if dataRowsNeeded <= maxDataRows {
|
||||||
|
// We have enough space for all connections
|
||||||
|
tableHeight += dataRowsNeeded
|
||||||
|
} else {
|
||||||
|
// We need to limit to available space
|
||||||
|
tableHeight += maxDataRows
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add one extra line to prevent the last row from being hidden
|
||||||
|
tableHeight += 1
|
||||||
|
|
||||||
|
return tableHeight
|
||||||
|
}
|
||||||
|
|
||||||
|
// filterConnections filters connections based on search input
|
||||||
|
func (m *HistoryModel) filterConnections() {
|
||||||
|
searchTerm := strings.ToLower(m.searchInput.Value())
|
||||||
|
if searchTerm == "" {
|
||||||
|
m.filteredConns = m.connections
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
m.filteredConns = []history.ConnectionInfo{}
|
||||||
|
for _, conn := range m.connections {
|
||||||
|
// Search in hostname
|
||||||
|
if strings.Contains(strings.ToLower(conn.HostName), searchTerm) {
|
||||||
|
m.filteredConns = append(m.filteredConns, conn)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// For manual connections, search in parsed fields
|
||||||
|
if history.IsManualConnection(conn.HostName) {
|
||||||
|
user, hostname, _, ok := history.ParseManualConnectionID(conn.HostName)
|
||||||
|
if ok {
|
||||||
|
if strings.Contains(strings.ToLower(user), searchTerm) ||
|
||||||
|
strings.Contains(strings.ToLower(hostname), searchTerm) {
|
||||||
|
m.filteredConns = append(m.filteredConns, conn)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// connectToHistory connects to a host from history
|
||||||
|
func (m HistoryModel) connectToHistory(conn history.ConnectionInfo) tea.Cmd {
|
||||||
|
var sshArgs []string
|
||||||
|
|
||||||
|
if history.IsManualConnection(conn.HostName) {
|
||||||
|
// Manual connection
|
||||||
|
user, hostname, port, ok := history.ParseManualConnectionID(conn.HostName)
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if port != "" && port != "22" {
|
||||||
|
sshArgs = append(sshArgs, "-p", port)
|
||||||
|
}
|
||||||
|
|
||||||
|
if user != "" {
|
||||||
|
sshArgs = append(sshArgs, fmt.Sprintf("%s@%s", user, hostname))
|
||||||
|
} else {
|
||||||
|
sshArgs = append(sshArgs, hostname)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Configured host
|
||||||
|
if m.configFile != "" {
|
||||||
|
sshArgs = append(sshArgs, "-F", m.configFile)
|
||||||
|
}
|
||||||
|
sshArgs = append(sshArgs, conn.HostName)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Execute SSH using tea.ExecProcess for proper terminal handling
|
||||||
|
sshCmd := exec.Command("ssh", sshArgs...)
|
||||||
|
return tea.ExecProcess(sshCmd, func(err error) tea.Msg {
|
||||||
|
return tea.Quit()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// deleteFromHistory removes a connection from history
|
||||||
|
func (m HistoryModel) deleteFromHistory(conn history.ConnectionInfo) tea.Cmd {
|
||||||
|
return func() tea.Msg {
|
||||||
|
historyManager, err := history.NewHistoryManager()
|
||||||
|
if err != nil {
|
||||||
|
return tea.Quit
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove from history
|
||||||
|
// This would need a new method in history manager
|
||||||
|
// For now, just quit
|
||||||
|
_ = historyManager
|
||||||
|
|
||||||
|
return tea.Quit
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// createAddFormFromConnection creates an add form pre-filled with connection details
|
||||||
|
func (m *HistoryModel) createAddFormFromConnection(conn history.ConnectionInfo) *addFormModel {
|
||||||
|
user, hostname, port, ok := history.ParseManualConnectionID(conn.HostName)
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create form with empty name (user will choose)
|
||||||
|
form := NewAddForm("", m.styles, m.width, m.height, m.configFile)
|
||||||
|
|
||||||
|
// Pre-fill the form with connection details
|
||||||
|
form.inputs[hostnameInput].SetValue(hostname)
|
||||||
|
form.inputs[userInput].SetValue(user)
|
||||||
|
if port != "22" && port != "" {
|
||||||
|
form.inputs[portInput].SetValue(port)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Leave name field empty for user to choose
|
||||||
|
// form.inputs[nameInput].SetValue("") // Already empty by default
|
||||||
|
|
||||||
|
return form
|
||||||
|
}
|
||||||
|
|
||||||
|
// formatTimeSince formats a time duration in human-readable format
|
||||||
|
func formatTimeSince(t time.Time) string {
|
||||||
|
duration := time.Since(t)
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case duration < time.Minute:
|
||||||
|
return "just now"
|
||||||
|
case duration < time.Hour:
|
||||||
|
mins := int(duration.Minutes())
|
||||||
|
if mins == 1 {
|
||||||
|
return "1 minute ago"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d minutes ago", mins)
|
||||||
|
case duration < 24*time.Hour:
|
||||||
|
hours := int(duration.Hours())
|
||||||
|
if hours == 1 {
|
||||||
|
return "1 hour ago"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d hours ago", hours)
|
||||||
|
case duration < 7*24*time.Hour:
|
||||||
|
days := int(duration.Hours() / 24)
|
||||||
|
if days == 1 {
|
||||||
|
return "1 day ago"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d days ago", days)
|
||||||
|
case duration < 30*24*time.Hour:
|
||||||
|
weeks := int(duration.Hours() / 24 / 7)
|
||||||
|
if weeks == 1 {
|
||||||
|
return "1 week ago"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d weeks ago", weeks)
|
||||||
|
default:
|
||||||
|
months := int(duration.Hours() / 24 / 30)
|
||||||
|
if months == 1 {
|
||||||
|
return "1 month ago"
|
||||||
|
}
|
||||||
|
if months < 12 {
|
||||||
|
return fmt.Sprintf("%d months ago", months)
|
||||||
|
}
|
||||||
|
years := months / 12
|
||||||
|
if years == 1 {
|
||||||
|
return "1 year ago"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d years ago", years)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -42,6 +42,7 @@ const (
|
|||||||
ViewPortForward
|
ViewPortForward
|
||||||
ViewHelp
|
ViewHelp
|
||||||
ViewFileSelector
|
ViewFileSelector
|
||||||
|
ViewHistory
|
||||||
)
|
)
|
||||||
|
|
||||||
// PortForwardType defines the type of port forwarding
|
// PortForwardType defines the type of port forwarding
|
||||||
@@ -80,6 +81,9 @@ type Model struct {
|
|||||||
sortMode SortMode
|
sortMode SortMode
|
||||||
configFile string // Path to the SSH config file
|
configFile string // Path to the SSH config file
|
||||||
|
|
||||||
|
// Application configuration
|
||||||
|
appConfig *config.AppConfig
|
||||||
|
|
||||||
// Version update information
|
// Version update information
|
||||||
updateInfo *version.UpdateInfo
|
updateInfo *version.UpdateInfo
|
||||||
currentVersion string
|
currentVersion string
|
||||||
@@ -93,6 +97,7 @@ type Model struct {
|
|||||||
portForwardForm *portForwardModel
|
portForwardForm *portForwardModel
|
||||||
helpForm *helpModel
|
helpForm *helpModel
|
||||||
fileSelectorForm *fileSelectorModel
|
fileSelectorForm *fileSelectorModel
|
||||||
|
historyView *HistoryModel
|
||||||
|
|
||||||
// Terminal size and styles
|
// Terminal size and styles
|
||||||
width int
|
width int
|
||||||
|
|||||||
@@ -37,35 +37,64 @@ func sortHostsByName(hosts []config.SSHHost) []config.SSHHost {
|
|||||||
|
|
||||||
// filterHosts filters hosts according to the search query (name or tags)
|
// filterHosts filters hosts according to the search query (name or tags)
|
||||||
func (m Model) filterHosts(query string) []config.SSHHost {
|
func (m Model) filterHosts(query string) []config.SSHHost {
|
||||||
|
subqueries := strings.Split(query, " ")
|
||||||
|
subqueriesLength := len(subqueries)
|
||||||
|
subfilteredHosts := make([][]config.SSHHost, subqueriesLength)
|
||||||
|
for i, subquery := range subqueries {
|
||||||
|
subfilteredHosts[i] = m.filterHostsByWord(subquery)
|
||||||
|
}
|
||||||
|
|
||||||
|
// return the intersection of search results
|
||||||
|
result := make([]config.SSHHost, 0)
|
||||||
|
tempMap := map[string]int{}
|
||||||
|
for _, hosts := range subfilteredHosts {
|
||||||
|
for _, host := range hosts {
|
||||||
|
if _, ok := tempMap[host.Name]; !ok {
|
||||||
|
tempMap[host.Name] = 1
|
||||||
|
} else {
|
||||||
|
tempMap[host.Name] = tempMap[host.Name] + 1
|
||||||
|
}
|
||||||
|
|
||||||
|
if tempMap[host.Name] == subqueriesLength {
|
||||||
|
result = append(result, host)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// filterHostsByWord filters hosts according to a single word
|
||||||
|
func (m Model) filterHostsByWord(word string) []config.SSHHost {
|
||||||
var filtered []config.SSHHost
|
var filtered []config.SSHHost
|
||||||
|
|
||||||
if query == "" {
|
if word == "" {
|
||||||
filtered = m.hosts
|
filtered = m.hosts
|
||||||
} else {
|
} else {
|
||||||
query = strings.ToLower(query)
|
word = strings.ToLower(word)
|
||||||
|
|
||||||
for _, host := range m.hosts {
|
for _, host := range m.hosts {
|
||||||
// Check the hostname
|
// Check the hostname
|
||||||
if strings.Contains(strings.ToLower(host.Name), query) {
|
if strings.Contains(strings.ToLower(host.Name), word) {
|
||||||
filtered = append(filtered, host)
|
filtered = append(filtered, host)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the hostname
|
// Check the hostname
|
||||||
if strings.Contains(strings.ToLower(host.Hostname), query) {
|
if strings.Contains(strings.ToLower(host.Hostname), word) {
|
||||||
filtered = append(filtered, host)
|
filtered = append(filtered, host)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the user
|
// Check the user
|
||||||
if strings.Contains(strings.ToLower(host.User), query) {
|
if strings.Contains(strings.ToLower(host.User), word) {
|
||||||
filtered = append(filtered, host)
|
filtered = append(filtered, host)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the tags
|
// Check the tags
|
||||||
for _, tag := range host.Tags {
|
for _, tag := range host.Tags {
|
||||||
if strings.Contains(strings.ToLower(tag), query) {
|
if strings.Contains(strings.ToLower(tag), word) {
|
||||||
filtered = append(filtered, host)
|
filtered = append(filtered, host)
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -34,6 +34,7 @@ type Styles struct {
|
|||||||
|
|
||||||
// Error and confirmation styles
|
// Error and confirmation styles
|
||||||
Error lipgloss.Style
|
Error lipgloss.Style
|
||||||
|
ErrorText lipgloss.Style
|
||||||
|
|
||||||
// Form styles (for add/edit forms)
|
// Form styles (for add/edit forms)
|
||||||
FormTitle lipgloss.Style
|
FormTitle lipgloss.Style
|
||||||
@@ -62,12 +63,14 @@ func NewStyles(width int) Styles {
|
|||||||
SearchFocused: lipgloss.NewStyle().
|
SearchFocused: lipgloss.NewStyle().
|
||||||
BorderStyle(lipgloss.RoundedBorder()).
|
BorderStyle(lipgloss.RoundedBorder()).
|
||||||
BorderForeground(lipgloss.Color(PrimaryColor)).
|
BorderForeground(lipgloss.Color(PrimaryColor)).
|
||||||
Padding(0, 1),
|
Padding(0, 1).
|
||||||
|
Width(50), // Fixed width to prevent expansion
|
||||||
|
|
||||||
SearchUnfocused: lipgloss.NewStyle().
|
SearchUnfocused: lipgloss.NewStyle().
|
||||||
BorderStyle(lipgloss.RoundedBorder()).
|
BorderStyle(lipgloss.RoundedBorder()).
|
||||||
BorderForeground(lipgloss.Color(SecondaryColor)).
|
BorderForeground(lipgloss.Color(SecondaryColor)).
|
||||||
Padding(0, 1),
|
Padding(0, 1).
|
||||||
|
Width(50), // Fixed width to prevent expansion
|
||||||
|
|
||||||
// Table styles
|
// Table styles
|
||||||
TableFocused: lipgloss.NewStyle().
|
TableFocused: lipgloss.NewStyle().
|
||||||
@@ -97,6 +100,11 @@ func NewStyles(width int) Styles {
|
|||||||
BorderForeground(lipgloss.Color(ErrorColor)).
|
BorderForeground(lipgloss.Color(ErrorColor)).
|
||||||
Padding(1, 2),
|
Padding(1, 2),
|
||||||
|
|
||||||
|
// Error text style (no border, just red text)
|
||||||
|
ErrorText: lipgloss.NewStyle().
|
||||||
|
Foreground(lipgloss.Color(ErrorColor)).
|
||||||
|
Bold(true),
|
||||||
|
|
||||||
// Form styles
|
// Form styles
|
||||||
FormTitle: lipgloss.NewStyle().
|
FormTitle: lipgloss.NewStyle().
|
||||||
Foreground(lipgloss.Color("#FFFDF5")).
|
Foreground(lipgloss.Color("#FFFDF5")).
|
||||||
|
|||||||
@@ -17,6 +17,15 @@ import (
|
|||||||
|
|
||||||
// NewModel creates a new TUI model with the given SSH hosts
|
// NewModel creates a new TUI model with the given SSH hosts
|
||||||
func NewModel(hosts []config.SSHHost, configFile, currentVersion string) Model {
|
func NewModel(hosts []config.SSHHost, configFile, currentVersion string) Model {
|
||||||
|
// Load application configuration
|
||||||
|
appConfig, err := config.LoadAppConfig()
|
||||||
|
if err != nil {
|
||||||
|
// Log the error but continue with default configuration
|
||||||
|
fmt.Printf("Warning: Could not load application config: %v, using defaults\n", err)
|
||||||
|
defaultConfig := config.GetDefaultAppConfig()
|
||||||
|
appConfig = &defaultConfig
|
||||||
|
}
|
||||||
|
|
||||||
// Initialize the history manager
|
// Initialize the history manager
|
||||||
historyManager, err := history.NewHistoryManager()
|
historyManager, err := history.NewHistoryManager()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -39,6 +48,7 @@ func NewModel(hosts []config.SSHHost, configFile, currentVersion string) Model {
|
|||||||
sortMode: SortByName,
|
sortMode: SortByName,
|
||||||
configFile: configFile,
|
configFile: configFile,
|
||||||
currentVersion: currentVersion,
|
currentVersion: currentVersion,
|
||||||
|
appConfig: appConfig,
|
||||||
styles: styles,
|
styles: styles,
|
||||||
width: 80,
|
width: 80,
|
||||||
height: 24,
|
height: 24,
|
||||||
|
|||||||
@@ -20,6 +20,8 @@ type (
|
|||||||
versionCheckMsg *version.UpdateInfo
|
versionCheckMsg *version.UpdateInfo
|
||||||
versionErrorMsg error
|
versionErrorMsg error
|
||||||
errorMsg string
|
errorMsg string
|
||||||
|
returnToListMsg struct{}
|
||||||
|
refreshHostsMsg struct{}
|
||||||
)
|
)
|
||||||
|
|
||||||
// startPingAllCmd creates a command to ping all hosts concurrently
|
// startPingAllCmd creates a command to ping all hosts concurrently
|
||||||
@@ -166,6 +168,40 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|||||||
}
|
}
|
||||||
return m, nil
|
return m, nil
|
||||||
|
|
||||||
|
case returnToListMsg:
|
||||||
|
// Return to list view from history
|
||||||
|
m.viewMode = ViewList
|
||||||
|
m.historyView = nil
|
||||||
|
return m, nil
|
||||||
|
|
||||||
|
case refreshHostsMsg:
|
||||||
|
// Refresh hosts after adding from history
|
||||||
|
var hosts []config.SSHHost
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if m.configFile != "" {
|
||||||
|
hosts, err = config.ParseSSHConfigFile(m.configFile)
|
||||||
|
} else {
|
||||||
|
hosts, err = config.ParseSSHConfig()
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
m.hosts = m.sortHosts(hosts)
|
||||||
|
|
||||||
|
// Reapply search filter if there is one active
|
||||||
|
if m.searchInput.Value() != "" {
|
||||||
|
m.filteredHosts = m.filterHosts(m.searchInput.Value())
|
||||||
|
} else {
|
||||||
|
m.filteredHosts = m.hosts
|
||||||
|
}
|
||||||
|
|
||||||
|
m.updateTableRows()
|
||||||
|
m.viewMode = ViewList
|
||||||
|
m.historyView = nil
|
||||||
|
return m, nil
|
||||||
|
|
||||||
case addFormSubmitMsg:
|
case addFormSubmitMsg:
|
||||||
if msg.err != nil {
|
if msg.err != nil {
|
||||||
// Show error in form
|
// Show error in form
|
||||||
@@ -394,9 +430,9 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|||||||
}
|
}
|
||||||
case ViewEdit:
|
case ViewEdit:
|
||||||
if m.editForm != nil {
|
if m.editForm != nil {
|
||||||
var newForm *editFormModel
|
var updatedModel tea.Model
|
||||||
newForm, cmd = m.editForm.Update(msg)
|
updatedModel, cmd = m.editForm.Update(msg)
|
||||||
m.editForm = newForm
|
m.editForm = updatedModel.(*editFormModel)
|
||||||
return m, cmd
|
return m, cmd
|
||||||
}
|
}
|
||||||
case ViewMove:
|
case ViewMove:
|
||||||
@@ -434,6 +470,14 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|||||||
m.fileSelectorForm = newForm
|
m.fileSelectorForm = newForm
|
||||||
return m, cmd
|
return m, cmd
|
||||||
}
|
}
|
||||||
|
case ViewHistory:
|
||||||
|
if m.historyView != nil {
|
||||||
|
newView, cmd := m.historyView.Update(msg)
|
||||||
|
if histView, ok := newView.(HistoryModel); ok {
|
||||||
|
m.historyView = &histView
|
||||||
|
return m, cmd
|
||||||
|
}
|
||||||
|
}
|
||||||
case ViewList:
|
case ViewList:
|
||||||
// Handle list view keys
|
// Handle list view keys
|
||||||
return m.handleListViewKeys(msg)
|
return m.handleListViewKeys(msg)
|
||||||
@@ -445,8 +489,9 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|||||||
|
|
||||||
func (m Model) handleListViewKeys(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
|
func (m Model) handleListViewKeys(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
|
||||||
var cmd tea.Cmd
|
var cmd tea.Cmd
|
||||||
|
key := msg.String()
|
||||||
|
|
||||||
switch msg.String() {
|
switch key {
|
||||||
case "esc", "ctrl+c":
|
case "esc", "ctrl+c":
|
||||||
if m.deleteMode {
|
if m.deleteMode {
|
||||||
// Exit delete mode
|
// Exit delete mode
|
||||||
@@ -455,11 +500,17 @@ func (m Model) handleListViewKeys(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
|
|||||||
m.table.Focus()
|
m.table.Focus()
|
||||||
return m, nil
|
return m, nil
|
||||||
}
|
}
|
||||||
|
// Use configurable key bindings for quit
|
||||||
|
if m.appConfig != nil && m.appConfig.KeyBindings.ShouldQuitOnKey(key) {
|
||||||
return m, tea.Quit
|
return m, tea.Quit
|
||||||
|
}
|
||||||
case "q":
|
case "q":
|
||||||
if !m.searchMode && !m.deleteMode {
|
if !m.searchMode && !m.deleteMode {
|
||||||
|
// Use configurable key bindings for quit
|
||||||
|
if m.appConfig != nil && m.appConfig.KeyBindings.ShouldQuitOnKey(key) {
|
||||||
return m, tea.Quit
|
return m, tea.Quit
|
||||||
}
|
}
|
||||||
|
}
|
||||||
case "/", "ctrl+f":
|
case "/", "ctrl+f":
|
||||||
if !m.searchMode && !m.deleteMode {
|
if !m.searchMode && !m.deleteMode {
|
||||||
// Enter search mode
|
// Enter search mode
|
||||||
@@ -698,6 +749,22 @@ func (m Model) handleListViewKeys(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
|
|||||||
m.viewMode = ViewHelp
|
m.viewMode = ViewHelp
|
||||||
return m, nil
|
return m, nil
|
||||||
}
|
}
|
||||||
|
case "ctrl+h":
|
||||||
|
if !m.searchMode && !m.deleteMode {
|
||||||
|
// Switch to history view
|
||||||
|
if m.historyManager != nil {
|
||||||
|
connections := m.historyManager.GetAllConnectionsInfo()
|
||||||
|
historyView := NewHistoryModel(connections, m.configFile, m.currentVersion)
|
||||||
|
historyView.width = m.width
|
||||||
|
historyView.height = m.height
|
||||||
|
historyView.styles = m.styles
|
||||||
|
// Force table update with correct dimensions
|
||||||
|
historyView.updateTable()
|
||||||
|
m.historyView = &historyView
|
||||||
|
m.viewMode = ViewHistory
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
case "s":
|
case "s":
|
||||||
if !m.searchMode && !m.deleteMode {
|
if !m.searchMode && !m.deleteMode {
|
||||||
// Cycle through sort modes (only 2 modes now)
|
// Cycle through sort modes (only 2 modes now)
|
||||||
|
|||||||
@@ -43,6 +43,10 @@ func (m Model) View() string {
|
|||||||
if m.fileSelectorForm != nil {
|
if m.fileSelectorForm != nil {
|
||||||
return m.fileSelectorForm.View()
|
return m.fileSelectorForm.View()
|
||||||
}
|
}
|
||||||
|
case ViewHistory:
|
||||||
|
if m.historyView != nil {
|
||||||
|
return m.historyView.View()
|
||||||
|
}
|
||||||
case ViewList:
|
case ViewList:
|
||||||
return m.renderListView()
|
return m.renderListView()
|
||||||
}
|
}
|
||||||
@@ -106,7 +110,7 @@ func (m Model) renderListView() string {
|
|||||||
// Add the help text
|
// Add the help text
|
||||||
var helpText string
|
var helpText string
|
||||||
if !m.searchMode {
|
if !m.searchMode {
|
||||||
helpText = " ↑/↓: navigate • Enter: connect • p: ping all • i: info • h: help • q: quit"
|
helpText = " ↑/↓: navigate • Enter: connect • Ctrl+H: history • i: info • h: help • q: quit"
|
||||||
} else {
|
} else {
|
||||||
helpText = " Type to filter • Enter: validate • Tab: switch • ESC: quit"
|
helpText = " Type to filter • Enter: validate • Tab: switch • ESC: quit"
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -128,7 +128,8 @@ func TestValidateIdentityFile(t *testing.T) {
|
|||||||
{"empty path", "", true}, // Optional field
|
{"empty path", "", true}, // Optional field
|
||||||
{"valid file", validFile, true},
|
{"valid file", validFile, true},
|
||||||
{"non-existent file", "/path/to/nonexistent", false},
|
{"non-existent file", "/path/to/nonexistent", false},
|
||||||
{"tilde path", "~/.ssh/id_rsa", true}, // Will pass if file exists
|
// Skip tilde path test in CI environments where ~/.ssh/id_rsa may not exist
|
||||||
|
// {"tilde path", "~/.ssh/id_rsa", true}, // Will pass if file exists
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@@ -138,6 +139,15 @@ func TestValidateIdentityFile(t *testing.T) {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Test tilde path separately, but only if the file actually exists
|
||||||
|
t.Run("tilde path", func(t *testing.T) {
|
||||||
|
tildeFile := "~/.ssh/id_rsa"
|
||||||
|
// Just test that it doesn't crash, don't assume file exists
|
||||||
|
result := ValidateIdentityFile(tildeFile)
|
||||||
|
// Result can be true or false depending on file existence
|
||||||
|
_ = result // We just care that it doesn't panic
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestValidateHost(t *testing.T) {
|
func TestValidateHost(t *testing.T) {
|
||||||
|
|||||||
Reference in New Issue
Block a user