package tests

import (
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/http/httptest"

	"github.com/gomodule/redigo/redis"
	"github.com/tidwall/pretty"
	"github.com/tidwall/sjson"
)

func fence_roaming_webhook_test(mc *mockServer) error {
	car1, car2, expected := roamingTestData()
	finalErr := make(chan error)

	// Create a connection for subscribing to geofence notifications
	sc, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
	if err != nil {
		return err
	}
	defer sc.Close()

	actual := []string{}
	// Create the test http server that will capture all messages
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if err := func() error {
			// Read the request body
			body, err := ioutil.ReadAll(r.Body)
			if err != nil {
				return err
			}

			// If the new message doesn't match whats expected an error
			// should be returned
			actual = append(actual, cleanMessage(body))
			pos := len(actual) - 1
			if len(expected) < pos+1 {
				return fmt.Errorf("More messages than expected were received : '%s'", actual[pos])
			}
			if actual[pos] != expected[pos] {
				return fmt.Errorf("Expected '%s' but got '%s'", expected[pos],
					actual[pos])
			}
			if len(actual) == len(expected) {
				finalErr <- nil
			}
			return nil
		}(); err != nil {
			finalErr <- err
		}
		fmt.Fprintln(w, "OK!")
	}))
	defer ts.Close()

	_, err = sc.Do("SETHOOK", "carshook", ts.URL, "NEARBY", "cars", "FENCE", "ROAM", "cars", "*", 1000)
	if err != nil {
		return err
	}

	// Create the base connection for setting up points and geofences
	bc, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
	if err != nil {
		return err
	}
	defer bc.Close()

	// Fire all car movement commands on the base client
	for i := range car1 {
		if _, err := bc.Do("SET", "cars", "car1", "POINT", car1[i][1],
			car1[i][0]); err != nil {
			return err
		}
		if _, err := bc.Do("SET", "cars", "car2", "POINT", car2[i][1],
			car2[i][0]); err != nil {
			return err
		}
	}

	return <-finalErr
}

func fence_roaming_live_test(mc *mockServer) error {
	car1, car2, expected := roamingTestData()
	finalErr := make(chan error)

	go func() {
		// Create a connection for subscribing to geofence notifications
		sc, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
		if err != nil {
			finalErr <- err
			return
		}
		defer sc.Close()

		// Set up a live geofence stream
		if _, err := sc.Do("NEARBY", "cars", "FENCE", "ROAM", "cars", "*", 1000); err != nil {
			finalErr <- err
			return
		}

		actual := []string{}
		for sc.Err() == nil {
			if err := func() error {
				bodyi, err := sc.Receive()
				if err != nil {
					return err
				}
				body, ok := bodyi.([]byte)
				if !ok {
					return errors.New("Non byte-slice received")
				}

				// If the new message doesn't match whats expected an error
				// should be returned
				actual = append(actual, cleanMessage(body))
				pos := len(actual) - 1
				if len(expected) < pos+1 {
					return fmt.Errorf("More messages than expected were received : '%s'", actual[pos])
				}
				if actual[pos] != expected[pos] {
					return fmt.Errorf("Expected '%s' but got '%s'", expected[pos],
						actual[pos])
				}
				if len(actual) == len(expected) {
					finalErr <- nil
				}
				return nil
			}(); err != nil {
				finalErr <- err
			}
		}
	}()

	// Create the base connection for setting up points and geofences
	bc, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
	if err != nil {
		return err
	}
	defer bc.Close()

	// Fire all car movement commands on the base client
	for i := range car1 {
		if _, err := bc.Do("SET", "cars", "car1", "POINT", car1[i][1],
			car1[i][0]); err != nil {
			return err
		}
		if _, err := bc.Do("SET", "cars", "car2", "POINT", car2[i][1],
			car2[i][0]); err != nil {
			return err
		}
	}

	return <-finalErr
}

