added refresh tocken

internal/auth/repository.go

@@ -10,33 +10,49 @@ import (
 )
 
 type Repository struct {
-	collection *mongo.Collection
+	usersCollection         *mongo.Collection
+	refreshTokensCollection *mongo.Collection
 }
 
 func NewRepository(db *mongo.Database) *Repository {
 	return &Repository{
-		collection: db.Collection("users"),
+		usersCollection:         db.Collection("users"),
+		refreshTokensCollection: db.Collection("refresh_tokens"),
 	}
 }
 
 func (r *Repository) EnsureIndexes(ctx context.Context) error {
-	_, err := r.collection.Indexes().CreateOne(ctx, mongo.IndexModel{
+	_, err := r.usersCollection.Indexes().CreateOne(ctx, mongo.IndexModel{
 		Keys:    bson.D{{Key: "email", Value: 1}},
 		Options: options.Index().SetUnique(true),
 	})
+	if err != nil {
+		return err
+	}
+
+	_, err = r.refreshTokensCollection.Indexes().CreateMany(ctx, []mongo.IndexModel{
+		{
+			Keys:    bson.D{{Key: "token_hash", Value: 1}},
+			Options: options.Index().SetUnique(true),
+		},
+		{
+			Keys:    bson.D{{Key: "expires_at", Value: 1}},
+			Options: options.Index().SetExpireAfterSeconds(0),
+		},
+	})
 	return err
 }
 
-func (r *Repository) Create(ctx context.Context, user *User) error {
+func (r *Repository) CreateUser(ctx context.Context, user *User) error {
 	user.ID = bson.NewObjectID()
 	user.CreatedAt = time.Now().UTC()
-	_, err := r.collection.InsertOne(ctx, user)
+	_, err := r.usersCollection.InsertOne(ctx, user)
 	return err
 }
 
 func (r *Repository) FindByEmail(ctx context.Context, email string) (*User, error) {
 	var user User
-	err := r.collection.FindOne(ctx, bson.M{"email": email}).Decode(&user)
+	err := r.usersCollection.FindOne(ctx, bson.M{"email": email}).Decode(&user)
 	if err != nil {
 		return nil, err
 	}
@@ -45,9 +61,32 @@ func (r *Repository) FindByEmail(ctx context.Context, email string) (*User, erro
 
 func (r *Repository) FindByID(ctx context.Context, id bson.ObjectID) (*User, error) {
 	var user User
-	err := r.collection.FindOne(ctx, bson.M{"_id": id}).Decode(&user)
+	err := r.usersCollection.FindOne(ctx, bson.M{"_id": id}).Decode(&user)
 	if err != nil {
 		return nil, err
 	}
 	return &user, nil
 }
+
+//Refresh
+
+func (r *Repository) CreateRefreshToken(ctx context.Context, doc *RefreshTokenDoc) error {
+	doc.ID = bson.NewObjectID()
+	doc.CreatedAt = time.Now().UTC()
+	_, err := r.refreshTokensCollection.InsertOne(ctx, doc)
+	return err
+}
+
+func (r *Repository) FindRefreshTokenByHash(ctx context.Context, hash string) (*RefreshTokenDoc, error) {
+	var doc RefreshTokenDoc
+	err := r.refreshTokensCollection.FindOne(ctx, bson.M{"token_hash": hash}).Decode(&doc)
+	if err != nil {
+		return nil, err
+	}
+	return &doc, nil
+}
+
+func (r *Repository) DeleteRefreshToken(ctx context.Context, id bson.ObjectID) error {
+	_, err := r.refreshTokensCollection.DeleteOne(ctx, bson.M{"_id": id})
+	return err
+}