Merge "Junit for Duplicate Record Exception"
[portal.git] / ecomp-portal-BE-common / src / main / java / org / onap / portalapp / portal / controller / WidgetsCatalogController.java
index 15e1982..a403737 100644 (file)
  *
  * ============LICENSE_END============================================
  *
- * ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ *
  */
 package org.onap.portalapp.portal.controller;
 
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
-import java.io.IOException;
 import java.io.OutputStream;
 import java.util.ArrayList;
 import java.util.List;
@@ -49,6 +48,7 @@ import javax.servlet.ServletContext;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import org.apache.commons.lang.StringUtils;
 import org.onap.portalapp.controller.EPRestrictedBaseController;
 import org.onap.portalapp.portal.domain.EPUser;
 import org.onap.portalapp.portal.domain.MicroserviceParameter;
@@ -59,9 +59,10 @@ import org.onap.portalapp.portal.domain.WidgetServiceHeaders;
 import org.onap.portalapp.portal.ecomp.model.PortalRestResponse;
 import org.onap.portalapp.portal.ecomp.model.PortalRestStatusEnum;
 import org.onap.portalapp.portal.logging.aop.EPAuditLog;
-import org.onap.portalapp.portal.service.ConsulHealthService;
+import org.onap.portalapp.portal.service.WidgetMService;
 import org.onap.portalapp.portal.service.MicroserviceService;
 import org.onap.portalapp.portal.service.WidgetParameterService;
+import org.onap.portalapp.portal.utils.EPCommonSystemProperties;
 import org.onap.portalapp.portal.utils.EcompPortalUtils;
 import org.onap.portalapp.util.EPUserUtils;
 import org.onap.portalsdk.core.logging.logic.EELFLoggerDelegate;
@@ -99,12 +100,18 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
 
        private static final String MS_WIDGET_LOCAL_PORT = "microservices.widget.local.port";
 
+       private static final String UNAUTHORIZED_OR_FORBIDDEN_FOR_A_DISABLED_USER = "Unauthorized or  Forbidden for a disabled user";
+
+    private static final String MS_WIDGET_CATALOG_URL = "/widget/microservices/widgetCatalog/";
+
+    private static final String MS_WIDGET_URL = "/widget/microservices/";
+
        private RestTemplate template = new RestTemplate();
 
        private String whatService = "widgets-service";
 
        @Autowired
-       private ConsulHealthService consulHealthService;
+       private WidgetMService widgetMService;
 
        @Autowired
        private MicroserviceService microserviceService;
