#!/bin/sh

#================================================================#

# GitSSH SETUP MODULE

# Interactive SSH key setup for GitHub and GitLab

#================================================================#

# Configuration

SSH_CONFIG_FILE="$HOME/.ssh/config"

SSH_DIR="$HOME/.ssh"

#================================================================#

# MAIN SETUP COMMANDS

#================================================================#

# Setup GitHub SSH key

setup_github() {

if ! _check_setup_dependencies; then

return 1

fi

printf "\n"

_print_color blue "=========================================="

printf "\n"

_print_color blue " GitHub SSH Key Setup Wizard"

printf "\n"

_print_color blue "=========================================="

printf "\n\n"

_print_info "This wizard will help you set up SSH authentication for GitHub"

printf "We'll guide you through each step with explanations.\n\n"

# Step 1: Gather user information

if ! _gather_github_info; then

return 1

fi

# Step 2: Check existing SSH setup

if ! _check_existing_ssh_setup "github"; then

return 1

fi

# Step 3: Generate SSH key if needed

if [ "$GENERATE_NEW_KEY" = "true" ]; then

if ! _generate_ssh_key "github"; then

return 1

fi

fi

# Step 4: Configure SSH

if ! _setup_ssh_config "github"; then

return 1

fi

# Step 5: Add to SSH agent

if ! _add_to_ssh_agent "github"; then

return 1

fi

# Step 6: Display public key and guide user

if ! _guide_github_key_addition; then

return 1

fi

# Step 7: Test connection

if ! _test_github_connection; then

return 1

fi

# Step 8: Integration with GitSSH Session Manager

_integrate_with_session_manager "github"

_print_success "GitHub SSH setup completed successfully!"

printf "\n"

_print_info "You can now use the GitSSH Session Manager:"

printf " • Run 'gitssh session set' to configure your identity for repositories\n"

printf " • Use 'gitssh clone', 'gitssh push', etc. for enhanced Git operations\n"

printf " • Run 'gitssh remote convert' to convert HTTPS repositories to SSH\n"

printf "\n"

}

# Setup GitLab SSH key

setup_gitlab() {

if ! _check_setup_dependencies; then

return 1

fi

printf "\n"

_print_color blue "=========================================="

printf "\n"

_print_color blue " GitLab SSH Key Setup Wizard"

printf "\n"

_print_color blue "=========================================="

printf "\n\n"

_print_info "This wizard will help you set up SSH authentication for GitLab"

printf "We'll guide you through each step with explanations.\n\n"

# Step 1: Gather user information

if ! _gather_gitlab_info; then

return 1

fi

# Step 2: Check existing SSH setup

if ! _check_existing_ssh_setup "gitlab"; then

return 1

fi

# Step 3: Generate SSH key if needed

if [ "$GENERATE_NEW_KEY" = "true" ]; then

if ! _generate_ssh_key "gitlab"; then

return 1

fi

fi

# Step 4: Configure SSH

if ! _setup_ssh_config "gitlab"; then

return 1

fi

# Step 5: Add to SSH agent

if ! _add_to_ssh_agent "gitlab"; then

return 1

fi

# Step 6: Display public key and guide user

if ! _guide_gitlab_key_addition; then

return 1

fi

# Step 7: Test connection

if ! _test_gitlab_connection; then

return 1

fi

# Step 8: Integration with GitSSH Session Manager

_integrate_with_session_manager "gitlab"

_print_success "GitLab SSH setup completed successfully!"

printf "\n"

_print_info "You can now use the GitSSH Session Manager:"

printf " • Run 'gitssh session set' to configure your identity for repositories\n"

printf " • Use 'gitssh clone', 'gitssh push', etc. for enhanced Git operations\n"

printf " • Run 'gitssh remote convert' to convert HTTPS repositories to SSH\n"

printf "\n"

}

# Show SSH setup status

