package logger

import (
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"
)

func testService(t *testing.T, prilagodi func(*Config)) *Service {
	t.Helper()
	cfg := DefaultConfig()
	cfg.Dir = t.TempDir()
	if prilagodi != nil {
		prilagodi(&cfg)
	}
	s, err := New(cfg)
	if err != nil {
		t.Fatalf("New: %v", err)
	}
	t.Cleanup(s.store.Close)
	return s
}

func TestServiceHandleUpisujeDnevnikIPamtiHeartbeat(t *testing.T) {
	s := testService(t, nil)
	t0 := time.Date(2026, 7, 25, 12, 0, 0, 0, time.UTC)
	s.now = func() time.Time { return t0 }

	s.Handle([]byte(`{"app":"TERMINIA","name":"TERM-001","mac":"1C:DB:D4:A6:A4:60","ip":"192.168.0.20","fw":"0.13.4","hb":1}`), "192.168.0.20")
	s.Handle([]byte("#TERMINIA TERM-001 1C:DB:D4:A6:A4:60 0.13.4\n[stampac] javio se\n"), "192.168.0.20")
	s.Handle([]byte{0x00, 0x01, 0x02, 0xff, 0xfe}, "9.9.9.9") // binarno smece

	if got := s.stats.Heartbeat.Load(); got != 1 {
		t.Errorf("heartbeata = %d", got)
	}
	if got := s.stats.LogPkts.Load(); got != 1 {
		t.Errorf("paketa dnevnika = %d", got)
	}
	if got := s.stats.WriteErrs.Load(); got != 0 {
		t.Errorf("gresaka upisa = %d", got)
	}
	lines, _ := s.store.Tail("1CDBD4A6A460", 10, "")
	if len(lines) != 1 || !strings.Contains(lines[0], "[stampac] javio se") {
		t.Errorf("dnevnik = %q", lines)
	}
}

func TestServiceStranicaUredjaja(t *testing.T) {
	s := testService(t, nil)
	t0 := time.Date(2026, 7, 25, 12, 0, 0, 0, time.UTC)
	s.now = func() time.Time { return t0 }
	s.Handle([]byte("#TERMINIA TERM-001 1C:DB:D4:A6:A4:60 0.13.4\n[stampac] javio se\n[kapija] prolaz\n"), "192.168.0.20")
	h := s.Handler()

	rec := httptest.NewRecorder()
	h.ServeHTTP(rec, httptest.NewRequest("GET", "/", nil))
	if rec.Code != http.StatusOK {
		t.Fatalf("/ = %d", rec.Code)
	}
	body := rec.Body.String()
	if !strings.Contains(body, "TERM-001") || !strings.Contains(body, "1C:DB:D4:A6:A4:60") {
		t.Errorf("spisak ne prikazuje uredjaj: %s", body)
	}

	rec = httptest.NewRecorder()
	h.ServeHTTP(rec, httptest.NewRequest("GET", "/uredjaj?k=1CDBD4A6A460&q=stampac", nil))
	if rec.Code != http.StatusOK {
		t.Fatalf("/uredjaj = %d", rec.Code)
	}
	body = rec.Body.String()
	if !strings.Contains(body, "[stampac] javio se") {
		t.Error("pretraga ne vraca ocekivanu liniju")
	}
	if strings.Contains(body, "[kapija] prolaz") {
		t.Error("pretraga vraca i linije koje ne odgovaraju upitu")
	}
}

func TestServiceNepoznatUredjajJeOznacen(t *testing.T) {
	s := testService(t, nil)
	s.Handle([]byte("goli tekst bez zaglavlja\n"), "192.168.0.77")

	rec := httptest.NewRecorder()
	s.Handler().ServeHTTP(rec, httptest.NewRequest("GET", "/", nil))
	body := rec.Body.String()
	if !strings.Contains(body, "nepoznat uređaj") {
		t.Errorf("uredjaj bez zaglavlja nije oznacen kao nepoznat:\n%s", body)
	}
	if !strings.Contains(body, "192.168.0.77") {
		t.Error("nije prikazana izvorna adresa nepoznatog uredjaja")
	}
}

func TestServiceTihUredjajJeCrven(t *testing.T) {
	s := testService(t, nil)
	t0 := time.Date(2026, 7, 25, 12, 0, 0, 0, time.UTC)
	s.now = func() time.Time { return t0 }
	s.Handle([]byte(`{"name":"TERM-001","mac":"1C:DB:D4:A6:A4:60","hb":1}`), "192.168.0.20")

	s.now = func() time.Time { return t0.Add(10 * time.Minute) } // 3x60 s je davno proslo
	rec := httptest.NewRecorder()
	s.Handler().ServeHTTP(rec, httptest.NewRequest("GET", "/", nil))
	if !strings.Contains(rec.Body.String(), `class="tih"`) {
		t.Error("tih uredjaj nije oznacen crvenim redom")
	}
}

