ghset

#!/bin/bash

# Script para alternar perfis SSH
# Uso: ghset <perfil>
# Perfis disponíveis: belmo, oppy, personal

set -e

PROFILE="$1"
SSH_CONFIG="$HOME/.ssh/config"
BACKUP_CONFIG="$HOME/.ssh/config.backup"

# Cores para output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

show_help() {
    echo "Uso: ghset <perfil>"
    echo ""
    echo "Perfis disponíveis:"
    echo "  belmo    - Perfil pessoal (chave id_github)"
    echo "  oppy     - Perfil empresa (chave id_oppy)"
    echo "  personal - Perfil pessoal (chave id_ed25519)"
    echo ""
    echo "Exemplos:"
    echo "  ghset belmo"
    echo "  ghset oppy"
    echo "  ghset personal"
}

backup_config() {
    if [[ -f "$SSH_CONFIG" ]]; then
        cp "$SSH_CONFIG" "$BACKUP_CONFIG"
        echo -e "${YELLOW}Backup criado: $BACKUP_CONFIG${NC}"
    fi
}

create_config() {
    local profile="$1"
    
    case "$profile" in
        "belmo")
            cat > "$SSH_CONFIG" << 'EOF'
# GitHub - Perfil Belmo (Pessoal)
Host github.com
  HostName github.com
  User git
  IdentityFile ~/.ssh/id_github
  IdentitiesOnly yes

# Bitbucket
Host bitbucket.org
  HostName bitbucket.org
  User git
  IdentityFile ~/.ssh/id_bitbucket
  IdentitiesOnly yes

# Repositórios da empresa (desabilitado no perfil belmo)
# Host github.com-koware
#   HostName github.com
#   User git
#   IdentityFile ~/.ssh/id_oppy
#   IdentitiesOnly yes
EOF
            ;;
        "oppy")
            cat > "$SSH_CONFIG" << 'EOF'
# GitHub - Perfil Oppy (Empresa)
Host github.com
  HostName github.com
  User git
  IdentityFile ~/.ssh/id_oppy
  IdentitiesOnly yes

# Bitbucket
Host bitbucket.org
  HostName bitbucket.org
  User git
  IdentityFile ~/.ssh/id_bitbucket
  IdentitiesOnly yes

# Repositórios da empresa
Host github.com-koware
  HostName github.com
  User git
  IdentityFile ~/.ssh/id_oppy
  IdentitiesOnly yes
EOF
            ;;
        "personal")
            cat > "$SSH_CONFIG" << 'EOF'
# GitHub - Perfil Personal (Pessoal)
Host github.com
  HostName github.com
  User git
  IdentityFile ~/.ssh/id_ed25519
  IdentitiesOnly yes

# Bitbucket
Host bitbucket.org
  HostName bitbucket.org
  User git
  IdentityFile ~/.ssh/id_bitbucket
  IdentitiesOnly yes

# Repositórios da empresa (desabilitado no perfil personal)
# Host github.com-koware
#   HostName github.com
#   User git
#   IdentityFile ~/.ssh/id_oppy
#   IdentitiesOnly yes
EOF
            ;;
        *)
            echo -e "${RED}Erro: Perfil '$profile' não encontrado${NC}"
            show_help
            exit 1
            ;;
    esac
}

verify_keys() {
    local profile="$1"
    local key_file=""
    
    case "$profile" in
        "belmo")
            key_file="$HOME/.ssh/id_github"
            ;;
        "oppy")
            key_file="$HOME/.ssh/id_oppy"
            ;;
        "personal")
            key_file="$HOME/.ssh/id_ed25519"
            ;;
    esac
    
    if [[ ! -f "$key_file" ]]; then
        echo -e "${RED}Erro: Chave SSH não encontrada: $key_file${NC}"
        exit 1
    fi
    
    if [[ ! -f "${key_file}.pub" ]]; then
        echo -e "${RED}Erro: Chave pública não encontrada: ${key_file}.pub${NC}"
        exit 1
    fi
}

clear_ssh_agent() {
    # Remove todas as chaves do ssh-agent
    ssh-add -D 2>/dev/null || true
}

add_ssh_key() {
    local profile="$1"
    local key_file=""
    
    case "$profile" in
        "belmo")
            key_file="$HOME/.ssh/id_github"
            ;;
        "oppy")
            key_file="$HOME/.ssh/id_oppy"
            ;;
        "personal")
            key_file="$HOME/.ssh/id_ed25519"
            ;;
    esac
    
    # Adiciona a chave específica ao ssh-agent
    ssh-add "$key_file" 2>/dev/null || {
        echo -e "${YELLOW}Aviso: Não foi possível adicionar a chave ao ssh-agent${NC}"
    }
}

show_current_profile() {
    if [[ ! -f "$SSH_CONFIG" ]]; then
        echo -e "${RED}Arquivo de configuração SSH não encontrado${NC}"
        return
    fi
    
    local github_key=$(grep -A 5 "Host github.com" "$SSH_CONFIG" | grep "IdentityFile" | head -1 | awk '{print $2}' | sed 's/~/$HOME/g')
    
    case "$github_key" in
        *"id_github")
            echo -e "${GREEN}Perfil atual: belmo (pessoal)${NC}"
            ;;
        *"id_oppy")
            echo -e "${GREEN}Perfil atual: oppy (empresa)${NC}"
            ;;
        *"id_ed25519")
            echo -e "${GREEN}Perfil atual: personal (pessoal)${NC}"
            ;;
        *)
            echo -e "${YELLOW}Perfil atual: desconhecido${NC}"
            ;;
    esac
}

# Função principal
main() {
    if [[ $# -eq 0 ]]; then
        show_current_profile
        echo ""
        show_help
        exit 0
    fi
    
    if [[ "$1" == "-h" || "$1" == "--help" ]]; then
        show_help
        exit 0
    fi
    
    verify_keys "$PROFILE"
    backup_config
    create_config "$PROFILE"
    clear_ssh_agent
    add_ssh_key "$PROFILE"
    
    echo -e "${GREEN}Perfil alterado para: $PROFILE${NC}"
    
    # Mostra informações da chave ativa
    case "$PROFILE" in
        "belmo")
            echo -e "Chave SSH: ${YELLOW}~/.ssh/id_github${NC}"
            ;;
        "oppy")
            echo -e "Chave SSH: ${YELLOW}~/.ssh/id_oppy${NC}"
            ;;
        "personal")
            echo -e "Chave SSH: ${YELLOW}~/.ssh/id_ed25519${NC}"
            ;;
    esac
    
    # Testa a conexão
    echo ""
    echo -e "${YELLOW}Testando conexão...${NC}"
    ssh -T git@github.com 2>&1 | head -1
}

main "$@"