func fence_roaming_channel_test(mc *mockServer) error {
	car1, car2, expected := roamingTestData()
	finalErr := make(chan error)

	go func() {
		// Create a connection for subscribing to geofence notifications
		sc, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
		if err != nil {
			finalErr <- err
			return
		}
		defer sc.Close()

		if _, err := sc.Do("SETCHAN", "carschan", "NEARBY", "cars", "FENCE", "ROAM", "cars", "*", 1000); err != nil {
			finalErr <- err
			return
		}

		// Subscribe the subscription client to the * pattern
		psc := redis.PubSubConn{Conn: sc}
		if err := psc.PSubscribe("carschan"); err != nil {
			finalErr <- err
			return
		}

		actual := []string{}
		for sc.Err() == nil {
			if err := func() error {
				var body []byte
				switch v := psc.Receive().(type) {
				case redis.Message:
					body = v.Data
				case error:
					return err
				}
				if len(body) == 0 {
					return nil
				}

				// If the new message doesn't match whats expected an error
				// should be returned
				actual = append(actual, cleanMessage(body))
				pos := len(actual) - 1
				if len(expected) < pos+1 {
					return fmt.Errorf("More messages than expected were received : '%s'", actual[pos])
				}
				if actual[pos] != expected[pos] {
					return fmt.Errorf("Expected '%s' but got '%s'", expected[pos],
						actual[pos])
				}
				if len(actual) == len(expected) {
					finalErr <- nil
				}
				return nil
			}(); err != nil {
				finalErr <- err
			}
		}
	}()

	// Create the base connection for setting up points and geofences
	bc, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
	if err != nil {
		return err
	}
	defer bc.Close()

	// Fire all car movement commands on the base client
	for i := range car1 {
		if _, err := bc.Do("SET", "cars", "car1", "POINT", car1[i][1],
			car1[i][0]); err != nil {
			return err
		}
		if _, err := bc.Do("SET", "cars", "car2", "POINT", car2[i][1],
			car2[i][0]); err != nil {
			return err
		}
	}

	return <-finalErr
}

func cleanMessage(body []byte) string {
	// Remove fields that are non-deterministic or use case specific
	msg, _ := sjson.Delete(string(body), "group")
	msg, _ = sjson.Delete(msg, "time")
	msg, _ = sjson.Delete(msg, "hook")
	msg = string(pretty.Ugly([]byte(msg)))
	return msg
}

func roamingTestData() (car1 [][]float64, car2 [][]float64, output []string) {
	car1 = [][]float64{
		{-111.93669319152832, 33.414750027566235},
		{-111.93051338195801, 33.414750027566235},
		{-111.92416191101074, 33.414750027566235},
		{-111.91789627075195, 33.414750027566235},
		{-111.9111156463623, 33.414750027566235},
		{-111.90510749816895, 33.414750027566235},
		{-111.89746856689453, 33.414750027566235},
	}
	car2 = [][]float64{
		{-111.89746856689453, 33.414750027566235},
		{-111.90519332885742, 33.414750027566235},
		{-111.91154479980467, 33.414750027566235},
		{-111.91781044006346, 33.414750027566235},
		{-111.92416191101074, 33.414750027566235},
		{-111.93059921264648, 33.414750027566235},
		{-111.93660736083984, 33.414750027566235},
	}
	output = []string{
		`{"command":"set","detect":"roam","key":"cars","id":"car1","object":{"type":"Point","coordinates":[-111.91789627075195,33.414750027566235]},"nearby":{"key":"cars","id":"car2","object":{"type":"Point","coordinates":[-111.91154479980467,33.414750027566235]},"meters":589.512}}`,
		`{"command":"set","detect":"roam","key":"cars","id":"car2","object":{"type":"Point","coordinates":[-111.91781044006346,33.414750027566235]},"nearby":{"key":"cars","id":"car1","object":{"type":"Point","coordinates":[-111.91789627075195,33.414750027566235]},"meters":7.966}}`,
		`{"command":"set","detect":"roam","key":"cars","id":"car1","object":{"type":"Point","coordinates":[-111.9111156463623,33.414750027566235]},"nearby":{"key":"cars","id":"car2","object":{"type":"Point","coordinates":[-111.91781044006346,33.414750027566235]},"meters":621.377}}`,
		`{"command":"set","detect":"roam","key":"cars","id":"car2","object":{"type":"Point","coordinates":[-111.92416191101074,33.414750027566235]},"faraway":{"key":"cars","id":"car1","object":{"type":"Point","coordinates":[-111.9111156463623,33.414750027566235]},"meters":1210.89}}`,
		`{"command":"set","detect":"roam","key":"cars","id":"car1","object":{"type":"Point","coordinates":[-111.90510749816895,33.414750027566235]},"faraway":{"key":"cars","id":"car2","object":{"type":"Point","coordinates":[-111.92416191101074,33.414750027566235]},"meters":1768.536}}`,
	}
	return
}