From d7ae8016693fe60e787d92aad4aa6a50929f80c8 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jan-sebastian.tuomi@aalto.fi>
Date: Sun, 9 Oct 2016 13:59:13 +0300
Subject: Fixes to edit command

---
 editor.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/editor.cpp b/editor.cpp
index e714a15..51b9077 100644
--- a/editor.cpp
+++ b/editor.cpp
@@ -46,7 +46,8 @@ int Editor::print_contents_on_line(int line) const {
 int Editor::edit_command() {
     std::cout << "Edit which line? ";
     int line;
-    std::cin >> line; if (std::cin.fail() || line < 0 || line >= m_contents->size()) {
+    std::cin >> line; 
+    if (std::cin.fail() || line < 0) {
         std::cout << "Not a valid line number!\n";
         std::cin.clear();
         std::cin.ignore(std::numeric_limits<std::streamsize>::max(),'\n');
@@ -57,6 +58,11 @@ int Editor::edit_command() {
 }
 
 int Editor::edit_line(int line) {
+
+    if (line >= m_contents->size()) {
+        m_contents->resize(line + 1, "");
+    }
+
     std::cout << "> ";
     std::string input;
 
-- 
cgit v1.3