Add prepend before line

2 files changed, 24 insertions, 0 deletions

diff --git a/editor.cpp b/editor.cpp
index 6c86447..f8c8ade 100644
--- a/editor.cpp
+++ b/editor.cpp
@@ -36,7 +36,11 @@ int Editor::run_editor() {
         case 'a':
             append_after_command();
             break;
+        case 'b':
+            prepend_before_command();
+            break;
         default:
+            std::cout << "Not a recognized command " << input << ".\n";
             break;
         }
     }
@@ -188,3 +192,21 @@ int Editor::append_after_line(int line) {
     return edit_line(line + 1);
 }
 
+int Editor::prepend_before_command() {
+    std::cout << "Prepend before which line? ";
+    int line;
+    std::cin >> line;
+    if (std::cin.fail() || line < 0 || line >= m_contents->size()) {
+        std::cout << "Not a valid line number!\n";
+        std::cin.clear();
+        std::cin.ignore(std::numeric_limits<std::streamsize>::max(),'\n');
+        return 1;
+    }
+
+    return prepend_before_line(line);
+}
+
+int Editor::prepend_before_line(int line) {
+    m_contents->insert(m_contents->begin() + line, "");
+    return edit_line(line);
+}
diff --git a/editor.h b/editor.h
index bb429bd..5cc5fdf 100644
--- a/editor.h
+++ b/editor.h
@@ -23,12 +23,14 @@ class Editor {
         int quit_command();
         int delete_and_clamp_command();
         int append_after_command();
+        int prepend_before_command();
 
         int edit_line(int line);
         int move_to_line(int line);
         int write();
         int delete_and_clamp(int line);
         int append_after_line(int line);
+        int prepend_before_line(int line);
 };
 
 #endif