批量优化
This commit is contained in:
hero920103 committed
1 parent
e3db229f41
commit
aa92ca40a2
79 files changed
+7495
-1104
No files matched your search
@@ -0,0 +1,97 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/models"
|
||||
)
|
||||
|
||||
// =============================================================
|
||||
// 移动端免登录票据
|
||||
//
|
||||
// 背景：移动端用「密码直连」拿到的是 client_id=yz-uniapp、user_type=app 的令牌，
|
||||
// Web 端（管理后台）只认自己 localStorage 里 client_id=yz-backend 的令牌，
|
||||
// 两者登录态不共享，导致移动端打开 webview 时又要走一遍统一认证登录。
|
||||
//
|
||||
// 方案：移动端令牌 → 一次性票据 → Web 端令牌。
|
||||
// 票据本身是一次性的短时效凭证，即使泄露也无法二次使用。
|
||||
// =============================================================
|
||||
|
||||
// AppTicketTTL 票据有效期。移动端拿到后应立即跳转，不需要长时效。
|
||||
const AppTicketTTL = 60 * time.Second
|
||||
|
||||
var (
|
||||
ErrAppTicketInvalid = errors.New("登录票据无效")
|
||||
ErrAppTicketUsed = errors.New("登录票据已被使用")
|
||||
ErrAppTicketExpired = errors.New("登录票据已过期")
|
||||
)
|
||||
|
||||
// CreateAppTicket 签发一次性票据
|
||||
func CreateAppTicket(identityID, tid uint64, clientID string) (string, error) {
|
||||
if identityID == 0 {
|
||||
return "", errors.New("身份无效")
|
||||
}
|
||||
b := make([]byte, 32)
|
||||
if _, err := rand.Read(b); err != nil {
|
||||
return "", err
|
||||
}
|
||||
ticket := hex.EncodeToString(b)
|
||||
if clientID == "" {
|
||||
clientID = "yz-backend"
|
||||
}
|
||||
row := &models.AuthAppLoginTicket{
|
||||
Ticket: ticket,
|
||||
IdentityID: identityID,
|
||||
Tid: tid,
|
||||
ClientID: clientID,
|
||||
ExpireAt: time.Now().Add(AppTicketTTL),
|
||||
}
|
||||
if _, err := models.Orm.Insert(row); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return ticket, nil
|
||||
}
|
||||
|
||||
// ConsumeAppTicket 核销一次性票据（CAS 防重放），返回票据携带的身份与租户。
|
||||
func ConsumeAppTicket(ticket string) (*models.AuthAppLoginTicket, error) {
|
||||
ticket = strings.TrimSpace(ticket)
|
||||
if ticket == "" {
|
||||
return nil, ErrAppTicketInvalid
|
||||
}
|
||||
var row models.AuthAppLoginTicket
|
||||
if err := models.Orm.QueryTable(new(models.AuthAppLoginTicket)).
|
||||
Filter("ticket", ticket).One(&row); err != nil {
|
||||
return nil, ErrAppTicketInvalid
|
||||
}
|
||||
if row.UsedAt != nil {
|
||||
return nil, ErrAppTicketUsed
|
||||
}
|
||||
if row.ExpireAt.Before(time.Now()) {
|
||||
return nil, ErrAppTicketExpired
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
affected, err := models.Orm.QueryTable(new(models.AuthAppLoginTicket)).
|
||||
Filter("id", row.ID).
|
||||
Filter("used_at__isnull", true).
|
||||
Update(map[string]interface{}{"used_at": now})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if affected == 0 {
|
||||
return nil, ErrAppTicketUsed
|
||||
}
|
||||
row.UsedAt = &now
|
||||
return &row, nil
|
||||
}
|
||||
|
||||
// PurgeExpiredAppTickets 清理过期票据（防止表无限增长，由调用方定期触发）
|
||||
func PurgeExpiredAppTickets() (int64, error) {
|
||||
return models.Orm.QueryTable(new(models.AuthAppLoginTicket)).
|
||||
Filter("expire_at__lt", time.Now().Add(-24*time.Hour)).
|
||||
Delete()
|
||||
}
|
||||
@@ -70,7 +70,7 @@ func IssueTokens(opt TokenIssue) (*TokenPair, error) {
|
||||
userType := opt.UserType
|
||||
if userType == "" {
|
||||
userType = "backend"
|
||||
if opt.ClientID == "yz-uniapp" {
|
||||
if opt.ClientID == "yz-uniapp" || opt.ClientID == "yz-miniapp" {
|
||||
userType = "app"
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,154 @@
|
||||
// Package services 中本文件负责「展示端（platforms）」维度的读写与过滤。
|
||||
//
|
||||
// 背景：模块与菜单原先只有 views（[1]平台端 / [2]租户端）一个维度，
|
||||
// 无法表达「这个子系统/功能只在 Web 端出现，还是只在小程序出现，还是两端都出现」。
|
||||
// 这里新增 platforms 维度：
|
||||
// ["web"] —— 仅在 Web 端（租户管理后台）展示
|
||||
// ["app"] —— 仅在移动端（uniapp APP / 微信小程序）展示
|
||||
// ["web","app"] —— 两端都展示
|
||||
// 空串 / NULL —— 视为全端展示（保证历史数据行为不变）
|
||||
package services
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strings"
|
||||
|
||||
"server/models"
|
||||
)
|
||||
|
||||
// ParsePlatforms 解析展示端配置（JSON 数组字符串）。
|
||||
// 兼容非数组的旧值（如单个字符串），解析失败一律返回空切片。
|
||||
func ParsePlatforms(raw *string) []string {
|
||||
if raw == nil || strings.TrimSpace(*raw) == "" {
|
||||
return nil
|
||||
}
|
||||
var arr []string
|
||||
if err := json.Unmarshal([]byte(*raw), &arr); err == nil {
|
||||
out := make([]string, 0, len(arr))
|
||||
for _, s := range arr {
|
||||
s = strings.ToLower(strings.TrimSpace(s))
|
||||
if s != "" {
|
||||
out = append(out, s)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
// 非数组格式：按单个值处理
|
||||
v := strings.ToLower(strings.TrimSpace(*raw))
|
||||
if v == "" {
|
||||
return nil
|
||||
}
|
||||
return []string{v}
|
||||
}
|
||||
|
||||
// HasPlatform 判断是否在指定端展示。
|
||||
// 未配置（空）视为全端展示，保证历史数据与新建记录默认两端可见。
|
||||
func HasPlatform(list []string, platform string) bool {
|
||||
if len(list) == 0 {
|
||||
return true
|
||||
}
|
||||
p := strings.ToLower(strings.TrimSpace(platform))
|
||||
for _, s := range list {
|
||||
if strings.ToLower(strings.TrimSpace(s)) == p {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// NormalizePlatforms 规范化并去重展示端配置，返回待落库的 JSON 字符串。
|
||||
// 空输入返回空串（表示全端），便于前端「未配置 = 全端」的语义保持一致。
|
||||
func NormalizePlatforms(list []string) string {
|
||||
set := map[string]bool{}
|
||||
out := make([]string, 0, len(list))
|
||||
for _, s := range list {
|
||||
v := strings.ToLower(strings.TrimSpace(s))
|
||||
if v == "" || set[v] {
|
||||
continue
|
||||
}
|
||||
set[v] = true
|
||||
out = append(out, v)
|
||||
}
|
||||
if len(out) == 0 {
|
||||
return ""
|
||||
}
|
||||
if b, err := json.Marshal(out); err == nil {
|
||||
return string(b)
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// ListAllModulesForScope 读取全部未删除模块（含停用与 type=2 系统设置）。
|
||||
// 展示端继承判定必须用全量模块：系统设置类模块默认不启用，
|
||||
// 若只用启用模块，其菜单会落到"无归属"被误判为全端可见。
|
||||
func ListAllModulesForScope() ([]models.SystemModules, error) {
|
||||
var rows []models.SystemModules
|
||||
_, err := models.Orm.QueryTable(new(models.SystemModules)).
|
||||
Filter("delete_time__isnull", true).
|
||||
OrderBy("sort", "id").
|
||||
All(&rows)
|
||||
return rows, err
|
||||
}
|
||||
|
||||
// FilterModulesByPlatform 按展示端过滤模块
|
||||
func FilterModulesByPlatform(rows []models.SystemModules, platform string) []models.SystemModules {
|
||||
out := make([]models.SystemModules, 0, len(rows))
|
||||
for _, m := range rows {
|
||||
if HasPlatform(ParsePlatforms(m.Platforms), platform) {
|
||||
out = append(out, m)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// FilterMenusByPlatform 按展示端过滤菜单（只看菜单自身的 platforms，不做模块继承）
|
||||
func FilterMenusByPlatform(rows []models.SystemMenu, platform string) []models.SystemMenu {
|
||||
out := make([]models.SystemMenu, 0, len(rows))
|
||||
for _, m := range rows {
|
||||
if HasPlatform(ParsePlatforms(m.Platforms), platform) {
|
||||
out = append(out, m)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// SortedModulesForScope 把模块按 path 长度倒序，供 MenuVisibleOn 的归属判定复用。
|
||||
func SortedModulesForScope(modules []models.SystemModules) []models.SystemModules {
|
||||
return sortModulesByPathDesc(modules)
|
||||
}
|
||||
|
||||
// MenuVisibleOn 判断单条菜单是否在指定端展示。
|
||||
// 判定顺序：菜单自身 platforms → 继承所属模块 platforms（按 path 前缀归属）→ 全端可见。
|
||||
// 这样后台只需在模块上配一次「仅 Web / 仅小程序」，其下未单独配置的菜单自动跟随。
|
||||
func MenuVisibleOn(m models.SystemMenu, sortedModules []models.SystemModules, allModules []models.SystemModules, platform string) bool {
|
||||
if own := ParsePlatforms(m.Platforms); len(own) > 0 {
|
||||
return HasPlatform(own, platform)
|
||||
}
|
||||
path := ""
|
||||
if m.Path != nil {
|
||||
path = strings.TrimSpace(*m.Path)
|
||||
}
|
||||
code, ok := ownerModuleCode(sortedModules, path)
|
||||
if !ok {
|
||||
return true // 不属于任何模块 → 全端
|
||||
}
|
||||
for _, mod := range allModules {
|
||||
if NormalizeModuleCode(mod.Code) == NormalizeModuleCode(code) {
|
||||
return HasPlatform(ParsePlatforms(mod.Platforms), platform)
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
// FilterMenusByPlatformWithModules 按展示端过滤菜单，并支持继承所属模块的展示端。
|
||||
// modules 应传入全部模块（含停用与 type=2 系统设置），否则系统设置类菜单会误判为全端。
|
||||
func FilterMenusByPlatformWithModules(rows []models.SystemMenu, modules []models.SystemModules, platform string) []models.SystemMenu {
|
||||
sorted := sortModulesByPathDesc(modules)
|
||||
out := make([]models.SystemMenu, 0, len(rows))
|
||||
for _, m := range rows {
|
||||
if MenuVisibleOn(m, sorted, modules, platform) {
|
||||
out = append(out, m)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
@@ -0,0 +1,91 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"server/models"
|
||||
)
|
||||
|
||||
func strPtr(s string) *string { return &s }
|
||||
|
||||
func TestParsePlatforms(t *testing.T) {
|
||||
cases := []struct {
|
||||
name string
|
||||
raw *string
|
||||
want int
|
||||
}{
|
||||
{"nil 视为全端", nil, 0},
|
||||
{"空串视为全端", strPtr(""), 0},
|
||||
{"空格视为全端", strPtr(" "), 0},
|
||||
{"单值 web", strPtr(`["web"]`), 1},
|
||||
{"双值", strPtr(`["web","app"]`), 2},
|
||||
{"大小写归一化", strPtr(`["WEB"]`), 1},
|
||||
{"非法 JSON 退化为单值", strPtr("web"), 1},
|
||||
}
|
||||
for _, c := range cases {
|
||||
got := ParsePlatforms(c.raw)
|
||||
if len(got) != c.want {
|
||||
t.Errorf("%s: got %d items (%v), want %d", c.name, len(got), got, c.want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestHasPlatform(t *testing.T) {
|
||||
if !HasPlatform(nil, models.PlatformWeb) || !HasPlatform(nil, models.PlatformApp) {
|
||||
t.Error("未配置时应当全端可见")
|
||||
}
|
||||
web := []string{"web"}
|
||||
if !HasPlatform(web, "web") {
|
||||
t.Error("web 配置应在 web 端可见")
|
||||
}
|
||||
if HasPlatform(web, "app") {
|
||||
t.Error("web 配置不应在 app 端可见")
|
||||
}
|
||||
both := []string{"web", "app"}
|
||||
if !HasPlatform(both, "web") || !HasPlatform(both, "app") {
|
||||
t.Error("双端配置应两端都可见")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNormalizePlatforms(t *testing.T) {
|
||||
if got := NormalizePlatforms(nil); got != "" {
|
||||
t.Errorf("空输入应返回空串（表示全端），got %q", got)
|
||||
}
|
||||
if got := NormalizePlatforms([]string{"WEB", " app ", "web"}); got != `["web","app"]` {
|
||||
t.Errorf("应去重并小写归一化，got %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
// 模块级过滤：覆盖三种业务场景
|
||||
func TestFilterModulesByPlatform(t *testing.T) {
|
||||
web := `["web"]`
|
||||
app := `["app"]`
|
||||
both := `["web","app"]`
|
||||
rows := []models.SystemModules{
|
||||
{Code: "crm", Platforms: &web}, // 仅 Web
|
||||
{Code: "oa", Platforms: &app}, // 仅小程序
|
||||
{Code: "cms", Platforms: &both}, // 双端
|
||||
{Code: "erp", Platforms: nil}, // 未配置 → 全端
|
||||
}
|
||||
|
||||
got := FilterModulesByPlatform(rows, models.PlatformWeb)
|
||||
if len(got) != 3 {
|
||||
t.Errorf("Web 端应看到 crm/cms/erp 共 3 个，got %d", len(got))
|
||||
}
|
||||
gotApp := FilterModulesByPlatform(rows, models.PlatformApp)
|
||||
if len(gotApp) != 3 {
|
||||
t.Errorf("移动端应看到 oa/cms/erp 共 3 个，got %d", len(gotApp))
|
||||
}
|
||||
}
|
||||
|
||||
func TestFilterMenusByPlatform(t *testing.T) {
|
||||
app := `["app"]`
|
||||
rows := []models.SystemMenu{
|
||||
{Title: "线索", Platforms: strPtr(`["web"]`)},
|
||||
{Title: "日程", Platforms: &app},
|
||||
{Title: "客户", Platforms: nil},
|
||||
}
|
||||
if got := FilterMenusByPlatform(rows, models.PlatformApp); len(got) != 2 {
|
||||
t.Errorf("移动端应看到 日程/客户 共 2 条，got %d", len(got))
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,234 @@
|
||||
package wechatmini
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// =============================================================
|
||||
// 微信小程序服务端 API
|
||||
// 1. GET /sns/jscode2session code → openid / unionid / session_key
|
||||
// 2. POST /wxa/business/getuserphonenumber phone_code → 手机号（需 access_token）
|
||||
// =============================================================
|
||||
|
||||
const apiBase = "https://api.weixin.qq.com"
|
||||
|
||||
var httpClient = &http.Client{Timeout: 10 * time.Second}
|
||||
|
||||
// APIError 微信接口业务错误（HTTP 200 但 errcode 非 0）
|
||||
type APIError struct {
|
||||
API string
|
||||
Code int
|
||||
Msg string
|
||||
}
|
||||
|
||||
func (e *APIError) Error() string {
|
||||
return fmt.Sprintf("%s 失败：errcode=%d errmsg=%s", e.API, e.Code, e.Msg)
|
||||
}
|
||||
|
||||
// Explain 返回常见 errcode 的中文排查建议；未知错误码返回空串
|
||||
func (e *APIError) Explain() string {
|
||||
switch e.Code {
|
||||
case 40029:
|
||||
return "code 无效或已过期：请重新调用 wx.login / uni.login 获取 code 后立即重试"
|
||||
case 40163:
|
||||
return "code 已被使用：同一次 code 只能兑换一次，请重新获取"
|
||||
case 40013:
|
||||
return "AppID 无效：请核对小程序后台「开发 → 开发管理」中的 AppID"
|
||||
case 40125:
|
||||
return "AppSecret 错误：请在小程序后台重置后复制新的 AppSecret 填入本页"
|
||||
case 40226:
|
||||
return "该小程序未开通手机号快速验证能力：请在「开发管理 → 接口设置」中开通"
|
||||
case 45009:
|
||||
return "接口调用频次超限：请稍后重试"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// ExplainError 提取中文排查建议；无可用建议时返回 ("", false)
|
||||
func ExplainError(err error) (string, bool) {
|
||||
if ae, ok := err.(*APIError); ok {
|
||||
if hint := ae.Explain(); hint != "" {
|
||||
return hint, true
|
||||
}
|
||||
}
|
||||
return "", false
|
||||
}
|
||||
|
||||
// ============================ code 换 openid ============================
|
||||
|
||||
// Session jscode2session 返回结果
|
||||
type Session struct {
|
||||
OpenID string `json:"openid"`
|
||||
UnionID string `json:"unionid"`
|
||||
SessionKey string `json:"session_key"`
|
||||
}
|
||||
|
||||
// Code2Session 用 wx.login / uni.login 拿到的 code 换 openid（+ unionid）
|
||||
func Code2Session(cfg *Config, code string) (*Session, error) {
|
||||
if cfg == nil || cfg.AppID == "" || cfg.AppSecret == "" {
|
||||
return nil, ErrNotConfigured
|
||||
}
|
||||
api := fmt.Sprintf("%s/sns/jscode2session?appid=%s&secret=%s&js_code=%s&grant_type=authorization_code",
|
||||
apiBase, url.QueryEscape(cfg.AppID), url.QueryEscape(cfg.AppSecret), url.QueryEscape(code))
|
||||
|
||||
var resp struct {
|
||||
OpenID string `json:"openid"`
|
||||
UnionID string `json:"unionid"`
|
||||
SessionKey string `json:"session_key"`
|
||||
ErrCode int `json:"errcode"`
|
||||
ErrMsg string `json:"errmsg"`
|
||||
}
|
||||
if err := httpGetJSON(api, &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.OpenID == "" {
|
||||
return nil, &APIError{API: "code 换取 openid", Code: resp.ErrCode, Msg: resp.ErrMsg}
|
||||
}
|
||||
return &Session{OpenID: resp.OpenID, UnionID: resp.UnionID, SessionKey: resp.SessionKey}, nil
|
||||
}
|
||||
|
||||
// ============================ 手机号快速验证 ============================
|
||||
|
||||
// PhoneNumber getuserphonenumber 返回结果
|
||||
type PhoneNumber struct {
|
||||
PhoneNumber string `json:"phoneNumber"`
|
||||
PurePhoneNumber string `json:"purePhoneNumber"`
|
||||
CountryCode string `json:"countryCode"`
|
||||
}
|
||||
|
||||
// GetPhoneNumber 用 getPhoneNumber 事件返回的动态 code 换真实手机号。
|
||||
// 仅企业主体小程序可用；未开通该能力时微信返回 40226。
|
||||
func GetPhoneNumber(cfg *Config, phoneCode string) (*PhoneNumber, error) {
|
||||
if cfg == nil || cfg.AppID == "" || cfg.AppSecret == "" {
|
||||
return nil, ErrNotConfigured
|
||||
}
|
||||
token, err := getAccessToken(cfg)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
api := fmt.Sprintf("%s/wxa/business/getuserphonenumber?access_token=%s", apiBase, url.QueryEscape(token))
|
||||
payload := map[string]interface{}{"code": phoneCode}
|
||||
|
||||
var resp struct {
|
||||
PhoneInfo struct {
|
||||
PhoneNumber string `json:"phoneNumber"`
|
||||
PurePhoneNumber string `json:"purePhoneNumber"`
|
||||
CountryCode string `json:"countryCode"`
|
||||
} `json:"phone_info"`
|
||||
ErrCode int `json:"errcode"`
|
||||
ErrMsg string `json:"errmsg"`
|
||||
}
|
||||
if err := httpPostJSON(api, payload, &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.ErrCode != 0 || resp.PhoneInfo.PhoneNumber == "" {
|
||||
return nil, &APIError{API: "获取手机号", Code: resp.ErrCode, Msg: resp.ErrMsg}
|
||||
}
|
||||
return &PhoneNumber{
|
||||
PhoneNumber: resp.PhoneInfo.PhoneNumber,
|
||||
PurePhoneNumber: resp.PhoneInfo.PurePhoneNumber,
|
||||
CountryCode: resp.PhoneInfo.CountryCode,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// ============================ access_token 缓存 ============================
|
||||
|
||||
type tokenEntry struct {
|
||||
appID string
|
||||
token string
|
||||
expiresAt time.Time
|
||||
}
|
||||
|
||||
var (
|
||||
tokenMu sync.Mutex
|
||||
tokenCache tokenEntry
|
||||
tokenMinTTL = 5 * time.Minute
|
||||
)
|
||||
|
||||
// GetAccessToken 获取小程序 access_token（内存缓存，提前 5 分钟刷新）
|
||||
func GetAccessToken(cfg *Config) (string, error) { return getAccessToken(cfg) }
|
||||
|
||||
func getAccessToken(cfg *Config) (string, error) {
|
||||
tokenMu.Lock()
|
||||
if tokenCache.appID == cfg.AppID && tokenCache.token != "" && time.Now().Before(tokenCache.expiresAt) {
|
||||
t := tokenCache.token
|
||||
tokenMu.Unlock()
|
||||
return t, nil
|
||||
}
|
||||
tokenMu.Unlock()
|
||||
|
||||
api := fmt.Sprintf("%s/cgi-bin/token?grant_type=client_credential&appid=%s&secret=%s",
|
||||
apiBase, url.QueryEscape(cfg.AppID), url.QueryEscape(cfg.AppSecret))
|
||||
var resp struct {
|
||||
AccessToken string `json:"access_token"`
|
||||
ExpiresIn int `json:"expires_in"`
|
||||
ErrCode int `json:"errcode"`
|
||||
ErrMsg string `json:"errmsg"`
|
||||
}
|
||||
if err := httpGetJSON(api, &resp); err != nil {
|
||||
return "", err
|
||||
}
|
||||
if resp.AccessToken == "" {
|
||||
return "", &APIError{API: "获取 access_token", Code: resp.ErrCode, Msg: resp.ErrMsg}
|
||||
}
|
||||
ttl := time.Duration(resp.ExpiresIn) * time.Second
|
||||
if ttl <= tokenMinTTL {
|
||||
ttl = time.Hour
|
||||
}
|
||||
tokenMu.Lock()
|
||||
tokenCache = tokenEntry{appID: cfg.AppID, token: resp.AccessToken, expiresAt: time.Now().Add(ttl - tokenMinTTL)}
|
||||
tokenMu.Unlock()
|
||||
return resp.AccessToken, nil
|
||||
}
|
||||
|
||||
// ============================ HTTP 辅助 ============================
|
||||
|
||||
func httpGetJSON(api string, out interface{}) error {
|
||||
resp, err := httpClient.Get(api)
|
||||
if err != nil {
|
||||
return fmt.Errorf("请求微信接口失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return fmt.Errorf("读取微信响应失败: %w", err)
|
||||
}
|
||||
if err := json.Unmarshal(body, out); err != nil {
|
||||
return fmt.Errorf("解析微信响应失败: %w（body: %s）", err, truncate(string(body), 200))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func httpPostJSON(api string, payload interface{}, out interface{}) error {
|
||||
bs, err := json.Marshal(payload)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
resp, err := httpClient.Post(api, "application/json; charset=utf-8", bytes.NewReader(bs))
|
||||
if err != nil {
|
||||
return fmt.Errorf("请求微信接口失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return fmt.Errorf("读取微信响应失败: %w", err)
|
||||
}
|
||||
if err := json.Unmarshal(body, out); err != nil {
|
||||
return fmt.Errorf("解析微信响应失败: %w（body: %s）", err, truncate(string(body), 200))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func truncate(s string, n int) string {
|
||||
if len(s) <= n {
|
||||
return s
|
||||
}
|
||||
return s[:n] + "..."
|
||||
}
|
||||
@@ -0,0 +1,126 @@
|
||||
// Package wechatmini 微信小程序服务端能力：配置、code 换 openid、手机号快速验证。
|
||||
//
|
||||
// 与 services/wechatmp（公众号）分开：小程序走 sns/jscode2session 与
|
||||
// wxa/business/getuserphonenumber，两者的 AppID / AppSecret 互不相通。
|
||||
package wechatmini
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/beego/beego/v2/client/orm"
|
||||
"server/models"
|
||||
"server/services/wechatmp"
|
||||
)
|
||||
|
||||
// Config 小程序运行配置（敏感字段已解密）
|
||||
type Config struct {
|
||||
ID uint64
|
||||
AppID string
|
||||
AppSecret string // 明文
|
||||
Enabled bool
|
||||
AutoBind bool
|
||||
Remark string
|
||||
}
|
||||
|
||||
// ErrNotConfigured 未配置或未启用
|
||||
var ErrNotConfigured = errors.New("微信小程序未配置或未启用")
|
||||
|
||||
// LoadConfig 读取小程序配置（单行，敏感字段解密）；未配置返回 ErrNotConfigured
|
||||
func LoadConfig() (*Config, error) {
|
||||
var row models.WechatMiniConfig
|
||||
err := models.Orm.QueryTable(new(models.WechatMiniConfig)).OrderBy("id").One(&row)
|
||||
if err != nil {
|
||||
if err == orm.ErrNoRows {
|
||||
return nil, ErrNotConfigured
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
cfg := &Config{
|
||||
ID: row.ID,
|
||||
AppID: strings.TrimSpace(row.AppID),
|
||||
Enabled: row.Enabled == 1,
|
||||
AutoBind: row.AutoBind != 0,
|
||||
Remark: row.Remark,
|
||||
}
|
||||
if row.AppSecret != "" {
|
||||
plain, derr := wechatmp.DecryptSecret(row.AppSecret)
|
||||
if derr != nil {
|
||||
return nil, derr
|
||||
}
|
||||
cfg.AppSecret = plain
|
||||
}
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
// LoadEnabledConfig 读取并校验启用状态
|
||||
func LoadEnabledConfig() (*Config, error) {
|
||||
cfg, err := LoadConfig()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !cfg.Enabled || cfg.AppID == "" || cfg.AppSecret == "" {
|
||||
return nil, ErrNotConfigured
|
||||
}
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
// SaveInput 保存配置入参（AppSecret 支持掩码表示不修改）
|
||||
type SaveInput struct {
|
||||
AppID string
|
||||
AppSecret string // 掩码 / 空串 = 保持不变
|
||||
Enabled bool
|
||||
AutoBind bool
|
||||
Remark string
|
||||
}
|
||||
|
||||
// SaveConfig 保存小程序配置（不存在则创建）
|
||||
func SaveConfig(in SaveInput) error {
|
||||
if strings.TrimSpace(in.AppID) == "" {
|
||||
return errors.New("AppID 不能为空")
|
||||
}
|
||||
|
||||
var row models.WechatMiniConfig
|
||||
err := models.Orm.QueryTable(new(models.WechatMiniConfig)).OrderBy("id").One(&row)
|
||||
now := time.Now()
|
||||
isNew := false
|
||||
if err != nil {
|
||||
if err != orm.ErrNoRows {
|
||||
return err
|
||||
}
|
||||
isNew = true
|
||||
row = models.WechatMiniConfig{}
|
||||
}
|
||||
|
||||
secret := strings.TrimSpace(in.AppSecret)
|
||||
if secret != "" && !wechatmp.IsMasked(secret) {
|
||||
enc, eerr := wechatmp.EncryptSecret(secret)
|
||||
if eerr != nil {
|
||||
return eerr
|
||||
}
|
||||
row.AppSecret = enc
|
||||
}
|
||||
|
||||
row.AppID = strings.TrimSpace(in.AppID)
|
||||
row.Remark = strings.TrimSpace(in.Remark)
|
||||
if in.Enabled {
|
||||
row.Enabled = 1
|
||||
} else {
|
||||
row.Enabled = 0
|
||||
}
|
||||
if in.AutoBind {
|
||||
row.AutoBind = 1
|
||||
} else {
|
||||
row.AutoBind = 0
|
||||
}
|
||||
row.UpdateTime = &now
|
||||
|
||||
if isNew {
|
||||
row.CreateTime = now
|
||||
_, err = models.Orm.Insert(&row)
|
||||
return err
|
||||
}
|
||||
_, err = models.Orm.Update(&row)
|
||||
return err
|
||||
}
|
||||
Reference in new issue
Block a user