WIP
This commit is contained in:
6
.idea/vcs.xml
generated
Normal file
6
.idea/vcs.xml
generated
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="VcsDirectoryMappings">
|
||||||
|
<mapping directory="$PROJECT_DIR$" vcs="Git" />
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
@@ -72,7 +72,7 @@ func installRunE(_ *cobra.Command, args []string) error {
|
|||||||
spaceWorker := space_worker.NewSpaceWorker(logger.LogVerbose(flagVerbose), workDir)
|
spaceWorker := space_worker.NewSpaceWorker(logger.LogVerbose(flagVerbose), workDir)
|
||||||
spaceWorker.AnsibleBin = flagAnsibleBin
|
spaceWorker.AnsibleBin = flagAnsibleBin
|
||||||
spaceWorker.AnsibleVerbose = flagAnsibleVerbose
|
spaceWorker.AnsibleVerbose = flagAnsibleVerbose
|
||||||
return spaceWorker.Install(flagGenOnly, args...)
|
return spaceWorker.Install(flagGenOnly, flagTargetHost, args...)
|
||||||
}
|
}
|
||||||
|
|
||||||
func containsInSlice(element string, slice []string) bool {
|
func containsInSlice(element string, slice []string) bool {
|
||||||
|
|||||||
@@ -81,7 +81,12 @@ func (app *SpaceCreator) CreateSpace(name string, props *ServerProps) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
app.log.Trace("generate playbook.yml")
|
app.log.Trace("generate playbook.yml")
|
||||||
if err := generatePlaybook(spacePath, props.Name); err != nil {
|
if err := generatePlaybook(spacePath); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
app.log.Trace("generate default.host.txt")
|
||||||
|
if err := utils.WriteStringFile(filepath.Join(spacePath, "default.host.txt"), props.Name); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -134,9 +139,9 @@ func generateHosts(spacePath string, props *ServerProps) error {
|
|||||||
return types.WriteHosts(spacePath, &hosts)
|
return types.WriteHosts(spacePath, &hosts)
|
||||||
}
|
}
|
||||||
|
|
||||||
func generatePlaybook(spacePath string, name string) error {
|
func generatePlaybook(spacePath string) error {
|
||||||
playbook := types.Playbook{
|
playbook := types.Playbook{
|
||||||
Hosts: name,
|
Hosts: "all",
|
||||||
GatherFacts: true,
|
GatherFacts: true,
|
||||||
PreTasks: []types.Task{{
|
PreTasks: []types.Task{{
|
||||||
Name: "Include vars",
|
Name: "Include vars",
|
||||||
|
|||||||
@@ -176,9 +176,24 @@ func (app *SpaceWorker) ListRoles() error {
|
|||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
func (app *SpaceWorker) Install(generateOnly bool, roles ...string) error {
|
func (app *SpaceWorker) Install(generateOnly bool, targetHost string, roles ...string) error {
|
||||||
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
|
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
|
||||||
ansibleArgs := setupAnsibleArgs(app.AnsibleVerbose, roles)
|
|
||||||
|
var host string
|
||||||
|
if targetHost != "" {
|
||||||
|
host = targetHost
|
||||||
|
} else {
|
||||||
|
var err error
|
||||||
|
host, err = app.GetDefaultHost()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ansibleArgs, err := app.setupAnsibleArgs(app.AnsibleVerbose, host, roles)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
command := exec.Command(app.AnsibleBin, ansibleArgs...)
|
command := exec.Command(app.AnsibleBin, ansibleArgs...)
|
||||||
command.Dir = app.workDir
|
command.Dir = app.workDir
|
||||||
@@ -196,7 +211,7 @@ func (app *SpaceWorker) Install(generateOnly bool, roles ...string) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func setupAnsibleArgs(verbose uint8, roles []string) []string {
|
func (app *SpaceWorker) setupAnsibleArgs(verbose uint8, targetHost string, roles []string) ([]string, error) {
|
||||||
var ansibleArgs []string
|
var ansibleArgs []string
|
||||||
|
|
||||||
switch verbose {
|
switch verbose {
|
||||||
@@ -209,6 +224,7 @@ func setupAnsibleArgs(verbose uint8, roles []string) []string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
ansibleArgs = append(ansibleArgs, "-i", "hosts.yml")
|
ansibleArgs = append(ansibleArgs, "-i", "hosts.yml")
|
||||||
|
ansibleArgs = append(ansibleArgs, "-l", targetHost)
|
||||||
|
|
||||||
if len(roles) > 0 {
|
if len(roles) > 0 {
|
||||||
for _, role := range roles {
|
for _, role := range roles {
|
||||||
@@ -218,7 +234,7 @@ func setupAnsibleArgs(verbose uint8, roles []string) []string {
|
|||||||
ansibleArgs = append(ansibleArgs, "--extra-vars", "dd_install=true")
|
ansibleArgs = append(ansibleArgs, "--extra-vars", "dd_install=true")
|
||||||
}
|
}
|
||||||
|
|
||||||
return append(ansibleArgs, "playbook.yml")
|
return append(ansibleArgs, "playbook.yml"), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -232,9 +248,19 @@ func (app *SpaceWorker) ListHosts() error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
app.log.Trace("try read default.host.txt")
|
||||||
|
defaultHost, err := utils.ReadStringFile(filepath.Join(app.workDir, "default.host.txt"))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
fmt.Println("Хосты:")
|
fmt.Println("Хосты:")
|
||||||
for serverName, props := range *hosts {
|
for serverName, props := range *hosts {
|
||||||
fmt.Printf(" - %s [%s:%d] (%s)\n", serverName, props.Host, props.Port, props.User)
|
defaultMarker := ""
|
||||||
|
if serverName == defaultHost {
|
||||||
|
defaultMarker = " *"
|
||||||
|
}
|
||||||
|
fmt.Printf(" - %s [%s:%d] (%s)%s\n", serverName, props.Host, props.Port, props.User, defaultMarker)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
@@ -302,20 +328,24 @@ func (app *SpaceWorker) HostRemove(name string) error {
|
|||||||
func (app *SpaceWorker) SetDefaultHost(name string) error {
|
func (app *SpaceWorker) SetDefaultHost(name string) error {
|
||||||
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
|
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
|
||||||
|
|
||||||
app.log.Trace("try read playbook.yml")
|
app.log.Trace("try write default.host.txt")
|
||||||
playbook, err := types.ReadPlaybook(app.workDir)
|
err := utils.WriteStringFile(filepath.Join(app.workDir, "default.host.txt"), name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
app.log.Trace("set default host")
|
|
||||||
playbook.Hosts = name
|
|
||||||
|
|
||||||
app.log.Trace("write playbook.yml")
|
|
||||||
if err := types.WritePlaybook(app.workDir, playbook); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
app.log.Info(fmt.Sprintf("Хост '%s' установлен по-умолчанию", name))
|
app.log.Info(fmt.Sprintf("Хост '%s' установлен по-умолчанию", name))
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (app *SpaceWorker) GetDefaultHost() (string, error) {
|
||||||
|
app.log.Debug(fmt.Sprintf("workDir: %s", app.workDir))
|
||||||
|
|
||||||
|
app.log.Trace("try read default.host.txt")
|
||||||
|
value, err := utils.ReadStringFile(filepath.Join(app.workDir, "default.host.txt"))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return value, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
import "os"
|
import (
|
||||||
|
"bytes"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
func SaveStaticFile(path string, data []byte) error {
|
func SaveStaticFile(path string, data []byte) error {
|
||||||
return os.WriteFile(path, data, 0644)
|
return os.WriteFile(path, data, 0644)
|
||||||
@@ -13,3 +16,19 @@ func CreateDir(path string) error {
|
|||||||
func RemoveFile(path string) error {
|
func RemoveFile(path string) error {
|
||||||
return os.Remove(path)
|
return os.Remove(path)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func WriteStringFile(path string, data string) error {
|
||||||
|
buffer := &bytes.Buffer{}
|
||||||
|
buffer.WriteString(data)
|
||||||
|
|
||||||
|
return os.WriteFile(path, buffer.Bytes(), 0644)
|
||||||
|
}
|
||||||
|
|
||||||
|
func ReadStringFile(path string) (string, error) {
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return string(data), nil
|
||||||
|
}
|
||||||
|
|||||||
Binary file not shown.
Reference in New Issue
Block a user