check.go

package cmd

import (
	"fmt"
	"net"

	"github.com/sirupsen/logrus"
	"github.com/urfave/cli/v2"
)

func Check() *cli.Command {
	return &cli.Command{
		Name:      "check",
		Aliases:   []string{"c"},
		Usage:     "Check if ip or host exists",
		Action:    check,
		ArgsUsage: "[IP|HOST]",
	}
}
func check(c *cli.Context) error {
	if c.Args().Len() < 1 {
		logrus.Infof("No input, pass an ip address or hostname to check.")
		return nil
	}

	hf, err := loadHostsfile(c, true)
	if err != nil {
		return err
	}
	input := c.Args().First()

	if net.ParseIP(input) != nil {
		if hf.HasIP(input) {
			logrus.Infof("%s exists in hosts file\n", input)
			return nil
		}
	}

	if hf.HasHostname(input) {
		logrus.Infof("%s exists in hosts file\n", input)
		return nil
	}

	return cli.Exit(fmt.Sprintf("%s does not match anything in the hosts file", input), 1)
}