123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- package server
- import (
- "fmt"
- "net/http"
- "net/url"
- "github.com/labstack/echo/v4"
- getter "github.com/usememos/memos/plugin/http-getter"
- )
- func registerGetterPublicRoutes(g *echo.Group) {
- g.GET("/get/httpmeta", func(c echo.Context) error {
- urlStr := c.QueryParam("url")
- if urlStr == "" {
- return echo.NewHTTPError(http.StatusBadRequest, "Missing website url")
- }
- if _, err := url.Parse(urlStr); err != nil {
- return echo.NewHTTPError(http.StatusBadRequest, "Wrong url").SetInternal(err)
- }
- htmlMeta, err := getter.GetHTMLMeta(urlStr)
- if err != nil {
- return echo.NewHTTPError(http.StatusNotAcceptable, fmt.Sprintf("Failed to get website meta with url: %s", urlStr)).SetInternal(err)
- }
- return c.JSON(http.StatusOK, composeResponse(htmlMeta))
- })
- g.GET("/get/image", func(c echo.Context) error {
- urlStr := c.QueryParam("url")
- if urlStr == "" {
- return echo.NewHTTPError(http.StatusBadRequest, "Missing image url")
- }
- if _, err := url.Parse(urlStr); err != nil {
- return echo.NewHTTPError(http.StatusBadRequest, "Wrong url").SetInternal(err)
- }
- image, err := getter.GetImage(urlStr)
- if err != nil {
- return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Failed to get image url: %s", urlStr)).SetInternal(err)
- }
- c.Response().Writer.WriteHeader(http.StatusOK)
- c.Response().Writer.Header().Set("Content-Type", image.Mediatype)
- c.Response().Writer.Header().Set(echo.HeaderCacheControl, "max-age=31536000, immutable")
- if _, err := c.Response().Writer.Write(image.Blob); err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to write image blob").SetInternal(err)
- }
- return nil
- })
- }
|