96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package yatwiki3
|
|
|
|
import (
|
|
"errors"
|
|
"html/template"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type WikiServer struct {
|
|
db *WikiDB
|
|
opts *ServerOptions
|
|
pageTmp *template.Template
|
|
}
|
|
|
|
func NewWikiServer(opts *ServerOptions) (*WikiServer, error) {
|
|
wdb, err := NewWikiDB(opts.DBFilePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tmpl, err := template.New("yatwiki/page").Parse(pageTemplate)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
ws := WikiServer{
|
|
db: wdb,
|
|
opts: opts,
|
|
pageTmp: tmpl,
|
|
}
|
|
return &ws, nil
|
|
}
|
|
|
|
func (this *WikiServer) Close() {
|
|
this.db.Close()
|
|
}
|
|
|
|
func (this *WikiServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Server", "YATWiki3")
|
|
|
|
if r.Method == "GET" {
|
|
if r.URL.Path == this.opts.ExpectBaseURL+"wiki.css" {
|
|
w.Header().Set("Content-Type", "text/css")
|
|
content, _ := wikiCssBytes()
|
|
w.Write(content)
|
|
return
|
|
|
|
} else if r.URL.Path == this.opts.ExpectBaseURL+"highlight.js" {
|
|
w.Header().Set("Content-Type", "application/javascript")
|
|
content, _ := highlightJsBytes()
|
|
w.Write(content)
|
|
return
|
|
|
|
} else if r.URL.Path == this.opts.ExpectBaseURL+"favicon.ico" && len(this.opts.FaviconFilePath) > 0 {
|
|
w.Header().Set("Content-Type", "image/x-icon")
|
|
http.ServeFile(w, r, this.opts.FaviconFilePath)
|
|
return
|
|
|
|
} else if r.URL.Path == this.opts.ExpectBaseURL+"formatting" {
|
|
this.routeFormatting(w, r)
|
|
return
|
|
|
|
} else if r.URL.Path == this.opts.ExpectBaseURL+"index" {
|
|
this.routeIndex(w, r)
|
|
return
|
|
|
|
} else if strings.HasPrefix(r.URL.Path, this.opts.ExpectBaseURL+"view/") {
|
|
articleTitle, err := url.QueryUnescape(r.URL.Path[len(this.opts.ExpectBaseURL+"view/"):])
|
|
if err != nil {
|
|
this.serveErrorMessage(w, err)
|
|
return
|
|
}
|
|
this.routeView(w, r, articleTitle)
|
|
return
|
|
|
|
} else if strings.HasPrefix(r.URL.Path, this.opts.ExpectBaseURL+"raw/") {
|
|
revId, err := strconv.Atoi(r.URL.Path[len(this.opts.ExpectBaseURL+"raw/"):])
|
|
if err != nil {
|
|
this.serveErrorMessage(w, err)
|
|
return
|
|
}
|
|
|
|
this.routeRawView(w, r, revId)
|
|
return
|
|
}
|
|
|
|
}
|
|
|
|
// No match? Add 'Page not found' to next session response, and redirect to homepage
|
|
this.serveErrorMessage(w, errors.New("Page not found"))
|
|
|
|
}
|