@@ -134,9 +141,9 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                try {
                        ResponseEntity<List> ans = template.exchange(
                                        EcompPortalUtils.widgetMsProtocol() + "://"
-                                                       + consulHealthService.getServiceLocation(whatService,
-                                                                       SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                                                       + "/widget/microservices/widgetCatalog/" + loginName,
+                                                       + widgetMService.getServiceLocation(whatService,
+                                                                       SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_LOCAL_PORT))
+                            + MS_WIDGET_CATALOG_URL + loginName,
                                        HttpMethod.GET, new HttpEntity<>(WidgetServiceHeaders.getInstance()), List.class);
                        widgets = ans.getBody();
                } catch (Exception e) {
@@ -154,8 +161,8 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                try {
                        ResponseEntity<List> ans = template.exchange(
                                        EcompPortalUtils.widgetMsProtocol() + "://"
-                                                       + consulHealthService.getServiceLocation(whatService,
-                                                                       SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
+                                                       + widgetMService.getServiceLocation(whatService,
+                                                                       SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_LOCAL_PORT))
                                                        + "/widget/microservices/widgetCatalog",
                                        HttpMethod.GET, new HttpEntity<>(WidgetServiceHeaders.getInstance()), List.class);
                        widgets = ans.getBody();
@@ -173,9 +180,9 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
        public void updateWidgetCatalog(@RequestBody WidgetCatalog newWidgetCatalog, @PathVariable("widgetId") long widgetId) throws Exception {
                template.exchange(
                                EcompPortalUtils.widgetMsProtocol() + "://"
-                                               + consulHealthService.getServiceLocation(whatService,
+                                               + widgetMService.getServiceLocation(whatService,
                                                                SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                                               + "/widget/microservices/widgetCatalog/" + widgetId,
+                        + MS_WIDGET_CATALOG_URL + widgetId,
                                HttpMethod.PUT, new HttpEntity<>(newWidgetCatalog, WidgetServiceHeaders.getInstance()), String.class);
        }
 
@@ -183,9 +190,9 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
        public void deleteOnboardingWidget(@PathVariable("widgetId") long widgetId) throws Exception {
                template.exchange(
                                EcompPortalUtils.widgetMsProtocol() + "://"
-                                               + consulHealthService.getServiceLocation(whatService,
+                                               + widgetMService.getServiceLocation(whatService,
                                                                SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                                               + "/widget/microservices/widgetCatalog/" + widgetId,
+                        + MS_WIDGET_CATALOG_URL + widgetId,
                                HttpMethod.DELETE, new HttpEntity<>(WidgetServiceHeaders.getInstance()), String.class);
        }
 
@@ -197,16 +204,13 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                String fileName;
                String tmpFolderName = "/tmp/";
                String respond = null;
-               FileOutputStream fo = null;
-               try {
-                       mRequest = (MultipartHttpServletRequest) request;
-                       MultipartFile mFile = mRequest.getFile("file");
-                       fileName = mFile.getOriginalFilename();
-                       fo = new FileOutputStream(tmpFolderName + fileName);
-                       fo.write(mFile.getBytes());
-                       // silence sonar scan by calling close here
-                       fo.close();
-                       fo = null;
+        try {
+            mRequest = (MultipartHttpServletRequest) request;
+            MultipartFile mFile = mRequest.getFile("file");
+            fileName = mFile.getOriginalFilename();
+            try (FileOutputStream fo = new FileOutputStream(tmpFolderName + fileName)) {
+                fo.write(mFile.getBytes());
+            }
 
                        HttpHeaders header = new HttpHeaders();
                        header.setContentType(MediaType.MULTIPART_FORM_DATA);
@@ -214,43 +218,40 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                        multipartRequest.add("widget", request.getParameter("newWidget"));
                        respond = template.postForObject(
                                        EcompPortalUtils.widgetMsProtocol() + "://"
-                                                       + consulHealthService.getServiceLocation(whatService,
-                                                                       SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                                                       + "/widget/microservices/widgetCatalog/" + widgetId,
+                                                       + widgetMService.getServiceLocation(whatService,
+                                                                       SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_LOCAL_PORT))
+                            + MS_WIDGET_CATALOG_URL + widgetId,
                                        new HttpEntity<>(multipartRequest, WidgetServiceHeaders.getInstance()), String.class);
                        File f = new File(tmpFolderName + fileName);
                        f.delete();
                } catch (Exception e) {
                        logger.error(EELFLoggerDelegate.errorLogger, "updateWidgetCatalogWithFiles failed", e);
-               } finally {
-                       try {
-                               if (fo != null)
-                                       fo.close();
-                       } catch (IOException e) {
-                               logger.error(EELFLoggerDelegate.errorLogger, "updateWidgetCatalogWithFiles failed 2", e);
-                       }
-               }
+        }
                return respond;
        }
 
        @RequestMapping(value = { "/portalApi/microservices/widgetCatalog" }, method = RequestMethod.POST)
        public String createWidgetCatalog(HttpServletRequest request)
                        throws Exception {
+
+               if (StringUtils.isNotBlank(SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_UPLOAD_FLAG))
+                && "false".equalsIgnoreCase(
+                        SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_UPLOAD_FLAG))) {
+                       return UNAUTHORIZED_OR_FORBIDDEN_FOR_A_DISABLED_USER;
+               }
+
                MultipartHttpServletRequest mRequest;
                MultiValueMap<String, Object> multipartRequest = new LinkedMultiValueMap<>();
                String fileName;
                String tmpFolderName = "/tmp/";
                String respond = null;
-               FileOutputStream fo = null;
                try {
                        mRequest = (MultipartHttpServletRequest) request;
                        MultipartFile mFile = mRequest.getFile("file");
                        fileName = mFile.getOriginalFilename();
-                       fo = new FileOutputStream(tmpFolderName + fileName);
-                       fo.write(mFile.getBytes());
-                       // silence sonar scan by calling close here
-                       fo.close();
-                       fo = null;
+            try (FileOutputStream fo = new FileOutputStream(tmpFolderName + fileName)) {
+                fo.write(mFile.getBytes());
+            }
 
                        HttpHeaders header = new HttpHeaders();
                        header.setContentType(MediaType.MULTIPART_FORM_DATA);
@@ -259,21 +260,14 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
 
                        respond = template.postForObject(
                                        EcompPortalUtils.widgetMsProtocol() + "://"
-                                                       + consulHealthService.getServiceLocation(whatService,
-                                                                       SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
+                                                       + widgetMService.getServiceLocation(whatService,
+                                                                       SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_LOCAL_PORT))
                                                        + "/widget/microservices/widgetCatalog",
                                        new HttpEntity<>(multipartRequest, WidgetServiceHeaders.getInstance()), String.class);
                        File f = new File(tmpFolderName + fileName);
                        f.delete();
                } catch (Exception e) {
                        logger.error(EELFLoggerDelegate.errorLogger, "createWidgetCatalog failed", e);
-               } finally {
-                       try {
-                               if (fo != null)
-                                       fo.close();
-                       } catch (IOException e) {
-                               logger.error(EELFLoggerDelegate.errorLogger, "createWidgetCatalog failed 2", e);
-                       }
                }
                return respond;
        }
@@ -281,27 +275,27 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
        @RequestMapping(value = "/portalApi/microservices/{widgetId}/framework.js", method = RequestMethod.GET)
        public String getWidgetFramework(@PathVariable("widgetId") long widgetId) throws Exception {
                return template.getForObject(EcompPortalUtils.widgetMsProtocol() + "://"
-                               + consulHealthService.getServiceLocation(whatService,
+                               + widgetMService.getServiceLocation(whatService,
                                                SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                               + "/widget/microservices/" + widgetId + "/framework.js", String.class,
+                + MS_WIDGET_URL + widgetId + "/framework.js", String.class,
                                WidgetServiceHeaders.getInstance());
        }
 
        @RequestMapping(value = "/portalApi/microservices/{widgetId}/controller.js", method = RequestMethod.GET)
        public String getWidgetController(@PathVariable("widgetId") long widgetId) throws Exception {
                return template.getForObject(EcompPortalUtils.widgetMsProtocol() + "://"
-                               + consulHealthService.getServiceLocation(whatService,
+                               + widgetMService.getServiceLocation(whatService,
                                                SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                               + "/widget/microservices/" + widgetId + "/controller.js", String.class,
+                + MS_WIDGET_URL + widgetId + "/controller.js", String.class,
                                WidgetServiceHeaders.getInstance());
        }
 
        @RequestMapping(value = "/portalApi/microservices/{widgetId}/style.css", method = RequestMethod.GET)
        public String getWidgetCSS(@PathVariable("widgetId") long widgetId) throws Exception {
                return template.getForObject(EcompPortalUtils.widgetMsProtocol() + "://"
-                               + consulHealthService.getServiceLocation(whatService,
+                               + widgetMService.getServiceLocation(whatService,
                                                SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
-                               + "/widget/microservices/" + widgetId + "/styles.css", String.class,
+                + MS_WIDGET_URL + widgetId + "/styles.css", String.class,
                                WidgetServiceHeaders.getInstance());
        }
 
@@ -313,13 +307,13 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                List<WidgetParameterResult> list = new ArrayList<>();
                Long serviceId = template.exchange(
                                EcompPortalUtils.widgetMsProtocol() + "://"
-                                               + consulHealthService.getServiceLocation(whatService,
-                                                               SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
+                                               + widgetMService.getServiceLocation(whatService,
+                                                               SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_LOCAL_PORT))
                                                + "/widget/microservices/widgetCatalog/parameters/" + widgetId,
                                HttpMethod.GET, new HttpEntity<>(WidgetServiceHeaders.getInstance()), Long.class).getBody();
                if (serviceId == null) {
                        // return ok/sucess and no service parameter for this widget
-                       return new PortalRestResponse<List<WidgetParameterResult>>(PortalRestStatusEnum.WARN,
+            return new PortalRestResponse<>(PortalRestStatusEnum.WARN,
                                        "No service parameters for this widget", list);
                } else {
                        List<MicroserviceParameter> defaultParam = microserviceService.getParametersById(serviceId);
@@ -338,13 +332,12 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                                list.add(userResult);
                        }
                }
-               return new PortalRestResponse<List<WidgetParameterResult>>(PortalRestStatusEnum.OK, "SUCCESS", list);
+        return new PortalRestResponse<>(PortalRestStatusEnum.OK, "SUCCESS", list);
        }
 
        @RequestMapping(value = { "/portalApi/microservices/services/{paramId}" }, method = RequestMethod.GET)
        public List<WidgetCatalogParameter> getUserParameterById(       @PathVariable("paramId") long paramId) {
-               List<WidgetCatalogParameter> list = widgetParameterService.getUserParameterById(paramId);
-               return list;
+        return widgetParameterService.getUserParameterById(paramId);
        }
 
        @RequestMapping(value = { "/portalApi/microservices/services/{paramId}" }, method = RequestMethod.DELETE)
@@ -360,39 +353,45 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
                byte[] byteFile = template
                                .exchange(
                                                EcompPortalUtils.widgetMsProtocol() + "://"
-                                                               + consulHealthService.getServiceLocation(whatService,
-                                                                               SystemProperties.getProperty(MS_WIDGET_LOCAL_PORT))
+                                                               + widgetMService.getServiceLocation(whatService,
+                                                                               SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_LOCAL_PORT))
                                                                + "/widget/microservices/download/" + widgetId,
                                                HttpMethod.GET, new HttpEntity<>(WidgetServiceHeaders.getInstance()), byte[].class)
                                .getBody();
 
                File downloadFile = File.createTempFile("temp", ".zip");
-               FileOutputStream stream = new FileOutputStream(downloadFile.getPath());
-               stream.write(byteFile);
-               stream.close();
-
-               FileInputStream inputStream = new FileInputStream(downloadFile);
-               String mimeType = context.getMimeType(downloadFile.getPath());
-               if (mimeType == null) {
-                       mimeType = "application/octet-stream";
+               try(FileOutputStream stream = new FileOutputStream(downloadFile.getPath())){
+                       stream.write(byteFile);
+               }catch(Exception e)
+               {
+                       logger.error(EELFLoggerDelegate.errorLogger, "doDownload failed", e);
+                       throw e;
                }
 
-               response.setContentType(mimeType);
-               response.setContentLength((int) downloadFile.length());
-               String headerKey = "Content-Disposition";
-               String headerValue = String.format("attachment; filename=\"%s\"", downloadFile.getName());
-               downloadFile.delete();
-               response.setHeader(headerKey, headerValue);
-
-               OutputStream outStream = response.getOutputStream();
-               byte[] buffer = new byte[32 * 1024];
-               int bytesRead;
-               while ((bytesRead = inputStream.read(buffer)) != -1) {
-                       outStream.write(buffer, 0, bytesRead);
-               }
+               try(FileInputStream inputStream = new FileInputStream(downloadFile);
+                       OutputStream outStream = response.getOutputStream()){
+                       String mimeType = context.getMimeType(downloadFile.getPath());
+                       if (mimeType == null) {
+                               mimeType = "application/octet-stream";
+                       }
 
-               inputStream.close();
-               outStream.close();
+                       response.setContentType(mimeType);
+                       response.setContentLength((int) downloadFile.length());
+                       String headerKey = "Content-Disposition";
+                       String headerValue = String.format("attachment; filename=\"%s\"", downloadFile.getName());
+                       downloadFile.delete();
+                       response.setHeader(headerKey, headerValue);
+
+                       byte[] buffer = new byte[32 * 1024];
+                       int bytesRead;
+                       while ((bytesRead = inputStream.read(buffer)) != -1) {
+                               outStream.write(buffer, 0, bytesRead);
+                       }
+               }catch(Exception e)
+               {
+                       logger.error(EELFLoggerDelegate.errorLogger, "doDownload failed", e);
+                       throw e;
+               }
        }
 
        @RequestMapping(value = { "/portalApi/microservices/parameters" }, method = RequestMethod.POST)
@@ -410,8 +409,20 @@ public class WidgetsCatalogController extends EPRestrictedBaseController {
 
                } catch (Exception e) {
                        logger.error(EELFLoggerDelegate.errorLogger, "saveWidgetParameter failed", e);
-                       return new PortalRestResponse<String>(PortalRestStatusEnum.ERROR, "FAILURE", e.getMessage());
+                       return new PortalRestResponse<>(PortalRestStatusEnum.ERROR, "FAILURE", e.getMessage());
+               }
+               return new PortalRestResponse<>(PortalRestStatusEnum.OK, "SUCCESS", "");
+       }
+
+       @RequestMapping(value = { "/portalApi/microservices/uploadFlag" }, method = RequestMethod.GET)
+       public String getUploadFlag() {
+            String uplaodFlag="";
+               try {
+                       uplaodFlag = SystemProperties.getProperty(EPCommonSystemProperties.MS_WIDGET_UPLOAD_FLAG);
+               } catch (Exception e) {
+                       logger.error(EELFLoggerDelegate.errorLogger, "uploadFlag failed", e);
+                       return null;
                }
-               return new PortalRestResponse<String>(PortalRestStatusEnum.OK, "SUCCESS", "");
+               return uplaodFlag;
        }
 }