1 files changed, 69 insertions, 0 deletions
diff --git a/src/db/talk.go b/src/db/talk.go
new file mode 100644
index 0000000..c286aa8
--- /dev/null
+++ b/src/db/talk.go
@@ -0,0 +1,69 @@
+package db
+
+import (
+	"fmt"
+
+	"git.darknebu.la/chaosdorf/freitagsfoo/src/structs"
+	pg "github.com/go-pg/pg/v9"
+	"github.com/google/uuid"
+)
+
+// InsertTalk inserts the given talk into the database
+func InsertTalk(db *pg.DB, talk *structs.Talk) error {
+	err := db.Insert(talk)
+	if err != nil {
+		return fmt.Errorf("could not insert talk into the db: %s", err)
+	}
+
+	return nil
+}
+
+// UpcomingTalksLimited returns the next 3 upcoming talks
+func UpcomingTalksLimited(db *pg.DB) ([]structs.Talk, error) {
+	var talks []structs.Talk
+	err := db.Model(&talks).Order("id DESC").Limit(3).Select()
+	if err != nil {
+		return []structs.Talk{}, fmt.Errorf("could not get the talks from the db: %s", err)
+	}
+
+	return talks, nil
+}
+
+// UpcomingTalks returns the next upcoming talks
+func UpcomingTalks(db *pg.DB) ([]structs.Talk, error) {
+	var talks []structs.Talk
+	err := db.Model(&talks).Order("id DESC").Select()
+	if err != nil {
+		return []structs.Talk{}, fmt.Errorf("could not get the talks from the db: %s", err)
+	}
+
+	return talks, nil
+}
+
+// CountUpcomingTalks counts the amount of talks upcoming
+func CountUpcomingTalks(db *pg.DB) (int, error) {
+
+	var talks []structs.Talk
+	count, err := db.Model(&talks).Where("upcoming = ?", true).SelectAndCount()
+	if err != nil {
+		return -1, fmt.Errorf("could not get the talks from the db: %s", err)
+	}
+
+	return count, nil
+}
+
+// TalkByUUID returns the talk with the given UUID
+func TalkByUUID(db *pg.DB, uuidString string) (structs.Talk, error) {
+	parsedUUID, err := uuid.Parse(uuidString)
+	if err != nil {
+		return structs.Talk{}, fmt.Errorf("could not parse the UUID: %s", err)
+	}
+
+	var talk structs.Talk
+	err = db.Model(&talk).Where("uuid = ?", parsedUUID).Select()
+	if err != nil {
+		return structs.Talk{}, fmt.Errorf("could not get the talks from the db: %s", err)
+	}
+
+	return talk, nil
+}