ssh_status() {

printf "\n"

_print_color blue "=========================================="

printf "\n"

_print_color blue " SSH Setup Status"

printf "\n"

_print_color blue "=========================================="

printf "\n\n"

# Check SSH directory

printf "SSH Directory Status:\n"

if [ -d "$SSH_DIR" ]; then

_print_success "SSH directory exists: $SSH_DIR"

printf " Permissions: %s\n" "$(ls -ld "$SSH_DIR" | awk '{print $1}')"

else

_print_warning "SSH directory not found: $SSH_DIR"

fi

printf "\n"

# Check SSH keys

printf "SSH Keys:\n"

key_count=0

for key_type in ed25519 rsa ecdsa dsa; do

if [ -f "$SSH_DIR/id_$key_type" ]; then

_print_success "Found $key_type key: id_$key_type"

key_count=$((key_count + 1))

fi

done

# Check for custom GitHub/GitLab keys

for service in github gitlab; do

for key_type in ed25519 rsa; do

if [ -f "$SSH_DIR/${service}_$key_type" ]; then

_print_success "Found $service $key_type key: ${service}_$key_type"

key_count=$((key_count + 1))

fi

done

done

if [ "$key_count" -eq 0 ]; then

_print_warning "No SSH keys found"

printf " Run 'gitssh setup github' or 'gitssh setup gitlab' to create keys\n"

fi

printf "\n"

# Check SSH agent

printf "SSH Agent Status:\n"

if _is_ssh_agent_running; then

_print_success "SSH Agent is running"

printf "Loaded keys:\n"

_get_loaded_ssh_keys | sed 's/^/ /'

else

_print_warning "SSH Agent not running"

printf " Start with: eval \$(ssh-agent -s)\n"

fi

printf "\n"

# Check SSH config

printf "SSH Configuration:\n"

if [ -f "$SSH_CONFIG_FILE" ]; then

_print_success "SSH config exists: $SSH_CONFIG_FILE"

# Check for GitHub/GitLab hosts

github_hosts=$(grep '^Host github-' "$SSH_CONFIG_FILE" 2>/dev/null | wc -l)

gitlab_hosts=$(grep '^Host gitlab-' "$SSH_CONFIG_FILE" 2>/dev/null | wc -l)

printf " GitHub hosts configured: %s\n" "$github_hosts"

printf " GitLab hosts configured: %s\n" "$gitlab_hosts"

else

_print_warning "SSH config not found"

printf " Will be created during setup\n"

fi

printf "\n"

# Test connections

printf "Connection Tests:\n"

_test_service_connections

printf "\n"

_print_color blue "=========================================="

printf "\n"

}

# Test SSH

