138 lines
3.4 KiB
Go
138 lines
3.4 KiB
Go
package yatwiki3
|
|
|
|
import (
|
|
"database/sql"
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
"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" {
|
|
pto := DefaultPageTemplateOptions(this.opts)
|
|
pto.CurrentPageName = "Formatting help"
|
|
pto.Content = `
|
|
<h2>Formatting help</h2><br><br>
|
|
<ul>
|
|
<li>[h]header[/h]</li>
|
|
<li>[b]bold[/b]</li>
|
|
<li>[u]underline[/u]</li>
|
|
<li>[i]italic[/i]</li>
|
|
<li>[s]strikethrough[/s]</li>
|
|
<li>[spoiler]spoiler[/spoiler]</li>
|
|
<li>[list] item [*] item [/list]</li>
|
|
<li>[url=address]title[/url]</li>
|
|
<li>[article=page name]title[/article] or [rev=id]title[/rev]</li>
|
|
<li>[img]image-url[/img]</li>
|
|
<li>[imgur]asdf.jpg[/imgur]</li>
|
|
<li>[code]fixed width[/code]</li>
|
|
<li>[section=header]content[/section]</li>
|
|
<li>[html]raw html[/html]</li>
|
|
</ul>`
|
|
this.servePageResponse(w, pto)
|
|
return
|
|
|
|
} else if strings.HasPrefix(r.URL.Path, this.opts.ExpectBaseURL+"view/") {
|
|
articleTitle := r.URL.Path[len(this.opts.ExpectBaseURL+"view/"):]
|
|
a, err := this.db.GetLatestVersion(articleTitle)
|
|
if err != nil {
|
|
this.serveErrorMessage(w, err)
|
|
return
|
|
}
|
|
|
|
pto := DefaultPageTemplateOptions(this.opts)
|
|
pto.CurrentPageName = articleTitle
|
|
pto.CurrentPageIsArticle = true
|
|
|
|
bcr := NewBBCodeRenderer(this.opts.ExpectBaseURL)
|
|
pto.Content = bcr.RenderHTML(string(a.Body))
|
|
pto.LoadCodeResources = bcr.CodePresent
|
|
|
|
this.servePageResponse(w, pto)
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// No match? Add 'Page not found' to next session response, and redirect to homepage
|
|
http.Error(w, "Page not found", 404)
|
|
|
|
/*
|
|
pto.SessionMessage = `Invalid request.`
|
|
//pto.CurrentPageIsArticle = true
|
|
//pto.CurrentPageName = "quotes/\"2017"
|
|
|
|
bcr := NewBBCodeRenderer(this.opts.ExpectBaseURL)
|
|
pto.Content = bcr.RenderHTML(`[h]Hello World[/h] this content is [b]bold[/b] <script>alert("hi");</script>`)
|
|
|
|
this.servePageResponse(w, pto)
|
|
*/
|
|
}
|
|
|
|
func (this *WikiServer) serveErrorMessage(w http.ResponseWriter, message error) {
|
|
if message == sql.ErrNoRows {
|
|
// 404 not found
|
|
}
|
|
|
|
// TODO Add error message to session response, and redirect to homepage
|
|
http.Error(w, message.Error(), 500)
|
|
}
|
|
|
|
func (this *WikiServer) servePageResponse(w http.ResponseWriter, pto *pageTemplateOptions) {
|
|
err := this.pageTmp.Execute(w, pto)
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
}
|
|
}
|