com.netflix.spinnaker.gate.services.SecurityGroupService.groovy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gate-web Show documentation
Spinnaker Gate
There is a newer version: 6.64.0
/*
 * Copyright 2014 Netflix, Inc.
 *
 * 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.
 */

package com.netflix.spinnaker.gate.services

import com.netflix.spinnaker.gate.services.internal.ClouddriverServiceSelector
import groovy.transform.CompileStatic
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Component

@CompileStatic
@Component
class SecurityGroupService {

  @Autowired
  ClouddriverServiceSelector clouddriverServiceSelector

  /**
   * Keyed by account
   */
  Map getAll(String selectorKey) {
    clouddriverServiceSelector.select().securityGroups
  }

  /**
   * Looks for a security group by its id (uses clouddriver's search api)
   *
   * @param id
   * @return
   */
  Map getById(String id, String selectorKey) {
    def result = clouddriverServiceSelector.select().search(id, "securityGroups", null, 10000, 1, [:])[0]
    if (result.results) {
      Map firstResult = ((List)result.results)[0]
      String uriString = firstResult.url
      String vpcId = firstResult.vpcId
      def uri = new URI(uriString)
      def path = uri.path
      def query = uri.query
      def region = query.split('=')[1]
      def pathParts = path.split('/')
      def account = pathParts[2]
      def provider = pathParts[3]
      def sgName = pathParts[-1]
      return getSecurityGroup(account, provider, sgName, region, selectorKey, vpcId)
    } else {
      [:]
    }
  }

  /**
   * @param account account name
   * @param provider provider name (aws, gce, docker)
   * @param region optional. nullable
   */
  Map getForAccountAndProvider(String account, String provider, String selectorKey) {
    clouddriverServiceSelector.select().getSecurityGroups(account, provider)
  }

  List getForAccountAndProviderAndRegion(String account, String provider, String region, String selectorKey) {
    clouddriverServiceSelector.select().getSecurityGroupsForRegion(account, provider, region)
  }

  /**
   * @param account account name
   * @param provider provider name (aws, gce, docker)
   * @param name security group name
   * @param region optional. nullable
   */
  Map getSecurityGroup(String account, String provider, String name, String region, String selectorKey, String vpcId = null) {
    clouddriverServiceSelector.select().getSecurityGroup(account, provider, name, region, vpcId)
  }
}