ssh_test() {

if [ $# -eq 0 ]; then

printf "Usage: gitssh ssh test <hostname> [timeout]\n"

printf "Example: gitssh ssh test github-work 10\n"

return 1

fi

hostname="$1"

timeout="${2:-5}"

printf "Testing SSH connection to %s...\n" "$hostname"

if _test_ssh_connection "$hostname" "$timeout"; then

_print_success "SSH connection to $hostname successful!"

# Try to get more details

test_output=$(ssh -o ConnectTimeout="$timeout" -T "git@$hostname" 2>&1)

if printf "%s" "$test_output" | grep -q "successfully authenticated\|Welcome to GitLab"; then

printf "Authentication details:\n"

printf "%s\n" "$test_output" | head -3

fi

else

_print_error "SSH connection to $hostname failed"

printf "Troubleshooting:\n"

printf " 1. Check if SSH host exists: grep 'Host %s' ~/.ssh/config\n" "$hostname"

printf " 2. Test SSH agent: ssh-add -l\n"

printf " 3. Run diagnostics: gitssh ssh doctor\n"

return 1

fi

}

#================================================================#

# INFORMATION GATHERING

#================================================================#

# Gather GitHub user information

_gather_github_info() {

printf "Step 1: GitHub Account Information\n"

printf "===================================\n"

printf "GitHub username: "

read -r GITHUB_USERNAME

if _is_empty "$GITHUB_USERNAME"; then

_print_error "GitHub username cannot be empty"

return 1

fi

if ! _validate_username "$GITHUB_USERNAME"; then

_print_error "Invalid GitHub username format"

return 1

fi

printf "Email address (for Git commits): "

read -r USER_EMAIL

if _is_empty "$USER_EMAIL"; then

_print_error "Email address cannot be empty"

return 1

fi

if ! _validate_email "$USER_EMAIL"; then

_print_error "Invalid email format"

return 1

fi

printf "Display name [%s]: " "$GITHUB_USERNAME"

read -r USER_NAME

USER_NAME=${USER_NAME:-$GITHUB_USERNAME}

USER_NAME=$(_trim "$USER_NAME")

# Set SSH host name

SSH_HOST="github-$GITHUB_USERNAME"

KEY_COMMENT="$USER_EMAIL"

printf "\n"

_print_success "Account information collected:"

printf " Username: %s\n" "$GITHUB_USERNAME"

printf " Name: %s\n" "$USER_NAME"

printf " Email: %s\n" "$USER_EMAIL"

printf " SSH Host: %s\n" "$SSH_HOST"

printf "\n"

return 0

}

# Gather GitLab user information

_gather_gitlab_info() {

printf "Step 1: GitLab Account Information\n"

printf "===================================\n"

printf "GitLab instance URL [gitlab.com]: "

read -r GITLAB_URL

GITLAB_URL=${GITLAB_URL:-gitlab.com}

GITLAB_URL=$(_trim "$GITLAB_URL")

# Remove protocol if provided

GITLAB_URL=$(printf "%s" "$GITLAB_URL" | sed 's|^https\?://||')

printf "GitLab username: "

read -r GITLAB_USERNAME

if _is_empty "$GITLAB_USERNAME"; then

_print_error "GitLab username cannot be empty"

return 1

fi

if ! _validate_username "$GITLAB_USERNAME"; then

_print_error "Invalid GitLab username format"

return 1

fi

printf "Email address (for Git commits): "

read -r USER_EMAIL

if _is_empty "$USER_EMAIL"; then

_print_error "Email address cannot be empty"

return 1

fi

if ! _validate_email "$USER_EMAIL"; then

_print_error "Invalid email format"

return 1

fi

printf "Display name [%s]: " "$GITLAB_USERNAME"

read -r USER_NAME

USER_NAME=${USER_NAME:-$GITLAB_USERNAME}

USER_NAME=$(_trim "$USER_NAME")

# Set SSH host name

if [ "$GITLAB_URL" = "gitlab.com" ]; then

SSH_HOST="gitlab-$GITLAB_USERNAME"

else

# For custom GitLab instances, include domain

sanitized_domain=$(printf "%s" "$GITLAB_URL" | sed 's/[^a-zA-Z0-9]/-/g')

SSH_HOST="gitlab-$sanitized_domain-$GITLAB_USERNAME"

fi

KEY_COMMENT="$USER_EMAIL"

printf "\n"

_print_success "Account information collected:"

printf " GitLab URL: %s\n" "$GITLAB_URL"

printf " Username: %s\n" "$GITLAB_USERNAME"

printf " Name: %s\n" "$USER_NAME"

printf " Email: %s\n" "$USER_EMAIL"

printf " SSH Host: %s\n" "$SSH_HOST"

printf "\n"

return 0

}

#================================================================#

# SSH KEY MANAGEMENT

#================================================================#

# Check existing SSH setup

_check_existing_ssh_setup() {

service="$1" # github or gitlab

printf "Step 2: Checking Existing SSH Setup\n"

printf "====================================\n"

# Check SSH directory

if [ ! -d "$SSH_DIR" ]; then

printf "Creating SSH directory...\n"

mkdir -p "$SSH_DIR"

chmod 700 "$SSH_DIR"

_print_success "Created $SSH_DIR with proper permissions"

else

_print_success "SSH directory exists: $SSH_DIR"

fi

# Check for existing keys

existing_keys=""

key_files=""

# Check for service-specific keys first

for key_type in ed25519 rsa; do

key_name="${service}_${key_type}"

if [ -f "$SSH_DIR/$key_name" ] && [ -f "$SSH_DIR/$key_name.pub" ]; then

existing_keys="$existing_keys $key_name"

key_files="$key_files $SSH_DIR/$key_name"

fi

done

# Check for default keys

for key_type in ed25519 rsa ecdsa dsa; do

key_name="id_$key_type"

if [ -f "$SSH_DIR/$key_name" ] && [ -f "$SSH_DIR/$key_name.pub" ]; then

existing_keys="$existing_keys $key_name"

key_files="$key_files $SSH_DIR/$key_name"

fi

done

if [ -n "$existing_keys" ]; then

_print_info "Found existing SSH keys:"

for key in $existing_keys; do

printf " • %s\n" "$key"

done

printf "\n"

printf "Options:\n"

printf " 1. Use existing key\n"

printf " 2. Generate new key\n"

printf " 3. Exit setup\n"

printf "\n"

while true; do

printf "Choose option (1-3): "

read -r choice

case "$choice" in

1)

GENERATE_NEW_KEY="false"

if ! _select_existing_key; then

return 1

fi

break

;;

2)

