package repository import ( "database/sql" "errors" "fmt" "time" "git.jamestombleson.com/jtom38/newsbot-api/internal/domain" "github.com/huandu/go-sqlbuilder" ) const ( refreshTokenTableName = "RefreshTokens" ) type RefreshToken interface { Create(username string, token string) (int64, error) GetByUsername(name string) (domain.RefreshTokenEntity, error) DeleteById(id int64) (int64, error) } type RefreshTokenRepository struct { connection *sql.DB } func NewRefreshTokenRepository(conn *sql.DB) RefreshTokenRepository { return RefreshTokenRepository{ connection: conn, } } func (rt RefreshTokenRepository) Create(username string, token string) (int64, error) { dt := time.Now() builder := sqlbuilder.NewInsertBuilder() builder.InsertInto(refreshTokenTableName) builder.Cols("Username", "Token", "CreatedAt", "UpdatedAt") builder.Values(username, token, dt, dt) query, args := builder.Build() _, err := rt.connection.Exec(query, args...) if err != nil { return 0, err } return 1, nil } func (rt RefreshTokenRepository) GetByUsername(name string) (domain.RefreshTokenEntity, error) { builder := sqlbuilder.NewSelectBuilder() builder.Select("*").From(refreshTokenTableName).Where( builder.E("Username", name), ) query, args := builder.Build() rows, err := rt.connection.Query(query, args...) if err != nil { return domain.RefreshTokenEntity{}, err } data := rt.processRows(rows) if len(data) == 0 { return domain.RefreshTokenEntity{}, errors.New("no token found for user") } return data[0], nil } func (rt RefreshTokenRepository) DeleteById(id int64) (int64, error) { builder := sqlbuilder.NewDeleteBuilder() builder.DeleteFrom(refreshTokenTableName) builder.Where( builder.EQ("Id", id), ) query, args := builder.Build() rows, err := rt.connection.Exec(query, args...) if err != nil { return -1, err } return rows.RowsAffected() } func (rd RefreshTokenRepository) processRows(rows *sql.Rows) []domain.RefreshTokenEntity { items := []domain.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 := domain.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()