Initial test, but meh

This commit is contained in:
Pablu23
2024-05-21 11:56:14 +02:00
parent 0904a1214e
commit 20ad56b155
6 changed files with 63 additions and 36 deletions

View File

@@ -1,7 +1,8 @@
package database
type Chapter struct {
Id int `gorm:"primary_key;AUTO_INCREMENT"`
Id int `gorm:"primary_key;autoIncrement;"`
ChapterId int
Url string
Name string
Number string
@@ -11,7 +12,7 @@ type Chapter struct {
func NewChapter(id int, mangaId int, url string, name string, number string, timeStampUnix int64) Chapter {
return Chapter{
Id: id,
ChapterId: id,
Url: url,
Name: name,
Number: number,

View File

@@ -54,6 +54,6 @@ func (dbMgr *Manager) Delete(mangaId int) {
}
func (dbMgr *Manager) createDatabaseIfNotExists() error {
err := dbMgr.Db.AutoMigrate(&Manga{}, &Chapter{}, &Setting{})
err := dbMgr.Db.AutoMigrate(&MangaDefinition{}, &User{}, &Manga{}, &Chapter{}, &Setting{})
return err
}

View File

@@ -1,17 +1,26 @@
package database
type Manga struct {
type MangaDefinition struct {
Id int `gorm:"primary_key;AUTO_INCREMENT"`
Title string
TimeStampUnix int64
Thumbnail []byte
LastChapterNum string
Chapters []Chapter
// Chapters []Chapter
//`gorm:"foreignkey:MangaID"`
}
func NewManga(id int, title string, timeStampUnix int64) Manga {
return Manga{
type Manga struct {
Id int `gorm:"primary_key;AUTO_INCREMENT"`
MangaDefinitionId int
Definition MangaDefinition `gorm:"foreignKey:MangaDefinitionId"`
UserId int
TimeStampUnix int64
Chapters []Chapter `gorm:"foreignKey:MangaId"`
}
func NewMangaDefinition(id int, title string, timeStampUnix int64) MangaDefinition {
return MangaDefinition{
Id: id,
Title: title,
TimeStampUnix: timeStampUnix,
@@ -21,20 +30,20 @@ func NewManga(id int, title string, timeStampUnix int64) Manga {
// GetLatestChapter TODO: Cache this somehow
func (m *Manga) GetLatestChapter() (*Chapter, bool) {
highest := int64(0)
index := 0
for i, chapter := range m.Chapters {
if chapter.MangaId == m.Id && highest < chapter.TimeStampUnix {
highest = chapter.TimeStampUnix
index = i
}
}
// highest := int64(0)
// index := 0
// for i, chapter := range m.Chapters {
// if chapter.MangaId == m.Manga.Id && highest < chapter.TimeStampUnix {
// highest = chapter.TimeStampUnix
// index = i
// }
// }
if highest == 0 {
return nil, false
}
// if highest == 0 {
return nil, false
// }
return &m.Chapters[index], true
// return &m.Chapters[index], true
//result := db.Where("manga.id = ?", m.Id).Order("TimeStampUnix desc").Take(&chapter)
//if result.Error != nil && !errors.Is(result.Error, gorm.ErrRecordNotFound) {

18
internal/database/user.go Normal file
View File

@@ -0,0 +1,18 @@
package database
type User struct {
Id int `gorm:"primary_key;AUTO_INCREMENT"`
DisplayName string
LoginName string
PwdHash []byte
Salt []byte
Mangas []Manga `gorm:"foreignKey:UserId"`
}
// type UserManga struct {
// Id int `gorm:"primary_key;AUTO_INCREMENT"`
// DisplayName string
// Manga Manga
// User User
// // Chapters []Chapter `gorm:"ForeignKey:ChapterId,UserId;References:Id,UserId"`
// }

View File

@@ -38,7 +38,7 @@ func (s *Server) HandleNew(w http.ResponseWriter, r *http.Request) {
func (s *Server) HandleMenu(w http.ResponseWriter, _ *http.Request) {
tmpl := template.Must(view.GetViewTemplate(view.Menu))
var all []*database.Manga
_ = s.DbMgr.Db.Preload("Chapters").Find(&all)
_ = s.DbMgr.Db.Preload("Chapters").Where("user_id = ?", 1).Find(&all)
l := len(all)
mangaViewModels := make([]view.MangaViewModel, l)
counter := 0
@@ -57,7 +57,7 @@ func (s *Server) HandleMenu(w http.ResponseWriter, _ *http.Request) {
//TODO: Change all this to be more performant
for _, manga := range all {
title := cases.Title(language.English, cases.Compact).String(strings.Replace(manga.Title, "-", " ", -1))
title := cases.Title(language.English, cases.Compact).String(strings.Replace(manga.Definition.Title, "-", " ", -1))
t1 := time.Now().UnixNano()
@@ -78,30 +78,29 @@ func (s *Server) HandleMenu(w http.ResponseWriter, _ *http.Request) {
// This is very slow
// TODO: put this into own Method
if manga.LastChapterNum == "" {
if manga.Definition.LastChapterNum == "" {
err, updated := s.UpdateLatestAvailableChapter(manga)
if err != nil {
fmt.Println(err)
}
if updated {
s.DbMgr.Db.Save(manga)
s.DbMgr.Db.Save(manga.Definition)
}
}
t2 = time.Now().UnixNano()
titNs += t2 - t1
latestChapter, ok := manga.GetLatestChapter()
if !ok {
continue
}
mangaViewModels[counter] = view.MangaViewModel{
ID: manga.Id,
ID: manga.Definition.Id,
Title: title,
Number: latestChapter.Number,
LastNumber: manga.LastChapterNum,
LastNumber: manga.Definition.LastChapterNum,
// I Hate this time Format... 15 = hh, 04 = mm, 02 = DD, 01 = MM, 06 == YY
LastTime: time.Unix(manga.TimeStampUnix, 0).Format("15:04 (02-01-06)"),
Url: latestChapter.Url,
@@ -214,10 +213,10 @@ func (s *Server) HandleCurrent(w http.ResponseWriter, _ *http.Request) {
fmt.Println(err)
}
var manga database.Manga
var manga database.MangaDefinition
result := s.DbMgr.Db.First(&manga, mangaId)
if result.Error != nil && errors.Is(result.Error, gorm.ErrRecordNotFound) {
manga = database.NewManga(mangaId, title, time.Now().Unix())
manga = database.NewMangaDefinition(mangaId, title, time.Now().Unix())
} else {
manga.TimeStampUnix = time.Now().Unix()
}

View File

@@ -212,9 +212,9 @@ func (s *Server) LoadCurr() {
}
func (s *Server) UpdateLatestAvailableChapter(manga *database.Manga) (error, bool) {
fmt.Printf("Updating Manga: %s\n", manga.Title)
fmt.Printf("Updating Manga: %s\n", manga.Definition.Title)
l, err := s.Provider.GetChapterList("/title/" + strconv.Itoa(manga.Id))
l, err := s.Provider.GetChapterList("/title/" + strconv.Itoa(manga.Definition.Id))
if err != nil {
return err, false
}
@@ -227,16 +227,16 @@ func (s *Server) UpdateLatestAvailableChapter(manga *database.Manga) (error, boo
chapterNumberStr := strings.Replace(c, "ch_", "", 1)
if manga.LastChapterNum == chapterNumberStr {
if manga.Definition.LastChapterNum == chapterNumberStr {
return nil, false
} else {
manga.LastChapterNum = chapterNumberStr
manga.Definition.LastChapterNum = chapterNumberStr
return nil, true
}
}
func (s *Server) LoadThumbnail(manga *database.Manga) (path string, updated bool, err error) {
strId := strconv.Itoa(manga.Id)
strId := strconv.Itoa(manga.Definition.Id)
s.Mutex.Lock()
defer s.Mutex.Unlock()
@@ -244,8 +244,8 @@ func (s *Server) LoadThumbnail(manga *database.Manga) (path string, updated bool
return strId, false, nil
}
if manga.Thumbnail != nil {
s.ImageBuffers[strId] = manga.Thumbnail
if manga.Definition.Thumbnail != nil {
s.ImageBuffers[strId] = manga.Definition.Thumbnail
return strId, false, nil
}
@@ -257,7 +257,7 @@ func (s *Server) LoadThumbnail(manga *database.Manga) (path string, updated bool
if err != nil {
return "", false, err
}
manga.Thumbnail = ram
manga.Definition.Thumbnail = ram
s.ImageBuffers[strId] = ram
return strId, true, nil
}