mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 04:22:33 +08:00
332 lines
8.5 KiB
Go
332 lines
8.5 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/go-rat/chix"
|
|
"github.com/go-rat/utils/collect"
|
|
"github.com/hashicorp/go-version"
|
|
"github.com/shirou/gopsutil/disk"
|
|
"github.com/shirou/gopsutil/host"
|
|
"github.com/spf13/cast"
|
|
|
|
"github.com/TheTNB/panel/internal/app"
|
|
"github.com/TheTNB/panel/internal/biz"
|
|
"github.com/TheTNB/panel/internal/data"
|
|
"github.com/TheTNB/panel/internal/http/request"
|
|
"github.com/TheTNB/panel/pkg/api"
|
|
"github.com/TheTNB/panel/pkg/db"
|
|
"github.com/TheTNB/panel/pkg/shell"
|
|
"github.com/TheTNB/panel/pkg/tools"
|
|
"github.com/TheTNB/panel/pkg/types"
|
|
)
|
|
|
|
type DashboardService struct {
|
|
api *api.API
|
|
taskRepo biz.TaskRepo
|
|
websiteRepo biz.WebsiteRepo
|
|
appRepo biz.AppRepo
|
|
settingRepo biz.SettingRepo
|
|
cronRepo biz.CronRepo
|
|
}
|
|
|
|
func NewDashboardService() *DashboardService {
|
|
return &DashboardService{
|
|
api: api.NewAPI(app.Version),
|
|
taskRepo: data.NewTaskRepo(),
|
|
websiteRepo: data.NewWebsiteRepo(),
|
|
appRepo: data.NewAppRepo(),
|
|
settingRepo: data.NewSettingRepo(),
|
|
cronRepo: data.NewCronRepo(),
|
|
}
|
|
}
|
|
|
|
func (s *DashboardService) Panel(w http.ResponseWriter, r *http.Request) {
|
|
name, _ := s.settingRepo.Get(biz.SettingKeyName)
|
|
if name == "" {
|
|
name = "耗子面板"
|
|
}
|
|
|
|
Success(w, chix.M{
|
|
"name": name,
|
|
"locale": app.Conf.MustString("app.locale"),
|
|
})
|
|
}
|
|
|
|
func (s *DashboardService) HomeApps(w http.ResponseWriter, r *http.Request) {
|
|
apps, err := s.appRepo.GetHomeShow()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取首页应用失败: %v", err)
|
|
return
|
|
}
|
|
|
|
Success(w, apps)
|
|
}
|
|
|
|
func (s *DashboardService) Current(w http.ResponseWriter, r *http.Request) {
|
|
req, err := Bind[request.DashboardCurrent](r)
|
|
if err != nil {
|
|
Error(w, http.StatusUnprocessableEntity, "%v", err)
|
|
return
|
|
}
|
|
|
|
Success(w, tools.CurrentInfo(req.Nets, req.Disks))
|
|
}
|
|
|
|
func (s *DashboardService) SystemInfo(w http.ResponseWriter, r *http.Request) {
|
|
hostInfo, err := host.Info()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取系统信息失败")
|
|
return
|
|
}
|
|
|
|
// 所有网卡名称
|
|
var nets []types.LV
|
|
netInterfaces, _ := net.Interfaces()
|
|
for _, v := range netInterfaces {
|
|
nets = append(nets, types.LV{
|
|
Value: v.Name,
|
|
Label: v.Name,
|
|
})
|
|
}
|
|
// 所有硬盘名称
|
|
var disks []types.LV
|
|
partitions, _ := disk.Partitions(false)
|
|
for _, v := range partitions {
|
|
disks = append(disks, types.LV{
|
|
Value: v.Device,
|
|
Label: fmt.Sprintf("%s (%s)", v.Device, v.Mountpoint),
|
|
})
|
|
}
|
|
|
|
Success(w, chix.M{
|
|
"procs": hostInfo.Procs,
|
|
"hostname": hostInfo.Hostname,
|
|
"panel_version": app.Version,
|
|
"commit_hash": app.CommitHash,
|
|
"build_id": app.BuildID,
|
|
"build_time": app.BuildTime,
|
|
"build_user": app.BuildUser,
|
|
"build_host": app.BuildHost,
|
|
"go_version": app.GoVersion,
|
|
"kernel_arch": hostInfo.KernelArch,
|
|
"kernel_version": hostInfo.KernelVersion,
|
|
"os_name": hostInfo.Platform + " " + hostInfo.PlatformVersion,
|
|
"boot_time": hostInfo.BootTime,
|
|
"uptime": hostInfo.Uptime,
|
|
"nets": nets,
|
|
"disks": disks,
|
|
})
|
|
}
|
|
|
|
func (s *DashboardService) CountInfo(w http.ResponseWriter, r *http.Request) {
|
|
websiteCount, err := s.websiteRepo.Count()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取网站数量失败")
|
|
return
|
|
}
|
|
|
|
mysqlInstalled, _ := s.appRepo.IsInstalled("slug = ?", "mysql")
|
|
postgresqlInstalled, _ := s.appRepo.IsInstalled("slug = ?", "postgresql")
|
|
|
|
var databaseCount int
|
|
if mysqlInstalled {
|
|
rootPassword, _ := s.settingRepo.Get(biz.SettingKeyMySQLRootPassword)
|
|
mysql, err := db.NewMySQL("root", rootPassword, "/tmp/mysql.sock", "unix")
|
|
if err == nil {
|
|
defer mysql.Close()
|
|
databases, err := mysql.Databases()
|
|
if err == nil {
|
|
databaseCount += len(databases)
|
|
}
|
|
}
|
|
}
|
|
if postgresqlInstalled {
|
|
postgres, err := db.NewPostgres("postgres", "", "127.0.0.1", 5432)
|
|
if err == nil {
|
|
defer postgres.Close()
|
|
databases, err := postgres.Databases()
|
|
if err == nil {
|
|
databaseCount += len(databases)
|
|
}
|
|
}
|
|
}
|
|
|
|
var ftpCount int
|
|
ftpInstalled, _ := s.appRepo.IsInstalled("slug = ?", "pureftpd")
|
|
if ftpInstalled {
|
|
listRaw, err := shell.Execf("pure-pw list")
|
|
if len(listRaw) != 0 && err == nil {
|
|
listArr := strings.Split(listRaw, "\n")
|
|
ftpCount = len(listArr)
|
|
}
|
|
}
|
|
|
|
cronCount, err := s.cronRepo.Count()
|
|
if err != nil {
|
|
cronCount = -1
|
|
}
|
|
|
|
Success(w, chix.M{
|
|
"website": websiteCount,
|
|
"database": databaseCount,
|
|
"ftp": ftpCount,
|
|
"cron": cronCount,
|
|
})
|
|
}
|
|
|
|
func (s *DashboardService) InstalledDbAndPhp(w http.ResponseWriter, r *http.Request) {
|
|
mysqlInstalled, _ := s.appRepo.IsInstalled("slug = ?", "mysql")
|
|
postgresqlInstalled, _ := s.appRepo.IsInstalled("slug = ?", "postgresql")
|
|
php, _ := s.appRepo.GetInstalledAll("slug like ?", "php%")
|
|
|
|
var phpData []types.LVInt
|
|
var dbData []types.LV
|
|
phpData = append(phpData, types.LVInt{Value: 0, Label: "不使用"})
|
|
dbData = append(dbData, types.LV{Value: "0", Label: "不使用"})
|
|
for _, p := range php {
|
|
// 过滤 phpmyadmin
|
|
match := regexp.MustCompile(`php(\d+)`).FindStringSubmatch(p.Slug)
|
|
if len(match) == 0 {
|
|
continue
|
|
}
|
|
|
|
item, _ := s.appRepo.Get(p.Slug)
|
|
phpData = append(phpData, types.LVInt{Value: cast.ToInt(strings.ReplaceAll(p.Slug, "php", "")), Label: item.Name})
|
|
}
|
|
|
|
if mysqlInstalled {
|
|
dbData = append(dbData, types.LV{Value: "mysql", Label: "MySQL"})
|
|
}
|
|
if postgresqlInstalled {
|
|
dbData = append(dbData, types.LV{Value: "postgresql", Label: "PostgreSQL"})
|
|
}
|
|
|
|
Success(w, chix.M{
|
|
"php": phpData,
|
|
"db": dbData,
|
|
})
|
|
}
|
|
|
|
func (s *DashboardService) CheckUpdate(w http.ResponseWriter, r *http.Request) {
|
|
if offline, _ := s.settingRepo.GetBool(biz.SettingKeyOfflineMode); offline {
|
|
Error(w, http.StatusForbidden, "离线模式下无法检查更新")
|
|
return
|
|
}
|
|
|
|
current := app.Version
|
|
latest, err := s.api.LatestVersion()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取最新版本失败")
|
|
return
|
|
}
|
|
|
|
v1, err := version.NewVersion(current)
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "版本号解析失败")
|
|
return
|
|
}
|
|
v2, err := version.NewVersion(latest.Version)
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "版本号解析失败")
|
|
return
|
|
}
|
|
if v1.GreaterThanOrEqual(v2) {
|
|
Success(w, chix.M{
|
|
"update": false,
|
|
})
|
|
return
|
|
}
|
|
|
|
Success(w, chix.M{
|
|
"update": true,
|
|
})
|
|
}
|
|
|
|
func (s *DashboardService) UpdateInfo(w http.ResponseWriter, r *http.Request) {
|
|
if offline, _ := s.settingRepo.GetBool(biz.SettingKeyOfflineMode); offline {
|
|
Error(w, http.StatusForbidden, "离线模式下无法检查更新")
|
|
return
|
|
}
|
|
|
|
current := app.Version
|
|
latest, err := s.api.LatestVersion()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取最新版本失败")
|
|
return
|
|
}
|
|
|
|
v1, err := version.NewVersion(current)
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "版本号解析失败")
|
|
return
|
|
}
|
|
v2, err := version.NewVersion(latest.Version)
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "版本号解析失败")
|
|
return
|
|
}
|
|
if v1.GreaterThanOrEqual(v2) {
|
|
Error(w, http.StatusInternalServerError, "当前版本已是最新版本")
|
|
return
|
|
}
|
|
|
|
versions, err := s.api.IntermediateVersions()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取更新信息失败：%v", err)
|
|
return
|
|
}
|
|
|
|
Success(w, versions)
|
|
}
|
|
|
|
func (s *DashboardService) Update(w http.ResponseWriter, r *http.Request) {
|
|
if offline, _ := s.settingRepo.GetBool(biz.SettingKeyOfflineMode); offline {
|
|
Error(w, http.StatusForbidden, "离线模式下无法更新")
|
|
return
|
|
}
|
|
|
|
if s.taskRepo.HasRunningTask() {
|
|
Error(w, http.StatusInternalServerError, "后台任务正在运行，禁止更新，请稍后再试")
|
|
return
|
|
}
|
|
|
|
panel, err := s.api.LatestVersion()
|
|
if err != nil {
|
|
Error(w, http.StatusInternalServerError, "获取最新版本失败：%v", err)
|
|
return
|
|
}
|
|
|
|
download := collect.First(panel.Downloads)
|
|
if download == nil {
|
|
Error(w, http.StatusInternalServerError, "获取下载链接失败")
|
|
return
|
|
}
|
|
ver, url, checksum := panel.Version, download.URL, download.Checksum
|
|
|
|
app.Status = app.StatusUpgrade
|
|
if err = s.settingRepo.UpdatePanel(ver, url, checksum); err != nil {
|
|
app.Status = app.StatusFailed
|
|
Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
|
|
app.Status = app.StatusNormal
|
|
Success(w, nil)
|
|
tools.RestartPanel()
|
|
}
|
|
|
|
func (s *DashboardService) Restart(w http.ResponseWriter, r *http.Request) {
|
|
if s.taskRepo.HasRunningTask() {
|
|
Error(w, http.StatusInternalServerError, "后台任务正在运行，禁止重启，请稍后再试")
|
|
return
|
|
}
|
|
|
|
tools.RestartPanel()
|
|
Success(w, nil)
|
|
}
|