mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Roey BermanandGitHub
4170bc8e64
## What changed? Moved the code from github.com/temporalio/version-info-model into this repository and removed insecure HTTPS setting. ## Why? No reason to maintain a whole separate repo with the server being the only dependent.
97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package versioninfo
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
type Caller struct {
|
|
Scheme string
|
|
Host string
|
|
}
|
|
|
|
func NewCaller() Caller {
|
|
return Caller{"https", "version-info.temporal.io"}
|
|
}
|
|
|
|
func (c Caller) Call(r *VersionCheckRequest) (*VersionCheckResponse, error) {
|
|
err := validateRequest(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
u := c.getUrl(r)
|
|
tr := &http.Transport{
|
|
DisableKeepAlives: true,
|
|
MaxIdleConnsPerHost: -1,
|
|
}
|
|
if c.Scheme == "https" {
|
|
tr.TLSClientConfig = &tls.Config{}
|
|
}
|
|
client := &http.Client{Transport: tr}
|
|
reqBody, err := json.Marshal(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req, err := http.NewRequest("POST", u.String(), bytes.NewReader(reqBody))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Set("Content-Type", "application/json")
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode != 200 {
|
|
return nil, errors.New(fmt.Sprintf("bad response code %v", resp.StatusCode))
|
|
}
|
|
defer resp.Body.Close()
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
versionCheckResponse := &VersionCheckResponse{}
|
|
if err := json.Unmarshal(body, versionCheckResponse); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := validateResponse(versionCheckResponse); err != nil {
|
|
return nil, err
|
|
}
|
|
return versionCheckResponse, nil
|
|
}
|
|
|
|
func validateResponse(r *VersionCheckResponse) error {
|
|
if len(r.Products) == 0 {
|
|
return errors.New("invalid response: missing product list")
|
|
}
|
|
firstProduct := r.Products[0]
|
|
if firstProduct.Product == "" || firstProduct.Current.Version == "" || firstProduct.Recommended.Version == "" {
|
|
return errors.New("invalid response: missing product name, current or recommended version")
|
|
}
|
|
return nil
|
|
}
|
|
func validateRequest(r *VersionCheckRequest) error {
|
|
if r.Product == "" || r.Version == "" || r.ClusterID == "" || r.DB == "" || r.OS == "" || r.Arch == "" || r.Timestamp == 0 {
|
|
return errors.New("invalid request: missing required fields")
|
|
}
|
|
for _, info := range r.SDKInfo {
|
|
if info.Name == "" || info.Version == "" {
|
|
return errors.New("invalid request: missing required fields")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c Caller) getUrl(r *VersionCheckRequest) *url.URL {
|
|
var u url.URL
|
|
u.Scheme = c.Scheme
|
|
u.Host = c.Host
|
|
u.Path = fmt.Sprintf("check")
|
|
return &u
|
|
}
|