dns-updater/ip/resolver/http/service.go

package http

import (
	"context"
	"io"
	"net"
	"net/http"

	httputils "dnsupdater/http"
	"dnsupdater/ip/internal"
	"dnsupdater/ip/resolver/decoder"
)

type Decoder func(io.Reader) (string, error)

type Service struct {
	ServiceName string
	Url         string
	Headers     http.Header
	Decoder     Decoder
}

func (s Service) Name() string {
	return s.ServiceName
}

func (s Service) Lookup(ctx context.Context) (net.IP, error) {
	resp, err := httputils.Get(ctx, s.Url, s.Headers)
	if err != nil {
		return nil, err
	}

	if s.Decoder == nil {
		s.Decoder = decoder.Text
	}

	body, err := s.Decoder(resp.Body)
	if err != nil && err != io.EOF {
		return nil, err
	}
	return internal.ParseIP(string(body))
}