tstor/cmd/tstor/main.go

package main

import (
	"context"
	"fmt"
	"log/slog"

	"net"
	nethttp "net/http"
	_ "net/http/pprof"
	"os"
	"os/signal"
	"path/filepath"
	"syscall"

	"git.kmsign.ru/royalcat/tstor/pkg/ctxbilly"
	wnfs "git.kmsign.ru/royalcat/tstor/pkg/go-nfs"
	"git.kmsign.ru/royalcat/tstor/pkg/rlog"
	"git.kmsign.ru/royalcat/tstor/src/config"
	"git.kmsign.ru/royalcat/tstor/src/delivery"
	"git.kmsign.ru/royalcat/tstor/src/sources"
	"git.kmsign.ru/royalcat/tstor/src/sources/torrent"
	"git.kmsign.ru/royalcat/tstor/src/sources/ytdlp"
	"git.kmsign.ru/royalcat/tstor/src/telemetry"
	"git.kmsign.ru/royalcat/tstor/src/vfs"
	"github.com/go-git/go-billy/v5/osfs"
	"github.com/urfave/cli/v2"

	_ "git.kmsign.ru/royalcat/tstor/pkg/rlog"
	"git.kmsign.ru/royalcat/tstor/src/export/fuse"
	"git.kmsign.ru/royalcat/tstor/src/export/httpfs"
	"git.kmsign.ru/royalcat/tstor/src/export/nfs"
	"git.kmsign.ru/royalcat/tstor/src/export/webdav"
)

const (
	configFlag     = "config"
	portFlag       = "http-port"
	webDAVPortFlag = "webdav-port"
)

func main() {
	app := &cli.App{
		Name:  "tstor",
		Usage: "Torrent client with on-demand file downloading as a filesystem.",
		Flags: []cli.Flag{
			&cli.StringFlag{
				Name:  configFlag,
				Value: "./config.yaml",
				Usage: "YAML file containing tstor configuration.",
			},
		},

		Action: func(c *cli.Context) error {
			return run(c.String(configFlag))
		},

		HideHelpCommand: true,
	}

	if err := app.Run(os.Args); err != nil {
		print("problem starting application: ", err.Error())
	}
}

func run(configPath string) error {
	conf, err := config.Load(configPath)
	if err != nil {
		return fmt.Errorf("error loading configuration: %w", err)
	}
	// dlog.Load(&conf.Log)

	ctx := context.Background()
	client, err := telemetry.Setup(ctx, conf.OtelHttp)
	if err != nil {
		return err
	}

	defer client.Shutdown(ctx)

	log := rlog.Component("run")

	// TODO make optional
	err = syscall.Setpriority(syscall.PRIO_PGRP, 0, 19)
	if err != nil {
		log.Error(ctx, "set priority failed", rlog.Error(err))
	}

	if err := os.MkdirAll(conf.SourceDir, 0744); err != nil {
		return fmt.Errorf("error creating data folder: %w", err)
	}

	sourceFs := osfs.New(conf.SourceDir, osfs.WithBoundOS())
	tsrv, err := torrent.NewService(sourceFs, conf.TorrentClient)
	if err != nil {
		return fmt.Errorf("error creating service: %w", err)
	}

	err = os.MkdirAll("./ytdlp", 0744)
	if err != nil {
		return err
	}
	ytdlpsrv := ytdlp.NewService("./ytdlp")

	sfs := sources.NewHostedFS(
		vfs.NewCtxBillyFs("/", ctxbilly.WrapFileSystem(sourceFs)),
		tsrv, ytdlpsrv,
	)
	sfs = vfs.WrapLogFS(sfs)

	if conf.Mounts.Fuse.Enabled {
		mh := fuse.NewHandler(conf.Mounts.Fuse.AllowOther, conf.Mounts.Fuse.Path)
		err := mh.Mount(sfs)
		if err != nil {
			return fmt.Errorf("mount fuse error: %w", err)
		}
		defer mh.Unmount()
	}

	if conf.Mounts.WebDAV.Enabled {
		go func() {
			if err := webdav.NewWebDAVServer(sfs, conf.Mounts.WebDAV.Port, conf.Mounts.WebDAV.User, conf.Mounts.WebDAV.Pass); err != nil {
				log.Error(ctx, "error starting webDAV", rlog.Error(err))
			}

			log.Warn(ctx, "webDAV configuration not found!")
		}()
	}
	if conf.Mounts.HttpFs.Enabled {
		go func() {
			httpfs := httpfs.NewHTTPFS(sfs)
			addr := fmt.Sprintf("0.0.0.0:%d", conf.Mounts.HttpFs.Port)
			err = nethttp.ListenAndServe(addr, nethttp.FileServer(httpfs))
			if err != nil {
				log.Error(ctx, "error starting HTTPFS", rlog.Error(err))
			}
			// r := gin.New()

			// r.GET("*filepath", func(c *gin.Context) {
			// 	path := c.Param("filepath")
			// 	c.FileFromFS(path, httpfs)
			// })

			log.Info(ctx, "starting HTTPFS", slog.String("address", addr))
			// if err := r.Run(fmt.Sprintf("0.0.0.0:%d", conf.Mounts.HttpFs.Port)); err != nil {
			// 	log.Error().Err(err).Msg("error starting HTTPFS")
			// }
		}()
	}

	if conf.Mounts.NFS.Enabled {
		go func() {
			log := log.WithComponent("NFS")
			listener, err := net.Listen("tcp", fmt.Sprintf("0.0.0.0:%d", conf.Mounts.NFS.Port))
			if err != nil {
				log.Error(ctx, "failed to start TCP  listener", rlog.Error(err))
				return
			}
			log.Info(ctx, "starting NFS server", slog.String("address", listener.Addr().String()))
			handler, err := nfs.NewNFSv3Handler(sfs)
			if err != nil {
				log.Error(ctx, "failed to create NFS handler", rlog.Error(err))
				return
			}

			err = wnfs.Serve(listener, handler)
			if err != nil {
				log.Error(ctx, "error serving nfs", rlog.Error(err))
				return
			}
		}()
	}

	go func() {
		err := webdav.NewDirServer(conf.SourceDir, 36912, conf.Mounts.WebDAV.User, conf.Mounts.WebDAV.Pass)
		if err != nil {
			log.Error(ctx, "error starting webDAV", rlog.Error(err))
		}
	}()

	go func() {
		logFilename := filepath.Join(conf.Log.Path, "logs")

		err := delivery.New(nil, tsrv, sfs, logFilename, conf)
		if err != nil {
			log.Error(ctx, "error initializing HTTP server", rlog.Error(err))
		}
	}()

	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
	<-sigChan

	return tsrv.Close(ctx)
}