Blame view

common/models/Media.php 3.26 KB
14a09168   Alex Savenko   init commit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
  <?php

  

  namespace common\models;

  

  use common\modules\blog\models\Article;

  use common\modules\blog\models\ArticleCategoryMedia;

  use common\modules\blog\models\ArticleMedia;

  use Yii;

  use yii\db\ActiveRecord;

  use yii\web\UploadedFile;

  

  /**

   * This is the model class for table "media".

   *

   * @property integer $media_id

   * @property string $hash

   * @property string $extension

   *

   * @property ArticleCategoryMedia[] $articleCategoryMedia

   * @property ArticleMedia[] $articleMedia

   */

  class Media extends ActiveRecord

  {

      /**

       * @var UploadedFile[];

       */

      public $imageFile;

  

      /**

       * @inheritdoc

       */

      public static function tableName()

      {

          return 'media';

      }

  

      /**

       * @inheritdoc

       */

      public function rules()

      {

          return [

              [['media_id'], 'integer'],

              [['hash', 'extension'], 'string'],

              [['imageFile'], 'file', 'extensions' => 'png, gif, jpeg, jpg', 'skipOnEmpty' => true],

          ];

      }

  

      /**

       * @inheritdoc

       */

      public function  attributeLabels()

      {

          return [

              'media_id' => Yii::t('app', 'media_id'),

              'hash' => Yii::t('app', 'hash'),

              'extension' => Yii::t('app', 'extension'),

          ];

      }

  

      /**

       * @return \yii\db\ActiveQuery

       */

      public function getArticleCategoryMedia()

      {

          return $this->hasMany(ArticleCategoryMedia::className(), ['media_id' => 'media_id']);

      }

  

      /**

       * @return \yii\db\ActiveQuery

       */

      public function getArticleMedia()

      {

          return $this->hasMany(ArticleMedia::className(), ['media_id' => 'media_id']);

      }

  

      public function getArticle()

      {

          return $this->hasMany(Article::className(), ['article_id' => 'article_id'])->via('articleMedia');

      }

  

      public function upload()

      {

          if(!empty($this->imageFile) && $this->validate('imageFile'))

          {

              $uploaddir = \Yii::getAlias('@saveImageDir');

              $this->hash = md5_file($this->imageFile->tempName).\Yii::$app->security->generateRandomString(5);

              $this->extension = $this->imageFile->extension;

              if(is_dir($uploaddir.$this->hash)) {

                  return false;

              } else {

                  if(!mkdir($uploaddir.$this->hash, 0755)) {

                      return false;

                  }

              }

              $this->imageFile->saveAs($uploaddir.$this->hash.'/original.'.$this->imageFile->extension, false);

              if($this->save(false)) {

                  return true;

              } else {

                  $this->addError('imageFile', \Yii::t('app', 'Cannot load file'));

                  return false;

              }

          } else {

              return false;

          }

      }

  

      public function delete()

      {

          $uploaddir = \Yii::getAlias('@saveImageDir');

          if(is_dir($uploaddir.$this->hash)) {

              $this->removeDir($uploaddir.$this->hash);

          }

          return parent::delete();

      }

  

      public function removeDir($dir)

      {

          if($objs = glob($dir."/*")) {

              foreach($objs as $obj) {

                  is_dir($obj) ? removeDir($obj) : unlink($obj);

              }

          }

          rmdir($dir);

      }

  }