dorisbot/route/private/private.go

40 lines
758 B
Go

package private
import (
pubmodel "dorisbot/models/public"
"dorisbot/pkg/config"
"dorisbot/pkg/context"
"net/http"
"net/textproto"
)
// VerifyKey -
func VerifyKey(c *context.Context) {
conf := config.GetConfig()
key := c.GetHeader(textproto.CanonicalMIMEHeaderKey("x-mtfos-key"))
if len(key) == 0 || key != conf.SelfKey {
c.CustomRes(http.StatusForbidden, map[string]string{
"message": "key empty or not match",
})
return
}
c.Next()
}
// GetFacebookIDs -
func GetFacebookIDs(c *context.Context) {
pages, err := pubmodel.GetAllFacebookPage()
if err != nil {
c.ServerError()
return
}
ids := make([]string, 0, len(pages))
for _, v := range pages {
ids = append(ids, v.ID)
}
c.Success(map[string]interface{}{"list": ids})
}