com.iheart.thomas.cli.EligibilityControlCommand.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of thomas-cli_2.12 Show documentation
thomas-cli
The newest version!
package com.iheart.thomas
package cli

import cats.effect.Async
import cats.implicits._
import com.iheart.thomas.abtest.json.play.Formats._
import com.iheart.thomas.abtest.model.UserMetaCriterion
import com.iheart.thomas.cli.OptsSyntax._
import com.iheart.thomas.cli.SharedOpts._
import com.monovore.decline._
import _root_.play.api.libs.json.Json.{prettyPrint, toJson}

class EligibilityControlCommand[F[_]](implicit F: Async[F]) {

  val showCommand = Command("show", "show current user metas filters") {
    (tidOrFnOps, AbtestHttpClientOpts.opts[F]).mapN { (tidOrF, client) =>
      client.use { c =>
        c.getUserMetaCriteria(tidOrF).map { r =>
          s"""
                             |
                             |User Meta Criteria for ${show(tidOrF)}:
                             | ${prettyPrint(toJson(r))}
             """.stripMargin
        }
      }
    }
  }

  val criteriaOpts = Opts
    .option[String](
      "criteria",
      "A Json object representing the user meta criteria, see https://github.com/iheartradio/thomas/blob/master/docs/src/main/tut/FAQ.md#how-to-manage-user-eligibility"
    )
    .parseJson[UserMetaCriterion.And] orElse criteriaFileOpts

  lazy val criteriaFileOpts = Opts
    .option[String](
      "criteriaFile",
      "the location of the file containing the json criteria"
    )
    .readJsonFile[UserMetaCriterion.And]

  val newRevOpts = Opts
    .flag("new", "create a new revision if the current one has already started")
    .orFalse

  val updateCommand = Command("update", "update user meta criteria") {
    (tidOrFnOps, criteriaOpts, newRevOpts, AbtestHttpClientOpts.opts[F]).mapN {
      (tidOrFeature, criteria, nt, clientR) =>
        clientR
          .use { _.updateUserMetaCriteria(tidOrFeature, Some(criteria), nt) }
          .as(
            s"User meta criteria for ${show(tidOrFeature)} is updated"
          )
    }
  }

  val removeCommand = Command("remove", "user meta criteria") {
    (tidOrFnOps, newRevOpts, AbtestHttpClientOpts.opts[F]).mapN {
      (tidOrFeature, nt, clientR) =>
        clientR
          .use { _.updateUserMetaCriteria(tidOrFeature, None, nt) }
          .as(
            s"User meta criteria for ${show(tidOrFeature)} is removed"
          )

    }
  }

  val userMetaCriteriaCommand: Command[F[String]] = Command(
    "userMetaCriteria",
    "managing eligibility control through user meta"
  ) {
    Opts.subcommand(showCommand) orElse Opts.subcommand(updateCommand) orElse Opts
      .subcommand(removeCommand)
  }

}