MeBox/internal/service/qbittorrent_adp_list.go

package service

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"
	"strings"
	"time"
)

// List 列出种子。
func (a *QBitAdapter) List(ctx context.Context, filter string) ([]TorrentInfo, error) {
	a.mu.Lock()
	defer a.mu.Unlock()
	if err := a.ensureAuthLocked(ctx); err != nil {
		return nil, err
	}
	baseURL := strings.TrimRight(a.cfg.Host, "/")
	u := baseURL + "/api/v2/torrents/info"
	req, err := newDownloadClientHTTPRequest(ctx, http.MethodGet, u, nil)
	if err != nil {
		return nil, err
	}
	if filter != "" {
		query := req.URL.Query()
		query.Set("filter", filter)
		req.URL.RawQuery = query.Encode()
	}
	req.Header.Set("Referer", baseURL)
	resp, err := a.client.Do(req)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()
	if resp.StatusCode >= 400 {
		return nil, fmt.Errorf("qbittorrent list: %d", resp.StatusCode)
	}

	var qbList []qbitTorrentListItem
	if err := json.NewDecoder(resp.Body).Decode(&qbList); err != nil {
		return nil, err
	}
	return qbitTorrentListToInfo(qbList), nil
}

// GetInfo 获取单个种子信息。
func (a *QBitAdapter) GetInfo(ctx context.Context, hash string) (*TorrentInfo, error) {
	list, err := a.List(ctx, "")
	if err != nil {
		return nil, err
	}
	for _, t := range list {
		if t.Hash == hash {
			return &t, nil
		}
	}
	return nil, fmt.Errorf("torrent %s not found", hash)
}

type qbitTorrentListItem struct {
	Hash         string  `json:"hash"`
	Name         string  `json:"name"`
	State        string  `json:"state"`
	Progress     float32 `json:"progress"`
	DLSpeed      int64   `json:"dlspeed"`
	UPSpeed      int64   `json:"upspeed"`
	NumSeeds     int     `json:"num_seeds"`
	NumLeechs    int     `json:"num_leechs"`
	Size         int64   `json:"size"`
	SavePath     string  `json:"save_path"`
	AddedOn      int64   `json:"added_on"`
	Category     string  `json:"category"`
	Tags         string  `json:"tags"`
	ContentPath  string  `json:"content_path"`
	CompletionOn int64   `json:"completion_on"`
}

func qbitTorrentListToInfo(items []qbitTorrentListItem) []TorrentInfo {
	result := make([]TorrentInfo, 0, len(items))
	for _, item := range items {
		result = append(result, TorrentInfo{
			Hash:         item.Hash,
			Name:         item.Name,
			Size:         item.Size,
			Progress:     normalizedTorrentProgress(float64(item.Progress)),
			DLSpeed:      item.DLSpeed,
			UPSpeed:      item.UPSpeed,
			State:        canonicalTorrentState(item.State, float64(item.Progress)),
			SavePath:     item.SavePath,
			NumSeeds:     item.NumSeeds,
			NumLeechs:    item.NumLeechs,
			AddedOn:      time.Unix(item.AddedOn, 0),
			Category:     item.Category,
			Tags:         item.Tags,
			ContentPath:  item.ContentPath,
			CompletionOn: item.CompletionOn,
		})
	}
	return result
}