YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: implement memo route

Browse Source
This commit is contained in:
Steven
2023-12-15 07:32:49 +08:00
parent 6cf7192d6a
commit 52743017a3
3 changed files with 58 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
server/frontend/frontend.go
View File

@ -2,6 +2,7 @@ package frontend
import (
"embed"
"fmt"
"io/fs"
"net/http"
"strings"
@ -10,6 +11,8 @@ import (
"github.com/labstack/echo/v4/middleware"
"github.com/usememos/memos/internal/util"
"github.com/usememos/memos/server/profile"
"github.com/usememos/memos/store"
)
//go:embed dist
@ -18,7 +21,19 @@ var embeddedFiles embed.FS
//go:embed dist/index.html
var rawIndexHTML string
func Serve(e *echo.Echo) {
type FrontendService struct {
Profile *profile.Profile
Store *store.Store
}
func NewFrontendService(profile *profile.Profile, store *store.Store) *FrontendService {
return &FrontendService{
Profile: profile,
Store: store,
}
}
func (s *FrontendService) Serve(e *echo.Echo) {
// Use echo static middleware to serve the built dist folder
// refer: https://github.com/labstack/echo/blob/master/middleware/static.go
e.Use(middleware.StaticWithConfig(middleware.StaticConfig{
@ -44,16 +59,50 @@ func Serve(e *echo.Echo) {
Filesystem: getFileSystem("dist/assets"),
}))
registerRoutes(e)
s.registerRoutes(e)
}
func registerRoutes(e *echo.Echo) {
func (s *FrontendService) registerRoutes(e *echo.Echo) {
e.GET("/m/:memoID", func(c echo.Context) error {
indexHTML := strings.ReplaceAll(rawIndexHTML, "<!-- memos.metadata -->", "<meta name=\"memos-memo-id\" content=\""+c.Param("memoID")+"\">"+"\n")
ctx := c.Request().Context()
memoID, err := util.ConvertStringToInt32(c.Param("memoID"))
if err != nil {
return echo.NewHTTPError(http.StatusBadRequest, "invalid memo id")
}
memo, err := s.Store.GetMemo(ctx, &store.FindMemo{
ID: &memoID,
})
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, "failed to retrieve memo")
}
if memo == nil {
return echo.NewHTTPError(http.StatusNotFound, "memo not found")
}
if memo.Visibility != store.Public {
return echo.NewHTTPError(http.StatusForbidden, "memo is not public")
}
indexHTML := strings.ReplaceAll(rawIndexHTML, "<!-- memos.metadata -->", generateMemoMetadata(memo))
return c.HTML(http.StatusOK, indexHTML)
})
}
func generateMemoMetadata(memo *store.Memo) string {
metadataList := []string{
fmt.Sprintf(`<meta name="description" content="%s" />`, memo.Content),
fmt.Sprintf(`<meta property="og:title" content="%s" />`, fmt.Sprintf("Memos - %d", memo.ID)),
fmt.Sprintf(`<meta property="og:description" content="%s" />`, memo.Content),
fmt.Sprintf(`<meta property="og:image" content="%s" />`, "https://www.usememos.com/logo.png"),
`<meta property="og:type" content="website" />`,
// Twitter related metadata.
fmt.Sprintf(`<meta name="twitter:title" content="%s" />`, fmt.Sprintf("Memos - %d", memo.ID)),
fmt.Sprintf(`<meta name="twitter:description" content="%s" />`, memo.Content),
fmt.Sprintf(`<meta name="twitter:image" content="%s" />`, "https://www.usememos.com/logo.png"),
`<meta name="twitter:card" content="summary" />`,
}
return strings.Join(metadataList, "\n")
}
func getFileSystem(path string) http.FileSystem {
fs, err := fs.Sub(embeddedFiles, path)
if err != nil {