Skip to content

Commit 48af820

Browse files
committed
main: remove unnecessary buffer
1 parent 08f3c07 commit 48af820

File tree

1 file changed

+9
-14
lines changed

1 file changed

+9
-14
lines changed

main.go

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
package main
22

33
import (
4-
"bytes"
54
"context"
65
"flag"
76
"fmt"
@@ -170,22 +169,10 @@ func execPerTable(ctx context.Context, db db.Database) error {
170169
}
171170

172171
func tableToMarkdown(ctx context.Context, db db.Database, w io.Writer, schema, tableName string) error {
173-
var buf bytes.Buffer
174-
if conf.Documentation.Stdout {
175-
w = io.MultiWriter(w, &buf)
176-
defer func() {
177-
out, err := glamour.Render(buf.String(), "dark")
178-
if err != nil {
179-
fmt.Fprintf(os.Stderr, "rendering markdown to terminal: %v\n", err)
180-
return
181-
}
182-
fmt.Println(out)
183-
}()
184-
}
185172
md := markdown.NewMarkdown(w)
186173

187174
md.H1(markdown.Code(tableName))
188-
md.Text("\n")
175+
md.PlainText("\n")
189176
mdTableRows := [][]string{}
190177
columns, err := db.ListColumns(ctx, schema, tableName)
191178
if err != nil {
@@ -211,6 +198,14 @@ func tableToMarkdown(ctx context.Context, db db.Database, w io.Writer, schema, t
211198
Header: []string{"Name", "Type", "Nullable", "Default"},
212199
Rows: mdTableRows,
213200
})
201+
if conf.Documentation.Stdout {
202+
out, err := glamour.Render(md.String(), "dark")
203+
if err != nil {
204+
fmt.Fprintf(os.Stderr, "Error rendering markdown to terminal: %v\n", err)
205+
} else {
206+
fmt.Println(out)
207+
}
208+
}
214209

215210
return md.Build()
216211
}

0 commit comments

Comments
 (0)