NFVRoutingController.java

package org.opennaas.gui.nfvrouting.controllers;

import java.util.Locale;
import java.util.logging.Level;
import javax.servlet.http.HttpSession;
import org.apache.log4j.Logger;
import org.opennaas.gui.nfvrouting.beans.insertRoutes;
import org.opennaas.gui.nfvrouting.bos.NFVRoutingBO;
import org.opennaas.gui.nfvrouting.entities.settings.Settings;
import org.opennaas.gui.nfvrouting.entities.Route;
import org.opennaas.gui.nfvrouting.services.rest.RestServiceException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.support.ReloadableResourceBundleMessageSource;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.SessionAttributes;

/**
 * @author Josep Batallé (josep.batalle@i2cat.net)
 */
@Controller
@SessionAttributes("settings")
public class NFVRoutingController {

    private static final Logger LOGGER = Logger.getLogger(NFVRoutingController.class);
    @Autowired
    protected NFVRoutingBO nfvRoutingBO;
    @Autowired
    protected ReloadableResourceBundleMessageSource messageSource;

    /**
     * Redirect to Configure view. Get the Route table of the given IP type.
     *
     * @param type
     * @param model
     * @param locale
     * @param session
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/getRouteTable")
    public String getRouteTable(@RequestParam("type") String type, ModelMap model, Locale locale, HttpSession session) {
        LOGGER.error("Get Route Table ------------------> IPv" + type);
        Settings settings = null;
        if ((Settings) session.getAttribute("settings") != null) {
            model.put("settings", (Settings) session.getAttribute("settings"));
            settings = (Settings) session.getAttribute("settings");
        } else {
            model.addAttribute("errorMsg", "Session time out. Return to <a href='http://nfv.opennaas.i2cat.net/secure/nfvRouting/home'>Home</a>");
//		return "home";
        }
        if (settings == null) {
            settings = new Settings();
        }
        model.addAttribute("settings", settings);

        if ((String) session.getAttribute("topologyName") != null) {
            model.put("topologyName", (String) session.getAttribute("topologyName"));
        }

        int typ;
        if (type.equals("IPv4")) {
            typ = 4;
        } else if (type.equals("IPv6")) {
            typ = 6;
        } else {
            model.addAttribute("errorMsg", "This type of table does not exist.");
            model.addAttribute("json", "'null'");
            return "configRoute";
        }
        try {
            String response = nfvRoutingBO.getRouteTable(typ);
            if (response.equals("OpenNaaS is not started")) {
                model.addAttribute("errorMsg", response);
            }
            LOGGER.info("received json: " + response);
            model.addAttribute("json", response);
        } catch (RestServiceException e) {
            model.addAttribute("errorMsg", e.getMessage());
            model.addAttribute("json", "null");
            return "configRoute";
        }

        return "configRoute";
    }

    /**
     * **
     * Redirect to insert view
     *
     * @param model
     * @param session
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/insertRoute")
    public String insertRoute(ModelMap model, HttpSession session) {
        if ((Settings) session.getAttribute("settings") != null) {
            model.put("settings", (Settings) session.getAttribute("settings"));
        } else {
            model.addAttribute("errorMsg", "Session time out. Return to <a href='http://nfv.opennaas.i2cat.net/secure/nfvRouting/home'>Home</a>");
//                return "home";
        }
        if ((String) session.getAttribute("topologyName") != null) {
            model.put("topologyName", (String) session.getAttribute("topologyName"));
        }

        model.addAttribute(new insertRoutes());
        return "insertRoute";
    }

    /**
     * Redirect to insert view and insert the values received by POST
     *
     * @param route
     * @param result
     * @param model
     * @param session
     * @return
     */
    @RequestMapping(method = RequestMethod.POST, value = "/secure/noc/nfvRouting/insertRoute")
    public String insertRoutePost(insertRoutes route, BindingResult result, ModelMap model, HttpSession session) {
        LOGGER.info("Insert route ------------------> " + route.getListRoutes());
        if ((String) session.getAttribute("topologyName") != null) {
            model.put("topologyName", (String) session.getAttribute("topologyName"));
        }
        try {
            for (Route r : route.getListRoutes()) {
                String response = nfvRoutingBO.insertRoute(r);
                model.addAttribute("json", response);
            }
            model.addAttribute("infoMsg", "Route addded correctly.");
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }

        return "insertRoute";
    }

