// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

package rpc

import (
	"crypto/tls"

	"storj.io/common/identity"
	"storj.io/drpc"
)

// Conn is a wrapper around a drpc client connection.
type Conn struct {
	state tls.ConnectionState
	drpc.Conn
}

// Close closes the connection.
func (c *Conn) Close() error { return c.Conn.Close() }

// ConnectionState returns the tls connection state.
func (c *Conn) ConnectionState() tls.ConnectionState { return c.state }

// PeerIdentity returns the peer identity on the other end of the connection.
func (c *Conn) PeerIdentity() (*identity.PeerIdentity, error) {
	return identity.PeerIdentityFromChain(c.state.PeerCertificates)
}

// Raw returns the underlying connection.
func (c *Conn) Raw() drpc.Conn {
	return c.Conn
}