add config: allow surrounding output with backticks (#4)

config/config.go

@@ -10,6 +10,7 @@ const (
 	ConnectionStringKey            = "connectionString"
 	ConnectionStringSuggestionsKey = "connectionStringSuggestions"
 	OutputFileNameKey              = "outputFileName"
+	EncloseWithMermaidBackticksKey = "encloseWithMermaidBackticks"
 )
 
 func ShowAllConstraints() bool {
@@ -39,3 +40,7 @@ func ConnectionStringSuggestions() []string {
 func SelectedTables() []string {
 	return viper.GetStringSlice(SelectedTablesKey)
 }
+
+func EncloseWithMermaidBackticks() bool {
+	return viper.GetBool(EncloseWithMermaidBackticksKey)
+}

diagram/diagram.go

@@ -18,6 +18,13 @@ func Create(result *database.Result) error {
 	defer f.Close()
 
 	buffer := bufio.NewWriter(f)
+
+	if config.EncloseWithMermaidBackticks() {
+		_, err = buffer.WriteString("```mermaid\n")
+		if err != nil {
+			return err
+		}
+	}
 	_, err = buffer.WriteString("erDiagram\n")
 	if err != nil {
 		return err
@@ -46,6 +53,11 @@ func Create(result *database.Result) error {
 		if err != nil {
 			return err
 		}
+
+		if config.EncloseWithMermaidBackticks() {
+			_, err = buffer.WriteString("\n```")
+		}
+
 		_, err = buffer.WriteString("\n\n")
 
 	}

go.mod

@@ -13,32 +13,8 @@ require (
 )
 
 require (
-	github.com/fsnotify/fsnotify v1.5.1 // indirect
-	github.com/hashicorp/hcl v1.0.0 // indirect
-	github.com/inconshreveable/mousetrap v1.0.0 // indirect
-	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
-	github.com/jackc/pgconn v1.11.0 // indirect
-	github.com/jackc/pgio v1.0.0 // indirect
-	github.com/jackc/pgpassfile v1.0.0 // indirect
-	github.com/jackc/pgproto3/v2 v2.2.0 // indirect
-	github.com/jackc/pgservicefile v0.0.0-20200714003250-2b9c44734f2b // indirect
-	github.com/jackc/pgtype v1.10.0 // indirect
-	github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 // indirect
-	github.com/magiconair/properties v1.8.5 // indirect
-	github.com/mattn/go-colorable v0.1.12 // indirect
-	github.com/mattn/go-isatty v0.0.14 // indirect
 	github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d // indirect
-	github.com/mitchellh/mapstructure v1.4.3 // indirect
-	github.com/pelletier/go-toml v1.9.4 // indirect
-	github.com/spf13/afero v1.6.0 // indirect
-	github.com/spf13/cast v1.4.1 // indirect
-	github.com/spf13/jwalterweatherman v1.1.0 // indirect
-	github.com/spf13/pflag v1.0.5 // indirect
-	github.com/subosito/gotenv v1.2.0 // indirect
 	golang.org/x/crypto v0.0.0-20220307211146-efcb8507fb70 // indirect
 	golang.org/x/sys v0.0.0-20220310020820-b874c991c1a5 // indirect
 	golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 // indirect
-	golang.org/x/text v0.3.7 // indirect
-	gopkg.in/ini.v1 v1.66.2 // indirect
-	gopkg.in/yaml.v2 v2.4.0 // indirect
 )