refactor: simplification of removals

Author: vyadh

nut/project.nu

@@ -71,14 +71,16 @@ export def "remove dependency" [
     let dependencies = $project | child dependencies
     let id = $package | package to id
 
+    let category = $project | find category $package
+    if $category == null {
+        error make { msg: $"Package doesn't exist in project: ($id)" }
+    }
+
     {
         ...($project | reject --ignore-errors dependencies)
         dependencies: {
-            runtime: {
-                ...($dependencies | child runtime | reject --ignore-errors $id)
-            }
-            development: {
-                ...($dependencies | child development | reject --ignore-errors $id)
+            $category: {
+                ...($dependencies | child $category | reject --ignore-errors $id)
             }
         }
     }

test/project_test.nu

@@ -245,14 +245,11 @@ def "remove package that was not added" [] {
         fragment: ""
     }
 
-    let result = $project | project remove dependency $pkg
-
-    assert equal $result {
-        dependencies: {
-            runtime: { }
-            development: { }
-        }
+    let error = catch-error {
+        $project | project remove dependency $pkg
     }
+
+    assert equal $error "Package doesn't exist in project: github.com/example/project"
 }
 
 # [test]
@@ -295,7 +292,6 @@ def "remove package that was previously added" [] {
                     revision: "01"
                 }
             }
-            development: { }
         }
     }
 }