GENERATE_NEW_KEY="true"

_generate_key_name "$service"

break

;;

3)

printf "Setup cancelled\n"

return 1

;;

*)

_print_warning "Invalid choice. Please enter 1, 2, or 3."

;;

esac

done

else

_print_info "No existing SSH keys found"

GENERATE_NEW_KEY="true"

_generate_key_name "$service"

fi

printf "\n"

return 0

}

# Select existing SSH key

_select_existing_key() {

printf "\nSelect SSH key to use:\n"

i=1

key_list=""

for key_type in ed25519 rsa ecdsa dsa; do

for prefix in "${service}_" "id_"; do

key_name="${prefix}${key_type}"

if [ -f "$SSH_DIR/$key_name" ] && [ -f "$SSH_DIR/$key_name.pub" ]; then

printf " %d. %s\n" "$i" "$key_name"

key_list="$key_list $key_name"

i=$((i + 1))

fi

done

done

if [ "$i" -eq 1 ]; then

_print_error "No valid key pairs found"

return 1

fi

while true; do

printf "\nSelect key (1-%d): " "$((i-1))"

read -r choice

if [ "$choice" -ge 1 ] && [ "$choice" -le "$((i-1))" ] 2>/dev/null; then

SELECTED_KEY=$(printf "%s" "$key_list" | awk "{print \$$choice}")

KEY_PATH="$SSH_DIR/$SELECTED_KEY"

_print_success "Selected: $SELECTED_KEY"

break

else

_print_warning "Invalid choice"

fi

done

return 0

}

# Generate key name for new key

_generate_key_name() {

service="$1"

# Use ed25519 as default for new keys (best practice)

case "$service" in

github)

KEY_NAME="github_${GITHUB_USERNAME}_id_ed25519"

;;

gitlab)

if [ "$GITLAB_URL" = "gitlab.com" ]; then

KEY_NAME="gitlab_${GITLAB_USERNAME}_id_ed25519"

else

sanitized_domain=$(printf "%s" "$GITLAB_URL" | sed 's/[^a-zA-Z0-9]/-/g')

KEY_NAME="gitlab-${sanitized_domain}_id_ed25519"

fi

;;

esac

KEY_PATH="$SSH_DIR/$KEY_NAME"

_print_info "Will generate new key: $KEY_NAME"

}

# Generate SSH key with user guidance

_generate_ssh_key() {

service="$1"

printf "Step 3: Generating SSH Key\n"

printf "==========================\n"

_print_info "About SSH Keys:"

printf "SSH keys provide secure, passwordless authentication to Git services.\n"

printf "We'll use Ed25519 encryption (recommended).\n\n"

# Check if key already exists

if [ -f "$KEY_PATH" ]; then

_print_warning "Key file already exists: $KEY_PATH"

printf "Overwrite existing key? (y/N): "

read -r overwrite

case "$overwrite" in

[Yy]*) ;;

*)

_print_info "Using existing key"

return 0

;;

esac

fi

printf "Generating SSH key...\n"

printf "Key type: Ed25519 (recommended)\n"

printf "Key file: %s\n" "$KEY_PATH"

printf "Comment: %s\n\n" "$KEY_COMMENT"

_print_info "About Passphrases:"

printf "A passphrase adds extra security to your SSH key.\n"

printf "Recommended: Use a strong passphrase\n"

printf "Note: You can use ssh-agent to avoid entering it repeatedly\n\n"

# Generate the key

if ssh-keygen -t ed25519 -f "$KEY_PATH" -C "$KEY_COMMENT"; then

_print_success "SSH key generated successfully!"

# Set proper permissions

chmod 600 "$KEY_PATH"

chmod 644 "$KEY_PATH.pub"

printf "\nKey details:\n"

printf " Private key: %s (keep this secret!)\n" "$KEY_PATH"

printf " Public key: %s (this will be added to $service)\n" "$KEY_PATH.pub"

printf " Fingerprint: "gitssh ssh

ssh-keygen -lf "$KEY_PATH.pub" 2>/dev/null | awk '{print $2}'

else

_print_error "Failed to generate SSH key"

return 1

fi

printf "\n"

return 0

}

# Setup SSH configuration

