<template>
  <view class="container">
    <button class="wechat-btn" open-type="getUserInfo" @click="handleWechatLogin">
      <uni-icons type="weixin" color="#fff" size="30"></uni-icons>
      微信一鍵登錄
    </button>
  </view>
</template>

<script>
export default {
  methods: {
    async handleWechatLogin() {
      try {
        // 獲取微信授權(quán)碼
        const loginRes = await uni.login({
          provider: "weixin",
          onlyAuthorize: true, // 關(guān)鍵參數(shù)：僅授權(quán)不獲取用戶(hù)信息
        });
        
        // 調(diào)用后端登錄接口
        const res = await uni.request({
          url: "你的后端地址/weChatLogin", // 記得換成你的實(shí)際地址！
          method: 'POST',
          data: { socialCode: loginRes.code }
        });
        
        if (res.data.token) {
          // 登錄成功處理
          uni.showToast({ title: '登錄成功！', icon: 'success' });
        } else {
          // 需要綁定手機(jī)號(hào)
          this.goToBindPhone();
        }
      } catch (error) {
        uni.showToast({ title: '登錄失敗，請(qǐng)重試', icon: 'none' });
      }
    }
  }
}
</script>

@Tag(name = "微信登錄")
@RestController
public class SysLoginController {
    
    @Autowired
    private SysLoginService loginService;
    
    @Operation(summary = "微信授權(quán)登錄")
    @PostMapping("/weChatLogin")
    public AjaxResult weChatLogin(@RequestBody LoginBodyWeChat loginBody) {
        // 三步走：授權(quán) -> 獲取用戶(hù)信息 -> 生成令牌
        Map<String, String> authResult = weiXinApiService.authorize(loginBody.getSocialCode());
        Map<String, String> userInfo = weiXinApiService.getUserInfo(
            authResult.get("access_token"), 
            authResult.get("openid")
        );
        
        String token = loginService.weChatLogin(userInfo);
        
        // 返回令牌給前端
        return AjaxResult.success()
            .put(Constants.TOKEN, token)
            .put("unionid", userInfo.get("unionid"));
    }
}

@Component
public class WeiXinApiService {
    
    public Map<String, String> authorize(String socialCode) {
        String url = String.format(
            "https://api.weixin.qq.com/sns/oauth2/access_token?appid=%s&secret=%s&code=%s&grant_type=authorization_code", 
            appid, secret, socialCode
        );
        
        String response = HttpUtil.get(url);
        JSONObject jsonObject = JSONUtil.parseObj(response);
        
        // 錯(cuò)誤處理絕不能少！
        if (jsonObject.containsKey("errcode")) {
            throw new ServiceException("微信登錄失敗：" + jsonObject.get("errmsg"));
        }
        
        // 返回access_token和openid
        return Map.of(
            "access_token", jsonObject.getStr("access_token"),
            "openid", jsonObject.getStr("openid")
        );
    }
    public Map<String, String> getUserInfo(String accessToken, String openId) {
        String url = String.format("https://api.weixin.qq.com/sns/userinfo?access_token=%s&openid=%s", 
        accessToken, openId);
        String response = HttpUtil.get(url);
        JSONObject jsonObject = JSONUtil.parseObj(response);
        if (jsonObject.containsKey("errcode")) {
            throw new ServiceException("獲取微信用戶(hù)信息失?。? + jsonObject.get("errmsg"));
        }
        Map<String, String> result = new HashMap<>();
        result.put("nickname", jsonObject.getStr("nickname"));
        result.put("sex", jsonObject.getStr("sex"));
        result.put("headimgurl", jsonObject.getStr("headimgurl"));
        result.put("unionid", jsonObject.getStr("unionid"));
        result.put("openid", jsonObject.getStr("openid"));
        return result;
    }
}

public String weChatLogin(Map<String, String> userInfo) {
    // 用unionid查詢(xún)用戶(hù)
    SysUser user = userService.selectUserByOpenId(userInfo.get("unionid"));//這里可以根據(jù)自己的業(yè)務(wù)框架來(lái)處理
    
    if (user == null) {
        // 新用戶(hù)：自動(dòng)注冊(cè)
        user = createNewUser(userInfo);
        userService.insertUser(user);//這里可以根據(jù)自己的業(yè)務(wù)框架來(lái)處理
    }
    
    // 未綁定手機(jī)號(hào)？引導(dǎo)綁定
    if (StringUtils.isBlank(user.getPhonenumber())) {
        return ""; // 前端根據(jù)空token跳轉(zhuǎn)綁定頁(yè)面
    }
    
    // 執(zhí)行登錄，生成token
    return performLogin(user.getPhonenumber());//這里可以根據(jù)自己的業(yè)務(wù)框架來(lái)處理
}

@PostMapping("/bindPhoneNumber")
public AjaxResult bindPhoneNumber(@RequestBody LoginBodyBindPhone loginBody) {
    // 驗(yàn)證碼校驗(yàn) + 綁定手機(jī)號(hào)
    String token = loginService.bindPhoneNumber(loginBody);
    return AjaxResult.success().put(Constants.TOKEN, token);
}
@Data
public class LoginBodyBindPhone {

    @Schema(description = "手機(jī)號(hào)")
    private String phonenumber;
    @Schema(description = "微信用戶(hù)統(tǒng)一標(biāo)識(shí)")
    private String unionid;
    @Schema(description = "驗(yàn)證碼")
    private String code;
}