newsbot-api/internal/repository/refreshTokens.go

package repository

import (
	"context"
	"database/sql"
	"errors"
	"fmt"
	"time"

	"git.jamestombleson.com/jtom38/newsbot-api/internal/entity"
	"github.com/huandu/go-sqlbuilder"
)

const (
	refreshTokenTableName = "RefreshTokens"
)

type RefreshToken interface {
	Create(ctx context.Context, username string, token string) (int64, error)
	GetByUsername(ctx context.Context, name string) (entity.RefreshTokenEntity, error)
	DeleteById(ctx context.Context, id int64) (int64, error)
}

type RefreshTokenRepository struct {
	connection *sql.DB
}

func NewRefreshTokenRepository(conn *sql.DB) RefreshTokenRepository {
	return RefreshTokenRepository{
		connection: conn,
	}
}

func (rt RefreshTokenRepository) Create(ctx context.Context, username string, token string) (int64, error) {
	dt := time.Now()
	builder := sqlbuilder.NewInsertBuilder()
	builder.InsertInto(refreshTokenTableName)
	builder.Cols("Username", "Token", "CreatedAt", "UpdatedAt", "DeletedAt")
	builder.Values(username, token, dt, dt, time.Time{})
	query, args := builder.Build()

	_, err := rt.connection.ExecContext(ctx, query, args...)
	if err != nil {
		return 0, err
	}

	return 1, nil
}

func (rt RefreshTokenRepository) GetByUsername(ctx context.Context, name string) (entity.RefreshTokenEntity, error) {
	builder := sqlbuilder.NewSelectBuilder()
	builder.Select("*").From(refreshTokenTableName).Where(
		builder.E("Username", name),
	)

	query, args := builder.Build()
	rows, err := rt.connection.QueryContext(ctx, query, args...)
	if err != nil {
		return entity.RefreshTokenEntity{}, err
	}

	data := rt.processRows(rows)
	if len(data) == 0 {
		return entity.RefreshTokenEntity{}, errors.New("no token found for user")
	}

	return data[0], nil
}

func (rt RefreshTokenRepository) DeleteById(ctx context.Context, id int64) (int64, error) {
	builder := sqlbuilder.NewDeleteBuilder()
	builder.DeleteFrom(refreshTokenTableName)
	builder.Where(
		builder.EQ("Id", id),
	)

	query, args := builder.Build()
	rows, err := rt.connection.ExecContext(ctx, query, args...)
	if err != nil {
		return -1, err
	}

	return rows.RowsAffected()
}

func (rd RefreshTokenRepository) processRows(rows *sql.Rows) []entity.RefreshTokenEntity {
	items := []entity.RefreshTokenEntity{}

	for rows.Next() {
		var id int64
		var username string
		var token string
		var createdAt time.Time
		var updatedAt time.Time
		var deletedAt sql.NullTime

		err := rows.Scan(&id, &createdAt, &updatedAt, &deletedAt, &username, &token)
		if err != nil {
			fmt.Println(err)
		}

		item := entity.RefreshTokenEntity{
			ID:        id,
			Username:  username,
			Token:     token,
			CreatedAt: createdAt,
			UpdatedAt: updatedAt,
		}

		if deletedAt.Valid {
			item.DeletedAt = deletedAt.Time
		}

		items = append(items, item)
	}

	return items
}

//func (rt RefreshTokenRepository) Delete()