com.loopfor.zookeeper.cli.command.GetACL.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zookeeper-cli Show documentation

ZooKeeper CLI

The newest version!

/*
 * Copyright 2020 David Edwards
 *
 * 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.loopfor.zookeeper.cli.command

import com.loopfor.scalop._
import com.loopfor.zookeeper._
import com.loopfor.zookeeper.cli._

object GetACL {
  val Usage = """usage: getacl [PATH...]

  Gets the ACL for the node specified by each PATH. PATH may be omitted, in
  which case the current working path is assumed.
"""

  private lazy val opts = OptParser(Seq.empty)

  def command(zk: Zookeeper) = new CommandProcessor {
    implicit val _zk: Zookeeper = zk

    def apply(cmd: String, args: Seq[String], context: Path): Path = {
      val optr = opts <~ args
      val nodes = optr.args.map { path => Node(context.resolve(path)) }
      getACL(nodes)
      context
    }
  }

  def find(zk: Zookeeper, args: Seq[String]) = new FindProcessor {
    val optr = opts <~ args

    def apply(node: Node): Unit = {
      getACL(Seq(node))
    }
  }

  private def getACL(nodes: Seq[Node]): Unit = {
    val count = nodes.size
    nodes.foldLeft(1) { case (i, node) =>
      try {
        val (acl, _) = node.getACL()
        if (count > 1) println(s"${node.path}:")
        acl foreach { println _ }
        if (count > 1 && i < count) println()
      } catch {
        case _: NoNodeException => println(s"${node.path}: no such node")
      }
      i + 1
    }
  }
}