|
| 1 | +package myaddr |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "encoding/json" |
| 6 | + "fmt" |
| 7 | + "net/http" |
| 8 | + "net/netip" |
| 9 | + "net/url" |
| 10 | + "strings" |
| 11 | + |
| 12 | + "github.com/qdm12/ddns-updater/internal/models" |
| 13 | + "github.com/qdm12/ddns-updater/internal/provider/constants" |
| 14 | + "github.com/qdm12/ddns-updater/internal/provider/errors" |
| 15 | + "github.com/qdm12/ddns-updater/internal/provider/headers" |
| 16 | + "github.com/qdm12/ddns-updater/internal/provider/utils" |
| 17 | + "github.com/qdm12/ddns-updater/pkg/publicip/ipversion" |
| 18 | +) |
| 19 | + |
| 20 | +type Provider struct { |
| 21 | + domain string |
| 22 | + owner string |
| 23 | + ipVersion ipversion.IPVersion |
| 24 | + ipv6Suffix netip.Prefix |
| 25 | + key string |
| 26 | +} |
| 27 | + |
| 28 | +func New(data json.RawMessage, domain, owner string, |
| 29 | + ipVersion ipversion.IPVersion, ipv6Suffix netip.Prefix, |
| 30 | +) (*Provider, error) { |
| 31 | + var providerSpecificSettings struct { |
| 32 | + Key string `json:"key"` |
| 33 | + } |
| 34 | + err := json.Unmarshal(data, &providerSpecificSettings) |
| 35 | + if err != nil { |
| 36 | + return nil, fmt.Errorf("json decoding provider specific settings: %w", err) |
| 37 | + } |
| 38 | + err = validateSettings(domain, providerSpecificSettings.Key) |
| 39 | + if err != nil { |
| 40 | + return nil, fmt.Errorf("validating provider specific settings: %w", err) |
| 41 | + } |
| 42 | + return &Provider{ |
| 43 | + domain: domain, |
| 44 | + owner: owner, |
| 45 | + ipVersion: ipVersion, |
| 46 | + ipv6Suffix: ipv6Suffix, |
| 47 | + key: providerSpecificSettings.Key, |
| 48 | + }, nil |
| 49 | +} |
| 50 | + |
| 51 | +func validateSettings(domain, key string) (err error) { |
| 52 | + err = utils.CheckDomain(domain) |
| 53 | + if err != nil { |
| 54 | + return fmt.Errorf("%w: %w", errors.ErrDomainNotValid, err) |
| 55 | + } |
| 56 | + if key == "" { |
| 57 | + return fmt.Errorf("%w", errors.ErrKeyNotSet) |
| 58 | + } |
| 59 | + return nil |
| 60 | +} |
| 61 | + |
| 62 | +func (p *Provider) String() string { |
| 63 | + return utils.ToString(p.Domain(), p.Owner(), constants.Myaddr, p.IPVersion()) |
| 64 | +} |
| 65 | + |
| 66 | +func (p *Provider) Domain() string { |
| 67 | + return p.domain |
| 68 | +} |
| 69 | + |
| 70 | +func (p *Provider) Owner() string { |
| 71 | + return p.owner |
| 72 | +} |
| 73 | + |
| 74 | +func (p *Provider) BuildDomainName() string { |
| 75 | + return utils.BuildDomainName(p.owner, p.domain) |
| 76 | +} |
| 77 | + |
| 78 | +func (p *Provider) HTML() models.HTMLRow { |
| 79 | + return models.HTMLRow{ |
| 80 | + Domain: fmt.Sprintf("<a href=\"http://%s\">%s</a>", p.BuildDomainName(), p.BuildDomainName()), |
| 81 | + Owner: p.Owner(), |
| 82 | + Provider: "<a href=\"https://myaddr.tools/\">myaddr</a>", |
| 83 | + IPVersion: p.IPVersion().String(), |
| 84 | + } |
| 85 | +} |
| 86 | + |
| 87 | +func (p *Provider) Proxied() bool { |
| 88 | + return false |
| 89 | +} |
| 90 | + |
| 91 | +func (p *Provider) IPVersion() ipversion.IPVersion { |
| 92 | + return p.ipVersion |
| 93 | +} |
| 94 | + |
| 95 | +func (p *Provider) IPv6Suffix() netip.Prefix { |
| 96 | + return p.ipv6Suffix |
| 97 | +} |
| 98 | + |
| 99 | +func (p *Provider) Update(ctx context.Context, client *http.Client, ip netip.Addr) (netip.Addr, error) { |
| 100 | + u := &url.URL{ |
| 101 | + Scheme: "https", |
| 102 | + Host: "myaddr.tools", |
| 103 | + Path: "/update", |
| 104 | + } |
| 105 | + v := url.Values{} |
| 106 | + v.Set("key", p.key) |
| 107 | + v.Set("ip", ip.String()) |
| 108 | + buffer := strings.NewReader(v.Encode()) |
| 109 | + request, err := http.NewRequestWithContext(ctx, http.MethodPost, u.String(), buffer) |
| 110 | + if err != nil { |
| 111 | + return netip.Addr{}, fmt.Errorf("creating http request: %w", err) |
| 112 | + } |
| 113 | + headers.SetContentType(request, "application/x-www-form-urlencoded") |
| 114 | + headers.SetUserAgent(request) |
| 115 | + response, err := client.Do(request) |
| 116 | + if err != nil { |
| 117 | + return netip.Addr{}, err |
| 118 | + } |
| 119 | + defer response.Body.Close() |
| 120 | + switch response.StatusCode { |
| 121 | + case http.StatusOK: |
| 122 | + return ip, nil |
| 123 | + case http.StatusBadRequest: |
| 124 | + return netip.Addr{}, fmt.Errorf("%w: %s", errors.ErrBadRequest, utils.BodyToSingleLine(response.Body)) |
| 125 | + case http.StatusNotFound: |
| 126 | + return netip.Addr{}, fmt.Errorf("%w: %s", errors.ErrKeyNotValid, utils.BodyToSingleLine(response.Body)) |
| 127 | + default: |
| 128 | + return netip.Addr{}, fmt.Errorf("%w: %d: %s", |
| 129 | + errors.ErrHTTPStatusNotValid, response.StatusCode, utils.BodyToSingleLine(response.Body)) |
| 130 | + } |
| 131 | +} |
0 commit comments