init

app/http/helpers/admin.go

@@ -0,0 +1,40 @@
+package helpers
+
+import (
+	"errors"
+
+	"github.com/goravel/framework/contracts/http"
+
+	"goravel/app/models"
+)
+
+// GetAdminFromContext 从 context 中获取 admin 对象
+// 如果 context 中没有 admin 或类型不匹配，返回错误
+func GetAdminFromContext(ctx http.Context) (*models.Admin, error) {
+	adminValue := ctx.Value("admin")
+	if adminValue == nil {
+		return nil, errors.New("admin not found in context")
+	}
+
+	// 尝试值类型
+	if admin, ok := adminValue.(models.Admin); ok {
+		return &admin, nil
+	}
+
+	// 尝试指针类型
+	if adminPtr, ok := adminValue.(*models.Admin); ok {
+		return adminPtr, nil
+	}
+
+	return nil, errors.New("invalid admin type in context")
+}
+
+// GetAdminIDFromContext 从 context 中获取 admin ID
+// 如果 context 中没有 admin 或类型不匹配，返回 0 和错误
+func GetAdminIDFromContext(ctx http.Context) (uint, error) {
+	admin, err := GetAdminFromContext(ctx)
+	if err != nil {
+		return 0, err
+	}
+	return admin.ID, nil
+}