bitmagnet/internal/app/cmd/processcmd/command.go

package processcmd

import (
	"github.com/bitmagnet-io/bitmagnet/internal/boilerplate/lazy"
	"github.com/bitmagnet-io/bitmagnet/internal/processor"
	"github.com/bitmagnet-io/bitmagnet/internal/protocol"
	"github.com/urfave/cli/v2"
	"go.uber.org/fx"
	"go.uber.org/zap"
)

type Params struct {
	fx.In
	Processor lazy.Lazy[processor.Processor]
	Logger    *zap.SugaredLogger
}

type Result struct {
	fx.Out
	Command *cli.Command `group:"commands"`
}

func New(p Params) (Result, error) {
	return Result{Command: &cli.Command{
		Name: "process",
		Flags: []cli.Flag{
			&cli.StringSliceFlag{
				Name: "infoHash",
			},
		},
		Action: func(ctx *cli.Context) error {
			pr, err := p.Processor.Get()
			if err != nil {
				return err
			}
			var infoHashes []protocol.ID
			for _, infoHash := range ctx.StringSlice("infoHash") {
				id, err := protocol.ParseID(infoHash)
				if err != nil {
					return err
				}
				infoHashes = append(infoHashes, id)
			}
			if err != nil {
				return err
			}
			return pr.Process(ctx.Context, processor.MessageParams{
				ClassifyMode: processor.ClassifyModeRematch,
				InfoHashes:   infoHashes,
			})
		},
	},
	}, nil
}