Save credentials for google in the database

starter/google/database.go

@@ -2,10 +2,14 @@ package google
 
 import (
 	"context"
+
 	"git.gorlug.de/code/ersteller"
 	google_http "git.gorlug.de/code/ersteller/authentication/google/http"
 	"git.gorlug.de/code/ersteller/starter/ent"
+	"git.gorlug.de/code/ersteller/starter/ent/googleauth"
+	"git.gorlug.de/code/ersteller/starter/ent/schema"
 	"git.gorlug.de/code/ersteller/starter/ent/user"
+	"golang.org/x/oauth2"
 )
 
 type Database struct {
@@ -30,3 +34,19 @@ func (d *Database) CreateUser(ctx context.Context, email string) (int, error) {
 	}
 	return newUser.ID, nil
 }
+
+func (d *Database) SaveCredentials(ctx context.Context, userId int, token *oauth2.Token) error {
+	existing, err := d.db.GoogleAuth.Query().Where(googleauth.HasUserWith(user.ID(userId))).Only(ctx)
+	if err == nil {
+		_, err = existing.Update().SetCredentials(schema.Credentials{Token: *token}).Save(ctx)
+		return err
+	}
+	if !ent.IsNotFound(err) {
+		return err
+	}
+	_, err = d.db.GoogleAuth.Create().
+		SetUserID(userId).
+		SetCredentials(schema.Credentials{Token: *token}).
+		Save(ctx)
+	return err
+}