package main import ( "bytes" "context" "crypto/rand" "fmt" "io" "log" "math/big" "net/http" "net/url" "time" h "gitlab.codemonkeysoftware.net/b/hatmill" a "gitlab.codemonkeysoftware.net/b/hatmill/attribute" e "gitlab.codemonkeysoftware.net/b/hatmill/element" "gitlab.codemonkeysoftware.net/b/henwen" _ "gitlab.codemonkeysoftware.net/b/henwen" ) const Addr = ":8080" const Title = "Henwen" const baseURL = "http://localhost:8080" const dbFileName = "./henwen.db" const ( pathRoot = "/" pathCreate = "/create" pathDoCreate = "/create/do" pathAdmin = "/admin" ) var store *henwen.Store func writePage(w io.Writer, contents h.Term) error { page := e.Html()( e.Head()( e.Title()(h.Text(Title)), ), e.Body()( e.H1()(h.Text(Title)), e.Div()(contents), ), ) _, err := h.WriteDocument(w, page) return err } func pageRoot() h.Term { return h.Terms{ e.H2()(h.Text("Welcome!")), e.A(a.Href(pathCreate))( h.Text("Create event"), ), } } const ( fieldNameEarliest = "earliestDate" fieldNameLatest = "latestDate" fieldNameEventName = "eventName" fieldNameDescription = "eventDescription" ) func pageCreate() h.Term { return h.Terms{ e.H2()(h.Text("Create an event")), e.Form(a.Action(pathDoCreate), a.Method(http.MethodPost))( e.Label(a.For(fieldNameEventName))(h.Text("Event name")), e.Input(a.Name(fieldNameEventName)), e.Label(a.For(fieldNameDescription))(h.Text("Description")), e.Textarea(a.Name(fieldNameEventName), a.Placeholder("What's going on?"))(), e.Label(a.For(fieldNameEarliest))(h.Text("Earliest date")), e.Input(a.Name(fieldNameEarliest), a.Type("date")), e.Label(a.For(fieldNameLatest))(h.Text("Latest date")), e.Input(a.Name(fieldNameLatest), a.Type("date")), e.Input(a.Type("submit")), ), } } func pageAdmin(alphaID, adminCode string) h.Term { event, err := store.GetEvent(context.Background(), henwen.GetEventQuery{ AlphaID: alphaID, AdminCode: adminCode, }) if err != nil { return h.Text(err.Error()) } return h.Terms{ e.H2()(h.Text("Edit your event")), e.Form(a.Action(pathDoCreate), a.Method(http.MethodPost))( e.Label(a.For(fieldNameEventName))(h.Text("Event name")), e.Input( a.Name(fieldNameEventName), a.Value(event.Name), ), e.Br(), e.Label(a.For(fieldNameDescription))(h.Text("Description")), e.Textarea(a.Name(fieldNameEventName))(h.Text(event.Description)), e.Br(), e.Label(a.For(fieldNameEarliest))(h.Text("Earliest date")), e.Input( a.Name(fieldNameEarliest), a.Type("date"), a.Value(event.EarliestDate.Format(dateFmt)), ), e.Br(), e.Label(a.For(fieldNameLatest))(h.Text("Latest date")), e.Input( a.Name(fieldNameLatest), a.Type("date"), a.Value(event.LatestDate.Format(dateFmt)), ), e.Br(), e.Input(a.Type("submit")), ), } } const keyEventID = "event_id" const keyAdminCode = "admin_code" func pageDoCreate(name, description string, earliest, latest time.Time) h.Term { event, err := store.CreateEvent(context.Background(), henwen.CreateEventCommand{ Name: name, Description: description, EarliestDate: earliest, LatestDate: latest, Duration: 1, }) if err != nil { return h.Text(err.Error()) } var adminQuery = make(url.Values) adminQuery.Add(keyEventID, event.AlphaID) adminQuery.Add(keyAdminCode, event.AdminCode) adminURL := baseURL + pathAdmin + "?" + adminQuery.Encode() const dateDisplayFmt = "Monday, January 2, 2006" return h.Terms{ e.H2()(h.Text("Created event!")), e.P()( h.Text("You can find it again at "), e.A(a.Href(adminURL))(h.Text(adminURL)), h.Text("."), ), e.H3()(h.Text("Name")), h.Text(name), e.H3()(h.Text("Earliest date")), h.Text(earliest.Format(dateDisplayFmt)), e.H3()(h.Text("Latest date")), h.Text(latest.Format(dateDisplayFmt)), } } var chars = []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ") func genString() (string, error) { const length = 10 charsLength := big.NewInt(int64(len(chars))) var maxN big.Int maxN.Exp(charsLength, big.NewInt(length), nil) n, err := rand.Int(rand.Reader, &maxN) if err != nil { return "", err } var buf bytes.Buffer for n.Cmp(&big.Int{}) == 1 { var charIdx big.Int n.DivMod(n, charsLength, &charIdx) _ = buf.WriteByte(chars[charIdx.Int64()]) } return buf.String(), nil } const dateFmt = "2006-01-02" func main() { var err error store, err = henwen.NewStore(dbFileName, genString) if err != nil { log.Fatal(err) } mux := http.NewServeMux() mux.HandleFunc(pathRoot, func(w http.ResponseWriter, r *http.Request) { _ = writePage(w, pageRoot()) }) mux.HandleFunc(pathCreate, func(w http.ResponseWriter, r *http.Request) { _ = writePage(w, pageCreate()) }) mux.HandleFunc(pathDoCreate, func(w http.ResponseWriter, r *http.Request) { earliest, err := time.Parse(dateFmt, r.FormValue(fieldNameEarliest)) if err != nil { fmt.Fprint(w, "bad earliest date") return } latest, err := time.Parse(dateFmt, r.FormValue(fieldNameLatest)) if err != nil { fmt.Fprint(w, "bad latest date") return } eventName := r.FormValue(fieldNameEventName) if eventName == "" { fmt.Fprint(w, "event name is required") return } description := r.FormValue(fieldNameDescription) _ = writePage(w, pageDoCreate(eventName, description, earliest, latest)) }) mux.HandleFunc(pathAdmin, func(w http.ResponseWriter, r *http.Request) { query := r.URL.Query() _ = writePage(w, pageAdmin(query.Get(keyEventID), query.Get(keyAdminCode))) }) srv := http.Server{ Addr: Addr, Handler: mux, } log.Println(srv.ListenAndServe()) }