com.sksamuel.elastic4s.handlers.security.roles.CreateOrUpdateRoleContentBuilder.scala Maven / Gradle / Ivy

Go to download
package com.sksamuel.elastic4s.handlers.security.roles

import com.sksamuel.elastic4s.json.{XContentBuilder, XContentFactory}
import com.sksamuel.elastic4s.requests.security.roles.{
  ApplicationPrivileges,
  CreateOrUpdateRoleRequest,
  GlobalPrivileges,
  IndexPrivileges
}

object CreateOrUpdateRoleContentBuilder {
  def apply(c: CreateOrUpdateRoleRequest): XContentBuilder = {
    val builder = XContentFactory.jsonBuilder()

    if (c.runAs.nonEmpty) {
      builder.autoarray("run_as", c.runAs)
    }
    if (c.clusterPermissions.nonEmpty) {
      builder.autoarray("cluster", c.clusterPermissions)
    }
    if (c.global.nonEmpty) {
      builder.rawField("global", GlobalPrivilegesContentBuilder(c.global.get))
    }
    if (c.indices.nonEmpty) {
      builder.array("indices", c.indices.map(IndexPrivilegesContentBuilder(_)).toArray)
    }
    if (c.applications.nonEmpty) {
      builder.array("applications", c.applications.map(ApplicationPrivilegesContentBuilder(_)).toArray)
    }

    builder.endObject()
    builder
  }
}

object IndexPrivilegesContentBuilder {
  def apply(i: IndexPrivileges): XContentBuilder = {
    val builder = XContentFactory.jsonBuilder()

    builder.autoarray("names", i.names)
    builder.autoarray("privileges", i.privileges)

    if (
      i.field_security.nonEmpty &&
      (i.field_security.get.grant.nonEmpty || i.field_security.get.except.nonEmpty)
    ) {
      builder.startObject("field_security")
      if (i.field_security.get.grant.nonEmpty) {
        builder.autoarray("grant", i.field_security.get.grant)
      }
      if (i.field_security.get.except.nonEmpty) {
        builder.autoarray("except", i.field_security.get.except)
      }
      builder.endObject()
    }

    if (i.query.nonEmpty) {
      builder.field("query", i.query.get)
    }

    builder.endObject()
    builder
  }
}

object GlobalPrivilegesContentBuilder {
  def apply(g: GlobalPrivileges): XContentBuilder = {
    val builder = XContentFactory.jsonBuilder()

    builder.startObject("application")
    builder.startObject("manage")
    builder.autoarray("applications", g.application.manage.applications)
    builder.endObject()
    builder.endObject()

    builder
  }
}

object ApplicationPrivilegesContentBuilder {
  def apply(a: ApplicationPrivileges): XContentBuilder = {
    val builder = XContentFactory.jsonBuilder()

    builder.field("application", a.application)

    if (a.privileges.nonEmpty) {
      builder.autoarray("privileges", a.privileges)
    }

    if (a.resources.nonEmpty) {
      builder.autoarray("resources", a.resources)
    }

    builder
  }
}