You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
3.4 KiB
126 lines
3.4 KiB
// Copyright © 2023 OpenIM. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package user
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
gettoken "github.com/openimsdk/open-im-server/v3/test/e2e/api/token"
|
|
"github.com/openimsdk/open-im-server/v3/test/e2e/framework/config"
|
|
)
|
|
|
|
// ForceLogoutRequest represents a request to force a user logout.
|
|
type ForceLogoutRequest struct {
|
|
PlatformID int `json:"platformID"`
|
|
UserID string `json:"userID"`
|
|
}
|
|
|
|
// CheckUserAccountRequest represents a request to check a user account.
|
|
type CheckUserAccountRequest struct {
|
|
CheckUserIDs []string `json:"checkUserIDs"`
|
|
}
|
|
|
|
// GetUsersRequest represents a request to get a list of users.
|
|
type GetUsersRequest struct {
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Pagination specifies the page number and number of items per page.
|
|
type Pagination struct {
|
|
PageNumber int `json:"pageNumber"`
|
|
ShowNumber int `json:"showNumber"`
|
|
}
|
|
|
|
// ForceLogout forces a user to log out.
|
|
func ForceLogout(token, userID string, platformID int) error {
|
|
|
|
url := fmt.Sprintf("http://%s:%s/auth/force_logout", config.LoadConfig().APIHost, config.LoadConfig().APIPort)
|
|
|
|
requestBody := ForceLogoutRequest{
|
|
PlatformID: platformID,
|
|
UserID: userID,
|
|
}
|
|
return sendPostRequestWithToken(url, token, requestBody)
|
|
}
|
|
|
|
// CheckUserAccount checks if the user accounts exist.
|
|
func CheckUserAccount(token string, userIDs []string) error {
|
|
|
|
url := fmt.Sprintf("http://%s:%s/user/account_check", config.LoadConfig().APIHost, config.LoadConfig().APIPort)
|
|
|
|
requestBody := CheckUserAccountRequest{
|
|
CheckUserIDs: userIDs,
|
|
}
|
|
return sendPostRequestWithToken(url, token, requestBody)
|
|
}
|
|
|
|
// GetUsers retrieves a list of users with pagination.
|
|
func GetUsers(token string, pageNumber, showNumber int) error {
|
|
|
|
url := fmt.Sprintf("http://%s:%s/user/account_check", config.LoadConfig().APIHost, config.LoadConfig().APIPort)
|
|
|
|
requestBody := GetUsersRequest{
|
|
Pagination: Pagination{
|
|
PageNumber: pageNumber,
|
|
ShowNumber: showNumber,
|
|
},
|
|
}
|
|
return sendPostRequestWithToken(url, token, requestBody)
|
|
}
|
|
|
|
// sendPostRequestWithToken sends a POST request with a token in the header.
|
|
func sendPostRequestWithToken(url, token string, body any) error {
|
|
reqBytes, err := json.Marshal(body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client := &http.Client{}
|
|
req, err := http.NewRequest("POST", url, bytes.NewBuffer(reqBytes))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req.Header.Add("Content-Type", "application/json")
|
|
req.Header.Add("operationID", gettoken.OperationID)
|
|
req.Header.Add("token", token)
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
respBody, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var respData map[string]any
|
|
if err := json.Unmarshal(respBody, &respData); err != nil {
|
|
return err
|
|
}
|
|
|
|
if errCode, ok := respData["errCode"].(float64); ok && errCode != 0 {
|
|
return fmt.Errorf("error in response: %v", respData)
|
|
}
|
|
|
|
return nil
|
|
}
|