types

type ByteLimiter interface {
	WaitN(ctx context.Context, n int64) error
}

type CancelResult struct {
	Found     bool
	Filename  string
	DestPath  string
	Completed bool
	WasQueued bool
}

type ChunkStatus int

type DownloadConfig struct {
	URL        string
	OutputPath string
	DestPath   string
	ID         string
	Filename   string
	IsResume   bool
	ProgressCh chan<- any
	State      *ProgressState
	SavedState *DownloadState
	Runtime    *RuntimeConfig
	Mirrors    []string
	Headers    map[string]string
	Limiter    ByteLimiter

	IsExplicitCategory bool
	TotalSize          int64
	SupportsRange      bool
	RateLimitBps       int64
	RateLimitSet       bool
}

type DownloadEntry struct {
	ID           string   `json:"id"`
	URLHash      string   `json:"url_hash"`
	URL          string   `json:"url"`
	DestPath     string   `json:"dest_path"`
	Filename     string   `json:"filename"`
	Status       string   `json:"status"`
	TotalSize    int64    `json:"total_size"`
	Downloaded   int64    `json:"downloaded"`
	CompletedAt  int64    `json:"completed_at"`
	TimeTaken    int64    `json:"time_taken"`
	AvgSpeed     float64  `json:"avg_speed"`
	Mirrors      []string `json:"mirrors,omitempty"`
	RateLimit    int64    `json:"rate_limit,omitempty"`
	RateLimitSet bool     `json:"rate_limit_set,omitempty"`
}

type DownloadState struct {
	ID         string   `json:"id"`
	URLHash    string   `json:"url_hash"`
	URL        string   `json:"url"`
	DestPath   string   `json:"dest_path"`
	TotalSize  int64    `json:"total_size"`
	Downloaded int64    `json:"downloaded"`
	Tasks      []Task   `json:"tasks"`
	Filename   string   `json:"filename"`
	CreatedAt  int64    `json:"created_at"`
	PausedAt   int64    `json:"paused_at"`
	Elapsed    int64    `json:"elapsed"`
	Mirrors    []string `json:"mirrors,omitempty"`

	ChunkBitmap     []byte `json:"chunk_bitmap,omitempty"`
	ActualChunkSize int64  `json:"actual_chunk_size,omitempty"`

	FileHash     string `json:"file_hash,omitempty"`
	RateLimit    int64  `json:"rate_limit,omitempty"`
	RateLimitSet bool   `json:"rate_limit_set,omitempty"`
}

type DownloadStatus struct {
	ID           string  `json:"id"`
	URL          string  `json:"url"`
	Filename     string  `json:"filename"`
	DestPath     string  `json:"dest_path,omitempty"`
	TotalSize    int64   `json:"total_size"`
	Downloaded   int64   `json:"downloaded"`
	Progress     float64 `json:"progress"`
	Speed        float64 `json:"speed"`
	Status       string  `json:"status"`
	Error        string  `json:"error,omitempty"`
	ETA          int64   `json:"eta"`
	Connections  int     `json:"connections"`
	AddedAt      int64   `json:"added_at"`
	TimeTaken    int64   `json:"time_taken"`
	AvgSpeed     float64 `json:"avg_speed"`
	RateLimit    int64   `json:"rate_limit,omitempty"`
	RateLimitSet bool    `json:"rate_limit_set,omitempty"`
}

type MasterList struct {
	Downloads []DownloadEntry `json:"downloads"`
}

type MirrorStatus struct {
	URL    string
	Active bool
	Error  bool
}

type ProgressState struct {
	ID            string
	Downloaded    atomic.Int64
	TotalSize     int64
	DestPath      string // Initial destination path
	Filename      string // Initial filename
	URL           string // Source URL
	StartTime     time.Time
	ActiveWorkers atomic.Int32
	Done          atomic.Bool
	Error         atomic.Pointer[error]
	Paused        atomic.Bool
	Pausing       atomic.Bool // Intermediate state: Pause requested but workers not yet exited

	VerifiedProgress  atomic.Int64  // Verified bytes written to disk (for UI progress)
	SessionStartBytes int64         // SessionStartBytes tracks how many bytes were already downloaded when the current session started
	SavedElapsed      time.Duration // Time spent in previous sessions
	RateLimitBps      int64         // Effective per-download rate limit in bytes/sec
	RateLimitSet      bool          // Whether RateLimitBps is an explicit per-download override

	Mirrors []MirrorStatus

	ChunkBitmap     []byte
	ChunkProgress   []int64
	ActualChunkSize int64
	BitmapWidth     int
	// contains filtered or unexported fields
}

type RuntimeConfig struct {
	MaxConnectionsPerDownload   int
	UserAgent                   string
	ProxyURL                    string
	CustomDNS                   string
	SequentialDownload          bool
	MinChunkSize                int64
	GlobalRateLimitBps          int64
	DefaultDownloadRateLimitBps int64

	WorkerBufferSize      int
	MaxTaskRetries        int
	DialHedgeCount        int
	SlowWorkerThreshold   float64
	SlowWorkerGracePeriod time.Duration
	StallTimeout          time.Duration
	SpeedEmaAlpha         float64
}

type Task struct {
	Offset          int64         `json:"offset"`
	Length          int64         `json:"length"`
	SharedMaxOffset *atomic.Int64 `json:"-"`
}