mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-18 00:06:38 +00:00
ac135ab11d
Fixes panic occurring when body is nil (this usually happens when connections is refused) due to lack of nil check by centralizing response.Body.Close() behavior.
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package rest
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/netbirdio/netbird/management/server/http/api"
|
|
)
|
|
|
|
// UsersAPI APIs for users, do not use directly
|
|
type UsersAPI struct {
|
|
c *Client
|
|
}
|
|
|
|
// List list all users, only returns one user always
|
|
// See more: https://docs.netbird.io/api/resources/users#list-all-users
|
|
func (a *UsersAPI) List(ctx context.Context) ([]api.User, error) {
|
|
resp, err := a.c.newRequest(ctx, "GET", "/api/users", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.Body != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
ret, err := parseResponse[[]api.User](resp)
|
|
return ret, err
|
|
}
|
|
|
|
// Create create user
|
|
// See more: https://docs.netbird.io/api/resources/users#create-a-user
|
|
func (a *UsersAPI) Create(ctx context.Context, request api.PostApiUsersJSONRequestBody) (*api.User, error) {
|
|
requestBytes, err := json.Marshal(request)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp, err := a.c.newRequest(ctx, "POST", "/api/users", bytes.NewReader(requestBytes))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.Body != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
ret, err := parseResponse[api.User](resp)
|
|
return &ret, err
|
|
}
|
|
|
|
// Update update user settings
|
|
// See more: https://docs.netbird.io/api/resources/users#update-a-user
|
|
func (a *UsersAPI) Update(ctx context.Context, userID string, request api.PutApiUsersUserIdJSONRequestBody) (*api.User, error) {
|
|
requestBytes, err := json.Marshal(request)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp, err := a.c.newRequest(ctx, "PUT", "/api/users/"+userID, bytes.NewReader(requestBytes))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.Body != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
ret, err := parseResponse[api.User](resp)
|
|
return &ret, err
|
|
}
|
|
|
|
// Delete delete user
|
|
// See more: https://docs.netbird.io/api/resources/users#delete-a-user
|
|
func (a *UsersAPI) Delete(ctx context.Context, userID string) error {
|
|
resp, err := a.c.newRequest(ctx, "DELETE", "/api/users/"+userID, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if resp.Body != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ResendInvitation resend user invitation
|
|
// See more: https://docs.netbird.io/api/resources/users#resend-user-invitation
|
|
func (a *UsersAPI) ResendInvitation(ctx context.Context, userID string) error {
|
|
resp, err := a.c.newRequest(ctx, "POST", "/api/users/"+userID+"/invite", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if resp.Body != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Current gets the current user info
|
|
// See more: https://docs.netbird.io/api/resources/users#retrieve-current-user
|
|
func (a *UsersAPI) Current(ctx context.Context) (*api.User, error) {
|
|
resp, err := a.c.newRequest(ctx, "GET", "/api/users/current", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.Body != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
|
|
ret, err := parseResponse[api.User](resp)
|
|
return &ret, err
|
|
}
|