forked from lug/matterbridge
120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package slack
|
|
|
|
import (
|
|
"errors"
|
|
"net/url"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
DEFAULT_LOGINS_COUNT = 100
|
|
DEFAULT_LOGINS_PAGE = 1
|
|
)
|
|
|
|
type TeamResponse struct {
|
|
Team TeamInfo `json:"team"`
|
|
SlackResponse
|
|
}
|
|
|
|
type TeamInfo struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Domain string `json:"domain"`
|
|
EmailDomain string `json:"email_domain"`
|
|
Icon map[string]interface{} `json:"icon"`
|
|
}
|
|
|
|
type LoginResponse struct {
|
|
Logins []Login `json:"logins"`
|
|
Paging `json:"paging"`
|
|
SlackResponse
|
|
}
|
|
|
|
|
|
type Login struct {
|
|
UserID string `json:"user_id"`
|
|
Username string `json:"username"`
|
|
DateFirst int `json:"date_first"`
|
|
DateLast int `json:"date_last"`
|
|
Count int `json:"count"`
|
|
IP string `json:"ip"`
|
|
UserAgent string `json:"user_agent"`
|
|
ISP string `json:"isp"`
|
|
Country string `json:"country"`
|
|
Region string `json:"region"`
|
|
}
|
|
|
|
// AccessLogParameters contains all the parameters necessary (including the optional ones) for a GetAccessLogs() request
|
|
type AccessLogParameters struct {
|
|
Count int
|
|
Page int
|
|
}
|
|
|
|
// NewAccessLogParameters provides an instance of AccessLogParameters with all the sane default values set
|
|
func NewAccessLogParameters() AccessLogParameters {
|
|
return AccessLogParameters{
|
|
Count: DEFAULT_LOGINS_COUNT,
|
|
Page: DEFAULT_LOGINS_PAGE,
|
|
}
|
|
}
|
|
|
|
|
|
func teamRequest(path string, values url.Values, debug bool) (*TeamResponse, error) {
|
|
response := &TeamResponse{}
|
|
err := post(path, values, response, debug)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !response.Ok {
|
|
return nil, errors.New(response.Error)
|
|
}
|
|
|
|
return response, nil
|
|
}
|
|
|
|
func accessLogsRequest(path string, values url.Values, debug bool) (*LoginResponse, error) {
|
|
response := &LoginResponse{}
|
|
err := post(path, values, response, debug)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !response.Ok {
|
|
return nil, errors.New(response.Error)
|
|
}
|
|
return response, nil
|
|
}
|
|
|
|
|
|
// GetTeamInfo gets the Team Information of the user
|
|
func (api *Client) GetTeamInfo() (*TeamInfo, error) {
|
|
values := url.Values{
|
|
"token": {api.config.token},
|
|
}
|
|
|
|
response, err := teamRequest("team.info", values, api.debug)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &response.Team, nil
|
|
}
|
|
|
|
// GetAccessLogs retrieves a page of logins according to the parameters given
|
|
func (api *Client) GetAccessLogs(params AccessLogParameters) ([]Login, *Paging, error) {
|
|
values := url.Values{
|
|
"token": {api.config.token},
|
|
}
|
|
if params.Count != DEFAULT_LOGINS_COUNT {
|
|
values.Add("count", strconv.Itoa(params.Count))
|
|
}
|
|
if params.Page != DEFAULT_LOGINS_PAGE {
|
|
values.Add("page", strconv.Itoa(params.Page))
|
|
}
|
|
response, err := accessLogsRequest("team.accessLogs", values, api.debug)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return response.Logins, &response.Paging, nil
|
|
}
|
|
|