mirror of
https://github.com/josmo/drone-rancher.git
synced 2026-06-16 14:50:35 +08:00
Joachim Hill-Grannec
30 lines
625 B
Go
30 lines
625 B
Go
package hostaccess
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/rancher/go-rancher/v2"
|
|
)
|
|
|
|
type RancherWebsocketClient client.RancherClient
|
|
|
|
func (c *RancherWebsocketClient) GetHostAccess(resource client.Resource, action string, input interface{}) (*websocket.Conn, error) {
|
|
var resp client.HostAccess
|
|
url := resource.Actions[action]
|
|
if url == "" {
|
|
return nil, fmt.Errorf("Failed to find action: %s", action)
|
|
}
|
|
|
|
err := c.Post(url, input, &resp)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
url = fmt.Sprintf("%s?token=%s", resp.Url, resp.Token)
|
|
|
|
conn, _, err := c.Websocket(url, nil)
|
|
|
|
return conn, err
|
|
}
|