mongo4cats.models.client.MongoConnection.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2020 Kirill5k
 *
 * 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 mongo4cats.models.client

sealed abstract class MongoConnectionType(val `type`: String)

object MongoConnectionType {
  case object Classic extends MongoConnectionType("mongodb")

  case object Srv extends MongoConnectionType("mongodb+srv")
}

final case class MongoCredential(username: String, password: String)

/** A data model representation of a MongoDB Connection String
  *
  * @param host
  *   The host that serves MongoDB
  * @param port
  *   Port where the MongoDB is served in the host
  * @param credential
  *   Optional credentials that maybe used to establish authentication with the MongoDB
  * @param connectionType
  *   For switching between different MongoDB connection types, see [[MongoConnectionType]] for possible options
  */
sealed abstract class MongoConnection(
    val host: String,
    val port: Option[Int],
    val credential: Option[MongoCredential],
    val connectionType: MongoConnectionType
) {
  override def toString: String = {
    val credentialString = credential.fold("")(cred => s"${cred.username}:${cred.password}@")
    val portString       = port.fold("")(p => s":$p")
    s"${connectionType.`type`}://$credentialString$host$portString"
  }
}

object MongoConnection {

  def classic(
      host: String,
      port: Int = 27017,
      credential: Option[MongoCredential] = None
  ): MongoConnection = apply(host, Some(port), credential, MongoConnectionType.Classic)

  def srv(
      host: String,
      credential: Option[MongoCredential] = None
  ): MongoConnection = apply(host, None, credential, MongoConnectionType.Srv)

  def apply(
      host: String,
      port: Option[Int] = Some(27017),
      credential: Option[MongoCredential] = None,
      connectionType: MongoConnectionType = MongoConnectionType.Classic
  ): MongoConnection =
    new MongoConnection(host = host, port = port, credential = credential, connectionType = connectionType) {}
}