web/activity-oauth-svr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化discord的检查流程

Browse Source
This commit is contained in:
CounterFire2023 2023-06-15 12:24:42 +08:00
parent af312dc4d3
commit 66c737f0ea
5 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/controllers/discord.controller.ts
View File

@ -12,7 +12,7 @@ class DiscordController extends BaseController {
let { code, state } = req.params let { code, state } = req.params
if (code && state) { if (code && state) {
const stateArr = state.split('|') const stateArr = state.split('|')
const address = stateArr[0] const address = stateArr[0].toLowerCase()
const record = await AuthRecord.insertOrUpdate( const record = await AuthRecord.insertOrUpdate(
{ address, platform: 7 }, { address, platform: 7 },
{ address, platform: 7, $inc: { version: 1 } }, { address, platform: 7, $inc: { version: 1 } },

11
src/controllers/main.controller.ts
View File

@ -21,6 +21,7 @@ class MainController extends BaseController {
if (sign !== signCheck) { if (sign !== signCheck) {
throw new ZError(11, 'invalid sign') throw new ZError(11, 'invalid sign')
} }
address = address.toLowerCase()
let records = await AuthRecord.find({ address }) let records = await AuthRecord.find({ address })
let result: any = { let result: any = {
discord: {}, discord: {},
@ -47,11 +48,17 @@ class MainController extends BaseController {
} }
} }
if (result.discord.id && !result.discord.verified) { if (result.discord.id && !result.discord.verified) {
setImmediate(async () => {
try {
let verified = await new DiscordSvr().checkUserRole(result.discord.id) let verified = await new DiscordSvr().checkUserRole(result.discord.id)
if (verified) { if (verified) {
distcordRecord.condition = 1 logger.info('check role with verified: ' + verified)
result.discord.verified = 1 await AuthRecord.updateOne({ openId: result.discord.id, platform: 7 }, { $set: { condition: 1 } })
} }
} catch (err) {
logger.info('check role with err: ' + err.message || err)
}
})
} }
return result return result
} }

2
src/controllers/twitter.controller.ts
View File

@ -13,7 +13,7 @@ class TwitterController extends BaseController {
const { code, state } = req.params const { code, state } = req.params
if (code && state) { if (code && state) {
const stateArr = state.split('|') const stateArr = state.split('|')
const address = stateArr[0] const address = stateArr[0].toLowerCase()
const record = await AuthRecord.insertOrUpdate( const record = await AuthRecord.insertOrUpdate(
{ address, platform: 4 }, { address, platform: 4 },
{ address, platform: 4, $inc: { version: 1 } }, { address, platform: 4, $inc: { version: 1 } },

1
src/modules/AuthRecord.ts
View File

@ -15,6 +15,7 @@ export enum PlatEnum {
@dbconn() @dbconn()
@index({ address: 1, platform: 1 }, { unique: true }) @index({ address: 1, platform: 1 }, { unique: true })
@index({ address: 1 }, { unique: false })
@modelOptions({ @modelOptions({
schemaOptions: { collection: 'auth_record', timestamps: true }, schemaOptions: { collection: 'auth_record', timestamps: true },
options: { allowMixed: Severity.ALLOW }, options: { allowMixed: Severity.ALLOW },

1
templates/discord_redirect.ejs
View File

@ -15,6 +15,7 @@
<body> <body>
<script> <script>
(function() { (function() {
// window.opener.postMessage('I am ready');
window.close(); window.close();
})(); })();
</script> </script>