09-Pagination add create post

controller/g.go

@@ -10,13 +10,15 @@ var (
 	homeController home
 	templates      map[string]*template.Template
 	sessionName    string
+	flashName      string
 	store          *sessions.CookieStore
 )
 
 func init() {
 	templates = PopulateTemplates()
 	store = sessions.NewCookieStore([]byte("something-very-secret"))
 	sessionName = "go-mega"
+	flashName = "go-flash"
 }
 
 // Startup func

controller/home.go

@@ -29,8 +29,27 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
 	tpName := "index.html"
 	vop := vm.IndexViewModelOp{}
 	username, _ := getSessionUser(r)
-	v := vop.GetVM(username)
-	templates[tpName].Execute(w, &v)
+	if r.Method == http.MethodGet {
+		flash := getFlash(w, r)
+		v := vop.GetVM(username, flash)
+		templates[tpName].Execute(w, &v)
+	}
+	if r.Method == http.MethodPost {
+		r.ParseForm()
+		body := r.Form.Get("body")
+		errMessage := checkLen("Post", body, 1, 180)
+		if errMessage != "" {
+			setFlash(w, r, errMessage)
+		} else {
+			err := vm.CreatePost(username, body)
+			if err != nil {
+				log.Println("add Post error:", err)
+				w.Write([]byte("Error insert Post in database"))
+				return
+			}
+		}
+		http.Redirect(w, r, "/", http.StatusSeeOther)
+	}
 }
 
 func loginHandler(w http.ResponseWriter, r *http.Request) {

controller/utils.go

@@ -176,3 +176,19 @@ func checkRegister(username, email, pwd1, pwd2 string) []string {
 func addUser(username, password, email string) error {
 	return vm.AddUser(username, password, email)
 }
+
+func setFlash(w http.ResponseWriter, r *http.Request, message string) {
+	session, _ := store.Get(r, sessionName)
+	session.AddFlash(message, flashName)
+	session.Save(r, w)
+}
+
+func getFlash(w http.ResponseWriter, r *http.Request) string {
+	session, _ := store.Get(r, sessionName)
+	fm := session.Flashes(flashName)
+	if fm == nil {
+		return ""
+	}
+	session.Save(r, w)
+	return fmt.Sprintf("%v", fm[0])
+}

templates/content/index.html

@@ -1,7 +1,23 @@
 {{define "content"}}
     <h1>Hello, {{.CurrentUser}}!</h1>
 
+    <form action="/" method="post">
+
+        <p><textarea  name="body" rows="3" cols="80" value="" placeholder="say something..."></textarea></p>
+        <p><input type="submit" name="submit" value="Post"></p>
+
+        {{ if .Flash }}
+            <span style="color: red;">[{{.Flash}}]</span>
+        {{ end }}
+    </form>
+
+
     {{range .Posts}}
-        <div><p>{{ .User.Username }} says: <b>{{ .Body }}</b></p></div>
+        <table>
+            <tr valign="top">
+                <td><img src="{{.User.Avatar}}&s=36"></td>
+                <td>{{ .User.Username }} says:<br>{{ .Body }}</td>
+            </tr>
+        </table>
     {{end}}
 {{end}}

vm/index.go

@@ -6,16 +6,23 @@ import "github.com/bonfy/go-mega-code/model"
 type IndexViewModel struct {
 	BaseViewModel
 	Posts []model.Post
+	Flash string
 }
 
 // IndexViewModelOp struct
 type IndexViewModelOp struct{}
 
 // GetVM func
-func (IndexViewModelOp) GetVM(username string) IndexViewModel {
-	u1, _ := model.GetUserByUsername(username)
-	posts, _ := model.GetPostsByUserID(u1.ID)
-	v := IndexViewModel{BaseViewModel{Title: "Homepage"}, *posts}
+func (IndexViewModelOp) GetVM(username string, flash string) IndexViewModel {
+	u, _ := model.GetUserByUsername(username)
+	posts, _ := u.FollowingPosts()
+	v := IndexViewModel{BaseViewModel{Title: "Homepage"}, *posts, flash}
 	v.SetCurrentUser(username)
 	return v
 }
+
+// CreatePost func
+func CreatePost(username, post string) error {
+	u, _ := model.GetUserByUsername(username)
+	return u.CreatePost(post)
+}