84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package mgmt
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
|
|
telebit "git.coolaj86.com/coolaj86/go-telebitd/mplexer"
|
|
"git.coolaj86.com/coolaj86/go-telebitd/mplexer/mgmt/authstore"
|
|
)
|
|
|
|
type Grants struct {
|
|
Domains []string `json:"domains"`
|
|
}
|
|
|
|
type SuccessResponse struct {
|
|
Success bool `json:"success"`
|
|
}
|
|
|
|
func Ping(authURL, token string) error {
|
|
msg, err := telebit.Request("POST", authURL+"/ping", token, nil)
|
|
if nil != err {
|
|
return err
|
|
}
|
|
if nil == msg {
|
|
return fmt.Errorf("invalid response")
|
|
}
|
|
resp := SuccessResponse{}
|
|
err = json.NewDecoder(msg).Decode(&resp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if true != resp.Success {
|
|
return fmt.Errorf("expected successful response")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Inspect(authURL, token string) (*Grants, error) {
|
|
msg, err := telebit.Request("GET", authURL+"/inspect", token, nil)
|
|
if nil != err {
|
|
return nil, err
|
|
}
|
|
if nil == msg {
|
|
return nil, fmt.Errorf("invalid response")
|
|
}
|
|
|
|
grants := &Grants{}
|
|
err = json.NewDecoder(msg).Decode(grants)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return grants, nil
|
|
}
|
|
|
|
func Register(authURL, secret, ppid string) (kid string, err error) {
|
|
pub := authstore.ToPublicKeyString(ppid)
|
|
jsonb := bytes.NewBuffer([]byte(
|
|
fmt.Sprintf(`{ "machine_ppid": "%s", "public_key": "%s" }`, ppid, pub),
|
|
))
|
|
msg, err := telebit.Request("POST", authURL+"/register-device/"+secret, "", jsonb)
|
|
if nil != err {
|
|
return "", err
|
|
}
|
|
if nil == msg {
|
|
return "", fmt.Errorf("invalid response")
|
|
}
|
|
|
|
auth := &authstore.Authorization{}
|
|
err = json.NewDecoder(msg).Decode(auth)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
msgBytes, _ := ioutil.ReadAll(msg)
|
|
if "" == auth.PublicKey {
|
|
return "", fmt.Errorf("unexpected server response: no public key: %s", string(msgBytes))
|
|
}
|
|
if pub != auth.PublicKey {
|
|
return "", fmt.Errorf("server disagrees about public key id: %s vs %s", kid, auth.PublicKey)
|
|
}
|
|
return auth.PublicKey, nil
|
|
}
|