package cmd import ( "bufio" "context" "fmt" "os/exec" "regexp" "strconv" "strings" "sync" "time" tea "github.com/charmbracelet/bubbletea" ) const ( playerMaxTitleLength = 30 ) var ( regexpPosDur = regexp.MustCompile(`AV: (\d+):(\d+):(\d+) \/ (\d+):(\d+):(\d+)`) ) type playerModel struct { entry *feedEntry isPlayingVideo bool // video information videoDuration time.Duration videoPosition time.Duration // TODO: make this smarter // finished entries to send messages about finishedEntries []int64 sync.RWMutex } func (m *playerModel) Init() tea.Cmd { return nil } func (m *playerModel) playEntry(f feedEntry) tea.Cmd { if m.isPlayingVideo == true { return nil } m.entry = &f if m.entry != nil && m.isPlayingVideo == false { go m.playVideo(context.Background(), m.entry.Link) } else { // TODO: we are already playing something, we should not allow this // return MsgTick(time.Now()) } return nil } func (m *playerModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) { switch msg := msg.(type) { case MsgPlayEntry: f := feedEntry(msg) return m, m.playEntry(f) } // send message because it is finished m.Lock() defer m.Unlock() if len(m.finishedEntries) > 0 { watchedEntryID := m.finishedEntries[0] m.finishedEntries = m.finishedEntries[1:] return m, func() tea.Msg { return MsgWatchedEntry(watchedEntryID) } } return m, nil } func (m *playerModel) View() string { if m.isPlayingVideo && m.entry != nil { timePos := time.Time{}.Add(m.videoPosition) timeDur := time.Time{}.Add(m.videoDuration) // truncate name if needed name := m.entry.Name if len(name) > playerMaxTitleLength { name = name[0:playerMaxTitleLength] + "..." } return fmt.Sprintf("Playing: %s (%s/%s %.1f%%)", name, timePos.Format(time.TimeOnly), timeDur.Format(time.TimeOnly), videoPercentageWatched(m.videoPosition, m.videoDuration), ) } return "" } func (m *playerModel) finishedVideo() { m.Lock() defer m.Unlock() m.isPlayingVideo = false if videoPercentageWatched(m.videoPosition, m.videoDuration) > 90.0 { m.finishedEntries = append(m.finishedEntries, m.entry.ID) } } func (m *playerModel) playVideo(ctx context.Context, url string) (bool, error) { m.Lock() m.isPlayingVideo = true m.Unlock() defer m.finishedVideo() cmd := exec.CommandContext(ctx, "mpv", url) output, err := cmd.StdoutPipe() if err != nil { return false, err } err = cmd.Start() if err != nil { return false, err } s := bufio.NewScanner(output) go func(s *bufio.Scanner) { var finishedVideo bool for s.Scan() { t := s.Text() if strings.Contains(t, "End of file") { finishedVideo = true } else if strings.HasPrefix(t, "AV: ") { matches := regexpPosDur.FindStringSubmatch(t) posh, _ := strconv.Atoi(matches[1]) posm, _ := strconv.Atoi(matches[2]) poss, _ := strconv.Atoi(matches[3]) durh, _ := strconv.Atoi(matches[4]) durm, _ := strconv.Atoi(matches[5]) durs, _ := strconv.Atoi(matches[6]) m.videoPosition = time.Duration((posh*60*60 + posm*60 + poss) * int(time.Second)) m.videoDuration = time.Duration((durh*60*60 + durm*60 + durs) * int(time.Second)) } } _ = finishedVideo }(s) return false, cmd.Wait() }