main.go

package main

import (
	"context"
	"log"
	"os"
	"os/signal"
	"syscall"

	"github.com/urfave/cli/v3"

	"github.com/jacobbrewer1/trackify/ipaddress"
	"github.com/jacobbrewer1/trackify/username"
)

func coreContext() (context.Context, context.CancelFunc) {
	return signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
}

func main() {
	cmd := &cli.Command{
		Commands: []*cli.Command{
			{
				Name:  "username",
				Usage: "Track online accounts for a given username",
				Flags: []cli.Flag{
					&cli.StringSliceFlag{
						Name:    "target",
						Aliases: []string{"t"},
						Usage:   "Specify a target platform to search (can be specified multiple times). If not specified, all platforms will be searched.",
					},
				},
				Before: func(ctx context.Context, cmd *cli.Command) (context.Context, error) {
					args := cmd.Args()
					if args.Len() < 1 {
						return nil, cli.Exit("at least one username is required", 1)
					}
					return ctx, nil
				},
				Action: func(ctx context.Context, cmd *cli.Command) error {
					return username.TrackUsernames(
						ctx,
						cmd.Args().Slice(),
						cmd.StringSlice("target"),
					)
				},
			},
			{
				Name:  "ip",
				Usage: "Track given IP addresses",
				Before: func(ctx context.Context, cmd *cli.Command) (context.Context, error) {
					args := cmd.Args()
					if args.Len() < 1 {
						return nil, cli.Exit("at least one IP Address is required", 1)
					}
					return ctx, nil
				},
				Action: func(ctx context.Context, cmd *cli.Command) error {
					return ipaddress.TrackIPAddresses(
						ctx,
						cmd.Args().Slice(),
					)
				},
			},
		},
	}

	ctx, cancel := coreContext()
	defer cancel()

	if err := cmd.Run(ctx, os.Args); err != nil {
		cancel()
		log.Fatal(err) // nolint:gocritic // Calling cancel() before exiting on the line above
	}
}