tstor/pkg/qbittorrent/client_impl.go

package qbittorrent

import (
	"context"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strings"
	"time"
)

type responseResult struct {
	code    int
	body    []byte
	cookies []*http.Cookie
}

type requestData struct {
	method      string
	url         string
	contentType string
	body        io.Reader
}

var _ Client = (*client)(nil)

type client struct {
	config *Config
	client *http.Client
}

func (c *client) Authentication() Authentication {
	return c
}

func (c *client) Application() Application {
	return c
}

func (c *client) Log() Log {
	return c
}

func (c *client) Sync() Sync {
	return c
}

func (c *client) Transfer() Transfer {
	return c
}

func (c *client) Torrent() Torrent {
	return c
}

func (c *client) Search() Search {
	return c
}

func (c *client) RSS() RSS {
	return c
}

// doRequest send request
func (c *client) doRequest(ctx context.Context, data *requestData) (*responseResult, error) {
	if data.method == "" {
		data.method = "GET"
	}
	if data.contentType == "" {
		data.contentType = ContentTypeFormUrlEncoded
	}
	request, err := http.NewRequestWithContext(ctx, data.method, data.url, data.body)
	if err != nil {
		return nil, err
	}

	request.Header.Set("Content-Type", data.contentType)
	for key, value := range c.config.CustomHeaders {
		request.Header.Set(key, value)
	}

	resp, err := c.client.Do(request)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()

	readAll, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}

	return &responseResult{code: resp.StatusCode, body: readAll, cookies: resp.Cookies()}, nil
}

func (c *client) cookies() (string, error) {
	if c.client.Jar == nil {
		return "", ErrNotLogin
	}
	u, err := url.Parse(c.config.Address)
	if err != nil {
		return "", err
	}
	cookies := c.client.Jar.Cookies(u)
	if len(cookies) == 0 {
		return "", ErrNotLogin
	}
	var builder strings.Builder
	for _, cookie := range cookies {
		builder.WriteString(fmt.Sprintf("%s=%s; ", cookie.Name, cookie.Value))
	}

	return builder.String(), nil
}

func (c *client) refreshCookie() {
	ctx := context.Background()
	if c.config.RefreshIntervals == 0 {
		c.config.RefreshIntervals = time.Hour
	}
	var ticker = time.NewTicker(c.config.RefreshIntervals)
	for range ticker.C {
		if err := c.Authentication().Logout(ctx); err != nil {
			// todo
		}
	}
}