feat: Added sesReceiptRuleSet service configurations

Author: Marco Franceschi

src/services/sesReceiptRuleSet/data.ts

@@ -0,0 +1,116 @@
+import SES, {
+  ListReceiptRuleSetsResponse,
+  ReceiptRuleSetMetadata,
+  ReceiptRule,
+} from 'aws-sdk/clients/ses'
+import { AWSError } from 'aws-sdk/lib/error'
+import { Config } from 'aws-sdk/lib/config'
+
+import CloudGraph from '@cloudgraph/sdk'
+import groupBy from 'lodash/groupBy'
+
+import awsLoggerText from '../../properties/logger'
+import { initTestEndpoint } from '../../utils'
+import AwsErrorLog from '../../utils/errorLog'
+
+const lt = { ...awsLoggerText }
+const { logger } = CloudGraph
+const serviceName = 'SES Receipt Rule Set'
+const errorLog = new AwsErrorLog(serviceName)
+const endpoint = initTestEndpoint(serviceName)
+
+/**
+ * SES Receipt Rule Set
+ */
+
+export interface RawAwsSesReceiptRuleSet {
+  ReceiptRuleSet: ReceiptRuleSetMetadata
+  Rules: ReceiptRule[]
+  region: string
+}
+
+
+const getReceiptRuleSets = async (ses: SES): Promise<ReceiptRuleSetMetadata[]> =>
+  new Promise(resolve => {
+    try {
+      ses.listReceiptRuleSets(
+        (err: AWSError, data: ListReceiptRuleSetsResponse) => {
+          if (err) {
+            errorLog.generateAwsErrorLog({
+              functionName: 'sesReceiptRuleSet:listReceiptRuleSets',
+              err,
+            })
+            return resolve([])
+          }
+          const { RuleSets = [] } = data || {}
+          resolve(RuleSets)
+        }
+      )
+    } catch (error) {
+      resolve([])
+    }
+  })
+
+const getReceiptRules = async (ses: SES, ruleSets: ReceiptRuleSetMetadata[]): Promise<{ ruleSet: ReceiptRuleSetMetadata, rules: ReceiptRule[] }[]> => {
+  const receiptRules: { ruleSet: ReceiptRuleSetMetadata, rules: ReceiptRule[] }[] = []
+  for (const ruleSet of ruleSets) {
+    try {
+      const response = await ses.describeReceiptRuleSet({
+        RuleSetName: ruleSet.Name,
+      }).promise()
+      const { Metadata, Rules = [] } = response
+
+      receiptRules.push({
+        ruleSet: Metadata,
+        rules: Rules
+      })
+    } catch (error) {
+      errorLog.generateAwsErrorLog({
+        functionName: 'sesReceiptRuleSet:describeReceiptRuleSet',
+        err: error,
+      })
+    }
+  }
+  return receiptRules
+}
+
+
+export default async ({
+  regions,
+  config,
+}: {
+  regions: string
+  config: Config
+}): Promise<{ [property: string]: RawAwsSesReceiptRuleSet[] }> =>
+  new Promise(async resolve => {
+    const sesData: RawAwsSesReceiptRuleSet[] = []
+    const regionPromises = []
+
+    regions.split(',').map(region => {
+      const regionPromise = new Promise<void>(async resolveRegion => {
+        const ses = new SES({ ...config, region, endpoint })
+
+        const ruleSets = await getReceiptRuleSets(ses)
+        const receiptRuleSets = await getReceiptRules(ses, ruleSets)
+
+        for (const { ruleSet, rules } of receiptRuleSets) {
+          sesData.push({
+            ReceiptRuleSet: ruleSet,
+            Rules: rules,
+            region
+          })
+        }
+
+        resolveRegion()
+
+
+      })
+      regionPromises.push(regionPromise)
+    })
+
+
+    await Promise.all(regionPromises)
+    errorLog.reset()
+
+    resolve(groupBy(sesData, 'region'))
+  })

src/services/sesReceiptRuleSet/format.ts

@@ -0,0 +1,46 @@
+import { RawAwsSesReceiptRuleSet } from './data'
+import { AwsSesReceiptRuleSet } from '../../types/generated'
+import { generateUniqueId } from '@cloudgraph/sdk'
+
+/**
+ * SES Receipt Rule Set
+ */
+
+export default ({
+  service,
+  account,
+  region,
+}: {
+  service: RawAwsSesReceiptRuleSet
+  account: string
+  region: string
+}): AwsSesReceiptRuleSet => {
+  const {
+    ReceiptRuleSet,
+    Rules = []
+  } = service
+
+  const rules = Rules.map(r => ({
+    id: generateUniqueId({
+      ...r,
+      account,
+      region
+    }),
+    name: r.Name,
+    enabled: r.Enabled,
+    tlsPolicy: r.TlsPolicy,
+    scanEnabled: r.ScanEnabled
+  }))
+
+  return {
+    id: generateUniqueId({
+      ...service,
+      account,
+      region
+    }),
+    accountId: account,
+    region,
+    name: ReceiptRuleSet?.Name,
+    rules
+  }
+}

src/services/sesReceiptRuleSet/index.ts

@@ -0,0 +1,13 @@
+import { Service } from '@cloudgraph/sdk'
+import BaseService from '../base'
+import format from './format'
+import getData from './data'
+import mutation from './mutation'
+
+export default class SESReceiptRuleSet extends BaseService implements Service {
+  format = format.bind(this)
+
+  getData = getData.bind(this)
+
+  mutation = mutation
+}

src/services/sesReceiptRuleSet/mutation.ts

@@ -0,0 +1,5 @@
+export default `mutation($input: [AddawsSesReceiptRuleSetInput!]!) {
+  addawsSesReceiptRuleSet(input: $input, upsert: true) {
+    numUids
+  }
+}`

src/services/sesReceiptRuleSet/schema.graphql

@@ -0,0 +1,19 @@
+type awsSesReceiptRuleSetRule
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  ) {
+  id: String! @id
+  name: String @search(by: [hash, regexp])
+  enabled: Boolean @search
+  tlsPolicy: String @search(by: [hash, regexp])
+  scanEnabled: Boolean @search
+}
+
+
+type awsSesReceiptRuleSet implements awsOptionalService @key(fields: "id") {
+  accountId: String @search(by: [hash, regexp])
+  name: String @search(by: [hash, regexp])
+  rules: [awsSesReceiptRuleSetRule]
+}