Wiki codelab, complete with tests.

R=r, rsc, gri
CC=golang-dev
http://codereview.appspot.com/887045

1 files changed, 93 insertions, 0 deletions

diff --git a/doc/codelab/wiki/final.go b/doc/codelab/wiki/final.go
new file mode 100644
index 000000000..0186729c2
--- /dev/null
+++ b/doc/codelab/wiki/final.go
@@ -0,0 +1,93 @@
+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)
+}
+
+var templates = make(map[string]*template.Template)
+
+func init() {
+	for _, tmpl := range []string{"edit", "view"} {
+		templates[tmpl] = template.MustParseFile(tmpl+".html", nil)
+	}
+}
+
+func renderTemplate(c *http.Conn, tmpl string, p *page) {
+	err := templates[tmpl].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)
+}