-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Revised config handling, logging and other functions, use latest Matt…
…ermost driver, implement auto reconnect
- Loading branch information
1 parent
0dada2a
commit ee7b102
Showing
14 changed files
with
2,357 additions
and
393 deletions.
There are no files selected for viewing
This file contains 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -3,6 +3,7 @@ builds: | |
| goos: | ||
| - darwin | ||
| - linux | ||
| - windows | ||
| goarch: | ||
| - amd64 | ||
| - arm64 | ||
This file contains 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
This file contains 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,18 +1,21 @@ | ||
| ROLES=govc,oc,terraform,tf | ||
| MATTERMOST_HOST=mattermost-host.tld | ||
| MATTERMOST_EMAIL=email@account.tld | ||
| MATTERMOST_USER=bpt-username | ||
| MATTERMOST_PASSWORD=secretpassword | ||
| MATTERMOST_TEAM=TEAM | ||
| MATTERMOST_CHANNEL=bot-channel | ||
| PRIVATEBIN_ENABLE=true | ||
| PRIVATEBIN_HOST=https://paste.yourdomain.tld | ||
| PRIVATEBIN_FORMATTER=syntaxhighlighting | ||
| PRIVATEBIN_EXPIRE=6days | ||
| PRIVATEBIN_OPENDISCUSSION=false | ||
| PRIVATEBIN_BURNAFTERREADING=false | ||
| PRIVATEBIN_PASSWORD= | ||
| GOVC_URL=https://vcsa.domain | ||
| GOVC_DATACENTER=datacenter | ||
| GOVC_USERNAME=vcsa_username | ||
| GOVC_PASSWORD=secretpassword | ||
| MB_DEBUG=true | ||
| MB_ROLES=govc,oc,terraform,tf | ||
| MB_MATTERMOST_URL=mattermost.example.com | ||
| MB_MATTERMOST_USERTOKEN=mysecretusertoken | ||
| MB_MATTERMOST_TEAM=test | ||
| MB_MATTERMOST_CHANNEL=town-square | ||
| MB_PRIVATEBIN_ENABLE=true | ||
| MB_PRIVATEBIN_HOST=paste.example.com | ||
| MB_PRIVATEBIN_FORMATTER=syntaxhighlighting | ||
| MB_PRIVATEBIN_EXPIRE=6days | ||
| MB_PRIVATEBIN_OPENDISCUSSION=false | ||
| MB_PRIVATEBIN_BURNAFTERREADING=false | ||
| MB_PRIVATEBIN_PASSWORD= | ||
| MB_GOVC_HOST=vcenter-server-appliance-endpoint.example.com | ||
| MB_GOVC_DATACENTER=mydatacenter | ||
| MB_GOVC_USERNAME=myvsphereusername | ||
| MB_GOVC_PASSWORD=mysecretpasswordformyvsphereusername | ||
| MB_TERRAFORM_EXECUTABLE=/usr/bin/terraform | ||
| MB_MB_TERRAFORM_WRAPPER_EXECUTABLE=/opt/terraform/tf.sh | ||
| MB_OC_EXECUTABLE=/usr/local/bin/oc | ||
| MB_GOVC_EXECUTABLE=/usr/local/bin/govc |
This file contains 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,205 @@ | ||
| package confighandler | ||
|
|
||
| import ( | ||
| "log" | ||
| "os" | ||
|
|
||
| _ "github.com/joho/godotenv/autoload" | ||
| "github.com/mattermost/mattermost-server/v6/model" | ||
| "github.com/rs/zerolog" | ||
| "github.com/spf13/viper" | ||
| ) | ||
|
|
||
| var version = "DEV" | ||
| var App = &Application{} | ||
|
|
||
| var Settings Config | ||
|
|
||
| // application struct to hold the dependencies for our bot | ||
| type Application struct { | ||
| Config Config | ||
| Logger zerolog.Logger | ||
| MattermostClient *model.Client4 | ||
| MattermostWebSocketClient *model.WebSocketClient | ||
| MattermostUser *model.User | ||
| MattermostChannel *model.Channel | ||
| MattermostTeam *model.Team | ||
| } | ||
|
|
||
| type Config struct { | ||
| MB_ROLES string | ||
| MB_MATTERMOST_URL string | ||
| MB_MATTERMOST_USER string | ||
| MB_MATTERMOST_USERTOKEN string | ||
| MB_MATTERMOST_TEAM string | ||
| MB_MATTERMOST_CHANNEL string | ||
| MB_PRIVATEBIN_ENABLE string | ||
| MB_PRIVATEBIN_HOST string | ||
| MB_PRIVATEBIN_FORMATTER string | ||
| MB_PRIVATEBIN_EXPIRE string | ||
| MB_PRIVATEBIN_OPENDISCUSSION string | ||
| MB_PRIVATEBIN_BURNAFTERREADING string | ||
| MB_PRIVATEBIN_PASSWORD string | ||
| MB_TERRAFORM_EXECUTABLE string | ||
| MB_TERRAFORM_WRAPPER_EXECUTABLE string | ||
| MB_OC_EXECUTABLE string | ||
| MB_GOVC_EXECUTABLE string | ||
| MB_GOVC_HOST string | ||
| MB_GOVC_DATACENTER string | ||
| MB_GOVC_USERNAME string | ||
| MB_GOVC_PASSWORD string | ||
| } | ||
|
|
||
| func Init() { | ||
| // App = &Application{ | ||
| // Logger: zerolog.New( | ||
| // zerolog.ConsoleWriter{ | ||
| // Out: os.Stdout, | ||
| // TimeFormat: time.RFC822, | ||
| // NoColor: true, | ||
| // }, | ||
| // ).With().Timestamp().Logger(), | ||
| // } | ||
|
|
||
| App = &Application{ | ||
| Logger: zerolog.New( | ||
| os.Stdout, | ||
| ).With().Timestamp().Logger(), | ||
| } | ||
|
|
||
| App.Config = LoadConfig() | ||
| } | ||
|
|
||
| func LoadConfig() Config { | ||
| App.Logger.Info().Msg("Mattermost-Bot (v" + version + ") started.") | ||
|
|
||
| viper.SetConfigName("config") // name of config file (without extension) | ||
| viper.SetConfigType("env") // REQUIRED if the config file does not have the extension in the name | ||
| //viper.AddConfigPath("/etc/bitbucket-mattermost-notifier/") // path to look for the config file in | ||
| viper.AddConfigPath(".") // optionally look for config in the working directory | ||
| viper.AutomaticEnv() | ||
| if err := viper.ReadInConfig(); err != nil { | ||
| if _, ok := err.(viper.ConfigFileNotFoundError); ok { | ||
| // Config file not found; ignore error if desired | ||
| } else { | ||
| // Config file was found but another error was produced | ||
| log.Fatal("Fatal error config file: %w", err) | ||
| } | ||
| } | ||
| // If MB_DEBUG is true, enable debug mode | ||
| zerolog.SetGlobalLevel(zerolog.InfoLevel) | ||
| if GetConfigBoolValue("MB_DEBUG") { | ||
| zerolog.SetGlobalLevel(zerolog.DebugLevel) | ||
| } | ||
| // if !checkParams("MATTERMOST_WEBHOOKURL", "MATTERMOST_CHANNEL") { | ||
| // log.Fatal("MATTERMOST_WEBHOOKURL or MATTERMOST_CHANNEL can NOT be empty") | ||
| // } | ||
| // settings.mattermostTeamName = os.Getenv("MM_TEAM") | ||
| // settings.mattermostUserName = os.Getenv("MM_USERNAME") | ||
| // settings.mattermostToken = os.Getenv("MM_TOKEN") | ||
| // settings.mattermostChannel = os.Getenv("MM_CHANNEL") | ||
| // settings.mattermostServer, _ = url.Parse(os.Getenv("MM_SERVER")) | ||
|
|
||
| settings := map[string]string{ | ||
| "MB_ROLES": GetConfigValue("MB_ROLES"), | ||
| "MB_MATTERMOST_URL": GetConfigValue("MB_MATTERMOST_URL"), | ||
| "MB_MATTERMOST_USERTOKEN": GetConfigValue("MB_MATTERMOST_USERTOKEN"), | ||
| "MB_MATTERMOST_TEAM": GetConfigValue("MB_MATTERMOST_TEAM"), | ||
| "MB_MATTERMOST_CHANNEL": GetConfigValue("MB_MATTERMOST_CHANNEL"), | ||
| "MB_PRIVATEBIN_ENABLE": GetConfigValue("MB_PRIVATEBIN_ENABLE"), | ||
| "MB_PRIVATEBIN_HOST": GetConfigValue("MB_PRIVATEBIN_HOST"), | ||
| "MB_PRIVATEBIN_FORMATTER": GetConfigValue("MB_PRIVATEBIN_FORMATTER"), | ||
| "MB_PRIVATEBIN_EXPIRE": GetConfigValue("MB_PRIVATEBIN_EXPIRE"), | ||
| "MB_PRIVATEBIN_OPENDISCUSSION": GetConfigValue("MB_PRIVATEBIN_OPENDISCUSSION"), | ||
| "MB_PRIVATEBIN_BURNAFTERREADING": GetConfigValue("MB_PRIVATEBIN_BURNAFTERREADING"), | ||
| "MB_PRIVATEBIN_PASSWORD": GetConfigValue("MB_PRIVATEBIN_PASSWORD"), | ||
| "MB_TERRAFORM_EXECUTABLE": GetConfigValue("MB_TERRAFORM_EXECUTABLE"), | ||
| "MB_TERRAFORM_WRAPPER_EXECUTABLE": GetConfigValue("MB_TERRAFORM_WRAPPER_EXECUTABLE"), | ||
| "MB_OC_EXECUTABLE": GetConfigValue("MB_OC_EXECUTABLE"), | ||
| "MB_GOVC_EXECUTABLE": GetConfigValue("MB_GOVC_EXECUTABLE"), | ||
| "MB_GOVC_HOST": GetConfigValue("MB_GOVC_HOST"), | ||
| "MB_GOVC_DATACENTER": GetConfigValue("MB_GOVC_DATACENTER"), | ||
| "MB_GOVC_USERNAME": GetConfigValue("MB_GOVC_USERNAME"), | ||
| "MB_GOVC_PASSWORD": GetConfigValue("MB_GOVC_PASSWORD"), | ||
| } | ||
|
|
||
| // Settings.MB_ROLES = GetConfigValue("MB_ROLES") | ||
| // Settings.MB_MATTERMOST_URL = GetConfigValue("MB_MATTERMOST_URL") | ||
| // Settings.MB_MATTERMOST_USERTOKEN = GetConfigValue("MB_MATTERMOST_USERTOKEN") | ||
| // Settings.MB_MATTERMOST_TEAM = GetConfigValue("MB_MATTERMOST_TEAM") | ||
| // Settings.MB_MATTERMOST_CHANNEL = GetConfigValue("MB_MATTERMOST_CHANNEL") | ||
| // Settings.MB_PRIVATEBIN_ENABLE = GetConfigBoolValue("MB_PRIVATEBIN_ENABLE") | ||
| // Settings.MB_PRIVATEBIN_HOST = GetConfigValue("MB_PRIVATEBIN_HOST") | ||
| // Settings.MB_PRIVATEBIN_FORMATTER = GetConfigValue("MB_PRIVATEBIN_FORMATTER") | ||
| // Settings.MB_PRIVATEBIN_EXPIRE = GetConfigValue("MB_PRIVATEBIN_EXPIRE") | ||
| // Settings.MB_PRIVATEBIN_OPENDISCUSSION = GetConfigBoolValue("MB_PRIVATEBIN_OPENDISCUSSION") | ||
| // Settings.MB_PRIVATEBIN_BURNAFTERREADING = GetConfigBoolValue("MB_PRIVATEBIN_BURNAFTERREADING") | ||
| // Settings.MB_PRIVATEBIN_PASSWORD = GetConfigValue("MB_PRIVATEBIN_PASSWORD") | ||
| // Settings.MB_TERRAFORM_EXECUTABLE = GetConfigValue("MB_TERRAFORM_EXECUTABLE") | ||
| // Settings.MB_TERRAFORM_WRAPPER_EXECUTABLE = GetConfigValue("MB_TERRAFORM_WRAPPER_EXECUTABLE") | ||
| // Settings.MB_OC_EXECUTABLE = GetConfigValue("MB_OC_EXECUTABLE") | ||
| // Settings.MB_GOVC_EXECUTABLE = GetConfigValue("MB_GOVC_EXECUTABLE") | ||
| // Settings.MB_GOVC_HOST = GetConfigValue("MB_GOVC_HOST") | ||
| // Settings.MB_GOVC_DATACENTER = GetConfigValue("MB_GOVC_DATACENTER") | ||
| // Settings.MB_GOVC_USERNAME = GetConfigValue("MB_GOVC_USERNAME") | ||
| // Settings.MB_GOVC_PASSWORD = GetConfigValue("MB_GOVC_PASSWORD") | ||
|
|
||
| // TODO: No secret credentials in output | ||
| //App.Logger.Info().Msg(fmt.Sprint(Settings)) | ||
|
|
||
| // Output all keys and values of the current loaded config | ||
| message := "Current configuration: " | ||
| for key, value := range settings { | ||
| if key == "MB_MATTERMOST_USERTOKEN" || key == "MB_PRIVATEBIN_PASSWORD" || key == "MB_GOVC_PASSWORD" { | ||
| continue | ||
| } | ||
| message += key + "=" + value + " " | ||
| } | ||
| message = message[:len(message)-1] | ||
|
|
||
| App.Logger.Info().Msg(message) | ||
|
|
||
| Settings := Config{ | ||
| MB_ROLES: settings["MB_ROLES"], | ||
| MB_MATTERMOST_URL: settings["MB_MATTERMOST_URL"], | ||
| MB_MATTERMOST_USERTOKEN: settings["MB_MATTERMOST_USERTOKEN"], | ||
| MB_MATTERMOST_TEAM: settings["MB_MATTERMOST_TEAM"], | ||
| MB_MATTERMOST_CHANNEL: settings["MB_MATTERMOST_CHANNEL"], | ||
| MB_PRIVATEBIN_ENABLE: settings["MB_PRIVATEBIN_ENABLE"], | ||
| MB_PRIVATEBIN_HOST: settings["MB_PRIVATEBIN_HOST"], | ||
| MB_PRIVATEBIN_FORMATTER: settings["MB_PRIVATEBIN_FORMATTER"], | ||
| MB_PRIVATEBIN_EXPIRE: settings["MB_PRIVATEBIN_EXPIRE"], | ||
| MB_PRIVATEBIN_OPENDISCUSSION: settings["MB_PRIVATEBIN_OPENDISCUSSION"], | ||
| MB_PRIVATEBIN_BURNAFTERREADING: settings["MB_PRIVATEBIN_BURNAFTERREADING"], | ||
| MB_PRIVATEBIN_PASSWORD: settings["MB_PRIVATEBIN_PASSWORD"], | ||
| MB_TERRAFORM_EXECUTABLE: settings["MB_TERRAFORM_EXECUTABLE"], | ||
| MB_TERRAFORM_WRAPPER_EXECUTABLE: settings["MB_TERRAFORM_WRAPPER_EXECUTABLE"], | ||
| MB_OC_EXECUTABLE: settings["MB_OC_EXECUTABLE"], | ||
| MB_GOVC_EXECUTABLE: settings["MB_GOVC_EXECUTABLE"], | ||
| MB_GOVC_HOST: settings["MB_GOVC_HOST"], | ||
| MB_GOVC_DATACENTER: settings["MB_GOVC_DATACENTER"], | ||
| MB_GOVC_USERNAME: settings["MB_GOVC_USERNAME"], | ||
| MB_GOVC_PASSWORD: settings["MB_GOVC_PASSWORD"], | ||
| } | ||
|
|
||
| return Settings | ||
| } | ||
|
|
||
| func checkParams(params ...string) bool { | ||
| for _, param := range params { | ||
| if !viper.IsSet(param) { | ||
| return false | ||
| } | ||
| } | ||
| return true | ||
| } | ||
|
|
||
| func GetConfigValue(key string) string { | ||
| //LoadConfig() | ||
| return viper.GetString(key) | ||
| } | ||
|
|
||
| func GetConfigBoolValue(key string) bool { | ||
| //LoadConfig() | ||
| return viper.GetBool(key) | ||
| } |
Oops, something went wrong.