package v1 import ( "encoding/json" "fmt" "html" "net/http" "net/url" "strings" "github.com/labstack/echo/v4" "github.com/pkg/errors" storepb "github.com/boojack/slash/proto/gen/store" "github.com/boojack/slash/server/metric" "github.com/boojack/slash/store" ) func (s *APIV1Service) registerRedirectorRoutes(g *echo.Group) { g.GET("/*", func(c echo.Context) error { ctx := c.Request().Context() if len(c.ParamValues()) == 0 { return echo.NewHTTPError(http.StatusBadRequest, "invalid shortcut name") } shortcutName := c.ParamValues()[0] shortcut, err := s.Store.GetShortcut(ctx, &store.FindShortcut{ Name: &shortcutName, }) if err != nil { return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get shortcut, err: %s", err)).SetInternal(err) } if shortcut == nil { return c.Redirect(http.StatusSeeOther, fmt.Sprintf("/404?shortcut=%s", shortcutName)) } if shortcut.Visibility != storepb.Visibility_PUBLIC { userID, ok := c.Get(userIDContextKey).(int32) if !ok { return echo.NewHTTPError(http.StatusUnauthorized, "Unauthorized") } if shortcut.Visibility == storepb.Visibility_PRIVATE && shortcut.CreatorId != userID { return echo.NewHTTPError(http.StatusUnauthorized, "Unauthorized") } } if err := s.createShortcutViewActivity(c, shortcut); err != nil { return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to create activity, err: %s", err)).SetInternal(err) } metric.Enqueue("shortcut redirect") return redirectToShortcut(c, shortcut) }) } func redirectToShortcut(c echo.Context, shortcut *storepb.Shortcut) error { isValidURL := isValidURLString(shortcut.Link) if shortcut.OgMetadata == nil || (shortcut.OgMetadata.Title == "" && shortcut.OgMetadata.Description == "" && shortcut.OgMetadata.Image == "") { if isValidURL { return c.Redirect(http.StatusSeeOther, shortcut.Link) } return c.String(http.StatusOK, shortcut.Link) } htmlTemplate := `
%s%s` metadataList := []string{ fmt.Sprintf(`