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("/{slug}", updateBlogPostBySlug) 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 updateBlogPostBySlug(w http.ResponseWriter, r *http.Request) { 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 }