Fixed Thumbnail Db Entry not loading, added favicon.ico, added going back to menu, if no more Chapters exist, added auto start of default browser with url, added dark mode to main Menu (doesnt save state), added Constructor for Server

internal/server/handler.go

@@ -212,13 +212,20 @@ func (s *Server) HandleNext(w http.ResponseWriter, r *http.Request) {
 		}(*s.PrevViewModel, s)
 	}
 
+	if s.NextViewModel == nil || s.NextSubUrl == "" {
+		http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
+		err := s.DbMgr.Save()
+		if err != nil {
+			fmt.Println(err)
+		}
+		return
+	}
+
 	s.PrevViewModel = s.CurrViewModel
 	s.CurrViewModel = s.NextViewModel
 	s.PrevSubUrl = s.CurrSubUrl
 	s.CurrSubUrl = s.NextSubUrl
 
-	<-s.NextReady
-
 	go s.LoadNext()
 
 	http.Redirect(w, r, "/current/", http.StatusTemporaryRedirect)
@@ -236,13 +243,20 @@ func (s *Server) HandlePrev(w http.ResponseWriter, r *http.Request) {
 		}(*s.NextViewModel, s)
 	}
 
+	if s.PrevViewModel == nil || s.PrevSubUrl == "" {
+		http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
+		err := s.DbMgr.Save()
+		if err != nil {
+			fmt.Println(err)
+		}
+		return
+	}
+
 	s.NextViewModel = s.CurrViewModel
 	s.CurrViewModel = s.PrevViewModel
 	s.NextSubUrl = s.CurrSubUrl
 	s.CurrSubUrl = s.PrevSubUrl
 
-	<-s.PrevReady
-
 	go s.LoadPrev()
 
 	http.Redirect(w, r, "/current/", http.StatusTemporaryRedirect)

internal/server/server.go

@@ -2,6 +2,7 @@ package server
 
 import (
 	"bytes"
+	_ "embed"
 	"fmt"
 	"io"
 	"mangaGetter/internal/database"
@@ -32,35 +33,51 @@ type Server struct {
 	IsLast  bool
 
 	DbMgr *database.Manager
+}
 
-	// I'm not even sure if this helps.
-	// If you press next and then prev too fast you still lock yourself out
-	NextReady chan bool
-	PrevReady chan bool
+func New(provider provider.Provider, db *database.Manager) *Server {
+	s := Server{
+		ImageBuffers: make(map[string]*bytes.Buffer),
+		Provider:     provider,
+		DbMgr:        db,
+		Mutex:        &sync.Mutex{},
+	}
+
+	s.AddIco()
+
+	return &s
 }
 
 func (s *Server) LoadNext() {
 	c, err := s.Provider.GetHtml(s.CurrSubUrl)
 	if err != nil {
 		fmt.Println(err)
+		s.NextSubUrl = ""
+		s.NextViewModel = nil
 		return
 	}
 
 	next, err := s.Provider.GetNext(c)
 	if err != nil {
 		fmt.Println(err)
+		s.NextSubUrl = ""
+		s.NextViewModel = nil
 		return
 	}
 
 	html, err := s.Provider.GetHtml(next)
 	if err != nil {
 		fmt.Println(err)
+		s.NextSubUrl = ""
+		s.NextViewModel = nil
 		return
 	}
 
 	imagesNext, err := s.AppendImagesToBuf(html)
 	if err != nil {
 		fmt.Println(err)
+		s.NextSubUrl = ""
+		s.NextViewModel = nil
 		return
 	}
 
@@ -73,32 +90,38 @@ func (s *Server) LoadNext() {
 	full := strings.Replace(title, "-", " ", -1) + " - " + strings.Replace(chapter, "_", " ", -1)
 
 	s.NextViewModel = &view.ImageViewModel{Images: imagesNext, Title: full}
-
 	s.NextSubUrl = next
 	fmt.Println("Loaded next")
-	s.NextReady <- true
 }
 
 func (s *Server) LoadPrev() {
 	c, err := s.Provider.GetHtml(s.CurrSubUrl)
 	if err != nil {
 		fmt.Println(err)
+		s.PrevSubUrl = ""
+		s.PrevViewModel = nil
 		return
 	}
 	prev, err := s.Provider.GetPrev(c)
 	if err != nil {
 		fmt.Println(err)
+		s.PrevSubUrl = ""
+		s.PrevViewModel = nil
 		return
 	}
 	html, err := s.Provider.GetHtml(prev)
 	if err != nil {
 		fmt.Println(err)
+		s.PrevSubUrl = ""
+		s.PrevViewModel = nil
 		return
 	}
 
 	imagesNext, err := s.AppendImagesToBuf(html)
 	if err != nil {
 		fmt.Println(err)
+		s.PrevSubUrl = ""
+		s.PrevViewModel = nil
 		return
 	}
 
@@ -114,7 +137,6 @@ func (s *Server) LoadPrev() {
 
 	s.PrevSubUrl = prev
 	fmt.Println("Loaded prev")
-	s.PrevReady <- true
 }
 
 func (s *Server) LoadCurr() {
@@ -187,6 +209,16 @@ func (s *Server) AppendImagesToBuf(html string) ([]view.Image, error) {
 	return images, nil
 }
 
+//go:embed favicon.ico
+var ico []byte
+
+func (s *Server) AddIco() {
+	buf := bytes.NewBuffer(ico)
+	s.Mutex.Lock()
+	s.ImageBuffers["favicon.ico"] = buf
+	s.Mutex.Unlock()
+}
+
 func addFileToRam(url string) (*bytes.Buffer, error) {
 	// Get the data
 	resp, err := http.Get(url)