feat: 初始化 ChatGPT Team 管理后端项目

- 添加用户认证模块 (JWT + 密码管理)
- 添加 ChatGPT 账户管理功能
- 添加卡密管理功能 (创建、批量生成、查询)
- 添加邀请功能
- 配置数据库迁移和路由系统
This commit is contained in:
sar
2026-01-13 14:42:56 +08:00
commit 42c423bd32
29 changed files with 2969 additions and 0 deletions

View File

@@ -0,0 +1,83 @@
package repository
import (
"database/sql"
"time"
"gpt-manager-go/internal/models"
)
// InvitationRepository 邀请记录仓储
type InvitationRepository struct {
db *sql.DB
}
// NewInvitationRepository 创建仓储
func NewInvitationRepository(db *sql.DB) *InvitationRepository {
return &InvitationRepository{db: db}
}
// Create 创建邀请记录
func (r *InvitationRepository) Create(invitation *models.Invitation) error {
return r.db.QueryRow(`
INSERT INTO invitations (card_key_id, account_id, invited_email, status, error_message, expires_at, created_at)
VALUES ($1, $2, $3, $4, $5, $6, $7)
RETURNING id
`,
invitation.CardKeyID,
invitation.AccountID,
invitation.InvitedEmail,
invitation.Status,
invitation.ErrorMessage,
invitation.ExpiresAt,
time.Now(),
).Scan(&invitation.ID)
}
// FindByEmail 根据邮箱查找邀请记录
func (r *InvitationRepository) FindByEmail(email string) ([]*models.Invitation, error) {
rows, err := r.db.Query(`
SELECT id, card_key_id, account_id, invited_email, status, error_message, expires_at, created_at, updated_at
FROM invitations WHERE invited_email = $1 ORDER BY created_at DESC
`, email)
if err != nil {
return nil, err
}
defer rows.Close()
var invitations []*models.Invitation
for rows.Next() {
inv := &models.Invitation{}
if err := rows.Scan(
&inv.ID, &inv.CardKeyID, &inv.AccountID, &inv.InvitedEmail,
&inv.Status, &inv.ErrorMessage, &inv.ExpiresAt,
&inv.CreatedAt, &inv.UpdatedAt,
); err != nil {
return nil, err
}
invitations = append(invitations, inv)
}
return invitations, nil
}
// UpdateStatus 更新邀请状态
func (r *InvitationRepository) UpdateStatus(id int, status models.InvitationStatus, errMsg string) error {
var errMsgSQL sql.NullString
if errMsg != "" {
errMsgSQL = sql.NullString{String: errMsg, Valid: true}
}
_, err := r.db.Exec(`
UPDATE invitations SET status = $1, error_message = $2, updated_at = $3
WHERE id = $4
`, status, errMsgSQL, time.Now(), id)
return err
}
// DeleteByEmailAndAccountID 根据邮箱和账号ID删除邀请记录
func (r *InvitationRepository) DeleteByEmailAndAccountID(email string, accountID int) error {
_, err := r.db.Exec(`
DELETE FROM invitations WHERE invited_email = $1 AND account_id = $2
`, email, accountID)
return err
}