From f127de3048b187ff02a7a7afcf75695f8e8d9471 Mon Sep 17 00:00:00 2001 From: Miguel Mota Date: Sun, 25 Apr 2021 13:51:20 -0700 Subject: [PATCH] ssh: Add support for holdings command --- pkg/ssh/server.go | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/pkg/ssh/server.go b/pkg/ssh/server.go index fa06699..50222a5 100644 --- a/pkg/ssh/server.go +++ b/pkg/ssh/server.go @@ -169,11 +169,22 @@ func (s *Server) ListenAndServe() error { colorsDir, } - for i, arg := range cmdUserArgs { - if i == 0 { - continue + if len(cmdUserArgs) > 0 { + if cmdUserArgs[0] == "cointop" { + cmdUserArgs = cmdUserArgs[1:] } + } + + if len(cmdUserArgs) > 0 { + if cmdUserArgs[0] == "holdings" { + flags = []string{ + "--config", + configPath, + } + } + } + for _, arg := range cmdUserArgs { flags = append(flags, arg) }