This repository has been archived on 2023-07-09. You can view files and clone it, but cannot push or open issues or pull requests.
sudoscientist-go-backend/packages/blog/blog.go
2019-04-20 17:45:01 -04:00

185 lines
4.4 KiB
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
}