package upload

import (
	"bytes"
	_ "embed"
	"github.com/julienschmidt/httprouter"
	"github.com/spf13/afero"
	"github.com/stretchr/testify/assert"
	"io"
	"mime/multipart"
	"net/http"
	"net/http/httptest"
	"testing"
)

//go:embed test-archive.tar.gz
var testArchiveTarGz []byte

func assertUploadedFile(t *testing.T, fs afero.Fs) {
	// check uploaded file exists
	stat, err := fs.Stat("example.com/main/test.txt")
	assert.NoError(t, err)
	assert.False(t, stat.IsDir())
	assert.Equal(t, int64(13), stat.Size())

	// check contents
	o, err := fs.Open("example.com/main/test.txt")
	assert.NoError(t, err)
	all, err := io.ReadAll(o)
	assert.NoError(t, err)
	assert.Equal(t, "Hello world!\n", string(all))
}

func TestHandler_Handle(t *testing.T) {
	fs := afero.NewMemMapFs()
	h := &Handler{fs}
	mpBuf := new(bytes.Buffer)
	mp := multipart.NewWriter(mpBuf)
	file, err := mp.CreateFormFile("upload", "test-archive.tar.gz")
	assert.NoError(t, err)
	_, err = file.Write(testArchiveTarGz)
	assert.NoError(t, err)
	assert.NoError(t, mp.Close())
	req, err := http.NewRequest(http.MethodPost, "https://example.com/u/example.com?branch=main", mpBuf)
	assert.NoError(t, err)
	req.Header.Set("Content-Type", mp.FormDataContentType())
	rec := httptest.NewRecorder()
	h.Handle(rec, req, httprouter.Params{{Key: "site", Value: "example.com"}})
	res := rec.Result()
	assert.Equal(t, http.StatusAccepted, res.StatusCode)
	assert.NotNil(t, res.Body)
	all, err := io.ReadAll(res.Body)
	assert.NoError(t, err)
	assert.Equal(t, "", string(all))

	assertUploadedFile(t, fs)
}

func TestHandler_extractTarGzUpload(t *testing.T) {
	fs := afero.NewMemMapFs()
	h := &Handler{fs}
	buffer := bytes.NewBuffer(testArchiveTarGz)
	assert.NoError(t, h.extractTarGzUpload(buffer, "example.com", "main"))

	assertUploadedFile(t, fs)
}