mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-12-14 16:49:34 +08:00
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package docker
|
|
|
|
import (
|
|
"context"
|
|
"github.com/compose-spec/compose-go/v2/loader"
|
|
"github.com/compose-spec/compose-go/v2/types"
|
|
"github.com/docker/compose/v2/pkg/api"
|
|
"github.com/joho/godotenv"
|
|
"path"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type ComposeService struct {
|
|
api.Service
|
|
project *types.Project
|
|
}
|
|
|
|
func GetComposeProject(projectName, workDir string, yml []byte, env []byte, skipNormalization bool) (*types.Project, error) {
|
|
var configFiles []types.ConfigFile
|
|
configFiles = append(configFiles, types.ConfigFile{
|
|
Filename: "docker-compose.yml",
|
|
Content: yml},
|
|
)
|
|
envMap, err := godotenv.UnmarshalBytes(env)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
details := types.ConfigDetails{
|
|
WorkingDir: workDir,
|
|
ConfigFiles: configFiles,
|
|
Environment: envMap,
|
|
}
|
|
projectName = strings.ToLower(projectName)
|
|
reg, _ := regexp.Compile(`[^a-z0-9_-]+`)
|
|
projectName = reg.ReplaceAllString(projectName, "")
|
|
project, err := loader.LoadWithContext(context.Background(), details, func(options *loader.Options) {
|
|
options.SetProjectName(projectName, true)
|
|
options.ResolvePaths = true
|
|
options.SkipNormalization = skipNormalization
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
project.ComposeFiles = []string{path.Join(workDir, "docker-compose.yml")}
|
|
return project, nil
|
|
}
|
|
|
|
type ComposeProject struct {
|
|
Version string
|
|
Services map[string]Service `yaml:"services"`
|
|
}
|
|
|
|
type Service struct {
|
|
Image string `yaml:"image"`
|
|
}
|
|
|
|
func GetDockerComposeImages(projectName string, env, yml []byte) ([]string, error) {
|
|
var (
|
|
configFiles []types.ConfigFile
|
|
images []string
|
|
)
|
|
configFiles = append(configFiles, types.ConfigFile{
|
|
Filename: "docker-compose.yml",
|
|
Content: yml},
|
|
)
|
|
envMap, err := godotenv.UnmarshalBytes(env)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
details := types.ConfigDetails{
|
|
ConfigFiles: configFiles,
|
|
Environment: envMap,
|
|
}
|
|
|
|
project, err := loader.LoadWithContext(context.Background(), details, func(options *loader.Options) {
|
|
options.SetProjectName(projectName, true)
|
|
options.ResolvePaths = true
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, service := range project.AllServices() {
|
|
images = append(images, service.Image)
|
|
}
|
|
return images, nil
|
|
}
|