_setup_ssh_config() {

service="$1"

printf "Step 4: Configuring SSH\n"

printf "=======================\n"

_print_info "About SSH Config:"

printf "The SSH config file (~/.ssh/config) tells SSH how to connect to different hosts.\n"

printf "This allows you to use custom hostnames like 'github-username'.\n\n"

# Determine hostname and host entry

case "$service" in

github)

REAL_HOSTNAME="github.com"

HOST_ENTRY="Host $SSH_HOST

HostName github.com

User git

IdentityFile $KEY_PATH

IdentitiesOnly yes"

;;

gitlab)

REAL_HOSTNAME="$GITLAB_URL"

HOST_ENTRY="Host $SSH_HOST

HostName $GITLAB_URL

User git

IdentityFile $KEY_PATH

IdentitiesOnly yes"

;;

esac

# Create SSH config if it doesn't exist

if [ ! -f "$SSH_CONFIG_FILE" ]; then

printf "Creating SSH config file...\n"

touch "$SSH_CONFIG_FILE"

chmod 600 "$SSH_CONFIG_FILE"

_print_success "Created $SSH_CONFIG_FILE"

fi

# Check if host entry already exists

if grep -q "^Host $SSH_HOST$" "$SSH_CONFIG_FILE"; then

_print_warning "SSH host '$SSH_HOST' already exists in config"

printf "Update the existing entry? (Y/n): "

read -r update_config

case "$update_config" in

[Nn]*)

_print_info "Keeping existing SSH config"

return 0

;;

esac

# Remove existing entry

_remove_ssh_host_entry "$SSH_HOST"

fi

# Add new host entry

printf "Adding SSH host configuration...\n"

printf "\n# %s SSH configuration\n%s\n\n" "$service" "$HOST_ENTRY" >> "$SSH_CONFIG_FILE"

_print_success "Added SSH configuration for $SSH_HOST"

printf " You can now use: git@%s:username/repository.git\n" "$SSH_HOST"

printf "\n"

return 0

}

# Add key to SSH agent

_add_to_ssh_agent() {

service="$1"

printf "Step 5: Adding Key to SSH Agent\n"

printf "================================\n"

_print_info "About SSH Agent:"

printf "SSH Agent manages your SSH keys and handles authentication.\n"

printf "This allows you to enter your passphrase once per session.\n\n"

# Check if SSH agent is running

if ! _is_ssh_agent_running; then

printf "Starting SSH agent...\n"

eval "$(ssh-agent -s)" >/dev/null

if _is_ssh_agent_running; then

_print_success "SSH agent started"

else

_print_error "Failed to start SSH agent"

return 1

fi

else

_print_success "SSH agent is already running"

fi

# Add key to agent

printf "Adding SSH key to agent...\n"

if ssh-add "$KEY_PATH"; then

_print_success "SSH key added to agent successfully!"

# Show loaded keys

printf "\nCurrently loaded keys:\n"

_get_loaded_ssh_keys | sed 's/^/ /'

else

_print_error "Failed to add SSH key to agent"

return 1

fi

printf "\n"

return 0

}

#================================================================#

# SERVICE-SPECIFIC GUIDES

#================================================================#

# Guide user through GitHub key addition

_guide_github_key_addition() {

printf "Step 6: Adding Key to GitHub\n"

printf "============================\n"

_print_info "Now you need to add your public key to GitHub:"

printf "\n"

# Display the public key

printf "Your public key (copy this entire text):\n"

printf "%s\n" "$(cat "$KEY_PATH.pub")"

printf "%s\n" "$(printf '=%.0s' {1..60})"

printf "\n"

_print_info "GitHub Setup Steps:"

printf "1. Go to https://github.com/settings/keys\n"

printf "2. Click 'New SSH key'\n"

printf "3. Give it a title (e.g., 'My Laptop - %s')\n" "$(date '+%Y-%m-%d')"

printf "4. Paste the public key above into the 'Key' field\n"

printf "5. Click 'Add SSH key'\n\n"

# Auto-copy to clipboard if possible

if command -v xclip >/dev/null 2>&1; then

cat "$KEY_PATH.pub" | xclip -selection clipboard

_print_success "Public key copied to clipboard!"

elif command -v pbcopy >/dev/null 2>&1; then

cat "$KEY_PATH.pub" | pbcopy

_print_success "Public key copied to clipboard!"

elif command -v wl-copy >/dev/null 2>&1; then

cat "$KEY_PATH.pub" | wl-copy

_print_success "Public key copied to clipboard!"

else

_print_info "Tip: Select and copy the public key above"

fi

printf "Press Enter after adding the key to GitHub..."

read -r _

return 0

}

