connect get threads controller

This commit is contained in:
Yanislav Igonin 2022-04-06 11:45:39 +03:00
parent 67fbcf8c17
commit b0727315fe
2 changed files with 323 additions and 336 deletions

View File

@ -1,403 +1,392 @@
package controllers
import (
"context"
"log"
"math"
"net/http"
"path/filepath"
"strconv"
"strings"
"github.com/dchest/captcha"
"github.com/gin-gonic/gin"
"github.com/gofiber/fiber/v2"
Config "micrach/config"
Db "micrach/db"
Repositories "micrach/repositories"
Utils "micrach/utils"
"micrach/config"
"micrach/repositories"
)
func GetThreads(c *gin.Context) {
pageString := c.DefaultQuery("page", "1")
func GetThreads(c *fiber.Ctx) error {
pageString := c.Query("page", "1")
page, err := strconv.Atoi(pageString)
if err != nil {
c.HTML(http.StatusNotFound, "404.html", nil)
return
return c.Status(fiber.StatusNotFound).Render("pages/404", nil)
}
if page <= 0 {
c.HTML(http.StatusNotFound, "404.html", nil)
return
return c.Status(fiber.StatusNotFound).Render("pages/404", nil)
}
limit := 10
offset := limit * (page - 1)
threads, err := Repositories.Posts.Get(limit, offset)
threads, err := repositories.Posts.Get(limit, offset)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
return c.Status(fiber.StatusInternalServerError).Render("pages/500", nil)
}
count, err := Repositories.Posts.GetThreadsCount()
count, err := repositories.Posts.GetThreadsCount()
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
return c.Status(fiber.StatusInternalServerError).Render("pages/500", nil)
}
pagesCount := int(math.Ceil(float64(count) / 10))
if page > pagesCount && count != 0 {
c.HTML(http.StatusNotFound, "404.html", nil)
return
return c.Status(fiber.StatusNotFound).Render("pages/404", nil)
}
captchaID := captcha.New()
htmlData := Repositories.GetThreadsHtmlData{
htmlData := repositories.GetThreadsHtmlData{
Threads: threads,
Pagination: Repositories.HtmlPaginationData{
Pagination: repositories.HtmlPaginationData{
PagesCount: pagesCount,
Page: page,
},
FormData: Repositories.HtmlFormData{
FormData: repositories.HtmlFormData{
CaptchaID: captchaID,
IsCaptchaActive: Config.App.IsCaptchaActive,
IsCaptchaActive: config.App.IsCaptchaActive,
},
}
c.HTML(http.StatusOK, "index.html", htmlData)
return c.Status(fiber.StatusOK).Render("pages/index", htmlData)
}
func GetThread(c *gin.Context) {
threadIDString := c.Param("threadID")
threadID, err := strconv.Atoi(threadIDString)
if err != nil {
c.HTML(http.StatusNotFound, "404.html", nil)
return
}
thread, err := Repositories.Posts.GetThreadByPostID(threadID)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
if thread == nil {
c.HTML(http.StatusNotFound, "404.html", nil)
return
}
// func GetThread(c *gin.Context) {
// threadIDString := c.Param("threadID")
// threadID, err := strconv.Atoi(threadIDString)
// if err != nil {
// c.HTML(http.StatusNotFound, "404.html", nil)
// return
// }
// thread, err := Repositories.Posts.GetThreadByPostID(threadID)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// if thread == nil {
// c.HTML(http.StatusNotFound, "404.html", nil)
// return
// }
firstPost := thread[0]
captchaID := captcha.New()
htmlData := Repositories.GetThreadHtmlData{
Thread: thread,
FormData: Repositories.HtmlFormData{
FirstPostID: firstPost.ID,
CaptchaID: captchaID,
IsCaptchaActive: Config.App.IsCaptchaActive,
},
}
c.HTML(http.StatusOK, "thread.html", htmlData)
}
// firstPost := thread[0]
// captchaID := captcha.New()
// htmlData := Repositories.GetThreadHtmlData{
// Thread: thread,
// FormData: Repositories.HtmlFormData{
// FirstPostID: firstPost.ID,
// CaptchaID: captchaID,
// IsCaptchaActive: Config.App.IsCaptchaActive,
// },
// }
// c.HTML(http.StatusOK, "thread.html", htmlData)
// }
func CreateThread(c *gin.Context) {
form, err := c.MultipartForm()
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// func CreateThread(c *gin.Context) {
// form, err := c.MultipartForm()
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// TODO: dat shit crashes if no fields in request
text := form.Value["text"][0]
title := form.Value["title"][0]
filesInRequest := form.File["files"]
validationErrorMessage := Utils.ValidatePost(title, text, filesInRequest)
if validationErrorMessage != "" {
errorHtmlData := Repositories.BadRequestHtmlData{
Message: validationErrorMessage,
}
c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
return
}
// // TODO: dat shit crashes if no fields in request
// text := form.Value["text"][0]
// title := form.Value["title"][0]
// filesInRequest := form.File["files"]
// validationErrorMessage := Utils.ValidatePost(title, text, filesInRequest)
// if validationErrorMessage != "" {
// errorHtmlData := Repositories.BadRequestHtmlData{
// Message: validationErrorMessage,
// }
// c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
// return
// }
if Config.App.IsCaptchaActive {
captchaID := form.Value["captchaId"][0]
captchaString := form.Value["captcha"][0]
isCaptchaValid := captcha.VerifyString(captchaID, captchaString)
if !isCaptchaValid {
errorHtmlData := Repositories.BadRequestHtmlData{
Message: Repositories.InvalidCaptchaErrorMessage,
}
c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
return
}
}
// if Config.App.IsCaptchaActive {
// captchaID := form.Value["captchaId"][0]
// captchaString := form.Value["captcha"][0]
// isCaptchaValid := captcha.VerifyString(captchaID, captchaString)
// if !isCaptchaValid {
// errorHtmlData := Repositories.BadRequestHtmlData{
// Message: Repositories.InvalidCaptchaErrorMessage,
// }
// c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
// return
// }
// }
conn, err := Db.Pool.Acquire(context.TODO())
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
defer conn.Release()
// conn, err := Db.Pool.Acquire(context.TODO())
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// defer conn.Release()
threadsCount, err := Repositories.Posts.GetThreadsCount()
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// threadsCount, err := Repositories.Posts.GetThreadsCount()
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
if threadsCount >= Config.App.ThreadsMaxCount {
oldestThreadUpdatedAt, err := Repositories.Posts.GetOldestThreadUpdatedAt()
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
err = Repositories.Posts.ArchiveThreadsFrom(oldestThreadUpdatedAt)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
}
// if threadsCount >= Config.App.ThreadsMaxCount {
// oldestThreadUpdatedAt, err := Repositories.Posts.GetOldestThreadUpdatedAt()
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// err = Repositories.Posts.ArchiveThreadsFrom(oldestThreadUpdatedAt)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// }
tx, err := conn.Begin(context.TODO())
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
defer tx.Rollback(context.TODO())
// tx, err := conn.Begin(context.TODO())
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// defer tx.Rollback(context.TODO())
post := Repositories.Post{
IsParent: true,
Title: title,
Text: text,
IsSage: false,
}
threadID, err := Repositories.Posts.CreateInTx(tx, post)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// post := Repositories.Post{
// IsParent: true,
// Title: title,
// Text: text,
// IsSage: false,
// }
// threadID, err := Repositories.Posts.CreateInTx(tx, post)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
err = Utils.CreateThreadFolder(threadID)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// err = Utils.CreateThreadFolder(threadID)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
for _, fileInRequest := range filesInRequest {
file := Repositories.File{
PostID: threadID,
Name: fileInRequest.Filename,
// image/jpeg -> jpeg
Ext: strings.Split(fileInRequest.Header["Content-Type"][0], "/")[1],
Size: int(fileInRequest.Size),
}
// for _, fileInRequest := range filesInRequest {
// file := Repositories.File{
// PostID: threadID,
// Name: fileInRequest.Filename,
// // image/jpeg -> jpeg
// Ext: strings.Split(fileInRequest.Header["Content-Type"][0], "/")[1],
// Size: int(fileInRequest.Size),
// }
fileID, err := Repositories.Files.CreateInTx(tx, file)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// fileID, err := Repositories.Files.CreateInTx(tx, file)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
path := filepath.Join(
Utils.UPLOADS_DIR_PATH,
strconv.Itoa(threadID),
"o",
strconv.Itoa(fileID)+"."+file.Ext,
)
err = c.SaveUploadedFile(fileInRequest, path)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// creating thumbnail
thumbImg, err := Utils.MakeImageThumbnail(path, file.Ext, threadID, fileID)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// saving thumbnail
err = Utils.SaveImageThumbnail(thumbImg, threadID, fileID, file.Ext)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
}
// path := filepath.Join(
// Utils.UPLOADS_DIR_PATH,
// strconv.Itoa(threadID),
// "o",
// strconv.Itoa(fileID)+"."+file.Ext,
// )
// err = c.SaveUploadedFile(fileInRequest, path)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// // creating thumbnail
// thumbImg, err := Utils.MakeImageThumbnail(path, file.Ext, threadID, fileID)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// // saving thumbnail
// err = Utils.SaveImageThumbnail(thumbImg, threadID, fileID, file.Ext)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// }
tx.Commit(context.TODO())
// tx.Commit(context.TODO())
c.Redirect(http.StatusFound, "/"+strconv.Itoa(threadID))
}
// c.Redirect(http.StatusFound, "/"+strconv.Itoa(threadID))
// }
// Add new post in thread
func UpdateThread(c *gin.Context) {
threadIDString := c.Param("threadID")
threadID, err := strconv.Atoi(threadIDString)
if err != nil {
c.HTML(http.StatusNotFound, "500.html", nil)
return
}
// // Add new post in thread
// func UpdateThread(c *gin.Context) {
// threadIDString := c.Param("threadID")
// threadID, err := strconv.Atoi(threadIDString)
// if err != nil {
// c.HTML(http.StatusNotFound, "500.html", nil)
// return
// }
isArchived, err := Repositories.Posts.GetIfThreadIsArchived(threadID)
if isArchived {
errorHtmlData := Repositories.BadRequestHtmlData{
Message: Repositories.ThreadIsArchivedErrorMessage,
}
c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
return
}
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// isArchived, err := Repositories.Posts.GetIfThreadIsArchived(threadID)
// if isArchived {
// errorHtmlData := Repositories.BadRequestHtmlData{
// Message: Repositories.ThreadIsArchivedErrorMessage,
// }
// c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
// return
// }
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
form, err := c.MultipartForm()
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// form, err := c.MultipartForm()
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// TODO: dat shit crashes if no fields in request
text := form.Value["text"][0]
filesInRequest := form.File["files"]
validationErrorMessage := Utils.ValidatePost("", text, filesInRequest)
if validationErrorMessage != "" {
errorHtmlData := Repositories.BadRequestHtmlData{
Message: validationErrorMessage,
}
c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
return
}
// // TODO: dat shit crashes if no fields in request
// text := form.Value["text"][0]
// filesInRequest := form.File["files"]
// validationErrorMessage := Utils.ValidatePost("", text, filesInRequest)
// if validationErrorMessage != "" {
// errorHtmlData := Repositories.BadRequestHtmlData{
// Message: validationErrorMessage,
// }
// c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
// return
// }
if Config.App.IsCaptchaActive {
captchaID := form.Value["captchaId"][0]
captchaString := form.Value["captcha"][0]
isCaptchaValid := captcha.VerifyString(captchaID, captchaString)
if !isCaptchaValid {
errorHtmlData := Repositories.BadRequestHtmlData{
Message: Repositories.InvalidCaptchaErrorMessage,
}
c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
return
}
}
// if Config.App.IsCaptchaActive {
// captchaID := form.Value["captchaId"][0]
// captchaString := form.Value["captcha"][0]
// isCaptchaValid := captcha.VerifyString(captchaID, captchaString)
// if !isCaptchaValid {
// errorHtmlData := Repositories.BadRequestHtmlData{
// Message: Repositories.InvalidCaptchaErrorMessage,
// }
// c.HTML(http.StatusBadRequest, "400.html", errorHtmlData)
// return
// }
// }
isSageField := form.Value["sage"]
var isSageString string
if len(isSageField) != 0 {
isSageString = isSageField[0]
}
isSage := isSageString == "on"
// isSageField := form.Value["sage"]
// var isSageString string
// if len(isSageField) != 0 {
// isSageString = isSageField[0]
// }
// isSage := isSageString == "on"
conn, err := Db.Pool.Acquire(context.TODO())
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
defer conn.Release()
// conn, err := Db.Pool.Acquire(context.TODO())
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// defer conn.Release()
tx, err := conn.Begin(context.TODO())
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
defer tx.Rollback(context.TODO())
// tx, err := conn.Begin(context.TODO())
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// defer tx.Rollback(context.TODO())
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
post := Repositories.Post{
IsParent: false,
ParentID: &threadID,
Title: "",
Text: text,
IsSage: isSage,
}
postID, err := Repositories.Posts.CreateInTx(tx, post)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// post := Repositories.Post{
// IsParent: false,
// ParentID: &threadID,
// Title: "",
// Text: text,
// IsSage: isSage,
// }
// postID, err := Repositories.Posts.CreateInTx(tx, post)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
postsCountInThread, err := Repositories.Posts.GetThreadPostsCount(threadID)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
isBumpLimit := postsCountInThread >= Config.App.ThreadBumpLimit
isThreadBumped := !isBumpLimit && !isSage && !post.IsParent
if isThreadBumped {
err = Repositories.Posts.BumpThreadInTx(tx, threadID)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
}
// postsCountInThread, err := Repositories.Posts.GetThreadPostsCount(threadID)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// isBumpLimit := postsCountInThread >= Config.App.ThreadBumpLimit
// isThreadBumped := !isBumpLimit && !isSage && !post.IsParent
// if isThreadBumped {
// err = Repositories.Posts.BumpThreadInTx(tx, threadID)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// }
for _, fileInRequest := range filesInRequest {
file := Repositories.File{
PostID: postID,
Name: fileInRequest.Filename,
// image/jpeg -> jpeg
Ext: strings.Split(fileInRequest.Header["Content-Type"][0], "/")[1],
Size: int(fileInRequest.Size),
}
// for _, fileInRequest := range filesInRequest {
// file := Repositories.File{
// PostID: postID,
// Name: fileInRequest.Filename,
// // image/jpeg -> jpeg
// Ext: strings.Split(fileInRequest.Header["Content-Type"][0], "/")[1],
// Size: int(fileInRequest.Size),
// }
fileID, err := Repositories.Files.CreateInTx(tx, file)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// fileID, err := Repositories.Files.CreateInTx(tx, file)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
path := filepath.Join(
Utils.UPLOADS_DIR_PATH,
strconv.Itoa(threadID),
"o",
strconv.Itoa(fileID)+"."+file.Ext,
)
err = c.SaveUploadedFile(fileInRequest, path)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// creating thumbnail
thumbImg, err := Utils.MakeImageThumbnail(path, file.Ext, threadID, fileID)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
// saving thumbnail
err = Utils.SaveImageThumbnail(thumbImg, threadID, fileID, file.Ext)
if err != nil {
log.Println("error:", err)
c.HTML(http.StatusInternalServerError, "500.html", nil)
return
}
}
// path := filepath.Join(
// Utils.UPLOADS_DIR_PATH,
// strconv.Itoa(threadID),
// "o",
// strconv.Itoa(fileID)+"."+file.Ext,
// )
// err = c.SaveUploadedFile(fileInRequest, path)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// // creating thumbnail
// thumbImg, err := Utils.MakeImageThumbnail(path, file.Ext, threadID, fileID)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// // saving thumbnail
// err = Utils.SaveImageThumbnail(thumbImg, threadID, fileID, file.Ext)
// if err != nil {
// log.Println("error:", err)
// c.HTML(http.StatusInternalServerError, "500.html", nil)
// return
// }
// }
tx.Commit(context.TODO())
// tx.Commit(context.TODO())
c.Header("Refresh", "0")
}
// c.Header("Refresh", "0")
// }

View File

@ -12,6 +12,7 @@ import (
"micrach/build"
"micrach/config"
"micrach/controllers"
"micrach/db"
"micrach/repositories"
"micrach/templates"
@ -132,10 +133,7 @@ func main() {
app.Static("/uploads", "./uploads")
app.Static("/static", "./static")
app.Get("/", func(c *fiber.Ctx) error {
// return c.Render("components/index", fiber.Map{})
return c.SendString("get threads")
})
app.Get("/", controllers.GetThreads)
app.Post("/", func(c *fiber.Ctx) error {
return c.SendString("create thread")
})