add pre group one bot setting

This commit is contained in:
Jay
2019-06-25 17:02:03 +08:00
parent c605b2d9e9
commit 8c432b9b68
13 changed files with 185 additions and 56 deletions
+12 -14
View File
@@ -13,7 +13,6 @@ import (
"strings"
"git.trj.tw/golang/mtfosbot/module/apis"
"git.trj.tw/golang/mtfosbot/module/config"
)
// TextMessage - line text message object
@@ -66,11 +65,10 @@ func getURL(p string) (string, bool) {
return str, true
}
func getHeaders() map[string]string {
func getHeaders(token string) map[string]string {
m := make(map[string]string)
conf := config.GetConf()
m["Content-Type"] = "application/json"
m["Authorization"] = fmt.Sprintf("Bearer %s", conf.Line.Access)
m["Authorization"] = fmt.Sprintf("Bearer %s", token)
return m
}
@@ -104,7 +102,7 @@ func checkMessageObject(m interface{}) interface{} {
}
// PushMessage -
func PushMessage(target string, message ...interface{}) {
func PushMessage(accessToken, target string, message ...interface{}) {
log.Println("push target :::: ", target)
if len(target) == 0 || len(message) == 0 {
return
@@ -145,7 +143,7 @@ func PushMessage(target string, message ...interface{}) {
reqObj := apis.RequestObj{
Method: "POST",
URL: apiURL,
Headers: getHeaders(),
Headers: getHeaders(accessToken),
Body: byteReader,
}
@@ -163,7 +161,7 @@ func PushMessage(target string, message ...interface{}) {
}
// ReplyMessage -
func ReplyMessage(replyToken string, message ...interface{}) {
func ReplyMessage(accessToken, replyToken string, message ...interface{}) {
if len(replyToken) == 0 || len(message) == 0 {
return
}
@@ -201,7 +199,7 @@ func ReplyMessage(replyToken string, message ...interface{}) {
reqObj := apis.RequestObj{
Method: "POST",
URL: apiURL,
Headers: getHeaders(),
Headers: getHeaders(accessToken),
Body: byteReader,
}
@@ -217,9 +215,9 @@ func ReplyMessage(replyToken string, message ...interface{}) {
}
// GetUserInfo -
func GetUserInfo(u, g string) (user *LineUserInfo, err error) {
func GetUserInfo(accessToken, u, g string) (user *LineUserInfo, err error) {
urlPath := fmt.Sprintf("/v2/bot/group/%s/member/%s", g, u)
header := getHeaders()
header := getHeaders(accessToken)
apiURL, ok := getURL(urlPath)
if !ok {
return nil, errors.New("url parser fail")
@@ -263,9 +261,9 @@ func GetUserInfo(u, g string) (user *LineUserInfo, err error) {
}
// GetContentHead -
func GetContentHead(id string) (mime string, err error) {
func GetContentHead(accessToken, id string) (mime string, err error) {
urlPath := fmt.Sprintf("/v2/bot/message/%s/content", id)
header := getHeaders()
header := getHeaders(accessToken)
u, ok := getURL(urlPath)
if !ok {
return "", errors.New("get url fail")
@@ -294,9 +292,9 @@ func GetContentHead(id string) (mime string, err error) {
}
// DownloadContent -
func DownloadContent(id string, w io.Writer) (err error) {
func DownloadContent(accessToken, id string, w io.Writer) (err error) {
urlPath := fmt.Sprintf("/v2/bot/message/%s/content", id)
header := getHeaders()
header := getHeaders(accessToken)
u, ok := getURL(urlPath)
if !ok {
return errors.New("get url fail")