Finished generate.go, modified main.go

I completed the process of generating the configuration file for
idiots like me who like to delete configuration files. Don't
worry, it's only by accident ;)

I also modified main.go so that the generating process actually
becomes useful.

What happens is that the program checks if the
configuration file exists (which will have to change to become
more modular as I have it set to look in the current directory
you're in) and if it doesn't, it goes ahead creates it and inputs
some random junk in there that is actually supposed to help with
your own configuration.
This commit is contained in:
Wyatt J. Miller 2019-09-29 13:38:58 -04:00
parent df6f5b4973
commit dbde9b3cab
2 changed files with 24 additions and 8 deletions

View File

@ -19,29 +19,38 @@ func doesConfigExist() bool {
if err != nil { if err != nil {
return false return false
} else {
return true
} }
return true
} }
func createTemplateConfig(result bool) { func createTemplateConfig(result bool) {
if result == false { if result == false {
os.Create(config) os.Create(config)
} else { } else {
fmt.Println() fmt.Println("Configuration file already created")
} }
} }
func writeTemplateConfig(filename string) { func writeTemplateConfig(filename string) {
file := []byte(filename) hosts := make([]string, 1)
_ = ioutil.WriteFile("config.yaml", file, 0600) logs := make([]string, 1)
config := Configuration{} config := Configuration{
"jbob",
"jbobisawesome",
22,
hosts,
logs,
}
_, err := yaml.Marshal(config) data, err := yaml.Marshal(&config)
fmt.Printf(string(data))
_ = ioutil.WriteFile(filename, data, 0600)
if err != nil { if err != nil {
panic(err) fmt.Errorf("Couldn't create the configuration file", err)
} }
fmt.Println("Template for logman's configuration complete!") fmt.Println("Template for logman's configuration complete!")

View File

@ -3,6 +3,13 @@ package main
import "log" import "log"
func main() { func main() {
isCreated := doesConfigExist()
if isCreated == false {
createTemplateConfig(isCreated)
writeTemplateConfig("config.yaml")
}
config := initializeConfig("config.yaml") config := initializeConfig("config.yaml")
sshConn, sshConfig := initializeConnection(config) sshConn, sshConfig := initializeConnection(config)
clientConns := sshConn.dialConnection(sshConfig) clientConns := sshConn.dialConnection(sshConfig)