add classes to work with archives

Mihail
1 parent 8b0defd0
Showing 7 changed files with 207 additions and 29 deletions Show diff stats
common/components/archive_reader/ArchiveCreator.php
common/components/archive_reader/ArchiveReader.php
common/components/archive_reader/RarArchiveReader.php
common/components/mail/ImapMailReader.php
common/components/mail/MailAttachmentsSaver.php
common/components/mail/MailReader.php
console/controllers/ParserController.php
+<?php
+/**
+ * Created by PhpStorm.
+ * User: Tsurkanov
+ * Date: 03.11.2015
+ * Time: 14:51
+ */
+
+namespace common\components\archive_reader;
+
+
+class ArchiveCreator {
+
+  protected  $handleExtension = [];
+
+    public  function create( $file, $ext ){
+        if ( isHandableExtension( $ext )) {
+            $arh_class = ucfirst( $ext ) . 'ArchiveReader';
+            if ( class_exists( $arh_class ) ) {
+
+                $arh_reader =  new $arh_class();
+
+                if ($arh_reader instanceof ArchiveReader ) {
+                    $arh_reader->open($file);
+                    return $arh_reader;
+                }
+
+            }
+
+        }
+        // не найден подходящий обработчик
+        throw new \Exception( "Для расширения {$ext} не найден подходящий распаковщик" );
+    }
+
+    protected  function isHandableExtension( $ext ){
+
+        $this->setHandleExtension( );
+        return (bool) array_search( $ext, $this->handleExtension);
+    }
+
+    protected  function setHandleExtension( ){
+        if ( !$this->handleExtension ) {
+            foreach (get_declared_classes() as $class) {
+                if (is_subclass_of( $class, ArchiveReader::class ))
+
+                    $this->handleExtension[] = $class::getExtension();
+
+            }
+        }
+    }
+
+    public  function getHandleExtension( ){
+
+       return $this->handleExtension;
+
+    }
+
+
+}
 \ No newline at end of file
+<?php
+/**
+ * Created by PhpStorm.
+ * User: Tsurkanov
+ * Date: 03.11.2015
+ * Time: 14:48
+ */
+namespace common\components\archive_reader;
+abstract class ArchiveReader
+{
+
+    protected $extracted_files = [];
+
+    public abstract function open( $file, $password = '');
+
+    public abstract function extractTo($destination);
+
+    public static abstract function getExtension();
+
+    public  function getExtractedFiles(){
+        return $this->extracted_files;
+    }
+    public  function setExtractedFiles($name, $ext){
+        $this->extracted_files[$name] = $ext;
+    }
+
+}
 \ No newline at end of file
+<?php
+/**
+ * Created by PhpStorm.
+ * User: Tsurkanov
+ * Date: 03.11.2015
+ * Time: 15:12
+ */
+
+namespace common\components\archive_reader;
+
+
+class RarArchiveReader extends  ArchiveReader {
+
+    public  function open( $file, $password = '' ){
+        $rar_arch = \RarArchive::open( $file, $password );
+        if ($rar_arch === FALSE)
+            die("Failed opening file");
+    }
+
+    public  function extractTo( $destination){
+
+    }
+    public static  function getExtension(){
+        return 'rar';
+    }
+
+
+
+
+}
 \ No newline at end of file
@@ -17,7 +17,7 @@ class ImapMailReader extends  MailReader {
         $this->connection = imap_open($hostname, $username, $password);
  
         if ($this->connection === false)
-            throw new \Exception('Cannot connect to Gmail: ' . imap_last_error());
+            throw new \Exception('Cannot connect to mail: ' . imap_last_error());
  
     }
  
@@ -26,6 +26,11 @@ class ImapMailReader extends  MailReader {
        return imap_search( $this->connection, $flag );
     }
  
+    public function getEmailBody( $email_number, $section )
+    {
+       return imap_fetchbody($this->connection, $email_number, $section );
+    }
+
     /**
      * @return mixed
      */
@@ -11,24 +11,47 @@ namespace common\components\mail;
  
 use common\components\CustomVarDamp;
  
