wticreatorstudio/Modules/ReferralSystem/Http/Controllers/ReferralSystemController.php

<?php

namespace Modules\ReferralSystem\Http\Controllers;

use Illuminate\Contracts\Support\Renderable;
use Illuminate\Http\Request;
use Illuminate\Routing\Controller;
use App\Models\User;
use Str;

class ReferralSystemController extends Controller
{

    public function index()
    {
        if(auth()->user()->role->type == 'staff'){

            $user = auth()->user();
            if($user->referral_code == NULL){
                $user->referral_code = $user->username ? Str::slug($user->username.''.$user->id) : Str::slug($user->first_name.''.$user->last_name.''.$user->id);
                $user->save();
            }


            $referrals = User::where('referrer_id',$user->id)->get();
            return view('referralsystem::index', compact('referrals'));
        } 

        return redirect()->back();
    }

    public function clientRegister($referral_code)
    {
        //check first if the referralcode exist
        $user = User::where('referral_code',$referral_code)->first();
        if($user){
            session(['register_type' => 'client', 'referral_code' => $referral_code]);
            return redirect()->to('/register');
        } 

        return redirect()->back();
    }

    public function staffRegister($referral_code)
    {
        //check first if the referralcode exist
        $user = User::where('referral_code',$referral_code)->first();
        if($user){
            session(['register_type' => 'staff', 'referral_code' => $referral_code]);
            return redirect()->to('/register');
        } 

        return redirect()->back();
    }

}