tstor/mount/handler.go

package mount

import (
	"fmt"
	"os"

	"github.com/ajnavarro/distribyted/config"
	"github.com/ajnavarro/distribyted/node"
	"github.com/ajnavarro/distribyted/stats"
	"github.com/anacrolix/torrent"
	"github.com/hanwen/go-fuse/v2/fs"
	"github.com/hanwen/go-fuse/v2/fuse"
	log "github.com/sirupsen/logrus"
)

type Handler struct {
	c    *torrent.Client
	s    *stats.Torrent
	opts *fs.Options

	servers map[string]*fuse.Server
}

func NewHandler(c *torrent.Client, s *stats.Torrent) *Handler {
	return &Handler{
		c:       c,
		s:       s,
		opts:    &fs.Options{},
		servers: make(map[string]*fuse.Server),
	}
}

func (s *Handler) Mount(mpc *config.MountPoint) error {
	var torrents []*torrent.Torrent
	for _, mpcTorrent := range mpc.Torrents {
		var t *torrent.Torrent
		var err error

		switch {
		case mpcTorrent.MagnetURI != "":
			t, err = s.c.AddMagnet(mpcTorrent.MagnetURI)
			break
		case mpcTorrent.TorrentPath != "":
			t, err = s.c.AddTorrentFromFile(mpcTorrent.TorrentPath)
			break
		default:
			err = fmt.Errorf("no magnet URI or torrent path provided")
		}

		if err != nil {
			return err
		}

		// only get info if name is not available
		if t.Name() == "" {
			log.WithField("hash", t.InfoHash()).Info("getting torrent info")
			<-t.GotInfo()
		}

		s.s.Add(mpc.Path, t)
		log.WithField("name", t.Name()).WithField("path", mpc.Path).Info("torrent added to mountpoint")

		torrents = append(torrents, t)
	}

	// TODO change permissions
	if err := os.MkdirAll(mpc.Path, 0770); err != nil && !os.IsExist(err) {
		return err
	}

	node := node.NewRoot(torrents)
	server, err := fs.Mount(mpc.Path, node, s.opts)
	if err != nil {
		return err
	}

	s.servers[mpc.Path] = server

	return nil
}

func (s *Handler) Close() {
	for path, server := range s.servers {
		log.WithField("path", path).Info("unmounting")
		err := server.Unmount()
		if err != nil {
			//TODO try to force unmount if possible
			log.WithError(err).WithField("path", path).Error("unmount failed")
		}
	}
}