mirror of
https://github.com/containers/podman.git
synced 2025-06-05 05:57:24 +08:00
a1c654de94
Refactor client code to break out building connection string from making the connection. Example: client: Connection: unix:/run/podman/io.podman Connection Type: DirectConnection . : Signed-off-by: Jhon Honce <jhonce@redhat.com>
75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
// +build remoteclient
|
|
|
|
package adapter
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/varlink/go/varlink"
|
|
)
|
|
|
|
var remoteEndpoint *Endpoint
|
|
|
|
func (r RemoteRuntime) RemoteEndpoint() (remoteEndpoint *Endpoint, err error) {
|
|
if remoteEndpoint == nil {
|
|
remoteEndpoint = &Endpoint{Unknown, ""}
|
|
} else {
|
|
return remoteEndpoint, nil
|
|
}
|
|
|
|
// I'm leaving this here for now as a document of the birdge format. It can be removed later once the bridge
|
|
// function is more flushed out.
|
|
// bridge := `ssh -T root@192.168.122.1 "/usr/bin/varlink -A '/usr/bin/podman varlink \$VARLINK_ADDRESS' bridge"`
|
|
if len(r.cmd.RemoteHost) > 0 {
|
|
// The user has provided a remote host endpoint
|
|
if len(r.cmd.RemoteUserName) < 1 {
|
|
return nil, errors.New("you must provide a username when providing a remote host name")
|
|
}
|
|
remoteEndpoint.Type = BridgeConnection
|
|
remoteEndpoint.Connection = fmt.Sprintf(
|
|
`ssh -T %s@%s /usr/bin/varlink -A \'/usr/bin/podman --log-level=%s varlink \\\$VARLINK_ADDRESS\' bridge`,
|
|
r.cmd.RemoteUserName, r.cmd.RemoteHost, r.cmd.LogLevel)
|
|
|
|
} else if bridge := os.Getenv("PODMAN_VARLINK_BRIDGE"); bridge != "" {
|
|
remoteEndpoint.Type = BridgeConnection
|
|
remoteEndpoint.Connection = bridge
|
|
} else {
|
|
address := os.Getenv("PODMAN_VARLINK_ADDRESS")
|
|
if address == "" {
|
|
address = DefaultAddress
|
|
}
|
|
remoteEndpoint.Type = DirectConnection
|
|
remoteEndpoint.Connection = address
|
|
}
|
|
return
|
|
}
|
|
|
|
// Connect provides a varlink connection
|
|
func (r RemoteRuntime) Connect() (*varlink.Connection, error) {
|
|
ep, err := r.RemoteEndpoint()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch ep.Type {
|
|
case DirectConnection:
|
|
return varlink.NewConnection(ep.Connection)
|
|
case BridgeConnection:
|
|
return varlink.NewBridge(ep.Connection)
|
|
}
|
|
return nil, errors.New(fmt.Sprintf("Unable to determine type of varlink connection: %s", ep.Connection))
|
|
}
|
|
|
|
// RefreshConnection is used to replace the current r.Conn after things like
|
|
// using an upgraded varlink connection
|
|
func (r RemoteRuntime) RefreshConnection() error {
|
|
newConn, err := r.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
r.Conn = newConn
|
|
return nil
|
|
}
|