summaryrefslogtreecommitdiff
path: root/doc/codelab/wiki/final-parsetemplate.go
diff options
context:
space:
mode:
Diffstat (limited to 'doc/codelab/wiki/final-parsetemplate.go')
-rw-r--r--doc/codelab/wiki/final-parsetemplate.go90
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)
+}