Initial commit

core/lib/Thelia/Model/Folder.php

@@ -0,0 +1,168 @@
+<?php
+
+namespace Thelia\Model;
+
+use Propel\Runtime\ActiveQuery\Criteria;
+use Propel\Runtime\Collection\ObjectCollection;
+use Propel\Runtime\Connection\ConnectionInterface;
+use Thelia\Core\Event\Folder\FolderEvent;
+use Thelia\Core\Event\TheliaEvents;
+use Thelia\Files\FileModelParentInterface;
+use Thelia\Model\Base\Folder as BaseFolder;
+use Thelia\Model\Tools\ModelEventDispatcherTrait;
+use Thelia\Model\Tools\PositionManagementTrait;
+use Thelia\Model\Tools\UrlRewritingTrait;
+
+class Folder extends BaseFolder implements FileModelParentInterface
+{
+    use ModelEventDispatcherTrait;
+
+    use PositionManagementTrait;
+
+    use UrlRewritingTrait;
+
+    /**
+     * {@inheritDoc}
+     */
+    public function getRewrittenUrlViewName()
+    {
+        return 'folder';
+    }
+
+    /**
+     * @return int number of contents for the folder
+     */
+    public function countChild()
+    {
+        return FolderQuery::countChild($this->getId());
+    }
+
+    /**
+     * count all products for current category and sub categories
+     *
+     * @param bool|string $contentVisibility: true (default) to count only visible products, false to count only hidden
+     *                    products, or * to count all products.
+     * @return int
+     */
+    public function countAllContents($contentVisibility = true)
+    {
+        $children = FolderQuery::findAllChild($this->getId());
+        array_push($children, $this);
+
+
+        $query = ContentQuery::create()->filterByFolder(new ObjectCollection($children), Criteria::IN);
+
+        if ($contentVisibility !== '*') {
+            $query->filterByVisible($contentVisibility);
+        }
+
+        return $query->count();
+    }
+
+    /**
+     * Get the root folder
+     * @param  int   $folderId
+     * @return mixed
+     */
+    public function getRoot($folderId)
+    {
+        $folder = FolderQuery::create()->findPk($folderId);
+
+        if (0 !== $folder->getParent()) {
+            $parentFolder = FolderQuery::create()->findPk($folder->getParent());
+
+            if (null !== $parentFolder) {
+                $folderId = $this->getRoot($parentFolder->getId());
+            }
+        }
+
+        return $folderId;
+    }
+
+    /**
+     * Calculate next position relative to our parent
+
+     * @param FolderQuery $query
+     */
+    protected function addCriteriaToPositionQuery($query)
+    {
+        $query->filterByParent($this->getParent());
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function preInsert(ConnectionInterface $con = null)
+    {
+        parent::preInsert($con);
+
+        $this->setPosition($this->getNextPosition());
+
+        $this->dispatchEvent(TheliaEvents::BEFORE_CREATEFOLDER, new FolderEvent($this));
+
+        return true;
+    }
+
+    public function postInsert(ConnectionInterface $con = null)
+    {
+        parent::postInsert($con);
+
+        $this->dispatchEvent(TheliaEvents::AFTER_CREATEFOLDER, new FolderEvent($this));
+    }
+
+    public function preUpdate(ConnectionInterface $con = null)
+    {
+        parent::preUpdate($con);
+
+        $this->dispatchEvent(TheliaEvents::BEFORE_UPDATEFOLDER, new FolderEvent($this));
+
+        return true;
+    }
+
+    public function postUpdate(ConnectionInterface $con = null)
+    {
+        parent::postUpdate($con);
+
+        $this->dispatchEvent(TheliaEvents::AFTER_UPDATEFOLDER, new FolderEvent($this));
+    }
+
+    public function preDelete(ConnectionInterface $con = null)
+    {
+        parent::preDelete($con);
+
+        $this->dispatchEvent(TheliaEvents::BEFORE_DELETEFOLDER, new FolderEvent($this));
+        $this->reorderBeforeDelete(
+            array(
+                "parent" => $this->getParent(),
+            )
+        );
+
+        return true;
+    }
+
+    public function postDelete(ConnectionInterface $con = null)
+    {
+        parent::postDelete($con);
+
+        $this->markRewrittenUrlObsolete();
+
+        $this->dispatchEvent(TheliaEvents::AFTER_DELETEFOLDER, new FolderEvent($this));
+    }
+
+    /**
+     * Overload for the position management
+     * @param Base\ContentFolder $contentFolder
+     * @inheritdoc
+     */
+    protected function doAddContentFolder($contentFolder)
+    {
+        parent::doAddContentFolder($contentFolder);
+
+        $contentFolderPosition = ContentFolderQuery::create()
+            ->filterByFolderId($contentFolder->getFolderId())
+            ->orderByPosition(Criteria::DESC)
+            ->findOne();
+
+        $contentFolder->setPosition($contentFolderPosition !== null ? $contentFolderPosition->getPosition() + 1 : 1);
+    }
+}