From e0e8f8742c05b39c254a53908dfa713ab6512c77 Mon Sep 17 00:00:00 2001
From: Sebastiaan de Schaetzen <sebastiaan.de.schaetzen@gmail.com>
Date: Tue, 17 Jun 2025 20:05:40 +0200
Subject: [PATCH] Add watch progress feature to video model and UI

---
 .gitignore                          |  1 +
 migrations/6_add_watch_progress.sql |  1 +
 static/index.html                   | 27 ++++++++++++++++++++++++++-
 webview.go                          | 17 +++++++++--------
 4 files changed, 37 insertions(+), 9 deletions(-)
 create mode 100644 migrations/6_add_watch_progress.sql

diff --git a/.gitignore b/.gitignore
index b5b4739..a581eb7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,6 +1,7 @@
 .idea
 *.iml
 *.db3
+*.db3.*
 *.db3-*
 *.bak*
 /downloads/
diff --git a/migrations/6_add_watch_progress.sql b/migrations/6_add_watch_progress.sql
new file mode 100644
index 0000000..86cd36b
--- /dev/null
+++ b/migrations/6_add_watch_progress.sql
@@ -0,0 +1 @@
+ALTER TABLE videos ADD COLUMN watch_progress REAL;
diff --git a/static/index.html b/static/index.html
index 0bc5b07..d75f81b 100644
--- a/static/index.html
+++ b/static/index.html
@@ -93,6 +93,25 @@
 			opacity: 1;
 		}
 
+		.progress-bar-container {
+			position: absolute;
+			bottom: 0;
+			left: 0;
+			width: 100%;
+			height: 5px; /* Or adjust as needed */
+			background-color: rgba(0, 0, 0, 0.5);
+			border-bottom-left-radius: 8px;
+			border-bottom-right-radius: 8px;
+			overflow: hidden;
+		}
+
+		.progress-bar {
+			height: 100%;
+			background-color: #3498db; /* A nice blue */
+			width: 0; /* Initially no progress */
+			transition: width 0.2s ease-in-out;
+		}
+
 		.play-button {
 			position: absolute;
 			top: 50%;
@@ -339,7 +358,13 @@
 			let container = $(`<div class="video-thumbnail" tabindex="0" role="button" aria-label="${data.title}"></div>`);
 			let thumbnailImg = $(`<img src="${data.thumbnail}" alt="" style="width: 100%; height: 100%; object-fit: cover;">`);
 			let title = $(`<div class="video-title">${data.title}</div>`);
-			container.append(thumbnailImg).append(title);
+			let progressBarContainer = $(`<div class="progress-bar-container"><div class="progress-bar"></div></div>`);
+
+			container.append(thumbnailImg).append(title).append(progressBarContainer);
+
+			if (data.watchProgress > 0) {
+				progressBarContainer.find(".progress-bar").css("width", (data.watchProgress * 100) + "%");
+			}
 
 			let index;
 			if (append) {
diff --git a/webview.go b/webview.go
index 8401f02..dc5dd14 100644
--- a/webview.go
+++ b/webview.go
@@ -52,10 +52,11 @@ func serveWebview(db *mysqlite.Db) {
 }
 
 type VideoModel struct {
-	ID        int    `json:"id"`
-	Thumbnail string `json:"thumbnail"`
-	URL       string `json:"url"`
-	Title     string `json:"title"`
+	ID            int     `json:"id"`
+	Thumbnail     string  `json:"thumbnail"`
+	URL           string  `json:"url"`
+	Title         string  `json:"title"`
+	WatchProgress float64 `json:"watchProgress"`
 }
 
 type HomePageModel struct {
@@ -66,13 +67,13 @@ type HomePageModel struct {
 
 func (a *App) homePage(c *gin.Context) {
 	model := HomePageModel{}
-	errCurrent := a.db.Query("SELECT id, thumbnail, title FROM videos WHERE (watch_state IS NULL OR watch_state != 'watched') ORDER BY upload_date, episode LIMIT 1").
-		ScanSingle(&model.Current.ID, &model.Current.Thumbnail, &model.Current.Title)
+	errCurrent := a.db.Query("SELECT id, thumbnail, title, watch_progress FROM videos WHERE (watch_state IS NULL OR watch_state != 'watched') ORDER BY upload_date, episode LIMIT 1").
+		ScanSingle(&model.Current.ID, &model.Current.Thumbnail, &model.Current.Title, &model.Current.WatchProgress)
 
 	var errNext, errPrevious error
-	for row := range a.db.Query("SELECT id, thumbnail, title FROM videos WHERE (watch_state IS NULL OR watch_state != 'watched') ORDER BY upload_date, episode LIMIT 3 OFFSET 1").Range(&errNext) {
+	for row := range a.db.Query("SELECT id, thumbnail, title, watch_progress FROM videos WHERE (watch_state IS NULL OR watch_state != 'watched') ORDER BY upload_date, episode LIMIT 3 OFFSET 1").Range(&errNext) {
 		video := VideoModel{}
-		err := row.Scan(&video.ID, &video.Thumbnail, &video.Title)
+		err := row.Scan(&video.ID, &video.Thumbnail, &video.Title, &video.WatchProgress)
 		if err != nil {
 			log.Printf("Failed to scan next video: %v", err)
 			c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve next videos"})