com.github.jy2.commandline.picocli.param.completion.ParameterValueCompletionCandidates Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jy2-tool Show documentation

There is a newer version: 0.0.39

package com.github.jy2.commandline.picocli.param.completion;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;

import com.github.jy2.commandline.picocli.Main;

import picocli.AutoComplete;
import picocli.CommandLine.Model.CommandSpec;
import picocli.CommandLine.Model.OptionSpec;
import picocli.CommandLine.Model.PositionalParamSpec;

public class ParameterValueCompletionCandidates implements Iterable {

	@Override
	public Iterator iterator() {
		if (AutoComplete.argIndex != getParameterValueIndex()) {
			return new ArrayList().iterator();
		}

		ArrayList list = new ArrayList<>();
		String name = getParameterName();
		if (name != null) {
			Object param = null;
			try {
				param = Main.di.getParameterClient().getParameter(name);
			} catch (IOException e) {
				e.printStackTrace();
			}
			if (param == null) {
				return list.iterator();
			}
			Object value = null;
			try {
				value = Main.di.getParameterClient().getParameter(name);
			} catch (IOException e) {
				e.printStackTrace();
			}
			if (value != null) {
				list.add(value.toString());
			}
			// disabled for now
//			if ("true".equalsIgnoreCase(value)) {
//				list.add("false");
//			} else if ("false".equalsIgnoreCase(value)) {
//				list.add("true");
//			}
		}
		list.sort(String::compareToIgnoreCase);
		return list.iterator();
	}

	private String getParameterName() {
		if (AutoComplete.tentativeMatch == null) {
			return null;
		}
		for (Object obj : AutoComplete.tentativeMatch) {
			if (obj instanceof CommandSpec) { // subcommand
			} else if (obj instanceof OptionSpec) { // option
			} else if (obj instanceof PositionalParamSpec) { // positional
				PositionalParamSpec pos = (PositionalParamSpec) obj;
				if (pos.index().min == 0) {
					String value = pos.stringValues().get(0);
					return value;
				}
			}
		}
		return null;
	}

	private int getParameterValueIndex() {
		if (AutoComplete.tentativeMatch == null) {
			return -1;
		}
		int i = 0;
		for (Object obj : AutoComplete.tentativeMatch) {
			if (obj instanceof CommandSpec) { // subcommand
			} else if (obj instanceof OptionSpec) { // option
			} else if (obj instanceof PositionalParamSpec) { // positional
				PositionalParamSpec pos = (PositionalParamSpec) obj;
				if (pos.index().min == 1) {
					return i;
				}
			}
			i++;
		}
		return -1;
	}
}