commonMain.com.github.ajalt.clikt.parsers.OptionWithValuesParser.kt Maven / Gradle / Ivy

Go to download
package com.github.ajalt.clikt.parsers

import com.github.ajalt.clikt.core.IncorrectOptionValueCount
import com.github.ajalt.clikt.parameters.options.Option
import com.github.ajalt.clikt.parsers.OptionParser.ParseResult

/** An option that takes one more values */
object OptionWithValuesParser : OptionParser {
    override fun parseLongOpt(option: Option, name: String, argv: List,
                              index: Int, explicitValue: String?): ParseResult {
        require(option.nvalues > 0) {
            "This parser can only be used with a fixed number of arguments. Try the flag parser instead."
        }
        val hasIncludedValue = explicitValue != null
        val consumedCount = if (hasIncludedValue) option.nvalues else option.nvalues + 1
        val endIndex = index + consumedCount - 1

        if (endIndex > argv.lastIndex) {
            throw IncorrectOptionValueCount(option, name)
        }

        val invocation = if (option.nvalues > 1) {
            var args = argv.slice((index + 1)..endIndex)
            if (explicitValue != null) args = listOf(explicitValue) + args
            OptionParser.Invocation(name, args)
        } else {
            OptionParser.Invocation(name, listOf(explicitValue
                    ?: argv[index + 1]))
        }
        return ParseResult(consumedCount, invocation)
    }

    override fun parseShortOpt(option: Option, name: String, argv: List,
                               index: Int, optionIndex: Int): ParseResult {
        val opt = argv[index]
        val hasIncludedValue = optionIndex != opt.lastIndex
        val explicitValue = if (hasIncludedValue) opt.substring(optionIndex + 1) else null
        return parseLongOpt(option, name, argv, index, explicitValue)
    }
}