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

2026-06-02 22:29:53 +02:00
parent f918b76542
commit 2401f6d5ec
25 changed files with 2449 additions and 48 deletions
@@ -0,0 +1,212 @@
+package handler
+
+import (
+	"html/template"
+	"net/http"
+	"time"
+
+	"ntech/internal/auth"
+)
+
+const imeKolacica = "ntech_sesija"
+const trajanjeSeije = 7 * 24 * time.Hour
+const trajanjePredSeije = 5 * time.Minute
+
+// PrikazPrijave renderuje formu za prijavu
+func (h *Handler) PrikazPrijave(w http.ResponseWriter, r *http.Request) {
+	// ako nema korisnika, preusmeri na setup wizard
+	postoji, err := h.KorisniciRepo.PostojiIjedan(r.Context())
+	if err == nil && !postoji {
+		http.Redirect(w, r, "/setup", http.StatusSeeOther)
+		return
+	}
+
+	greska := r.URL.Query().Get("greska")
+	renderujStandaloneTemplate(w, "web/templates/stranice/prijava.html", map[string]any{
+		"Greska": greska,
+	})
+}
+
+// Prijava obrađuje POST formu za prijavu
+func (h *Handler) Prijava(w http.ResponseWriter, r *http.Request) {
+	if err := r.ParseForm(); err != nil {
+		http.Redirect(w, r, "/prijava?greska=1", http.StatusSeeOther)
+		return
+	}
+
+	korisnickoIme := r.FormValue("korisnicko_ime")
+	lozinka := r.FormValue("lozinka")
+
+	korisnik, err := h.KorisniciRepo.DohvatiPoImenu(r.Context(), korisnickoIme)
+	if err != nil || !korisnik.Aktivan {
+		http.Redirect(w, r, "/prijava?greska=1", http.StatusSeeOther)
+		return
+	}
+
+	if !auth.ProveriLozinku(korisnik.LozinkaHash, lozinka) {
+		http.Redirect(w, r, "/prijava?greska=1", http.StatusSeeOther)
+		return
+	}
+
+	token := auth.GenerisiToken()
+
+	if korisnik.TotpTajna != "" {
+		// kreira privremenu sesiju koja čeka TOTP verifikaciju
+		if err := h.SesijeRepo.Kreiraj(r.Context(), korisnik.ID, token, time.Now().Add(trajanjePredSeije), false); err != nil {
+			http.Redirect(w, r, "/prijava?greska=2", http.StatusSeeOther)
+			return
+		}
+		http.SetCookie(w, napraviKolacic(token, time.Now().Add(trajanjePredSeije)))
+		http.Redirect(w, r, "/prijava/totp", http.StatusSeeOther)
+		return
+	}
+
+	// direktna sesija bez TOTP
+	if err := h.SesijeRepo.Kreiraj(r.Context(), korisnik.ID, token, time.Now().Add(trajanjeSeije), true); err != nil {
+		http.Redirect(w, r, "/prijava?greska=2", http.StatusSeeOther)
+		return
+	}
+	http.SetCookie(w, napraviKolacic(token, time.Now().Add(trajanjeSeije)))
+	http.Redirect(w, r, "/dashboard", http.StatusSeeOther)
+}
+
+// PrikazTotp renderuje formu za unos TOTP koda
+func (h *Handler) PrikazTotp(w http.ResponseWriter, r *http.Request) {
+	kolacic, err := r.Cookie(imeKolacica)
+	if err != nil {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+	sesija, err := h.SesijeRepo.DohvatiPoTokenu(r.Context(), kolacic.Value)
+	if err != nil || sesija.TotpPotvrdjeno || time.Now().After(sesija.DatumIsteka) {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+
+	greska := r.URL.Query().Get("greska")
+	renderujStandaloneTemplate(w, "web/templates/stranice/totp_provera.html", map[string]any{
+		"Greska": greska,
+	})
+}
+
+// VerifikujTotp obrađuje POST formu za TOTP verifikaciju
+func (h *Handler) VerifikujTotp(w http.ResponseWriter, r *http.Request) {
+	kolacic, err := r.Cookie(imeKolacica)
+	if err != nil {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+
+	sesija, err := h.SesijeRepo.DohvatiPoTokenu(r.Context(), kolacic.Value)
+	if err != nil || sesija.TotpPotvrdjeno || time.Now().After(sesija.DatumIsteka) {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+
+	korisnik, err := h.KorisniciRepo.DohvatiPoID(r.Context(), sesija.KorisnikID)
+	if err != nil {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+
+	kod := r.FormValue("kod")
+	if !auth.VerifikujTotpKod(kod, korisnik.TotpTajna) {
+		http.Redirect(w, r, "/prijava/totp?greska=1", http.StatusSeeOther)
+		return
+	}
+
+	novoIstice := time.Now().Add(trajanjeSeije)
+	if err := h.SesijeRepo.PotvrdiTotp(r.Context(), kolacic.Value, novoIstice); err != nil {
+		http.Redirect(w, r, "/prijava?greska=2", http.StatusSeeOther)
+		return
+	}
+
+	http.SetCookie(w, napraviKolacic(kolacic.Value, novoIstice))
+	http.Redirect(w, r, "/dashboard", http.StatusSeeOther)
+}
+
+// PrikazSetupa renderuje setup wizard za prvog korisnika
+func (h *Handler) PrikazSetupa(w http.ResponseWriter, r *http.Request) {
+	postoji, err := h.KorisniciRepo.PostojiIjedan(r.Context())
+	if err == nil && postoji {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+	greska := r.URL.Query().Get("greska")
+	renderujStandaloneTemplate(w, "web/templates/stranice/setup.html", map[string]any{
+		"Greska": greska,
+	})
+}
+
+// SacuvajSetup kreira prvog superadmin korisnika
+func (h *Handler) SacuvajSetup(w http.ResponseWriter, r *http.Request) {
+	postoji, err := h.KorisniciRepo.PostojiIjedan(r.Context())
+	if err == nil && postoji {
+		http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+		return
+	}
+
+	if err := r.ParseForm(); err != nil {
+		http.Redirect(w, r, "/setup?greska=1", http.StatusSeeOther)
+		return
+	}
+
+	ime := r.FormValue("korisnicko_ime")
+	lozinka := r.FormValue("lozinka")
+	potvrda := r.FormValue("lozinka_potvrda")
+
+	if len(ime) < 3 || len(lozinka) < 8 || lozinka != potvrda {
+		http.Redirect(w, r, "/setup?greska=1", http.StatusSeeOther)
+		return
+	}
+
+	hash, err := auth.HashujLozinku(lozinka)
+	if err != nil {
+		http.Redirect(w, r, "/setup?greska=2", http.StatusSeeOther)
+		return
+	}
+
+	if _, err := h.KorisniciRepo.Kreiraj(r.Context(), ime, hash, "superadmin"); err != nil {
+		http.Redirect(w, r, "/setup?greska=2", http.StatusSeeOther)
+		return
+	}
+
+	http.Redirect(w, r, "/prijava?sacuvano=1", http.StatusSeeOther)
+}
+
+// Odjava briše sesiju i kolačić
+func (h *Handler) Odjava(w http.ResponseWriter, r *http.Request) {
+	if kolacic, err := r.Cookie(imeKolacica); err == nil {
+		_ = h.SesijeRepo.Obrisi(r.Context(), kolacic.Value)
+	}
+	http.SetCookie(w, &http.Cookie{
+		Name:    imeKolacica,
+		Value:   "",
+		Path:    "/",
+		Expires: time.Unix(0, 0),
+		MaxAge:  -1,
+	})
+	http.Redirect(w, r, "/prijava", http.StatusSeeOther)
+}
+
+func napraviKolacic(token string, istice time.Time) *http.Cookie {
+	return &http.Cookie{
+		Name:     imeKolacica,
+		Value:    token,
+		Path:     "/",
+		Expires:  istice,
+		HttpOnly: true,
+		SameSite: http.SameSiteStrictMode,
+	}
+}
+
+func renderujStandaloneTemplate(w http.ResponseWriter, putanja string, podaci any) {
+	tmpl, err := template.ParseFiles(putanja)
+	if err != nil {
+		http.Error(w, "Greška pri učitavanju stranice", http.StatusInternalServerError)
+		return
+	}
+	if err := tmpl.Execute(w, podaci); err != nil {
+		http.Error(w, "Greška pri prikazu stranice", http.StatusInternalServerError)
+	}
+}