Feat: add session

implements/bunDatabase.go

@@ -1,4 +1,69 @@
 package implements
 
+import (
+	"context"
+
+	"gitea.konchin.com/go2025/backend/models"
+	"gitea.konchin.com/go2025/backend/tracing"
+	"github.com/uptrace/bun"
+	"go.uber.org/zap"
+)
+
 type BunDatabase struct {
+	db *bun.DB
+}
+
+func NewBunDatabase(db *bun.DB) *BunDatabase {
+	return &BunDatabase{db: db}
+}
+
+func (self *BunDatabase) GetSession(
+	ctx context.Context,
+	userId string,
+) (models.Session, error) {
+	ret := models.Session{
+		UserId: userId,
+	}
+	err := self.db.NewSelect().
+		Model(&ret).
+		WherePK().
+		Scan(ctx)
+	if err != nil {
+		return models.Session{}, err
+	}
+	return ret, nil
+}
+
+func (self *BunDatabase) UpdateRefreshToken(
+	ctx context.Context,
+	userId string,
+) (models.Session, error) {
+	ret := models.Session{
+		UserId: userId,
+	}
+	err := self.db.NewSelect().
+		Model(&ret).
+		WherePK().
+		Scan(ctx)
+	if err != nil {
+		return models.Session{}, err
+	}
+
+	if err := ret.RotateRefreshToken(); err != nil {
+		tracing.Logger.Ctx(ctx).
+			Error("failed to rotate refresh token",
+				zap.Error(err))
+		return models.Session{}, err
+	}
+
+	err = self.db.NewUpdate().
+		Model((*models.Session)(nil)).
+		Set("refresh_token = ?", ret.RefreshToken).
+		Where("user_id = ?", ret.UserId).
+		Returning("*").
+		Scan(ctx, &ret)
+	if err != nil {
+		return models.Session{}, err
+	}
+	return ret, nil
 }