2023-12-31 22:54:55 +00:00
|
|
|
package storage
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
|
|
|
|
"git.kmsign.ru/royalcat/tstor/src/config"
|
|
|
|
"github.com/anacrolix/torrent/storage"
|
|
|
|
)
|
|
|
|
|
2024-01-01 18:17:32 +00:00
|
|
|
func SetupStorage(cfg config.TorrentClient) (storage.ClientImplCloser, storage.PieceCompletion, error) {
|
2023-12-31 22:54:55 +00:00
|
|
|
pcp := filepath.Join(cfg.DataFolder, "piece-completion")
|
|
|
|
if err := os.MkdirAll(pcp, 0744); err != nil {
|
|
|
|
return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
|
|
|
|
}
|
|
|
|
pc, err := storage.NewBoltPieceCompletion(pcp)
|
|
|
|
if err != nil {
|
|
|
|
return nil, nil, fmt.Errorf("error creating servers piece completion: %w", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// pc, err := NewBadgerPieceCompletion(pcp)
|
|
|
|
// if err != nil {
|
|
|
|
// return nil, nil, fmt.Errorf("error creating servers piece completion: %w", err)
|
|
|
|
// }
|
|
|
|
|
|
|
|
// TODO implement cache/storage switching
|
|
|
|
// cacheDir := filepath.Join(tcfg.DataFolder, "cache")
|
|
|
|
// if err := os.MkdirAll(cacheDir, 0744); err != nil {
|
|
|
|
// return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
|
|
|
|
// }
|
|
|
|
// fc, err := filecache.NewCache(cacheDir)
|
|
|
|
// if err != nil {
|
|
|
|
// return nil, nil, fmt.Errorf("error creating cache: %w", err)
|
|
|
|
// }
|
|
|
|
// log.Info().Msg(fmt.Sprintf("setting cache size to %d MB", 1024))
|
|
|
|
// fc.SetCapacity(1024 * 1024 * 1024)
|
|
|
|
|
|
|
|
// rp := storage.NewResourcePieces(fc.AsResourceProvider())
|
|
|
|
// st := &stc{rp}
|
|
|
|
|
2024-01-01 18:17:32 +00:00
|
|
|
// filesDir := filepath.Join(cfg.DataFolder, "files")
|
|
|
|
// if err := os.MkdirAll(filesDir, 0744); err != nil {
|
|
|
|
// return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
|
|
|
|
// }
|
|
|
|
|
|
|
|
//st := NewFileStorage(filesDir, pc)
|
|
|
|
|
|
|
|
piecesDir := filepath.Join(cfg.DataFolder, "pieces")
|
|
|
|
if err := os.MkdirAll(piecesDir, 0744); err != nil {
|
2023-12-31 22:54:55 +00:00
|
|
|
return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
|
|
|
|
}
|
|
|
|
|
2024-01-01 18:17:32 +00:00
|
|
|
st := storage.NewMMapWithCompletion(piecesDir, pc)
|
2023-12-31 22:54:55 +00:00
|
|
|
|
|
|
|
return st, pc, nil
|
|
|
|
}
|