func TestServiceHTMLEscapeIzPaketa(t *testing.T) {
	// Sadrzaj paketa je tudji tekst - ne sme da postane HTML.
	s := testService(t, nil)
	s.Handle([]byte("#TERMINIA <script>alert(1)</script> 1C:DB:D4:A6:A4:60 0.13.4\n<img src=x onerror=alert(2)>\n"), "1.2.3.4")

	for _, url := range []string{"/", "/uredjaj?k=1CDBD4A6A460"} {
		rec := httptest.NewRecorder()
		s.Handler().ServeHTTP(rec, httptest.NewRequest("GET", url, nil))
		if strings.Contains(rec.Body.String(), "<script>alert(1)</script>") ||
			strings.Contains(rec.Body.String(), "<img src=x") {
			t.Errorf("%s: HTML iz paketa nije ekranovan", url)
		}
	}
}

func TestServiceToken(t *testing.T) {
	s := testService(t, func(c *Config) { c.Token = "tajna" })
	h := s.Handler()

	rec := httptest.NewRecorder()
	h.ServeHTTP(rec, httptest.NewRequest("GET", "/", nil))
	if rec.Code != http.StatusForbidden {
		t.Errorf("bez tokena = %d, ocekivano 403", rec.Code)
	}

	rec = httptest.NewRecorder()
	h.ServeHTTP(rec, httptest.NewRequest("GET", "/?t=tajna", nil))
	if rec.Code != http.StatusOK {
		t.Errorf("sa tokenom = %d", rec.Code)
	}
}

func TestServiceAPIJson(t *testing.T) {
	s := testService(t, nil)
	s.Handle([]byte(`{"name":"TERM-001","mac":"1C:DB:D4:A6:A4:60","ip":"192.168.0.20","fw":"0.13.4","hb":1}`), "192.168.0.20")

	rec := httptest.NewRecorder()
	s.Handler().ServeHTTP(rec, httptest.NewRequest("GET", "/api/uredjaji", nil))
	if rec.Code != http.StatusOK {
		t.Fatalf("/api/uredjaji = %d", rec.Code)
	}
	if !strings.Contains(rec.Body.String(), `"mac": "1C:DB:D4:A6:A4:60"`) {
		t.Errorf("JSON = %s", rec.Body.String())
	}
}

func TestServiceRawTekst(t *testing.T) {
	s := testService(t, nil)
	s.Handle([]byte("#TERMINIA T 1C:DB:D4:A6:A4:60 0.13.4\nprva\ndruga\n"), "1.2.3.4")

	rec := httptest.NewRecorder()
	s.Handler().ServeHTTP(rec, httptest.NewRequest("GET", "/dnevnik.txt?k=1CDBD4A6A460", nil))
	if ct := rec.Header().Get("Content-Type"); !strings.HasPrefix(ct, "text/plain") {
		t.Errorf("Content-Type = %q", ct)
	}
	if !strings.Contains(rec.Body.String(), "prva") {
		t.Errorf("telo = %q", rec.Body.String())
	}
}

func TestConfigFromEnv(t *testing.T) {
	t.Setenv("LOGGER_UDP_PORT", "46465")
	t.Setenv("LOGGER_HTTP_PORT", "8081")
	t.Setenv("LOGGER_DIR", "/var/log/terminia")
	cfg, err := ConfigFromEnv()
	if err != nil {
		t.Fatalf("ConfigFromEnv: %v", err)
	}
	if cfg.UDPPort != 46465 || cfg.HTTPPort != 8081 || cfg.Dir != "/var/log/terminia" {
		t.Errorf("cfg = %+v", cfg)
	}

	t.Setenv("LOGGER_UDP_PORT", "nije-broj")
	if _, err := ConfigFromEnv(); err == nil {
		t.Error("neispravan port mora dati gresku, ne tihu podrazumevanu vrednost")
	}

	t.Setenv("LOGGER_UDP_PORT", "46465")
	t.Setenv("LOGGER_MAX_DEVICE_MB", "5000")
	t.Setenv("LOGGER_MAX_TOTAL_MB", "100")
	if _, err := ConfigFromEnv(); err == nil {
		t.Error("kvota po uredjaju veca od ukupne mora dati gresku")
	}
}

func TestDefaultConfigNijeNaPortuUredjaja(t *testing.T) {
	// 46464 je port na kom UREDJAJI slusaju discovery - prijemnik tamo ne sme.
	if DefaultConfig().UDPPort == 46464 {
		t.Error("podrazumevani UDP port se sudara sa discovery portom uredjaja")
	}
}