From 717c53f6e533b34ba2743155a93623566ed56d60 Mon Sep 17 00:00:00 2001
From: Yang Luo <hsluoyz@gmail.com>
Date: Wed, 25 Sep 2024 19:28:53 +0800
Subject: [PATCH] feat: support enableErrorMask2 config

---
 controllers/user.go | 6 ++++++
 controllers/util.go | 9 +++++++++
 2 files changed, 15 insertions(+)

diff --git a/controllers/user.go b/controllers/user.go
index 86105e6e..02beb544 100644
--- a/controllers/user.go
+++ b/controllers/user.go
@@ -410,6 +410,12 @@ func (c *ApiController) GetEmailAndPhone() {
 	organization := c.Ctx.Request.Form.Get("organization")
 	username := c.Ctx.Request.Form.Get("username")
 
+	enableErrorMask2 := conf.GetConfigBool("enableErrorMask2")
+	if enableErrorMask2 {
+		c.ResponseError("Error")
+		return
+	}
+
 	user, err := object.GetUserByFields(organization, username)
 	if err != nil {
 		c.ResponseError(err.Error())
diff --git a/controllers/util.go b/controllers/util.go
index 433c88ef..27a67786 100644
--- a/controllers/util.go
+++ b/controllers/util.go
@@ -45,6 +45,15 @@ func (c *ApiController) ResponseOk(data ...interface{}) {
 
 // ResponseError ...
 func (c *ApiController) ResponseError(error string, data ...interface{}) {
+	enableErrorMask2 := conf.GetConfigBool("enableErrorMask2")
+	if enableErrorMask2 {
+		error = c.T("subscription:Error")
+
+		resp := &Response{Status: "error", Msg: error}
+		c.ResponseJsonData(resp, data...)
+		return
+	}
+
 	enableErrorMask := conf.GetConfigBool("enableErrorMask")
 	if enableErrorMask {
 		if strings.HasPrefix(error, "The user: ") && strings.HasSuffix(error, " doesn't exist") || strings.HasPrefix(error, "用户: ") && strings.HasSuffix(error, "不存在") {