First version of web UI

internal/database/articles.go

@@ -0,0 +1,88 @@
+package database
+
+import (
+	"fmt"
+	_ "github.com/lib/pq"
+	"gitlab.com/kbr4/svevijesti/internal/model"
+	"math"
+	"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)
+	}
+
+	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
+	}
+
+	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
+}

internal/database/database.go

@@ -4,7 +4,6 @@ import (
 	"database/sql"
 	"fmt"
 	_ "github.com/lib/pq"
-	"gitlab.com/kbr4/svevijesti/internal/model"
 )
 
 const (
@@ -24,36 +23,3 @@ func Connect() (*Store, error) {
 	db, err := sql.Open("postgres", psqlInfo)
 	return db, err
 }
-
-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)
-	}
-
-	row := query.QueryRow(url)
-	err = row.Scan(&exists)
-	if err != nil {
-		panic(err)
-	}
-	return exists
-}