11package main
22
33import (
4- "bytes"
54"context"
6- "encoding/json"
75"fmt"
86"io"
97stdlog"log"
4139logFile := viper .GetString ("log-file" )
4240readOnly := viper .GetBool ("read-only" )
4341exportTranslations := viper .GetBool ("export-translations" )
44- prettyPrintJSON := viper .GetBool ("pretty-print-json" )
4542logger ,err := initLogger (logFile )
4643if err != nil {
4744stdlog .Fatal ("Failed to initialize logger:" ,err )
5249logger :logger ,
5350logCommands :logCommands ,
5451exportTranslations :exportTranslations ,
55- prettyPrintJSON :prettyPrintJSON ,
5652}
5753if err := runStdioServer (cfg );err != nil {
5854stdlog .Fatal ("failed to run stdio server:" ,err )
@@ -70,15 +66,13 @@ func init() {
7066rootCmd .PersistentFlags ().Bool ("enable-command-logging" ,false ,"When enabled, the server will log all command requests and responses to the log file" )
7167rootCmd .PersistentFlags ().Bool ("export-translations" ,false ,"Save translations to a JSON file" )
7268rootCmd .PersistentFlags ().String ("gh-host" ,"" ,"Specify the GitHub hostname (for GitHub Enterprise etc.)" )
73- rootCmd .PersistentFlags ().Bool ("pretty-print-json" ,false ,"Pretty print JSON output" )
7469
7570// Bind flag to viper
7671_ = viper .BindPFlag ("read-only" ,rootCmd .PersistentFlags ().Lookup ("read-only" ))
7772_ = viper .BindPFlag ("log-file" ,rootCmd .PersistentFlags ().Lookup ("log-file" ))
7873_ = viper .BindPFlag ("enable-command-logging" ,rootCmd .PersistentFlags ().Lookup ("enable-command-logging" ))
7974_ = viper .BindPFlag ("export-translations" ,rootCmd .PersistentFlags ().Lookup ("export-translations" ))
8075_ = viper .BindPFlag ("gh-host" ,rootCmd .PersistentFlags ().Lookup ("gh-host" ))
81- _ = viper .BindPFlag ("pretty-print-json" ,rootCmd .PersistentFlags ().Lookup ("pretty-print-json" ))
8276
8377// Add subcommands
8478rootCmd .AddCommand (stdioCmd )
@@ -112,20 +106,6 @@ type runConfig struct {
112106logger * log.Logger
113107logCommands bool
114108exportTranslations bool
115- prettyPrintJSON bool
116- }
117-
118- // JSONPrettyPrintWriter is a Writer that pretty prints input to indented JSON
119- type JSONPrettyPrintWriter struct {
120- writer io.Writer
121- }
122-
123- func (j JSONPrettyPrintWriter )Write (p []byte ) (n int ,err error ) {
124- var prettyJSON bytes.Buffer
125- if err := json .Indent (& prettyJSON ,p ,"" ,"\t " );err != nil {
126- return 0 ,err
127- }
128- return j .writer .Write (prettyJSON .Bytes ())
129109}
130110
131111func runStdioServer (cfg runConfig )error {
@@ -179,9 +159,6 @@ func runStdioServer(cfg runConfig) error {
179159in ,out = loggedIO ,loggedIO
180160}
181161
182- if cfg .prettyPrintJSON {
183- out = JSONPrettyPrintWriter {writer :out }
184- }
185162errC <- stdioServer .Listen (ctx ,in ,out )
186163}()
187164