package consts

import (
	"github.com/labstack/echo/v4"
)

type contextKey string

const ContextKeyEdition contextKey = "edition"

type LicenseEdition int32

const (
	LicenseEditionFree       LicenseEdition = 0 // 开源版
	LicenseEditionProfession LicenseEdition = 1 // 专业版
	LicenseEditionEnterprise LicenseEdition = 2 // 企业版
	LicenseEditionBusiness   LicenseEdition = 3 // 商业版
)

func GetLicenseEdition(c echo.Context) LicenseEdition {
	edition, ok := c.Get("edition").(LicenseEdition)
	if !ok {
		return LicenseEditionFree
	}
	return edition
}