Расширение опций и рефакторинг:

* Больше опций настройки клиента
* Добавлен типа DateOnly, т.к. приходится повторять везде
* Немного переработан uncompress тела ответа
* пр. мелкий рефакторинг

Author: stvoidit

apiClient.go

@@ -1,36 +1,39 @@
 package megaplan
 
 import (
-	"bytes"
-	"compress/gzip"
-	"context"
 	"crypto/tls"
-	"errors"
-	"io"
 	"net/http"
+	"net/http/cookiejar"
+	"net/url"
 	"runtime"
 	"strconv"
+	"strings"
 	"time"
 )
 
 // DefaultClient - клиент по умаолчанию для API.
 var (
-	cpus          = runtime.NumCPU()
+	cpus             = runtime.NumCPU()
+	DefaultTransport = &http.Transport{
+		Proxy:               http.ProxyFromEnvironment,
+		MaxIdleConns:        cpus,
+		MaxConnsPerHost:     cpus,
+		MaxIdleConnsPerHost: cpus,
+	}
 	DefaultClient = &http.Client{
-		Transport: &http.Transport{
-			Proxy:               http.ProxyFromEnvironment,
-			MaxIdleConns:        cpus,
-			MaxConnsPerHost:     cpus,
-			MaxIdleConnsPerHost: cpus,
-		},
-		Timeout: time.Minute,
+		Transport: DefaultTransport,
+		Timeout:   time.Minute,
 	}
 	// DefaultHeaders - заголовок по умолчанию - версия go. Используется при инициализации клиента в NewClient.
 	DefaultHeaders = http.Header{"User-Agent": {runtime.Version()}}
 )
 
 // NewClient - обертка над http.Client для удобной работы с API v3
 func NewClient(domain, token string, opts ...ClientOption) (c *ClientV3) {
+	if strings.HasPrefix("http", domain) {
+		_URL, _ := url.Parse(domain)
+		domain = _URL.Host
+	}
 	c = &ClientV3{
 		client:         DefaultClient,
 		domain:         domain,
@@ -48,91 +51,7 @@ type ClientV3 struct {
 	client         *http.Client
 }
 
-// Do - http.Do + установка обязательных заголовков + декомпрессия ответа, если ответ сжат
-func (c *ClientV3) Do(req *http.Request) (*http.Response, error) {
-	const ct = "Content-Type"
-	for h := range c.defaultHeaders {
-		req.Header.Set(h, c.defaultHeaders.Get(h))
-	}
-	if _, ok := req.Header[ct]; !ok {
-		req.Header.Set(ct, "application/json")
-	}
-	res, err := c.client.Do(req)
-	if err != nil {
-		return nil, err
-	}
-	if err := unzipResponse(res); err != nil {
-		return nil, err
-	}
-	return res, nil
-}
-
-// DoRequestAPI - т.к. в v3 параметры запроса для GET (json маршализируется и будет иметь вид: "*?{params}=")
-func (c ClientV3) DoRequestAPI(method string, endpoint string, search QueryParams, body io.Reader) (*http.Response, error) {
-	var args string // параметры строки запроса
-	if search != nil {
-		args = search.QueryEscape()
-	}
-	request, err := http.NewRequest(method, c.domain, body)
-	if err != nil {
-		return nil, err
-	}
-	request.URL.Path = endpoint
-	request.URL.RawQuery = args
-	return c.Do(request)
-}
-
-// DoRequestAPI - т.к. в v3 параметры запроса для GET (json маршализируется и будет иметь вид: "*?{params}=")
-func (c ClientV3) DoCtxRequestAPI(ctx context.Context, method string, endpoint string, search QueryParams, body io.Reader) (*http.Response, error) {
-	var args string // параметры строки запроса
-	if search != nil {
-		args = search.QueryEscape()
-	}
-	request, err := http.NewRequestWithContext(ctx, method, c.domain, body)
-	if err != nil {
-		return nil, err
-	}
-	request.URL.Path = endpoint
-	request.URL.RawQuery = args
-	return c.Do(request)
-}
-
-// ErrUnknownCompressionMethod - неизвестное значение в заголовке "Content-Encoding"
-// не является фатальной ошибкой, должна возвращаться вместе с http.Response.Body,
-// чтобы пользователь мог реализовать свой метод обработки сжатого сообщения
-var ErrUnknownCompressionMethod = errors.New("unknown compression method")
-
-// unzipResponse - распаковка сжатого ответа
-func unzipResponse(response *http.Response) (err error) {
-	if response.Uncompressed {
-		return nil
-	}
-	switch response.Header.Get("Content-Encoding") {
-	case "":
-		return nil
-	case "gzip":
-		gz, err := gzip.NewReader(response.Body)
-		if err != nil {
-			return err
-		}
-		b, err := io.ReadAll(gz)
-		if err != nil {
-			return err
-		}
-		if err := response.Body.Close(); err != nil {
-			return err
-		}
-		if err := gz.Close(); err != nil {
-			return err
-		}
-		response.Body = io.NopCloser(bytes.NewReader(b))
-		response.Header.Del("Content-Encoding")
-		response.Uncompressed = true
-		return nil
-	default:
-		return ErrUnknownCompressionMethod
-	}
-}
+func (c *ClientV3) Close() { c.client.CloseIdleConnections() }
 
 // SetOptions - применить опции
 func (c *ClientV3) SetOptions(opts ...ClientOption) {
@@ -149,13 +68,13 @@ type ClientOption func(*ClientV3)
 
 // OptionInsecureSkipVerify - переключение флага bool в http.Client.Transport.TLSClientConfig.InsecureSkipVerify - отключать или нет проверку сертификтов
 // Если домен использует самоподписанные сертифика, то удобно включать на время отладки и разработки
-func OptionInsecureSkipVerify(b bool) ClientOption {
+func OptionInsecureSkipVerify(yes bool) ClientOption {
 	return func(c *ClientV3) {
 		if c.client.Transport != nil {
 			if (c.client.Transport.(*http.Transport)).TLSClientConfig == nil {
-				(c.client.Transport.(*http.Transport)).TLSClientConfig = &tls.Config{InsecureSkipVerify: b}
+				(c.client.Transport.(*http.Transport)).TLSClientConfig = &tls.Config{InsecureSkipVerify: yes}
 			} else {
-				(c.client.Transport.(*http.Transport)).TLSClientConfig.InsecureSkipVerify = b
+				(c.client.Transport.(*http.Transport)).TLSClientConfig.InsecureSkipVerify = yes
 			}
 		}
 	}
@@ -168,10 +87,10 @@ func OptionsSetHTTPTransport(tr http.RoundTripper) ClientOption {
 
 // OptionEnableAcceptEncodingGzip - доабвить заголов Accept-Encoding=gzip к запросу
 // т.е. объекм трафика на хуках может быть большим, то удобно запрашивать сжатый ответ
-func OptionEnableAcceptEncodingGzip(b bool) ClientOption {
+func OptionEnableAcceptEncodingGzip(yes bool) ClientOption {
 	const header = "Accept-Encoding"
 	return func(c *ClientV3) {
-		if b {
+		if yes {
 			c.defaultHeaders.Set(header, "gzip")
 		} else {
 			c.defaultHeaders.Del(header)
@@ -200,3 +119,34 @@ func OptionSetXUserID(userID int) ClientOption {
 		}
 	}
 }
+
+func OptionDisableCookie(yes bool) ClientOption {
+	return func(c *ClientV3) {
+		if yes {
+			c.client.Jar = nil
+		} else {
+			jar, _ := cookiejar.New(nil)
+			c.client.Jar = jar
+		}
+	}
+}
+
+func OptionForceAttemptHTTP2(yes bool) ClientOption {
+	return func(c *ClientV3) {
+		if c.client.Transport != nil {
+			(c.client.Transport.(*http.Transport)).ForceAttemptHTTP2 = yes
+		} else {
+			c.client.Transport = &http.Transport{ForceAttemptHTTP2: yes}
+		}
+	}
+}
+
+func OptionDisablekeepAlive(yes bool) ClientOption {
+	return func(c *ClientV3) {
+		if c.client.Transport != nil {
+			(c.client.Transport.(*http.Transport)).DisableKeepAlives = yes
+		} else {
+			c.client.Transport = &http.Transport{DisableKeepAlives: yes}
+		}
+	}
+}

apiQueryParams.go

@@ -5,8 +5,12 @@ import (
 	"encoding/json"
 	"io"
 	"net/url"
+	"time"
 )
 
+// ISO8601 - формат даты для api
+const ISO8601 = `2006-01-02T15:04:05-07:00`
+
 // QueryParams - параметры запроса
 type QueryParams map[string]any
 
@@ -34,3 +38,22 @@ func (qp QueryParams) PrettyPrintJSON(w io.Writer) error {
 	enc.SetIndent("", "  ")
 	return enc.Encode(qp)
 }
+
+type DateOnly struct {
+	ContentType string `json:"contentType"`
+	Day         int    `json:"day"`
+	Month       int    `json:"month"`
+	Year        int    `json:"year"`
+}
+
+func (d DateOnly) ToTime() time.Time {
+	return time.Date(d.Year, time.Month(d.Month)+1, d.Day, 0, 0, 0, 0, time.UTC)
+}
+func NewDateOnly(t time.Time) DateOnly {
+	return DateOnly{
+		ContentType: typeDateOnly,
+		Day:         t.Day(),
+		Month:       int(t.Month()) - 1,
+		Year:        t.Year(),
+	}
+}

apiRequest.go

@@ -2,14 +2,24 @@ package megaplan
 
 import (
 	"bytes"
+	"context"
 	"io"
 	"mime/multipart"
 	"net/http"
+	"net/url"
 	"time"
 )
 
-// ISO8601 - формат даты для api
-const ISO8601 = `2006-01-02T15:04:05-07:00`
+const (
+	headerContentType    = "Content-Type"
+	valueApplicationJSON = "application/json"
+)
+
+const (
+	typeDateOnly     = "DateOnly"
+	typeDateTime     = "DateTime"
+	typeDateInterval = "DateInterval"
+)
 
 // BuildQueryParams - сборка объекта для запроса
 func BuildQueryParams(opts ...QueryBuildingFunc) (qp QueryParams) {
@@ -28,23 +38,22 @@ type QueryBuildingFunc func(QueryParams)
 func CreateEnity(contentType string, value any) (qp QueryParams) {
 	qp = make(QueryParams, 2)
 	qp["contentType"] = contentType
-
 	switch contentType {
-	case "DateOnly":
+	case typeDateOnly:
 		t, isTime := value.(time.Time)
 		if !isTime {
 			return nil
 		}
 		qp["year"] = t.Year()
 		qp["month"] = t.Month() - 1
 		qp["day"] = t.Day()
-	case "DateTime":
+	case typeDateTime:
 		t, isTime := value.(time.Time)
 		if !isTime {
 			return nil
 		}
 		qp["value"] = t.Format(ISO8601)
-	case "DateInterval":
+	case typeDateInterval:
 		// если передается не время, то должно указываться кол-во секунд (актуальная документация мегаплана пишет что миллисекунды - это ошибка)
 		switch v := value.(type) {
 		case time.Time:
@@ -106,6 +115,72 @@ func (c *ClientV3) UploadFile(filename string, fileReader io.Reader) (*http.Resp
 		return nil, err
 	}
 	request.URL.Path = "/api/file"
-	request.Header.Set("Content-Type", mw.FormDataContentType())
+	request.Header.Set(headerContentType, mw.FormDataContentType())
+	return c.Do(request)
+}
+
+// Do - http.Do + установка обязательных заголовков + декомпрессия ответа, если ответ сжат
+func (c *ClientV3) Do(req *http.Request) (*http.Response, error) {
+	for h := range c.defaultHeaders.Clone() {
+		req.Header.Set(h, c.defaultHeaders.Get(h))
+	}
+	if _, ok := req.Header[headerContentType]; !ok {
+		req.Header.Set(headerContentType, valueApplicationJSON)
+	}
+	res, err := c.client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	// slog.Debug("response",
+	// 	slog.String("proto", res.Proto),
+	// 	slog.Int("status", res.StatusCode),
+	// 	slog.String("connection", res.Header.Get("connection")),
+	// )
+	if err := unzipResponse(res); err != nil {
+		return nil, err
+	}
+	return res, nil
+}
+
+func (c ClientV3) makeRequestURL(endpoint string, search QueryParams) string {
+	var args string // параметры строки запроса
+	if search != nil {
+		args = search.QueryEscape()
+	}
+	return (&url.URL{
+		Scheme:   "https",
+		Host:     c.domain,
+		Path:     endpoint,
+		RawQuery: args,
+	}).String()
+}
+
+// DoRequestAPI - т.к. в v3 параметры запроса для GET (json маршализируется и будет иметь вид: "*?{params}=")
+func (c ClientV3) DoRequestAPI(method, endpoint string, search QueryParams, body io.Reader) (*http.Response, error) {
+	request, err := http.NewRequest(
+		method,
+		c.makeRequestURL(endpoint, search),
+		body)
+	if err != nil {
+		return nil, err
+	}
+	// slog.Debug("DoRequestAPI.NewRequest",
+	// 	slog.String("method", method),
+	// 	slog.String("endpoint", request.URL.String()))
+	return c.Do(request)
+}
+
+// DoRequestAPI - т.к. в v3 параметры запроса для GET (json маршализируется и будет иметь вид: "*?{params}=")
+func (c ClientV3) DoCtxRequestAPI(ctx context.Context, method, endpoint string, search QueryParams, body io.Reader) (*http.Response, error) {
+	request, err := http.NewRequestWithContext(ctx,
+		method,
+		c.makeRequestURL(endpoint, search),
+		body)
+	if err != nil {
+		return nil, err
+	}
+	// slog.Debug("DoCtxRequestAPI.NewRequestWithContext",
+	// 	slog.String("method", method),
+	// 	slog.String("endpoint", request.URL.String()))
 	return c.Do(request)
 }