YetiForceCompany/YetiForceCRM

<?php
/* +***********************************************************************************
 * The contents of this file are subject to the vtiger CRM Public License Version 1.0
 * ("License"); You may not use this file except in compliance with the License
 * The Original Code is:  vtiger CRM Open Source
 * The Initial Developer of the Original Code is vtiger.
 * Portions created by vtiger are Copyright (C) vtiger.
 * All Rights Reserved.
 * Contributor(s): YetiForce S.A.
 * *********************************************************************************** */

class Users_MassSave_Action extends Vtiger_MassSave_Action
{
    /** {@inheritdoc} */
    public function checkPermission(App\Request $request)
    {
        $currentUserModel = Users_Record_Model::getCurrentUserModel();
        if (!$currentUserModel->isAdminUser()) {
            throw new \App\Exceptions\NoPermitted('LBL_PERMISSION_DENIED', 406);
        }
    }

    /** {@inheritdoc} */
    public function process(App\Request $request)
    {
        $moduleName = $request->getModule();
        $recordModels = $this->getRecordModelsFromRequest($request);
        foreach ($recordModels as $recordId => $recordModel) {
            if (\App\Privilege::isPermitted($moduleName, 'Save', $recordId)) {
                $recordModel->save();
            }
        }

        $response = new Vtiger_Response();
        $response->setResult(true);
        $response->emit();
    }

    /**
     * Function to get the record model based on the request parameters.
     *
     * @param \App\Request $request
     *
     * @return Vtiger_Record_Model or Module specific Record Model instance
     */
    public function getRecordModelsFromRequest(App\Request $request)
    {
        $moduleName = $request->getModule();
        $moduleModel = Vtiger_Module_Model::getInstance($moduleName);
        $recordIds = self::getRecordsListFromRequest($request);
        if (empty($recordIds) && 'all' === $request->getRaw('selected_ids')) {
            $recordIds = (new \App\Db\Query())->select(['id'])->from('vtiger_users')->column();
        }
        $recordModels = [];
        $fieldModelList = $moduleModel->getFields();
        foreach ($recordIds as $recordId) {
            $recordModel = Vtiger_Record_Model::getInstanceById($recordId, $moduleModel);
            $recordModel->setId($recordId);
            if (!$recordModel->isEditable()) {
                $recordModels[$recordId] = false;
                continue;
            }
            foreach ($fieldModelList as $fieldName => $fieldModel) {
                if ($fieldModel->isWritable() && $request->has($fieldName)) {
                    $fieldModel->getUITypeModel()->setValueFromRequest($request, $recordModel);
                }
            }
            $recordModels[$recordId] = $recordModel;
        }
        return $recordModels;
    }
}