89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
|
package torrent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"git.kmsign.ru/royalcat/tstor/pkg/kvsingle"
|
||
|
|
"github.com/anacrolix/torrent"
|
||
|
|
"github.com/anacrolix/torrent/metainfo"
|
||
|
|
"github.com/anacrolix/torrent/types"
|
||
|
|
"github.com/royalcat/kv"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FileController struct {
|
||
|
|
file *torrent.File
|
||
|
|
properties *kvsingle.Value[string, FileProperties]
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewFileController(f *torrent.File, properties *kvsingle.Value[string, FileProperties]) *FileController {
|
||
|
|
return &FileController{
|
||
|
|
file: f,
|
||
|
|
properties: properties,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) Properties(ctx context.Context) (FileProperties, error) {
|
||
|
|
p, err := s.properties.Get(ctx)
|
||
|
|
if err == kv.ErrKeyNotFound {
|
||
|
|
return FileProperties{
|
||
|
|
Excluded: false,
|
||
|
|
Priority: defaultPriority,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return FileProperties{}, err
|
||
|
|
}
|
||
|
|
return p, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) SetPriority(ctx context.Context, priority types.PiecePriority) error {
|
||
|
|
err := s.properties.Edit(ctx, func(ctx context.Context, v FileProperties) (FileProperties, error) {
|
||
|
|
v.Priority = priority
|
||
|
|
return v, nil
|
||
|
|
})
|
||
|
|
if err == kv.ErrKeyNotFound {
|
||
|
|
seterr := s.properties.Set(ctx, FileProperties{
|
||
|
|
Priority: priority,
|
||
|
|
})
|
||
|
|
if seterr != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
err = nil
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
s.file.SetPriority(priority)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) FileInfo() metainfo.FileInfo {
|
||
|
|
return s.file.FileInfo()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) Excluded(ctx context.Context) (bool, error) {
|
||
|
|
p, err := s.properties.Get(ctx)
|
||
|
|
if err == kv.ErrKeyNotFound {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
return p.Excluded, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) Path() string {
|
||
|
|
return s.file.Path()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) Size() int64 {
|
||
|
|
return s.file.Length()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *FileController) BytesCompleted() int64 {
|
||
|
|
return s.file.BytesCompleted()
|
||
|
|
}
|