From 0d50f5bd08f8f16b5f76179978f301d4f161cc95 Mon Sep 17 00:00:00 2001
From: boojack <stevenlgtm@gmail.com>
Date: Wed, 6 Sep 2023 21:59:20 +0800
Subject: [PATCH] chore: update comments (#2195)

---
 server/server.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/server/server.go b/server/server.go
index a3c16f7b..3edd4c85 100644
--- a/server/server.go
+++ b/server/server.go
@@ -85,10 +85,13 @@ func NewServer(ctx context.Context, profile *profile.Profile, store *store.Store
 	}
 	s.ID = serverID
 
+	// Serve frontend.
 	embedFrontend(e)
 
-	// This will serve Swagger UI at /api/index.html and Swagger 2.0 spec at /api/doc.json
-	e.GET("/api/*", echoSwagger.WrapHandler)
+	// Serve swagger in dev/demo mode.
+	if profile.Mode == "dev" || profile.Mode == "demo" {
+		e.GET("/api/*", echoSwagger.WrapHandler)
+	}
 
 	secret := "usememos"
 	if profile.Mode == "prod" {