initial commit

This commit is contained in:
2021-10-21 02:51:32 +02:00
commit d9c1c42150
30 changed files with 1617 additions and 0 deletions

View File

@ -0,0 +1,164 @@
package compose
import (
"bytes"
"errors"
"fmt"
"github.com/Masterminds/semver"
"gopkg.in/yaml.v3"
"io/ioutil"
"ledo/app/modules/context"
"log"
"os/exec"
"regexp"
"strings"
)
const DockerComposeVersion = ">= 1.28.6"
func CheckDockerComposeVersion() {
cmd := exec.Command("docker-compose", "--version")
var output bytes.Buffer
cmd.Stdout = &output
err := cmd.Run()
if err != nil {
log.Fatal("No docker-compose installed. Please install docker-compose ie. via `pip3 install docker-compose`")
}
r := regexp.MustCompile("(.*){1}(version\\ ){1}(([0-9]+)\\.([0-9]+)\\.([0-9]+))")
result := r.FindStringSubmatch(output.String())
composeVersion := result[3]
verConstraint, _ := semver.NewConstraint(DockerComposeVersion)
composeSemVer, _ := semver.NewVersion(composeVersion)
if !verConstraint.Check(composeSemVer) {
log.Fatal("Wrong docker-compose version, please update to 1.28.6 or higher.")
}
}
func MergeComposerFiles(filenames ...string) (string, error) {
var resultValues map[string]interface{}
if len(filenames) <= 0 {
return "", errors.New("You must provide at least one filename for reading Values")
}
for _, filename := range filenames {
var override map[string]interface{}
bs, err := ioutil.ReadFile(filename)
if err != nil {
log.Print(err)
continue
}
if err := yaml.Unmarshal(bs, &override); err != nil {
log.Print(err)
continue
}
if resultValues == nil {
resultValues = override
} else {
for k, v := range override {
resultValues[k] = v
}
}
}
bs, err := yaml.Marshal(resultValues)
if err != nil {
log.Fatal(err)
return "", err
}
return string(bs), nil
}
func ShowDockerImageFQN(ctx *context.LedoContext) string {
return fmt.Sprintf("%s/%s/%s/master:latest", ctx.Config.Docker.Registry, ctx.Config.Docker.Namespace, ctx.Config.Docker.Name)
}
func ExecComposerUp(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "up", "-d")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerPull(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "pull")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerStop(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "stop")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerBuild(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "build", "--pull")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerDown(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "down")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerStart(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "start")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerRestart(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "restart")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerLogs(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "logs", "--follow", "--tail", "100")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerPs(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "ps")
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerShell(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "exec", strings.ToLower(ctx.Config.Docker.MainService), ctx.Config.Docker.Shell)
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerDebug(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "run", "--entrypoint=", strings.ToLower(ctx.Config.Docker.MainService), ctx.Config.Docker.Shell)
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerRun(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "run", strings.ToLower(ctx.Config.Docker.MainService), ctx.Config.Docker.Shell)
ctx.ExecCmd("docker-compose", args[0:])
}
func ExecComposerUpOnce(ctx *context.LedoContext) {
args := ctx.ComposeArgs
args = append(args, "up", "--force-recreate", "--renew-anon-volumes", "--abort-on-container-exit", "--exit-code-from", ctx.Config.Docker.MainService)
ctx.ExecCmd("docker-compose", args[0:])
}

View File

@ -0,0 +1,33 @@
package config
import (
"gopkg.in/yaml.v3"
"io/ioutil"
)
type LedoFile struct {
Docker DockerMap `yaml:"docker,omitempty"`
Modes map[string]string `yaml:"modes,omitempty"`
Project string `yaml:"project,omitempty"`
}
type DockerMap struct {
Registry string `yaml:"registry,omitempty"`
Namespace string `yaml:"namespace,omitempty"`
Name string `yaml:"name,omitempty"`
MainService string `yaml:"main_service,omitempty"`
Shell string `yaml:"shell,omitempty"`
}
func NewLedoFile(s string) (*LedoFile, error) {
yamlFile, err := ioutil.ReadFile(s)
if err != nil {
return nil, err
}
t := &LedoFile{}
err = yaml.Unmarshal(yamlFile, t)
if err != nil {
return nil, err
}
return t, nil
}

View File

