From 19939f5d7c268938a62ac32b47296cc7d43146e5 Mon Sep 17 00:00:00 2001
From: Trent Huber <trentmhuber@gmail.com>
Date: Sat, 8 Nov 2025 03:18:14 -0500
Subject: [PATCH] Update set builtin too

---
 src/builtins/set.c   | 17 +++++++++++------
 src/builtins/unset.c |  8 +++-----
 2 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/src/builtins/set.c b/src/builtins/set.c
index df28cc8..f8df49e 100644
--- a/src/builtins/set.c
+++ b/src/builtins/set.c
@@ -4,12 +4,17 @@
 #include "utils.h"
 
 BUILTIN(set) {
-	if (argc != 3) return usage(argv[0], "name value");
-
-	if (setenv(argv[1], argv[2], 1) == -1) {
-		note("Unable to set %s to %s", argv[1], argv[2]);
-		return EXIT_FAILURE;
+	switch (argc) {
+	case 3:
+		if (setenv(argv[1], argv[2], 1) == -1) {
+			note("Unable to set %s to %s", argv[1], argv[2]);
+			return EXIT_FAILURE;
+		}
+	case 2:
+		break;
+	default:
+		return usage(argv[0], "name [value]");
 	}
-	
+
 	return EXIT_SUCCESS;
 }
diff --git a/src/builtins/unset.c b/src/builtins/unset.c
index c7eb78f..d7a5679 100644
--- a/src/builtins/unset.c
+++ b/src/builtins/unset.c
@@ -6,10 +6,8 @@
 BUILTIN(unset) {
 	if (argc != 2) return usage(argv[0], "name");
 
-	if (unsetenv(argv[1]) == -1) {
-		note("Unable to unset $%s$", argv[1]);
-		return EXIT_FAILURE;
-	}
+	if (unsetenv(argv[1]) != -1) return EXIT_SUCCESS;
 
-	return EXIT_SUCCESS;
+	note("Unable to unset $%s$", argv[1]);
+	return EXIT_FAILURE;
 }
-- 
2.51.0