# Guide user through GitLab key addition

_guide_gitlab_key_addition() {

printf "Step 6: Adding Key to GitLab\n"

printf "============================\n"

_print_info "Now you need to add your public key to GitLab:"

printf "\n"

# Display the public key

printf "Your public key (copy this entire text):\n"

printf "%s\n" "$(cat "$KEY_PATH.pub")"

printf "%s\n" "$(printf '=%.0s' {1..60})"

printf "\n"

_print_info "GitLab Setup Steps:"

if [ "$GITLAB_URL" = "gitlab.com" ]; then

printf "1. Go to https://gitlab.com/-/user_settings/ssh_keys\n"

else

printf "1. Go to https://%s/-/user_settings/ssh_keys\n" "$GITLAB_URL"

fi

printf "2. Paste the public key above into the 'Key' field\n"

printf "3. Give it a title (e.g., 'My Laptop - %s')\n" "$(date '+%Y-%m-%d')"

printf "4. Set an expiration date (optional but recommended)\n"

printf "5. Click 'Add key'\n\n"

# Auto-copy to clipboard if possible

if command -v xclip >/dev/null 2>&1; then

cat "$KEY_PATH.pub" | xclip -selection clipboard

_print_success "Public key copied to clipboard!"

elif command -v pbcopy >/dev/null 2>&1; then

cat "$KEY_PATH.pub" | pbcopy

_print_success "Public key copied to clipboard!"

elif command -v wl-copy >/dev/null 2>&1; then

cat "$KEY_PATH.pub" | wl-copy

_print_success "Public key copied to clipboard!"

else

_print_info "Tip: Select and copy the public key above"

fi

printf "Press Enter after adding the key to GitLab..."

read -r _

return 0

}

#================================================================#

# CONNECTION TESTING

#================================================================#

# Test GitHub connection

_test_github_connection() {

printf "Step 7: Testing GitHub Connection\n"

printf "==================================\n"

_print_info "Testing SSH connection to GitHub..."

printf "This verifies that your key was added correctly.\n\n"

printf "Testing connection to %s...\n" "$SSH_HOST"

# Test with timeout and capture output

test_output=$(ssh -o ConnectTimeout=10 -T "git@$SSH_HOST" 2>&1)

test_result=$?

if printf "%s" "$test_output" | grep -q "successfully authenticated"; then

_print_success "SSH connection successful!"

# Extract username from GitHub response

github_user=$(printf "%s" "$test_output" | sed -n 's/.*Hi \([^!]*\)!.*/\1/p')

if [ -n "$github_user" ]; then

printf "Authenticated as GitHub user: %s\n" "$github_user"

if [ "$github_user" != "$GITHUB_USERNAME" ]; then

_print_warning "Username mismatch!"

printf " Expected: %s\n" "$GITHUB_USERNAME"

printf " Actual: %s\n" "$github_user"

printf "This might indicate a key conflict or wrong account.\n"

fi

fi

else

_print_error "SSH connection failed"

printf "Error output:\n%s\n\n" "$test_output"

_print_info "Troubleshooting tips:"

printf "1. Make sure you added the public key to the correct GitHub account\n"

printf "2. Check that the key was pasted completely without extra spaces\n"

printf "3. Verify your SSH agent is running: ssh-add -l\n"

printf "4. Try adding the key to SSH agent again: ssh-add %s\n" "$KEY_PATH"

printf "\nWould you like to try again? (y/N): "

read -r retry

case "$retry" in

[Yy]*) return _test_github_connection ;;

*) return 1 ;;

esac

fi

printf "\n"

return 0

}

# Test GitLab connection

