package studentlife

import (
	"bytes"
	"fmt"
	"image/color"
	"image/png"
	"os"
	"regexp"
	"strings"

	"github.com/bwmarrin/discordgo"
	sm "github.com/flopp/go-staticmaps"
	"github.com/golang/geo/s2"
	mapbox "github.com/jackmerrill/go-mapbox/lib"
	"github.com/jackmerrill/go-mapbox/lib/base"
	"github.com/jackmerrill/go-mapbox/lib/directions"
	"github.com/jackmerrill/hampbot/internal/utils/config"
	"github.com/jackmerrill/hampbot/internal/utils/embed"
	"github.com/zekroTJA/shireikan"
)

// string is the name of the building in regex, []float64 is the lat/long
type Location map[string][]float64

var LOCATIONS = Location{
	"^(kern|kern kafe|kern cafe|kern center|r\\.w\\. kern center|rw kern center|admissions|financial aid|finaid)$": []float64{42.325490, -72.530425},
	"^(bridge|bridge cafe|the bridge)$": []float64{42.32560055268511, -72.53171782489403},
	"^(rcc|robert crown center|gym)$":   []float64{42.3259021736783, -72.53138568980566},
	"^(library|harold f. johnson library center|mail room|post office|hampstore|hamp store|campus store|duplications|it|art gallery|hampshire college art gallery|sparc|diagnostic center|media services)$": []float64{42.325503815980404, -72.53234665636477},
	"^(cole|the cole|cole science center|csc)$": []float64{42.325005132910455, -72.53264274547023},
	"^(mixed nuts)$":                               []float64{42.32438520935449, -72.53314741618625},
	"^(prescott|cff|qcac)$":                        []float64{42.32370121438042, -72.53400755941487},
	"^(prescott tavern|tavern)$":                   []float64{42.32330551141666, -72.53410313088433},
	"^(c4d|center for design|lemelson building)$":  []float64{42.323653164859145, -72.53277277596145},
	"^(central records|casa)$":                     []float64{42.32400565840685, -72.53267685200223},
	"^(arts barn|art barn|art)$":                   []float64{42.323484930985096, -72.53270230871183},
	"^(jlc|jerome liebling center)$":               []float64{42.32343637951593, -72.53186281901529},
	"^(solar canopy)$":                             []float64{42.3231160046793, -72.5323111913881},
	"^(music and dance building|mdb|music|dance)$": []float64{42.32296880488658, -72.53257385728799},
	"^(ash|adele simmons hall)$":                   []float64{42.32286199484466, -72.53189407069387},
	"^(carle|the carle|the eric carle museum|the eric carle museum of picture book art)$": []float64{42.32109682435725, -72.53332586596198},
	"^(elc|early learning center)$":                          []float64{42.32130800774271, -72.53485513102083},
	"^(multisport center|multisport|weight room|msc)$":       []float64{42.32138500151223, -72.53614340300827},
	"^(dakin student life center|dslc)$":                     []float64{42.32313128927387, -72.53022445236283},
	"^(merrill student life center|mslc)$":                   []float64{42.32347857352905, -72.53034306624653},
	"^(merrill pavillion|pavillion)$":                        []float64{42.323322471247494, -72.53030510980373},
	"^(dakin|hell|dakin house)$":                             []float64{42.322597743840625, -72.53027825479788},
	"^(merrill|merrill house)$":                              []float64{42.323732718386175, -72.52977892147766},
	"^(dining commons|dc|saga)$":                             []float64{42.32319356233955, -72.52917576638322},
	"^(yiddish book center|ybc)$":                            []float64{42.32175452980474, -72.5276151025249},
	"^(franklin patterson hall|fph)$":                        []float64{42.324221538083805, -72.5306271019939},
	"^(the yurt|yurt|radio)$":                                []float64{42.324071744881785, -72.53136719263956},
	"^(enfield|enfield mods|enfield house)$":                 []float64{42.32646094303269, -72.52929170510444},
	"^(wellness center)$":                                    []float64{42.32709812095117, -72.5291661607627},
	"^(spiritual life center|slc)$":                          []float64{42.32705614551552, -72.52973716786117},
	"^(edh|emily dickinson hall|theatre|theater)$":           []float64{42.327652942240896, -72.53062317253445},
	"^(writing center)$":                                     []float64{42.32780110830725, -72.53112929268784},
	"^(greenwich|greenwich mods|greenwich house)$":           []float64{42.32748781455265, -72.53192281696715},
	"^(soccer|soccer field|hampshire college soccer field)$": []float64{42.32651025401564, -72.53454675256307},
	"^(cultural center|cc|lebron-wiggins-pran cultural center|Lebrón-Wiggins-Pran Cultural Center)$": []float64{42.32487763771362, -72.5339769868106},
	"^(basketball|basketball courts|basketball court|tennis courts|tennis court)$":                   []float64{42.32557748709811, -72.53720546848393},
	"^(red barn|the red barn|barn)$": []float64{42.32642633315529, -72.52551730246282},
	"^(the hitchcock center|hitchcock center|hitchcock center for the environment)$": []float64{42.32771760017527, -72.52570886249646},
	"^(csa|farm|hampshire college farm center|hampshire college farm)$":              []float64{42.32903799744424, -72.52573139895138},
	"^(health services|hampshire college health services)$":                          []float64{42.32666929509394, -72.52524995366625},
	"^(atkins|atkins farms country market)$":                                         []float64{42.319355104827295, -72.52927125654328},
	"^(res|the res|the reservoir|reservoir)$":                                        []float64{42.317285834368995, -72.5406219720153},
}

