package ratchet
import (
"fmt"
"html/template"
"log"
"net/http"
"strconv"
)
type RatchetServer struct {
http.Handler
//Services used by HTTP routes
UserService UserService
}
func NewRatchetServer() *RatchetServer {
r := new(RatchetServer)
router := http.NewServeMux()
// /{$} is used to prevent subtree path patterns from acting like a wildcard
// resulting in this route requiring an exact match on "/" only
// You can only include one HTTP method in a route pattern if you choose
// GET will match GET & HEAD http request methods
router.HandleFunc("GET /{$}", home)
router.HandleFunc("GET /snippet/view/{id}", snippetView)
router.HandleFunc("GET /snippet/create", snippetCreate)
router.HandleFunc("POST /snippet/create", snippetCreatePost)
r.Handler = router
return r
}
func home(w http.ResponseWriter, r *http.Request) {
w.Header().Add("Server", "Go")
// Initialize a slice containing the paths to the two files. It's important
// to note that the file containing our base template must be the *first*
// file in the slice.
files := []string{
"./ui/html/base.go.tmpl",
"./ui/html/partials/nav.go.tmpl",
"./ui/html/pages/home.go.tmpl",
}
// read template file into template set.
ts, err := template.ParseFiles(files...)
if err != nil {
log.Print(err.Error())
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
return
}
// Write template content to response body
err = ts.ExecuteTemplate(w, "base", nil)
if err != nil {
log.Print(err.Error())
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
}
}
func snippetView(w http.ResponseWriter, r *http.Request) {
id, err := strconv.Atoi(r.PathValue("id"))
if err != nil || id < 1 {
http.NotFound(w, r)
return
}
// Set a new cache-control header. If an existing "Cache-Control" header exists
// it will be overwritten.
w.Header().Set("Cache-Control", "public, max-age=31536000")
// msg := fmt.Sprintf("Snippet %d...", id)
// w.Write([]byte(msg))
// we can rely on the Write() interface to use a differnent
// function to write out our response
fmt.Fprintf(w, "Snippet %d...", id)
}
// snippetCreate handles display of the form used to create snippets
func snippetCreate(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Displaying snippetCreate form..."))
}
// snippetCreate handles display of the form used to create snippets
func snippetCreatePost(w http.ResponseWriter, r *http.Request) {
// example of a custom header. Must be done before calling WriteHeader
// or they will fail to take effect.
w.Header().Add("Server", "Dirp")
w.WriteHeader(http.StatusCreated)
w.Write([]byte("Created snippet..."))
}