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 = '';
        if (is_array($data) && isset($data['invite_code'])) {
            $invite = trim($data['invite_code']);
        }
        if ($invite === '' && request()) {
            $invite = (string)request()->post('invite_code', '');
        }
        Factory::ensureBound($user, $invite);
    }
}