package exif

import (
	"bytes"
	"fmt"
	"path"
	"reflect"
	"testing"

	"encoding/binary"
	"io/ioutil"

	"github.com/dsoprea/go-logging"
)

func TestIfdTagEntry_ValueBytes(t *testing.T) {
	byteOrder := binary.BigEndian
	ve := NewValueEncoder(byteOrder)

	original := []byte("original text")

	ed, err := ve.encodeBytes(original)
	log.PanicIf(err)

	// Now, pass the raw encoded value as if it was the entire addressable area
	// and provide an offset of 0 as if it was a real block of data and this
	// value happened to be recorded at the beginning.

	ite := IfdTagEntry{
		TagType:     TypeByte,
		UnitCount:   uint32(len(original)),
		ValueOffset: 0,
	}

	decodedBytes, err := ite.ValueBytes(ed.Encoded, byteOrder)
	log.PanicIf(err)

	if bytes.Compare(decodedBytes, original) != 0 {
		t.Fatalf("Bytes not decoded correctly.")
	}
}

func TestIfdTagEntry_ValueBytes_RealData(t *testing.T) {
	defer func() {
		if state := recover(); state != nil {
			err := log.Wrap(state.(error))
			log.PrintErrorf(err, "Test failure.")
		}
	}()

	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	eh, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	var ite *IfdTagEntry
	for _, thisIte := range index.RootIfd.Entries {
		if thisIte.TagId == 0x0110 {
			ite = thisIte
			break
		}
	}

	if ite == nil {
		t.Fatalf("Tag not found.")
	}

	addressableData := rawExif[ExifAddressableAreaStart:]
	decodedBytes, err := ite.ValueBytes(addressableData, eh.ByteOrder)
	log.PanicIf(err)

	expected := []byte("Canon EOS 5D Mark III")
	expected = append(expected, 0)

	if len(decodedBytes) != int(ite.UnitCount) {
		t.Fatalf("Decoded bytes not the right count.")
	} else if bytes.Compare(decodedBytes, expected) != 0 {
		t.Fatalf("Decoded bytes not correct.")
	}
}

func TestIfdTagEntry_Resolver_ValueBytes(t *testing.T) {
	defer func() {
		if state := recover(); state != nil {
			err := log.Wrap(state.(error))
			log.PrintErrorf(err, "Test failure.")
		}
	}()

	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	eh, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	var ite *IfdTagEntry
	for _, thisIte := range index.RootIfd.Entries {
		if thisIte.TagId == 0x0110 {
			ite = thisIte
			break
		}
	}

	if ite == nil {
		t.Fatalf("Tag not found.")
	}

	itevr := NewIfdTagEntryValueResolver(rawExif, eh.ByteOrder)

	decodedBytes, err := itevr.ValueBytes(ite)
	log.PanicIf(err)

	expected := []byte("Canon EOS 5D Mark III")
	expected = append(expected, 0)

	if len(decodedBytes) != int(ite.UnitCount) {
		t.Fatalf("Decoded bytes not the right count.")
	} else if bytes.Compare(decodedBytes, expected) != 0 {
		t.Fatalf("Decoded bytes not correct.")
	}
}

func TestIfdTagEntry_Resolver_ValueBytes__Unknown_Field_And_Nonroot_Ifd(t *testing.T) {
	defer func() {
		if state := recover(); state != nil {
			err := log.Wrap(state.(error))
			log.PrintErrorf(err, "Test failure.")
		}
	}()

	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	eh, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifdExif := index.Lookup[IfdPathStandardExif][0]

	var ite *IfdTagEntry
	for _, thisIte := range ifdExif.Entries {
		if thisIte.TagId == 0x9000 {
			ite = thisIte
			break
		}
	}

	if ite == nil {
		t.Fatalf("Tag not found.")
	}

	itevr := NewIfdTagEntryValueResolver(rawExif, eh.ByteOrder)

	decodedBytes, err := itevr.ValueBytes(ite)
	log.PanicIf(err)

	expected := []byte{'0', '2', '3', '0'}

	if len(decodedBytes) != int(ite.UnitCount) {
		t.Fatalf("Decoded bytes not the right count.")
	} else if bytes.Compare(decodedBytes, expected) != 0 {
		t.Fatalf("Recovered unknown value is not correct.")
	}
}

