Provider Notion

cmd/integration-test/integration.go

@@ -26,6 +26,7 @@ var (
 		//		"smtp":     &smtp{},
 		//		"pushover": &pushover{},
 		"gotify": &gotify{},
+		"notion": &notion{},
 	}
 )
 

cmd/integration-test/providers.go

@@ -75,6 +75,12 @@ func (h *gotify) Execute() error {
 	return run("gotify")
 }
 
+type notion struct{}
+
+func (h *notion) Execute() error {
+	return run("notion")
+}
+
 func errIncorrectResultsCount(results []string) error {
 	return fmt.Errorf("incorrect number of results %s", strings.Join(results, "\n\t"))
 }

cmd/integration-test/test-config.yaml

@@ -47,3 +47,8 @@ gotify:
     gotify_token: "${GOTIFY_APP_TOKEN}"
     gotify_format: "{{data}}"
     gotify_disabletls: true
+notion:
+  - id: "notion"
+    notion_api_key: "${notion_api_key}"
+    notion_database_id: "${notion_database_id}"
+    notion_in_page_title: "${notion_in_page_title}" # optional

pkg/providers/notion/notion.go

@@ -0,0 +1,61 @@
+package notion
+
+import (
+	"fmt"
+	"time"
+
+	"go.uber.org/multierr"
+
+	"github.com/projectdiscovery/gologger"
+	sliceutil "github.com/projectdiscovery/utils/slice"
+)
+
+type Provider struct {
+	Notion  []*Options `yaml:"notion,omitempty"`
+	counter int
+}
+
+type Options struct {
+	ID                     string `yaml:"id,omitempty"`
+	NotionAPIKey         string `yaml:"notion_api_key,omitempty"`
+	NotionDatabaseId       string `yaml:"notion_database_id,omitempty"`
+	NotionInPageTitle string `yaml:"notion_in_page_title,omitempty"` // optional
+}
+
+func New(options []*Options, ids []string) (*Provider, error) {
+	provider := &Provider{}
+
+	for _, o := range options {
+		if len(ids) == 0 || sliceutil.Contains(ids, o.ID) {
+			provider.Notion = append(provider.Notion, o)
+		}
+	}
+
+	provider.counter = 0
+
+	return provider, nil
+}
+
+func (p *Provider) Send(message, CliFormat string) error {
+	var errs []error
+	for _, pr := range p.Notion {
+
+		newPage := func() Page {
+			if pr.NotionInPageTitle != "" {
+				titleWithTimestamp := fmt.Sprintf("%s %s", pr.NotionInPageTitle, time.Now().Format("2006-01-02 15:04:05 MST"))
+				return pr.CreateInpageText(titleWithTimestamp, message)
+			}
+			return pr.CreateNormalText(message)
+
+		}()
+
+		if success := pr.CreatePage(newPage); success != nil {
+			err := fmt.Errorf("failed to send notion notification for id: %s", pr.ID)
+			errs = append(errs, err)
+			continue
+		}
+
+		gologger.Verbose().Msgf("Notion notification sent for id: %s", pr.ID)
+	}
+	return multierr.Combine(errs...)
+}

pkg/providers/notion/notion_api.go

