Support --fmt

2 files changed, 12 insertions, 2 deletions

diff --git a/src/flags.rs b/src/flags.rs
index ecc60922e..d68bb5e09 100644
--- a/src/flags.rs
+++ b/src/flags.rs
@@ -30,6 +30,7 @@ pub struct DenoFlags {
   pub types: bool,
   pub prefetch: bool,
   pub info: bool,
+  pub fmt: bool,
 }
 
 pub fn get_usage(opts: &Options) -> String {
@@ -115,6 +116,9 @@ fn set_recognized_flags(
         if matches.opt_present("info") {
           flags.info = true;
         }
+        if matches.opt_present("fmt") {
+          flags.fmt = true;
+        }
 
         if !matches.free.is_empty() {
           rest.extend(matches.free);
@@ -152,6 +156,7 @@ pub fn set_flags(
   opts.optflag("", "types", "Print runtime TypeScript declarations.");
   opts.optflag("", "prefetch", "Prefetch the dependencies.");
   opts.optflag("", "info", "Show source file related info");
+  opts.optflag("", "fmt", "Format code.");
 
   let mut flags = DenoFlags::default();
 
diff --git a/src/main.rs b/src/main.rs
index f6a9c49ee..36a39dbf0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -63,8 +63,8 @@ fn print_err_and_exit(err: errors::RustOrJsError) {
 fn main() {
   log::set_logger(&LOGGER).unwrap();
   let args = env::args().collect();
-  let (flags, rest_argv, usage_string) =
-    flags::set_flags(args).unwrap_or_else(|err| {
+  let (mut flags, mut rest_argv, usage_string) = flags::set_flags(args)
+    .unwrap_or_else(|err| {
       eprintln!("{}", err);
       std::process::exit(1)
     });
@@ -80,6 +80,11 @@ fn main() {
     LevelFilter::Warn
   });
 
+  if flags.fmt {
+    rest_argv.insert(1, "https://deno.land/x/std/prettier/main.ts".to_string());
+    flags.allow_write = true;
+  }
+
   let should_prefetch = flags.prefetch;
   let should_display_info = flags.info;