[Filebeat][CometD] Add CometD input for Salesforce connector

CHANGELOG.asciidoc

@@ -5784,6 +5784,7 @@ https://github.com/elastic/beats/compare/v6.2.3\...v6.3.0[View commits]
 - Add support human friendly size for the UDP input. {pull}6886[6886]
 - Add Syslog input to ingest RFC3164 Events via TCP and UDP {pull}6842[6842]
 - Remove the undefined `username` option from the Redis input and clarify the documentation. {pull}6662[6662]
+- Add CometD input {pull}30089[30089]
 
 *Heartbeat*
 

NOTICE.txt

@@ -5805,6 +5805,37 @@ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 
 
 
+--------------------------------------------------------------------------------
+Dependency : github.com/elastic/bayeux
+Version: v1.0.5
+Licence type (autodetected): MIT
+--------------------------------------------------------------------------------
+
+Contents of probable licence file $GOMODCACHE/github.com/elastic/[email protected]/LICENSE:
+
+MIT License
+
+Copyright (c) 2017 Zander Hill
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+
 --------------------------------------------------------------------------------
 Dependency : github.com/elastic/elastic-agent-client/v7
 Version: v7.0.0-20210727140539-f0905d9377f6

filebeat/docs/filebeat-options.asciidoc

@@ -66,6 +66,7 @@ You can configure {beatname_uc} to use the following inputs:
 * <<{beatname_lc}-input-aws-s3>>
 * <<{beatname_lc}-input-azure-eventhub>>
 * <<{beatname_lc}-input-cloudfoundry>>
+* <<{beatname_lc}-input-cometd>>
 * <<{beatname_lc}-input-container>>
 * <<{beatname_lc}-input-filestream>>
 * <<{beatname_lc}-input-gcp-pubsub>>
@@ -94,6 +95,8 @@ include::../../x-pack/filebeat/docs/inputs/input-azure-eventhub.asciidoc[]
 
 include::../../x-pack/filebeat/docs/inputs/input-cloudfoundry.asciidoc[]
 
+include::../../x-pack/filebeat/docs/inputs/input-cometd.asciidoc[]
+
 include::inputs/input-container.asciidoc[]
 
 include::inputs/input-filestream.asciidoc[]

go.mod

