2 files changed, 397 insertions, 0 deletions
diff --git a/src/main/java/org/openecomp/sparky/viewandinspect/servlet/SearchServlet.java b/src/main/java/org/openecomp/sparky/viewandinspect/servlet/SearchServlet.java
new file mode 100644
index 0000000..c011e9c
--- /dev/null
+++ b/src/main/java/org/openecomp/sparky/viewandinspect/servlet/SearchServlet.java
@@ -0,0 +1,194 @@
+/**
+ * ============LICENSE_START===================================================
+ * SPARKY (AAI UI service)
+ * ============================================================================
+ * Copyright © 2017 AT&T Intellectual Property.
+ * Copyright © 2017 Amdocs
+ * All rights reserved.
+ * ============================================================================
+ * 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.
+ * ============LICENSE_END=====================================================
+ *
+ * ECOMP and OpenECOMP are trademarks
+ * and service marks of AT&T Intellectual Property.
+ */
+
+package org.openecomp.sparky.viewandinspect.servlet;
+
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.openecomp.cl.api.Logger;
+import org.openecomp.cl.eelf.LoggerFactory;
+import org.openecomp.sparky.config.oxm.OxmModelLoader;
+import org.openecomp.sparky.dal.elasticsearch.SearchAdapter;
+import org.openecomp.sparky.dal.rest.OperationResult;
+import org.openecomp.sparky.dal.sas.config.SearchServiceConfig;
+import org.openecomp.sparky.logging.AaiUiMsgs;
+import org.openecomp.sparky.search.VnfSearchService;
+import org.openecomp.sparky.search.config.SuggestionConfig;
+import org.openecomp.sparky.util.NodeUtils;
+import org.openecomp.sparky.viewandinspect.services.SearchServiceWrapper;
+
+import org.openecomp.cl.mdc.MdcContext;
+
+/**
+ * The Class SearchServlet.
+ */
+
+public class SearchServlet extends HttpServlet {
+
+  private static final long serialVersionUID = 1L;
+
+  private static final Logger LOG = LoggerFactory.getInstance().getLogger(SearchServlet.class);
+
+  private SearchServiceWrapper searchWrapper = null;
+
+  private static final String KEY_PAYLOAD = "payload";
+  
+  /**
+   * Instantiates a new search servlet.
+   */
+  public SearchServlet() {
+  }
+
+  /*
+   * (non-Javadoc)
+   * 
+   * @see javax.servlet.http.HttpServlet#doGet(javax.servlet.http.HttpServletRequest,
+   * javax.servlet.http.HttpServletResponse)
+   */
+  @Override
+  public void doGet(HttpServletRequest request, HttpServletResponse response)
+      throws ServletException, IOException {
+    doPost(request, response);
+  }
+
+   public void destroy() {
+    // TODO Auto-generated method stub
+    super.destroy();
+  }
+  
+  public void init() throws ServletException {
+    super.init();
+    searchWrapper = new SearchServiceWrapper();
+  }
+
+  protected Map<String, String> getPayloadParams(JSONObject parameters) {
+    Map<String, String> payloadParams = new HashMap<String, String>();
+    try {
+      JSONObject payload = parameters.getJSONObject(KEY_PAYLOAD);
+      if (payload.length() > 0) {
+        for (String key : JSONObject.getNames(payload)) {
+          payloadParams.put(key, payload.getString(key));
+        }
+      }
+    } catch (JSONException exc) {
+      LOG.error(AaiUiMsgs.ERROR_PARSING_PARAMS, exc);
+    }
+    return payloadParams;
+  }
+
+  /*
+   * (non-Javadoc)
+   * 
+   * @see javax.servlet.http.HttpServlet#doPost(javax.servlet.http.HttpServletRequest,
+   * javax.servlet.http.HttpServletResponse)
+   */
+  @Override
+  public void doPost(HttpServletRequest request, HttpServletResponse response)
+      throws ServletException, IOException {
+    String txnID = request.getHeader("X-TransactionId");
+    if (txnID == null) { 
+      txnID = NodeUtils.getRandomTxnId();
+    }
+
+    String partnerName = request.getHeader("X-FromAppId");
+    if (partnerName == null) {
+      partnerName = "Browser";
+    }
+    MdcContext.initialize(txnID, "AAI_UI", "", partnerName, request.getRemoteAddr());
+    searchWrapper.doPost(request, response);
+  }
+
+  /**
+   * Generate json error response.
+   *
+   * @param message the message
+   * @return the string
+   */
+  /*
+   * This is the manual approach, however we could also create an object container for the error
+   * then use the Jackson ObjectWrite to dump the object to json instead. If it gets any more
+   * complicated we could do that approach so we don't have to manually trip over the JSON
+   * formatting.
+   */
+  protected String generateJsonErrorResponse(String message) {
+    return String.format("{ \"errorMessage\" : %s }", message);
+  }
+
+  /**
+   * Handle search servlet errors.
+   *
+   * @param errorMsg the error msg
+   * @param exc the exc
+   * @param response the response
+   * @throws IOException Signals that an I/O exception has occurred.
+   */
+  public void handleSearchServletErrors(String errorMsg, Exception exc,
+      HttpServletResponse response) throws IOException {
+
+    String errorLogMsg =
+        (exc == null ? errorMsg : errorMsg + ". Error:" + exc.getLocalizedMessage());
+
+    LOG.error(AaiUiMsgs.SEARCH_SERVLET_ERROR, errorLogMsg);
+
+    response.setContentType("application/json");
+    PrintWriter out = response.getWriter();
+    out.println(generateJsonErrorResponse(errorMsg));
+    out.close();
+  }
+
+
+  /**
+   * Sets the servlet response.
+   * 
+   * @param response the response
+   * @param postPayload the post payload
+   *
+   * @throws IOException Signals that an I/O exception has occurred.
+   */
+  private void setServletResponse(HttpServletResponse response, String postPayload)
+      throws IOException {
+
+    if (postPayload != null) {
+      response.setContentType("application/json");
+      PrintWriter out = response.getWriter();
+      out.println(postPayload);
+      out.close();
+    }
+  }
+
+  
+
+  
+}
diff --git a/src/main/java/org/openecomp/sparky/viewandinspect/servlet/VisualizationServlet.java b/src/main/java/org/openecomp/sparky/viewandinspect/servlet/VisualizationServlet.java
new file mode 100644
index 0000000..b64d0e8
--- /dev/null
+++ b/src/main/java/org/openecomp/sparky/viewandinspect/servlet/VisualizationServlet.java
@@ -0,0 +1,203 @@
+/**
+ * ============LICENSE_START===================================================
+ * SPARKY (AAI UI service)
+ * ============================================================================
+ * Copyright © 2017 AT&T Intellectual Property.
+ * Copyright © 2017 Amdocs
+ * All rights reserved.
+ * ============================================================================
+ * 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.
+ * ============LICENSE_END=====================================================
+ *
+ * ECOMP and OpenECOMP are trademarks
+ * and service marks of AT&T Intellectual Property.
+ */
+
+package org.openecomp.sparky.viewandinspect.servlet;
+
+import java.io.IOException;
+import java.io.PrintWriter;
+
+import javax.servlet.FilterConfig;
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.apache.commons.io.IOUtils;
+import org.openecomp.cl.api.Logger;
+import org.openecomp.cl.eelf.LoggerFactory;
+import org.openecomp.sparky.config.oxm.OxmModelLoader;
+import org.openecomp.sparky.dal.rest.OperationResult;
+import org.openecomp.sparky.dal.servlet.ResettableStreamHttpServletRequest;
+import org.openecomp.sparky.logging.AaiUiMsgs;
+import org.openecomp.sparky.util.NodeUtils;
+import org.openecomp.sparky.viewandinspect.entity.QueryRequest;
+import org.openecomp.sparky.viewandinspect.services.VisualizationService;
+
+import org.openecomp.cl.mdc.MdcContext;
+
+/**
+ * A dedicated servlet for handling Front-End Visualization Requests and performing feats of magic
+ * to execute the right model/type/config driven queries to build the D3 visualization output JSON
+ * back to the FE.
+ * 
+ * @author DAVEA
+ *
+ */
+public class VisualizationServlet extends HttpServlet {
+
+  /**
+   * 
+   */
+  private static final long serialVersionUID = 4678831934652478571L;
+  private static final Logger LOG =
+      LoggerFactory.getInstance().getLogger(VisualizationServlet.class);
+  private static final String VISUALIZATION_API_ENDPOINT = "prepareVisualization"; 
+  private final VisualizationService visualizationService;
+  /**
+   * Instantiates a new visualization servlet.
+   *
+   * @throws Exception the exception
+   */
+  public VisualizationServlet() throws Exception {
+    this.visualizationService = new VisualizationService(OxmModelLoader.getInstance());
+  }
+
+  /**
+   * Inits the.
+   *
+   * @param filterConfig the filter config
+   * @throws ServletException the servlet exception
+   */
+  public void init(FilterConfig filterConfig) throws ServletException {
+    LOG.debug(AaiUiMsgs.DEBUG_GENERIC, "init()");
+  }
+
+  /**
+   * Gets the request body.
+   *
+   * @param request the request
+   * @return the request body
+   */
+  private String getRequestBody(HttpServletRequest request) {
+
+    ResettableStreamHttpServletRequest requestWrapper =
+        new ResettableStreamHttpServletRequest(request);
+
+    String body = null;
+    try {
+      body = IOUtils.toString(requestWrapper.getRequestBody());
+    } catch (IOException exc) {
+      LOG.error(AaiUiMsgs.EXCEPTION_CAUGHT, "Trying to get body from request",
+          exc.getLocalizedMessage());
+    }
+
+    return body;
+  }
+
+  /*
+   * (non-Javadoc)
+   * 
+   * @see javax.servlet.http.HttpServlet#doGet(javax.servlet.http.HttpServletRequest,
+   * javax.servlet.http.HttpServletResponse)
+   */
+  @Override
+  protected void doGet(HttpServletRequest request, HttpServletResponse response)
+      throws ServletException, IOException {
+    doPost(request, response);
+  }
+
+  /*
+   * (non-Javadoc)
+   * 
+   * @see javax.servlet.http.HttpServlet#doPost(javax.servlet.http.HttpServletRequest,
+   * javax.servlet.http.HttpServletResponse)
+   */
+  @Override
+  protected void doPost(HttpServletRequest request, HttpServletResponse response)
+      throws ServletException, IOException {
+    String txnID = request.getHeader("X-TransactionId");
+    if (txnID == null) {
+      txnID = NodeUtils.getRandomTxnId();
+    }
+
+    String partnerName = request.getHeader("X-FromAppId");
+    if (partnerName == null) {
+      partnerName = "Browser";
+    }
+
+    MdcContext.initialize(txnID, "AAI-UI", "", partnerName, request.getRemoteAddr());
+
+    String postRequestBody = getRequestBody(request);
+
+    String requestUri = request.getRequestURI();
+    OperationResult operationResult = null;
+
+    /*
+     * For now we only have a single API call but there could be more in the future
+     */
+    if (requestUri.endsWith(VISUALIZATION_API_ENDPOINT)) {
+
+      /*
+       * Work our magic and determine the best way to interrogate AAI to get the stuff we are
+       * interested in. Perhaps it should be an edge-tag-query or perhaps it is a straight up
+       * derived self-link query.
+       */
+
+      /*
+       * Map request body to an interpreted API PoJo object
+       */
+      QueryRequest queryRequest = visualizationService.analyzeQueryRequestBody(postRequestBody);
+
+      if (queryRequest != null) {
+        operationResult = visualizationService.buildVisualizationUsingGenericQuery(queryRequest);
+      } else {
+        LOG.error(AaiUiMsgs.FAILED_TO_ANALYZE,
+            String.format("Failed to analyze post request query body = '%s'", postRequestBody));
+
+        operationResult = new OperationResult();
+        operationResult.setResult(500,
+            String.format("Failed to analyze post request query body = '%s'", postRequestBody));
+
+      }
+
+    } else {
+      // unhandled type
+      LOG.error(AaiUiMsgs.UNKNOWN_SERVER_ERROR, "Unhandled requestUri - " + requestUri);
+      operationResult = new OperationResult();
+      operationResult.setResult(500, "Unknown Server Error: Unhandled requestUri = " + requestUri);
+    }
+
+    PrintWriter out = response.getWriter();
+    response.addHeader("Content-Type", "application/xml");
+
+    response.setStatus(operationResult.getResultCode());
+
+    if (operationResult.getResultCode() == 200) {
+      response.setContentLength(operationResult.getResult().length());
+      out.print(operationResult.getResult());
+      out.print("\n");
+    } else {
+      response.setContentLength(operationResult.getResult().length());
+      out.print(operationResult.getResult());
+      out.print("\n");
+    }
+  }
+  
+  @Override
+  public void destroy() {
+      super.destroy();
+      visualizationService.shutdown();
+  }
+}