mtfosbot/router/api/twitch.go
2018-09-18 23:54:36 +08:00

320 lines
5.6 KiB
Go

package api
import (
"fmt"
"time"
"git.trj.tw/golang/mtfosbot/model"
"git.trj.tw/golang/mtfosbot/module/context"
"git.trj.tw/golang/mtfosbot/module/twitch-irc"
"git.trj.tw/golang/mtfosbot/module/utils"
"github.com/gin-gonic/contrib/sessions"
)
func sessionTypeTwitch(id string) (ch []*model.TwitchChannel, err error) {
chdata, err := model.GetTwitchChannelWithID(id)
if err != nil {
return
}
if chdata == nil {
return nil, nil
}
list := make([]*model.TwitchChannel, 1)
list = append(list, chdata)
return list, nil
}
func sessionTypeSystem() (ch []*model.TwitchChannel, err error) {
ch, err = model.GetAllTwitchChannel()
return
}
// GetChannels - middleware
func GetChannels(c *context.Context) {
sess := sessions.Default(c.Context)
logingType := sess.Get("loginType")
if logingType == nil {
c.LoginFirst(nil)
return
}
ltype, ok := logingType.(string)
if !ok {
c.LoginFirst(nil)
return
}
if ltype == "twitch" {
u := sess.Get("user")
if u == nil {
c.LoginFirst(nil)
return
}
user, ok := u.(model.TwitchChannel)
if !ok {
c.LoginFirst(nil)
return
}
chs, err := sessionTypeTwitch(user.ID)
if err != nil || chs == nil {
c.ServerError(nil)
return
}
c.Set("channels", chs)
} else if ltype == "system" {
u := sess.Get("user")
if u == nil {
c.LoginFirst(nil)
return
}
_, ok := u.(model.Account)
if !ok {
c.LoginFirst(nil)
return
}
chs, err := sessionTypeSystem()
if err != nil || chs == nil {
c.LoginFirst(nil)
return
}
c.Set("channels", chs)
} else {
c.LoginFirst(nil)
return
}
c.Next()
}
func hasChannel(id string, c *context.Context) *model.TwitchChannel {
if len(id) == 0 {
return nil
}
channels, exists := c.Get("channels")
if !exists {
return nil
}
chs, ok := channels.([]*model.TwitchChannel)
if !ok {
return nil
}
for _, v := range chs {
if v.ID == id {
return v
}
}
return nil
}
// GetChannelList -
func GetChannelList(c *context.Context) {
channels, exists := c.Get("channels")
if !exists {
c.ServerError(nil)
return
}
list, ok := channels.([]*model.TwitchChannel)
if !ok {
c.ServerError(nil)
return
}
mapList := make([]map[string]interface{}, len(list))
for k, v := range list {
mapList[k] = utils.ToMap(v)
}
c.Success(map[string]interface{}{
"list": mapList,
})
}
// GetChannelData -
func GetChannelData(c *context.Context) {
chid := c.Param("chid")
chdata := hasChannel(chid, c)
if chdata == nil {
c.NotFound(nil)
return
}
c.Success(map[string]interface{}{
"channel": utils.ToMap(chdata),
})
}
// BotJoinChannel -
func BotJoinChannel(c *context.Context) {
chid := c.Param("chid")
chdata := hasChannel(chid, c)
if chdata == nil {
c.NotFound(nil)
return
}
bodyArg := struct {
Join int `json:"join"`
}{}
err := c.BindData(&bodyArg)
if err != nil {
c.DataFormat(nil)
return
}
if bodyArg.Join != 0 && bodyArg.Join != 1 {
c.DataFormat(nil)
return
}
err = chdata.UpdateJoin(bodyArg.Join == 1)
if err != nil {
c.ServerError(nil)
return
}
if bodyArg.Join == 1 {
twitchirc.JoinChannel(chdata.Name)
} else {
twitchirc.LeaveChannel(chdata.Name)
}
c.Success(nil)
}
// OpayIDChange -
func OpayIDChange(c *context.Context) {
chid := c.Param("chid")
chdata := hasChannel(chid, c)
if chdata == nil {
c.NotFound(nil)
return
}
bodyArg := struct {
Opay string `json:"opay" binding:"required"`
}{}
err := c.BindData(&bodyArg)
if err != nil {
c.DataFormat(nil)
return
}
err = chdata.UpdateOpayID(bodyArg.Opay)
if err != nil {
c.ServerError(nil)
return
}
c.Success(nil)
}
// GetDonateSetting -
func GetDonateSetting(c *context.Context) {
chid := c.Param("chid")
chdata := hasChannel(chid, c)
if chdata == nil {
c.NotFound(nil)
return
}
ds, err := model.GetDonateSettingByChannel(chdata.ID)
if err != nil {
fmt.Println(ds, err)
c.ServerError(nil)
return
}
var mapData map[string]interface{}
if ds != nil {
mapData = utils.ToMap(ds)
} else {
mapData = map[string]interface{}{}
}
c.Success(map[string]interface{}{
"setting": mapData,
})
}
// UpdateDonateSetting -
func UpdateDonateSetting(c *context.Context) {
chid := c.Param("chid")
chdata := hasChannel(chid, c)
if chdata == nil {
c.NotFound(nil)
return
}
bodyArg := struct {
End int64 `json:"end" binding:"exists"`
Title string `json:"title" binding:"required"`
Amount int `json:"amount" binding:"exists"`
StartAmount int `json:"start_amount"`
}{}
err := c.BindData(&bodyArg)
if err != nil {
c.DataFormat(nil)
return
}
if bodyArg.End > 10000000000-1 {
bodyArg.End = bodyArg.End / 1000
}
t := time.Unix(bodyArg.End, 0)
ds := &model.DonateSetting{
Title: bodyArg.Title,
EndDate: t,
StartDate: time.Now(),
StartAmount: bodyArg.StartAmount,
TargetAmount: bodyArg.Amount,
Twitch: chdata.ID,
}
err = ds.InsertOrUpdate()
if err != nil {
c.ServerError(nil)
return
}
c.Success(nil)
}
// GetDonateBarStatus -
func GetDonateBarStatus(c *context.Context) {
chid := c.Param("chid")
chdata, err := model.GetTwitchChannelWithID(chid)
if err != nil {
c.ServerError(nil)
return
}
if chdata == nil {
c.NotFound(nil)
return
}
ds, err := model.GetDonateSettingByChannel(chdata.ID)
if err != nil {
c.ServerError(nil)
return
}
sum := 0
mapData := map[string]interface{}{}
if ds != nil {
sum, err = model.SumChannelDonatePriceSinceTime(chdata.ID, ds.StartDate)
if err != nil {
c.ServerError(nil)
return
}
sum += ds.StartAmount
mapData = utils.ToMap(ds)
mapData["total"] = sum
}
c.Success(map[string]interface{}{
"setting": mapData,
})
}