mirror of
https://github.com/ultravioletrs/cocos.git
synced 2026-06-23 04:10:25 +00:00
Sammy Kerata Oina
d5badba547
CI / lint (push) Has been cancelled
CI / test (agent) (push) Has been cancelled
CI / test (cli) (push) Has been cancelled
CI / test (cmd) (push) Has been cancelled
CI / test (internal) (push) Has been cancelled
CI / test (manager, true) (push) Has been cancelled
CI / test (pkg) (push) Has been cancelled
CI / upload-coverage (push) Has been cancelled
* feat: Implement per-resource KBS configuration, allowing algorithms and datasets to specify individual KBS URLs. Signed-off-by: Sammy Oina <sammyoina@gmail.com> * refactor: Encapsulate CLI error handling and CVM certificate paths within the CLI struct, and add algorithm type to agent's algorithm structure. Signed-off-by: Sammy Oina <sammyoina@gmail.com> * style: Remove blank lines and fix indentation in CLI commands. Signed-off-by: Sammy Oina <sammyoina@gmail.com> * refactor: Update downloadAndDecryptGenericResource to accept KBS URL as a parameter and adjust related tests Signed-off-by: Sammy Oina <sammyoina@gmail.com> * refactor: group CLI configuration into structured types and simplify skopeo decryption key handling Signed-off-by: Sammy Oina <sammyoina@gmail.com> --------- Signed-off-by: Sammy Oina <sammyoina@gmail.com>
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
// Copyright (c) Ultraviolet
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package cli
|
|
|
|
import (
|
|
"github.com/absmach/supermq/pkg/errors"
|
|
"github.com/fatih/color"
|
|
"github.com/spf13/cobra"
|
|
"github.com/ultravioletrs/cocos/agent/auth"
|
|
"github.com/ultravioletrs/cocos/pkg/clients/grpc/agent"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
var (
|
|
errAgentUnavailable = errors.New("agent is unavailable on the current address")
|
|
errDigitalSignatureVerificationFailed = errors.New("digital signature verification failed, check the provided public key")
|
|
)
|
|
|
|
func decodeErros(err error) error {
|
|
statusErr, ok := status.FromError(err)
|
|
if ok {
|
|
switch statusErr.Code() {
|
|
case codes.PermissionDenied:
|
|
return errDigitalSignatureVerificationFailed
|
|
case codes.Unavailable:
|
|
return errAgentUnavailable
|
|
case codes.Unknown:
|
|
return err
|
|
}
|
|
}
|
|
switch {
|
|
case errors.Contains(err, auth.ErrSignatureVerificationFailed):
|
|
return auth.ErrSignatureVerificationFailed
|
|
|
|
case errors.Contains(err, agent.ErrAgentServiceUnavailable):
|
|
return agent.ErrAgentServiceUnavailable
|
|
default:
|
|
return err
|
|
}
|
|
}
|
|
|
|
func (c *CLI) printError(cmd *cobra.Command, message string, err error) {
|
|
if !c.Verbose {
|
|
err = decodeErros(err)
|
|
}
|
|
msg := color.New(color.FgRed).Sprintf(message, err)
|
|
cmd.Println(msg)
|
|
}
|