Patch for xlsx processor

unknown [2011-08-07 22:Aug:th]
Patch for xlsx processor
Filename
configs/config.php
lib/processors/xlsx_processor.php
tests/test_files/test.xlsx
tests/xlsx_processor_test.php
diff --git a/configs/config.php b/configs/config.php
index f6d7c4a17..4a71cfa13 100755
--- a/configs/config.php
+++ b/configs/config.php
@@ -259,7 +259,8 @@ $INDEXED_FILE_TYPES =
             "tab",
             "tsv",
             "txt",
-            "xml");
+            "xml",
+            "xlsx");

 /**
  * @global array filetypes which should be considered images
@@ -293,7 +294,9 @@ $PAGE_PROCESSORS = array(   "text/html" => "HtmlProcessor",
                             "image/gif" => "GifProcessor",
                             "image/png" => "PngProcessor",
                             "image/bmp" => "BmpProcessor",
-                            "image/svg+xml"=> "SvgProcessor"
+                            "image/svg+xml"=> "SvgProcessor",
+                            "application/vnd.openxmlformats-officedocument.
+                                spreadsheetml.sheet" => "XlsxProcessor"
 );

 $INDEXING_PLUGINS = array("recipe");
diff --git a/lib/processors/xlsx_processor.php b/lib/processors/xlsx_processor.php
new file mode 100644
index 000000000..fa25729ee
--- /dev/null
+++ b/lib/processors/xlsx_processor.php
@@ -0,0 +1,306 @@
+<?php
+/**
+ *  SeekQuarry/Yioop --
+ *  Open Source Pure PHP Search Engine, Crawler, and Indexer
+ *
+ *  Copyright (C) 2009, 2010, 2011  Chris Pollett chris@pollett.org
+ *
+ *  LICENSE:
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ *  END LICENSE
+ *
+ * @author Tarun Ramaswamy tarun.pepira@gmail.com
+ * @package seek_quarry
+ * @subpackage processor
+ * @license http://www.gnu.org/licenses/ GPL3
+ * @link http://www.seekquarry.com/
+ * @copyright 2009, 2010, 2011
+ * @filesource
+ */
+
+if(!defined('BASE_DIR')) {echo "BAD REQUEST"; exit();}
+
+/**
+ * Load base class, if needed.
+ */
+require_once BASE_DIR."/lib/processors/text_processor.php";
+/**
+ * Load so can parse urls
+ */
+require_once BASE_DIR."/lib/url_parser.php";
+
+ /**
+ * Used to create crawl summary information
+ * for xlsx files
+ *
+ * @author Tarun Ramaswamy
+ * @package seek_quarry
+ * @subpackage processor
+ */
+
+class XlsxProcessor extends TextProcessor
+{
+    const MAX_DESCRIPTION_LEN = 2000;
+
+    /**
+     *  Used to extract the title, description and links from
+     *  a xlsx file.
+     *
+     *  @param string $page contents of xlsx file in zip format
+     *  @param string $url the url where the page contents came from,
+     *     used to canonicalize relative links
+     *
+     *  @return array  a summary of the contents of the page
+     *
+     */
+    function process($page, $url)
+    {
+
+        $summary = NULL;
+        $sites = array();
+
+        // Create a temporary xlsx file
+        $file_name=CRAWL_DIR . "/temp.xlsx";
+
+        file_put_contents($file_name, $page);
+
+        // Open a zip archive
+        $zip = new ZipArchive;
+        if ($zip->open($file_name) === TRUE) {
+
+            //Count of the sheets in xlsx
+            $file_count = 0;
+
+            //Getting the title from xlsx file
+            $buf= $zip->getFromName("docProps/app.xml");
+            if ($buf) {
+
+               $dom = self::dom($buf);
+               if($dom !== false) {
+                   // Get the title
+                   $summary[self::TITLE] = self::title($dom);
+
+                   $file_count = self::sheetCount($dom);
+               }
+            }
+
+            //Getting the description from xlsx file
+            $buf= $zip->getFromName("xl/sharedStrings.xml");
+            if ($buf) {
+
+                $dom = self::dom($buf);
+                if($dom !== false) {
+                    // Get the description
+                    $summary[self::DESCRIPTION] = self::description($dom);
+                }
+
+                //Getting the language from xlsx file
+                $summary[self::LANG] = self::lang($summary[self::DESCRIPTION]);
+            }
+
+            $summary[self::LINKS]=$sites;
+            //Getting links from each worksheet
+            for ($i = 1; $i<= $file_count; $i++) {
+                $buf= $zip->getFromName("xl/worksheets/_rels/sheet" . $i .
+                    ".xml.rels");
+                if ($buf) {
+
+                    $dom = self::dom($buf);
+                    if($dom !== false) {
+                        // Get the links
+                        $summary[self::LINKS] = array_merge(
+                            $summary[self::LINKS],self::links($dom,$url));
+                    }
+                }
+            }
+        }
+        else {
+            $summary=parent::process($page, $url);
+        }
+        // Close the zip
+        $zip->close();
+        //delete the temporarily created file
+        @unlink("$file_name");
+
+        return $summary;
+    }
+
+    /**
+     * Return a document object based on a string containing the contents of
+     * a xml file
+     *
+     *  @param string $page   xml document
+     *
+     *  @return object  document object
+     */
+    static function dom($page)
+    {
+        $dom = new DOMDocument();
+
+        @$dom->loadXML($page);
+
+        return $dom;
+    }
+
+    /**
+     *  Returns title of a xlsx file from each worksheet
+     *
+     *  @param object $dom   a document object to extract a title from.
+     *  @return string  a title of the xlsx file
+     *
+     */
+    static function title($dom)
+    {
+        $properties = $dom->getElementsByTagName("Properties");
+        $title = "";
+        foreach ($properties as $property) {
+            $titles = $property->getElementsByTagName("TitlesOfParts");
+            $title = $titles->item(0)->nodeValue;
+        }
+
+        return $title;
+     }
+
+    /**
+     *  Returns the count of worksheets in the xlsx file
+     *
+     *  @param object $dom   a document object to extract a title from.
+     *  @return integer  number of worksheets in the xlsx file
+     *
+     */
+    static function sheetCount($dom)
+    {
+        $count = 0;
+        $properties = $dom->getElementsByTagName("Properties");
+        foreach ($properties as $property) {
+            $titles = $property->getElementsByTagName("TitlesOfParts");
+            foreach ($titles as $vector) {
+                $vt_vector = $vector->getElementsByTagName("vector");
+                foreach($vt_vector as $vector_attribute) {
+                    $count = $vector_attribute->getAttribute("size");
+                }
+            }
+        }
+
+        return $count;
+    }
+
+    /**
+     * Returns descriptive text concerning a xlsx file based on its document
+     * object
+     *
+     * @param object $dom   a document object to extract a description from.
+     * @return string a description of the slide
+     */
+    static function description($dom)
+    {
+        $xpath = new DOMXPath($dom);
+
+        $sst_tag = $dom->getElementsByTagName("sst");
+        $descriptions = "";
+        foreach ($sst_tag as $sst_value) {
+            $t_tag = $sst_value->getElementsByTagName("si");
+            foreach ($t_tag as $t_value) {
+                $si_value = $t_value->getElementsByTagName("t");
+                $descriptions .= $si_value->item(0)->nodeValue;
+            }
+        }
+        return $descriptions;
+    }
+
+    /**
+     *  Determines the language of the xlsx document by looking at data
+     *  and guessing the language
+     *  Used the code from html_processor.php
+     *
+     *  @param string $sample_text sample text to try guess the language from
+     *
+     *  @return string language tag for guessed language
+     */
+    static function lang($sample_text = NULL)
+    {
+        $lang = NULL;
+
+        if($lang == NULL && $sample_text != NULL) {
+            $words = mb_split("[[:space:]]|".PUNCT, $sample_text);
+            $num_words = count($words);
+            $ascii_count = 0;
+            foreach($words as $word) {
+                if(strlen($word) == mb_strlen($word)) {
+                    $ascii_count++;
+                }
+            }
+            // crude, but let's guess ASCII == english
+            if($ascii_count/$num_words > EN_RATIO) {
+                $lang = 'en';
+            } else {
+                $lang = NULL;
+            }
+        } else {
+            $lang = NULL;
+        }
+
+        return $lang;
+    }
+
+    /**
+     * Returns up to MAX_LINK_PER_PAGE many links from the supplied
+     * dom object where links have been canonicalized according to
+     * the supplied $site information.
+     *
+     * @param object $dom   a document object with links on it
+     * @param string $site   a string containing a url
+     *
+     * @return array   links from the $dom object
+     */
+    static function links($dom, $site)
+    {
+        $sites = array();
+        $hyperlink = "http://schemas.openxmlformats.org/officeDocument/2006/".
+            "relationships/hyperlink";
+        $i = 0;
+        $relationships = $dom->getElementsByTagName("Relationships");
+
+        foreach ($relationships as $relationship) {
+            $relations = $relationship->getElementsByTagName("Relationship");
+            foreach ($relations as $relation) {
+                if( strcmp( $relation->getAttribute('Type'),
+                    $hyperlink) == 0 ) {
+
+                    if($i < MAX_LINKS_PER_PAGE) {
+                        $link = $relation->getAttribute('Target');
+                        $url = UrlParser::canonicalLink(
+                            $link, $site);
+                        if(!UrlParser::checkRecursiveUrl($url)  &&
+                            strlen($url) < MAX_URL_LENGTH) {
+                            if(isset($sites[$url])) {
+                                $sites[$url] .=" ".$link;
+                            } else {
+                                $sites[$url] = $link;
+                            }
+                            $i++;
+                        }
+                    }
+                }
+            }
+        }
+
+        return $sites;
+    }
+
+}
+
+?>
\ No newline at end of file
diff --git a/tests/test_files/test.xlsx b/tests/test_files/test.xlsx
new file mode 100644
index 000000000..83babc72e
Binary files /dev/null and b/tests/test_files/test.xlsx differ
diff --git a/tests/xlsx_processor_test.php b/tests/xlsx_processor_test.php
new file mode 100644
index 000000000..d60147e97
--- /dev/null
+++ b/tests/xlsx_processor_test.php
@@ -0,0 +1,129 @@
+<?php
+/**
+ *  SeekQuarry/Yioop --
+ *  Open Source Pure PHP Search Engine, Crawler, and Indexer
+ *
+ *  Copyright (C) 2009, 2010, 2011  Chris Pollett chris@pollett.org
+ *
+ *  LICENSE:
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ *  END LICENSE
+ *
+ * @author Tarun Ramaswamy tarun.pepira@gmail.com
+ * @package seek_quarry
+ * @subpackage test
+ * @license http://www.gnu.org/licenses/ GPL3
+ * @link http://www.seekquarry.com/
+ * @copyright 2009, 2010, 2011
+ * @filesource
+ */
+if(!defined('BASE_DIR')) {echo "BAD REQUEST"; exit();}
+
+/**
+ *  Load in the xlsx_processor class we are testing
+ *
+ */
+require_once BASE_DIR."/lib/processors/xlsx_processor.php";
+
+/**
+ * Used to test that the XlsxProcessor class provides the basic functionality
+ * of getting the tile, description, languages and links
+ *
+ * @author Tarun Ramaswamy
+ * @package seek_quarry
+ * @subpackage test
+ */
+class XlsxProcessorTest extends UnitTest implements CrawlConstants
+{
+    /**
+     * sets up the initial content for the testcase by extracting
+     * it from the xlsx file
+     */
+    public function setUp()
+    {
+        $file_name="test_files/test.xlsx";
+        $page=file_get_contents($file_name);
+        $xlsx_processor = new XlsxProcessor();
+        $summary = NULL;
+        $url = "http://localhost:80/";
+        $summary = $xlsx_processor->process($page,$url);
+        $this->test_objects['title'] = $summary[self::TITLE];
+        $this->test_objects['description'] = $summary[self::DESCRIPTION];
+        $this->test_objects['language'] = $summary[self::LANG];
+        $this->test_objects['links'] = $summary[self::LINKS];
+    }
+
+    /**
+     * Can be used for clenup activity
+     */
+    public function tearDown()
+    {
+
+    }
+
+    /**
+     * Tests that the title is correct
+     */
+    public function titleTestCase()
+    {
+
+        $title = "SampleTitle";
+
+        $this->assertEqual($this->test_objects['title'], $title,
+            "check for title");
+    }
+
+    /**
+     * Tests that the description is correct
+     */
+    public function descriptionTestCase()
+    {
+
+        $description = "This is a sample descriptionlink1link2";
+
+        $this->assertEqual($this->test_objects['description'], $description,
+            "check for description");
+    }
+
+    /**
+     * Tests that the language is correct
+     */
+    public function languageTestCase()
+    {
+
+        $language = "en";
+
+        $this->assertEqual($this->test_objects['language'], $language,
+            "check for language");
+    }
+
+    /**
+     * Tests that the links are correct
+     */
+    public function linksTestCase()
+    {
+
+        $sites = array();
+        $sites[0] = "http://www.yahoo.com/";
+        $sites[1] = "http://www.google.com/";
+        $i = 0;
+        foreach($this->test_objects['links'] as $link) {
+            $this->assertEqual($link, $sites[$i], "check for Links");
+            $i++;
+        }
+    }
+}
+?>
\ No newline at end of file
ViewGit