Function NewMonitorPartnerHashDownloadCmd

cmd/monitorpartner.go:111–144

NewMonitorPartnerHashDownloadCmd returns a command for downloading files from your monitor account.

()

// NewMonitorPartnerHashDownloadCmd returns a command for downloading files from your
// monitor account.
func NewMonitorPartnerHashDownloadCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:     "download [sha256]...",
		Short:   "Download files from your monitor partner account",
		Long:    monitorPartnerHashDownloadCmdHelp,
		Example: monitorPartnerHashDownloadCmdExample,
		RunE: func(cmd *cobra.Command, args []string) error {
			var argReader utils.StringReader
			if len(args) == 0 {
				return errors.New("No hash provided")
			} else if len(args) == 1 && args[0] == "-" {
				argReader = utils.NewStringIOReader(os.Stdin)
			} else {
				argReader = utils.NewStringArrayReader(args)
			}
			client, err := NewAPIClient()
			if err != nil {
				return err
			}
			re, _ := regexp.Compile("[[:xdigit:]]{64}")
			monitorHashes := utils.NewFilteredStringReader(argReader, re)

			c := utils.NewCoordinator(viper.GetInt("threads"))
			c.DoWithStringsFromReader(
				&monitorPartnerDownloader{fileDownloader: newFileDownloader(client)},
				monitorHashes)
			return err
		},
	}

	addThreadsFlag(cmd.Flags())
	addOutputFlag(cmd.Flags())
	return cmd
}

var monitorPartnerCmdHelp = `Manage your VirusTotal Monitor Partner account.