New command line flag: --fix (#554)

Author: nene

README.md

@@ -106,6 +106,7 @@ optional arguments:
   -h, --help      show this help message and exit
   -o, --output    OUTPUT
                     File to write SQL output (defaults to stdout)
+  --fix           Update the file in-place
   -l, --language  {bigquery,db2,hive,mariadb,mysql,n1ql,plsql,postgresql,redshift,singlestoredb,snowflake,spark,sql,sqlite,trino,tsql}
                     SQL dialect (defaults to basic sql)
   -c, --config    CONFIG

bin/sql-formatter-cli.cjs

@@ -16,7 +16,7 @@ class PrettierSQLArgs {
 
     this.query = this.getInput();
     const formattedQuery = format(this.query, this.cfg).trim() + '\n';
-    this.writeOutput(this.args.output, formattedQuery);
+    this.writeOutput(this.getOutputFile(this.args), formattedQuery);
   }
 
   getParser() {
@@ -35,6 +35,12 @@ class PrettierSQLArgs {
       help: 'File to write SQL output (defaults to stdout)',
     });
 
+    parser.add_argument('--fix', {
+      help: 'Update the file in-place',
+      action: 'store_const',
+      const: true,
+    });
+
     parser.add_argument('-l', '--language', {
       help: 'SQL Formatter dialect (defaults to basic sql)',
       choices: supportedDialects,
@@ -104,6 +110,22 @@ class PrettierSQLArgs {
     }
   }
 
+  getOutputFile(args) {
+    if (args.output && args.fix) {
+      console.error('Error: Cannot use both --output and --fix options simultaneously');
+      process.exit(1);
+    }
+    if (args.fix && !args.file) {
+      console.error('Error: The --fix option cannot be used without a filename');
+      process.exit(1);
+    }
+    if (args.fix) {
+      return args.file;
+    } else {
+      return args.output;
+    }
+  }
+
   writeOutput(file, query) {
     if (!file) {
       // No output file, write to console