AynaLivePlayer/controller/controller.go

package controller

import (
	"AynaLivePlayer/config"
	"AynaLivePlayer/event"
	"AynaLivePlayer/liveclient"
	"AynaLivePlayer/logger"
	"AynaLivePlayer/player"
	"AynaLivePlayer/provider"
	"AynaLivePlayer/util"
	"fmt"
	"github.com/sirupsen/logrus"
	"strconv"
)

const MODULE_CONTROLLER = "Controller"

func l() *logrus.Entry {
	return logger.Logger.WithField("Module", MODULE_CONTROLLER)
}

func SetDanmuClient(roomId string) {
	ResetDanmuClient()
	l().Infof("setting live client for %s", roomId)
	room, err := strconv.Atoi(roomId)
	if err != nil {
		l().Warn("parse room id error", err)
		return
	}
	if !util.StringSliceContains(config.LiveRoom.History, roomId) {
		config.LiveRoom.History = append(config.LiveRoom.History, roomId)
	}
	LiveClient = liveclient.NewBilibili(room)
	LiveClient.Handler().Register(&event.EventHandler{
		EventId: liveclient.EventMessageReceive,
		Name:    "controller.commandexecutor",
		Handler: danmuCommandHandler,
	})
	LiveClient.Handler().RegisterA(
		liveclient.EventMessageReceive,
		"controller.danmu.handler",
		danmuHandler)
	l().Infof("setting live client for %s success", roomId)
}

func StartDanmuClient() {
	LiveClient.Connect()
}

func ResetDanmuClient() {
	if LiveClient != nil {
		l().Infof("disconnect from current live client %s", LiveClient.ClientName())
		LiveClient.Disconnect()
		LiveClient.Handler().UnregisterAll()
		LiveClient = nil
	}
}

func AddPlaylist(pname string, uri string) *player.Playlist {
	l().Infof("try add playlist %s with provider %s", uri, pname)
	id, err := provider.FormatPlaylistUrl(pname, uri)
	if err != nil || id == "" {
		l().Warnf("fail to format %s playlist id for %s", uri, pname)
		return nil
	}
	p := player.NewPlaylist(fmt.Sprintf("%s-%s", pname, id), player.PlaylistConfig{})
	p.Meta = provider.Meta{
		Name: pname,
		Id:   id,
	}
	PlaylistManager = append(PlaylistManager, p)
	config.Player.Playlists = append(config.Player.Playlists, id)
	config.Player.PlaylistsProvider = append(config.Player.PlaylistsProvider, pname)
	return p
}

func RemovePlaylist(index int) {
	l().Infof("Try to remove playlist.index=%d", index)
	if index < 0 || index >= len(PlaylistManager) {
		l().Warnf("playlist.index=%d not found", index)
		return
	}
	if index == config.Player.PlaylistIndex {
		l().Info("Delete current system playlist, reset system playlist to index = 0")
		SetSystemPlaylist(0)
	}
	if index < config.Player.PlaylistIndex {
		l().Debugf("Delete playlist before system playlist (index=%d), reduce system playlist index by 1", config.Player.PlaylistIndex)
		config.Player.PlaylistIndex = config.Player.PlaylistIndex - 1
	}
	PlaylistManager = append(PlaylistManager[:index], PlaylistManager[index+1:]...)
	config.Player.Playlists = append(config.Player.Playlists[:index], config.Player.Playlists[index+1:]...)
	config.Player.PlaylistsProvider = append(config.Player.PlaylistsProvider[:index], config.Player.PlaylistsProvider[index+1:]...)
}

func SetSystemPlaylist(index int) {
	l().Infof("try set system playlist to playlist.id=%d", index)
	if index < 0 || index >= len(PlaylistManager) {
		l().Warn("playlist.index=%d not found", index)
		return
	}
	err := PreparePlaylist(PlaylistManager[index])
	if err != nil {
		return
	}
	medias := PlaylistManager[index].Playlist
	config.Player.PlaylistIndex = index
	ApplyUser(medias, player.PlaylistUser)
	SystemPlaylist.Replace(medias)
}

func PreparePlaylistByIndex(index int) {
	l().Infof("try prepare playlist.id=%d", index)
	if index < 0 || index >= len(PlaylistManager) {
		l().Warn("playlist.id=%d not found", index)
		return
	}
	err := PreparePlaylist(PlaylistManager[index])
	if err != nil {
		return
	}
}