diff options
Diffstat (limited to 'doc/codelab/wiki/final-parsetemplate.go')
-rw-r--r-- | doc/codelab/wiki/final-parsetemplate.go | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/doc/codelab/wiki/final-parsetemplate.go b/doc/codelab/wiki/final-parsetemplate.go new file mode 100644 index 000000000..93b956b9d --- /dev/null +++ b/doc/codelab/wiki/final-parsetemplate.go @@ -0,0 +1,90 @@ +package main + +import ( + "http" + "io/ioutil" + "os" + "regexp" + "template" +) + +type page struct { + title string + body []byte +} + +func (p *page) save() os.Error { + filename := p.title + ".txt" + return ioutil.WriteFile(filename, p.body, 0600) +} + +func loadPage(title string) (*page, os.Error) { + filename := title + ".txt" + body, err := ioutil.ReadFile(filename) + if err != nil { + return nil, err + } + return &page{title: title, body: body}, nil +} + +func viewHandler(c *http.Conn, r *http.Request, title string) { + p, err := loadPage(title) + if err != nil { + http.Redirect(c, "/edit/"+title, http.StatusFound) + return + } + renderTemplate(c, "view", p) +} + +func editHandler(c *http.Conn, r *http.Request, title string) { + p, err := loadPage(title) + if err != nil { + p = &page{title: title} + } + renderTemplate(c, "edit", p) +} + +func saveHandler(c *http.Conn, r *http.Request, title string) { + body := r.FormValue("body") + p := &page{title: title, body: []byte(body)} + err := p.save() + if err != nil { + http.Error(c, err.String(), http.StatusInternalServerError) + return + } + http.Redirect(c, "/view/"+title, http.StatusFound) +} + +func renderTemplate(c *http.Conn, tmpl string, p *page) { + t, err := template.ParseFile(tmpl+".html", nil) + if err != nil { + http.Error(c, err.String(), http.StatusInternalServerError) + return + } + err = t.Execute(p, c) + if err != nil { + http.Error(c, err.String(), http.StatusInternalServerError) + } +} + +const lenPath = len("/view/") + +var titleValidator = regexp.MustCompile("^[a-zA-Z0-9]+$") + +func makeHandler(fn func(*http.Conn, *http.Request, string)) http.HandlerFunc { + return func(c *http.Conn, r *http.Request) { + title := r.URL.Path[lenPath:] + if !titleValidator.MatchString(title) { + http.NotFound(c, r) + return + } + fn(c, r, title) + } +} + +func main() { + http.HandleFunc("/view/", makeHandler(viewHandler)) + http.HandleFunc("/edit/", makeHandler(editHandler)) + http.HandleFunc("/save/", makeHandler(saveHandler)) + http.ListenAndServe(":8080", nil) +} |