Compare commits

..

3 Commits

2 changed files with 82 additions and 9 deletions

66
main.go
View File

@@ -8,6 +8,7 @@ import (
"log" "log"
"net/http" "net/http"
"os" "os"
"strconv"
"time" "time"
"github.com/PuerkitoBio/goquery" "github.com/PuerkitoBio/goquery"
@@ -20,7 +21,7 @@ type Item struct {
URL string `json:"url" db:"url"` URL string `json:"url" db:"url"`
Title string `json:"title" db:"title"` Title string `json:"title" db:"title"`
Description string `json:"description" db:"description"` Description string `json:"description" db:"description"`
ImageURL string `json:"imageUrl" db:"image_url"` Image *string `json:"image" db:"image"`
CreatedAt time.Time `json:"createdAt" db:"created_at"` CreatedAt time.Time `json:"createdAt" db:"created_at"`
UpdatedAt time.Time `json:"updatedAt" db:"updated_at"` UpdatedAt time.Time `json:"updatedAt" db:"updated_at"`
} }
@@ -28,7 +29,7 @@ type Item struct {
type ArticleMetadata struct { type ArticleMetadata struct {
Title string Title string
Description string Description string
ImageURL string Image *string
} }
func getArticleMetadata(url string) (*ArticleMetadata, error) { func getArticleMetadata(url string) (*ArticleMetadata, error) {
@@ -38,6 +39,12 @@ func getArticleMetadata(url string) (*ArticleMetadata, error) {
} }
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode < 200 || resp.StatusCode >= 400 {
return &ArticleMetadata{
Title: url,
}, nil
}
doc, err := goquery.NewDocumentFromReader(resp.Body) doc, err := goquery.NewDocumentFromReader(resp.Body)
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to parse page html: %v", err) return nil, fmt.Errorf("failed to parse page html: %v", err)
@@ -50,12 +57,12 @@ func getArticleMetadata(url string) (*ArticleMetadata, error) {
description := doc.Find(`head>meta[name="og:description"]`).AttrOr("content", "") description := doc.Find(`head>meta[name="og:description"]`).AttrOr("content", "")
imageUrl := doc.Find(`head>meta[name="og:image"]`).AttrOr("content", "") image := doc.Find(`head>meta[name="og:image"]`).AttrOr("content", "")
if imageUrl == "" { if image == "" {
doc.Find("img").Each(func(i int, s *goquery.Selection) { doc.Find("img").Each(func(i int, s *goquery.Selection) {
src := s.AttrOr("src", "") src := s.AttrOr("src", "")
if src != "" { if src != "" {
imageUrl = src image = src
return return
} }
}) })
@@ -64,7 +71,7 @@ func getArticleMetadata(url string) (*ArticleMetadata, error) {
return &ArticleMetadata{ return &ArticleMetadata{
Title: title, Title: title,
Description: description, Description: description,
ImageURL: imageUrl, Image: &image,
}, nil }, nil
} }
@@ -91,7 +98,7 @@ CREATE TABLE IF NOT EXISTS items (
url varchar not null unique, url varchar not null unique,
title varchar not null, title varchar not null,
description varchar, description varchar,
image_url varchar not null, image varchar default null,
created_at datetime default current_timestamp, created_at datetime default current_timestamp,
updated_at datetime default current_timestamp updated_at datetime default current_timestamp
); );
@@ -127,11 +134,11 @@ END;
return return
} }
if _, err := db.NamedExec("INSERT INTO items (url, title, description, image_url) VALUES (:url, :title, :description, :image_url)", map[string]any{ if _, err := db.NamedExec("INSERT INTO items (url, title, description, image) VALUES (:url, :title, :description, :image)", map[string]any{
"url": pageUrl, "url": pageUrl,
"title": meta.Title, "title": meta.Title,
"description": meta.Description, "description": meta.Description,
"image_url": meta.ImageURL, "image": meta.Image,
}); err != nil { }); err != nil {
http.Error(w, fmt.Sprintf("failed to add item to db: %v", err), 500) http.Error(w, fmt.Sprintf("failed to add item to db: %v", err), 500)
return return
@@ -162,6 +169,47 @@ END;
} }
}) })
mux.HandleFunc("GET /items/{id}", func(w http.ResponseWriter, r *http.Request) {
id, err := strconv.Atoi(r.PathValue("id"))
if err != nil {
http.Error(w, err.Error(), 500)
return
}
row := db.QueryRowx("SELECT * FROM items WHERE id = ?", id)
if row.Err() != nil {
http.Error(w, fmt.Sprintf("item not found: %v", row.Err()), 404)
return
}
item := &Item{}
if err := row.StructScan(item); err != nil {
http.Error(w, fmt.Sprintf("failed to scan item to struct: %v", err), 500)
return
}
tmpl.ExecuteTemplate(w, "item.html", struct {
Item *Item
}{
Item: item,
})
})
mux.HandleFunc("POST /items/{id}/delete", func(w http.ResponseWriter, r *http.Request) {
id, err := strconv.Atoi(r.PathValue("id"))
if err != nil {
http.Error(w, err.Error(), 500)
return
}
if _, err := db.Exec("DELETE FROM items WHERE id = ?", id); err != nil {
http.Error(w, fmt.Sprintf("failed to delete item: %v", err), 500)
return
}
http.Redirect(w, r, "/", http.StatusFound)
})
log.Println("starting http server") log.Println("starting http server")
if err := http.ListenAndServe(":5000", mux); err != nil { if err := http.ListenAndServe(":5000", mux); err != nil {
log.Fatalf("failed to start http server: %v\n", err) log.Fatalf("failed to start http server: %v\n", err)

25
views/item.html Normal file
View File

@@ -0,0 +1,25 @@
<!doctype html>
<html lang="en">
<head>
<meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>{{.Item.Title}}</title>
<style>
.container {
max-width: 1200px;
width: 100%;
margin: 0 auto;
}
</style>
</head>
<body>
<div class="container">
<img src="{{.Item.Image}}" alt="" width="auto" height="100px" />
<h2>{{.Item.Title}}</h2>
<form method="POST" action="/items/{{.Item.ID}}/delete">
<button type="submit">Delete item</button>
</form>
</div>
</body>
</html>