added argument-parser and a human-readable flag

Author: scriptingosx

diskspace.xcodeproj/project.pbxproj

@@ -8,6 +8,7 @@
 
 /* Begin PBXBuildFile section */
 		C6F43F032731718900747000 /* main.swift in Sources */ = {isa = PBXBuildFile; fileRef = C6F43F022731718900747000 /* main.swift */; };
+		C6F43F0B2731762C00747000 /* ArgumentParser in Frameworks */ = {isa = PBXBuildFile; productRef = C6F43F0A2731762C00747000 /* ArgumentParser */; };
 /* End PBXBuildFile section */
 
 /* Begin PBXCopyFilesBuildPhase section */
@@ -32,6 +33,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				C6F43F0B2731762C00747000 /* ArgumentParser in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
@@ -78,6 +80,9 @@
 			dependencies = (
 			);
 			name = diskspace;
+			packageProductDependencies = (
+				C6F43F0A2731762C00747000 /* ArgumentParser */,
+			);
 			productName = diskspace;
 			productReference = C6F43EFF2731718900747000 /* diskspace */;
 			productType = "com.apple.product-type.tool";
@@ -106,6 +111,9 @@
 				Base,
 			);
 			mainGroup = C6F43EF62731718900747000;
+			packageReferences = (
+				C6F43F092731762C00747000 /* XCRemoteSwiftPackageReference "swift-argument-parser" */,
+			);
 			productRefGroup = C6F43F002731718900747000 /* Products */;
 			projectDirPath = "";
 			projectRoot = "";
@@ -286,6 +294,25 @@
 			defaultConfigurationName = Release;
 		};
 /* End XCConfigurationList section */
+
+/* Begin XCRemoteSwiftPackageReference section */
+		C6F43F092731762C00747000 /* XCRemoteSwiftPackageReference "swift-argument-parser" */ = {
+			isa = XCRemoteSwiftPackageReference;
+			repositoryURL = "https://github.com/apple/swift-argument-parser.git";
+			requirement = {
+				kind = upToNextMajorVersion;
+				minimumVersion = 1.0.0;
+			};
+		};
+/* End XCRemoteSwiftPackageReference section */
+
+/* Begin XCSwiftPackageProductDependency section */
+		C6F43F0A2731762C00747000 /* ArgumentParser */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = C6F43F092731762C00747000 /* XCRemoteSwiftPackageReference "swift-argument-parser" */;
+			productName = ArgumentParser;
+		};
+/* End XCSwiftPackageProductDependency section */
 	};
 	rootObject = C6F43EF72731718900747000 /* Project object */;
 }

diskspace.xcodeproj/xcuserdata/armin.xcuserdatad/xcschemes/xcschememanagement.plist

@@ -10,5 +10,13 @@
 			<integer>0</integer>
 		</dict>
 	</dict>
+	<key>SuppressBuildableAutocreation</key>
+	<dict>
+		<key>C6F43EFE2731718900747000</key>
+		<dict>
+			<key>primary</key>
+			<true/>
+		</dict>
+	</dict>
 </dict>
 </plist>

diskspace/main.swift

@@ -6,22 +6,47 @@
 //
 
 import Foundation
+import ArgumentParser
 
-let fileURL = URL(fileURLWithPath:"/")
-do {
-    let values = try fileURL.resourceValues(forKeys: [.volumeAvailableCapacityKey,.volumeAvailableCapacityForImportantUsageKey, .volumeAvailableCapacityForOpportunisticUsageKey, .volumeTotalCapacityKey])
-    if let importantCapacity = values.volumeAvailableCapacity {
-        print("Available:     \(importantCapacity)")
-    }
-    if let importantCapacity = values.volumeAvailableCapacityForImportantUsage {
-        print("Important:     \(importantCapacity)")
+// Code based on sample from here:
+// https://developer.apple.com/documentation/foundation/urlresourcekey/checking_volume_storage_capacity
+
+struct DiskSpace : ParsableCommand {
+    @Flag var humanReadable: Bool = false
+    
+    
+    func format(_ int: Int) -> String {
+        return format(Int64(int))
     }
-    if let opportunisticCapacity = values.volumeAvailableCapacityForOpportunisticUsage {
-        print("Opportunistic: \(opportunisticCapacity)")
+    
+    func format(_ int: Int64) -> String {
+        if humanReadable {
+            return ByteCountFormatter().string(fromByteCount: Int64(int))
+        } else {
+            return String(int)
+        }
     }
-    if let totalCapacity = values.volumeTotalCapacity {
-        print("Total:         \(totalCapacity)")
+    
+    func run() {
+        let systemVolume = URL(fileURLWithPath:"/")
+        do {
+            let values = try systemVolume.resourceValues(forKeys: [.volumeAvailableCapacityKey,.volumeAvailableCapacityForImportantUsageKey, .volumeAvailableCapacityForOpportunisticUsageKey, .volumeTotalCapacityKey])
+            if let totalCapacity = values.volumeAvailableCapacity {
+                print("Available:     \(format(totalCapacity))")
+            }
+            if let importantCapacity = values.volumeAvailableCapacityForImportantUsage {
+                print("Important:     \(format(importantCapacity))")
+            }
+            if let opportunisticCapacity = values.volumeAvailableCapacityForOpportunisticUsage {
+                print("Opportunistic: \(format(opportunisticCapacity))")
+            }
+            if let totalCapacity = values.volumeTotalCapacity {
+                print("Total:         \(format(totalCapacity))")
+            }
+        } catch {
+            print("Error retrieving capacity: \(error.localizedDescription)")
+        }
     }
-} catch {
-    print("Error retrieving capacity: \(error.localizedDescription)")
 }
+
+DiskSpace.main()