From 12bab7f7bb8049763711ec48a9a7a81427a25a05 Mon Sep 17 00:00:00 2001 From: peshwar9 Date: Wed, 25 Nov 2020 12:51:27 +0530 Subject: [PATCH] revisions for Chapter 7 FD --- chapter7/tui/src/bin/text-viewer1.rs | 4 ++++ chapter7/tui/src/bin/text-viewer2.rs | 6 +++++- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/chapter7/tui/src/bin/text-viewer1.rs b/chapter7/tui/src/bin/text-viewer1.rs index 6133322..f4f3b6f 100644 --- a/chapter7/tui/src/bin/text-viewer1.rs +++ b/chapter7/tui/src/bin/text-viewer1.rs @@ -104,6 +104,10 @@ impl TextViewer { fn main() { //Get arguments from command line let args: Vec = args().collect(); + if args.len() < 2 { + println!("Please provide file name as argument"); + std::process::exit(0); + } //Check if file exists. If not, print error message and exit process if !std::path::Path::new(&args[1]).exists() { println!("File does not exist"); diff --git a/chapter7/tui/src/bin/text-viewer2.rs b/chapter7/tui/src/bin/text-viewer2.rs index ca5e68b..3fee9c4 100644 --- a/chapter7/tui/src/bin/text-viewer2.rs +++ b/chapter7/tui/src/bin/text-viewer2.rs @@ -170,6 +170,10 @@ struct Coordinates { fn main() { //Get arguments from command line let args: Vec = args().collect(); + if args.len() < 2 { + println!("Please provide file name as argument"); + std::process::exit(0); + } //Check if file exists. If not, print error message and exit process if !std::path::Path::new(&args[1]).exists() { println!("File does not exist"); @@ -182,7 +186,7 @@ fn main() { println!("{}", termion::cursor::Goto(1, 1)); // Initialize editor let mut editor = TextViewer::init(&args[1]); - editor.show_document(); editor.set_pos(1, 1); + editor.show_document(); editor.run(); }