_test_gitlab_connection() {

printf "Step 7: Testing GitLab Connection\n"

printf "==================================\n"

_print_info "Testing SSH connection to GitLab..."

printf "This verifies that your key was added correctly.\n\n"

printf "Testing connection to %s...\n" "$SSH_HOST"

# Test with timeout and capture output

test_output=$(ssh -o ConnectTimeout=10 -T "git@$SSH_HOST" 2>&1)

test_result=$?

if printf "%s" "$test_output" | grep -q "Welcome to GitLab"; then

_print_success "SSH connection successful!"

# Extract username from GitLab response if available

gitlab_user=$(printf "%s" "$test_output" | sed -n 's/.*Welcome to GitLab, @\([^!]*\)!.*/\1/p')

if [ -n "$gitlab_user" ]; then

printf "Authenticated as GitLab user: %s\n" "$gitlab_user"

if [ "$gitlab_user" != "$GITLAB_USERNAME" ]; then

_print_warning "Username mismatch!"

printf " Expected: %s\n" "$GITLAB_USERNAME"

printf " Actual: %s\n" "$gitlab_user"

printf "This might indicate a key conflict or wrong account.\n"

fi

fi

else

_print_error "SSH connection failed"

printf "Error output:\n%s\n\n" "$test_output"

_print_info "Troubleshooting tips:"

printf "1. Make sure you added the public key to the correct GitLab account\n"

printf "2. Check that the key was pasted completely without extra spaces\n"

printf "3. Verify your SSH agent is running: ssh-add -l\n"

printf "4. Try adding the key to SSH agent again: ssh-add %s\n" "$KEY_PATH"

printf "\nWould you like to try again? (y/N): "

read -r retry

case "$retry" in

[Yy]*) return _test_gitlab_connection ;;

*) return 1 ;;

esac

fi

printf "\n"

return 0

}

#================================================================#

# INTEGRATION FUNCTIONS

#================================================================#

# Integrate with GitSSH Session Manager

_integrate_with_session_manager() {

service="$1"

printf "Step 8: Integration with GitSSH Session Manager\n"

printf "================================================\n"

_print_info "About GitSSH Session Manager:"

printf "This tool helps you manage multiple Git identities and SSH keys.\n"

printf "It automatically configures the right user for each repository.\n\n"

# Check if GitSSH Session Manager is available

if ! _check_dependencies 2>/dev/null; then

_print_warning "GitSSH Session Manager not fully configured"

printf "Run 'gitssh init' to complete the setup\n\n"

return 0

fi

# Add user to session manager if not exists

case "$service" in

github)

username="$GITHUB_USERNAME"

;;

gitlab)

username="$GITLAB_USERNAME"

;;

esac

if ! _user_exists "$username" 2>/dev/null; then

printf "Adding user to GitSSH Session Manager...\n"

# Add user to config

if _add_user_to_config "$username" "$USER_NAME" "$USER_EMAIL" "$SSH_HOST"; then

_print_success "Added '$username' to session manager"

else

_print_warning "Failed to add user to session manager"

printf "You can add manually later with 'gitssh user add'\n"

fi

else

_print_success "User '$username' already exists in session manager"

fi

printf "\n"

return 0

}

#================================================================#

# UTILITY FUNCTIONS

#================================================================#

# Check setup dependencies

_check_setup_dependencies() {

missing_deps=""

# Check for required commands

for cmd in ssh-keygen ssh-add ssh git; do

if ! command -v "$cmd" >/dev/null 2>&1; then

missing_deps="$missing_deps $cmd"

fi

done

if [ -n "$missing_deps" ]; then

_print_error "Missing required dependencies:$missing_deps"

printf "Please install them first:\n"

os=$(_detect_os)

case "$os" in

linux)

printf " Ubuntu/Debian: sudo apt install openssh-client git\n"

printf " RHEL/CentOS: sudo yum install openssh-clients git\n"

printf " Arch: sudo pacman -S openssh git\n"

;;

macos)

printf " macOS: git and ssh should be pre-installed\n"

printf " If missing: xcode-select --install\n"

;;

esac

return 1

fi

return 0

}

# Remove existing SSH host entry

_remove_ssh_host_entry() {

host_name="$1"

temp_file=$(_create_temp_file)

# Remove the host block (Host line and all following indented lines)

awk -v host="$host_name" '

BEGIN { skip = 0 }

/^Host / {

if ($2 == host) {

skip = 1

} else {

skip = 0

}

}

/^[[:space:]]/ && skip { next }

/^[^[:space:]#]/ && !/^Host / { skip = 0 }

!skip { print }

' "$SSH_CONFIG_FILE" > "$temp_file"

_safe_file_replace "$temp_file" "$SSH_CONFIG_FILE"

}