JosephFerano/discord-bot-game
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rework check invite logic a little

Browse Source
This commit is contained in:
Joseph Ferano 2022-06-18 22:09:15 +07:00
parent 9e85a53de1
commit f5336fb9b7
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Bot/InviteTracker.fs
View File

@ -135,18 +135,14 @@ let private checkUserAlreadyInvited userId = async {
} }
let checkInviteAccepted (userId : uint64) = async { let checkInviteAccepted (userId : uint64) = async {
try
let! result = let! result =
connStr connStr
|> Sql.connect |> Sql.connect
|> Sql.parameters [ "did" , Sql.string (string userId) ] |> Sql.parameters [ "did" , Sql.string (string userId) ]
|> Sql.query "SELECT accepted FROM invited_user WHERE discord_id = @did" |> Sql.query "SELECT accepted FROM invited_user WHERE discord_id = @did"
|> Sql.executeRowAsync (fun read -> read.bool "accepted") |> Sql.executeAsync (fun read -> read.bool "accepted")
|> Async.AwaitTask |> Async.AwaitTask
return result return List.tryHead result |> Option.defaultValue false
with ex ->
printfn "%s %u" ex.Message userId
return false
} }
let private getInviteAttributions userId = let private getInviteAttributions userId =
@ -306,7 +302,7 @@ let acceptInvite (guild : DiscordGuild) (user : DiscordMember) =
task { task {
match! checkInviteAccepted user.Id with match! checkInviteAccepted user.Id with
| false -> | false ->
let! _ = markInvitedAccepted user.Id |> Async.Ignore do! markInvitedAccepted user.Id |> Async.Ignore
try try
let! invite = getInviteFromInvitedUser user.Id let! invite = getInviteFromInvitedUser user.Id
let! player = DbService.tryFindPlayer invite.Inviter let! player = DbService.tryFindPlayer invite.Inviter