@ -0,0 +1,66 @@
package context
import (
"fmt"
"github.com/urfave/cli/v2"
"ledo/app/modules/config"
"ledo/app/modules/mode"
"os"
"os/exec"
"strings"
)
type LedoContext struct {
*cli.Context
Config *config.LedoFile
ComposeArgs []string
Mode mode.Mode
Output string
}
func InitCommand(ctx *cli.Context) *LedoContext {
var (
c LedoContext
cfg *config.LedoFile
)
configYml := ".ledo.yml"
modeYml := ".ledo-mode"
// Compat with jzcli (StreamSage and Jazzy deployment tool)
if _, err := os.Stat(".jz-project.yml"); err == nil {
configYml = ".jz-project.yml"
modeYml = ".jz-mode"
}
mode := mode.InitMode(modeYml, configYml)
c.Mode = mode
c.Output = ctx.String("output")
cfg, _ = config.NewLedoFile(configYml)
c.Config = cfg
args := []string{"--env-file", ".env"}
args = append(args, "--project-name", strings.ToLower(strings.Replace(c.Config.Docker.Name, "/", "-", -1)))
composes, _ := mode.GetModeConfig()
for _, element := range composes {
args = append(args, "-f")
args = append(args, element)
}
c.ComposeArgs = args
return &c
}
func (lx *LedoContext) ExecCmd(cmd string, cmdArgs []string) error {
fmt.Printf("Execute: %v %v\n", cmd, strings.Join(cmdArgs, " "))
command := exec.Command(cmd, cmdArgs...)
command.Stdin = os.Stdin
command.Stdout = os.Stdout
command.Stderr = os.Stderr
return command.Run()
}

View File

@ -0,0 +1,32 @@
package interact
import (
survey "github.com/AlecAivazis/survey/v2"
"github.com/thoas/go-funk"
"ledo/app/modules/context"
)
func SelectMode(context *context.LedoContext, selectedMode string) (string, error) {
mode := context.Mode
if selectedMode == "" {
availableModes := mode.GetModes()
// the questions to ask
var qs = []*survey.Question{
{
Name: "providers",
Prompt: &survey.Select{
Message: "Select run mode",
PageSize: 10,
Options: funk.Keys(availableModes).([]string),
},
},
}
err := survey.Ask(qs, &selectedMode)
if err != nil {
return "", err
}
}
_, err := mode.SetMode(selectedMode)
return selectedMode, err
}

109
app/modules/mode/mode.go Normal file
View File

@ -0,0 +1,109 @@
package mode
import (
"errors"
"fmt"
"github.com/spf13/viper"
"io/ioutil"
"os"
"path/filepath"
"strings"
)
type Mode struct {
modeCfg string
CurrentMode string
availableModes map[string]string
}
func InitMode(modeFileName string, cfgFile string) Mode {
wd, err := os.Getwd()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
modeFile := filepath.FromSlash(filepath.Join(wd, modeFileName))
if _, err := os.Stat(modeFile); os.IsNotExist(err) {
err := ioutil.WriteFile(modeFile, []byte("dev"), 0644)
if err != nil {
fmt.Printf("Unable to write file: %v", err)
}
}
currentModeFromFile, err := ioutil.ReadFile(modeFile)
currentMode := strings.TrimSpace(string(currentModeFromFile))
// TODO: dirty implement mode read
// Start dirty code ;]
viper.AddConfigPath(wd)
viper.SetConfigFile(cfgFile)
viper.AutomaticEnv()
_ = viper.ReadInConfig()
elements := viper.Get("modes")
// End dirty code :)
var el = make(map[string]string)
for mode, cfg := range elements.(map[string]interface{}) {
el[mode] = fmt.Sprintf("%v", cfg)
}
return Mode{
CurrentMode: string(currentMode),
availableModes: el,
modeCfg: modeFileName,
}
}
func (c *Mode) GetMode() string {
return c.CurrentMode
}
func (c *Mode) PrintListModes() {
fmt.Printf("Available modes:\n")
for m, _ := range c.availableModes {
fmt.Printf("- %v\n", m)
}
}
func (c *Mode) CheckMode(mode string) (bool, error) {
for m := range c.availableModes {
if m == mode {
return true, nil
}
}
return false, errors.New("Selected mode not exists in config file")
}
func (c *Mode) GetModes() map[string]string {
return c.availableModes
}
func (c *Mode) SetMode(modeName string) (bool, error) {
_, err := c.CheckMode(modeName)
if err != nil {
return false, err
}
wd, err := os.Getwd()
if err != nil {
return false, errors.New("Get working directory error!?")
}
modeFile := filepath.FromSlash(filepath.Join(wd, c.modeCfg))
wrErr := ioutil.WriteFile(modeFile, []byte(modeName), 0644)
if wrErr != nil {
return false, errors.New("Write mode error")
}
return true, nil
}
func (c *Mode) GetModeConfig() ([]string, error) {
_, err := c.CheckMode(c.CurrentMode);
if err == nil {
composes := strings.Fields(c.availableModes[c.CurrentMode])
return composes, nil
}
return nil, errors.New("Selected mode not exists in config file")
}