sdd
This commit is contained in:
huangzhenpc
64
cmd/create_user/main.go
Normal file
64
cmd/create_user/main.go
Normal file
@@ -0,0 +1,64 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"stalwart_client/pkg/api"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 解析命令行参数
|
||||
emailPtr := flag.String("e", "", "邮箱地址")
|
||||
passwordPtr := flag.String("p", "", "用户密码")
|
||||
descriptionPtr := flag.String("d", "", "用户描述")
|
||||
quotaPtr := flag.Int64("q", 0, "邮箱配额（字节）")
|
||||
authTypePtr := flag.String("a", "basic", "认证方式: basic(基本认证)或apikey(API Key认证)")
|
||||
authUserPtr := flag.String("u", "", "基本认证用户名")
|
||||
authPassPtr := flag.String("P", "", "基本认证密码")
|
||||
apiKeyPtr := flag.String("k", "", "API Key")
|
||||
configPtr := flag.String("c", "", "配置文件路径")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
// 加载配置
|
||||
config, err := api.LoadConfig(*configPtr)
|
||||
if err != nil {
|
||||
fmt.Printf("加载配置失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 创建客户端
|
||||
client := api.NewClient(config)
|
||||
|
||||
// 确定认证类型
|
||||
var authType api.AuthType
|
||||
if *authTypePtr == "apikey" {
|
||||
authType = api.APIKeyAuth
|
||||
} else {
|
||||
authType = api.BasicAuth
|
||||
}
|
||||
|
||||
// 创建用户
|
||||
userID, username, email, password, err := client.CreateEmailUser(
|
||||
*emailPtr,
|
||||
*passwordPtr,
|
||||
*descriptionPtr,
|
||||
*quotaPtr,
|
||||
authType,
|
||||
*authUserPtr,
|
||||
*authPassPtr,
|
||||
*apiKeyPtr,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("创建用户失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 打印结果
|
||||
success := api.PrintUserResult(userID, username, email, password)
|
||||
if !success {
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
68
cmd/get_latest_email/main.go
Normal file
68
cmd/get_latest_email/main.go
Normal file
@@ -0,0 +1,68 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"stalwart_client/pkg/api"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 解析命令行参数
|
||||
emailPtr := flag.String("e", "hayfzgul@evnmail.com", "邮箱地址")
|
||||
passwordPtr := flag.String("p", "", "邮箱密码")
|
||||
serverPtr := flag.String("s", "mail.evnmail.com", "IMAP服务器地址")
|
||||
portPtr := flag.Int("P", 993, "IMAP服务器端口")
|
||||
numPtr := flag.Int("n", 1, "获取的邮件数量")
|
||||
noSSLPtr := flag.Bool("no-ssl", false, "禁用SSL连接")
|
||||
configPtr := flag.String("c", "", "配置文件路径")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
// 加载配置
|
||||
_, err := api.LoadConfig(*configPtr)
|
||||
if err != nil {
|
||||
fmt.Printf("加载配置失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 检查密码
|
||||
password := *passwordPtr
|
||||
if password == "" {
|
||||
fmt.Println("错误: 必须提供邮箱密码")
|
||||
flag.Usage()
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 提取域名
|
||||
parts := strings.Split(*emailPtr, "@")
|
||||
domain := ""
|
||||
if len(parts) > 1 {
|
||||
domain = parts[1]
|
||||
}
|
||||
|
||||
// 如果未指定服务器，尝试使用邮箱域名
|
||||
server := *serverPtr
|
||||
if server == "" && domain != "" {
|
||||
server = fmt.Sprintf("mail.%s", domain)
|
||||
}
|
||||
|
||||
// 获取邮件
|
||||
emails, err := api.GetLatestEmails(
|
||||
*emailPtr,
|
||||
password,
|
||||
server,
|
||||
*portPtr,
|
||||
!*noSSLPtr,
|
||||
*numPtr,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("获取邮件失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 打印邮件信息
|
||||
api.PrintEmails(emails)
|
||||
}
|
||||
75
cmd/get_user/main.go
Normal file
75
cmd/get_user/main.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"stalwart_client/pkg/api"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 解析命令行参数
|
||||
idPtr := flag.String("i", "", "用户ID")
|
||||
emailPtr := flag.String("e", "", "用户邮箱地址")
|
||||
authTypePtr := flag.String("a", "basic", "认证方式: basic(基本认证)或apikey(API Key认证)")
|
||||
authUserPtr := flag.String("u", "", "基本认证用户名")
|
||||
authPassPtr := flag.String("P", "", "基本认证密码")
|
||||
apiKeyPtr := flag.String("k", "", "API Key")
|
||||
configPtr := flag.String("c", "", "配置文件路径")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
// 检查参数
|
||||
if *idPtr == "" && *emailPtr == "" {
|
||||
fmt.Println("错误: 必须指定用户ID或邮箱地址")
|
||||
flag.Usage()
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 加载配置
|
||||
config, err := api.LoadConfig(*configPtr)
|
||||
if err != nil {
|
||||
fmt.Printf("加载配置失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 创建客户端
|
||||
client := api.NewClient(config)
|
||||
|
||||
// 确定认证类型
|
||||
var authType api.AuthType
|
||||
if *authTypePtr == "apikey" {
|
||||
authType = api.APIKeyAuth
|
||||
} else {
|
||||
authType = api.BasicAuth
|
||||
}
|
||||
|
||||
// 获取用户信息
|
||||
var userInfo map[string]interface{}
|
||||
var err2 error
|
||||
|
||||
if *idPtr != "" {
|
||||
// 通过ID查询
|
||||
id, err := strconv.Atoi(*idPtr)
|
||||
if err != nil {
|
||||
fmt.Printf("无效的用户ID: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fmt.Printf("通过ID查询用户: %d\n", id)
|
||||
userInfo, err2 = client.GetUserDetails(id, authType, *authUserPtr, *authPassPtr, *apiKeyPtr)
|
||||
} else {
|
||||
// 通过邮箱查询
|
||||
fmt.Printf("通过邮箱查询用户: %s\n", *emailPtr)
|
||||
userInfo, err2 = client.GetUserByEmail(*emailPtr, authType, *authUserPtr, *authPassPtr, *apiKeyPtr)
|
||||
}
|
||||
|
||||
if err2 != nil {
|
||||
fmt.Printf("获取用户信息失败: %v\n", err2)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 格式化显示结果
|
||||
api.FormatUserDetails(userInfo)
|
||||
}
|
||||
58
cmd/list_users/main.go
Normal file
58
cmd/list_users/main.go
Normal file
@@ -0,0 +1,58 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"stalwart_client/pkg/api"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 解析命令行参数
|
||||
limitPtr := flag.Int("l", 100, "最大返回结果数量")
|
||||
pagePtr := flag.Int("p", 1, "页码")
|
||||
_ = flag.String("q", "", "搜索关键词") // 暂未实现搜索功能，保留参数
|
||||
authTypePtr := flag.String("a", "basic", "认证方式: basic(基本认证)或apikey(API Key认证)")
|
||||
authUserPtr := flag.String("u", "", "基本认证用户名")
|
||||
authPassPtr := flag.String("P", "", "基本认证密码")
|
||||
apiKeyPtr := flag.String("k", "", "API Key")
|
||||
configPtr := flag.String("c", "", "配置文件路径")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
// 加载配置
|
||||
config, err := api.LoadConfig(*configPtr)
|
||||
if err != nil {
|
||||
fmt.Printf("加载配置失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 创建客户端
|
||||
client := api.NewClient(config)
|
||||
|
||||
// 确定认证类型
|
||||
var authType api.AuthType
|
||||
if *authTypePtr == "apikey" {
|
||||
authType = api.APIKeyAuth
|
||||
} else {
|
||||
authType = api.BasicAuth
|
||||
}
|
||||
|
||||
// 获取用户列表
|
||||
userList, err := client.ListUsers(
|
||||
authType,
|
||||
*authUserPtr,
|
||||
*authPassPtr,
|
||||
*apiKeyPtr,
|
||||
*limitPtr,
|
||||
*pagePtr,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("获取用户列表失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 格式化显示结果
|
||||
api.FormatUserList(userList)
|
||||
}
|
||||
530
cmd/web_server/main.go
Normal file
530
cmd/web_server/main.go
Normal file
@@ -0,0 +1,530 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"os"
|
||||
"stalwart_client/pkg/api"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// 请求和响应结构
|
||||
type CreateUserRequest struct {
|
||||
Email string `json:"email"`
|
||||
Password string `json:"password,omitempty"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Quota int64 `json:"quota,omitempty"`
|
||||
}
|
||||
|
||||
type UserResponse struct {
|
||||
ID int `json:"id"`
|
||||
Username string `json:"username"`
|
||||
Email string `json:"email"`
|
||||
Password string `json:"password,omitempty"`
|
||||
Success bool `json:"success"`
|
||||
Message string `json:"message,omitempty"`
|
||||
}
|
||||
|
||||
type ListUsersRequest struct {
|
||||
Limit int `json:"limit,omitempty"`
|
||||
Page int `json:"page,omitempty"`
|
||||
}
|
||||
|
||||
type ListUsersResponse struct {
|
||||
Users []api.User `json:"users"`
|
||||
Total int `json:"total"`
|
||||
Success bool `json:"success"`
|
||||
Message string `json:"message,omitempty"`
|
||||
}
|
||||
|
||||
type GetUserRequest struct {
|
||||
ID int `json:"id,omitempty"`
|
||||
Email string `json:"email,omitempty"`
|
||||
}
|
||||
|
||||
type GetUserResponse struct {
|
||||
User map[string]interface{} `json:"user"`
|
||||
Success bool `json:"success"`
|
||||
Message string `json:"message,omitempty"`
|
||||
}
|
||||
|
||||
type GetEmailRequest struct {
|
||||
Email string `json:"email"`
|
||||
Password string `json:"password"`
|
||||
Server string `json:"server,omitempty"`
|
||||
Port int `json:"port,omitempty"`
|
||||
Count int `json:"count,omitempty"`
|
||||
}
|
||||
|
||||
type EmailResponse struct {
|
||||
Emails []*api.EmailMessage `json:"emails"`
|
||||
Success bool `json:"success"`
|
||||
Message string `json:"message,omitempty"`
|
||||
}
|
||||
|
||||
// 全局变量
|
||||
var (
|
||||
client *api.Client
|
||||
)
|
||||
|
||||
// API处理函数
|
||||
|
||||
// 创建用户
|
||||
func createUserHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var req CreateUserRequest
|
||||
// 尝试解析请求体，但即使失败也继续处理
|
||||
_ = json.NewDecoder(r.Body).Decode(&req)
|
||||
|
||||
// 如果email为空，自动生成一个随机用户名和域名
|
||||
if req.Email == "" {
|
||||
randomUsername := api.GenerateRandomPassword(8)
|
||||
req.Email = fmt.Sprintf("%s@%s", randomUsername, client.Config.User.DefaultDomain)
|
||||
}
|
||||
|
||||
// 如果密码为空，将由系统自动生成
|
||||
// 如果配额为0，使用默认配额
|
||||
if req.Quota == 0 {
|
||||
req.Quota = client.Config.User.DefaultQuota
|
||||
}
|
||||
|
||||
// 如果描述为空，添加默认描述
|
||||
if req.Description == "" {
|
||||
req.Description = "系统自动创建的用户"
|
||||
}
|
||||
|
||||
// 创建用户
|
||||
userID, username, email, password, err := client.CreateEmailUser(
|
||||
req.Email,
|
||||
req.Password,
|
||||
req.Description,
|
||||
req.Quota,
|
||||
api.APIKeyAuth, // 默认使用API Key认证
|
||||
"",
|
||||
"",
|
||||
client.Config.Auth.APIKey,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": fmt.Sprintf("创建用户失败: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 返回响应
|
||||
response := UserResponse{
|
||||
ID: userID,
|
||||
Username: username,
|
||||
Email: email,
|
||||
Password: password,
|
||||
Success: true,
|
||||
Message: "用户创建成功",
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(response)
|
||||
}
|
||||
|
||||
// 创建简单用户 - 无需任何参数，直接返回创建的用户信息
|
||||
func createSimpleUserHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet && r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
// 生成随机用户名 - 确保不包含特殊字符
|
||||
randomUsername := api.GenerateRandomUsername(8)
|
||||
// 确保用户名只包含字母和数字
|
||||
for i := 0; i < len(randomUsername); i++ {
|
||||
if !((randomUsername[i] >= 'a' && randomUsername[i] <= 'z') ||
|
||||
(randomUsername[i] >= '0' && randomUsername[i] <= '9')) {
|
||||
// 替换为字母
|
||||
randomUsername = randomUsername[:i] + "x" + randomUsername[i+1:]
|
||||
}
|
||||
}
|
||||
|
||||
email := fmt.Sprintf("%s@%s", randomUsername, client.Config.User.DefaultDomain)
|
||||
description := "自动创建的临时用户"
|
||||
quota := client.Config.User.DefaultQuota
|
||||
|
||||
// 创建用户
|
||||
userID, username, email, password, err := client.CreateEmailUser(
|
||||
email,
|
||||
"", // 空密码，系统会自动生成
|
||||
description,
|
||||
quota,
|
||||
api.APIKeyAuth,
|
||||
"",
|
||||
"",
|
||||
client.Config.Auth.APIKey,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": fmt.Sprintf("创建用户失败: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 返回简单格式的响应
|
||||
response := map[string]interface{}{
|
||||
"id": userID,
|
||||
"username": username,
|
||||
"email": email,
|
||||
"password": password,
|
||||
"server": fmt.Sprintf("mail.%s", strings.Split(email, "@")[1]),
|
||||
"success": true,
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(response)
|
||||
}
|
||||
|
||||
// 创建模拟用户 - 不实际调用API，只生成模拟数据
|
||||
func createMockUserHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet && r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
// 生成随机用户名 - 确保不包含特殊字符
|
||||
randomUsername := api.GenerateRandomUsername(8)
|
||||
// 确保用户名只包含字母和数字
|
||||
for i := 0; i < len(randomUsername); i++ {
|
||||
if !((randomUsername[i] >= 'a' && randomUsername[i] <= 'z') ||
|
||||
(randomUsername[i] >= '0' && randomUsername[i] <= '9')) {
|
||||
// 替换为字母
|
||||
randomUsername = randomUsername[:i] + "x" + randomUsername[i+1:]
|
||||
}
|
||||
}
|
||||
|
||||
domain := client.Config.User.DefaultDomain
|
||||
email := fmt.Sprintf("%s@%s", randomUsername, domain)
|
||||
password := api.GenerateRandomPassword(12)
|
||||
|
||||
// 返回模拟响应
|
||||
userID := rand.Intn(1000) + 1000 // 随机生成ID
|
||||
|
||||
response := map[string]interface{}{
|
||||
"id": userID,
|
||||
"username": randomUsername,
|
||||
"email": email,
|
||||
"password": password,
|
||||
"server": fmt.Sprintf("mail.%s", domain),
|
||||
"imap_server": fmt.Sprintf("mail.%s", domain),
|
||||
"imap_port": 993,
|
||||
"smtp_server": fmt.Sprintf("mail.%s", domain),
|
||||
"smtp_port": 465,
|
||||
"description": "模拟生成的测试用户",
|
||||
"quota": client.Config.User.DefaultQuota,
|
||||
"created_time": time.Now().Format(time.RFC3339),
|
||||
"success": true,
|
||||
"message": "模拟用户创建成功（仅本地数据，未实际调用API）",
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(response)
|
||||
}
|
||||
|
||||
// 列出用户
|
||||
func listUsersHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
// 解析查询参数
|
||||
limitStr := r.URL.Query().Get("limit")
|
||||
pageStr := r.URL.Query().Get("page")
|
||||
|
||||
limit := 100
|
||||
page := 1
|
||||
|
||||
if limitStr != "" {
|
||||
if l, err := strconv.Atoi(limitStr); err == nil && l > 0 {
|
||||
limit = l
|
||||
}
|
||||
}
|
||||
|
||||
if pageStr != "" {
|
||||
if p, err := strconv.Atoi(pageStr); err == nil && p > 0 {
|
||||
page = p
|
||||
}
|
||||
}
|
||||
|
||||
// 获取用户列表
|
||||
userList, err := client.ListUsers(
|
||||
api.APIKeyAuth, // 默认使用API Key认证
|
||||
"",
|
||||
"",
|
||||
client.Config.Auth.APIKey,
|
||||
limit,
|
||||
page,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": fmt.Sprintf("获取用户列表失败: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 返回响应
|
||||
response := ListUsersResponse{
|
||||
Users: userList.Data.Items,
|
||||
Total: userList.Data.Total,
|
||||
Success: true,
|
||||
Message: "获取用户列表成功",
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(response)
|
||||
}
|
||||
|
||||
// 获取单个用户信息
|
||||
func getUserHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
// 解析参数
|
||||
idStr := r.URL.Query().Get("id")
|
||||
email := r.URL.Query().Get("email")
|
||||
|
||||
// 如果两个参数都为空，尝试从路径中获取
|
||||
if idStr == "" && email == "" {
|
||||
pathParts := strings.Split(r.URL.Path, "/")
|
||||
if len(pathParts) > 3 {
|
||||
// 路径形式可能是 /api/get_user/123 或 /api/get_user/user@example.com
|
||||
lastPart := pathParts[len(pathParts)-1]
|
||||
if _, err := strconv.Atoi(lastPart); err == nil {
|
||||
// 是数字，当作ID处理
|
||||
idStr = lastPart
|
||||
} else if strings.Contains(lastPart, "@") {
|
||||
// 包含@，当作邮箱处理
|
||||
email = lastPart
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if idStr == "" && email == "" {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": "必须提供用户ID或邮箱地址",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
var userInfo map[string]interface{}
|
||||
var err error
|
||||
|
||||
if idStr != "" {
|
||||
// 通过ID查询
|
||||
id, err := strconv.Atoi(idStr)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": "无效的用户ID",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
userInfo, err = client.GetUserDetails(
|
||||
id,
|
||||
api.APIKeyAuth, // 默认使用API Key认证
|
||||
"",
|
||||
"",
|
||||
client.Config.Auth.APIKey,
|
||||
)
|
||||
} else {
|
||||
// 通过邮箱查询
|
||||
userInfo, err = client.GetUserByEmail(
|
||||
email,
|
||||
api.APIKeyAuth, // 默认使用API Key认证
|
||||
"",
|
||||
"",
|
||||
client.Config.Auth.APIKey,
|
||||
)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": fmt.Sprintf("获取用户信息失败: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 返回响应
|
||||
response := GetUserResponse{
|
||||
User: userInfo,
|
||||
Success: true,
|
||||
Message: "获取用户信息成功",
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(response)
|
||||
}
|
||||
|
||||
// 获取邮件
|
||||
func getEmailsHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var req GetEmailRequest
|
||||
// 尝试解析请求体，但即使失败也继续处理
|
||||
_ = json.NewDecoder(r.Body).Decode(&req)
|
||||
|
||||
// 检查是否提供了邮箱和密码参数
|
||||
if req.Email == "" || req.Password == "" {
|
||||
// 也可以从URL参数获取
|
||||
req.Email = r.URL.Query().Get("email")
|
||||
req.Password = r.URL.Query().Get("password")
|
||||
|
||||
// 如果仍然为空，则返回错误
|
||||
if req.Email == "" || req.Password == "" {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": "邮箱地址和密码必须提供",
|
||||
})
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// 设置默认参数
|
||||
server := req.Server
|
||||
if server == "" {
|
||||
// 尝试从邮箱地址提取域名
|
||||
parts := strings.Split(req.Email, "@")
|
||||
if len(parts) > 1 {
|
||||
server = fmt.Sprintf("mail.%s", parts[1])
|
||||
} else {
|
||||
server = "mail.evnmail.com"
|
||||
}
|
||||
}
|
||||
|
||||
port := req.Port
|
||||
if port == 0 {
|
||||
port = 993
|
||||
}
|
||||
|
||||
count := req.Count
|
||||
if count == 0 {
|
||||
count = 5 // 默认获取5封邮件
|
||||
}
|
||||
|
||||
// 获取邮件
|
||||
emails, err := api.GetLatestEmails(
|
||||
req.Email,
|
||||
req.Password,
|
||||
server,
|
||||
port,
|
||||
true, // 使用SSL
|
||||
count,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"success": "false",
|
||||
"message": fmt.Sprintf("获取邮件失败: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 返回响应
|
||||
response := EmailResponse{
|
||||
Emails: emails,
|
||||
Success: true,
|
||||
Message: fmt.Sprintf("成功获取 %d 封邮件", len(emails)),
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(response)
|
||||
}
|
||||
|
||||
// 启动Web服务器
|
||||
func main() {
|
||||
// 初始化随机数生成器
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
|
||||
// 解析命令行参数
|
||||
portPtr := flag.Int("p", 8080, "服务器端口")
|
||||
configPtr := flag.String("c", "", "配置文件路径")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
// 加载配置
|
||||
config, err := api.LoadConfig(*configPtr)
|
||||
if err != nil {
|
||||
fmt.Printf("加载配置失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 创建客户端
|
||||
client = api.NewClient(config)
|
||||
|
||||
// 设置路由 - 不再使用authMiddleware
|
||||
http.HandleFunc("/api/create_user", createUserHandler)
|
||||
http.HandleFunc("/api/simple_user", createSimpleUserHandler)
|
||||
http.HandleFunc("/api/mock_user", createMockUserHandler)
|
||||
http.HandleFunc("/api/list_users", listUsersHandler)
|
||||
http.HandleFunc("/api/get_user", getUserHandler)
|
||||
http.HandleFunc("/api/get_emails", getEmailsHandler)
|
||||
|
||||
// 添加一个简单的健康检查端点
|
||||
http.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("OK"))
|
||||
})
|
||||
|
||||
// 添加一个简单的根路径处理
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path != "/" {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("Stalwart 邮件 API 服务运行中"))
|
||||
})
|
||||
|
||||
// 启动服务器
|
||||
port := *portPtr
|
||||
addr := fmt.Sprintf(":%d", port)
|
||||
fmt.Printf("启动Web服务器，监听端口: %d\n", port)
|
||||
fmt.Printf("API端点 (无需认证):\n")
|
||||
fmt.Printf(" - POST /api/create_user (创建用户，可自定义参数)\n")
|
||||
fmt.Printf(" - GET /api/simple_user (创建随机用户，无需任何参数)\n")
|
||||
fmt.Printf(" - GET /api/mock_user (创建模拟用户，仅本地数据)\n")
|
||||
fmt.Printf(" - GET /api/list_users (获取用户列表)\n")
|
||||
fmt.Printf(" - GET /api/get_user (获取用户信息)\n")
|
||||
fmt.Printf(" - POST /api/get_emails (获取用户邮件)\n")
|
||||
|
||||
if err := http.ListenAndServe(addr, nil); err != nil {
|
||||
log.Fatalf("启动服务器失败: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user