The vt url command now accepts URL identifiers as returned by the API.

For example ` vt url f1177df4692356280844e1d5af67cc4a9eccecf77aa61c229d483b7082c70a8e` is equivalent to `vt url https://www.virustotal.com/`. This use case was already mentioned in the examples for the command, but it was not implemented.

Author: plusvic

cmd/url.go

@@ -17,6 +17,7 @@ import (
 	"encoding/base64"
 	"github.com/VirusTotal/vt-cli/utils"
 	"github.com/spf13/cobra"
+	"regexp"
 )
 
 var urlCmdHelp = `Get information about one or more URLs.
@@ -34,6 +35,10 @@ var urlCmdExample = `  vt url https://www.virustotal.com
   vt url f1177df4692356280844e1d5af67cc4a9eccecf77aa61c229d483b7082c70a8e
   cat list_of_urls | vt url -`
 
+
+// Regular expressions used for validating a URL identifier.
+var urlID = regexp.MustCompile(`[0-9a-fA-F]{64}`)
+
 // NewURLCmd returns a new instance of the 'url' command.
 func NewURLCmd() *cobra.Command {
 	cmd := &cobra.Command{
@@ -51,6 +56,14 @@ func NewURLCmd() *cobra.Command {
 			r := utils.NewMappedStringReader(
 				utils.StringReaderFromCmdArgs(args),
 				func (url string) string {
+					if urlID.MatchString(url) {
+						// The user provided a URL identifier as returned by
+						// VirusTotal's API, which consists in the URL's SHA-256.
+						// In that case use the identifier as is.
+						return url
+					}
+					// If the user provides an actual URL, it needs to be
+					// encoded as base64 before being used.
 					return base64.RawURLEncoding.EncodeToString([]byte(url))
 				})
 			return p.GetAndPrintObjects("urls/%s", r, nil)