[PD1] Fix adding user cert to db, also finished logs

This commit is contained in:
Afonso Franco 2024-04-28 23:32:19 +01:00
parent f2118fe570
commit 4ea8315aed
Signed by: afonso
SSH key fingerprint: SHA256:aiLbdlPwXKJS5wMnghdtod0SPy8imZjlVvCyUX9DJNk
4 changed files with 24 additions and 16 deletions

View file

@ -8,6 +8,7 @@ import (
"errors"
"flag"
"log"
"os"
"sort"
"strconv"
)
@ -30,8 +31,10 @@ func Run() {
command := flag.Arg(0)
switch command {
case "send":
if flag.NArg() < 3 {
log.Fatalln("Insufficient arguments for 'send' command. Usage: send <UID> <SUBJECT>")
if flag.NArg() != 3 {
printError("MSG SERVICE: command error!")
showHelp()
os.Exit(1)
}
uid := flag.Arg(1)
plainSubject := flag.Arg(2)
@ -42,6 +45,11 @@ func Run() {
}
case "askqueue":
if flag.NArg() > 3 {
printError("MSG SERVICE: command error!")
showHelp()
os.Exit(1)
}
pageInput := flag.Arg(1)
page := 1
if pageInput != "" {
@ -64,7 +72,9 @@ func Run() {
case "getmsg":
if flag.NArg() < 2 {
log.Fatalln("Insufficient arguments for 'getmsg' command. Usage: getmsg <NUM>")
printError("MSG SERVICE: command error!")
showHelp()
os.Exit(1)
}
numString := flag.Arg(1)
num, err := strconv.Atoi(numString)
@ -73,14 +83,15 @@ func Run() {
}
err = getMsgCommand(clientKeyStore, num)
if err != nil {
log.Fatalln(err)
printError(err.Error())
}
case "help":
showHelp()
default:
commandError()
printError("MSG SERVICE: command error!")
showHelp()
}
}