wraith-labs/wraith_module_comosum
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: list command with fancy md table

Browse Source
This commit is contained in:
0x1a8510f2 2023-06-12 04:10:23 +01:00
parent e000fa79f1
commit 310bf9b8dd
Signed by: 0x1a8510f2
GPG Key ID: 1C692E355D76775D
1 changed files with 27 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
cmd/pc3/commands.go
View File

@ -1,7 +1,6 @@
package main
import (
"encoding/json"
"fmt"
"reflect"
"strconv"
@ -13,13 +12,7 @@ import (
"github.com/traefik/yaegi/stdlib/unsafe"
)
func CmdX(ctx lib.CommandContext, arg string) (response string, errResponse error) {
defer func() {
if p := recover(); p != nil {
errResponse = fmt.Errorf("command panicked: %e", p)
}
}()
func CmdX(ctx lib.CommandContext, arg string) (string, error) {
i := interp.New(interp.Options{
Unrestricted: true,
})
@ -64,8 +57,25 @@ func CmdL(ctx lib.CommandContext, arg string) (string, error) {
if err != nil {
return "", fmt.Errorf("could not get page from database: %e", err)
}
clientListString, _ := json.Marshal(clients)
return string(clientListString), nil
clientListString := fmt.Sprintf("Client list page %d:\n", page)
clientListString += "\n| Client ID | Strain ID | Init Time | Hostname | Host OS | Host Arch | HostUser | Host User ID | Modules | Errors |"
clientListString += "\n| --- | --- | --- | --- | --- | --- | --- | --- | --- | --- |"
for _, client := range clients {
clientListString += fmt.Sprintf(
"\n| %s | %s | %s | %s | %s | %s | %s | %s | %s | %d |",
client.ID,
client.LastHeartbeat.StrainId,
client.LastHeartbeat.InitTime,
client.LastHeartbeat.Hostname,
client.LastHeartbeat.HostOS,
client.LastHeartbeat.HostArch,
client.LastHeartbeat.HostUser,
client.LastHeartbeat.HostUserId,
client.LastHeartbeat.Modules,
client.LastHeartbeat.Errors,
)
}
return clientListString, nil
}
func CmdH(ctx lib.CommandContext, arg string) (string, error) {
@ -77,7 +87,13 @@ func CmdH(ctx lib.CommandContext, arg string) (string, error) {
}
}
func ExecCmd(ctx lib.CommandContext, command string) (string, error) {
func ExecCmd(ctx lib.CommandContext, command string) (response string, errResponse error) {
defer func() {
if p := recover(); p != nil {
errResponse = fmt.Errorf("command panicked: %e", p)
}
}()
keyword, arg, _ := strings.Cut(command, " ")
switch strings.ToLower(keyword) {
case "x":