func Test_Ifd_FindTagWithId_Hit(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd := index.RootIfd
	results, err := ifd.FindTagWithId(0x011b)

	if len(results) != 1 {
		t.Fatalf("Exactly one result was not found: (%d)", len(results))
	} else if results[0].TagId != 0x011b {
		t.Fatalf("The result was not expected: %v", results[0])
	}
}

func Test_Ifd_FindTagWithId_Miss(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd := index.RootIfd

	_, err = ifd.FindTagWithId(0xffff)
	if err == nil {
		t.Fatalf("Expected error for not-found tag.")
	} else if log.Is(err, ErrTagNotFound) == false {
		log.Panic(err)
	}
}

func Test_Ifd_FindTagWithName_Hit(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd := index.RootIfd
	results, err := ifd.FindTagWithName("YResolution")

	if len(results) != 1 {
		t.Fatalf("Exactly one result was not found: (%d)", len(results))
	} else if results[0].TagId != 0x011b {
		t.Fatalf("The result was not expected: %v", results[0])
	}
}

func Test_Ifd_FindTagWithName_Miss(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd := index.RootIfd

	_, err = ifd.FindTagWithName("PlanarConfiguration")
	if err == nil {
		t.Fatalf("Expected error for not-found tag.")
	} else if log.Is(err, ErrTagNotFound) == false {
		log.Panic(err)
	}
}

func Test_Ifd_FindTagWithName_NonStandard(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd := index.RootIfd

	_, err = ifd.FindTagWithName("GeorgeNotAtHome")
	if err == nil {
		t.Fatalf("Expected error for not-found tag.")
	} else if log.Is(err, ErrTagNotStandard) == false {
		log.Panic(err)
	}
}

func Test_Ifd_Thumbnail(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd := index.RootIfd

	if ifd.NextIfd == nil {
		t.Fatalf("There is no IFD1.")
	}

	// The thumbnail is in IFD1 (The second root IFD).
	actual, err := ifd.NextIfd.Thumbnail()
	log.PanicIf(err)

	expectedFilepath := path.Join(assetsPath, "NDM_8901.jpg.thumbnail")

	expected, err := ioutil.ReadFile(expectedFilepath)
	log.PanicIf(err)

	if bytes.Compare(actual, expected) != 0 {
		t.Fatalf("thumbnail not correct")
	}
}

