first commit

Author: nonfu

config.json

@@ -0,0 +1,8 @@
+module github.com/xueyuanjun/chitchat
+
+go 1.13
+
+require (
+	github.com/go-sql-driver/mysql v1.5.0
+	github.com/gorilla/mux v1.7.4
+)

go.mod

@@ -0,0 +1,4 @@
+github.com/go-sql-driver/mysql v1.5.0 h1:ozyZYNQW3x3HtqT1jira07DN2PArx2v7/mN66gGcHOs=
+github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
+github.com/gorilla/mux v1.7.4 h1:VuZ8uybHlWmqV03+zRzdwKL4tUnIp1MAQtp1mIFE1bc=
+github.com/gorilla/mux v1.7.4/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=

go.sum

@@ -0,0 +1,16 @@
+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...))
+    threads, err := models.Threads();
+    if err == nil {
+        templates.ExecuteTemplate(w, "layout", threads)
+    }
+}

handlers/index.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+    . "github.com/xueyuanjun/chitchat/routes"
+    "log"
+    "net/http"
+)
+
+func main()  {
+    startWebServer("8080")
+}
+
+// 通过指定端口启动 Web 服务器
+func startWebServer(port string)  {
+    r := NewRouter() // 通过 router.go 中定义的路由器来分发请求
+
+    // 处理静态资源文件
+    assets := http.FileServer(http.Dir("public"))
+    r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", assets))
+
+    http.Handle("/", r)
+
+    log.Println("Starting HTTP service at " + port)
+    err := http.ListenAndServe(":" + port, nil) // 启动协程监听请求
+
+    if err != nil {
+        log.Println("An error occured starting HTTP listener at port " + port)
+        log.Println("Error: " + err.Error())
+    }
+}

main.go

@@ -0,0 +1,45 @@
+package models
+
+import (
+    "crypto/rand"
+    "crypto/sha1"
+    "database/sql"
+    "fmt"
+    _ "github.com/go-sql-driver/mysql"
+    "log"
+)
+
+var Db *sql.DB
+
+func init() {
+    var err error
+    Db, err = sql.Open("mysql", "root:root@/chitchat?charset=utf8&parseTime=true")
+    if err != nil {
+        log.Fatal(err)
+    }
+    return
+}
+
+// create a random UUID with from RFC 4122
+// adapted from http://github.com/nu7hatch/gouuid
+func createUUID() (uuid string) {
+    u := new([16]byte)
+    _, err := rand.Read(u[:])
+    if err != nil {
+        log.Fatalln("Cannot generate UUID", err)
+    }
+
+    // 0x40 is reserved variant from RFC 4122
+    u[8] = (u[8] | 0x40) & 0x7F
+    // Set the four most significant bits (bits 12 through 15) of the
+    // time_hi_and_version field to the 4-bit version number.
+    u[6] = (u[6] & 0xF) | (0x4 << 4)
+    uuid = fmt.Sprintf("%x-%x-%x-%x-%x", u[0:4], u[4:6], u[6:8], u[8:10], u[10:])
+    return
+}
+
+// hash plaintext with SHA-1
+func Encrypt(plaintext string) (cryptext string) {
+    cryptext = fmt.Sprintf("%x", sha1.Sum([]byte(plaintext)))
+    return
+}

models/db.go

@@ -0,0 +1,25 @@
+package models
+
+import "time"
+
+type Post struct {
+    Id        int
+    Uuid      string
+    Body      string
+    UserId    int
+    ThreadId  int
+    CreatedAt time.Time
+}
+
+func (post *Post) CreatedAtDate() string {
+    return post.CreatedAt.Format("Jan 2, 2006 at 3:04pm")
+}
+
+// Get the user who wrote the post
+func (post *Post) User() (user User) {
+    user = User{}
+    Db.QueryRow("SELECT id, uuid, name, email, created_at FROM users WHERE id = ?", post.UserId).
+        Scan(&user.Id, &user.Uuid, &user.Name, &user.Email, &user.CreatedAt)
+    return
+}
+

models/post.go

