storage.php

<?php
 
namespace Bitrix\Mail\Helper\Attachment;
 
use Bitrix\Main;
use Bitrix\Main\Localization\Loc;
use Bitrix\Mail;
 
class Storage
{
 
    public static function getStorage()
    {
        static $storage;
 
        if (!is_null($storage))
        {
            return $storage;
        }
 
        $storage = false;
 
        if (!Main\Loader::includeModule('disk'))
        {
            return $storage;
        }
 
        $storageId = Main\Config\Option::get('mail', 'disk_attachment_storage_id', 0);
        if ($storageId > 0)
        {
            $storage = \Bitrix\Disk\Storage::loadById($storageId);
            if (!$storage || $storage->getModuleId() != 'mail')
            {
                $storage = false;
            }
        }
 
        if (!$storage)
        {
            $driver = \Bitrix\Disk\Driver::getInstance();
 
            $storage = $driver->addStorageIfNotExist(array(
                'NAME' => Loc::getMessage('MAIL_ATTACHMENT_STORAGE_NAME'),
                'USE_INTERNAL_RIGHTS' => false,
                'MODULE_ID' => 'mail',
                'ENTITY_TYPE' => Mail\Disk\ProxyType\Mail::className(),
                'ENTITY_ID' => 'mail',
            ));
            if ($storage)
            {
                Main\Config\Option::set('mail', 'disk_attachment_storage_id', $storage->getId());
            }
            else
            {
                $storage = false;
            }
        }
 
        return $storage;
    }
 
    public static function getUrlManager()
    {
        static $urlManager;
 
        if (!is_null($urlManager))
        {
            return $urlManager;
        }
 
        $urlManager = false;
 
        if (!Main\Loader::includeModule('disk'))
        {
            return $urlManager;
        }
 
        $urlManager = \Bitrix\Disk\Driver::getInstance()->getUrlManager();
 
        return $urlManager;
    }
 
    public static function getObjectsByFileId($fileId, $limit = 0)
    {
        $storage = static::getStorage();
 
        if (!$storage)
        {
            return array();
        }
 
        return \Bitrix\Disk\File::getModelList(array(
            'filter' => array(
                '=STORAGE_ID' => $storage->getId(),
                '=TYPE' => \Bitrix\Disk\Internals\ObjectTable::TYPE_FILE,
                '=FILE_ID' => $fileId,
            ),
            'limit' => $limit,
        ));
    }
 
    public static function getObjectByAttachment(array $attachment, $create = false)
    {
        $list = static::getObjectsByFileId($attachment['FILE_ID'], 1);
        $object = reset($list);
 
        if (empty($object) && $create)
        {
            $object = static::registerAttachment($attachment);
        }
 
        return $object;
    }
 
    public static function registerAttachment(array $attachment)
    {
        $storage = static::getStorage();
 
        if (!$storage)
        {
            return false;
        }
 
        $folder = $storage->getChild(array(
            '=NAME' => date('Y-m'),
            '=TYPE' => \Bitrix\Disk\Internals\FolderTable::TYPE,
        ));
 
        if (!$folder)
        {
            $folder = $storage->addFolder(array(
                'NAME' => date('Y-m'),
                'CREATED_BY' => 1, // @TODO
            ));
        }
 
        if (!$folder)
        {
            $folder = $storage;
        }
 
        return $folder->addFile(
            array(
                'NAME' => \Bitrix\Disk\Ui\Text::correctFilename($attachment['FILE_NAME']) ?: sprintf('%x', rand(0, 0xffffff)),
                'FILE_ID' => $attachment['FILE_ID'],
                'SIZE' => $attachment['FILE_SIZE'],
                'CREATED_BY' => 1, // @TODO
            ),
            array(),
            true
        );
    }
 
    public static function unregisterAttachment($fileId)
    {
        foreach (static::getObjectsByFileId($fileId) as $item)
        {
            $item->delete(1); // @TODO
        }
    }
 
}