refactor(webconsole): api: add helper methods for response struct

This commit is contained in:
Yousong Zhou
2020-12-23 13:29:11 +08:00
parent c7d099b6fb
commit 4cd1d4ba7c
+27
View File
@@ -15,6 +15,9 @@
package webconsole
import (
"encoding/base64"
"net/url"
"yunion.io/x/onecloud/pkg/apis"
)
@@ -24,3 +27,27 @@ type ServerRemoteConsoleResponse struct {
apis.Meta
}
func (resp *ServerRemoteConsoleResponse) GetConnectParams() string {
params := resp.ConnectParams
if data, err := base64.StdEncoding.DecodeString(params); err == nil {
params = string(data)
}
return params
}
func (resp *ServerRemoteConsoleResponse) GetConnectProtocol() (string, error) {
var (
params = resp.GetConnectParams()
query url.Values
)
if uri, err := url.ParseRequestURI(params); err == nil {
query = uri.Query()
} else if q, err := url.ParseQuery(params); err == nil {
query = q
} else {
return "", err
}
protocol := query.Get("protocol")
return protocol, nil
}