process_sheet/fastadmin/application/common/behavior/PsFactory.php

<?php

namespace app\common\behavior;

use app\common\model\ps\Factory;
use app\common\model\User;

/**
 * 注册成功后建厂或凭邀请码入厂
 */
class PsFactory
{
    /**
     * ThinkPHP 行为默认入口
     * @param User  $user
     * @param mixed $data
     */
    public function run($user, $data = [])
    {
        $this->userRegisterSuccessed($user, $data);
    }

    /**
     * @param User  $user
     * @param mixed $data
     */
    public function userRegisterSuccessed($user, $data = [])
    {
        if (!$user instanceof User) {
            return;
        }
        $invite = '';
        $factoryName = '';
        if (is_array($data) && isset($data['invite_code'])) {
            $invite = trim($data['invite_code']);
        }
        if (is_array($data) && isset($data['factory_name'])) {
            $factoryName = trim((string)$data['factory_name']);
        }
        if (request()) {
            if ($invite === '') {
                $invite = (string)request()->post('invite_code', '');
            }
            if ($factoryName === '') {
                $factoryName = trim((string)request()->post('factory_name', ''));
            }
        }
        Factory::ensureBound($user, $invite, $factoryName);
    }
}