1 files changed, 12 insertions, 3 deletions

diff --git a/src/input.rs b/src/input.rs
index 41b0c29..ffc1213 100644
--- a/src/input.rs
+++ b/src/input.rs
@@ -1,5 +1,5 @@
 use std::env;
-use crate::{config::Cfg, list, modification, update, setup, download, error::{MLError, ErrorType, MLE}};
+use crate::{config::Cfg, list, modification, update, setup, download, io, error::{MLError, ErrorType, MLE}};
 
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub struct Input {
@@ -77,7 +77,8 @@ pub enum Cmd {
     List,
     Update,
     Download,
-    Setup
+    Setup,
+    Io
 }
 
 impl Cmd {
@@ -88,6 +89,7 @@ impl Cmd {
             "update" => Self::Update,
             "download" => Self::Download,
             "setup" => Self::Setup,
+            "io" => Self::Io,
             _ => return Err(MLError::new(ErrorType::ArgumentError, "Unknown command"))
         };
         Ok(cmd)
@@ -98,7 +100,9 @@ impl Cmd {
 pub enum Subcmd {
     Add,
     Remove,
-    Change
+    Change,
+    Export,
+    Import,
 }
 
 impl Subcmd {
@@ -107,6 +111,8 @@ impl Subcmd {
             "add" => Self::Add,
             "remove" => Self::Remove,
             "change" => Self::Change,
+            "export" => Self::Export,
+            "import" => Self::Import,
             _ => return Err(MLError::new(ErrorType::ArgumentError, "SUBCMD_NOT_FOUND"))
         };
         Ok(cmd)
@@ -136,6 +142,9 @@ pub async fn get_input(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
         },
         Cmd::Download => {
             download(config, input).await
+        },
+        Cmd::Io => {
+            io(config, input)
         }
     }
 }