tver/internal/web/library.go

package web

import (
	"errors"
	"fmt"
	"net/http"
	"os"
	"path/filepath"
	"strconv"

	"gitea.kazu634.com/kazu634/tver/internal/store"
	"github.com/jackc/pgx/v5"
)

const libraryLimit = 100

type libraryItemView struct {
	ID                               int64
	ThumbnailURL, SeriesTitle, Title string
	BroadcasterName, DurationText    string
	FileSizeText, FinishedAtText     string
}

type libraryView struct {
	Items []libraryItemView
}

type videoView struct {
	libraryItemView
	VideoURL string
}

func (s *server) libraryPage(w http.ResponseWriter, r *http.Request) {
	jobs, err := s.jobs.ListSucceededJobs(r.Context(), libraryLimit)
	if err != nil {
		s.logger.Printf("ライブラリの取得に失敗: %v", err)
		http.Error(w, "ライブラリを取得できませんでした", http.StatusInternalServerError)
		return
	}
	view := libraryView{Items: make([]libraryItemView, 0, len(jobs))}
	for _, job := range jobs {
		view.Items = append(view.Items, s.libraryItem(job))
	}
	s.render(w, "layout", page{Title: "ライブラリ", Library: &view})
}

func (s *server) videoPage(w http.ResponseWriter, r *http.Request) {
	job, ok := s.succeededJob(w, r)
	if !ok {
		return
	}
	item := s.libraryItem(job)
	s.render(w, "layout", page{Title: job.SeriesTitle + " " + job.Title,
		Video: &videoView{libraryItemView: item, VideoURL: "/videos/" + strconv.FormatInt(job.ID, 10)}})
}

func (s *server) libraryItem(job store.Job) libraryItemView {
	item := libraryItemView{ID: job.ID, ThumbnailURL: "https://statics.tver.jp" + job.ThumbnailPath,
		SeriesTitle: job.SeriesTitle, Title: job.Title, BroadcasterName: job.BroadcasterName,
		DurationText: fmt.Sprintf("%d分", durationMinutes(int64(job.DurationSeconds)))}
	if job.FileSizeBytes != nil {
		item.FileSizeText = formatBytes(float64(*job.FileSizeBytes))
	}
	if job.FinishedAt != nil {
		item.FinishedAtText = job.FinishedAt.In(s.zone).Format("2006-01-02 15:04 MST")
	}
	return item
}

func (s *server) serveVideo(w http.ResponseWriter, r *http.Request) {
	job, ok := s.succeededJob(w, r)
	if !ok || job.FilePath == nil {
		if ok {
			http.NotFound(w, r)
		}
		return
	}
	root, err := os.OpenRoot(s.videoDir)
	if err != nil {
		http.NotFound(w, r)
		return
	}
	defer root.Close()
	file, err := root.Open(*job.FilePath)
	if err != nil {
		http.NotFound(w, r)
		return
	}
	defer file.Close()
	info, err := file.Stat()
	if err != nil || !info.Mode().IsRegular() {
		http.NotFound(w, r)
		return
	}
	w.Header().Set("Content-Type", "video/mp4")
	http.ServeContent(w, r, filepath.Base(*job.FilePath), info.ModTime(), file)
}

func (s *server) deleteVideo(w http.ResponseWriter, r *http.Request) {
	job, ok := s.succeededJob(w, r)
	if !ok {
		return
	}
	if job.FilePath != nil {
		if !filepath.IsLocal(*job.FilePath) {
			http.Error(w, "動画ファイルの場所が不正です", http.StatusConflict)
			return
		}
		root, err := os.OpenRoot(s.videoDir)
		if err != nil {
			http.Error(w, "動画ファイルの場所が不正です", http.StatusConflict)
			return
		}
		defer root.Close()
		file, err := root.Open(*job.FilePath)
		if err == nil {
			_ = file.Close()
		} else if !errors.Is(err, os.ErrNotExist) {
			http.Error(w, "動画ファイルの場所が不正です", http.StatusConflict)
			return
		}
		if err := root.Remove(*job.FilePath); err != nil && !errors.Is(err, os.ErrNotExist) {
			s.logger.Printf("動画ファイルの削除に失敗: %v", err)
			http.Error(w, "動画ファイルを削除できませんでした", http.StatusInternalServerError)
			return
		}
	}
	if err := s.jobs.DeleteSucceededJob(r.Context(), job.ID); err != nil {
		s.logger.Printf("動画の記録削除に失敗: %v", err)
		http.Error(w, "動画の記録を削除できませんでした", http.StatusInternalServerError)
		return
	}
	http.Redirect(w, r, "/library", http.StatusSeeOther)
}

func (s *server) succeededJob(w http.ResponseWriter, r *http.Request) (store.Job, bool) {
	id, err := strconv.ParseInt(r.PathValue("id"), 10, 64)
	if err != nil || id <= 0 {
		http.NotFound(w, r)
		return store.Job{}, false
	}
	job, err := s.jobs.GetSucceededJob(r.Context(), id)
	if errors.Is(err, pgx.ErrNoRows) {
		http.NotFound(w, r)
		return store.Job{}, false
	}
	if err != nil {
		s.logger.Printf("完了ジョブの取得に失敗: %v", err)
		http.Error(w, "動画を取得できませんでした", http.StatusInternalServerError)
		return store.Job{}, false
	}
	return job, true
}