package rtc

import (
	"crypto/hmac"
	"crypto/sha1"
	"encoding/base64"
	"encoding/json"
	"fmt"
	"net/http"
	"strings"
	"time"

	"github.com/qiniu/api.v7/v7/auth"
)

var (
	// RtcHost 为 Qiniu RTC Server API服务域名
	RtcHost = "rtc.qiniuapi.com"
)

// Manager 提供了 Qiniu RTC Server API 相关功能
type Manager struct {
	mac        *auth.Credentials
	httpClient *http.Client
}

// MergePublishRtmp  连麦合流转推 RTMP 的配置
// Enable: 布尔类型，用于开启和关闭所有房间的合流功能。
// AudioOnly: 布尔类型，可选，指定是否只合成音频。
// Height, Width: int，可选，指定合流输出的高和宽，默认为 640 x 480。
// OutputFps: int，可选，指定合流输出的帧率，默认为 25 fps 。
// OutputKbps: int，可选，指定合流输出的码率，默认为 1000 。
// URL: 合流后转推旁路直播的地址，可选，支持魔法变量配置按照连麦房间号生成不同的推流地址。如果是转推到七牛直播云，不建议使用该配置
// StreamTitle: 转推七牛直播云的流名，可选，支持魔法变量配置按照连麦房间号生成不同的流名。例如，配置 Hub 为 qn-zhibo ，配置 StreamTitle 为 $(roomName) ，则房间 meeting-001 的合流将会被转推到 rtmp://pili-publish.qn-zhibo.***.com/qn-zhibo/meeting-001地址。详细配置细则，请咨询七牛技术支持。
type MergePublishRtmp struct {
	Enable      bool   `json:"enable,omitempty"`
	AudioOnly   bool   `json:"audioOnly,omitempty"`
	Height      int    `json:"height,omitempty"`
	Width       int    `json:"width,omitempty"`
	OutputFps   int    `json:"fps,omitempty"`
	OutputKbps  int    `json:"kbps,omitempty"`
	URL         string `json:"url,omitempty"`
	StreamTitle string `json:"streamTitle,omitempty"`
}

// App 完整信息
// AppID: app 的唯一标识，创建的时候由系统生成。
type App struct {
	AppID string `json:"appId"`
	AppInitConf
	MergePublishRtmp MergePublishRtmp `json:"mergePublishRtmp,omitempty"`
	CreatedAt        time.Time        `json:"createdAt"`
	UpdatedAt        time.Time        `json:"updatedAt"`
}

// AppInitConf 创建 App 请求参数
// Title: app 的名称， 可选。
// Hub: 绑定的直播 hub，可选，用于合流后 rtmp 推流。
// MaxUsers: int 类型，可选，连麦房间支持的最大在线人数。
// NoAutoKickUser: bool 类型，可选，禁止自动踢人。
type AppInitConf struct {
	Hub            string `json:"hub,omitempty"`
	Title          string `json:"title,omitempty"`
	MaxUsers       int    `json:"maxUsers,omitempty"`
	NoAutoKickUser bool   `json:"noAutoKickUser,omitempty"`
}

// MergePublishRtmpInfo 连麦合流转推 RTMP 的配置更改信息
type MergePublishRtmpInfo struct {
	Enable      *bool   `json:"enable,omitempty"`
	AudioOnly   *bool   `json:"audioOnly,omitempty"`
	Height      *int    `json:"height,omitempty"`
	Width       *int    `json:"width,omitempty"`
	OutputFps   *int    `json:"fps,omitempty"`
	OutputKbps  *int    `json:"kbps,omitempty"`
	URL         *string `json:"url,omitempty"`
	StreamTitle *string `json:"streamTitle,omitempty"`
}

// AppUpdateInfo 更改信息
// MergePublishRtmpInfo 连麦合流转推 RTMP 的配置更改信息
type AppUpdateInfo struct {
	Hub              *string               `json:"hub,omitempty"`
	Title            *string               `json:"title,omitempty"`
	MaxUsers         *int                  `json:"maxUsers,omitempty"`
	NoAutoKickUser   *bool                 `json:"noAutoKickUser,omitempty"`
	MergePublishRtmp *MergePublishRtmpInfo `json:"mergePublishRtmp,omitempty"`
}

// User 连麦房间里的用户
type User struct {
	UserID string `json:"userId"`
}

// NewManager 用来构建一个新的 Manager
func NewManager(mac *auth.Credentials) *Manager {
	httpClient := http.DefaultClient
	return &Manager{mac: mac, httpClient: httpClient}
}

// CreateApp 新建实时音视频云
func (r *Manager) CreateApp(appReq AppInitConf) (App, error) {
	url := buildURL("/v3/apps")
	ret := App{}
	info := postReq(r.httpClient, r.mac, url, &appReq, &ret)
	return ret, info.Err
}

// GetApp 根据 appID 获取 实时音视频云 信息
func (r *Manager) GetApp(appID string) (App, error) {
	url := buildURL("/v3/apps/" + appID)
	ret := App{}
	info := getReq(r.httpClient, r.mac, url, &ret)
	return ret, info.Err
}

// DeleteApp 根据 appID 删除 实时音视频云
func (r *Manager) DeleteApp(appID string) error {
	url := buildURL("/v3/apps/" + appID)
	info := delReq(r.httpClient, r.mac, url, nil)
	return info.Err
}

