Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add google chat #155

Merged
merged 1 commit into from Jul 14, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
51 changes: 51 additions & 0 deletions pkg/providers/googlechat/googlechat.go
@@ -0,0 +1,51 @@
package googlechat

import (
"fmt"

"github.com/containrrr/shoutrrr"
"github.com/pkg/errors"
"github.com/projectdiscovery/gologger"
"github.com/projectdiscovery/notify/pkg/utils"
"go.uber.org/multierr"
)

type Provider struct {
GoogleChat []*Options `yaml:"googleChat,omitempty"`
}

type Options struct {
ID string `yaml:"id,omitempty"`
Space string `yaml:"space,omitempty"`
Key string `yaml:"key,omitempty"`
Token string `yaml:"token,omitempty"`
GoogleChatFormat string `yaml:"google_chat_format,omitempty"`
}

func New(options []*Options, ids []string) (*Provider, error) {
provider := &Provider{}

for _, o := range options {
if len(ids) == 0 || utils.Contains(ids, o.ID) {
provider.GoogleChat = append(provider.GoogleChat, o)
}
}

return provider, nil
}

func (p *Provider) Send(message, CliFormat string) error {
var GoogleChatErr error
for _, pr := range p.GoogleChat {
msg := utils.FormatMessage(message, utils.SelectFormat(CliFormat, pr.GoogleChatFormat))
url := fmt.Sprintf("googlechat://chat.googleapis.com/v1/spaces/%s/messages?key=%s&token=%s", pr.Space, pr.Key, pr.Token)
err := shoutrrr.Send(url, msg)
if err != nil {
err = errors.Wrap(err, fmt.Sprintf("failed to send googleChat notification for id: %s ", pr.ID))
GoogleChatErr = multierr.Append(GoogleChatErr, err)
continue
}
gologger.Verbose().Msgf("googleChat notification sent for id: %s", pr.ID)
}
return GoogleChatErr
}
24 changes: 17 additions & 7 deletions pkg/providers/providers.go
Expand Up @@ -6,6 +6,7 @@ import (
"github.com/projectdiscovery/gologger"
"github.com/projectdiscovery/notify/pkg/providers/custom"
"github.com/projectdiscovery/notify/pkg/providers/discord"
"github.com/projectdiscovery/notify/pkg/providers/googlechat"
"github.com/projectdiscovery/notify/pkg/providers/pushover"
"github.com/projectdiscovery/notify/pkg/providers/slack"
"github.com/projectdiscovery/notify/pkg/providers/smtp"
Expand All @@ -18,13 +19,14 @@ import (

// ProviderOptions is configuration for notify providers
type ProviderOptions struct {
Slack []*slack.Options `yaml:"slack,omitempty"`
Discord []*discord.Options `yaml:"discord,omitempty"`
Pushover []*pushover.Options `yaml:"pushover,omitempty"`
SMTP []*smtp.Options `yaml:"smtp,omitempty"`
Teams []*teams.Options `yaml:"teams,omitempty"`
Telegram []*telegram.Options `yaml:"telegram,omitempty"`
Custom []*custom.Options `yaml:"custom,omitempty"`
Slack []*slack.Options `yaml:"slack,omitempty"`
Discord []*discord.Options `yaml:"discord,omitempty"`
Pushover []*pushover.Options `yaml:"pushover,omitempty"`
SMTP []*smtp.Options `yaml:"smtp,omitempty"`
Teams []*teams.Options `yaml:"teams,omitempty"`
Telegram []*telegram.Options `yaml:"telegram,omitempty"`
GoogleChat []*googlechat.Options `yaml:"googlechat,omitempty"`
Custom []*custom.Options `yaml:"custom,omitempty"`
}

// Provider is an interface implemented by providers
Expand Down Expand Up @@ -67,6 +69,14 @@ func New(providerOptions *ProviderOptions, options *types.Options) (*Client, err
}
client.providers = append(client.providers, provider)
}
if providerOptions.GoogleChat != nil && (len(options.Providers) == 0 || utils.Contains(options.Providers, "googlechat")) {

provider, err := googlechat.New(providerOptions.GoogleChat, options.IDs)
if err != nil {
return nil, errors.Wrap(err, "could not create googlechat provider client")
}
client.providers = append(client.providers, provider)
}
if providerOptions.SMTP != nil && (len(options.Providers) == 0 || utils.Contains(options.Providers, "smtp")) {

provider, err := smtp.New(providerOptions.SMTP, options.IDs)
Expand Down