创建群组和主题

Author: nonfu

handlers/auth.go

@@ -0,0 +1,75 @@
+package handlers
+
+import (
+    "fmt"
+    "github.com/xueyuanjun/chitchat/models"
+    "net/http"
+)
+
+// GET /login
+// 登录页面
+func Login(writer http.ResponseWriter, request *http.Request) {
+    t := parseTemplateFiles("auth.layout", "navbar", "login")
+    t.Execute(writer, nil)
+}
+
+// GET /signup
+// 注册页面
+func Signup(writer http.ResponseWriter, request *http.Request) {
+    generateHTML(writer, nil, "auth.layout", "navbar", "signup")
+}
+
+// POST /signup
+// 注册新用户
+func SignupAccount(writer http.ResponseWriter, request *http.Request) {
+    err := request.ParseForm()
+    if err != nil {
+        fmt.Println("Cannot parse form")
+    }
+    user := models.User{
+        Name:     request.PostFormValue("name"),
+        Email:    request.PostFormValue("email"),
+        Password: request.PostFormValue("password"),
+    }
+    if err := user.Create(); err != nil {
+        fmt.Println("Cannot create user")
+    }
+    http.Redirect(writer, request, "/login", 302)
+}
+
+// POST /authenticate
+// 通过邮箱和密码字段对用户进行认证
+func Authenticate(writer http.ResponseWriter, request *http.Request) {
+    err := request.ParseForm()
+    user, err := models.UserByEmail(request.PostFormValue("email"))
+    if err != nil {
+        fmt.Println("Cannot find user")
+    }
+    if user.Password == models.Encrypt(request.PostFormValue("password")) {
+        session, err := user.CreateSession()
+        if err != nil {
+            fmt.Println("Cannot create session")
+        }
+        cookie := http.Cookie{
+            Name:     "_cookie",
+            Value:    session.Uuid,
+            HttpOnly: true,
+        }
+        http.SetCookie(writer, &cookie)
+        http.Redirect(writer, request, "/", 302)
+    } else {
+        http.Redirect(writer, request, "/login", 302)
+    }
+}
+
+// GET /logout
+// 用户退出
+func Logout(writer http.ResponseWriter, request *http.Request) {
+    cookie, err := request.Cookie("_cookie")
+    if err != http.ErrNoCookie {
+        fmt.Println("Failed to get cookie")
+        session := models.Session{Uuid: cookie.Value}
+        session.DeleteByUUID()
+    }
+    http.Redirect(writer, request, "/", 302)
+}

handlers/helper.go

@@ -0,0 +1,48 @@
+package handlers
+
+import (
+    "errors"
+    "fmt"
+    "github.com/xueyuanjun/chitchat/models"
+    "html/template"
+    "net/http"
+)
+
+// Checks if the user is logged in and has a session, if not err is not nil
+func session(writer http.ResponseWriter, request *http.Request) (sess models.Session, err error) {
+    cookie, err := request.Cookie("_cookie")
+    if err == nil {
+        sess = models.Session{Uuid: cookie.Value}
+        if ok, _ := sess.Check(); !ok {
+            err = errors.New("Invalid session")
+        }
+    }
+    return
+}
+
+// parse HTML templates
+// pass in a list of file names, and get a template
+func parseTemplateFiles(filenames ...string) (t *template.Template) {
+    var files []string
+    t = template.New("layout")
+    for _, file := range filenames {
+        files = append(files, fmt.Sprintf("views/%s.html", file))
+    }
+    t = template.Must(t.ParseFiles(files...))
+    return
+}
+
+func generateHTML(writer http.ResponseWriter, data interface{}, filenames ...string) {
+    var files []string
+    for _, file := range filenames {
+        files = append(files, fmt.Sprintf("views/%s.html", file))
+    }
+
+    templates := template.Must(template.ParseFiles(files...))
+    templates.ExecuteTemplate(writer, "layout", data)
+}
+
+// version
+func Version() string {
+    return "0.1"
+}

handlers/index.go

@@ -2,15 +2,17 @@ package handlers
 
 import (
     "github.com/xueyuanjun/chitchat/models"
-    "html/template"
     "net/http"
 )
 
