125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
// Copyright (c) 2023 Joshua Rich <joshua.rich@gmail.com>
|
|
//
|
|
// This software is released under the MIT License.
|
|
// https://opensource.org/licenses/MIT
|
|
|
|
package hass
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"time"
|
|
|
|
"fyne.io/fyne/v2/data/binding"
|
|
"github.com/carlmjohnson/requests"
|
|
"github.com/go-playground/validator/v10"
|
|
)
|
|
|
|
type RegistrationDetails struct {
|
|
Server, Token binding.String
|
|
Device DeviceInfo
|
|
}
|
|
|
|
func (r *RegistrationDetails) Validate() bool {
|
|
validate := validator.New()
|
|
check := func(value string, validation string) bool {
|
|
if err := validate.Var(value, validation); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
if server, _ := r.Server.Get(); !check(server, "required,http_url") {
|
|
return false
|
|
}
|
|
if token, _ := r.Token.Get(); !check(token, "required") {
|
|
return false
|
|
}
|
|
if r.Device == nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
type RegistrationResponse struct {
|
|
CloudhookURL string `json:"cloudhook_url"`
|
|
RemoteUIURL string `json:"remote_ui_url"`
|
|
Secret string `json:"secret"`
|
|
WebhookID string `json:"webhook_id"`
|
|
}
|
|
|
|
func (rr *RegistrationResponse) GenerateAPIURL(host string) string {
|
|
switch {
|
|
case rr.CloudhookURL != "":
|
|
return rr.CloudhookURL
|
|
case rr.RemoteUIURL != "" && rr.WebhookID != "":
|
|
return rr.RemoteUIURL + webHookPath + rr.WebhookID
|
|
case rr.WebhookID != "":
|
|
u, _ := url.Parse(host)
|
|
u = u.JoinPath(webHookPath, rr.WebhookID)
|
|
return u.String()
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func (rr *RegistrationResponse) GenerateWebsocketURL(host string) string {
|
|
// TODO: look into websocket http upgrade method
|
|
u, _ := url.Parse(host)
|
|
switch u.Scheme {
|
|
case "https":
|
|
u.Scheme = "wss"
|
|
case "http":
|
|
u.Scheme = "ws"
|
|
case "ws":
|
|
// nothing to do
|
|
case "wss":
|
|
// nothing to do
|
|
default:
|
|
u.Scheme = "ws"
|
|
}
|
|
u = u.JoinPath(websocketPath)
|
|
return u.String()
|
|
}
|
|
|
|
type RegistrationRequest struct {
|
|
AppData interface{} `json:"app_data,omitempty"`
|
|
DeviceID string `json:"device_id"`
|
|
AppID string `json:"app_id"`
|
|
AppName string `json:"app_name"`
|
|
AppVersion string `json:"app_version"`
|
|
DeviceName string `json:"device_name"`
|
|
Manufacturer string `json:"manufacturer"`
|
|
Model string `json:"model"`
|
|
OsName string `json:"os_name"`
|
|
OsVersion string `json:"os_version"`
|
|
SupportsEncryption bool `json:"supports_encryption"`
|
|
}
|
|
|
|
func RegisterWithHass(registration *RegistrationDetails) (*RegistrationResponse, error) {
|
|
request, err := registration.Device.MarshalJSON()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
token, _ := registration.Token.Get()
|
|
host, _ := registration.Server.Get()
|
|
url, err := url.Parse(host)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
url = url.JoinPath("/api/mobile_app/registrations")
|
|
|
|
var response *RegistrationResponse
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
|
|
defer cancel()
|
|
err = requests.
|
|
URL(url.String()).
|
|
Header("Authorization", "Bearer "+token).
|
|
BodyBytes(request).
|
|
ToJSON(&response).
|
|
Fetch(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return response, nil
|
|
}
|