/*
    * This file is part of dependency-check-core.
    *
    * 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.
   *
   * Copyright (c) 2015 Institute for Defense Analyses. All Rights Reserved.
   */
  package org.owasp.dependencycheck.analyzer;
  
  import org.apache.commons.io.FileUtils;
  import org.apache.commons.io.filefilter.NameFileFilter;
  import org.apache.commons.io.filefilter.SuffixFileFilter;
  import org.owasp.dependencycheck.Engine;
  import org.owasp.dependencycheck.analyzer.exception.AnalysisException;
  import org.owasp.dependencycheck.dependency.Confidence;
  import org.owasp.dependencycheck.dependency.Dependency;
  import org.owasp.dependencycheck.dependency.EvidenceCollection;
  import org.owasp.dependencycheck.utils.FileFilterBuilder;
  import org.owasp.dependencycheck.utils.Settings;
  import org.owasp.dependencycheck.utils.UrlStringUtils;
  
  import java.io.File;
  import java.io.FileFilter;
  import java.io.IOException;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.regex.Matcher;
  import java.util.regex.Pattern;
  
  /**
   * Used to analyze a Python package, and collect information that can be used to determine the associated CPE.
   *
   * @author Dale Visser <dvisser@ida.org>
   */
  public class PythonPackageAnalyzer extends AbstractFileTypeAnalyzer {
  
      /**
       * Used when compiling file scanning regex patterns.
       */
      private static final int REGEX_OPTIONS = Pattern.DOTALL
              | Pattern.CASE_INSENSITIVE;
  
      /**
       * Filename extensions for files to be analyzed.
       */
      private static final String EXTENSIONS = "py";
  
      /**
       * Pattern for matching the module docstring in a source file.
       */
      private static final Pattern MODULE_DOCSTRING = Pattern.compile(
              "^(['\\\"]{3})(.*?)\\1", REGEX_OPTIONS);
  
      /**
       * Matches assignments to version variables in Python source code.
       */
      private static final Pattern VERSION_PATTERN = Pattern.compile(
              "\\b(__)?version(__)? *= *(['\"]+)(\\d+\\.\\d+.*?)\\3",
              REGEX_OPTIONS);
  
      /**
       * Matches assignments to title variables in Python source code.
       */
      private static final Pattern TITLE_PATTERN = compileAssignPattern("title");
  
      /**
       * Matches assignments to summary variables in Python source code.
       */
      private static final Pattern SUMMARY_PATTERN = compileAssignPattern("summary");
  
      /**
       * Matches assignments to URL/URL variables in Python source code.
       */
      private static final Pattern URI_PATTERN = compileAssignPattern("ur[il]");
  
      /**
       * Matches assignments to home page variables in Python source code.
       */
      private static final Pattern HOMEPAGE_PATTERN = compileAssignPattern("home_?page");
  
      /**
       * Matches assignments to author variables in Python source code.
       */
      private static final Pattern AUTHOR_PATTERN = compileAssignPattern("author");
  
      /**
       * Filter that detects files named "__init__.py".
       */
      private static final FileFilter INIT_PY_FILTER = new NameFileFilter("__init__.py");
 
     /**
      * The file filter for python files.
      */
     private static final FileFilter PY_FILTER = new SuffixFileFilter(".py");
 
     /**
      * Returns the name of the Python Package Analyzer.
      *
      * @return the name of the analyzer
      */
     @Override
     public String getName() {
         return "Python Package Analyzer";
     }
 
     /**
      * Tell that we are used for information collection.
      *
      * @return INFORMATION_COLLECTION
      */
     @Override
     public AnalysisPhase getAnalysisPhase() {
         return AnalysisPhase.INFORMATION_COLLECTION;
     }
 
     /**
      * The file filter used to determine which files this analyzer supports.
      */
     private static final FileFilter FILTER = FileFilterBuilder.newInstance().addExtensions(EXTENSIONS).build();
 
     /**
      * Returns the FileFilter
      *
      * @return the FileFilter
      */
     @Override
     protected FileFilter getFileFilter() {
         return FILTER;
     }
 
     /**
      * No-op initializer implementation.
      *
      * @throws Exception never thrown
      */
     @Override
     protected void initializeFileTypeAnalyzer() throws Exception {
         // Nothing to do here.
     }
 
     /**
      * Utility function to create a regex pattern matcher.
      *
      * @param name the value to use when constructing the assignment pattern
      * @return the compiled Pattern
      */
     private static Pattern compileAssignPattern(String name) {
         return Pattern.compile(
                 String.format("\\b(__)?%s(__)?\\b *= *(['\"]+)(.*?)\\3", name),
                 REGEX_OPTIONS);
     }
 
     /**
      * Analyzes python packages and adds evidence to the dependency.
      *
      * @param dependency the dependency being analyzed
      * @param engine     the engine being used to perform the scan
      * @throws AnalysisException thrown if there is an unrecoverable error analyzing the dependency
      */
     @Override
     protected void analyzeFileType(Dependency dependency, Engine engine)
             throws AnalysisException {
         final File file = dependency.getActualFile();
         final File parent = file.getParentFile();
         final String parentName = parent.getName();
         boolean found = false;
         if (INIT_PY_FILTER.accept(file)) {
             for (final File sourceFile : parent.listFiles(PY_FILTER)) {
                 found |= analyzeFileContents(dependency, sourceFile);
             }
         }
         if (found) {
             dependency.setDisplayFileName(parentName + "/__init__.py");
             dependency.getProductEvidence().addEvidence(file.getName(),
                     "PackageName", parentName, Confidence.MEDIUM);
         } else {
             // copy, alter and set in case some other thread is iterating over
             final List<Dependency> dependencies = new ArrayList<Dependency>(
                     engine.getDependencies());
             dependencies.remove(dependency);
             engine.setDependencies(dependencies);
         }
     }
 
     /**
      * This should gather information from leading docstrings, file comments, and assignments to __version__, __title__,
      * __summary__, __uri__, __url__, __home*page__, __author__, and their all caps equivalents.
      *
      * @param dependency the dependency being analyzed
      * @param file       the file name to analyze
      * @return whether evidence was found
      * @throws AnalysisException thrown if there is an unrecoverable error
      */
     private boolean analyzeFileContents(Dependency dependency, File file)
             throws AnalysisException {
         String contents;
         try {
             contents = FileUtils.readFileToString(file).trim();
         } catch (IOException e) {
             throw new AnalysisException(
                     "Problem occurred while reading dependency file.", e);
         }
         boolean found = false;
         if (!contents.isEmpty()) {
             final String source = file.getName();
             found = gatherEvidence(VERSION_PATTERN, contents, source,
                     dependency.getVersionEvidence(), "SourceVersion",
                     Confidence.MEDIUM);
             found |= addSummaryInfo(dependency, SUMMARY_PATTERN, 4, contents,
                     source, "summary");
             if (INIT_PY_FILTER.accept(file)) {
                 found |= addSummaryInfo(dependency, MODULE_DOCSTRING, 2,
                         contents, source, "docstring");
             }
             found |= gatherEvidence(TITLE_PATTERN, contents, source,
                     dependency.getProductEvidence(), "SourceTitle",
                     Confidence.LOW);
             final EvidenceCollection vendorEvidence = dependency
                     .getVendorEvidence();
             found |= gatherEvidence(AUTHOR_PATTERN, contents, source,
                     vendorEvidence, "SourceAuthor", Confidence.MEDIUM);
             found |= gatherHomePageEvidence(URI_PATTERN, vendorEvidence,
                     source, "URL", contents);
             found |= gatherHomePageEvidence(HOMEPAGE_PATTERN,
                     vendorEvidence, source, "HomePage", contents);
         }
         return found;
     }
 
     /**
      * Adds summary information to the dependency
      *
      * @param dependency the dependency being analyzed
      * @param pattern    the pattern used to perform analysis
      * @param group      the group from the pattern that indicates the data to use
      * @param contents   the data being analyzed
      * @param source     the source name to use when recording the evidence
      * @param key        the key name to use when recording the evidence
      * @return true if evidence was collected; otherwise false
      */
     private boolean addSummaryInfo(Dependency dependency, Pattern pattern,
                                    int group, String contents, String source, String key) {
         final Matcher matcher = pattern.matcher(contents);
         final boolean found = matcher.find();
         if (found) {
             JarAnalyzer.addDescription(dependency, matcher.group(group),
                     source, key);
         }
         return found;
     }
 
     /**
      * Collects evidence from the home page URL.
      *
      * @param pattern  the pattern to match
      * @param evidence the evidence collection to add the evidence to
      * @param source   the source of the evidence
      * @param name     the name of the evidence
      * @param contents the home page URL
      * @return true if evidence was collected; otherwise false
      */
     private boolean gatherHomePageEvidence(Pattern pattern,
                                            EvidenceCollection evidence, String source, String name,
                                            String contents) {
         final Matcher matcher = pattern.matcher(contents);
         boolean found = false;
         if (matcher.find()) {
             final String url = matcher.group(4);
             if (UrlStringUtils.isUrl(url)) {
                 found = true;
                 evidence.addEvidence(source, name, url, Confidence.MEDIUM);
             }
         }
         return found;
     }
 
     /**
      * Gather evidence from a Python source file using the given string assignment regex pattern.
      *
      * @param pattern    to scan contents with
      * @param contents   of Python source file
      * @param source     for storing evidence
      * @param evidence   to store evidence in
      * @param name       of evidence
      * @param confidence in evidence
      * @return whether evidence was found
      */
     private boolean gatherEvidence(Pattern pattern, String contents,
                                    String source, EvidenceCollection evidence, String name,
                                    Confidence confidence) {
         final Matcher matcher = pattern.matcher(contents);
         final boolean found = matcher.find();
         if (found) {
             evidence.addEvidence(source, name, matcher.group(4), confidence);
         }
         return found;
     }
 
     @Override
     protected String getAnalyzerEnabledSettingKey() {
         return Settings.KEYS.ANALYZER_PYTHON_PACKAGE_ENABLED;
     }
 }