#!/bin/bash
if [[ $TERM = screen* || $TERM = tmux* ]]; then
    TERM=screen exec vagrant ssh "$@"
elif [[ $TERM = xterm-kitty ]]; then
    TERM=xterm-256color exec vagrant ssh "$@"
else
    exec vagrant ssh "$@"
fi