47 lines
839 B
Go
47 lines
839 B
Go
|
package opsgenie
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
|
||
|
|
"github.com/opsgenie/opsgenie-go-sdk/client"
|
||
|
|
)
|
||
|
|
|
||
|
|
type OpsGenieClient struct {
|
||
|
|
apiKey string
|
||
|
|
|
||
|
|
StopContext context.Context
|
||
|
|
|
||
|
|
teams client.OpsGenieTeamClient
|
||
|
|
users client.OpsGenieUserClient
|
||
|
|
}
|
||
|
|
|
||
|
|
// Config defines the configuration options for the OpsGenie client
|
||
|
|
type Config struct {
|
||
|
|
ApiKey string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Client returns a new OpsGenie client
|
||
|
|
func (c *Config) Client() (*OpsGenieClient, error) {
|
||
|
|
opsGenie := new(client.OpsGenieClient)
|
||
|
|
opsGenie.SetAPIKey(c.ApiKey)
|
||
|
|
client := OpsGenieClient{}
|
||
|
|
|
||
|
|
log.Printf("[INFO] OpsGenie client configured")
|
||
|
|
|
||
|
|
teamsClient, err := opsGenie.Team()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
client.teams = *teamsClient
|
||
|
|
|
||
|
|
usersClient, err := opsGenie.User()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
client.users = *usersClient
|
||
|
|
|
||
|
|
return &client, nil
|
||
|
|
}
|