hq/app/aws/package.scala

import model.AwsAccount
import play.api.Logging
import utils.attempt.{Attempt, FailedAttempt, Failure}
import software.amazon.awssdk.regions.Region

import scala.reflect.ClassTag

package object aws extends Logging {
  type AwsClients[A] = List[AwsClient[A]]

  implicit class AwsClientsList[A](clients: AwsClients[A])(implicit classTag: ClassTag[A]) {
    def get(account: AwsAccount, region: Region): Attempt[AwsClient[A]] = {
      val maybeClient = clients.find { client =>
        client.account == account && client.region.id == region.id
      }

      val errorString = s"No ${classTag.runtimeClass.getSimpleName} client exists for ${account.id} and $region"
      if (maybeClient.isEmpty) {
        logger.warn(errorString)
      }

      Attempt.fromOption(maybeClient, FailedAttempt(Failure(
        errorString,
        s"Cannot find AWS client",
        500
      )))
    }

    // this method assumes that there is only one region per account and if this isn't true it will fail
    def get(account: AwsAccount): Attempt[AwsClient[A]] = {
      val clientsForAccount = clients.filter { client =>
        client.account == account
      }
      clientsForAccount match {
        case singleClient :: Nil => Attempt.Right(singleClient)
        case Nil =>
          val errorString = s"No ${classTag.runtimeClass.getSimpleName} client exists for ${account.id}"
          logger.warn(errorString)
          Attempt.Left(FailedAttempt(Failure(
          errorString,
          s"Cannot find AWS client",
          500
        )))
        case multipleClients =>
          val errorString = s"More than one ${classTag.runtimeClass.getSimpleName} client exists for ${account.id} - perhaps you need to use get(account, region)??"
          logger.warn(errorString)
          Attempt.Left(FailedAttempt(Failure(
          errorString,
          s"Multiple AWS clients found when only one was expected",
          500
        )))
      }
    }
  }
}