// UpdateApp 根据 appID, App 更改实时音视频云 信息
func (r *Manager) UpdateApp(appID string, appInfo AppUpdateInfo) (App, error) {
	url := buildURL("/v3/apps/" + appID)
	ret := App{}
	info := postReq(r.httpClient, r.mac, url, &appInfo, &ret)
	return ret, info.Err
}

// ListUser 根据 appID, roomName 获取连麦房间里在线的用户
// appID: 连麦房间所属的 app 。
// roomName: 操作所查询的连麦房间。
func (r *Manager) ListUser(appID, roomName string) ([]User, error) {
	url := buildURL("/v3/apps/" + appID + "/rooms/" + roomName + "/users")
	users := struct {
		Users []User `json:"users"`
	}{}
	info := getReq(r.httpClient, r.mac, url, &users)
	return users.Users, info.Err
}

// KickUser 根据 appID, roomName, UserID 剔除在线的用户
// appID: 连麦房间所属的 app 。
// roomName: 连麦房间。
// userID: 操作所剔除的用户。
func (r *Manager) KickUser(appID, roomName, userID string) error {
	url := buildURL("/v3/apps/" + appID + "/rooms/" + roomName + "/users/" + userID)
	info := delReq(r.httpClient, r.mac, url, nil)
	return info.Err
}

// RoomQuery 房间查询响应结果
// IsEnd: bool 类型，分页查询是否已经查完所有房间。
// Offset: int 类型，下次分页查询使用的位移标记。
// Rooms: 当前活跃的房间名列表。
type RoomQuery struct {
	IsEnd  bool       `json:"end"`
	Offset int        `json:"offset"`
	Rooms  []RoomName `json:"rooms"`
}

// RoomName 房间名
type RoomName string

// ListActiveRooms 根据 appID, roomNamePrefix, offset, limit 查询当前活跃的房间
// appID: 连麦房间所属的 app 。
// roomNamePrefix: 所查询房间名的前缀索引，可以为空。
// offset: int 类型，分页查询的位移标记。
// limit: int 类型，此次查询的最大长度。
func (r *Manager) ListActiveRooms(appID, roomNamePrefix string, offset, limit int) (RoomQuery, error) {
	ret, _, err := r.doListActiveRoom(appID, roomNamePrefix, offset, limit)
	return ret, err
}

// ListAllActiveRooms 根据 appID, roomNamePrefix 查询当前活跃的房间
// appID: 连麦房间所属的 app 。
// roomNamePrefix: 所查询房间名的前缀索引，可以为空。
func (r *Manager) ListAllActiveRooms(appID, roomNamePrefix string) ([]RoomName, error) {
	ns := []RoomName{}
	var outErr error
	for offset := 0; ; {
		q, info, err := r.doListActiveRoom(appID, roomNamePrefix, offset, 100)
		if err != nil && info.Code != 401 {
			time.Sleep(500 * time.Millisecond)
			q, info, err = r.doListActiveRoom(appID, roomNamePrefix, offset, 100)
		}

		if err != nil || len(q.Rooms) == 0 {
			outErr = err
			break
		}
		offset = q.Offset
		ns = append(ns, q.Rooms...)
		if q.IsEnd {
			break
		}
	}
	return ns, outErr
}

func (r *Manager) doListActiveRoom(appID, roomNamePrefix string, offset, limit int) (RoomQuery, resInfo, error) {
	query := ""
	roomNamePrefix = strings.TrimSpace(roomNamePrefix)
	if len(roomNamePrefix) != 0 {
		query = "prefix=" + roomNamePrefix + "&"
	}
	query += fmt.Sprintf("offset=%v&limit=%v", offset, limit)
	url := buildURL("/v3/apps/" + appID + "/rooms?" + query)
	ret := RoomQuery{}
	info := getReq(r.httpClient, r.mac, url, &ret)
	return ret, *info, info.Err
}

// RoomAccess 房间管理凭证
// AppID: 房间所属帐号的 app 。
// RoomName: 房间名称，需满足规格 ^[a-zA-Z0-9_-]{3,64}$
// UserID: 请求加入房间的用户 ID，需满足规格 ^[a-zA-Z0-9_-]{3,50}$
// ExpireAt: int64 类型，鉴权的有效时间，传入以秒为单位的64位Unix绝对时间，token 将在该时间后失效。
// Permission: 该用户的房间管理权限，"admin" 或 "user"，默认为 "user" 。当权限角色为 "admin" 时，拥有将其他用户移除出房间等特权.
type RoomAccess struct {
	AppID      string `json:"appId"`
	RoomName   string `json:"roomName"`
	UserID     string `json:"userId"`
	ExpireAt   int64  `json:"expireAt"`
	Permission string `json:"permission"`
}

// GetRoomToken 生成房间管理鉴权，连麦用户终端通过房间管理鉴权获取七牛连麦服务。
func (r *Manager) GetRoomToken(roomAccess RoomAccess) (token string, err error) {
	roomAccessByte, err := json.Marshal(roomAccess)
	if err != nil {
		return
	}
	buf := make([]byte, base64.URLEncoding.EncodedLen(len(roomAccessByte)))
	base64.URLEncoding.Encode(buf, roomAccessByte)

	hmacsha1 := hmac.New(sha1.New, r.mac.SecretKey)
	hmacsha1.Write(buf)
	sign := hmacsha1.Sum(nil)

	encodedSign := base64.URLEncoding.EncodeToString(sign)
	token = r.mac.AccessKey + ":" + encodedSign + ":" + string(buf)
	return
}