mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 13:47:15 +08:00
129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
package data
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/tnb-labs/panel/internal/app"
|
|
"github.com/tnb-labs/panel/internal/biz"
|
|
"github.com/tnb-labs/panel/pkg/shell"
|
|
"github.com/tnb-labs/panel/pkg/types"
|
|
)
|
|
|
|
type containerComposeRepo struct{}
|
|
|
|
func NewContainerComposeRepo() biz.ContainerComposeRepo {
|
|
return &containerComposeRepo{}
|
|
}
|
|
|
|
// List 列出所有编排文件名
|
|
func (r *containerComposeRepo) List() ([]types.ContainerCompose, error) {
|
|
raw, err := shell.Execf("docker compose ls --format json")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var composeRaws []types.ContainerComposeRaw
|
|
if err = json.Unmarshal([]byte(raw), &composeRaws); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
index := make(map[string]int)
|
|
var composes []types.ContainerCompose
|
|
err = filepath.WalkDir(filepath.Join(app.Root, "server", "compose"), func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !d.IsDir() {
|
|
return nil
|
|
}
|
|
var createdAt time.Time
|
|
if info, err := d.Info(); err == nil {
|
|
createdAt = info.ModTime()
|
|
}
|
|
composes = append(composes, types.ContainerCompose{
|
|
Name: filepath.Base(path),
|
|
Dir: path,
|
|
Status: "unknown",
|
|
CreatedAt: createdAt,
|
|
})
|
|
index[filepath.Base(path)] = len(composes) - 1
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 更新状态
|
|
for _, item := range composeRaws {
|
|
if i, ok := index[item.Name]; ok {
|
|
composes[i].Status = item.Status
|
|
}
|
|
}
|
|
|
|
return composes, nil
|
|
}
|
|
|
|
// Get 获取编排文件和环境变量内容
|
|
func (r *containerComposeRepo) Get(name string) (string, string, error) {
|
|
content, _ := os.ReadFile(filepath.Join(app.Root, "server", "compose", name, "docker-compose.yml"))
|
|
env, _ := os.ReadFile(filepath.Join(app.Root, "server", "compose", name, ".env"))
|
|
return string(content), string(env), nil // 有意忽略错误，这样可以允许新建文件
|
|
}
|
|
|
|
// Create 创建编排文件
|
|
func (r *containerComposeRepo) Create(name, compose, env string) error {
|
|
dir := filepath.Join(app.Root, "server", "compose", name)
|
|
if err := os.MkdirAll(dir, 0644); err != nil {
|
|
return err
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, "docker-compose.yml"), []byte(compose), 0644); err != nil {
|
|
return err
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, ".env"), []byte(env), 0644); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Update 更新编排文件
|
|
func (r *containerComposeRepo) Update(name, compose, env string) error {
|
|
dir := filepath.Join(app.Root, "server", "compose", name)
|
|
if err := os.WriteFile(filepath.Join(dir, "docker-compose.yml"), []byte(compose), 0644); err != nil {
|
|
return err
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, ".env"), []byte(env), 0644); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Up 启动编排
|
|
func (r *containerComposeRepo) Up(name string, force bool) error {
|
|
file := filepath.Join(app.Root, "server", "compose", name, "docker-compose.yml")
|
|
cmd := "docker compose -f %s up -d"
|
|
if force {
|
|
cmd += " --pull always" // 强制拉取镜像
|
|
}
|
|
_, err := shell.Execf(cmd, file)
|
|
return err
|
|
}
|
|
|
|
// Down 停止编排
|
|
func (r *containerComposeRepo) Down(name string) error {
|
|
file := filepath.Join(app.Root, "server", "compose", name, "docker-compose.yml")
|
|
_, err := shell.Execf("docker compose -f %s down", file)
|
|
return err
|
|
}
|
|
|
|
// Remove 删除编排
|
|
func (r *containerComposeRepo) Remove(name string) error {
|
|
dir := filepath.Join(app.Root, "server", "compose", name)
|
|
return os.RemoveAll(dir)
|
|
}
|