gerrit.onap Code Review - dcaegen2/collectors/hv-ves.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 2018 NOKIA
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.dcae.collectors.veshv.utils.commandline
  21
  22 import org.apache.commons.cli.CommandLine
  23 import org.apache.commons.cli.CommandLineParser
  24 import org.apache.commons.cli.Options
  25 import java.io.File
  26 import java.nio.file.Paths
  27
  28 abstract class ArgBasedConfiguration<T>(val parser: CommandLineParser) {
  29     abstract val cmdLineOptionsList: List<CommandLineOption>
  30
  31     fun parse(args: Array<out String>): T {
  32         val commandLineOptions = cmdLineOptionsList.map { it.option }.fold(Options(), Options::addOption)
  33         try {
  34             val cmdLine = parser.parse(commandLineOptions, args)
  35             return getConfiguration(cmdLine)
  36         } catch (ex: Exception) {
  37             throw WrongArgumentException(ex, commandLineOptions)
  38         }
  39     }
  40
  41     protected abstract fun getConfiguration(cmdLine: CommandLine): T
  42
  43     protected fun CommandLine.intValue(cmdLineOpt: CommandLineOption): Int =
  44             getOptionValue(cmdLineOpt.option.opt).toInt()
  45
  46     protected fun CommandLine.intValue(cmdLineOpt: CommandLineOption, default: Int): Int =
  47             getOptionValue(cmdLineOpt.option.opt)?.toInt() ?: default
  48
  49     protected fun CommandLine.longValue(cmdLineOpt: CommandLineOption): Long =
  50             getOptionValue(cmdLineOpt.option.opt).toLong()
  51
  52     protected fun CommandLine.longValue(cmdLineOpt: CommandLineOption, default: Long): Long =
  53             getOptionValue(cmdLineOpt.option.opt)?.toLong() ?: default
  54
  55     protected fun CommandLine.stringValue(cmdLineOpt: CommandLineOption): String =
  56             getOptionValue(cmdLineOpt.option.opt)
  57
  58     protected fun CommandLine.stringValue(cmdLineOpt: CommandLineOption, default: String): String =
  59             getOptionValue(cmdLineOpt.option.opt) ?: default
  60
  61     protected fun stringPathToPath(path: String) = Paths.get(File(path).toURI())
  62 }