Function NewScanFileCmd

cmd/scan.go:153–203

NewScanFileCmd returns a new instance of the 'scan file' command.

()

// NewScanFileCmd returns a new instance of the 'scan file' command.
func NewScanFileCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:     "file [[dir] | [file]...]",
		Short:   "Scan one or more files",
		Long:    scanFileCmdHelp,
		Example: scanFileCmdExample,
		Args:    cobra.MinimumNArgs(1),

		RunE: func(cmd *cobra.Command, args []string) error {
			c := utils.NewCoordinator(viper.GetInt("threads"))
			var argReader utils.StringReader
			if len(args) == 1 && args[0] == "-" {
				argReader = utils.NewStringIOReader(os.Stdin)
			} else if len(args) == 1 && utils.IsDir(args[0]) {
				recursive := viper.GetBool("recursive")
				maxDepth := viper.GetInt("maxDepth")
				argReader, _ = utils.NewFileDirReader(args[0], recursive, maxDepth)
			} else {
				argReader = utils.NewStringArrayReader(args)
			}
			client, err := NewAPIClient()
			if err != nil {
				return err
			}
			p, err := NewPrinter(cmd)
			if err != nil {
				return err
			}
			s := &fileScanner{
				scanner:           client.NewFileScanner(),
				showInVT:          viper.GetBool("open"),
				waitForCompletion: viper.GetBool("wait"),
				password:          viper.GetString("password"),
				printer:           p,
				cli:               client}
			c.DoWithStringsFromReader(s, argReader)
			return nil
		},
	}

	addRecursive(cmd.Flags())
	addMaxDepth(cmd.Flags())
	addThreadsFlag(cmd.Flags())
	addOpenInVTFlag(cmd.Flags())
	addPasswordFlag(cmd.Flags())
	addWaitForCompletionFlag(cmd.Flags())
	addIncludeExcludeFlags(cmd.Flags())
	cmd.MarkZshCompPositionalArgumentFile(1)

	return cmd
}

type urlScanner struct {
	scanner           *vt.URLScanner