package blog import ( "database/sql" "encoding/json" "fmt" "git.minhas.io/asara/sudoscientist-go-backend/packages/middleware" "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 { ID int `json:"id",db:"id"` Title string `json:"title",db:"title"` Slug string `json:"slug",db:"slug"` Author string `json:"author",db:"author"` Content string `json:"content",db:"content"` TimePublished time.Time `json:"time_published", db:"time_published"` Modified bool `json:"modified", db:"modified"` TimeModified time.Time `json:"last_modified", db:"last_modified"` } type BlogPosts []BlogPost 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"` } type ReferenceID struct { LastID int `json:"last_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, time_published timestamp, modified bool, last_modified 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.Verify(TokenAuth, auth_middleware.TokenFromSplitCookie)) r.Use(jwtauth.Authenticator) r.Post("/", createBlogPost) r.Patch("/by-id/{id}", updateBlogPostById) }) r.Get("/", getBlogPosts) r.Get("/by-id/{id}", getBlogPostById) r.Get("/by-tag/{tag}", getBlogPostsByTag) r.Get("/by-author/{author}", getBlogPostsByAuthor) r.Get("/by-slug/{slug}", getBlogPostBySlug) return r } func createBlogPost(w http.ResponseWriter, r *http.Request) { returnError := ReturnError{} newBlogPost := &NewBlogPost{} // basic checks _, claims, _ := jwtauth.FromContext(r.Context()) username := claims["username"].(string) err := json.NewDecoder(r.Body).Decode(newBlogPost) if err != nil { returnError.Message = "unknown error, try again later" w.WriteHeader(http.StatusBadRequest) render.JSON(w, r, returnError) 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. // checking to ensure the same slug doesn't exist... 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) render.JSON(w, r, returnError) fmt.Println(err) return } } s := `INSERT INTO posts (title, slug, author, content, time_published, modified, last_modified) VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING id` article_id := 0 // write the row and get back the id err = DB.QueryRow(s, newBlogPost.Title, as, username, newBlogPost.Content, time.Now().UTC(), false, 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) fmt.Println(err) return } returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } // if the article has 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) { returnError := ReturnError{} // Get the actual post id := chi.URLParam(r, "id") result := DB.QueryRow("SELECT id, title, slug, author, content, time_published FROM posts WHERE id=$1", id) post := BlogPost{} err := result.Scan(&post.ID, &post.Title, &post.Slug, &post.Author, &post.Content, &post.TimePublished) if err != nil { if err == sql.ErrNoRows { returnError.Message = "blog post requested for update not found" w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } // Verify the post belongs to the requester _, claims, _ := jwtauth.FromContext(r.Context()) username := claims["username"].(string) if username != post.Author { returnError.Message = "unauthorized..." w.WriteHeader(http.StatusUnauthorized) render.JSON(w, r, returnError) return } // update the post struct err = json.NewDecoder(r.Body).Decode(&post) s := ` UPDATE posts SET title = $1, content = $2, modified = $3, last_modified = $4 WHERE id = $5` // write the row update _, err = DB.Exec(s, post.Title, post.Content, true, time.Now().UTC(), post.ID) if err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } returnSuccess := ReturnSuccess{Message: "post updated", ID: post.ID} w.WriteHeader(http.StatusOK) render.JSON(w, r, returnSuccess) return } // This will search by the last id seen, descending. func getBlogPosts(w http.ResponseWriter, r *http.Request) { returnError := ReturnError{} referenceID := &ReferenceID{} err := json.NewDecoder(r.Body).Decode(referenceID) // hardcode 9001 for cool kid points if err != nil { referenceID.LastID = 9001 } search_id := referenceID.LastID // if someone is lame and sends up a negative number... if search_id < 1 { search_id = 9001 } search := ` SELECT id, title, slug, author, content, time_published, modified, last_modified FROM posts WHERE id < $1 ORDER BY id DESC FETCH FIRST 10 ROWS ONLY ` rows, err := DB.Query(search, search_id) if err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } defer rows.Close() post := BlogPost{} posts := make(BlogPosts, 0) for rows.Next() { if err := rows.Scan(&post.ID, &post.Title, &post.Slug, &post.Author, &post.Content, &post.TimePublished, &post.Modified, &post.TimeModified); err != nil { } posts = append(posts, post) } if err := rows.Err(); err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } w.WriteHeader(http.StatusOK) render.JSON(w, r, posts) return } func getBlogPostBySlug(w http.ResponseWriter, r *http.Request) { returnError := ReturnError{} slug := chi.URLParam(r, "slug") result := DB.QueryRow("SELECT id, title, slug, author, content, time_published, modified, last_modified FROM posts WHERE slug=$1", slug) post := BlogPost{} err := result.Scan(&post.ID, &post.Title, &post.Slug, &post.Author, &post.Content, &post.TimePublished, &post.Modified, &post.TimeModified) if err != nil { returnError.Message = "post not found" w.WriteHeader(http.StatusBadRequest) render.JSON(w, r, returnError) return } w.WriteHeader(http.StatusOK) render.JSON(w, r, post) return } func getBlogPostById(w http.ResponseWriter, r *http.Request) { returnError := ReturnError{} id := chi.URLParam(r, "id") result := DB.QueryRow("SELECT id, title, slug, author, content, time_published, modified, last_modified FROM posts WHERE id=$1", id) post := BlogPost{} err := result.Scan(&post.ID, &post.Title, &post.Slug, &post.Author, &post.Content, &post.TimePublished, &post.Modified, &post.TimeModified) if err != nil { returnError.Message = "post not found" w.WriteHeader(http.StatusBadRequest) render.JSON(w, r, returnError) return } w.WriteHeader(http.StatusOK) render.JSON(w, r, post) return } func getBlogPostsByTag(w http.ResponseWriter, r *http.Request) { returnError := ReturnError{} referenceID := &ReferenceID{} err := json.NewDecoder(r.Body).Decode(referenceID) // hardcode 9001 for cool kid points if err != nil { referenceID.LastID = 9001 } search_id := referenceID.LastID // if someone is lame and sends up a negative number... if search_id < 1 { search_id = 9001 } tag := chi.URLParam(r, "tag") search := ` SELECT posts.id, posts.title, posts.slug, posts.author, posts.content, posts.time_published, posts.modified, posts.last_modified FROM posts INNER JOIN tags on posts .id = tags.article_id WHERE posts.id < $1 AND tags.tag = $2 ORDER BY id DESC FETCH FIRST 10 ROWS ONLY ` rows, err := DB.Query(search, search_id, tag) if err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } defer rows.Close() post := BlogPost{} posts := make(BlogPosts, 0) for rows.Next() { if err := rows.Scan(&post.ID, &post.Title, &post.Slug, &post.Author, &post.Content, &post.TimePublished, &post.Modified, &post.TimeModified); err != nil { } posts = append(posts, post) } if err := rows.Err(); err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } w.WriteHeader(http.StatusOK) render.JSON(w, r, posts) return } func getBlogPostsByAuthor(w http.ResponseWriter, r *http.Request) { returnError := ReturnError{} referenceID := &ReferenceID{} err := json.NewDecoder(r.Body).Decode(referenceID) // hardcode 9001 for cool kid points if err != nil { referenceID.LastID = 9001 } search_id := referenceID.LastID // if someone is lame and sends up a negative number... if search_id < 1 { search_id = 9001 } author := chi.URLParam(r, "author") search := ` SELECT id, title, slug, author, content, time_published, modified, last_modified FROM posts WHERE id < $1 AND author = $2 ORDER BY id DESC FETCH FIRST 10 ROWS ONLY ` rows, err := DB.Query(search, search_id, author) if err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } defer rows.Close() post := BlogPost{} posts := make(BlogPosts, 0) for rows.Next() { if err := rows.Scan(&post.ID, &post.Title, &post.Slug, &post.Author, &post.Content, &post.TimePublished, &post.Modified, &post.TimeModified); err != nil { } posts = append(posts, post) } if err := rows.Err(); err != nil { returnError.Message = "something is super broken..." w.WriteHeader(http.StatusInternalServerError) render.JSON(w, r, returnError) fmt.Println(err) return } w.WriteHeader(http.StatusOK) render.JSON(w, r, posts) return } func getRssFeed(w http.ResponseWriter, r *http.Request) { return }