(ctx context.Context, workspaceIP string)
| 372 | } |
| 373 | |
| 374 | func (s *Server) GetWorkspaceSSHKey(ctx context.Context, workspaceIP string) (ssh.Signer, error) { |
| 375 | supervisorConn, err := grpc.Dial(workspaceIP+":22999", grpc.WithTransportCredentials(insecure.NewCredentials())) |
| 376 | if err != nil { |
| 377 | return nil, xerrors.Errorf("failed connecting to supervisor: %w", err) |
| 378 | } |
| 379 | defer supervisorConn.Close() |
| 380 | keyInfo, err := supervisor.NewControlServiceClient(supervisorConn).CreateSSHKeyPair(ctx, &supervisor.CreateSSHKeyPairRequest{}) |
| 381 | if err != nil { |
| 382 | return nil, xerrors.Errorf("failed getting ssh key pair info from supervisor: %w", err) |
| 383 | } |
| 384 | key, err := ssh.ParsePrivateKey([]byte(keyInfo.PrivateKey)) |
| 385 | if err != nil { |
| 386 | return nil, xerrors.Errorf("failed parse private key: %w", err) |
| 387 | } |
| 388 | return key, nil |
| 389 | } |
| 390 | |
| 391 | func (s *Server) Serve(l net.Listener) error { |
| 392 | for { |
no test coverage detected