package users import ( "context" "fmt" "time" "gitea.ravianand.me/Dan6erbond/listy/core" "gitea.ravianand.me/Dan6erbond/listy/models" "github.com/zmb3/spotify/v2" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" "golang.org/x/oauth2" ) func GetUsers(ctx context.Context, app *core.App) ([]*models.User, error) { coll := app.MongoClient.Database("listy").Collection("users") filter := bson.D{} opts := options.Find() cursor, err := coll.Find(context.TODO(), filter, opts) if err != nil { return nil, fmt.Errorf("couldn't fetch users %v", err) } var results []*models.User if err = cursor.All(ctx, &results); err != nil { return nil, fmt.Errorf("couldn't fetch users %v", err) } return results, nil } func GetUser(ctx context.Context, app *core.App, spotifyID string) (*models.User, error) { coll := app.MongoClient.Database("listy").Collection("users") filter := bson.D{{"spotifyid", spotifyID}} opts := options.FindOne() var result models.User err := coll.FindOne(context.TODO(), filter, opts).Decode(&result) if err != nil { return nil, fmt.Errorf("couldn't fetch user %v", err) } return &result, nil } func GetSpotifyClient(ctx context.Context, app *core.App, user *models.User) (*spotify.Client, error) { coll := app.MongoClient.Database("listy").Collection("users") if user.Token.Expiry.Before(time.Now()) { // expired so let's update it src := app.SpotifyOAuthConfig.TokenSource(ctx, &user.Token) newToken, err := src.Token() // this actually goes and renews the tokens if err != nil { panic(err) } if newToken.AccessToken != user.Token.AccessToken { _, err := coll.UpdateByID(ctx, user.ID, bson.D{{"$set", bson.D{{"token", *newToken}}}}) if err != nil { return nil, fmt.Errorf("couldn't update token %v", err) } user.Token = *newToken } } client := spotify.New(app.SpotifyAuth.Client(ctx, &user.Token)) return client, nil } func SaveUserToken(ctx context.Context, app *core.App, spotifyID string, token *oauth2.Token) (*models.User, error) { coll := app.MongoClient.Database("listy").Collection("users") result, err := GetUser(ctx, app, spotifyID) if err != nil && err != mongo.ErrNoDocuments { return nil, fmt.Errorf("couldn't fetch user %v", err) } if err == mongo.ErrNoDocuments { doc := models.User{Token: *token, SpotifyID: spotifyID} _, err = coll.InsertOne(context.TODO(), doc) if err != nil { return nil, fmt.Errorf("couldn't write user %v", err) } } else { _, err := coll.UpdateByID(ctx, result.ID, bson.D{{"$set", bson.D{{"token", *token}}}}) if err != nil { return nil, fmt.Errorf("couldn't write user %v", err) } result.Token = *token } return result, nil }