using Microsoft.AspNetCore.Authentication.OAuth; using Npgsql; namespace IOL.BookmarkThing.Server.Utilities; public static class HandleGithubCreatingTicket { public static async Task Handle(OAuthCreatingTicketContext context, IConfiguration configuration) { var githubId = context.Identity?.FindFirst(p => p.Type == ClaimTypes.NameIdentifier)?.Value; var githubUsername = context.Identity?.FindFirst(p => p.Type == ClaimTypes.Name)?.Value; if (githubId.IsNullOrWhiteSpace() || githubUsername.IsNullOrWhiteSpace() || context.Identity == default) { return; } var claims = context.Identity.Claims.ToList(); foreach (var claim in claims) { context.Identity.RemoveClaim(claim); } var connstring = ConnectionStrings.AppDatabaseConnectionString(configuration); var connection = new NpgsqlConnection(connstring); Console.WriteLine($"HandleGithubCreatingTicket: Getting user mappings for github user: {githubId}"); var getMappedUserQuery = @$"SELECT u.id,u.username FROM github_user_mappings INNER JOIN users u on u.id = github_user_mappings.user_id WHERE github_id='{githubId}'"; await connection.OpenAsync(); await using var getMappedUserCommand = new NpgsqlCommand(getMappedUserQuery, connection); await using var reader = await getMappedUserCommand.ExecuteReaderAsync(); var handled = false; while (await reader.ReadAsync()) { try { var userId = reader.GetGuid(0); var username = reader.GetString(1); context.Identity.AddClaim(new Claim(AppClaims.USER_ID, userId.ToString())); context.Identity.AddClaim(new Claim(AppClaims.NAME, username)); Console.WriteLine($"HandleGithubCreatingTicket: Found mapping for github id {githubId} mapped to user id {userId}"); handled = true; } catch (Exception e) { Console.WriteLine(e); handled = false; } } await connection.CloseAsync(); if (!handled) { var userId = Guid.NewGuid(); var insertUserQuery = $@"INSERT INTO users VALUES ('{userId}', '{githubUsername}', '', '{DateTime.UtcNow}')"; await connection.OpenAsync(); await using var insertUserCommand = new NpgsqlCommand(insertUserQuery, connection); await insertUserCommand.ExecuteNonQueryAsync(); await connection.CloseAsync(); var insertMappingQuery = $@"INSERT INTO github_user_mappings VALUES ('{Guid.NewGuid()}', '{userId}', '{githubId}', '{DateTime.UtcNow}')"; await connection.OpenAsync(); await using var insertMappingCommand = new NpgsqlCommand(insertMappingQuery, connection); await insertMappingCommand.ExecuteNonQueryAsync(); await connection.CloseAsync(); context.Identity.AddClaim(new Claim(AppClaims.USER_ID, userId.ToString())); context.Identity.AddClaim(new Claim(AppClaims.NAME, githubUsername)); Console.WriteLine($"HandleGithubCreatingTicket: Created mapping for github id {githubId} mapped to user id {userId}"); } } }