func TestIfd_GpsInfo(t *testing.T) {
	defer func() {
		if state := recover(); state != nil {
			err := log.Wrap(state.(error))
			log.PrintErrorf(err, "Test failure.")
		}
	}()

	filepath := path.Join(assetsPath, "gps.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd, err := index.RootIfd.ChildWithIfdPath(IfdPathStandardGps)
	log.PanicIf(err)

	gi, err := ifd.GpsInfo()
	log.PanicIf(err)

	if gi.Latitude.Orientation != 'N' || gi.Latitude.Degrees != 26 || gi.Latitude.Minutes != 35 || gi.Latitude.Seconds != 12 {
		t.Fatalf("latitude not correct")
	} else if gi.Longitude.Orientation != 'W' || gi.Longitude.Degrees != 80 || gi.Longitude.Minutes != 3 || gi.Longitude.Seconds != 13 {
		t.Fatalf("longitude not correct")
	} else if gi.Altitude != 0 {
		t.Fatalf("altitude not correct")
	} else if gi.Timestamp.Unix() != 1524964977 {
		t.Fatalf("timestamp not correct")
	} else if gi.Altitude != 0 {
		t.Fatalf("altitude not correct")
	}
}

func TestIfd_EnumerateTagsRecursively(t *testing.T) {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	collected := make([][2]interface{}, 0)

	cb := func(ifd *Ifd, ite *IfdTagEntry) error {
		item := [2]interface{}{
			ifd.IfdPath,
			int(ite.TagId),
		}

		collected = append(collected, item)

		return nil
	}

	err = index.RootIfd.EnumerateTagsRecursively(cb)
	log.PanicIf(err)

	expected := [][2]interface{}{
		[2]interface{}{"IFD", 0x010f},
		[2]interface{}{"IFD", 0x0110},
		[2]interface{}{"IFD", 0x0112},
		[2]interface{}{"IFD", 0x011a},
		[2]interface{}{"IFD", 0x011b},
		[2]interface{}{"IFD", 0x0128},
		[2]interface{}{"IFD", 0x0132},
		[2]interface{}{"IFD", 0x013b},
		[2]interface{}{"IFD", 0x0213},
		[2]interface{}{"IFD", 0x8298},
		[2]interface{}{"IFD/Exif", 0x829a},
		[2]interface{}{"IFD/Exif", 0x829d},
		[2]interface{}{"IFD/Exif", 0x8822},
		[2]interface{}{"IFD/Exif", 0x8827},
		[2]interface{}{"IFD/Exif", 0x8830},
		[2]interface{}{"IFD/Exif", 0x8832},
		[2]interface{}{"IFD/Exif", 0x9000},
		[2]interface{}{"IFD/Exif", 0x9003},
		[2]interface{}{"IFD/Exif", 0x9004},
		[2]interface{}{"IFD/Exif", 0x9101},
		[2]interface{}{"IFD/Exif", 0x9201},
		[2]interface{}{"IFD/Exif", 0x9202},
		[2]interface{}{"IFD/Exif", 0x9204},
		[2]interface{}{"IFD/Exif", 0x9207},
		[2]interface{}{"IFD/Exif", 0x9209},
		[2]interface{}{"IFD/Exif", 0x920a},
		[2]interface{}{"IFD/Exif", 0x927c},
		[2]interface{}{"IFD/Exif", 0x9286},
		[2]interface{}{"IFD/Exif", 0x9290},
		[2]interface{}{"IFD/Exif", 0x9291},
		[2]interface{}{"IFD/Exif", 0x9292},
		[2]interface{}{"IFD/Exif", 0xa000},
		[2]interface{}{"IFD/Exif", 0xa001},
		[2]interface{}{"IFD/Exif", 0xa002},
		[2]interface{}{"IFD/Exif", 0xa003},
		[2]interface{}{"IFD/Exif/Iop", 0x0001},
		[2]interface{}{"IFD/Exif/Iop", 0x0002},
		[2]interface{}{"IFD/Exif", 0xa20e},
		[2]interface{}{"IFD/Exif", 0xa20f},
		[2]interface{}{"IFD/Exif", 0xa210},
		[2]interface{}{"IFD/Exif", 0xa401},
		[2]interface{}{"IFD/Exif", 0xa402},
		[2]interface{}{"IFD/Exif", 0xa403},
		[2]interface{}{"IFD/Exif", 0xa406},
		[2]interface{}{"IFD/Exif", 0xa430},
		[2]interface{}{"IFD/Exif", 0xa431},
		[2]interface{}{"IFD/Exif", 0xa432},
		[2]interface{}{"IFD/Exif", 0xa434},
		[2]interface{}{"IFD/Exif", 0xa435},
		[2]interface{}{"IFD/GPSInfo", 0x0000},
		[2]interface{}{"IFD", 0x010f},
		[2]interface{}{"IFD", 0x0110},
		[2]interface{}{"IFD", 0x0112},
		[2]interface{}{"IFD", 0x011a},
		[2]interface{}{"IFD", 0x011b},
		[2]interface{}{"IFD", 0x0128},
		[2]interface{}{"IFD", 0x0132},
		[2]interface{}{"IFD", 0x013b},
		[2]interface{}{"IFD", 0x0213},
		[2]interface{}{"IFD", 0x8298},
		[2]interface{}{"IFD/Exif", 0x829a},
		[2]interface{}{"IFD/Exif", 0x829d},
		[2]interface{}{"IFD/Exif", 0x8822},
		[2]interface{}{"IFD/Exif", 0x8827},
		[2]interface{}{"IFD/Exif", 0x8830},
		[2]interface{}{"IFD/Exif", 0x8832},
		[2]interface{}{"IFD/Exif", 0x9000},
		[2]interface{}{"IFD/Exif", 0x9003},
		[2]interface{}{"IFD/Exif", 0x9004},
		[2]interface{}{"IFD/Exif", 0x9101},
		[2]interface{}{"IFD/Exif", 0x9201},
		[2]interface{}{"IFD/Exif", 0x9202},
		[2]interface{}{"IFD/Exif", 0x9204},
		[2]interface{}{"IFD/Exif", 0x9207},
		[2]interface{}{"IFD/Exif", 0x9209},
		[2]interface{}{"IFD/Exif", 0x920a},
		[2]interface{}{"IFD/Exif", 0x927c},
		[2]interface{}{"IFD/Exif", 0x9286},
		[2]interface{}{"IFD/Exif", 0x9290},
		[2]interface{}{"IFD/Exif", 0x9291},
		[2]interface{}{"IFD/Exif", 0x9292},
		[2]interface{}{"IFD/Exif", 0xa000},
		[2]interface{}{"IFD/Exif", 0xa001},
		[2]interface{}{"IFD/Exif", 0xa002},
		[2]interface{}{"IFD/Exif", 0xa003},
		[2]interface{}{"IFD/Exif/Iop", 0x0001},
		[2]interface{}{"IFD/Exif/Iop", 0x0002},
		[2]interface{}{"IFD/Exif", 0xa20e},
		[2]interface{}{"IFD/Exif", 0xa20f},
		[2]interface{}{"IFD/Exif", 0xa210},
		[2]interface{}{"IFD/Exif", 0xa401},
		[2]interface{}{"IFD/Exif", 0xa402},
		[2]interface{}{"IFD/Exif", 0xa403},
		[2]interface{}{"IFD/Exif", 0xa406},
		[2]interface{}{"IFD/Exif", 0xa430},
		[2]interface{}{"IFD/Exif", 0xa431},
		[2]interface{}{"IFD/Exif", 0xa432},
		[2]interface{}{"IFD/Exif", 0xa434},
		[2]interface{}{"IFD/Exif", 0xa435},
		[2]interface{}{"IFD/GPSInfo", 0x0000},
	}

	if reflect.DeepEqual(collected, expected) != true {
		fmt.Printf("ACTUAL:\n")
		fmt.Printf("\n")

		for _, item := range collected {
			fmt.Printf("[2]interface{} { \"%s\", 0x%04x },\n", item[0], item[1])
		}

		fmt.Printf("\n")

		fmt.Printf("EXPECTED:\n")
		fmt.Printf("\n")

		for _, item := range expected {
			fmt.Printf("[2]interface{} { \"%s\", 0x%04x },\n", item[0], item[1])
		}

		fmt.Printf("\n")

		t.Fatalf("tags not visited correctly")
	}
}

func ExampleIfd_EnumerateTagsRecursively() {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	cb := func(ifd *Ifd, ite *IfdTagEntry) error {

		// Something useful.

		return nil
	}

	err = index.RootIfd.EnumerateTagsRecursively(cb)
	log.PanicIf(err)

	// Output:
}

func ExampleIfd_GpsInfo() {
	filepath := path.Join(assetsPath, "gps.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	ifd, err := index.RootIfd.ChildWithIfdPath(IfdPathStandardGps)
	log.PanicIf(err)

	gi, err := ifd.GpsInfo()
	log.PanicIf(err)

	fmt.Printf("%s\n", gi)

	// Output:
	// GpsInfo<LAT=(26.58667) LON=(-80.05361) ALT=(0) TIME=[2018-04-29 01:22:57 +0000 UTC]>
}

func ExampleIfd_FindTagWithName() {
	filepath := path.Join(assetsPath, "NDM_8901.jpg")

	rawExif, err := SearchFileAndExtractExif(filepath)
	log.PanicIf(err)

	im := NewIfdMapping()

	err = LoadStandardIfds(im)
	log.PanicIf(err)

	ti := NewTagIndex()

	_, index, err := Collect(im, ti, rawExif)
	log.PanicIf(err)

	tagName := "Model"

	// We know the tag we want is on IFD0 (the first/root IFD).
	results, err := index.RootIfd.FindTagWithName(tagName)
	log.PanicIf(err)

	// This should never happen.
	if len(results) != 1 {
		log.Panicf("there wasn't exactly one result")
	}

	ite := results[0]

	valueRaw, err := index.RootIfd.TagValue(ite)
	log.PanicIf(err)

	value := valueRaw.(string)
	fmt.Println(value)

	// Output:
	// Canon EOS 5D Mark III
}