package com.example.controller;

import cn.com.digitalhainan.user.context.UserInfoContext;
import cn.com.digitalhainan.user.model.BaseUserInfo;
import cn.com.digitalhainan.user.model.UserAccount;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * 用户上下文使用示例
 * 通过 UserInfoContext 获取当前登录用户信息
 */
@RestController
public class UserContextDemoController {

    @GetMapping("/api/demo/current-user")
    public String getCurrentUser() {
        // 获取当前用户账号
        UserAccount account = UserInfoContext.getCurrentUserAccount();
        if (account == null) {
            return "未登录";
        }

        // 获取基础用户信息
        BaseUserInfo userInfo = account.getUserInfo(BaseUserInfo.class);
        String userId = userInfo.getUserId();
        String nickName = userInfo.getNickName();
        String phone = userInfo.getPhone();
        String userAccount = userInfo.getAccount();

        return "当前用户: " + nickName + " (" + userId + ")";
    }
}

package com.example.controller;

import cn.com.digitalhainan.user.annotation.LoginUser;
import cn.com.digitalhainan.user.context.UserInfoContext;
import cn.com.digitalhainan.user.model.BaseUserInfo;
import cn.com.digitalhainan.user.model.UserAccount;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * @LoginUser 注解使用示例
 * 标注在方法或类上，自动加载用户信息到 UserInfoContext
 */
@RestController
public class LoginUserDemoController {

    /**
     * 基础用法：加载用户基本信息
     */
    @LoginUser
    @GetMapping("/api/demo/profile")
    public String getProfile() {
        UserAccount account = UserInfoContext.getCurrentUserAccount();
        BaseUserInfo userInfo = account.getUserInfo(BaseUserInfo.class);
        return userInfo.getNickName();
    }

    /**
     * 加载用户角色权限
     */
    @LoginUser(getRoleAndPermission = true)
    @GetMapping("/api/demo/roles")
    public String getRoles() {
        UserAccount account = UserInfoContext.getCurrentUserAccount();
        return account.getRoleAndPermission().toString();
    }

    /**
     * 加载用户所有组织信息
     */
    @LoginUser(getAllOrgInfo = true)
    @GetMapping("/api/demo/orgs")
    public String getOrgs() {
        UserAccount account = UserInfoContext.getCurrentUserAccount();
        return account.getAllOrg(Object.class).toString();
    }

    /**
     * 跳过用户缓存（实时查询）
     */
    @LoginUser(skipUserCache = true)
    @GetMapping("/api/demo/realtime")
    public String getRealtimeInfo() {
        UserAccount account = UserInfoContext.getCurrentUserAccount();
        return account.getUserInfo(BaseUserInfo.class).getAccount();
    }
}

package com.example.controller;

import cn.com.digitalhainan.user.annotation.LoginUser;
import cn.com.digitalhainan.user.utils.AuthorizeUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * 权限校验使用示例
 */
@RestController
public class AuthDemoController {

    @Autowired
    private AuthorizeUtils authorizeUtils;

    @LoginUser(getRoleAndPermission = true)
    @GetMapping("/api/demo/admin")
    public String adminOnly() {
        if (!authorizeUtils.hasRole("ADMIN")) {
            return "无权访问";
        }
        return "管理员操作";
    }
}

package com.example.auth;

import cn.com.digitalhainan.ms.common.web.security.auth.AccessParameters;
import cn.com.digitalhainan.security.auth.AbstractAuthentication;
import cn.com.digitalhainan.security.auth.AuthInfo;
import jakarta.servlet.http.HttpServletRequest;

/**
 * 自定义认证实现示例
 */
public class CustomAuthentication extends AbstractAuthentication {

    @Override
    public AuthInfo doAuthentication(HttpServletRequest request, AccessParameters accessParameters) {
        // 自定义认证逻辑：解析 Token，查询用户信息
        AuthInfo authInfo = new AuthInfo();
        String token = parseToken(request);
        // ... 验证 token 并填充 authInfo
        authInfo.setUserId("user-id");
        authInfo.setAccount("username");
        authInfo.setNickName("昵称");
        return authInfo;
    }

    @Override
    public void doLogout(HttpServletRequest request) {
        // 登出清理逻辑
    }
}

package com.example.token;

import cn.com.digitalhainan.security.auth.token.strategy.TokenStrategy;

/**
 * 自定义 Token 生成策略
 */
public class CustomTokenStrategy implements TokenStrategy {

    @Override
    public String generateToken() {
        // 自定义 Token 生成逻辑
        return "custom-token-" + System.currentTimeMillis();
    }
}

package com.example.user;

import cn.com.digitalhainan.user.manager.UserManager;
import cn.com.digitalhainan.user.model.BaseUserInfo;
import cn.com.digitalhainan.user.model.RoleAndPermission;

/**
 * 自定义用户管理实现
 */
public class CustomUserManager implements UserManager {

    @Override
    public <T extends BaseUserInfo> T getUserInfoById(String userId) {
        // 从自定义数据源获取用户信息
        return null;
    }

    @Override
    public Object getUserAllOrg(String userId) {
        return null;
    }

    @Override
    public Object getUserOrgByAppKey(String userId) {
        return null;
    }

    @Override
    public RoleAndPermission getRolesAndPermissions(String userId) {
        return null;
    }
}