InputRequestHandler.java

/*
 * Copyright 2000-2009 JetBrains s.r.o.
 *
 * 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 org.napile.idea.thermit.config.execution;

import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;

import org.napile.idea.thermit.ThermitBundle;
import com.intellij.execution.junit2.segments.SegmentReader;
import com.intellij.execution.process.OSProcessHandler;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.ui.ex.MessagesEx;

class InputRequestHandler
{

	private InputRequestHandler()
	{
	}

	public static void processInput(Project project, SegmentReader reader, OSProcessHandler handler) throws IOException
	{
		String input = askUser(reader, project);
		Charset charset = handler.getCharset();
		OutputStream outputStream = handler.getProcessInput();
		sendInput(input, charset, outputStream);
	}

	private static void sendInput(String input, Charset charset, OutputStream outputStream) throws IOException
	{
		byte[] bytes = input.getBytes(charset.name());
		int length = bytes.length;
		byte[] packet = new byte[length + 4];
		System.arraycopy(bytes, 0, packet, 4, length);
		packet[0] = (byte) (length >> 24);
		packet[1] = (byte) (length >> 16);
		packet[2] = (byte) (length >> 8);
		packet[3] = (byte) length;
		outputStream.write(packet);
		outputStream.flush();
	}

	private static String askUser(SegmentReader reader, Project project)
	{
		String prompt = reader.readLimitedString();
		String defaultValue = reader.readLimitedString();
		String[] choices = reader.readStringArray();
		MessagesEx.BaseInputInfo question;
		if(choices.length == 0)
		{
			final MessagesEx.InputInfo info = new MessagesEx.InputInfo(project);
			info.setDefaultValue(defaultValue);
			question = info;
		}
		else
		{
			MessagesEx.ChoiceInfo choiceInfo = new MessagesEx.ChoiceInfo(project);
			choiceInfo.setChoices(choices, defaultValue);
			question = choiceInfo;
		}
		question.setIcon(Messages.getQuestionIcon());
		question.setTitle(ThermitBundle.message("user.inout.request.ant.build.input.dialog.title"));
		question.setMessage(prompt);
		question.setIcon(Messages.getQuestionIcon());
		return question.forceUserInput();
	}
}