Home | History | Annotate | Download | only in wiki
      1 // Copyright 2010 The Go Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style
      3 // license that can be found in the LICENSE file.
      4 
      5 package main
      6 
      7 import (
      8 	"html/template"
      9 	"io/ioutil"
     10 	"net/http"
     11 )
     12 
     13 type Page struct {
     14 	Title string
     15 	Body  []byte
     16 }
     17 
     18 func (p *Page) save() error {
     19 	filename := p.Title + ".txt"
     20 	return ioutil.WriteFile(filename, p.Body, 0600)
     21 }
     22 
     23 func loadPage(title string) (*Page, error) {
     24 	filename := title + ".txt"
     25 	body, err := ioutil.ReadFile(filename)
     26 	if err != nil {
     27 		return nil, err
     28 	}
     29 	return &Page{Title: title, Body: body}, nil
     30 }
     31 
     32 func editHandler(w http.ResponseWriter, r *http.Request) {
     33 	title := r.URL.Path[len("/edit/"):]
     34 	p, err := loadPage(title)
     35 	if err != nil {
     36 		p = &Page{Title: title}
     37 	}
     38 	renderTemplate(w, "edit", p)
     39 }
     40 
     41 func viewHandler(w http.ResponseWriter, r *http.Request) {
     42 	title := r.URL.Path[len("/view/"):]
     43 	p, _ := loadPage(title)
     44 	renderTemplate(w, "view", p)
     45 }
     46 
     47 func saveHandler(w http.ResponseWriter, r *http.Request) {
     48 	title := r.URL.Path[len("/save/"):]
     49 	body := r.FormValue("body")
     50 	p := &Page{Title: title, Body: []byte(body)}
     51 	p.save()
     52 	http.Redirect(w, r, "/view/"+title, http.StatusFound)
     53 }
     54 
     55 func renderTemplate(w http.ResponseWriter, tmpl string, p *Page) {
     56 	t, _ := template.ParseFiles(tmpl + ".html")
     57 	t.Execute(w, p)
     58 }
     59 
     60 func main() {
     61 	http.HandleFunc("/view/", viewHandler)
     62 	http.HandleFunc("/edit/", editHandler)
     63 	http.HandleFunc("/save/", saveHandler)
     64 	http.ListenAndServe(":8080", nil)
     65 }
     66