-
Notifications
You must be signed in to change notification settings - Fork 439
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
(BIDS-2428) improve email change flow
- Loading branch information
1 parent
f5ef71f
commit 325eedb
Showing
4 changed files
with
117 additions
and
62 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| -- +goose Up | ||
| -- +goose StatementBegin | ||
| SELECT 'up SQL query - add column stripe_email_pending'; | ||
| ALTER TABLE users ADD COLUMN IF NOT EXISTS stripe_email_pending BOOLEAN NOT NULL DEFAULT FALSE; | ||
| -- +goose StatementEnd | ||
|
|
||
| -- +goose Down | ||
| -- +goose StatementBegin | ||
| SELECT 'down SQL query - remove column stripe_email_pending'; | ||
| ALTER TABLE users DROP COLUMN IF EXISTS stripe_email_pending; | ||
| -- +goose StatementEnd |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,92 @@ | ||
| package services | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "eth2-exporter/db" | ||
| "eth2-exporter/utils" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
| "net/url" | ||
| "sync" | ||
| "time" | ||
|
|
||
| "github.com/lib/pq" | ||
| ) | ||
|
|
||
| func stripeEmailUpdater(wg *sync.WaitGroup) { | ||
| firstRun := true | ||
| for { | ||
| // fetch all users with pending stripe email updates | ||
| var pendingUsers []struct { | ||
| Email string `db:"email"` | ||
| StripeCustomerId string `db:"stripe_customer_id"` | ||
| } | ||
| err := db.FrontendReaderDB.Select(&pendingUsers, "SELECT email, COALESCE(stripe_customer_id. '') FROM users WHERE stripe_email_pending") | ||
| if err != nil { | ||
| utils.LogError(err, "error getting pending users for stripe email update service", 0) | ||
| time.Sleep(time.Second * 10) | ||
| continue | ||
| } | ||
|
|
||
| // update stripe customer email | ||
| var updatedUsers []string | ||
| for _, user := range pendingUsers { | ||
| if user.StripeCustomerId == "" { | ||
| utils.LogError(fmt.Errorf("user has no stripe_customer_id"), "error updating stripe customer email, this should never happen", 0, map[string]interface{}{"email": user.Email}) | ||
| continue | ||
| } | ||
| err := updateStripeCustomerEmail(user.StripeCustomerId, user.Email) | ||
| if err != nil { | ||
| utils.LogError(err, "error updating stripe customer email", 0, map[string]interface{}{"email": user.Email, "stripe_customer_id": user.StripeCustomerId}) | ||
| continue | ||
| } | ||
| updatedUsers = append(updatedUsers, user.Email) | ||
| } | ||
|
|
||
| // set stripe_email_pending flag to false for all users that were updated | ||
| if len(updatedUsers) > 0 { | ||
| _, err := db.FrontendWriterDB.Exec("UPDATE users SET stripe_email_pending = false WHERE email = ANY($1)", pq.Array(updatedUsers)) | ||
| if err != nil { | ||
| utils.LogError(err, "error setting stripe_email_pending flag false for users, stripe email was updated", 0, map[string]interface{}{"emails": updatedUsers}) | ||
| time.Sleep(time.Second * 10) | ||
| continue | ||
| } | ||
| } | ||
|
|
||
| if firstRun { | ||
| wg.Done() | ||
| firstRun = false | ||
| } | ||
| time.Sleep(time.Minute) | ||
| } | ||
| } | ||
|
|
||
| func updateStripeCustomerEmail(stripeCustomerId, newEmail string) error { | ||
| // see https://stripe.com/docs/api/customers/update | ||
| apiEndpoint := "https://api.stripe.com/v1/customers/" + stripeCustomerId | ||
|
|
||
| data := url.Values{} | ||
| data.Set("email", newEmail) | ||
| req, err := http.NewRequest(http.MethodPost, apiEndpoint, bytes.NewBufferString(data.Encode())) | ||
| if err != nil { | ||
| return fmt.Errorf("error creating email change request for stripe: %w", err) | ||
| } | ||
| req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", utils.Config.Frontend.Stripe.SecretKey)) | ||
| req.Header.Set("Content-Type", "application/x-www-form-urlencoded") | ||
|
|
||
| httpClient := http.Client{Timeout: time.Second * 10} | ||
| resp, err := httpClient.Do(req) | ||
| if err != nil { | ||
| return fmt.Errorf("error sending request to stripe: %w", err) | ||
| } | ||
| defer resp.Body.Close() | ||
| if resp.StatusCode != http.StatusOK { | ||
| body, err := io.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return fmt.Errorf("error updating email in stripe, also could not read body: %w", err) | ||
| } | ||
| return fmt.Errorf("error updating email in stripe: %w; body: %v", err, string(body)) | ||
| } | ||
| return nil | ||
| } |