type Where struct {
}

var mapboxToken = os.Getenv("MAPBOX_TOKEN")
var mapBox, _ = mapbox.NewMapbox(mapboxToken)

func (c *Where) GetInvokes() []string {
	return []string{"where"}
}

func (c *Where) GetDescription() string {
	return "Where is this building?"
}

func (c *Where) GetHelp() string {
	return "`where <building>` - Where is this building?"
}

func (c *Where) GetGroup() string {
	return config.GroupStudentLife
}

func (c *Where) GetDomainName() string {
	return "hamp.studentlife.where"
}

func (c *Where) GetSubPermissionRules() []shireikan.SubPermission {
	return nil
}
func (c *Where) IsExecutableInDMChannels() bool {
	return true
}

func (c *Where) Exec(ctx shireikan.Context) error {
	if len(ctx.GetArgs()) < 1 {
		// return all buildings
		e := embed.NewSuccessEmbed(ctx)

		e.SetTitle("Buildings")

		desc := "*Places with spaces must be quoted*\n\nPlaces:"
		locs := []string{}

		for regex := range LOCATIONS {
			l := strings.TrimLeft(regex, "^(")
			l = strings.TrimRight(l, ")$")

			ls := strings.Split(l, "|")

			for _, l := range ls {
				locs = append(locs, l)
			}

			desc += fmt.Sprintf("`%s`, ", strings.Join(ls, "`, `"))
		}

		e.SetDescription(desc)

		_, err := ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{
			Embed:     e.MessageEmbed,
			Reference: ctx.GetMessage().Reference(),
		})

		return err
	}

	// parse args, quoted args are one arg
	args := ctx.GetArgs()

	to := strings.ToLower(args[0])
	from := strings.ToLower(strings.Join(args[1:], " "))

	e := embed.NewSuccessEmbed(ctx)

	e.SetTitle("Where is " + to + "?")

	if from != "" {
		e.SetTitle("Directions from " + to + " to " + from)
	}

	e.SetDescription("It's right here!")

	mapCtx := sm.NewContext()
	mapCtx.SetSize(600, 600)
	mapCtx.SetTileProvider(sm.NewTileProviderOpenStreetMaps())

	if from == "" {
		mapCtx.SetZoom(17)
	}

	var toCoord []float64
	var fromCoord []float64

	for regex, coords := range LOCATIONS {
		toMatch, err := regexp.MatchString(regex, to)

		if err != nil {
			return err
		}

		fromMatch, err := regexp.MatchString(regex, from)

		if err != nil {
			return err
		}

		if toMatch || fromMatch {
			lat := coords[0]
			long := coords[1]

			if toMatch {
				toCoord = coords
				mapCtx.AddObject(
					sm.NewMarker(
						s2.LatLngFromDegrees(lat, long),
						color.RGBA{R: 0, G: 255, B: 0, A: 255},
						16.0,
					),
				)
			} else if fromMatch {
				fromCoord = coords
				mapCtx.AddObject(
					sm.NewMarker(
						s2.LatLngFromDegrees(lat, long),
						color.RGBA{R: 255, G: 0, B: 0, A: 255},
						16.0,
					),
				)
			}
		}
	}

	if len(toCoord) == 0 {
		errorEmbed := embed.NewErrorEmbed(ctx)

		errorEmbed.SetTitle("Error")
		errorEmbed.SetDescription(fmt.Sprintf("Building `%s` not found.", to))

		_, err := ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{
			Embed:     errorEmbed.MessageEmbed,
			Reference: ctx.GetMessage().Reference(),
		})

		return err
	}

	if len(fromCoord) == 0 && from != "" {
		errorEmbed := embed.NewErrorEmbed(ctx)

		errorEmbed.SetTitle("Error")
		errorEmbed.SetDescription(fmt.Sprintf("Building `%s` not found.", from))

		_, err := ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{
			Embed:     errorEmbed.MessageEmbed,
			Reference: ctx.GetMessage().Reference(),
		})

		return err
	}

	if from != "" {
		// get mapbox directions
		dir, err := mapBox.Directions.GetDirections([]base.Location{
			{
				Latitude:  toCoord[0],
				Longitude: toCoord[1],
			},
			{
				Latitude:  fromCoord[0],
				Longitude: fromCoord[1],
			},
		}, directions.RoutingWalking, &directions.RequestOpts{
			Steps:        false,
			Alternatives: false,
		})

		if err != nil {
			return err
		}

		var latLngs []s2.LatLng

		// add route to map
		for _, coord := range dir.Routes[0].Geometry.Coordinates {
			latLngs = append(latLngs, s2.LatLngFromDegrees(coord[1], coord[0]))
		}

		mapCtx.AddObject(
			sm.NewPath(
				latLngs,
				color.RGBA{R: 0, G: 0, B: 255, A: 255},
				3.0,
			),
		)

		// set the zoom accordingly to the route length
		if dir.Routes[0].Distance > 1000 {
			mapCtx.SetZoom(15)
		}
	}

	img, err := mapCtx.Render()

	if err != nil {
		return err
	}

	buf := new(bytes.Buffer)

	// convert to base64
	err = png.Encode(buf, img)

	if err != nil {
		return err
	}

	e.SetImage("attachment://map.png")

	if err != nil {
		return err
	}

	_, err = ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{
		Embed:     e.MessageEmbed,
		Reference: ctx.GetMessage().Reference(),
		Files: []*discordgo.File{
			{
				Name:        "map.png",
				ContentType: "image/png",
				Reader:      buf,
			},
		},
	})

	return err
}