Require -n when deleting a task as security

1 files changed, 6 insertions, 1 deletions

diff --git a/ws/main.go b/ws/main.go
index 57a279d..c66f720 100644
--- a/ws/main.go
+++ b/ws/main.go
@@ -460,10 +460,15 @@ func main() {
 		var shouldDeleteDone bool
 
 		flagSet := flag.NewFlagSet("del", flag.ExitOnError)
-		flagSet.IntVar(&n, "n", 1, "Set task number")
+		flagSet.IntVar(&n, "n", 0, "Set task number")
 		flagSet.BoolVar(&shouldDeleteDone, "d", false, "Should delete a done task")
 		flagSet.Parse(os.Args[2:])
 
+		if n == 0 {
+			fmt.Println("Task number is required.")
+			os.Exit(1)
+		}
+
 		if !shouldDeleteDone {
 			if len(TasksDo) == 0 {
 				fmt.Println("No tasks to edit.")