mirror of
https://github.com/jackmerrill/hampbot.git
synced 2024-10-16 13:47:41 -07:00
307 lines
10 KiB
Go
307 lines
10 KiB
Go
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
|
|
}
|