@@ -0,0 +1,116 @@
+package notion
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	_ "net/http"
+	"strings"
+)
+
+const notionApiUrl string = "https://api.notion.com/v1/"
+
+func (options *Options) CreatePage(page Page) error {
+	url := strings.Join([]string{notionApiUrl, "pages/"}, "")
+	apiKey := options.NotionAPIKey
+	requestBody := map[string]interface{}{
+		"parent": map[string]string{
+			"database_id": page.ParentId,
+		},
+		"properties": page.Properties,
+	}
+	if page.Children != nil {
+		requestBody["children"] = page.Children
+	}
+
+	jsonBody, err := json.Marshal(requestBody)
+	if err != nil {
+		fmt.Println("Error marshaling request body:", err)
+		return err
+	}
+
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBody))
+	if err != nil {
+		fmt.Println("Error creating request:", err)
+		return err
+	}
+
+	req.Header.Add("Content-Type", "application/json")
+	req.Header.Add("Notion-Version", "2022-02-22")
+	req.Header.Add("Authorization", "Bearer "+apiKey)
+
+	client := &http.Client{}
+	res, err := client.Do(req)
+	if err != nil {
+		fmt.Println("Error sending request:", err)
+		return err
+	}
+	defer res.Body.Close()
+
+	_, err = io.ReadAll(res.Body)
+	if err != nil {
+		fmt.Println("Error reading response body:", err)
+		return err
+	}
+
+	if res.StatusCode != 200 {
+		fmt.Println("Error creating page:", res.Status)
+		return fmt.Errorf("error creating page: %s", res.Status)
+	}
+	return nil
+}
+
+func (options *Options) CreateNormalText(text string) Page {
+	newPage := Page{
+		ParentId: options.NotionDatabaseId,
+		Properties: map[string]interface{}{
+			"Name": map[string]interface{}{
+				"title": []map[string]interface{}{
+					{
+						"text": map[string]string{
+							"content": text,
+						},
+					},
+				},
+			},
+		},
+	}
+
+	return newPage
+}
+
+func (options *Options) CreateInpageText(tite string, text string) Page {
+	newPage := Page{
+		ParentId: options.NotionDatabaseId,
+		Properties: map[string]interface{}{
+			"Name": map[string]interface{}{
+				"title": []map[string]interface{}{
+					{
+						"text": map[string]string{
+							"content": tite,
+						},
+					},
+				},
+			},
+		},
+		Children: []map[string]interface{}{
+			{
+				"type": "code",
+				"code": map[string]interface{}{
+					"rich_text": []map[string]interface{}{
+						{
+							"type": "text",
+							"text": map[string]string{
+								"content": text,
+							},
+						},
+					},
+					"language": "bash",
+				},
+			},
+		},
+	}
+	return newPage
+}

pkg/providers/notion/notion_types.go

@@ -0,0 +1,7 @@
+package notion
+
+type Page struct {
+	ParentId   string                   `json:"parent_id"`
+	Properties map[string]interface{}   `json:"properties"`
+	Children   []map[string]interface{} `json:"children"`
+}

pkg/providers/providers.go

@@ -10,6 +10,7 @@ import (
 	"github.com/projectdiscovery/notify/pkg/providers/discord"
 	"github.com/projectdiscovery/notify/pkg/providers/googlechat"
 	"github.com/projectdiscovery/notify/pkg/providers/gotify"
+	"github.com/projectdiscovery/notify/pkg/providers/notion"
 	"github.com/projectdiscovery/notify/pkg/providers/pushover"
 	"github.com/projectdiscovery/notify/pkg/providers/slack"
 	"github.com/projectdiscovery/notify/pkg/providers/smtp"
@@ -30,6 +31,7 @@ type ProviderOptions struct {
 	GoogleChat []*googlechat.Options `yaml:"googlechat,omitempty"`
 	Custom     []*custom.Options     `yaml:"custom,omitempty"`
 	Gotify     []*gotify.Options     `yaml:"gotify,omitempty"`
+	Notion     []*notion.Options     `yaml:"notion,omitempty"`
 }
 
 // Provider is an interface implemented by providers
@@ -123,6 +125,15 @@ func New(providerOptions *ProviderOptions, options *types.Options) (*Client, err
 		client.providers = append(client.providers, provider)
 	}
 
+	if providerOptions.Notion != nil && (len(options.Providers) == 0 || sliceutil.Contains(options.Providers, "notion")) {
+
+		provider, err := notion.New(providerOptions.Notion, options.IDs)
+		if err != nil {
+			return nil, errors.Wrap(err, "could not create notion provider client")
+		}
+		client.providers = append(client.providers, provider)
+	}
+
 	return client, nil
 }
 

pkg/utils/httpreq/httpreq.go

@@ -24,6 +24,7 @@ func (c *Client) Get(url string, response interface{}) error {
 	if err != nil {
 		return fmt.Errorf("error creating request: %v", err)
 	}
+	defer res.Body.Close()
 	if err := jsoniter.NewDecoder(res.Body).Decode(&response); err != nil {
 		return fmt.Errorf("error trying to unmarshal the response: %v", err)
 	}