Ispravka QR koda za 2FA — generisanje na serveru kao base64 PNG

internal/db/sqlite/sesije.go

@@ -0,0 +1,74 @@
+package sqlite
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"time"
+
+	"ntech/internal/model"
+)
+
+type sqliteSesijeRepo struct{ db *sql.DB }
+
+// NoviSesijeRepo kreira SQLite implementaciju SesijeRepository
+func NoviSesijeRepo(db *sql.DB) *sqliteSesijeRepo {
+	return &sqliteSesijeRepo{db: db}
+}
+
+func (r *sqliteSesijeRepo) Kreiraj(ctx context.Context, korisnikID int64, token string, istice time.Time, totpPotvrdjeno bool) error {
+	potvrdjen := 1
+	if !totpPotvrdjeno {
+		potvrdjen = 0
+	}
+	_, err := r.db.ExecContext(ctx,
+		`INSERT INTO sesije (korisnik_id, token, totp_potvrdjeno, datum_isteka) VALUES (?, ?, ?, ?)`,
+		korisnikID, token, potvrdjen, istice)
+	if err != nil {
+		return fmt.Errorf("ntech: sesije.Kreiraj: %w", err)
+	}
+	return nil
+}
+
+func (r *sqliteSesijeRepo) DohvatiPoTokenu(ctx context.Context, token string) (*model.Sesija, error) {
+	s := &model.Sesija{}
+	var totpPotvrdjeno int
+	var datumIsteka, datumKreiranja time.Time
+	err := r.db.QueryRowContext(ctx,
+		`SELECT id, korisnik_id, token, totp_potvrdjeno, datum_isteka, datum_kreiranja
+		 FROM sesije WHERE token = ?`, token).
+		Scan(&s.ID, &s.KorisnikID, &s.Token, &totpPotvrdjeno, &datumIsteka, &datumKreiranja)
+	if err != nil {
+		return nil, fmt.Errorf("ntech: sesije.DohvatiPoTokenu: %w", err)
+	}
+	s.TotpPotvrdjeno = totpPotvrdjeno == 1
+	s.DatumIsteka = datumIsteka
+	s.DatumKreiranja = datumKreiranja
+	return s, nil
+}
+
+func (r *sqliteSesijeRepo) PotvrdiTotp(ctx context.Context, token string, novoIstice time.Time) error {
+	_, err := r.db.ExecContext(ctx,
+		`UPDATE sesije SET totp_potvrdjeno = 1, datum_isteka = ? WHERE token = ?`,
+		novoIstice, token)
+	if err != nil {
+		return fmt.Errorf("ntech: sesije.PotvrdiTotp: %w", err)
+	}
+	return nil
+}
+
+func (r *sqliteSesijeRepo) Obrisi(ctx context.Context, token string) error {
+	_, err := r.db.ExecContext(ctx, `DELETE FROM sesije WHERE token = ?`, token)
+	if err != nil {
+		return fmt.Errorf("ntech: sesije.Obrisi: %w", err)
+	}
+	return nil
+}
+
+func (r *sqliteSesijeRepo) ObrisiIstekle(ctx context.Context) error {
+	_, err := r.db.ExecContext(ctx, `DELETE FROM sesije WHERE datum_isteka < ?`, time.Now())
+	if err != nil {
+		return fmt.Errorf("ntech: sesije.ObrisiIstekle: %w", err)
+	}
+	return nil
+}