refactor(ssh): use context control on keep alive between agent and client

Author: henrybarreto

ssh/server/channels/session.go

@@ -179,7 +179,7 @@ func DefaultSessionHandler() gliderssh.ChannelHandler {
 							return
 						}
 
-						if err := sess.KeepAlive(); err != nil {
+						if err := sess.KeepAlive(ctx); err != nil {
 							logger.WithError(err).Error("failed to send the API request to inform that the session is open")
 
 							return

ssh/session/session.go

@@ -691,8 +691,8 @@ func Event[D any](sess *Session, t string, data []byte, seat int) {
 	})
 }
 
-func (s *Session) KeepAlive() error {
-	if err := s.api.KeepAliveSession(context.TODO(), s.UID); err != nil {
+func (s *Session) KeepAlive(ctx context.Context) error {
+	if err := s.api.KeepAliveSession(ctx, s.UID); err != nil {
 		log.WithError(err).
 			WithFields(log.Fields{"session": s.UID, "sshid": s.SSHID}).
 			Error("Error when trying to keep alive the session")