package webui

import (
	"context"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"

	"tea.karlcom.de/karl/bucketeer/internal/store"
)

type fakeStore struct {
	lastFilter store.AssetFilter
}

func (f *fakeStore) ListAssets(ctx context.Context, fl store.AssetFilter) ([]store.Asset, error) {
	f.lastFilter = fl
	return []store.Asset{{ID: 1, Provider: "aws", Service: "s3", ProviderHost: "b.s3.amazonaws.com", Exposure: "public_read"}}, nil
}

func (f *fakeStore) Stats(ctx context.Context) (store.Stats, error) {
	return store.Stats{Total: 1, Readable: 1}, nil
}

func TestAPI(t *testing.T) {
	fs := &fakeStore{}
	srv := httptest.NewServer(New(fs, Config{}))
	defer srv.Close()

	resp, err := http.Get(srv.URL + "/api/assets?provider=aws&q=bucket&limit=5")
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		t.Fatalf("status = %d", resp.StatusCode)
	}
	var assets []store.Asset
	if err := json.NewDecoder(resp.Body).Decode(&assets); err != nil {
		t.Fatal(err)
	}
	if len(assets) != 1 || fs.lastFilter.Provider != "aws" || fs.lastFilter.Query != "bucket" || fs.lastFilter.Limit != 5 {
		t.Fatalf("unexpected assets/filter: %+v %+v", assets, fs.lastFilter)
	}

	statsResp, err := http.Get(srv.URL + "/api/stats")
	if err != nil {
		t.Fatal(err)
	}
	defer statsResp.Body.Close()
	var st store.Stats
	if err := json.NewDecoder(statsResp.Body).Decode(&st); err != nil {
		t.Fatal(err)
	}
	if st.Total != 1 {
		t.Fatalf("stats = %+v", st)
	}
}

func TestBasicAuth(t *testing.T) {
	srv := httptest.NewServer(New(&fakeStore{}, Config{User: "u", Pass: "p"}))
	defer srv.Close()

	resp, _ := http.Get(srv.URL + "/api/stats")
	if resp.StatusCode != http.StatusUnauthorized {
		t.Fatalf("status = %d, want 401", resp.StatusCode)
	}
	resp.Body.Close()

	req, _ := http.NewRequest(http.MethodGet, srv.URL+"/api/stats", nil)
	req.SetBasicAuth("u", "p")
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		t.Fatalf("status = %d, want 200", resp.StatusCode)
	}
}

func TestIndexServed(t *testing.T) {
	srv := httptest.NewServer(New(&fakeStore{}, Config{}))
	defer srv.Close()
	resp, err := http.Get(srv.URL + "/")
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()
	if ct := resp.Header.Get("Content-Type"); ct != "text/html; charset=utf-8" {
		t.Fatalf("content-type = %q", ct)
	}
}