Skip to content

Commit

Permalink
ssh-agent: autoload identities not already loaded (ohmyzsh#7174)
Browse files Browse the repository at this point in the history
With this PR the ssh-agent plugin checks the `ssh-add -l` output for the
identities added, and adds all those specified by the user that haven't been
added yet.

We also decouple the logic of starting ssh-agent from the logic of adding
identities, meaning that even if ssh-agent has been started by some other means
(like launchd) we can still ssh-add the user's identities.

Fixes ohmyzsh#3019
Fixes ohmyzsh#6979
  • Loading branch information
mcornella authored Jan 9, 2019
1 parent a299501 commit fabee55
Showing 1 changed file with 16 additions and 7 deletions.
23 changes: 16 additions & 7 deletions plugins/ssh-agent/ssh-agent.plugin.zsh
Original file line number Diff line number Diff line change
Expand Up @@ -2,20 +2,27 @@ typeset _agent_forwarding _ssh_env_cache

function _start_agent() {
local lifetime
local -a identities

# start ssh-agent and setup environment
zstyle -s :omz:plugins:ssh-agent lifetime lifetime

# start ssh-agent and setup environment
echo starting ssh-agent...
ssh-agent -s ${lifetime:+-t} ${lifetime} | sed 's/^echo/#echo/' >! $_ssh_env_cache
chmod 600 $_ssh_env_cache
. $_ssh_env_cache > /dev/null
}

# load identies
function _add_identities() {
local id line
local -a identities ids
zstyle -a :omz:plugins:ssh-agent identities identities

echo starting ssh-agent...
ssh-add $HOME/.ssh/${^identities}
# get list of loaded identities
for line in ${(f)"$(ssh-add -l)"}; do ids+=${${(z)line}[3]}; done

# add identities if not already loaded
for id in ${^identities}; do
[[ ${ids[(I)$HOME/.ssh/$id]} -le 0 ]] && ssh-add $HOME/.ssh/$id
done
}

# Get the filename to store/lookup the environment from
Expand All @@ -42,6 +49,8 @@ else
_start_agent
fi

_add_identities

# tidy up after ourselves
unset _agent_forwarding _ssh_env_cache
unfunction _start_agent
unfunction _start_agent _add_identities

0 comments on commit fabee55

Please sign in to comment.