153 lines
3.4 KiB
Go
153 lines
3.4 KiB
Go
package submit
|
|
|
|
import (
|
|
"context"
|
|
core "jokes-bapak2-api/app/core/submit"
|
|
"jokes-bapak2-api/app/core/validator"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/Masterminds/squirrel"
|
|
"github.com/georgysavva/scany/pgxscan"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/jackc/pgx/v4"
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
)
|
|
|
|
func (d *Dependencies) SubmitJoke(c *fiber.Ctx) error {
|
|
conn, err := d.DB.Acquire(c.Context())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Release()
|
|
|
|
var body Submission
|
|
err = c.BodyParser(&body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Image and/or Link should not be empty
|
|
if body.Image == "" && body.Link == "" {
|
|
return c.Status(fiber.StatusBadRequest).JSON(Error{
|
|
Error: "A link or an image should be supplied in a form of multipart/form-data",
|
|
})
|
|
}
|
|
|
|
// Author should be supplied
|
|
if body.Author == "" {
|
|
return c.Status(fiber.StatusBadRequest).JSON(Error{
|
|
Error: "An author key consisting on the format \"yourname <youremail@mail>\" must be supplied",
|
|
})
|
|
} else {
|
|
// Validate format
|
|
valid := validator.ValidateAuthor(body.Author)
|
|
if !valid {
|
|
return c.Status(fiber.StatusBadRequest).JSON(Error{
|
|
Error: "Please stick to the format of \"yourname <youremail@mail>\" and within 200 characters",
|
|
})
|
|
}
|
|
}
|
|
|
|
var link string
|
|
|
|
// Check link validity if link was provided
|
|
if body.Link != "" {
|
|
valid, err := validator.CheckImageValidity(d.HTTP, body.Link)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !valid {
|
|
return c.Status(fiber.StatusBadRequest).JSON(Error{
|
|
Error: "URL provided is not a valid image",
|
|
})
|
|
}
|
|
|
|
link = body.Link
|
|
}
|
|
|
|
// If image was provided
|
|
if body.Image != "" {
|
|
image := strings.NewReader(body.Image)
|
|
|
|
link, err = core.UploadImage(d.HTTP, image)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Validate if link already exists
|
|
validateLink, err := validateIfLinkExists(d.DB, c.Context(), d.Query, link)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if validateLink {
|
|
return c.Status(fiber.StatusConflict).JSON(Error{
|
|
Error: "Given link is already on the submission queue.",
|
|
})
|
|
}
|
|
|
|
now := time.Now().UTC().Format(time.RFC3339)
|
|
|
|
sql, args, err := d.Query.
|
|
Insert("submission").
|
|
Columns("link", "created_at", "author").
|
|
Values(link, now, body.Author).
|
|
Suffix("RETURNING id,created_at,link,author,status").
|
|
ToSql()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var submission []Submission
|
|
result, err := conn.Query(c.Context(), sql, args...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer result.Close()
|
|
|
|
err = pgxscan.ScanAll(&submission, result)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.
|
|
Status(fiber.StatusCreated).
|
|
JSON(ResponseSubmission{
|
|
Message: "Joke submitted. Please wait for a few days for admin to approve your submission.",
|
|
Submission: submission[0],
|
|
AuthorPage: "/submit?author=" + url.QueryEscape(body.Author),
|
|
})
|
|
}
|
|
|
|
func validateIfLinkExists(db *pgxpool.Pool, ctx context.Context, query squirrel.StatementBuilderType, link string) (bool, error) {
|
|
conn, err := db.Acquire(ctx)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer conn.Release()
|
|
|
|
sql, args, err := query.
|
|
Select("link").
|
|
From("submission").
|
|
Where(squirrel.Eq{"link": link}).
|
|
ToSql()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
var validateLink string
|
|
err = conn.QueryRow(context.Background(), sql, args...).Scan(&validateLink)
|
|
if err != nil && err != pgx.ErrNoRows {
|
|
return false, err
|
|
}
|
|
|
|
if err == nil && validateLink != "" {
|
|
return true, nil
|
|
}
|
|
|
|
return false, nil
|
|
}
|