118 lines
2.2 KiB
Go
118 lines
2.2 KiB
Go
package sinks
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"dynatron.me/x/stillbox/internal/forms"
|
|
"dynatron.me/x/stillbox/internal/version"
|
|
"dynatron.me/x/stillbox/pkg/calls"
|
|
"dynatron.me/x/stillbox/pkg/config"
|
|
)
|
|
|
|
type RelayManager struct {
|
|
xp *http.Transport
|
|
client *http.Client
|
|
|
|
relays []*Relay
|
|
}
|
|
|
|
type Relay struct {
|
|
config.Relay
|
|
mgr *RelayManager
|
|
Name string
|
|
|
|
url *url.URL
|
|
}
|
|
|
|
func NewRelayManager(s Sinks, cfgs []config.Relay) (*RelayManager, error) {
|
|
xp := http.DefaultTransport.(*http.Transport).Clone()
|
|
xp.MaxIdleConnsPerHost = 10
|
|
|
|
client := &http.Client{
|
|
Transport: xp,
|
|
}
|
|
|
|
rm := &RelayManager{
|
|
xp: xp,
|
|
client: client,
|
|
relays: make([]*Relay, 0, len(cfgs)),
|
|
}
|
|
|
|
for i, cfg := range cfgs {
|
|
rs, err := rm.newRelay(i, cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rm.relays = append(rm.relays, rs)
|
|
|
|
s.Register(rs.Name, rs, cfg.Required)
|
|
}
|
|
|
|
return rm, nil
|
|
}
|
|
|
|
func (rs *RelayManager) newRelay(idx int, cfg config.Relay) (*Relay, error) {
|
|
u, err := url.Parse(cfg.URL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if u.Path != "" && u.Path != "/" {
|
|
return nil, fmt.Errorf("relay path in %s must be instance root", cfg.URL)
|
|
}
|
|
|
|
u = u.JoinPath("/api/call-upload")
|
|
|
|
return &Relay{
|
|
Name: fmt.Sprintf("relay%d:%s", idx, u.Host),
|
|
Relay: cfg,
|
|
url: u,
|
|
mgr: rs,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Relay) Call(ctx context.Context, call *calls.Call) error {
|
|
var buf bytes.Buffer
|
|
body := multipart.NewWriter(&buf)
|
|
|
|
err := forms.Marshal(call, body)
|
|
if err != nil {
|
|
return fmt.Errorf("relay form parse: %w", err)
|
|
}
|
|
|
|
err = body.WriteField("key", s.APIKey)
|
|
if err != nil {
|
|
return fmt.Errorf("relay set API key: %w", err)
|
|
}
|
|
|
|
body.Close()
|
|
|
|
r, err := http.NewRequestWithContext(ctx, http.MethodPost, s.url.String(), &buf)
|
|
if err != nil {
|
|
return fmt.Errorf("relay newrequest: %w", err)
|
|
}
|
|
|
|
r.Header.Set("Content-Type", body.FormDataContentType())
|
|
r.Header.Set("User-Agent", version.HttpString("call-relay"))
|
|
|
|
resp, err := s.mgr.client.Do(r)
|
|
if err != nil {
|
|
return fmt.Errorf("relay %s: %w", s.Name, err)
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("relay %s: received HTTP %d", s.Name, resp.StatusCode)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Relay) SinkType() string {
|
|
return "relay"
|
|
}
|