+/**
+ * Class MailAttachmentsSaver
+ * @package common\components\mail
+ * сохраняет вложения в указанную папку по полученому соединению к ящику,
+ * а также хранит имена сохраненных файлов
+ */
 class MailAttachmentsSaver
 {
+
+    /**
+     * @var соединение с ящиком - экземляр класса - MailReader
+     */
     protected $mail_reader;
+
+    /**
+     * @var - string, тип сообщений - например UNSEEN. Значения можно перечислять разделяя запятой.
+     */
     protected $massage_type;
+
+    /**
+     * @var array - после сохранения будет содержать сохраненные файлы, ключ - путь к файлу, значение - расширение
+     */
+    protected $saved_files_arr;
+
+    /**
+     * @var - префикс который будет прибавлен к оригинальному имени сохраняемого файла
+     */
     public $file_name_prefix;
  
-    public function __construct( MailReader $mail_reader, $massage_type )
+    public function __construct(MailReader $mail_reader, $massage_type)
     {
         $this->mail_reader = $mail_reader;
         $this->massage_type = $massage_type;
+        $this->saved_files_arr = [];
  
     }
  
  
     public function saveAttachmentsTo( $destination )
     {
-
-        $emails = $this->mail_reader->getEmails( $this->massage_type );
+        $emails = $this->mail_reader->getEmails($this->massage_type);
  
         /* if emails are returned, cycle through each... */
  
@@ -39,7 +62,7 @@ class MailAttachmentsSaver
  
             /* put the newest emails on top */
             rsort($emails);
-           // CustomVarDamp::dump($emails);
+            // CustomVarDamp::dump($emails);
             foreach ($emails as $email_number) {
  
                 $structure = $this->mail_reader->getCurrentEmailStructure($email_number);
@@ -71,7 +94,7 @@ class MailAttachmentsSaver
                         }
  
                         if ($attachments[$i]['is_attachment']) {
-                            $attachments[$i]['attachment'] = imap_fetchbody($this->mail_reader->connection, $email_number, $i + 1);
+                            $attachments[$i]['attachment'] = $this->mail_reader->getEmailBody($email_number, $i + 1);
                             if ($structure->parts[$i]->encoding == 3) { // 3 = BASE64
                                 $attachments[$i]['attachment'] = base64_decode($attachments[$i]['attachment']);
                             } elseif ($structure->parts[$i]->encoding == 4) { // 4 = QUOTED-PRINTABLE
@@ -82,23 +105,20 @@ class MailAttachmentsSaver
  
                     if (count($attachments) != 0) {
  
-                                foreach($attachments as $key => &$val){
-                                    if ($val['is_attachment'] == 1) {
-
-                                        if( isset($this->file_name_prefix) ){
-                                            $name = $destination . '/' . $this->file_name_prefix . mb_decode_mimeheader($val['name']);
-                                        }else{
-                                            $name = $destination . '/' . mb_decode_mimeheader($val['name']);
-                                        }
-                                    mb_internal_encoding("UTF-8");
-
-                                    //file_put_contents(\Yii::getAlias('@temp_upload') . '/' . $name, $val['attachment']);
-                                    file_put_contents( $name, $val['attachment'] );
+                        foreach ($attachments as $key => &$val) {
+                            if ($val['is_attachment'] == 1) {
+                                if (isset($this->file_name_prefix)) {
+                                    $name = $destination . '/' . $this->file_name_prefix . mb_decode_mimeheader($val['name']);
+                                } else {
+                                    $name = $destination . '/' . mb_decode_mimeheader($val['name']);
                                 }
-
+                                $ext = pathinfo($name, PATHINFO_EXTENSION);
+                                mb_internal_encoding("UTF-8");
+                                file_put_contents($name, $val['attachment']);
+                                $this->setSavedFile( $name , $ext );
                             }
-                        }
  
+                        }
                     }
  
                 }
@@ -106,4 +126,24 @@ class MailAttachmentsSaver
             }
  
         }
+
+    }
+
+    /**
+     * @return array
+     */
+    public function getSavedFilesArr()
+    {
+        return $this->saved_files_arr;
+    }
+
+    /**
+     * @param array $saved_files_arr
+     */
+    public function setSavedFile($saved_file, $saved_file_ext)
+    {
+        $this->saved_files_arr[$saved_file] = $saved_file_ext;
+    }
+
+
 }
 \ No newline at end of file
@@ -35,12 +35,12 @@ abstract class  MailReader {
         return $this->hostname;
     }
  
-
-
     public abstract function getListMailboxes();
  
     public abstract function getEmails( $flag );
  
+    public abstract function getEmailBody( $email_number, $section );
+
     public abstract function getCurrentEmailStructure( $email );
  
     public abstract function reOpen( $hostname );
 <?php
 namespace console\controllers;
  
+use common\components\archive_reader\ArchiveCreator;
 use common\components\CustomVarDamp;
 use common\components\mail\ImapMailReader;
 use common\components\mail\MailAttachmentsSaver;
@@ -116,7 +117,7 @@ class ParserController extends Controller
  
     public function actionTest()
     {
-        Console::output('It is working ');
+        Console::output('It is working');
         \Yii::info('2', 'parser');
  
     }
@@ -127,19 +128,35 @@ class ParserController extends Controller
  
         $mail_reader = new ImapMailReader( '{imap.gmail.com:993/imap/ssl/novalidate-cert}', 'tsurkanovm@gmail.com', 'Wtvr@2000' );
         $mailboxes = $mail_reader->getListMailboxes();
-        foreach ($mailboxes as $custom_label) {
-            $words = explode(" ",str_replace(array($mail_reader->getHostname(),"!"),"",imap_utf7_decode($custom_label)));
-            $importer_id = (int)preg_replace("/[^A-Z0-9]+/","",strtoupper($words[0]));
+        foreach ( $mailboxes as $custom_label ) {
+            $words = explode(" ",str_replace( array($mail_reader->getHostname(),"!"),"",imap_utf7_decode($custom_label)) );
+            $importer_id = (int)preg_replace("/[^A-Z0-9]+/","", strtoupper($words[0]));
  
-            $mail_reader->reOpen($custom_label);
+            $mail_reader->reOpen( $custom_label );
             $saver = new MailAttachmentsSaver( $mail_reader, 'UNSEEN' );
             $saver->file_name_parefix = $importer_id . '~!~';
             $saver->saveAttachmentsTo(\Yii::getAlias('@temp_upload'));
  
-//            $files_model = new ImportersFiles();
-//            $files_model->importer_id = $importer_id;
         }
  
+        $files = $saver->getSavedFilesArr();
+        $arch_creator = new ArchiveCreator();
+        $arch_extensions = $arch_creator->getHandleExtension();
+        $files_on_unpack = array_intersect( $files , $arch_extensions );
+        foreach ($files_on_unpack as $arch_ext => $arch_name) {
+            $arch_reader = $arch_creator->create( $arch_name, $arch_ext);
+            $arch_reader->extractTo(\Yii::getAlias('@temp_upload'));
+            unset( $files[$arch_name] );
+            $files = array_merge( $files, $arch_reader->getExtractedFiles());
+        }
+
+        foreach ( $files as $file => $key ) {
+
+
+            //            $files_model = new ImportersFiles();
+//            $files_model->importer_id = $importer_id;
+            // $db->lastInsertId()
+        }
  
     }
 }
 \ No newline at end of file