75 lines
1.3 KiB
Go
75 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ModuleArgs struct {
|
||
|
|
Name string
|
||
|
|
}
|
||
|
|
|
||
|
|
type Response struct {
|
||
|
|
Msg string `json:"msg"`
|
||
|
|
Changed bool `json:"changed"`
|
||
|
|
Failed bool `json:"failed"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func ExitJson(responseBody Response) {
|
||
|
|
returnResponse(responseBody)
|
||
|
|
}
|
||
|
|
|
||
|
|
func FailJson(responseBody Response) {
|
||
|
|
responseBody.Failed = true
|
||
|
|
returnResponse(responseBody)
|
||
|
|
}
|
||
|
|
|
||
|
|
func returnResponse(responseBody Response) {
|
||
|
|
var response []byte
|
||
|
|
var err error
|
||
|
|
response, err = json.Marshal(responseBody)
|
||
|
|
if err != nil {
|
||
|
|
response, _ = json.Marshal(Response{Msg: "Invalid response object"})
|
||
|
|
}
|
||
|
|
fmt.Println(string(response))
|
||
|
|
if responseBody.Failed {
|
||
|
|
os.Exit(1)
|
||
|
|
} else {
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
var response Response
|
||
|
|
|
||
|
|
if len(os.Args) != 2 {
|
||
|
|
response.Msg = "No argument file provided"
|
||
|
|
FailJson(response)
|
||
|
|
}
|
||
|
|
|
||
|
|
argsFile := os.Args[1]
|
||
|
|
|
||
|
|
text, err := ioutil.ReadFile(argsFile)
|
||
|
|
if err != nil {
|
||
|
|
response.Msg = "Could not read configuration file: " + argsFile
|
||
|
|
FailJson(response)
|
||
|
|
}
|
||
|
|
|
||
|
|
var moduleArgs ModuleArgs
|
||
|
|
err = json.Unmarshal(text, &moduleArgs)
|
||
|
|
if err != nil {
|
||
|
|
response.Msg = "Configuration file not valid JSON: " + argsFile
|
||
|
|
FailJson(response)
|
||
|
|
}
|
||
|
|
|
||
|
|
var name string = "World"
|
||
|
|
if moduleArgs.Name != "" {
|
||
|
|
name = moduleArgs.Name
|
||
|
|
}
|
||
|
|
|
||
|
|
response.Msg = "Hello, " + name + "!"
|
||
|
|
ExitJson(response)
|
||
|
|
}
|