feat: implement redirector api

This commit is contained in:
Steven 2023-06-23 10:03:47 +08:00
parent 2aae515544
commit 60da9b7e7b
6 changed files with 53 additions and 13 deletions

37
api/v1/redirector.go Normal file
View File

@ -0,0 +1,37 @@
package v1
import (
"net/http"
"github.com/boojack/shortify/store"
"github.com/labstack/echo/v4"
)
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, "Failed to get shortcut").SetInternal(err)
}
if shortcut == nil {
return echo.NewHTTPError(http.StatusNotFound, "Shortcut not found")
}
if shortcut.Visibility != store.VisibilityPublic {
userID, ok := c.Get(getUserIDContextKey()).(int)
if !ok {
return echo.NewHTTPError(http.StatusUnauthorized, "Unauthorized")
}
if shortcut.Visibility == store.VisibilityPrivate && shortcut.CreatorID != userID {
return echo.NewHTTPError(http.StatusUnauthorized, "Unauthorized")
}
}
return c.Redirect(http.StatusSeeOther, shortcut.Link)
})
}

View File

@ -19,9 +19,13 @@ func NewAPIV1Service(profile *profile.Profile, store *store.Store) *APIV1Service
} }
} }
func (s *APIV1Service) Start(apiV1Group *echo.Group, secret string) { func (s *APIV1Service) Start(apiGroup *echo.Group, secret string) {
apiV1Group := apiGroup.Group("/api/v1")
s.registerSystemRoutes(apiV1Group) s.registerSystemRoutes(apiV1Group)
s.registerAuthRoutes(apiV1Group, secret) s.registerAuthRoutes(apiV1Group, secret)
s.registerUserRoutes(apiV1Group) s.registerUserRoutes(apiV1Group)
s.registerShortcutRoutes(apiV1Group) s.registerShortcutRoutes(apiV1Group)
redirectorGroup := apiGroup.Group("/o")
s.registerRedirectorRoutes(redirectorGroup)
} }

View File

@ -73,22 +73,17 @@ func audienceContains(audience jwt.ClaimStrings, token string) bool {
// will try to generate new access token and refresh token. // will try to generate new access token and refresh token.
func JWTMiddleware(server *Server, next echo.HandlerFunc, secret string) echo.HandlerFunc { func JWTMiddleware(server *Server, next echo.HandlerFunc, secret string) echo.HandlerFunc {
return func(c echo.Context) error { return func(c echo.Context) error {
path := c.Request().URL.Path path := c.Path()
method := c.Request().Method method := c.Request().Method
if server.defaultAuthSkipper(c) { if server.defaultAuthSkipper(c) {
return next(c) return next(c)
} }
// Skip validation for server status endpoints.
if hasPrefixes(path, "/api/ping", "/api/v1/idp", "/api/user/:id") && method == http.MethodGet {
return next(c)
}
token := findAccessToken(c) token := findAccessToken(c)
if token == "" { if token == "" {
// When the request is not authenticated, we allow the user to access the shortcut endpoints for those public shortcuts. // When the request is not authenticated, we allow the user to access the shortcut endpoints for those public shortcuts.
if hasPrefixes(path, "/api/status", "/api/shortcut") && method == http.MethodGet { if hasPrefixes(path, "/api/v1/status", "/o/*") && method == http.MethodGet {
return next(c) return next(c)
} }
return echo.NewHTTPError(http.StatusUnauthorized, "Missing access token") return echo.NewHTTPError(http.StatusUnauthorized, "Missing access token")

View File

@ -60,13 +60,13 @@ func NewServer(profile *profile.Profile, store *store.Store) (*Server, error) {
} }
e.Use(session.Middleware(sessions.NewCookieStore([]byte(secret)))) e.Use(session.Middleware(sessions.NewCookieStore([]byte(secret))))
apiGroup := e.Group("")
// Register API v1 routes. // Register API v1 routes.
apiV1Service := apiv1.NewAPIV1Service(profile, store) apiV1Service := apiv1.NewAPIV1Service(profile, store)
apiV1Group := e.Group("/api/v1") apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
apiV1Group.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
return JWTMiddleware(s, next, string(secret)) return JWTMiddleware(s, next, string(secret))
}) })
apiV1Service.Start(apiV1Group, secret) apiV1Service.Start(apiGroup, secret)
return s, nil return s, nil
} }

View File

@ -31,7 +31,7 @@ const ShortcutListView: React.FC<Props> = (props: Props) => {
}; };
const handleCopyButtonClick = (shortcut: Shortcut) => { const handleCopyButtonClick = (shortcut: Shortcut) => {
copy(absolutifyLink(`/${shortcut.name}`)); copy(absolutifyLink(`/o/${shortcut.name}`));
toast.success("Shortcut link copied to clipboard."); toast.success("Shortcut link copied to clipboard.");
}; };

View File

@ -1,5 +1,5 @@
import { defineConfig } from "vite";
import react from "@vitejs/plugin-react-swc"; import react from "@vitejs/plugin-react-swc";
import { defineConfig } from "vite";
// https://vitejs.dev/config/ // https://vitejs.dev/config/
export default defineConfig({ export default defineConfig({
@ -12,6 +12,10 @@ export default defineConfig({
target: "http://localhost:8082/", target: "http://localhost:8082/",
changeOrigin: true, changeOrigin: true,
}, },
"/o": {
target: "http://localhost:8082/",
changeOrigin: true,
},
}, },
}, },
}); });