/**
    * Copyright 2005-2015 The Kuali Foundation
    *
    * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
    *
   * 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.
   */
  package org.kuali.rice.testtools.selenium;
  
  import org.apache.commons.io.IOUtils;
  import org.openqa.selenium.By;
  
  import java.io.File;
  import java.io.FileOutputStream;
  import java.io.InputStream;
  import java.io.OutputStream;
  import java.net.URL;
  import java.net.URLDecoder;
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.Enumeration;
  import java.util.HashSet;
  import java.util.List;
  import java.util.Set;
  import java.util.jar.JarEntry;
  import java.util.jar.JarFile;
  
  /**
   * Methods for using files and resources in Afts
   *
   * @author Kuali Rice Team (rice.collab@kuali.org)
   */
  public abstract class WebDriverFileResourceAftBase extends WebDriverLegacyITBase {
  
      // values set by default for repeatable testing; left as configurable for load tests
      protected List<File> fileUploadList;
  
      protected String[] getResourceListing(Class clazz, String pathStartsWith) throws Exception {
          String classPath = clazz.getName().replace(".", "/")+".class";
          URL dirUrl = clazz.getClassLoader().getResource(classPath);
  
          if (!"jar".equals(dirUrl.getProtocol())) {
              throw new UnsupportedOperationException("Cannot list files for URL " + dirUrl);
          }
  
          String jarPath = dirUrl.getPath().substring(5, dirUrl.getPath().indexOf("!")); //strip out only the JAR file
          JarFile jar = new JarFile(URLDecoder.decode(jarPath, "UTF-8"));
          Enumeration<JarEntry> entries = jar.entries();
          Set<String> result = new HashSet<String>();
  
          while(entries.hasMoreElements()) {
              String entry = entries.nextElement().getName();
              if (entry.startsWith(pathStartsWith) && !entry.endsWith("/")) { //filter according to the pathStartsWith skipping directories
                  result.add(entry);
              }
          }
  
          return result.toArray(new String[result.size()]);
      }
  
      protected void setUpFiles(String path, String fileExtension) throws Exception {
          fileUploadList = new ArrayList<File>();
  
          File dir = new File(path);
  
          if (dir != null && dir.listFiles().length > 0) {
              Integer i = 1;
  
              for (File file : dir.listFiles()) {
                  if (file.getName().endsWith(fileExtension)) {
                      fileUploadList.add(file);
                  }
  
                  i++;
              }
  
              Collections.sort(fileUploadList);
          } else {
              throw new Exception("----Resources not found----");
          }
      }
  
      protected void setUpResourceDir(String resourceDir) {
          setUpResourceDir(resourceDir, "txt");
      }
  
      protected void setUpResourceDir(String resourceDir, String fileExtension) {
          try {
              setUpFiles("src/test/resources/" + resourceDir, fileExtension);
          } catch (Exception e) {
              System.out.println("Problem loading files from filesystem ( " + e.getMessage() +
                     "). If running from Intellij make sure working directory is " +
                     "rice-framework/krad-sampleapp/web attempt to load as resource.");
 
             try {
                 setUpResourceFiles(resourceDir);
             } catch (Exception e1) {
                 e1.printStackTrace();
                 jiraAwareFail("Problems loading files as resources " + e1.getMessage());
             }
         }
     }
 
     protected void setUpResourceFiles(String resourceDir) throws Exception {
         String[] resources = getResourceListing(getClass(), resourceDir);
         fileUploadList = new ArrayList<File>();
 
         for (String resource : resources) {
             InputStream inputStream = getClass().getResourceAsStream(resource);
             File file = new File(System.getProperty("java.io.tmpdir") + File.separator + resource);
             OutputStream outputStream = new FileOutputStream(file);
             IOUtils.copy(inputStream, outputStream);
             outputStream.close();
             fileUploadList.add(file);
         }
 
         Collections.sort(fileUploadList);
     }
 
     protected void fileIngesterBy(By by) {
         if(fileUploadList!=null && fileUploadList.size()>0) {
             for (File file : fileUploadList) {
                 String path = file.getAbsolutePath().toString();
                 driver.findElement(by).sendKeys(path);
             }
         }
     }
 
     protected void fileIngesterByName(String name) {
         fileIngesterBy(By.name(name));
     }
 
     protected void fileIngesterCollection() throws Exception {
         fileIngesterBy(By.xpath("//div[@data-label='Attached File']/fieldset/div/div/input[@type='file']"));
     }
 }