    /**
     * Redirect to insert view and insert the values received by POST
     *
     * @param route
     * @param result
     * @param model
     * @param session
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/demonstrator")
    public String demonstrator(insertRoutes route, BindingResult result, ModelMap model, HttpSession session) {
        LOGGER.info("Demonstator ------------------> " + route.getListRoutes());
        Settings settings = null;
        if ((Settings) session.getAttribute("settings") != null) {
            model.put("settings", (Settings) session.getAttribute("settings"));
        } else {
            model.addAttribute("errorMsg", "Session time out. REturn to home");
//                return "home";
        }
        if ((String) session.getAttribute("topologyName") != null) {
            model.put("topologyName", (String) session.getAttribute("topologyName"));
        }

        if (settings == null) {
            settings = new Settings();
        }
        model.addAttribute("settings", settings);

        return "demonstrator";
    }

    /**
     * Remove the Route without redirect
     *
     * @param type
     * @param id
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.POST, value = "/secure/noc/nfvRouting/deleteRoute/{id}")
    public @ResponseBody
    String deleteRoute(@RequestParam("type") String type, @PathVariable("id") int id, ModelMap model) {
        LOGGER.debug("Remove Route ------------------> " + id);
        LOGGER.error("REMOVE ROUTE " + id + " " + type);
        String response = "";
        int version;
        if (type.equals("IPv4")) {
            version = 4;
        } else if (type.equals("IPv6")) {
            version = 6;
        } else {
            model.addAttribute("errorMsg", "This type of table does not exist.");
            return "configRoute";
        }
        try {
            response = nfvRoutingBO.deleteRoute(id, version);
            model.addAttribute("json", response);
            model.addAttribute("infoMsg", "Route removed correctly.");
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }
        return response;
    }

    /**
     * Remove all Routes without redirect
     *
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.POST, value = "/secure/noc/nfvRouting/deleteAllRoutes")
    public @ResponseBody
    String deleteAllRoutes(ModelMap model) {
        LOGGER.debug("Remove All Route ------------------> ");
        String response = "";

        try {
            response = nfvRoutingBO.deleteAllRoutes();
            model.addAttribute("json", response);
            model.addAttribute("infoMsg", "Route removed correctly.");
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }
        return response;
    }

    /**
     * Obtain information of a switch. In which controller is connected and the
     * Flow table.
     *
     * @param dpid
     * @param model
     * @return the information of the switch (IP:port)
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/nfvRouting/getInfoSw/{dpid}")
    public @ResponseBody String getAllocatedFlows(@PathVariable("dpid") String dpid, ModelMap model) {
        LOGGER.debug("Get Information about switch ------------------");
        String response = nfvRoutingBO.getSwInfo(dpid);
        return response;
    }

    /**
     * Obtain information of a switch. In which controller is connected and the
     * Flow table.
     *
     * @param ipSrc
     * @param ipDst
     * @param dpid
     * @param inPort
     * @param model
     * @return the information of the switch (IP:port)
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/getRoute/{ipSrc}/{ipDst}/{dpid}/{inPort}")
    public @ResponseBody String getRoute(@PathVariable("ipSrc") String ipSrc, @PathVariable("ipDst") String ipDst,
            @PathVariable("dpid") String dpid, @PathVariable("inPort") String inPort, ModelMap model) {
        LOGGER.debug("Get Route ------------------");
        LOGGER.debug("Requested route: " + ipSrc + " " + ipDst + " " + dpid + " " + inPort + "------------------");
        String response = nfvRoutingBO.getRoute(ipSrc, ipDst, dpid, inPort);
        LOGGER.error("Response: " + response);

        return response.split(":", 2)[1];
    }

    /**
     * Redirect to insert view and insert the values received by POST
     *
     * @param ipSrc
     * @param ipDst
     * @param dpid
     * @param inPort
     * @param dstPort
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/insertRoute/{ipSrc}/{ipDst}/{dpid}/{inPort}/{outPort}")
    public @ResponseBody String insertRoute(@PathVariable("ipSrc") String ipSrc, @PathVariable("ipDst") String ipDst,
            @PathVariable("dpid") String dpid, @PathVariable("inPort") String inPort, @PathVariable("outPort") String dstPort, ModelMap model) {
        LOGGER.info("Insert route ------------------> ");
        String response = "";
        try {
            response = nfvRoutingBO.insertRoute(ipSrc, ipDst, dpid, inPort, dstPort);
            model.addAttribute("json", response);
            model.addAttribute("infoMsg", "Route addded correctly.");
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }

        return response;
    }

    /**
     * Get route paths given ipSrc and IpDest
     *
     * @param ipSrc
     * @param ipDst
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/route/{ipSrc}/{ipDst:.+}")
    public @ResponseBody String getRoute(@PathVariable("ipSrc") String ipSrc, @PathVariable("ipDst") String ipDst, ModelMap model) {
        LOGGER.error("Get ROUTE " + ipSrc + " " + ipDst);
        String response = "";
        try {
            response = nfvRoutingBO.getRoute(ipSrc, ipDst);
            LOGGER.error(response);
            model.addAttribute("json", response);
            model.addAttribute("infoMsg", "Route removed correctly.");
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }
        return response;
    }

    /**
     * Return a json file that contains the routes.
     *
     * @param type
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/routeAll")
    public @ResponseBody
    String getRouteAll(@RequestParam("type") String type, ModelMap model) {
        LOGGER.error("Get Route Table ------------------> IPv" + type);

        int typ;
        if (type.equals("IPv4")) {
            typ = 4;
        } else if (type.equals("IPv6")) {
            typ = 6;
        } else {
            model.addAttribute("errorMsg", "This type of table does not exist.");
            return "configRoute";
        }
        String response = "";
        try {
            response = nfvRoutingBO.getRouteTable(typ);
        } catch (RestServiceException ex) {
            java.util.logging.Logger.getLogger(NFVRoutingController.class.getName()).log(Level.SEVERE, null, ex);
        }

        return response;
    }

    /**
     * Return route mode (static/dynamic)
     *
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/getONRouteMode")
    public @ResponseBody
    String getONRouteMode(ModelMap model) {
        LOGGER.error("Get Route MODE ------------------>");

        String response = "";
        try {
            response = nfvRoutingBO.getONRouteMode();
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }
        return response;
    }

    /**
     * Set Route mode
     *
     * @param mode
     * @param model
     * @return
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/setONRouteMode/{mode}")
    public @ResponseBody
    String setONRouteMode(@PathVariable("mode") String mode, ModelMap model) {
        LOGGER.error("Set Route Mode ------------------> Mode" + mode);
        String response = "";
        try {
            response = nfvRoutingBO.setONRouteMode(mode);
        } catch (Exception e) {
            model.addAttribute("errorMsg", e.getMessage());
        }
        return response;
    }
    
    /**
     * Used in the Demo in order to show the Log
     *
     * @param model
     * @return the log of OpenNaaS
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/getLog")
    public @ResponseBody
    String getLog(ModelMap model) {
        LOGGER.debug("Get log ------------------");
        String response = nfvRoutingBO.getLog();

        return response;
    }

    /**
     * Used in the Demo in order to obtain a route each 5 seconds
     *
     * @param model
     * @return the log of OpenNaaS
     */
    @RequestMapping(method = RequestMethod.GET, value = "/secure/noc/nfvRouting/getStreamInfo")
    public @ResponseBody
    String getStreamRoute(ModelMap model) {
        LOGGER.debug("Get stream route ------------------");
        String response = nfvRoutingBO.getStream();

        return response;
    }
}