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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 package org.onap.dcae.collectors.veshv.utils.commandline
22 import arrow.core.Option
24 import arrow.core.getOrElse
25 import arrow.core.recoverWith
26 import org.apache.commons.cli.CommandLine
27 import org.apache.commons.cli.CommandLineParser
28 import org.apache.commons.cli.Options
30 import java.nio.file.Path
31 import java.nio.file.Paths
33 abstract class ArgBasedConfiguration<T>(val parser: CommandLineParser) {
34 abstract val cmdLineOptionsList: List<CommandLineOption>
36 fun parse(args: Array<out String>): Try<T> {
37 val commandLineOptions = cmdLineOptionsList.map { it.option }.fold(Options(), Options::addOption)
39 parser.parse(commandLineOptions, args)
41 Try.raise<CommandLine>(WrongArgumentException(ex, commandLineOptions))
42 }.map (this::getConfiguration)
45 protected abstract fun getConfiguration(cmdLine: CommandLine): T
47 protected fun CommandLine.intValue(cmdLineOpt: CommandLineOption, default: Int): Int =
48 intValue(cmdLineOpt).getOrElse { default }
50 protected fun CommandLine.longValue(cmdLineOpt: CommandLineOption, default: Long): Long =
51 longValue(cmdLineOpt).getOrElse { default }
53 protected fun CommandLine.stringValue(cmdLineOpt: CommandLineOption): Option<String> =
54 optionValue(cmdLineOpt)
56 protected fun CommandLine.stringValue(cmdLineOpt: CommandLineOption, default: String): String =
57 optionValue(cmdLineOpt).getOrElse { default }
59 protected fun CommandLine.hasOption(cmdLineOpt: CommandLineOption): Boolean =
60 this.hasOption(cmdLineOpt.option.opt)
62 protected fun stringPathToPath(path: String): Path = Paths.get(File(path).toURI())
64 private fun CommandLine.optionValue(cmdLineOpt: CommandLineOption): Option<String> =
65 Option.fromNullable(getOptionValue(cmdLineOpt.option.opt))
67 private fun CommandLine.intValue(cmdLineOpt: CommandLineOption): Option<Int> =
68 optionValue(cmdLineOpt).map(String::toInt)
70 private fun CommandLine.longValue(cmdLineOpt: CommandLineOption): Option<Long> =
71 optionValue(cmdLineOpt).map(String::toLong)