You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
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
|
|
}
|