渲染readme

2023-05-29 10:20:53 +08:00
parent 590113efbe
commit 35fd7495aa
2 changed files with 47 additions and 19 deletions
--- a/main.go
+++ b/main.go
@@ -1,7 +1,6 @@
 package main

 import (
-	"io/ioutil"
 	"log"
 	"net/http"
 	"runtime"
@@ -11,7 +10,6 @@ import (
 	"main/utils"

 	"github.com/gorilla/mux"
-	"github.com/russross/blackfriday"
 )

 func main() {
@@ -37,23 +35,7 @@ func main() {
 	})

 	// 設定路由
-	r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-		input, err := ioutil.ReadFile("./README.md")
-		if err != nil {
-			log.Println(err)
-			return
-		}
-		output := blackfriday.Markdown(input, blackfriday.HtmlRenderer(0, "", ""), blackfriday.EXTENSION_TABLES|blackfriday.EXTENSION_FENCED_CODE|blackfriday.EXTENSION_AUTOLINK)
-		css := `<style type="text/css">
-			body {
-				max-width: 960px;
-				margin: 0 auto;
-			}
-		</style>`
-		html := "<!DOCTYPE html><html><head><title>API Document</title>" + css + "</head><body>" + string(output) + "</body></html>"
-		w.Header().Set("Content-Type", "text/html; charset=utf-8")
-		w.Write([]byte(html))
-	})
+	r.HandleFunc("/", routers.GetDocs).Methods("GET")

 	r.HandleFunc("/api/users", routers.UsersGet).Methods("GET")
 	r.HandleFunc("/api/users", routers.UsersPost).Methods("POST")