Added start of db rework, also added Qol changes

internal/database/chapter.go

@@ -0,0 +1,62 @@
+package database
+
+import (
+	"database/sql"
+	"sync"
+)
+
+type Chapter struct {
+	Id            int
+	Manga         *Manga
+	Url           string
+	Name          string
+	Number        int
+	TimeStampUnix int64
+}
+
+type ChapterTable[K comparable] struct {
+	mutex    sync.Mutex
+	chapters map[K]Chapter
+	updated  map[K]DbStatus
+}
+
+func (c *ChapterTable[K]) Get(key K) (Chapter, bool) {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+	val, ok := c.chapters[key]
+	return val, ok
+}
+
+func (c *ChapterTable[K]) Set(key K, new Chapter) {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+	val, ok := c.updated[key]
+	if ok && val == Loaded {
+		c.updated[key] = Updated
+	} else {
+		c.updated[key] = New
+	}
+	c.chapters[key] = new
+}
+
+func (c *ChapterTable[K]) All() []Chapter {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+	res := make([]Chapter, len(c.chapters))
+	counter := 0
+	for _, chapter := range c.chapters {
+		res[counter] = chapter
+		counter++
+	}
+	return res
+}
+
+func (c *ChapterTable[K]) Save(db *sql.DB) error {
+	//TODO implement me
+	panic("implement me")
+}
+
+func (c *ChapterTable[K]) Load(db *sql.DB) error {
+	//TODO implement me
+	panic("implement me")
+}

internal/database/database.go

@@ -10,26 +10,6 @@ import (
 	_ "github.com/mattn/go-sqlite3"
 )
 
-type Manga struct {
-	Id             int
-	Title          string
-	TimeStampUnix  int64
-	Thumbnail      *bytes.Buffer
-	LastChapterNum int
-
-	// Not in DB
-	LatestChapter *Chapter
-}
-
-type Chapter struct {
-	Id            int
-	Manga         *Manga
-	Url           string
-	Name          string
-	Number        int
-	TimeStampUnix int64
-}
-
 type Manager struct {
 	ConnectionString string
 	db               *sql.DB

internal/database/manga.go

@@ -0,0 +1,94 @@
+package database
+
+import (
+	"bytes"
+	"database/sql"
+	"sync"
+)
+
+type Manga struct {
+	Id             int
+	Title          string
+	TimeStampUnix  int64
+	Thumbnail      *bytes.Buffer
+	LastChapterNum int
+
+	// Not in DB
+	LatestChapter *Chapter
+}
+
+type MangaTable[K comparable] struct {
+	mutex   sync.Mutex
+	mangas  map[K]Manga
+	updated map[K]DbStatus
+}
+
+func (m *MangaTable[K]) Get(key K) (Manga, bool) {
+	m.mutex.Lock()
+	defer m.mutex.Unlock()
+	val, ok := m.mangas[key]
+	return val, ok
+}
+
+func (m *MangaTable[K]) Set(key K, new Manga) {
+	m.mutex.Lock()
+	defer m.mutex.Unlock()
+	val, ok := m.updated[key]
+	if ok && val == Loaded {
+		m.updated[key] = Updated
+	} else {
+		m.updated[key] = New
+	}
+	m.mangas[key] = new
+}
+
+func (m *MangaTable[K]) All() []Manga {
+	m.mutex.Lock()
+	defer m.mutex.Unlock()
+	res := make([]Manga, len(m.mangas))
+	counter := 0
+	for _, manga := range m.mangas {
+		res[counter] = manga
+		counter++
+	}
+	return res
+}
+
+func (m *MangaTable[K]) Save(db *sql.DB) error {
+	m.mutex.Lock()
+	defer m.mutex.Unlock()
+	for k, status := range m.updated {
+		if status == Loaded {
+			continue
+		} else if status == Updated {
+			manga := m.mangas[k]
+			err := updateManga(db, &manga)
+			if err != nil {
+				return err
+			}
+		} else {
+			manga := m.mangas[k]
+			err := insertManga(db, &manga)
+			if err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}
+
+func (m *MangaTable[K]) Load(db *sql.DB) error {
+	panic("")
+}
+
+func updateManga(db *sql.DB, m *Manga) error {
+	const cmd = "UPDATE Manga set Title = ?, TimeStampUnixEpoch = ?, Thumbnail = ?, LatestAvailableChapter = ? WHERE ID = ?"
+	_, err := db.Exec(cmd, m.Title, m.TimeStampUnix, m.Thumbnail.Bytes(), m.LastChapterNum, m.Id)
+	return err
+}
+
+func insertManga(db *sql.DB, manga *Manga) error {
+	const cmd = "INSERT INTO Manga(ID, Title, TimeStampUnixEpoch, Thumbnail, LatestAvailableChapter) values(?, ?, ?, ?, ?)"
+	_, err := db.Exec(cmd, manga.Id, manga.Title, manga.TimeStampUnix, manga.Thumbnail.Bytes(), manga.LastChapterNum)
+	return err
+}

internal/database/table.go

@@ -0,0 +1,20 @@
+package database
+
+import "database/sql"
+
+type DbStatus int
+
+const (
+	New DbStatus = iota
+	Loaded
+	Updated
+)
+
+// Table TODO: This Could probably be a generic instead of interface / both
+type Table[K comparable, T any] interface {
+	Get(key K) (T, bool)
+	Set(key K, new T)
+	All() []T
+	Save(db *sql.DB) error
+	Load(db *sql.DB) error
+}