/*
 * Copyright 2013 the original author or authors.
 *
 * 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.springframework.yarn.config.annotation.configurers;

import java.util.Properties;

import org.springframework.data.hadoop.config.common.annotation.AnnotationConfigurerAdapter;
import org.springframework.yarn.YarnSystemConstants;
import org.springframework.yarn.am.YarnAppmaster;
import org.springframework.yarn.config.annotation.builders.YarnAppmasterBuilder;
import org.springframework.yarn.config.annotation.builders.YarnAppmasterConfigurer;
import org.springframework.yarn.container.CommandLineContainerRunner;
import org.springframework.yarn.launch.AbstractCommandLineRunner;
import org.springframework.yarn.launch.LaunchCommandsFactoryBean;

/**
 *
 *
 * @author Janne Valkealahti
 *
 */
public class DefaultMasterContainerRunnerConfigurer
		extends AnnotationConfigurerAdapter<YarnAppmaster, YarnAppmasterConfigurer, YarnAppmasterBuilder>
		implements MasterContainerRunnerConfigurer {

	private Class<?> contextClass;
	private String contextFile = "container-context.xml";
	private String stdout = "<LOG_DIR>/Container.stdout";
	private String stderr = "<LOG_DIR>/Container.stderr";
	private String beanName = YarnSystemConstants.DEFAULT_ID_CONTAINER;
	private Class<? extends AbstractCommandLineRunner<?>> runnerClass = CommandLineContainerRunner.class;

	private Properties arguments = new Properties();

	@Override
	public void configure(YarnAppmasterBuilder builder) throws Exception {
		LaunchCommandsFactoryBean fb = new LaunchCommandsFactoryBean();
		fb.setRunner(runnerClass);
		fb.setContextFile(contextClass != null ?  contextClass.getCanonicalName() : contextFile);
		fb.setBeanName(beanName);

		fb.setArguments(arguments);

		fb.setStdout(stdout);
		fb.setStderr(stderr);
		fb.afterPropertiesSet();
		builder.containerCommands(fb.getObject());
	}

	@Override
	public void init(YarnAppmasterBuilder builder) throws Exception {
		super.init(builder);
	}

	@Override
	public MasterContainerRunnerConfigurer contextClass(Class<?> contextClass) {
		this.contextClass = contextClass;
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer contextFile(String contextFile) {
		this.contextFile = contextFile;
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer stdout(String stdout) {
		this.stdout = stdout;
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer stderr(String stderr) {
		this.stderr = stderr;
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer beanName(String beanName) {
		this.beanName = beanName;
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer runnerClass(Class<? extends AbstractCommandLineRunner<?>> runnerClazz) {
		this.runnerClass = runnerClazz;
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer arguments(Properties arguments) {
		this.arguments.putAll(arguments);
		return this;
	}

	@Override
	public MasterContainerRunnerConfigurer argument(String key, String value) {
		this.arguments.put(key, value);
		return this;
	}

}