# Copyright 2019 Capital One Services, LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from c7n.actions import BaseNotify
from c7n import utils
from c7n.resolver import ValuesFrom
from c7n_gcp.provider import resources as gcp_resources
[docs]class Notify(BaseNotify):
"""
:example:
.. code-block:: yaml
policies:
- name: bad-instance-get
resource: gcp.instance
filters:
- Name: bad-instance
actions:
- type: notify
to:
- email@address
# which template for the email should we use
template: policy-template
transport:
type: pubsub
topic: projects/yourproject/topics/yourtopic
"""
batch_size = 1000
schema = {
'type': 'object',
'anyOf': [
{'required': ['type', 'transport', 'to']},
{'required': ['type', 'transport', 'to_from']}],
'properties': {
'type': {'enum': ['notify']},
'to': {'type': 'array', 'items': {'type': 'string'}},
'owner_absent_contact': {'type': 'array', 'items': {'type': 'string'}},
'to_from': ValuesFrom.schema,
'cc': {'type': 'array', 'items': {'type': 'string'}},
'cc_from': ValuesFrom.schema,
'cc_manager': {'type': 'boolean'},
'from': {'type': 'string'},
'subject': {'type': 'string'},
'template': {'type': 'string'},
'transport': {
'oneOf': [
{'type': 'object',
'required': ['type', 'topic'],
'properties': {
'topic': {'type': 'string'},
'type': {'enum': ['pubsub']},
}}],
},
}
}
[docs] @staticmethod
def register_notify_action(registry, _):
for resource in registry.keys():
klass = registry.get(resource)
klass.action_registry.register('notify', Notify)
[docs] def process(self, resources, event=None):
session = utils.local_session(self.manager.session_factory)
client = session.client('pubsub', 'v1', 'projects.topics')
project = session.get_default_project()
message = {
'event': event,
'account_id': project,
'account': project,
'region': 'all',
'policy': self.manager.data
}
message['action'] = self.expand_variables(message)
for batch in utils.chunks(resources, self.batch_size):
message['resources'] = batch
self.publish_message(message, client)
# Methods to handle GCP Pub Sub topic publishing
[docs] def publish_message(self, message, client):
"""Publish message to a GCP pub/sub topic
"""
return client.execute_command('publish', {
'topic': self.data['transport']['topic'],
'body': {
'messages': {
'data': self.pack(message)
}
}
})
gcp_resources.subscribe(
gcp_resources.EVENT_FINAL, Notify.register_notify_action)