-
Notifications
You must be signed in to change notification settings - Fork 42
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
3 changed files
with
140 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,137 @@ | ||
| package cmd | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "log" | ||
| "strings" | ||
|
|
||
| "github.com/nikoksr/notify" | ||
| "github.com/nikoksr/notify/service/pushbullet" | ||
| "github.com/urfave/cli/v2" | ||
| ) | ||
|
|
||
| // pushBullet struct holds data parsed via flags for pushbullet service. | ||
| type pushBullet struct { | ||
| Token string | ||
| Message string | ||
| Title string | ||
| Device string | ||
| PhoneNumber string | ||
| SMS bool | ||
| } | ||
|
|
||
| // SendToPushBullet parse values from *cli.context and return *cli.Command. | ||
| // Values include pushbullet token, Device, phone number, Message and Title. | ||
| // If multiple devices are provided they the string is split with "," separator and | ||
| // each device is added to receiver. | ||
| func SendToPushBullet() *cli.Command { | ||
| var pushBulletOpts pushBullet | ||
| return &cli.Command{ | ||
| Name: "pushbullet", | ||
| Usage: "Send message to pushbullet", | ||
| Description: `Pushbullet uses API token to authenticate & send messages to defined devices. | ||
| Multiple device nicknames or numbers can be used separated by comma.`, | ||
| UsageText: "pingme pushbullet --token '123' --device 'Web123, myAndroid' --msg 'some message'\n" + | ||
| "pingme pushbullet --token '123' --sms true --device 'Web123' --msg 'some message' --number '00123456789'", | ||
| Flags: []cli.Flag{ | ||
| &cli.StringFlag{ | ||
| Destination: &pushBulletOpts.Token, | ||
| Name: "token", | ||
| Aliases: []string{"t"}, | ||
| Required: true, | ||
| Usage: "Token of pushbullet api used for sending message.", | ||
| EnvVars: []string{"PUSHBULLET_TOKEN"}, | ||
| }, | ||
| &cli.StringFlag{ | ||
| Destination: &pushBulletOpts.Device, | ||
| Name: "device", | ||
| Aliases: []string{"d"}, | ||
| Required: true, | ||
| Usage: "Device's nickname of pushbullet.", | ||
| EnvVars: []string{"PUSHBULLET_DEVICE"}, | ||
| }, | ||
| &cli.StringFlag{ | ||
| Destination: &pushBulletOpts.PhoneNumber, | ||
| Name: "number", | ||
| Aliases: []string{"n"}, | ||
| Usage: "Target phone number", | ||
| EnvVars: []string{"PUSHBULLET_NUMBER"}, | ||
| }, | ||
| &cli.StringFlag{ | ||
| Destination: &pushBulletOpts.Message, | ||
| Name: "msg", | ||
| Aliases: []string{"m"}, | ||
| Usage: "Message content.", | ||
| EnvVars: []string{"PUSHBULLET_MESSAGE"}, | ||
| }, | ||
| &cli.StringFlag{ | ||
| Destination: &pushBulletOpts.Title, | ||
| Name: "title", | ||
| Value: TimeValue, | ||
| Usage: "Title of the message.", | ||
| EnvVars: []string{"PUSHBULLET_TITLE"}, | ||
| }, | ||
| &cli.BoolFlag{ | ||
| Destination: &pushBulletOpts.SMS, | ||
| Name: "sms", | ||
| Value: false, | ||
| Usage: "To send sms message set the value to 'true'", | ||
| EnvVars: []string{"PUSHBULLET_SMS"}, | ||
| }, | ||
| }, | ||
| Action: func(ctx *cli.Context) error { | ||
| notifier := notify.New() | ||
|
|
||
| switch pushBulletOpts.SMS { | ||
| case true: | ||
| pushBulletSmsSvc, err := pushbullet.NewSMS(pushBulletOpts.Token, pushBulletOpts.Device) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| devices := strings.Split(pushBulletOpts.PhoneNumber, ",") | ||
| for _, v := range devices { | ||
| if len(v) <= 0 { | ||
| return fmt.Errorf(EmptyChannel) | ||
| } | ||
| pushBulletSmsSvc.AddReceivers(v) | ||
|
|
||
| notifier.UseServices(pushBulletSmsSvc) | ||
|
|
||
| if err := notifier.Send( | ||
| context.Background(), | ||
| pushBulletOpts.Title, | ||
| pushBulletOpts.Message, | ||
| ); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| } | ||
| default: | ||
| pushBulletSvc := pushbullet.New(pushBulletOpts.Token) | ||
|
|
||
| devices := strings.Split(pushBulletOpts.Device, ",") | ||
| for _, v := range devices { | ||
| if len(v) <= 0 { | ||
| return fmt.Errorf(EmptyChannel) | ||
| } | ||
| pushBulletSvc.AddReceivers(v) | ||
| } | ||
|
|
||
| notifier.UseServices(pushBulletSvc) | ||
|
|
||
| if err := notifier.Send( | ||
| context.Background(), | ||
| pushBulletOpts.Title, | ||
| pushBulletOpts.Message, | ||
| ); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| } | ||
|
|
||
| log.Println("Successfully sent!") | ||
| return nil | ||
| }, | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters