78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package storage
|
|
|
|
import (
|
|
"path"
|
|
|
|
"github.com/anacrolix/missinggo/resource"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
)
|
|
|
|
type piecePerResource struct {
|
|
p resource.Provider
|
|
}
|
|
|
|
func NewResourcePieces(p resource.Provider) ClientImpl {
|
|
return &piecePerResource{
|
|
p: p,
|
|
}
|
|
}
|
|
|
|
func (s *piecePerResource) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (TorrentImpl, error) {
|
|
return s, nil
|
|
}
|
|
|
|
func (s *piecePerResource) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *piecePerResource) Piece(p metainfo.Piece) PieceImpl {
|
|
completed, err := s.p.NewInstance(path.Join("completed", p.Hash().HexString()))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
incomplete, err := s.p.NewInstance(path.Join("incomplete", p.Hash().HexString()))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return piecePerResourcePiece{
|
|
p: p,
|
|
c: completed,
|
|
i: incomplete,
|
|
}
|
|
}
|
|
|
|
type piecePerResourcePiece struct {
|
|
p metainfo.Piece
|
|
c resource.Instance
|
|
i resource.Instance
|
|
}
|
|
|
|
func (s piecePerResourcePiece) Completion() Completion {
|
|
fi, err := s.c.Stat()
|
|
return Completion{
|
|
Complete: err == nil && fi.Size() == s.p.Length(),
|
|
Ok: true,
|
|
}
|
|
}
|
|
|
|
func (s piecePerResourcePiece) MarkComplete() error {
|
|
return resource.Move(s.i, s.c)
|
|
}
|
|
|
|
func (s piecePerResourcePiece) MarkNotComplete() error {
|
|
return s.c.Delete()
|
|
}
|
|
|
|
func (s piecePerResourcePiece) ReadAt(b []byte, off int64) (int, error) {
|
|
if s.Completion().Complete {
|
|
return s.c.ReadAt(b, off)
|
|
} else {
|
|
return s.i.ReadAt(b, off)
|
|
}
|
|
}
|
|
|
|
func (s piecePerResourcePiece) WriteAt(b []byte, off int64) (n int, err error) {
|
|
return s.i.WriteAt(b, off)
|
|
}
|