@@ -0,0 +1,53 @@
+package models
+
+import "time"
+
+type Session struct {
+    Id        int
+    Uuid      string
+    Email     string
+    UserId    int
+    CreatedAt time.Time
+}
+
+// Check if session is valid in the database
+func (session *Session) Check() (valid bool, err error) {
+    err = Db.QueryRow("SELECT id, uuid, email, user_id, created_at FROM sessions WHERE uuid = ?", session.Uuid).
+        Scan(&session.Id, &session.Uuid, &session.Email, &session.UserId, &session.CreatedAt)
+    if err != nil {
+        valid = false
+        return
+    }
+    if session.Id != 0 {
+        valid = true
+    }
+    return
+}
+
+// Delete session from database
+func (session *Session) DeleteByUUID() (err error) {
+    statement := "delete from sessions where uuid = ?"
+    stmt, err := Db.Prepare(statement)
+    if err != nil {
+        return
+    }
+    defer stmt.Close()
+
+    _, err = stmt.Exec(session.Uuid)
+    return
+}
+
+// Get the user from the session
+func (session *Session) User() (user User, err error) {
+    user = User{}
+    err = Db.QueryRow("SELECT id, uuid, name, email, created_at FROM users WHERE id = ?", session.UserId).
+        Scan(&user.Id, &user.Uuid, &user.Name, &user.Email, &user.CreatedAt)
+    return
+}
+
+// Delete all sessions from database
+func SessionDeleteAll() (err error) {
+    statement := "delete from sessions"
+    _, err = Db.Exec(statement)
+    return
+}

models/session.go

@@ -0,0 +1,81 @@
+package models
+
+import "time"
+
+type Thread struct {
+    Id        int
+    Uuid      string
+    Topic     string
+    UserId    int
+    CreatedAt time.Time
+}
+
+// format the CreatedAt date to display nicely on the screen
+func (thread *Thread) CreatedAtDate() string {
+    return thread.CreatedAt.Format("Jan 2, 2006 at 3:04pm")
+}
+
+// get the number of posts in a thread
+func (thread *Thread) NumReplies() (count int) {
+    rows, err := Db.Query("SELECT count(*) FROM posts where thread_id = ?", thread.Id)
+    if err != nil {
+        return
+    }
+    for rows.Next() {
+        if err = rows.Scan(&count); err != nil {
+            return
+        }
+    }
+    rows.Close()
+    return
+}
+
+// get posts to a thread
+func (thread *Thread) Posts() (posts []Post, err error) {
+    rows, err := Db.Query("SELECT id, uuid, body, user_id, thread_id, created_at FROM posts where thread_id = ?", thread.Id)
+    if err != nil {
+        return
+    }
+    for rows.Next() {
+        post := Post{}
+        if err = rows.Scan(&post.Id, &post.Uuid, &post.Body, &post.UserId, &post.ThreadId, &post.CreatedAt); err != nil {
+            return
+        }
+        posts = append(posts, post)
+    }
+    rows.Close()
+    return
+}
+
+// Get all threads in the database and returns it
+func Threads() (threads []Thread, err error) {
+    rows, err := Db.Query("SELECT id, uuid, topic, user_id, created_at FROM threads ORDER BY created_at DESC")
+    if err != nil {
+        return
+    }
+    for rows.Next() {
+        conv := Thread{}
+        if err = rows.Scan(&conv.Id, &conv.Uuid, &conv.Topic, &conv.UserId, &conv.CreatedAt); err != nil {
+            return
+        }
+        threads = append(threads, conv)
+    }
+    rows.Close()
+    return
+}
+
+// Get a thread by the UUID
+func ThreadByUUID(uuid string) (conv Thread, err error) {
+    conv = Thread{}
+    err = Db.QueryRow("SELECT id, uuid, topic, user_id, created_at FROM threads WHERE uuid = ?", uuid).
+        Scan(&conv.Id, &conv.Uuid, &conv.Topic, &conv.UserId, &conv.CreatedAt)
+    return
+}
+
+// Get the user who started this thread
+func (thread *Thread) User() (user User) {
+    user = User{}
+    Db.QueryRow("SELECT id, uuid, name, email, created_at FROM users WHERE id = ?", thread.UserId).
+        Scan(&user.Id, &user.Uuid, &user.Name, &user.Email, &user.CreatedAt)
+    return
+}