diff options
Diffstat (limited to 'cmd/telegram.go')
| -rw-r--r-- | cmd/telegram.go | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/cmd/telegram.go b/cmd/telegram.go index 98092a6..23da10a 100644 --- a/cmd/telegram.go +++ b/cmd/telegram.go @@ -17,7 +17,7 @@ var phoneCmd = &cobra.Command{ Aliases: []string{"p"}, Short: "Get phone number for Telegram user", Long: `Retrieves the phone number associated with a Telegram username (with @ prefix) or user ID`, - Args: cobra.ExactArgs(1), + Args: argsOrStdin(1), RunE: runTelegramPhone, } @@ -27,21 +27,19 @@ func init() { } func runTelegramPhone(cmd *cobra.Command, args []string) error { - identifier := args[0] - c, err := newClient() if err != nil { return err } - response, err := c.GetTelegramPhone(identifier) - if err != nil { - return err - } - - if err := checkError(response.Error); err != nil { - return err - } - - return formatter.FormatTelegramResults(response, IsJSONOutput()) + return forEachQuery(args, func(identifier string) error { + response, err := c.GetTelegramPhone(identifier) + if err != nil { + return err + } + if err := checkError(response.Error); err != nil { + return err + } + return formatter.FormatTelegramResults(response, IsJSONOutput()) + }) } |
