refactor: imporve code style
This commit is contained in:
		
							parent
							
								
									eee3225f3a
								
							
						
					
					
						commit
						f75847d3d7
					
				@ -2,6 +2,7 @@ package controller
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"errors"
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
	"log"
 | 
						"log"
 | 
				
			||||||
	"net/http"
 | 
						"net/http"
 | 
				
			||||||
	"regexp"
 | 
						"regexp"
 | 
				
			||||||
@ -223,31 +224,30 @@ func (cp *commonPage) network(c *gin.Context) {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (cp *commonPage) getServerStat(c *gin.Context) ([]byte, error) {
 | 
					func (cp *commonPage) getServerStat(c *gin.Context) ([]byte, error) {
 | 
				
			||||||
	v, err, _ := cp.requestGroup.Do("serverStats", func() (any, error) {
 | 
						_, isMember := c.Get(model.CtxKeyAuthorizedUser)
 | 
				
			||||||
 | 
						_, isViewPasswordVerfied := c.Get(model.CtxKeyViewPasswordVerified)
 | 
				
			||||||
 | 
						authorized := isMember || isViewPasswordVerfied
 | 
				
			||||||
 | 
						v, err, _ := cp.requestGroup.Do(fmt.Sprintf("serverStats::%t", authorized), func() (interface{}, error) {
 | 
				
			||||||
		singleton.SortedServerLock.RLock()
 | 
							singleton.SortedServerLock.RLock()
 | 
				
			||||||
		defer singleton.SortedServerLock.RUnlock()
 | 
							defer singleton.SortedServerLock.RUnlock()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		_, isMember := c.Get(model.CtxKeyAuthorizedUser)
 | 
					 | 
				
			||||||
		_, isViewPasswordVerfied := c.Get(model.CtxKeyViewPasswordVerified)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		var servers []*model.Server
 | 
							var servers []*model.Server
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if isMember || isViewPasswordVerfied {
 | 
							if authorized {
 | 
				
			||||||
			servers = singleton.SortedServerList
 | 
								servers = singleton.SortedServerList
 | 
				
			||||||
		} else {
 | 
							} else {
 | 
				
			||||||
			servers = singleton.SortedServerListForGuest
 | 
								filteredServers := make([]*model.Server, len(singleton.SortedServerListForGuest))
 | 
				
			||||||
		}
 | 
								for i, server := range singleton.SortedServerListForGuest {
 | 
				
			||||||
 | 
					 | 
				
			||||||
		filteredServers := make([]*model.Server, len(servers))
 | 
					 | 
				
			||||||
		for i, server := range servers {
 | 
					 | 
				
			||||||
				filteredServer := *server
 | 
									filteredServer := *server
 | 
				
			||||||
				filteredServer.DDNSDomain = "redacted"
 | 
									filteredServer.DDNSDomain = "redacted"
 | 
				
			||||||
				filteredServers[i] = &filteredServer
 | 
									filteredServers[i] = &filteredServer
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
								servers = filteredServers
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		return utils.Json.Marshal(Data{
 | 
							return utils.Json.Marshal(Data{
 | 
				
			||||||
			Now:     time.Now().Unix() * 1000,
 | 
								Now:     time.Now().Unix() * 1000,
 | 
				
			||||||
			Servers: filteredServers,
 | 
								Servers: servers,
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
	return v.([]byte), err
 | 
						return v.([]byte), err
 | 
				
			||||||
 | 
				
			|||||||
@ -437,7 +437,6 @@ func (ma *memberAPI) addOrEditMonitor(c *gin.Context) {
 | 
				
			|||||||
		if m.NotificationTag == "" {
 | 
							if m.NotificationTag == "" {
 | 
				
			||||||
			m.NotificationTag = "default"
 | 
								m.NotificationTag = "default"
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		if err == nil {
 | 
					 | 
				
			||||||
		err = utils.Json.Unmarshal([]byte(mf.FailTriggerTasksRaw), &m.FailTriggerTasks)
 | 
							err = utils.Json.Unmarshal([]byte(mf.FailTriggerTasksRaw), &m.FailTriggerTasks)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if err == nil {
 | 
						if err == nil {
 | 
				
			||||||
@ -457,7 +456,6 @@ func (ma *memberAPI) addOrEditMonitor(c *gin.Context) {
 | 
				
			|||||||
			err = singleton.DB.Unscoped().Delete(&model.MonitorHistory{}, "monitor_id = ? and server_id not in (?)", m.ID, strings.Split(m.SkipServersRaw[1:len(m.SkipServersRaw)-1], ",")).Error
 | 
								err = singleton.DB.Unscoped().Delete(&model.MonitorHistory{}, "monitor_id = ? and server_id not in (?)", m.ID, strings.Split(m.SkipServersRaw[1:len(m.SkipServersRaw)-1], ",")).Error
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err == nil {
 | 
						if err == nil {
 | 
				
			||||||
		err = singleton.ServiceSentinelShared.OnMonitorUpdate(m)
 | 
							err = singleton.ServiceSentinelShared.OnMonitorUpdate(m)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
				
			|||||||
@ -5,12 +5,13 @@ import (
 | 
				
			|||||||
	"encoding/json"
 | 
						"encoding/json"
 | 
				
			||||||
	"errors"
 | 
						"errors"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
	"github.com/naiba/nezha/pkg/oidc/cloudflare"
 | 
					 | 
				
			||||||
	"net/http"
 | 
						"net/http"
 | 
				
			||||||
	"net/url"
 | 
						"net/url"
 | 
				
			||||||
	"strings"
 | 
						"strings"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/naiba/nezha/pkg/oidc/cloudflare"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"code.gitea.io/sdk/gitea"
 | 
						"code.gitea.io/sdk/gitea"
 | 
				
			||||||
	"github.com/gin-gonic/gin"
 | 
						"github.com/gin-gonic/gin"
 | 
				
			||||||
	GitHubAPI "github.com/google/go-github/v47/github"
 | 
						GitHubAPI "github.com/google/go-github/v47/github"
 | 
				
			||||||
@ -192,15 +193,12 @@ func (oa *oauth2controller) callback(c *gin.Context) {
 | 
				
			|||||||
				client = GitHubAPI.NewClient(oc)
 | 
									client = GitHubAPI.NewClient(oc)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
			var gu *GitHubAPI.User
 | 
								var gu *GitHubAPI.User
 | 
				
			||||||
			if err == nil {
 | 
					 | 
				
			||||||
			gu, _, err = client.Users.Get(ctx, "")
 | 
								gu, _, err = client.Users.Get(ctx, "")
 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if err == nil {
 | 
								if err == nil {
 | 
				
			||||||
				user = model.NewUserFromGitHub(gu)
 | 
									user = model.NewUserFromGitHub(gu)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err == nil && user.Login == "" {
 | 
						if err == nil && user.Login == "" {
 | 
				
			||||||
		err = errors.New("获取用户信息失败")
 | 
							err = errors.New("获取用户信息失败")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user