Souke/frontend/models/AddEtablissementForm.php

<?php

namespace frontend\models;

use Yii;
use yii\base\Model;
use common\models\UserEtablissement;
use common\models\Etablissement ;
use yii\helpers\Html ;

/**
 * ContactForm is the model behind the contact form.
 */
class AddEtablissementForm extends Model
{
    public $id_etablissement;
    public $code ;

    /**
     * @inheritdoc
     */
    public function rules()
    {
        return [
            ['id_etablissement','integer'],
            ['id_etablissement', 'required'],
            ['id_etablissement', function($attribute, $params) {
            
                $etablissement = Etablissement::findOne($this->id_etablissement) ;
                if(!$etablissement)
                {
                    $this->addError($attribute, 'Cet établissement n\'existe pas.') ;
                }
                
                $user_etablissement_exist = UserEtablissement::find()
                ->where([
                    'id_user' => Yii::$app->user->identity->id, 
                    'id_etablissement' => $this->id_etablissement,
                    'actif' => 1
                 ])
                ->one() ;
                
                if($user_etablissement_exist)
                {
                    $this->addError($attribute, 'Cet établissement est déjà sur votre tableau de bord.') ;
                }
            }],
            ['code', 'required', 'message' => 'Champs obligatoire', 'when' => function($model) {
                $etablissement = Etablissement::findOne($this->id_etablissement) ;
                if($etablissement)
                {
                    return strlen($etablissement->code) ;
                }
                else {
                    return false ;
                }
            }],
            ['code', function($attribute, $params) {
                $code = $this->$attribute ;
                $etablissement = Etablissement::findOne($this->id_etablissement) ;
                
                if($etablissement && strtolower(trim($code)) != strtolower(trim($etablissement->code))) 
                {
                     $this->addError($attribute, 'Code incorrect');
                }
            }],
        ];
    }

    /**
     * @inheritdoc
     */
    public function attributeLabels()
    {
        return [
            'id_etablissement' => 'Établissement',
            'code' => 'Code',
        ];
    }

    /**
     * Sends an email to the specified email address using the information collected by this model.
     *
     * @param  string  $email the target email address
     * @return boolean whether the email was sent
     */
    public function add()
    {       
        $etablissement = Etablissement::findOne($this->id_etablissement) ;
        
        $user_etablissement_exist = UserEtablissement::find()
            ->where([
                'id_user' => Yii::$app->user->identity->id, 
                'id_etablissement' => $this->id_etablissement,
                'actif' => 0
             ])
            ->one() ;
        
        if($user_etablissement_exist)
        {
            $user_etablissement_exist->actif = 1 ;
            $user_etablissement_exist->save() ;
        }
        else {
            $user_etablissement = new UserEtablissement() ;
            $user_etablissement->id_user = Yii::$app->user->identity->id ;
            $user_etablissement->id_etablissement = $this->id_etablissement ;
            $user_etablissement->credit = 0 ;
            $user_etablissement->actif = 1 ;
            $user_etablissement->save() ;
        }
        
        Yii::$app->session->setFlash('success', 'Le producteur <strong>'.Html::encode($etablissement->nom).'</strong> a bien été ajoutée à votre tableau de bord.') ;
    }
}