Opendistrib/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 [
            ['code', 'required', 'message' => 'Champs obligatoire'],
        ];
    }

    /**
     * @inheritdoc
     */
    public function attributeLabels()
    {
        return [
        	'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(['code' => $this->code]) ;
        
        if($etablissement)
        {
            $user_etablissement_exist = UserEtablissement::find()
                ->where(['id_user'=>Yii::$app->user->identity->id, 'id_etablissement' =>$etablissement->id])
                ->one() ;
        
            if(!$user_etablissement_exist)
            {
                $user_etablissement = new UserEtablissement() ;
                $user_etablissement->id_user = Yii::$app->user->identity->id ;
                $user_etablissement->id_etablissement = $etablissement->id ;
                $user_etablissement->save() ;
                
                Yii::$app->session->setFlash('success', 'La boulangerie <strong>'.Html::encode($etablissement->nom).'</strong> a bien été ajoutée à votre tableau de bord.') ;
            }
            else {
                Yii::$app->session->setFlash('error', 'Cette boulangerie est déjà sur votre tableau de bord.') ;
            }
        }
        else {
            Yii::$app->session->setFlash('error', 'Aucun établissement ne possède ce code d\'accès.') ;
        }
        
    }
}