-func Index(w http.ResponseWriter, r *http.Request) {
-    files := []string{"views/layout.html", "views/navbar.html", "views/index.html",}
-    templates := template.Must(template.ParseFiles(files...))
+func Index(writer http.ResponseWriter, request *http.Request) {
     threads, err := models.Threads();
     if err == nil {
-        templates.ExecuteTemplate(w, "layout", threads)
+        _, err := session(writer, request)
+        if err != nil {
+            generateHTML(writer, threads, "layout", "navbar", "index")
+        } else {
+            generateHTML(writer, threads, "layout", "auth.navbar", "index")
+        }
     }
 }

handlers/post.go

@@ -0,0 +1,36 @@
+package handlers
+
+import (
+    "fmt"
+    "github.com/xueyuanjun/chitchat/models"
+    "net/http"
+)
+
+// POST /thread/post
+// 在指定群组下创建新主题
+func PostThread(writer http.ResponseWriter, request *http.Request) {
+    sess, err := session(writer, request)
+    if err != nil {
+        http.Redirect(writer, request, "/login", 302)
+    } else {
+        err = request.ParseForm()
+        if err != nil {
+            fmt.Println("Cannot parse form")
+        }
+        user, err := sess.User()
+        if err != nil {
+            fmt.Println("Cannot get user from session")
+        }
+        body := request.PostFormValue("body")
+        uuid := request.PostFormValue("uuid")
+        thread, err := models.ThreadByUUID(uuid)
+        if err != nil {
+            fmt.Println("Cannot read thread")
+        }
+        if _, err := user.CreatePost(thread, body); err != nil {
+            fmt.Println("Cannot create post")
+        }
+        url := fmt.Sprint("/thread/read?id=", uuid)
+        http.Redirect(writer, request, url, 302)
+    }
+}

handlers/thread.go

@@ -0,0 +1,59 @@
+package handlers
+
+import (
+    "fmt"
+    "github.com/xueyuanjun/chitchat/models"
+    "net/http"
+)
+
+// GET /threads/new
+// 创建群组页面
+func NewThread(writer http.ResponseWriter, request *http.Request) {
+    _, err := session(writer, request)
+    if err != nil {
+        http.Redirect(writer, request, "/login", 302)
+    } else {
+        generateHTML(writer, nil, "layout", "auth.navbar", "new.thread")
+    }
+}
+
+// POST /thread/create
+// 执行群组创建逻辑
+func CreateThread(writer http.ResponseWriter, request *http.Request) {
+    sess, err := session(writer, request)
+    if err != nil {
+        http.Redirect(writer, request, "/login", 302)
+    } else {
+        err = request.ParseForm()
+        if err != nil {
+            fmt.Println("Cannot parse form")
+        }
+        user, err := sess.User()
+        if err != nil {
+            fmt.Println("Cannot get user from session")
+        }
+        topic := request.PostFormValue("topic")
+        if _, err := user.CreateThread(topic); err != nil {
+            fmt.Println("Cannot create thread")
+        }
+        http.Redirect(writer, request, "/", 302)
+    }
+}
+
+// GET /thread/read
+// 通过 ID 渲染指定群组页面
+func ReadThread(writer http.ResponseWriter, request *http.Request) {
+    vals := request.URL.Query()
+    uuid := vals.Get("id")
+    thread, err := models.ThreadByUUID(uuid)
+    if err != nil {
+        fmt.Println("Cannot read thread")
+    } else {
+        _, err := session(writer, request)
+        if err != nil {
+            generateHTML(writer, &thread, "layout", "navbar", "thread")
+        } else {
+            generateHTML(writer, &thread, "layout", "auth.navbar", "auth.thread")
+        }
+    }
+}

routes/routes.go

@@ -24,4 +24,58 @@ var webRoutes = WebRoutes{
         "/",
         handlers.Index,
     },
+    {
+        "signup",
+        "GET",
+        "/signup",
+        handlers.Signup,
+    },
+    {
+        "signupAccount",
+        "POST",
+        "/signup_account",
+        handlers.SignupAccount,
+    },
+    {
+        "login",
+        "GET",
+        "/login",
+        handlers.Login,
+    },
+    {
+        "auth",
+        "POST",
+        "/authenticate",
+        handlers.Authenticate,
+    },
+    {
+        "logout",
+        "GET",
+        "/logout",
+        handlers.Logout,
+    },
+    {
+            "newThread",
+            "GET",
+            "/thread/new",
+            handlers.NewThread,
+    },
+    {
+        "createThread",
+        "POST",
+        "/thread/create",
+        handlers.CreateThread,
+    },
+    {
+        "readThread",
+        "GET",
+        "/thread/read",
+        handlers.ReadThread,
+    },
+    {
+        "postThread",
+        "POST",
+        "/thread/post",
+        handlers.PostThread,
+    },
 }

views/auth.layout.html

@@ -0,0 +1,26 @@
+{{ define "layout" }}
+
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8">
+    <meta http-equiv="X-UA-Compatible" content="IE=9">
+    <meta name="viewport" content="width=device-width, initial-scale=1">
+    <title>ChitChat</title>
+    <link href="/static/css/bootstrap.min.css" rel="stylesheet">
+    <link href="/static/css/font-awesome.min.css" rel="stylesheet">
+    <link href="/static/css/login.css" rel="stylesheet">
+  </head>
+  <body>
+    <div class="container">
+      
+      {{ template "content" . }}
+      
+    </div> <!-- /container -->
+    
+    <script src="/static/js/jquery-2.1.1.min.js"></script>
+    <script src="/static/js/bootstrap.min.js"></script>
+  </body>
+</html>
+
+{{ end }}

views/auth.navbar.html

@@ -0,0 +1,26 @@
+{{ define "navbar" }}
+<div class="navbar navbar-default navbar-static-top" role="navigation">
+  <div class="container">
+    <div class="navbar-header">
+      <button type="button" class="navbar-toggle collapsed" data-toggle="collapse" data-target=".navbar-collapse">
+        <span class="sr-only">Toggle navigation</span>
+        <span class="icon-bar"></span>
+        <span class="icon-bar"></span>
+        <span class="icon-bar"></span>
+      </button>
+      <a class="navbar-brand" href="/">
+        <i class="fa fa-comments-o"></i>
+        ChitChat
+      </a>
+    </div>
+    <div class="navbar-collapse collapse">
+      <ul class="nav navbar-nav">
+        <li><a href="/">Home</a></li>
+      </ul>
+      <ul class="nav navbar-nav navbar-right">
+        <li><a href="/logout">Logout</a></li>
+      </ul>
+    </div>
+  </div>
+</div>
+{{ end }}