bog/integration/sqliterepo.go

package integration

import (
	"caj-larsson/bog/domain"
	"database/sql"
	"errors"
	"github.com/mattn/go-sqlite3"
)

type SQLiteUserAgentRepository struct {
	db *sql.DB
}

func NewSQLiteUserAgentRepository(db *sql.DB) *SQLiteUserAgentRepository {
	return &SQLiteUserAgentRepository{
		db: db,
	}
}

func (r *SQLiteUserAgentRepository) Migrate() error {
	query := `
	CREATE TABLE IF NOT EXISTS useragent(
		id INTEGER PRIMARY KEY AUTOINCREMENT,
		name TEXT NOT NULL UNIQUE,
		lastseen text,
		allowance_time bigint,
        quota_kb bigint,
        quota_usage_kb bigint
	);
	`

	_, err := r.db.Exec(query)
	return err
}

func (r *SQLiteUserAgentRepository) Create(useragent domain.UserAgent) (*domain.UserAgent, error) {
	var record, err = fromEntity(useragent)
	if err != nil {

	}

	res, err := r.db.Exec(
		"INSERT INTO useragent(name, lastseen, allowance_time, quota_kb, quota_usage_kb) values(?,?,?,?,?)",
		record.Name, record.LastSeen, record.AllowanceSeconds, record.QuotaKB, record.QuotaUsedKB,
	)
	if err != nil {
		var sqliteErr sqlite3.Error
		if errors.As(err, &sqliteErr) {
			if errors.Is(sqliteErr.ExtendedCode, sqlite3.ErrConstraintUnique) {
				return nil, domain.ErrDuplicate
			}
		}
		return nil, err
	}

	id, err := res.LastInsertId()
	if err != nil {
		return nil, err
	}
	useragent.ID = id

	return &useragent, nil
}

func (r *SQLiteUserAgentRepository) All() ([]domain.UserAgent, error) {
	rows, err := r.db.Query("SELECT * FROM useragent")
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var all []domain.UserAgent
	for rows.Next() {
		var record UserAgentDBRecord
		if err := rows.Scan(&record.ID, &record.Name, &record.LastSeen, &record.AllowanceSeconds, &record.QuotaKB, &record.QuotaUsedKB); err != nil {
			return nil, err
		}
		var useragent, err = record.toEntity()

		if err != nil {
			return nil, err
		}
		all = append(all, *useragent)
	}
	return all, nil
}

func (r *SQLiteUserAgentRepository) GetByName(name string) (*domain.UserAgent, error) {
	row := r.db.QueryRow("SELECT id, name, lastseen, allowance_time, allowance_kb FROM useragent WHERE name = ?", name)

	var record UserAgentDBRecord
	if err := row.Scan(&record.ID, &record.Name, &record.LastSeen, &record.AllowanceSeconds, &record.QuotaKB, &record.QuotaUsedKB); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return nil, domain.ErrNotExists
		}
		return nil, err
	}

	var useragent, err = record.toEntity()

	if err != nil {
		return nil, err
	}
	return useragent, nil
}

func (r *SQLiteUserAgentRepository) Update(id int64, updated domain.UserAgent) (*domain.UserAgent, error) {
	if id == 0 {
		return nil, errors.New("invalid updated ID")
	}

	var record, err = fromEntity(updated)

	res, err := r.db.Exec("UPDATE useragent SET name = ?, lastseen = ?, allowance_time = ?, quota_kb = ?, quota_usage_kb = ? WHERE id = ?",
		record.Name, record.LastSeen, record.AllowanceSeconds, &record.QuotaKB, &record.QuotaUsedKB, id)

	if err != nil {
		return nil, err
	}

	rowsAffected, err := res.RowsAffected()
	if err != nil {
		return nil, err
	}

	if rowsAffected == 0 {
		return nil, domain.ErrUpdateFailed
	}

	return &updated, nil
}

func (r *SQLiteUserAgentRepository) Delete(id int64) error {
	res, err := r.db.Exec("DELETE FROM useragent WHERE id = ?", id)
	if err != nil {
		return err
	}

	rowsAffected, err := res.RowsAffected()
	if err != nil {
		return err
	}

	if rowsAffected == 0 {
		return domain.ErrDeleteFailed
	}

	return err
}