Ticket #1765

modules/boonex/dolphin_migration/classes/BxDolMMemLevels.php

@@ -0,0 +1,204 @@
+<?php defined('BX_DOL') or die('hack attempt');
+/**
+ * Copyright (c) UNA, Inc - https://una.io
+ * MIT License - https://opensource.org/licenses/MIT
+ *
+ * @defgroup    DolphinMigration  Dolphin Migration
+ * @ingroup     UnaModules
+ *
+ * @{
+ */
+
+bx_import('BxDolProfile');
+bx_import('BxDolTranscoderImage');
+bx_import('BxDolStorage');
+	
+require_once('BxDolMData.php');	
+	
+class BxDolMMemLevels extends BxDolMData
+{
+    public function __construct(&$oMigrationModule, &$mDb)
+    {
+		parent::__construct($oMigrationModule, $mDb);
+		$this -> _sModuleName = 'membership_levels';
+		$this -> _sTableWithTransKey = 'sys_acl_levels';
+    }
+
+	public function getTotalRecords()
+	{
+		$iPaidMemberships = $this -> _mDb -> getOne("SELECT COUNT(*) FROM `" . $this -> _oConfig -> _aMigrationModules[$this -> _sModuleName]['table_name'] . "` WHERE `Active` = 'yes' AND `Purchasable` = 'yes'");
+		$iMemberWithSetMembership = $this -> _mDb -> getOne("SELECT COUNT(*) FROM `sys_acl_levels_members`");
+	    return array($iPaidMemberships, $iMemberWithSetMembership);
+	}
+	
+	public function runMigration()
+	{
+	    if (!$this -> getTotalRecords())
+		{
+			  $this -> setResultStatus(_t('_bx_dolphin_migration_no_data_to_transfer'));
+	          return BX_MIG_SUCCESSFUL;
+		}	
+		
+		$this -> setResultStatus(_t("_bx_dolphin_migration_started_migration_{$this -> _sModuleName}"));
+
+        $iPaidCount = $this -> transferMembershipLevels();
+        $aResult = $this -> _mDb -> getAll("SELECT * FROM `sys_acl_levels_members` ORDER BY `IDMember` ASC");
+        foreach($aResult as $iKey => $aValue)
+        {
+            $iProfileId = $this -> getProfileId($aValue['IDMember']);
+            if($iProfileId)
+            {
+                $sQuery = $this -> _oDb -> prepare(
+                    "INSERT INTO
+                     		`sys_acl_levels_members`
+                     	SET
+                     		`IDMember` = ?,
+                            `IDLevel` = ?,
+                            `DateStarts` = ?,
+                            `DateExpires` = ?,
+                            `TransactionID` = ?
+                     ",
+                    $iProfileId,
+                    $this-> getMembershipById($aValue['IDLevel']),
+                    $aValue['DateStarts'],
+                    $aValue['DateExpires'],
+                    $aValue['TransactionID']
+                );
+
+                if ($this -> _oDb -> query($sQuery))
+                    $this -> _iTransferred++;
+            }
+        }
+
+        $this -> setResultStatus(_t("_bx_dolphin_migration_started_migration_{$this -> _sModuleName}_finished", $iPaidCount, $this -> _iTransferred));
+		return BX_MIG_SUCCESSFUL;
+    }
+
+    private function getMembershipById($iOldId){
+        $aCommon = array(1 => 1, 2 => 3);
+        if (isset($aCommon[$iOldId]))
+            return $aCommon[$iOldId];
+
+        return $this -> _oDb -> getOne("SELECT `ID` FROM `sys_acl_levels` WHERE `{$this -> _sTransferFieldIdent}` = :id LIMIT 1", array('id' => $iOldId));
+    }
+
+	private function transferMembershipIcon($sIconName)
+    {
+       if (!$sIconName)
+           return 'user';
+
+       $iId = 0;
+       $sImagePath = $this -> _oDb -> getExtraParam('root') . 'media' . DIRECTORY_SEPARATOR . 'images' . DIRECTORY_SEPARATOR . 'membership' . DIRECTORY_SEPARATOR . $sIconName;
+       if($sImagePath){
+			$oStorage = BxDolStorage::getObjectInstance('sys_images');
+			$iId = $oStorage->storeFileFromPath($sImagePath, false);
+	   }
+
+		return $iId;
+    }
+
+    /**
+     * Transfer all purchasable membership levels
+     */
+    private function transferMembershipLevels()
+    {
+        $iCounter = 0;
+        $aMembershipLevels = $this -> _mDb -> getAll("SELECT * FROM `sys_acl_levels` WHERE `Active` = 'yes' AND `Purchasable` = 'yes'");
+        if (empty($aMembershipLevels))
+            return false;
+
+        $this -> createMIdField();
+        $iOrder = $this -> _oDb -> getOne("SELECT MAX(`Order`) FROM `sys_acl_levels` LIMIT 1");
+        foreach($aMembershipLevels as $iKey => $aLevel)
+        {
+            if ($this -> isItemExisted($aLevel['ID'], 'ID'))
+                continue;
+
+            $sName = strtolower($aLevel['Name']);
+            $sLangKeyName = "_adm_prm_txt_level_{$sName}";
+            $sLangKeyDescription = '';
+            foreach($this -> _aLanguages as $sLangKey => $sValue) {
+                $this->_oLanguage->addLanguageString($sLangKeyName, $aLevel['Name'], $this->_aLanguages[$sLangKey], 1, !$aLevel['Description']);
+
+                if ($aLevel['Description']) {
+                    $sLangKeyDescription = '_sys_form_input_description_' . (time() + $iCounter);
+                    $this->_oLanguage->addLanguageString($sLangKeyDescription, $aLevel['Description'], $this->_aLanguages[$sLangKey], 1, true);
+                }
+            }
+
+
+            $sQuery = $this -> _oDb -> prepare("INSERT INTO `sys_acl_levels` 
+                                                SET 
+                                                `Name` = ?,
+                                                `Icon` = ?,
+                                                `Description` = ?,
+                                                `Active` = 'yes',
+                                                `Order` = ?
+                                                 ",
+                $sLangKeyName,
+                $this -> transferMembershipIcon($aLevel['Icon']),
+                $sLangKeyDescription,
+                $iOrder++
+            );
+
+            if ($this -> _oDb -> query($sQuery) && $iId = $this -> _oDb -> lastId()) {
+                  $this -> setMID($iId, $aLevel['ID']);
+                  $aPrices = $this -> _mDb -> getAll("SELECT * FROM `sys_acl_level_prices` WHERE `IDLevel` =  :id", array('id' => $aLevel['ID']));
+                  foreach($aPrices as $iKey => $aPrice){
+                      $sQuery = $this -> _oDb -> prepare("REPLACE INTO `bx_acl_level_prices` 
+                                                SET 
+                                                    `level_id` = ?,
+                                                    `name` = ?,
+                                                    `period` = ?,
+                                                    `period_unit`='day',                                       
+                                                    `price` = ?,
+                                                    `order` = ?
+                                                 ",
+                          $iId,
+                          "{$sName}-{$aPrice['Days']}-day",
+                          $aPrice['Days'],
+                          $aPrice['Price'],
+                          $aPrice['id']
+                      );
+
+                      $this -> _oDb -> query($sQuery);
+                  }
+
+                $iCounter++;
+            }
+        }
+
+        return $iCounter;
+    }
+
+    public function removeContent()
+    {
+        if (!$this -> _oDb -> isTableExists($this -> _sTableWithTransKey) || !$this -> _oDb -> isFieldExists($this -> _sTableWithTransKey, $this -> _sTransferFieldIdent))
+            return false;
+
+        $iNumber = 0;
+        $aLevels = $this -> _oDb -> getAll("SELECT * FROM `{$this -> _sTableWithTransKey}` WHERE `{$this -> _sTransferFieldIdent}` !=0");
+        if (!empty($aLevels) && $this -> _oDb -> isTableExists('bx_acl_level_prices'))
+        {
+            foreach($aLevels as $iKey => $aLevel){
+                $this->_oDb->query("DELETE FROM `bx_acl_level_prices` WHERE `level_id` = :id", array('id' => $aLevel['ID']));
+
+                if(is_numeric($aLevel['Icon']))
+                    BxDolStorage::getObjectInstance(BX_DOL_STORAGE_OBJ_IMAGES)->deleteFile((int)$aLevel['Icon'], 0);
+
+                $this->_oLanguage->deleteLanguageString($aLevel['Name']);
+                $this->_oLanguage->deleteLanguageString($aLevel['Description']);
+                $this->_oDb->query("DELETE FROM `{$this -> _sTableWithTransKey}` WHERE `{$this -> _sTransferFieldIdent}` = :id", array('id' => $aLevel[$this -> _sTransferFieldIdent]));
+                $this->_oDb->query("DELETE FROM `sys_acl_levels_members` WHERE `IDLevel` = :id", array('id' => $aLevel['ID']));
+            }
+            $iNumber++;
+        }
+
+        parent::removeContent();
+        return $iNumber;
+    }
+	
+}
+
+	
+/** @} */