1
0
Fork 0
mirror of https://github.com/Luzifer/staticmap.git synced 2024-12-05 22:24:02 +00:00
staticmap/map.go
Knut Ahlers ad12e02404
Lint: Fix linter errors
Signed-off-by: Knut Ahlers <knut@ahlers.me>
2024-11-30 12:48:55 +01:00

118 lines
2.9 KiB
Go

package main
import (
"bytes"
"crypto/sha256"
"fmt"
"image/color"
"io"
"strings"
staticMap "github.com/Luzifer/go-staticmaps"
"github.com/fogleman/gg"
"github.com/golang/geo/s2"
"github.com/pkg/errors"
)
//nolint:mnd // these are the "constant" definitions
var markerColors = map[string]color.Color{
"black": color.RGBA{R: 145, G: 145, B: 145, A: 0xff},
"brown": color.RGBA{R: 178, G: 154, B: 123, A: 0xff},
"green": color.RGBA{R: 168, G: 196, B: 68, A: 0xff},
"purple": color.RGBA{R: 177, G: 150, B: 191, A: 0xff},
"yellow": color.RGBA{R: 237, G: 201, B: 107, A: 0xff},
"blue": color.RGBA{R: 163, G: 196, B: 253, A: 0xff},
"gray": color.RGBA{R: 204, G: 204, B: 204, A: 0xff},
"orange": color.RGBA{R: 229, G: 165, B: 68, A: 0xff},
"red": color.RGBA{R: 246, G: 118, B: 112, A: 0xff},
"white": color.RGBA{R: 245, G: 244, B: 241, A: 0xff},
}
type markerSize float64
//nolint:mnd // these are the "constant" definitions
var markerSizes = map[string]markerSize{
"tiny": 10,
"mid": 15,
"small": 20,
}
type marker struct {
pos s2.LatLng
color color.Color
size markerSize
}
func (m marker) String() string {
r, g, b, a := m.color.RGBA()
return fmt.Sprintf("%s|%.0f|%d,%d,%d,%d", m.pos.String(), m.size, r, g, b, a)
}
type generateMapConfig struct {
Center s2.LatLng
Zoom int
Markers []marker
Width int
Height int
DisableAttribution bool
Overlays []*staticMap.TileProvider
}
func (g generateMapConfig) getCacheKey() string {
markerString := []string{}
for _, m := range g.Markers {
markerString = append(markerString, m.String())
}
overlayString := []string{}
for _, o := range g.Overlays {
overlayString = append(overlayString, o.URLPattern)
}
hashString := fmt.Sprintf("%s:::%s|%d|%s|%dx%d|%v|%s",
version,
g.Center.String(),
g.Zoom,
strings.Join(markerString, "+"),
g.Width,
g.Height,
g.DisableAttribution,
fmt.Sprintf("%x", sha256.Sum256([]byte(strings.Join(overlayString, "::")))),
)
return fmt.Sprintf("%x", sha256.Sum256([]byte(hashString)))
}
func generateMap(opts generateMapConfig) (io.Reader, error) {
ctx := staticMap.NewContext()
ctx.SetUserAgent(fmt.Sprintf("Mozilla/5.0+(compatible; staticmap/%s; https://github.com/Luzifer/staticmap)", version))
ctx.SetSize(opts.Width, opts.Height)
ctx.SetCenter(opts.Center)
ctx.SetZoom(opts.Zoom)
if opts.DisableAttribution {
ctx.OverrideAttribution("")
}
if opts.Markers != nil {
for _, m := range opts.Markers {
ctx.AddObject(staticMap.NewMarker(m.pos, m.color, float64(m.size)))
}
}
if opts.Overlays != nil {
for _, o := range opts.Overlays {
ctx.AddOverlay(o)
}
}
img, err := ctx.Render()
if err != nil {
return nil, errors.Wrap(err, "rendering context")
}
pngCtx := gg.NewContextForImage(img)
pngBuf := new(bytes.Buffer)
return pngBuf, errors.Wrap(pngCtx.EncodePNG(pngBuf), "encoding to PNG")
}