import com.cisco.aesop.apphosting.iosapi.*;
import com.cisco.aesop.apphosting.appreapi.*;
import java.io.*;

/**
*Sample IOSTest class to configure the service module and modify the router *interface. Uses the IOS Service API for 
router modification and the CLI Service *API to configure the Service Module. In a production level program,the *System.out.println 
commands should be replaced by logging.
*Be sure to replace the [router-ip] [port] strings in the configureSM method
*with your router IP and the matching port number you set up on the router for *netconf communication.  In the modifyRouter method 
be sure to replace the
*[your description] string with a new description of the interface.  The *modifyRouter method also assumes that your router has an 
gigabitethernet 0/0 *interface.  If this interface is not present on your router please change the
*'int gig0/0' value to an interface your router has.
*/
public class IOSTest{

/**
*Constructor
*/
public IOSTest(){}

/**
* Checks if netconf is already configured on the Service Module
* @return true if netconf configured, otherwise false.
* @throws Exception
*/
public boolean isNetconfConfiguredOnSM() throws Exception{
  final String cmd                = "show run | include netconf";
  boolean result                  = false;
  CommonServiceImpl apiCall       = new CommonServiceImpl();
  AppreMessage msg                = new AppreMessage();

  System.out.println("Entering isNetconfConfiguredOnSM");
  msg.setRequest(cmd);
  if(apiCall.exec(msg) != AppreAPI.FAIL){
    String response = msg.getResponse();
    if(response.length()>0 && response.indexOf("netconf")!=-1){
      result = true;
    }
  }
  else{
    throw new Exception("Command failed " + cmd);
  }
  return result;
}


/**
* Copies the running configuration to the startup configuration
* on the Service Module.
* @throws Exception
*/
public boolean saveConfigurationOnSM() throws Exception{
  final String cmd                = "copy run start";
  boolean result                  = false;
  CommonServiceImpl apiCall       = new CommonServiceImpl();
  AppreMessage msg                = new AppreMessage();

  System.out.println("Entering saveConfigurationOnSM");
  msg.setRequest(cmd);
  if(apiCall.exec(msg) != AppreAPI.FAIL){
    result = true;
  }
  else{
    throw new Exception("Command failed " + cmd);
  }
  return result;
}


/**
* Configures netconf over beep on the service module
* @throws Exception
*/
public boolean configureSM() throws Exception {
  final String cmd          = "netconf beep initiator [router-ip] [port],netconf max-sessions 16";
  boolean result            = false;
  CommonServiceImpl apiCall = new CommonServiceImpl();
  AppreMessage msg          = new AppreMessage();

  System.out.println("Entering configureSM");
  msg.setRequest(cmd);
  if(apiCall.config(msg) != AppreAPI.FAIL){
    result = true;
  }
  else{
    throw new Exception("Command failed " + cmd);
  }
  return result;
}


/**
* Reconfigures the router running configuration.
* @throws Exception
*/
public void modifyRouter() throws Exception {
  final String cmd   = "int gig0/0;description [your description]";

  System.out.println("Entering modifyRouter");

  IosServiceAPI iosapi = IosapiFactory.getIosApi("commonservice");
  IosapiMessage msg = new IosapiMessage();
  msg.setRequest(cmd);
  int status = iosapi.config(msg);
  if(IosServiceAPI.FAIL == status){
    throw new Exception("Command Failed " + cmd + "  Message: " + msg.getResponse());
  }
  else{
    System.out.println("Status returned is:'" + status + "' Command '" + cmd + "'response is:" +  msg.getResponse());
  }
}


public static void main(String[] args){
  boolean isSMConfigured = false;
  try{
    System.out.println("Starting application.");
    IOSTest IOS = new IOSTest();
    if(!IOS.isNetconfConfiguredOnSM()){
      if(IOS.configureSM()){
        IOS.saveConfigurationOnSM();
        isSMConfigured=true;
      }
    }
    else{
      System.out.println("Skipping service module configuration");
      isSMConfigured=true;
    }
    if(isSMConfigured){
      IOS.modifyRouter();
    }
  }
  catch(Exception e){
    System.out.println(e.getMessage());
  }
  finally{
    System.out.println("Application done.");
  }
}