package coding_test

import (
	"bytes"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"strings"
	"testing"

	"github.com/jhillyerd/enmime/internal/coding"
)

func TestQPCleaner(t *testing.T) {
	ttable := []struct {
		input string
		want  string
	}{
		{"", ""},
		{"abcDEF_", "abcDEF_"},
		{"=5bSlack=5d", "=5bSlack=5d"},
		{"low: ,high:~", "low: ,high:~"},
		{"\r\n\t", "\r\n\t"},
		{"pédagogues", "p=C3=A9dagogues"},
		{"Stuffs’s", "Stuffs=E2=80=99s"},
		{"=", "=3D"},
		{"=a", "=3Da"},
	}

	for _, tc := range ttable {
		// Run cleaner
		cleaner := coding.NewQPCleaner(strings.NewReader(tc.input))
		buf := new(bytes.Buffer)
		_, err := buf.ReadFrom(cleaner)
		if err != nil {
			t.Fatal(err)
		}

		got := buf.String()
		if got != tc.want {
			t.Errorf("Got: %q, want: %q", got, tc.want)
		}
	}
}

// TestQPCleanerOverflow attempts to confuse the cleaner by issuing smaller subsequent reads.
func TestQPCleanerOverflow(t *testing.T) {
	input := bytes.Repeat([]byte("pédagogues =\r\n"), 1000)
	want := bytes.Repeat([]byte("p=C3=A9dagogues =\r\n"), 1000)
	inbuf := bytes.NewBuffer(input)
	qp := coding.NewQPCleaner(inbuf)

	offset := 0
	for len := 1000; len > 0; len -= 100 {
		p := make([]byte, len)
		n, err := qp.Read(p)
		if err != nil {
			t.Fatal(err)
		}
		if n < 1 {
			t.Fatalf("Read(p) = %v, wanted >0, at want[%v]", n, offset)
		}
		for i := 0; i < n; i++ {
			if p[i] != want[offset] {
				t.Errorf("p[%v] = %q, want: %q (want[%v])", i, p[i], want[offset], offset)
			}
			offset++
		}
	}
}

// TestQPCleanerSmallDest repeatedly calls Read with a small destination buffer.
func TestQPCleanerSmallDest(t *testing.T) {
	input := bytes.Repeat([]byte("pédagogues =z =\r\n"), 100)
	want := bytes.Repeat([]byte("p=C3=A9dagogues =3Dz =\r\n"), 100)

	for bufSize := 5; bufSize > 0; bufSize-- {
		t.Run(fmt.Sprintf("%v byte buffer", bufSize), func(t *testing.T) {
			inbuf := bytes.NewBuffer(input)
			qp := coding.NewQPCleaner(inbuf)

			offset := 0
			p := make([]byte, bufSize)
			for {
				n, err := qp.Read(p)
				if err != nil && err != io.EOF {
					t.Fatal(err)
				}
				if n < 1 && offset < len(want) {
					t.Fatalf("Read(p) = %v, wanted >0, at want[%v]", n, offset)
				}
				for i := 0; i < n; i++ {
					if p[i] != want[offset] {
						t.Errorf("p[%v] = %q, want: %q (want[%v])", i, p[i], want[offset], offset)
					}
					offset++
				}
				if err == io.EOF {
					break
				}
			}
		})
	}
}

// TestQPCleanerLineBreak verifies QPCleaner breaks long lines correctly.
func TestQPCleanerLineBreak(t *testing.T) {
	input := bytes.Repeat([]byte("pédagogues =z "), 10000)
	inbuf := bytes.NewBuffer(input)
	qp := coding.NewQPCleaner(inbuf)

	output, err := ioutil.ReadAll(qp)
	if err != nil {
		t.Fatal(err)
	}

	want := 1024 // Desired wrapping point.
	tolerance := 3

	if len(output) < want {
		t.Fatalf("wanted minimum output len %v, got %v", want, len(output))
	}

	// Examine each line of output long enough to wrap.
	for i := 0; len(output) > want; i++ {
		got := bytes.Index(output, []byte("=\r\n"))
		// Wrapping a few characters early is OK, but not late.
		if got > want || want-got > tolerance {
			t.Errorf("iteration %v: got line break at %v, wanted %v +/- %v",
				i, got, want, tolerance)
		}
		if got == 0 {
			break
		}
		output = output[got+3:] // Extend past =\r\n
	}
}

func TestQPCleanerLineBreakBufferFull(t *testing.T) {
	input := bytes.Repeat([]byte("abc"), 10000)
	inbuf := bytes.NewBuffer(input)
	qp := coding.NewQPCleaner(inbuf)

	dest := make([]byte, 1025)
	n, err := qp.Read(dest)
	if err != nil {
		t.Fatal(err)
	}
	if n != 1025 {
		t.Errorf("Unexpected result length: %d", n)
	}
}

var ErrPeek = errors.New("enmime test peek error")

type peekBreakReader string

// Read always returns a ErrPeek
func (r peekBreakReader) Read(p []byte) (int, error) {
	return copy(p, r), ErrPeek
}

func TestQPPeekError(t *testing.T) {
	qp := coding.NewQPCleaner(peekBreakReader("=a"))

	buf := make([]byte, 100)
	_, err := qp.Read(buf)
	if err != ErrPeek {
		t.Errorf("Got: %q, want: %q", err, ErrPeek)
	}
}

var result int

func BenchmarkQPCleaner(b *testing.B) {
	b.StopTimer()
	input := bytes.Repeat([]byte("pédagogues\t =zz =\r\n"), b.N)
	b.SetBytes(int64(len(input)))
	inbuf := bytes.NewBuffer(input)
	qp := coding.NewQPCleaner(inbuf)
	p := make([]byte, 1024)
	b.StartTimer()

	for {
		n, err := qp.Read(p)
		result += n
		if err == io.EOF {
			break
		}
		if err != nil {
			b.Fatalf("Read(): %v", err)
		}
	}
}