added disk-space

Author: Heliozoa

tmc-langs-cli/Cargo.toml

@@ -26,6 +26,8 @@ base64 = "0.13"
 schemars = "0.8"
 thiserror = "1"
 toml = "0.5"
+heim = { version = "0.1.0-beta.3", features = ["disk"] }
+smol = "1"
 
 [dev-dependencies]
 tempfile = "3"

tmc-langs-cli/src/app.rs

@@ -62,6 +62,15 @@ pub fn create_app() -> App<'static, 'static> {
 
         .subcommand(create_core_app()) // "core"
 
+        .subcommand(
+            SubCommand::with_name("disk-space")
+                .about("Returns the amount of free disk space in megabytes left on the partition that contains the given path")
+                .arg(Arg::with_name("path")
+                    .help("A path in the partition that should be inspected.")
+                    .required(true)
+                    .takes_value(true))
+        )
+
         .subcommand(SubCommand::with_name("extract-project")
             .about("Extracts an exercise from a ZIP archive. If the output-path is a project root, the plugin's student file policy will be used to avoid overwriting student files")
             .long_about(SCHEMA_NULL)
@@ -676,8 +685,7 @@ fn create_settings_app() -> App<'static, 'static> {
                 ),
         )
         .subcommand(
-            SubCommand::with_name("list")
-                .about("Prints every key=value pair in the settings file."),
+            SubCommand::with_name("list").about("Prints every key=value pair in the settings file"),
         )
         .subcommand(
             SubCommand::with_name("move-projects-dir")

tmc-langs-cli/src/main.rs

@@ -10,6 +10,7 @@ use clap::{ArgMatches, Error, ErrorKind};
 use config::ProjectsConfig;
 use config::{CourseConfig, Credentials, Exercise, TmcConfig};
 use error::{InvalidTokenError, SandboxTestError};
+use heim::disk;
 use output::{
     CombinedCourseData, ErrorData, Kind, Output, OutputData, OutputResult, Status, Warnings,
 };
@@ -273,6 +274,27 @@ fn run_app(matches: ArgMatches, pretty: bool, warnings: &mut Vec<anyhow::Error>)
                 }
             }
         }
+        ("disk-space", Some(matches)) => {
+            let path = matches.value_of("path").unwrap();
+            let path = Path::new(path);
+
+            let usage = smol::block_on(disk::usage(path)).with_context(|| {
+                format!("Failed to get disk usage from path {}", path.display())
+            })?;
+            let free = usage.free().get::<heim::units::information::megabyte>();
+
+            let output = Output::OutputData(OutputData {
+                status: Status::Finished,
+                message: Some(format!(
+                    "calculated free disk space for partition containing {}",
+                    path.display()
+                )),
+                result: OutputResult::ExecutedCommand,
+                percent_done: 1.0,
+                data: Some(free),
+            });
+            print_output(&output, pretty, &warnings)?
+        }
         ("extract-project", Some(matches)) => {
             let archive_path = matches.value_of("archive-path").unwrap();
             let archive_path = Path::new(archive_path);