136 lines
3.2 KiB
Go
136 lines
3.2 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
_ "github.com/lib/pq"
|
|
"gitlab.com/kbr4/svevijesti/internal/model"
|
|
"html/template"
|
|
"math"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func InsertArticle(store *Store, article model.ScrapedArticle) (err error) {
|
|
query := `
|
|
INSERT INTO articles
|
|
(title, content, slug, original_url, source_id)
|
|
VALUES
|
|
($1,$2,$3,$4,$5);`
|
|
|
|
_, err = store.Exec(query, article.Title, article.Content, article.Slug, article.OriginalUrl, article.SourceId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func IsSaved(store *Store, url string) bool {
|
|
|
|
exists := false
|
|
query, err := store.Prepare(`
|
|
select exists(select 1 from articles where original_url = $1);
|
|
`)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer query.Close()
|
|
|
|
row := query.QueryRow(url)
|
|
err = row.Scan(&exists)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func ArticlesForDay(store *Store, day time.Time) (articles []model.DisplayArticle, err error) {
|
|
|
|
result := []model.DisplayArticle{}
|
|
query, err := store.Prepare(`
|
|
select id,title, content, slug, original_url, source_id, created_at from articles where created_at > $1 and created_at < $2 and LENGTH(content) > 10 order by id desc;
|
|
`)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer query.Close()
|
|
|
|
tomorrow := day.AddDate(0, 0, 1)
|
|
todayDate := day.Format("2006-01-02")
|
|
tomorrowDate := tomorrow.Format("2006-01-02")
|
|
|
|
rows, err := query.Query(todayDate, tomorrowDate)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
r := model.DisplayArticle{}
|
|
err = rows.Scan(&r.ID, &r.Title, &r.Content, &r.Slug, &r.OriginalUrl, &r.SourceId, &r.CreatedAt)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
|
|
ago := time.Now().Sub(r.CreatedAt)
|
|
hours := ago.Hours()
|
|
|
|
if hours < 1 {
|
|
r.FormatedCreatedAt = fmt.Sprintf("Prije %d minuta.", int(math.Floor(ago.Minutes())))
|
|
|
|
} else if hours > 24 {
|
|
r.FormatedCreatedAt = r.CreatedAt.Format("01.02.2006. 15:04:05")
|
|
} else {
|
|
r.FormatedCreatedAt = fmt.Sprintf("Prije %d sati.", int(math.Floor(hours)))
|
|
}
|
|
r.SourceName = model.SourceName(r.SourceId)
|
|
|
|
result = append(result, r)
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func ArticleByID(store *Store, ID int, slug string) (article model.DisplayArticle, err error) {
|
|
|
|
result := model.DisplayArticle{}
|
|
query, err := store.Prepare(`
|
|
select id,title, content, slug, original_url, source_id, created_at from articles where id = $1 and slug = $2;
|
|
`)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer query.Close()
|
|
|
|
row := query.QueryRow(ID, slug)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
|
|
r := model.DisplayArticle{}
|
|
content := ""
|
|
err = row.Scan(&r.ID, &r.Title, &content, &r.Slug, &r.OriginalUrl, &r.SourceId, &r.CreatedAt)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
|
|
ago := time.Now().Sub(r.CreatedAt)
|
|
hours := ago.Hours()
|
|
|
|
r.Content = template.HTML(strings.Replace(content, "\n", "<br>\n", -1))
|
|
|
|
if hours < 1 {
|
|
r.FormatedCreatedAt = fmt.Sprintf("Prije %d minuta.", int(math.Floor(ago.Minutes())))
|
|
|
|
} else if hours > 24 {
|
|
r.FormatedCreatedAt = r.CreatedAt.Format("01.02.2006. 15:04:05")
|
|
} else {
|
|
r.FormatedCreatedAt = fmt.Sprintf("Prije %d sati.", int(math.Floor(hours)))
|
|
}
|
|
r.SourceName = model.SourceName(r.SourceId)
|
|
|
|
result = r
|
|
|
|
return result, nil
|
|
}
|