Created
October 5, 2018 16:00
-
-
Save aweiteka/969bd54466112ce139f688de2b716fad to your computer and use it in GitHub Desktop.
Golang CLI that can also be used as an ansible binary module
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment