diff options
Diffstat (limited to 'doc/codelab/wiki/final.go')
| -rw-r--r-- | doc/codelab/wiki/final.go | 93 | 
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) +} | 
