/*
    * Licensed under the GPL License. You may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *   https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
    *
    * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
    * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
    * PURPOSE.
   */
  package psiprobe.controllers.quickcheck;
  
  import jakarta.inject.Inject;
  import jakarta.servlet.http.HttpServletRequest;
  import jakarta.servlet.http.HttpServletResponse;
  
  import java.io.ByteArrayOutputStream;
  import java.io.File;
  import java.io.IOException;
  import java.io.OutputStream;
  import java.nio.charset.StandardCharsets;
  import java.nio.file.Files;
  import java.nio.file.Path;
  import java.util.ArrayList;
  import java.util.List;
  
  import org.apache.catalina.Context;
  import org.springframework.web.servlet.ModelAndView;
  
  import psiprobe.beans.ContainerListenerBean;
  import psiprobe.controllers.AbstractTomcatContainerController;
  import psiprobe.model.ApplicationResource;
  import psiprobe.model.DataSourceInfo;
  import psiprobe.model.TomcatTestReport;
  
  /**
   * "Quick check" base controller.
   */
  public class BaseTomcatAvailabilityController extends AbstractTomcatContainerController {
  
    /** The container listener bean. */
    @Inject
    protected ContainerListenerBean containerListenerBean;
  
    @Override
    public ModelAndView handleRequestInternal(HttpServletRequest request,
        HttpServletResponse response) throws Exception {
  
      final long start = System.currentTimeMillis();
      TomcatTestReport tomcatTestReport = new TomcatTestReport();
  
      // check datasource status
      tomcatTestReport.setDatasourceUsageScore(0);
  
      boolean allContextsAvailable = true;
      if (containerWrapper.getResourceResolver().supportsPrivateResources()) {
        for (Context appContext : containerWrapper.getTomcatContainer().findContexts()) {
          allContextsAvailable =
              allContextsAvailable && containerWrapper.getTomcatContainer().getAvailable(appContext);
  
          List<ApplicationResource> applicationResources = containerWrapper.getResourceResolver()
              .getApplicationResources(appContext, containerWrapper);
  
          for (ApplicationResource appResource : applicationResources) {
            DataSourceInfo dsi = appResource.getDataSourceInfo();
            if (dsi != null && dsi.getBusyScore() > tomcatTestReport.getDatasourceUsageScore()) {
              tomcatTestReport.setContextName(appContext.getName());
              tomcatTestReport.setDatasourceUsageScore(dsi.getBusyScore());
              tomcatTestReport.setDataSourceName(appResource.getName());
            }
          }
        }
  
        tomcatTestReport.setWebappAvailabilityTest(
            allContextsAvailable ? TomcatTestReport.TEST_PASSED : TomcatTestReport.TEST_FAILED);
  
      } else {
        List<ApplicationResource> resources =
            containerWrapper.getResourceResolver().getApplicationResources();
        for (ApplicationResource resource : resources) {
          DataSourceInfo dsi = resource.getDataSourceInfo();
          if (dsi != null && dsi.getBusyScore() > tomcatTestReport.getDatasourceUsageScore()) {
            tomcatTestReport.setDatasourceUsageScore(dsi.getBusyScore());
            tomcatTestReport.setDataSourceName(resource.getName());
          }
        }
      }
      tomcatTestReport.setDatasourceTest(TomcatTestReport.TEST_PASSED);
  
      // try to allocate some memory
      String word = "hello";
      int count = TomcatTestReport.DEFAULT_MEMORY_SIZE / word.length();
  
      try {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        for (; count > 0; count--) {
          bos.write(word.getBytes(StandardCharsets.UTF_8));
        }
        tomcatTestReport.setMemoryTest(TomcatTestReport.TEST_PASSED);
     } catch (IOException e) {
       tomcatTestReport.setMemoryTest(TomcatTestReport.TEST_FAILED);
       logger.trace("", e);
     }
 
     // try to open some files
     int fileCount = tomcatTestReport.getDefaultFileCount();
     List<File> files = new ArrayList<>();
     List<OutputStream> fileStreams = new ArrayList<>();
 
     try {
       for (; fileCount > 0; fileCount--) {
         File file = Path.of(System.getProperty("java.io.tmpdir"), "tctest_" + fileCount).toFile();
         try (OutputStream fos = Files.newOutputStream(file.toPath())) {
           files.add(file);
           fileStreams.add(fos);
           fos.write("this is a test".getBytes(StandardCharsets.UTF_8));
         }
       }
       tomcatTestReport.setFileTest(TomcatTestReport.TEST_PASSED);
     } catch (IOException e) {
       tomcatTestReport.setFileTest(TomcatTestReport.TEST_FAILED);
       logger.trace("", e);
     } finally {
       for (OutputStream fileStream : fileStreams) {
         try {
           fileStream.close();
         } catch (IOException e) {
           logger.trace("", e);
         }
       }
       for (File file : files) {
         Files.delete(file.toPath());
       }
     }
 
     tomcatTestReport.setTestDuration(System.currentTimeMillis() - start);
 
     // TODO JWL 9/19/2026 - implement a real service time test or remove this from the report
     long maxServiceTime = 0;
     tomcatTestReport.setMaxServiceTime(maxServiceTime);
 
     return new ModelAndView(getViewName(), "testReport", tomcatTestReport);
   }
 
 }