route_thread.go

package main

import (
	"github.com/anuragdhingra/lets-chat/data"
	"github.com/julienschmidt/httprouter"
	"log"
	"net/http"
	"strconv"
)

type ThreadInfoPublic struct {
	Thread data.Thread
	CreatedBy data.User
	Posts []PostInfoPublic
}

type ThreadInfoPrivate struct {
	Thread data.Thread
	CreatedBy data.User
	User data.User
	Posts []PostInfoPublic
}

func NewThread(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	sess, err := session(w, r)
	if err == nil {
		loggedInUser, err := sess.User()
		data := ThreadsInfoPrivate{nil, loggedInUser}
		if err != nil {
			log.Print(err)
			return
		} else {
			generateHTML(w, data, "layout", "private.navbar", "new.thread")
		}
	} else {
		log.Print(err)
		http.Redirect(w, r, "/login", http.StatusFound)
		return
	}
}

func CreateThread(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	err := r.ParseForm()
	throwError(err)
	sess, err := session(w, r)
	throwError(err)
	user, err := sess.User()
	throwError(err)

	createThreadRequest := data.CreateThreadRequest{
		r.PostFormValue("topic"),
		user.Id,

	}
	threadId, err := createThreadRequest.Create()
	log.Print(threadId)
	throwError(err)
	url := "/threads/" +  strconv.Itoa(threadId)
	log.Print(url)
	http.Redirect(w, r, url, http.StatusFound)
}

func FindThread(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
	threadId := p.ByName("id")
	thread, err := data.ThreadByID(threadId)
	if err != nil {
		log.Print(err)
		return
	} else {
		user, err := data.UserById(thread.UserId)
		throwError(err)
		posts, err := data.PostsByThreadId(thread.Id)
		throwError(err)
		postList := CreatePostList(posts)
		sess, err := session(w, r)
		if err != nil {
			data := ThreadInfoPublic{thread, user,  postList}
			generateHTML(w, data, "layout","public.navbar", "public.thread")
		} else {
			loggedInUser, err := sess.User()
			throwError(err)
			data := ThreadInfoPrivate{thread, user, loggedInUser, postList}
			generateHTML(w, data, "layout", "private.navbar","private.thread")
		}
	}
}

func CreateThreadList(threads []data.Thread) (threadListPublic []ThreadInfoPublic) {
	for _, thread := range threads {
		threadUserId := thread.UserId
		user, err := data.UserById(threadUserId)
		throwError(err)
		threadInfoPublic := ThreadInfoPublic{thread,user, nil}
		threadListPublic = append(threadListPublic, threadInfoPublic)
	}
	return
}