✨ feat: 添加微信登录功能

Author: imlinhanchao

game/backend/src/login/index.ts

@@ -29,5 +29,8 @@ export function getThirdPartyType() {
   if (config?.login?.steamApiKey) {
     types.push('steam');
   }
+  if (config?.login?.wechatAppId && config?.login?.wechatAppSecret) {
+    types.push('wechat');
+  }
   return types;
 }

game/backend/src/login/wechat.ts

@@ -0,0 +1,52 @@
+import { Request, Response } from "express";
+import utils from '../utils';
+import { saveUser } from './index';
+
+const appId = utils.config?.login.wechatAppId || '';
+
+export async function login(req: Request, res: Response) {
+  const domain = new URL(req.headers.referer || `${req.protocol}://${req.headers.host}`).host;
+  if (!appId) return res.end('微信OAuth 未配置，请联系管理员');
+  if (req.query['code']) {
+    const tokenData = await verify(req);
+    if (tokenData.access_token) {
+      const userInfo = await getUserInfo(tokenData.access_token, tokenData.openid);
+      req.session.player = await saveUser({
+        name: 'wechat-' + userInfo.openid,
+        nickname: userInfo.nickname,
+        id: userInfo.openid,
+        from: 'wechat',
+        avatar: userInfo.headimgurl,
+        ip: req.header('x-forwarded-for') || req.header('x-real-ip') || req.socket.remoteAddress || req.ip || ''
+      });
+      return res.redirect("/");
+    } else {
+      req.session.error = "登录验证失败，请重试";
+      return res.redirect("/login");
+    }
+  }
+  res.redirect(`https://open.weixin.qq.com/connect/oauth2/authorize?appid=${appId}&redirect_uri=https%3A%2F%2F${domain}%2Fapi%2Flogin%2Fwechat&response_type=code&scope=snsapi_login&state=STATE#wechat_redirect`);
+}
+
+async function verify(req: Request) {
+  const verifyReq = {
+    appid: utils.config?.login.wechatAppId,
+    secret: utils.config?.login.wechatAppSecret,
+    code: req.query['code'],
+    grant_type: 'authorization_code'
+  };
+  const params = new URLSearchParams(verifyReq as any);
+  const response = await fetch(`https://api.weixin.qq.com/sns/oauth2/access_token?${params}`, {
+    method: 'GET'
+  });
+  const data = await response.json();
+  return data;
+}
+
+async function getUserInfo(access_token: string, openid: string) {
+  const response = await fetch(`https://api.weixin.qq.com/sns/userinfo?access_token=${access_token}&openid=${openid}`, {
+    method: 'GET'
+  });
+  const data = await response.json();
+  return data;
+}

game/backend/src/routes/api.ts

@@ -3,6 +3,7 @@ import { Controller } from "../controller";
 import { login as fishpiLogin, register as fishpiRegister, updateUserInfo } from "../login/fishpi";
 import { login as steamLogin } from "../login/steam";
 import { login as githubLogin } from "../login/github";
+import { login as wechatLogin } from "../login/wechat";
 import { Record, RecordRepo, User, UserRepo, AppDataSource, PlayerStats, ManageRepo } from "@/entities";
 import { getPlayerStats, isConfigured } from "@/utils";
 import { FindOptionsWhere, Like } from "typeorm";
@@ -167,6 +168,7 @@ const createRoutes = (game: GameContext, gameName: string) => {
   router.get("/register/fishpi", fishpiRegister);
   router.get("/login/steam", steamLogin);
   router.get("/login/github", githubLogin);
+  router.get("/login/wechat", wechatLogin);
 
   router.post("/logout", (req: Request, res: Response) => {
     req.session.destroy((err) => {

game/backend/src/routes/config.ts

@@ -12,7 +12,7 @@ router.post("/", (req, res) => {
   const { 
     webport, goldenKey, host, port, username, password, database, prefix, marketKey,
     persistence_driver, persistence_host, persistence_port, persistence_username, persistence_password, persistence_database, persistence_prefix,
-    githubClientId, githubClientSecret, steamApiKey, steamMirror
+    githubClientId, githubClientSecret, steamApiKey, steamMirror, wechatAppId, wechatAppSecret
   } = req.body;
 
   const config: IConfig = {
@@ -39,6 +39,8 @@ router.post("/", (req, res) => {
       githubClientSecret,
       steamApiKey,
       steamMirror,
+      wechatAppId,
+      wechatAppSecret,
     }
   }
 

game/backend/types/index.d.ts

@@ -39,5 +39,7 @@ interface IConfig {
     githubClientSecret?: string;
     steamApiKey?: string;
     steamMirror?: string;
+    wechatAppId?: string;
+    wechatAppSecret?: string;
   }
 }