auto_ssh_setup.sh

#!/bin/bash

# GitHub Repository: https://github.com/Newton-Nganga

# Function to log messages
log() {
  echo -e "\n[LOG] $1"
}

# Function to log errors
error() {
  echo -e "\n[ERROR] $1"
}

# Function to log success messages
success() {
  echo -e "\n[SUCCESS] $1"
}

# Function to display a header
display_header() {
  clear
  echo "=============================================="
  echo "          SSH Setup Script for GitHub/Bitbucket"
  echo "          Created by: Newton-Nganga"
  echo "          GitHub: https://github.com/Newton-Nganga"
  echo "=============================================="
  echo ""
}

# Function to check and set Git global username and email
setup_git_config() {
  local git_user=$(git config --global user.name)
  local git_email=$(git config --global user.email)

  if [ -z "$git_user" ]; then
    log "Git global username is not set."
    read -p "Enter your Git username: " git_user
    git config --global user.name "$git_user"
  fi

  if [ -z "$git_email" ]; then
    log "Git global email is not set."
    read -p "Enter your Git email: " git_email
    git config --global user.email "$git_email"
  fi

  success "Git global username and email are set up!"
}

# Function to create or clear the ~/.ssh directory
setup_ssh_directory() {
  if [ -d ~/.ssh ]; then
    log "~/.ssh directory already exists. Clearing contents..."
    rm -rf ~/.ssh/*
  else
    log "Creating ~/.ssh directory..."
    mkdir -p ~/.ssh
  fi
  chmod 700 ~/.ssh
}

# Function to generate SSH keys
generate_ssh_key() {
  local key_name=$1
  local email=$2
  log "Generating SSH key for $email..."
  ssh-keygen -t ed25519 -C "$email" -f ~/.ssh/"$key_name"
  chmod 600 ~/.ssh/"$key_name"
  chmod 644 ~/.ssh/"$key_name.pub"
  success "SSH key generated successfully!"
}

# Function to add SSH key to the config file
add_to_config() {
  local key_name=$1
  local host=$2
  log "Adding $key_name to SSH config for $host..."
  cat <<EOF >> ~/.ssh/config
Host $host
  HostName $host
  IdentityFile ~/.ssh/$key_name
  User git
EOF
  success "SSH config updated for $host!"
}

# Function to display the public key
display_public_key() {
  local key_name=$1
  log "Here is your public key. Copy and add it to your account:"
  echo -e "\n$(cat ~/.ssh/"$key_name.pub")\n"
}

# Function to test the SSH connection
test_ssh_connection() {
  local host=$1
  log "Testing SSH connection to $host..."
  ssh -T git@$host
  if [ $? -eq 0 ]; then
    success "SSH connection to $host is working!"
  else
    error "SSH connection to $host failed. Please check your setup."
  fi
}

# Main script logic
display_header

# Check and set Git global username and email
setup_git_config

# Ask the user to choose a platform
PS3="Select the platform you want to set up SSH for: "
options=("GitHub" "Bitbucket" "Quit")
select opt in "${options[@]}"; do
  case $opt in
    "GitHub")
      platform="github.com"
      break
      ;;
    "Bitbucket")
      platform="bitbucket.org"
      break
      ;;
    "Quit")
      log "Exiting the script. Goodbye!"
      exit 0
      ;;
    *)
      error "Invalid option. Please try again."
      ;;
  esac
done

# Set up the SSH directory
setup_ssh_directory

# Ask for the key name
log "Please provide a name for your SSH key (e.g., id_github):"
read -r key_name

# Ask for the email associated with the account
log "Please enter the email associated with your $platform account:"
read -r email

# Generate the SSH key
generate_ssh_key "$key_name" "$email"

# Add the key to the SSH config
add_to_config "$key_name" "$platform"

# Display the public key
display_public_key "$key_name"

# Wait for the user to add the key to their account
log "Please add the public key to your $platform account. Once done, type 'yes' to continue."
while true; do
  read -r response
  if [ "$response" = "yes" ]; then
    break
  else
    error "Invalid input. Please type 'yes' to continue."
  fi
done

# Test the SSH connection
test_ssh_connection "$platform"

# Guide the user on cloning a repository
log "To clone a repository, use the following command:"
echo -e "\ngit clone git@$platform:username/repository.git\n"

# Change directory to ~
cd ~ || exit

# End of script
success "SSH setup completed successfully! You are now in the ~ directory."
log "Script terminated. Goodbye!"