From 4619bac708604fa2d3c2e5588f0e1a751c971c93 Mon Sep 17 00:00:00 2001 From: Raymaekers Luca Date: Wed, 8 Jan 2025 19:01:42 +0100 Subject: Require -n when deleting a task as security --- ws/main.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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.") -- cgit v1.2.3