mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 18:27:13 +08:00
267 lines
7.4 KiB
Go
267 lines
7.4 KiB
Go
package toolbox
|
|
|
|
import (
|
|
"net/http"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/go-rat/chix"
|
|
"github.com/spf13/cast"
|
|
|
|
"github.com/TheTNB/panel/internal/app"
|
|
"github.com/TheTNB/panel/internal/service"
|
|
"github.com/TheTNB/panel/pkg/io"
|
|
"github.com/TheTNB/panel/pkg/shell"
|
|
"github.com/TheTNB/panel/pkg/str"
|
|
"github.com/TheTNB/panel/pkg/types"
|
|
)
|
|
|
|
type Service struct{}
|
|
|
|
func NewService() *Service {
|
|
return &Service{}
|
|
}
|
|
|
|
// GetDNS 获取 DNS 信息
|
|
func (s *Service) GetDNS(w http.ResponseWriter, r *http.Request) {
|
|
raw, err := io.Read("/etc/resolv.conf")
|
|
if err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
match := regexp.MustCompile(`nameserver\s+(\S+)`).FindAllStringSubmatch(raw, -1)
|
|
if len(match) == 0 {
|
|
service.Error(w, http.StatusInternalServerError, "找不到 DNS 信息")
|
|
return
|
|
}
|
|
|
|
var dns []string
|
|
for _, m := range match {
|
|
dns = append(dns, m[1])
|
|
}
|
|
|
|
service.Success(w, dns)
|
|
}
|
|
|
|
// UpdateDNS 设置 DNS 信息
|
|
func (s *Service) UpdateDNS(w http.ResponseWriter, r *http.Request) {
|
|
req, err := service.Bind[DNS](r)
|
|
if err != nil {
|
|
service.Error(w, http.StatusUnprocessableEntity, "%v", err)
|
|
return
|
|
}
|
|
|
|
var dns string
|
|
dns += "nameserver " + req.DNS1 + "\n"
|
|
dns += "nameserver " + req.DNS2 + "\n"
|
|
|
|
if err := io.Write("/etc/resolv.conf", dns, 0644); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "写入 DNS 信息失败")
|
|
return
|
|
}
|
|
|
|
service.Success(w, nil)
|
|
}
|
|
|
|
// GetSWAP 获取 SWAP 信息
|
|
func (s *Service) GetSWAP(w http.ResponseWriter, r *http.Request) {
|
|
var total, used, free string
|
|
var size int64
|
|
if io.Exists(filepath.Join(app.Root, "swap")) {
|
|
file, err := io.FileInfo(filepath.Join(app.Root, "swap"))
|
|
if err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "获取 SWAP 信息失败")
|
|
return
|
|
}
|
|
|
|
size = file.Size() / 1024 / 1024
|
|
total = str.FormatBytes(float64(file.Size()))
|
|
} else {
|
|
size = 0
|
|
total = "0.00 B"
|
|
}
|
|
|
|
raw, err := shell.Execf("free | grep Swap")
|
|
if err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "获取 SWAP 信息失败")
|
|
return
|
|
}
|
|
|
|
match := regexp.MustCompile(`Swap:\s+(\d+)\s+(\d+)\s+(\d+)`).FindStringSubmatch(raw)
|
|
if len(match) >= 4 {
|
|
used = str.FormatBytes(cast.ToFloat64(match[2]) * 1024)
|
|
free = str.FormatBytes(cast.ToFloat64(match[3]) * 1024)
|
|
}
|
|
|
|
service.Success(w, chix.M{
|
|
"total": total,
|
|
"size": size,
|
|
"used": used,
|
|
"free": free,
|
|
})
|
|
}
|
|
|
|
// UpdateSWAP 设置 SWAP 信息
|
|
func (s *Service) UpdateSWAP(w http.ResponseWriter, r *http.Request) {
|
|
req, err := service.Bind[SWAP](r)
|
|
if err != nil {
|
|
service.Error(w, http.StatusUnprocessableEntity, "%v", err)
|
|
return
|
|
}
|
|
|
|
if io.Exists(filepath.Join(app.Root, "swap")) {
|
|
if _, err = shell.Execf("swapoff '%s'", filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
if _, err = shell.Execf("rm -f '%s'", filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
if _, err = shell.Execf(`sed -i "\|^%s|d" /etc/fstab`, filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
if req.Size > 1 {
|
|
var free string
|
|
free, err = shell.Execf("df -k %s | awk '{print $4}' | tail -n 1", app.Root)
|
|
if err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "获取硬盘空间失败")
|
|
return
|
|
}
|
|
if cast.ToInt64(free)*1024 < req.Size*1024*1024 {
|
|
service.Error(w, http.StatusInternalServerError, "硬盘空间不足，当前剩余%s", str.FormatBytes(cast.ToFloat64(free)))
|
|
return
|
|
}
|
|
|
|
btrfsCheck, _ := shell.Execf("df -T %s | awk '{print $2}' | tail -n 1", app.Root)
|
|
if strings.Contains(btrfsCheck, "btrfs") {
|
|
if _, err = shell.Execf("btrfs filesystem mkswapfile --size %dM --uuid clear %s", req.Size, filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
} else {
|
|
if _, err = shell.Execf("dd if=/dev/zero of=%s bs=1M count=%d", filepath.Join(app.Root, "swap"), req.Size); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
if _, err = shell.Execf("mkswap -f '%s'", filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
if err = io.Chmod(filepath.Join(app.Root, "swap"), 0600); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "设置 SWAP 权限失败")
|
|
return
|
|
}
|
|
}
|
|
if _, err = shell.Execf("swapon '%s'", filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
if _, err = shell.Execf("echo '%s swap swap defaults 0 0' >> /etc/fstab", filepath.Join(app.Root, "swap")); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
service.Success(w, nil)
|
|
}
|
|
|
|
// GetTimezone 获取时区
|
|
func (s *Service) GetTimezone(w http.ResponseWriter, r *http.Request) {
|
|
raw, err := shell.Execf("timedatectl | grep zone")
|
|
if err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "获取时区信息失败")
|
|
return
|
|
}
|
|
|
|
match := regexp.MustCompile(`zone:\s+(\S+)`).FindStringSubmatch(raw)
|
|
if len(match) == 0 {
|
|
service.Error(w, http.StatusInternalServerError, "找不到时区信息")
|
|
return
|
|
}
|
|
|
|
zonesRaw, err := shell.Execf("timedatectl list-timezones")
|
|
if err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "获取时区列表失败")
|
|
return
|
|
}
|
|
zones := strings.Split(zonesRaw, "\n")
|
|
|
|
var zonesList []types.LV
|
|
for _, z := range zones {
|
|
zonesList = append(zonesList, types.LV{
|
|
Label: z,
|
|
Value: z,
|
|
})
|
|
}
|
|
|
|
service.Success(w, chix.M{
|
|
"timezone": match[1],
|
|
"timezones": zonesList,
|
|
})
|
|
}
|
|
|
|
// UpdateTimezone 设置时区
|
|
func (s *Service) UpdateTimezone(w http.ResponseWriter, r *http.Request) {
|
|
req, err := service.Bind[Timezone](r)
|
|
if err != nil {
|
|
service.Error(w, http.StatusUnprocessableEntity, "%v", err)
|
|
return
|
|
}
|
|
|
|
if _, err = shell.Execf("timedatectl set-timezone '%s'", req.Timezone); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
|
|
service.Success(w, nil)
|
|
}
|
|
|
|
// GetHosts 获取 hosts 信息
|
|
func (s *Service) GetHosts(w http.ResponseWriter, r *http.Request) {
|
|
hosts, _ := io.Read("/etc/hosts")
|
|
service.Success(w, hosts)
|
|
}
|
|
|
|
// UpdateHosts 设置 hosts 信息
|
|
func (s *Service) UpdateHosts(w http.ResponseWriter, r *http.Request) {
|
|
req, err := service.Bind[Hosts](r)
|
|
if err != nil {
|
|
service.Error(w, http.StatusUnprocessableEntity, "%v", err)
|
|
return
|
|
}
|
|
|
|
if err = io.Write("/etc/hosts", req.Hosts, 0644); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "写入 hosts 信息失败")
|
|
return
|
|
}
|
|
|
|
service.Success(w, nil)
|
|
}
|
|
|
|
// UpdateRootPassword 设置 root 密码
|
|
func (s *Service) UpdateRootPassword(w http.ResponseWriter, r *http.Request) {
|
|
req, err := service.Bind[Password](r)
|
|
if err != nil {
|
|
service.Error(w, http.StatusUnprocessableEntity, "%v", err)
|
|
return
|
|
}
|
|
|
|
if !regexp.MustCompile(`^[a-zA-Z0-9·~!@#$%^&*()_+-=\[\]{};:'",./<>?]{6,20}$`).MatchString(req.Password) {
|
|
service.Error(w, http.StatusUnprocessableEntity, "密码必须为 6-20 位字母、数字或特殊字符")
|
|
return
|
|
}
|
|
|
|
req.Password = strings.ReplaceAll(req.Password, `'`, `\'`)
|
|
if _, err = shell.Execf(`yes '%s' | passwd root`, req.Password); err != nil {
|
|
service.Error(w, http.StatusInternalServerError, "%v", err)
|
|
return
|
|
}
|
|
|
|
service.Success(w, nil)
|
|
}
|