package image

import (
	rand "crypto/rand"
	"reflect"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("File format tests", func() {
	It("Copy known headers - map known headers", func() {
		got := CopyKnownHdrs()
		Expect(reflect.DeepEqual(got, knownHeaders)).To(BeTrue())
	})

	//tar bytes and offset
	token := make([]byte, 257)
	tarheader := []byte{0x75, 0x73, 0x74, 0x61, 0x72, 0x20}
	_, err := rand.Read(token)
	Expect(err).ToNot(HaveOccurred())
	tarbyte := append(token, tarheader...)

	DescribeTable("Header match", func(h Header, b []byte, want bool) {
		got := h.Match(b)
		Expect(got).To(Equal(want))
	},
		Entry("match gz",
			Header{"gz", []byte{0x1F, 0x8B}, 0, 0, 0},
			[]byte{0x1F, 0x8B},
			true),
		Entry("match qcow2",
			Header{"qcow2", []byte{'Q', 'F', 'I', 0xfb}, 0, 24, 8},
			[]byte{'Q', 'F', 'I', 0xfb},
			true),
		Entry("match tar",
			Header{"tar", []byte{0x75, 0x73, 0x74, 0x61, 0x72, 0x20}, 0x101, 124, 8},
			tarbyte,
			true),
		Entry("match xz",
			Header{"xz", []byte{0xFD, 0x37, 0x7A, 0x58, 0x5A, 0x00}, 0, 0, 0},
			[]byte{0xFD, 0x37, 0x7A, 0x58, 0x5A, 0x00},
			true),
		Entry("failed match",
			Header{"gz", []byte{0x1F, 0x8B}, 0, 0, 0},
			[]byte{'Q', 'F', 'I', 0xfb},
			false),
		Entry("match vmdk",
			Header{"vmdk", []byte("KDMV"), 0, 24, 8},
			[]byte("KDMV"),
			true),
		Entry("match vdi",
			Header{"vdi", []byte("<<< Oracle VM"), 0, 24, 8},
			[]byte("<<< Oracle VM"),
			true),
		Entry("match vhd",
			Header{"vpc", []byte("connectix"), 0, 24, 8},
			[]byte("connectix"),
			true),
		Entry("match vhdx",
			Header{"vhdx", []byte("vhdxfile"), 0, 24, 8},
			[]byte("vhdxfile"),
			true),
	)

	tokenQcow := make([]byte, 20)
	qcowMagic := []byte{'Q', 'F', 'I', 0xfb}
	qcowSize := []byte("10561056")
	_, err = rand.Read(tokenQcow)
	Expect(err).ToNot(HaveOccurred())
	qcowbyte := append(qcowMagic, tokenQcow...)
	qcowbyte = append(qcowbyte, qcowSize...)

	DescribeTable("Header size", func(h Header, b []byte, want int64, wantErr bool) {
		got, err := h.Size(b)
		if wantErr {
			Expect(err).To(HaveOccurred())
		} else {
			Expect(err).ToNot(HaveOccurred())
		}
		Expect(got).To(Equal(want))
	},
		Entry("get size of qcow2",
			Header{"qcow2", []byte{'Q', 'F', 'I', 0xfb}, 0, 24, 8},
			qcowbyte,
			int64(3544391413610329398),
			false),
		Entry("does not implement size",
			Header{"gz", []byte{0x1F, 0x8B}, 0, 0, 0},
			[]byte{0x1F, 0x8B},
			int64(0),
			false),
	)
})