sudoscientist-go-backend/packages/blog/blog.go

package blog

import (
	"database/sql"
	"encoding/json"
	"fmt"
	"github.com/go-chi/chi"
	"github.com/go-chi/jwtauth"
	"github.com/go-chi/render"
	"github.com/gosimple/slug"
	"net/http"
	"strings"
	"time"
)

var (
	DB        *sql.DB
	TokenAuth *jwtauth.JWTAuth
)

type BlogPost struct {
	Title         string    `json:"title",db:"title"`
	Slug          string    `json:"slug",db:"slug"`
	Author        string    `json:"author",db:"author"`
	Content       string    `json:"content",db:"content"`
	DatePublished time.Time `json:"date", db:"date"`
}

type Tag struct {
	TagList string `json:"tags"`
}

type NewBlogPost struct {
	Title   string `json:"title",db:"title"`
	Content string `json:"content",db:"content"`
	Tags    string `json:"tags"`
	Author  string `json:"author",db:"author"`
}

type ReturnError struct {
	Message string `json:"error"`
}

type ReturnSuccess struct {
	Message string `json:"success"`
	ID      int    `json:"id"`
}

func Init() {
	createPostsTable := `
	CREATE TABLE IF NOT EXISTS posts
	(id SERIAL PRIMARY KEY,
    title text,
    slug text,
	author text REFERENCES users (username),
    content text,
	date timestamp)`
	DB.Exec(createPostsTable)

	createTagsTable := `
	CREATE TABLE IF NOT EXISTS tags
	(id SERIAL PRIMARY KEY,
    tag text,
	article_id int REFERENCES posts (id))`
	DB.Exec(createTagsTable)
}

func Routes() *chi.Mux {
	r := chi.NewRouter()
	r.Group(func(r chi.Router) {
		r.Use(jwtauth.Verifier(TokenAuth))
		r.Use(jwtauth.Authenticator)
		r.Post("/", createBlogPost)
		r.Patch("/id/{id}", updateBlogPostById)
	})
	r.Get("/", getBlogPosts)
	r.Get("/{slug}", getBlogPostBySlug)
	r.Get("/by-tag/{tag}", getBlogPostByTag)
	r.Get("/by-id/{id}", getBlogPostById)
	r.Get("/by-author/{author}", getBlogPostByAuthor)
	return r
}

func createBlogPost(w http.ResponseWriter, r *http.Request) {
	returnError := ReturnError{}
	newBlogPost := &NewBlogPost{}
	_, claims, _ := jwtauth.FromContext(r.Context())
	username := claims["username"].(string)
	err := json.NewDecoder(r.Body).Decode(newBlogPost)
	if err != nil {
		w.WriteHeader(http.StatusBadRequest)
		return
	}
	if newBlogPost.Title == "" {
		returnError.Message = "title is required"
		w.WriteHeader(http.StatusBadRequest)
		render.JSON(w, r, returnError)
		return
	}
	if newBlogPost.Content == "" {
		returnError.Message = "content is required"
		w.WriteHeader(http.StatusBadRequest)
		render.JSON(w, r, returnError)
		return
	}
	as := slug.Make(newBlogPost.Title)
	slugCheck := DB.QueryRow("SELECT id FROM posts WHERE slug=$1", as)
	// wow this is ugly. someone pls send help.
	scr := 0
	err = slugCheck.Scan(&scr)
	if err == nil {
		returnError.Message = "slug already exists"
		w.WriteHeader(http.StatusBadRequest)
		render.JSON(w, r, returnError)
		return
	}
	if err != nil {
		if err != sql.ErrNoRows {
			returnError.Message = "something is super broken..."
			w.WriteHeader(http.StatusInternalServerError)
			return
		}
	}
	s := `INSERT INTO posts (title, slug, author, content, date)
            VALUES ($1, $2, $3, $4, $5)
            RETURNING id`
	article_id := 0
	err = DB.QueryRow(s, newBlogPost.Title, as, username, newBlogPost.Content, time.Now().UTC()).Scan(&article_id)
	if err != nil {
		if err == sql.ErrNoRows {
			returnError.Message = "something is super broken..."
			w.WriteHeader(http.StatusInternalServerError)
			render.JSON(w, r, returnError)
			return
		}
		fmt.Println(err)
		returnError.Message = "something is super broken..."
		w.WriteHeader(http.StatusInternalServerError)
		render.JSON(w, r, returnError)
		return
	}
	fmt.Println(newBlogPost.Tags)
	if newBlogPost.Tags != "" {
		t := `INSERT INTO tags (tag, article_id)
                VALUES ($1, $2)`
		tags := strings.Split(newBlogPost.Tags, ",")
		for i := range tags {
			DB.Exec(t, tags[i], article_id)
		}

	}
	returnSuccess := ReturnSuccess{Message: "post created", ID: article_id}
	w.WriteHeader(http.StatusCreated)
	render.JSON(w, r, returnSuccess)
	return
}

func updateBlogPostById(w http.ResponseWriter, r *http.Request) {
	return
}

func getBlogPosts(w http.ResponseWriter, r *http.Request) {
	return
}

func getBlogPostById(w http.ResponseWriter, r *http.Request) {
	return
}

func getBlogPostBySlug(w http.ResponseWriter, r *http.Request) {
	return
}

func getBlogPostByTag(w http.ResponseWriter, r *http.Request) {
	return
}

func getBlogPostByAuthor(w http.ResponseWriter, r *http.Request) {
	return
}

func getRssFeed(w http.ResponseWriter, r *http.Request) {
	return
}