@@ -186,6 +186,7 @@ require (
 )
 
 require (
+	github.com/elastic/bayeux v1.0.5
 	github.com/elastic/elastic-agent-libs v0.1.1
 	github.com/shirou/gopsutil/v3 v3.21.12
 	go.elastic.co/apm/module/apmelasticsearch/v2 v2.0.0

go.sum

@@ -561,6 +561,8 @@ github.com/eclipse/paho.mqtt.golang v1.2.0/go.mod h1:H9keYFcgq3Qr5OUJm/JZI/i6U7j
 github.com/eclipse/paho.mqtt.golang v1.3.5 h1:sWtmgNxYM9P2sP+xEItMozsR3w0cqZFlqnNN1bdl41Y=
 github.com/eclipse/paho.mqtt.golang v1.3.5/go.mod h1:eTzb4gxwwyWpqBUHGQZ4ABAV7+Jgm1PklsYT/eo8Hcc=
 github.com/edsrzf/mmap-go v1.0.0/go.mod h1:YO35OhQPt3KJa3ryjFM5Bs14WD66h8eGKpfaBNrHW5M=
+github.com/elastic/bayeux v1.0.5 h1:UceFq01ipmT3S8DzFK+uVAkbCdiPR0Bqei8qIGmUeY0=
+github.com/elastic/bayeux v1.0.5/go.mod h1:CSI4iP7qeo5MMlkznGvYKftp8M7qqP/3nzmVZoXHY68=
 github.com/elastic/dhcp v0.0.0-20200227161230-57ec251c7eb3 h1:lnDkqiRFKm0rxdljqrj3lotWinO9+jFmeDXIC4gvIQs=
 github.com/elastic/dhcp v0.0.0-20200227161230-57ec251c7eb3/go.mod h1:aPqzac6AYkipvp4hufTyMj5PDIphF3+At8zr7r51xjY=
 github.com/elastic/elastic-agent-client/v7 v7.0.0-20210727140539-f0905d9377f6 h1:nFvXHBjYK3e9+xF0WKDeAKK4aOO51uC28s+L9rBmilo=

testing/environments/docker/cometd/Dockerfile

@@ -0,0 +1,8 @@
+FROM docker.elastic.co/observability/stream:v0.6.1
+
+RUN apt update && \
+    apt -y install curl
+ENV PORT="8080"
+COPY files /files
+HEALTHCHECK --interval=1s --retries=600 CMD curl -X POST http://localhost:8080/token
+CMD [ "http-server", "--addr=:8080", "--config=/files/config.yml" ]

testing/environments/docker/cometd/files/config.yml

@@ -0,0 +1,27 @@
+rules:
+  - path: /token
+    methods: ["POST"]
+    responses:
+      - status_code: 200
+        body: |
+          {"instance_url": "http://cometd:8080", "expires_in": "60", "access_token": "abcd"}
+  - path: /cometd/38.0
+    methods: ["POST"]
+    request_body: '{"channel": "/meta/handshake", "supportedConnectionTypes": ["long-polling"], "version": "1.0"}'
+    responses:
+      - status_code: 200
+        body: |
+          [{"ext":{"replay":true,"payload.format":true},"minimumVersion":"1.0","clientId":"94b112sp7ph1c9s41mycpzik4rkj3","supportedConnectionTypes":["long-polling"],"channel":"/meta/handshake","version":"1.0","successful":true}]
+  - path: /cometd/38.0
+    methods: ["POST"]
+    request_body: '{"channel": "/meta/connect", "connectionType": "long-polling", "clientId": "94b112sp7ph1c9s41mycpzik4rkj3"} '
+    responses:
+      - status_code: 200
+        body: |
+          [{"data": {"payload": {"CountryIso": "IN"}, "event": {"replayId":1234}}, "channel": "/event/LoginEventStream"}]
+  - path: /cometd/38.0
+    methods: ["POST"]
+    responses:
+      - status_code: 200
+        body: |
+          [{"clientId": "94b112sp7ph1c9s41mycpzik4rkj3", "channel": "/meta/subscribe", "subscription": "/event/LoginEventStream", "successful":true}]

x-pack/filebeat/docker-compose.yml

@@ -23,6 +23,7 @@ services:
     image: busybox
     depends_on:
       elasticsearch: { condition: service_healthy }
+      cometd: { condition: service_healthy }
 
   elasticsearch:
     extends:
@@ -33,3 +34,8 @@ services:
       retries: 300
       interval: 1s
 
+  cometd:
+    build: ${ES_BEATS}/testing/environments/docker/cometd
+    hostname: cometd
+    ports:
+      - 8080:8080

x-pack/filebeat/docs/inputs/input-cometd.asciidoc

@@ -0,0 +1,65 @@
+[role="xpack"]
+
+:type: cometd
+
+[id="{beatname_lc}-input-{type}"]
+=== CometD input
+
+++++
+<titleabbrev>CometD</titleabbrev>
+++++
+
+Use the `https://docs.cometd.org/[cometd]` input to stream the real-time events from a https://resources.docs.salesforce.com/sfdc/pdf/api_streaming.pdf[Salesforce generic subscription Push Topic].
+
+This input can, for example, be used to receive Login and Logout events that are generated when users log in or out of the Salesforce instance.
+
+Example configuration:
+
+["source","yaml",subs="attributes"]
+----
+{beatname_lc}.inputs:
+- type: cometd
+  channel_name: /event/MyEventStream
+  auth.oauth2:
+    client.id: my-client-id
+    client.secret: my-client-secret
+    token_url: https://login.salesforce.com/services/oauth2/token
+    user: [email protected]
+    password: my-password
+----
+
+==== Configuration options
+
+The `cometd` input supports the following configuration options.
+
+[float]
+==== `channel_name`
+
+Salesforce generic subscription Push Topic name. Required.
+
+[float]
+==== `auth.oauth2.client.id`
+
+The client ID used as part of the authentication flow. Required.
+
+[float]
+==== `auth.oauth2.client.secret`
+
+The client secret used as part of the authentication flow. Required.
+
+[float]
+==== `auth.oauth2.token_url`
+
+The endpoint that will be used to generate the token during the oauth2 flow. Required.
+
+[float]
+==== `auth.oauth2.user`
+
+The user used as part of the authentication flow. It is required for authentication - grant type password. Required.
+
+[float]
+==== `auth.oauth2.password`
+
+The password used as part of the authentication flow. It is required for authentication - grant type password. Required.
+
+:type!:

x-pack/filebeat/input/cometd/client_mocked.go

@@ -0,0 +1,47 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License;
+// you may not use this file except in compliance with the Elastic License.
+
+package cometd
+
+import (
+	"github.com/elastic/beats/v7/filebeat/channel"
+	"github.com/elastic/beats/v7/libbeat/beat"
+	"github.com/elastic/beats/v7/libbeat/common"
+)
+
+type mockedConnector struct {
+	connectWithError error
+	outlet           channel.Outleter
+}
+
+var _ channel.Connector = new(mockedConnector)
+
+func (m *mockedConnector) Connect(c *common.Config) (channel.Outleter, error) {
+	return m.ConnectWith(c, beat.ClientConfig{})
+}
+
+func (m *mockedConnector) ConnectWith(*common.Config, beat.ClientConfig) (channel.Outleter, error) {
+	if m.connectWithError != nil {
+		return nil, m.connectWithError
+	}
+	return m.outlet, nil
+}
+
+type mockedOutleter struct {
+	onEventHandler func(event beat.Event) bool
+}
+
+var _ channel.Outleter = new(mockedOutleter)
+
+func (m mockedOutleter) Close() error {
+	panic("implement me")
+}
+
+func (m mockedOutleter) Done() <-chan struct{} {
+	panic("implement me")
+}
+
+func (m mockedOutleter) OnEvent(event beat.Event) bool {
+	return m.onEventHandler(event)
+}

x-pack/filebeat/input/cometd/cometd_integration_test.go

@@ -0,0 +1,106 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License;
+// you may not use this file except in compliance with the Elastic License.
+
+//go:build integration
+// +build integration
+
+package cometd
+
+import (
+	"sync"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/elastic/beats/v7/filebeat/channel"
+	"github.com/elastic/beats/v7/filebeat/input"
+	"github.com/elastic/beats/v7/libbeat/beat"
+	"github.com/elastic/beats/v7/libbeat/common"
+	"github.com/elastic/beats/v7/libbeat/logp"
+
+	bay "github.com/elastic/bayeux"
+)
+
+type eventCaptor struct {
+	c         chan struct{}
+	closeOnce sync.Once
+	closed    bool
+	events    chan beat.Event
+}
+
+func newEventCaptor(events chan beat.Event) channel.Outleter {
+	return &eventCaptor{
+		c:      make(chan struct{}),
+		events: events,
+	}
+}
+
+func (ec *eventCaptor) OnEvent(event beat.Event) bool {
+	ec.events <- event
+	return true
+}
+
+func (ec *eventCaptor) Close() error {
+	ec.closeOnce.Do(func() {
+		ec.closed = true
+		close(ec.c)
+	})
+	return nil
+}
+
+func (ec *eventCaptor) Done() <-chan struct{} {
+	return ec.c
+}
+
+func TestInput(t *testing.T) {
+	logp.TestingSetup(logp.WithSelectors("cometd input", "cometd")) //nolint:errcheck // Bad linter! no need to test this.
+
+	// Setup the input config.
+	config := common.MustNewConfigFrom(common.MapStr{
+		"channel_name":              "channel_name1",
+		"auth.oauth2.client.id":     "client.id",
+		"auth.oauth2.client.secret": "client.secret",
+		"auth.oauth2.user":          "user",
+		"auth.oauth2.password":      "password",
+		"auth.oauth2.token_url":     "http://cometd:8080/token",
+	})
+
+	// Route input events through our captor instead of sending through ES.
+	eventsCh := make(chan beat.Event)
+	defer close(eventsCh)
+
+	captor := newEventCaptor(eventsCh)
+	defer captor.Close()
+
+	connector := channel.ConnectorFunc(func(_ *common.Config, _ beat.ClientConfig) (channel.Outleter, error) {
+		return channel.SubOutlet(captor), nil
+	})
+
+	// Mock the context.
+	inputContext := input.Context{
+		Done:     make(chan struct{}),
+		BeatDone: make(chan struct{}),
+	}
+
+	// Setup the input
+	input, err := NewInput(config, connector, inputContext)
+	require.NoError(t, err)
+	require.NotNil(t, input)
+
+	var msg bay.MaybeMsg
+	msg.Msg.Data.Event.ReplayID = 1234
+	msg.Msg.Data.Payload = []byte(`{"CountryIso": "IN"}`)
+	msg.Msg.Channel = "channel_name1"
+
+	// Run the input.
+	input.Run()
+
+	event := <-eventsCh
+
+	val, err := event.GetValue("message")
+	require.NoError(t, err)
+	require.Equal(t, string(msg.Msg.Data.Payload), val)
+
+	input.Stop()
+}

x-pack/filebeat/input/cometd/config.go

@@ -0,0 +1,25 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License;
+// you may not use this file except in compliance with the Elastic License.
+
+package cometd
+
+import "fmt"
+
+type config struct {
+	ChannelName string      `config:"channel_name" validate:"required"`
+	Auth        *authConfig `config:"auth"`
+}
+
+func (c *config) Validate() error {
+	if c.ChannelName == "" {
+		return fmt.Errorf("no channel name was configured or detected")
+	}
+	return nil
+}
+
+func defaultConfig() config {
+	var c config
+	c.ChannelName = "cometd-channel"
+	return c
+}