print-api/app/controller/common/Login.php

<?php

namespace app\controller\common;

use app\extra\basic\Base;
use app\extra\service\basic\SmsService;
use app\middleware\AuthMiddleware;
use app\model\saas\SaasShop;
use app\model\system\SystemUser;
use Hzdad\Codecheck\Codecheck;
use LinFly\Annotation\Route\Controller;
use LinFly\Annotation\Route\Route;
use Shopwwi\WebmanAuth\Auth;
use support\Request;
use support\Response;
use think\facade\Db;
use Tinywan\Captcha\Captcha;
use LinFly\Annotation\Route\Middleware;


#[Controller(prefix: "/api/login")]
class Login extends Base
{

    /**
     * 登陆
     * @param Request $request
     * @return Response
     */
    #[Route(path: "user",methods: "post")]
    public function setLogin(Request $request): Response
    {
        try {
            $param = $this->_valid([
                "username.require"  => trans("empty.user"),
                "password.require"  => trans("empty.passwd"),
                "code.require"      => trans("empty.code"),
                "key.require"       => trans("empty.data"),
            ],"post");
            if (!is_array($param)) return error($param);
            if (Captcha::check($param['code'],$param['key']) === false) return errorTrans("error.captcha");
            $map = ["is_deleted" => 0,"username" => $param['username']];
            [$state,$msg,$user] = $this->checkLogin($map,2,$param);
            if (!$state) return error($msg);
            return successTrans("success.login",get_object_vars((new Auth)->guard("admin")->login($user)));
        } catch (\Throwable $throwable) {
            return error($throwable->getMessage());
        }
    }

    /**
     * 手机号码登陆
     * @param Request $request
     * @return Response
     */
    #[Route(path: "mobile",methods: "post")]
    public function setLogin2Mobile(Request $request): Response
    {
        try {
            $param = $this->_valid([
                "mobile.require"    => trans("empty.mobile"),
                "code.require"      => trans("empty.code"),
                "scene.require"     => trans("empty.data"),
            ],"post");
            if (!is_array($param)) return error($param);
            $code = (new Codecheck)->mobile($param['mobile'])->scene($param['scene'])->code($param['code'])->check();
            if (!$code) return errorTrans("error.captcha");
            $map = ["is_deleted" => 0,"mobile" => $param['mobile']];
            [$state,$msg,$user] = $this->checkLogin($map);
            if (!$state) return error($msg);
            return successTrans("success.login",get_object_vars((new Auth)->guard("admin")->login($user)));
        } catch (\Throwable $throwable) {
            return error($throwable->getMessage());
        }
    }


    /**
     * 登录验证处理
     * @param array $map
     * @param int $type
     * @param array $param
     * @return array
     */
    protected function checkLogin(array $map = [],int $type = 1,array $param = []): array
    {
        $user = (new SystemUser)->where($map)->findOrEmpty();
        if ($user->isEmpty()) return [0,trans("error.user-empty"),[]];
        if ($user['status'] <> 1) return [0,trans("error.user-status"),[]];
        if ($user['type'] > 1) {
            $typeUser = $this->getTypeUser($user['agent_id']);
            if (empty($typeUser)) return [0,trans("empty.agent"),[]];
            if ($typeUser['status'] <> 0) return [0,trans("error.agent"),[]];
            if (time() > strtotime($typeUser['vip_end'])) return [0,trans("error.agent-out"),[]];
            $user['shop_name'] = $typeUser['shop_name'];
        }
        if ($type == 2) {
            if (md5($param['password'].$user['salt']) <> $user['password']) return [0,trans("error.passwd"),[]];
        }
        $user->login_at = getDateFull();
        $user->login_ip = request()->getRealIp();
        $user->login_num = Db::raw("login_num+1");
        $user->save();
        return [1,'success',$user->toArray()];
    }

    /**
     * 获取代理信息
     * @param int $agentId
     * @return array
     */
    protected function getTypeUser(int $agentId = 0): array
    {
        return (new SaasShop)->where("shop_id",$agentId)->findOrEmpty()->toArray();
    }

    /**
     * @return Response
     */
    #[Route(path: "profile",methods: "get"),Middleware(AuthMiddleware::class)]
    public function getLoginUser(): Response
    {
        try {
            $userData = (new Auth)->guard("admin")->user()->toArray();
            if (isset($userData['password'])) unset($userData['password']);
            $agent = (new SaasShop)->where("shop_id",$userData['agent_id'])->findOrEmpty();
            if (empty($agent['vip_end']))
            {
                $userData['vip_end'] = 0;
            } else {
                $userData['vip_end'] = strtotime($agent['vip_end']);
            }
            return successTrans("success.data",[
                "username"  => $userData['username'],
                "truename"  => $userData['truename'],
                "vip_at"    => $userData['vip_end'],
                "agent_id"  => $userData['agent_id'],
                "super"     => $userData['is_super'],
                "shop"      => $agent['shop_name']??'',
                "type"      => $userData['type']
            ]);
        } catch (\Throwable $exception){
            return error($exception->getMessage());
        }
    }

}