/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

#ifndef APPLICATIONSUBMISSIONCONTEXT_H_
#define APPLICATIONSUBMISSIONCONTEXT_H_

#include <iostream>
#include "ContainerLaunchContext.h"
#include "YARN_yarn_protos.pb.h"
#include "ApplicationId.h"
#include "Priority.h"
#include "Resource.h"

using std::string; using std::list;
using namespace hadoop::yarn;

namespace libyarn {
/*
message ApplicationSubmissionContextProto {
 optional ApplicationIdProto application_id = 1;
 optional string application_name = 2 [default = "N/A"];
 optional string queue = 3 [default = "default"];
 optional PriorityProto priority = 4;
 optional ContainerLaunchContextProto am_container_spec = 5;
 optional bool cancel_tokens_when_complete = 6 [default = true];
 optional bool unmanaged_am = 7 [default = false];
 optional int32 maxAppAttempts = 8 [default = 0];
 optional ResourceProto resource = 9;
 optional string applicationType = 10 [default = "YARN"];
 }
*/
class ApplicationSubmissionContext {
public:
	ApplicationSubmissionContext();
	ApplicationSubmissionContext(const ApplicationSubmissionContextProto &proto);
	virtual ~ApplicationSubmissionContext();

	ApplicationSubmissionContextProto& getProto();

	void setApplicationId(ApplicationId &appId);
	ApplicationId getApplicationId();

	void setApplicationName(string &applicationName);
	string getApplicationName();

	void setQueue(string &queue);
	string getQueue();

	void setPriority(Priority &priority);
	Priority getPriority();

	void setAMContainerSpec(ContainerLaunchContext &ctx);
	ContainerLaunchContext getAMContainerSpec();

	void setCancelTokensWhenComplete(bool flag);
	bool getCancelTokensWhenComplete();

	void setUnmanagedAM(bool flag);
	bool getUnmanagedAM();

	void setMaxAppAttempts(int32_t max);
	int32_t getMaxAppAttempts();

	void setResource(Resource &resource);
	Resource getResource();

	void setApplicationType(string &type);
	string getApplicationType();

private:
	ApplicationSubmissionContextProto submitCtxProto;
};

} /* namespace libyarn */

#endif /* APPLICATIONSUBMISSIONCONTEXT_H_ */