mirror of
https://github.com/ultravioletrs/cocos.git
synced 2026-08-07 23:31:55 +00:00
Sammy Kerata Oina
e2fb7ea88d
* add test server Signed-off-by: SammyOina <sammyoina@gmail.com> * clean up and update docs Signed-off-by: SammyOina <sammyoina@gmail.com> * update docs Signed-off-by: SammyOina <sammyoina@gmail.com> * fix lint Signed-off-by: SammyOina <sammyoina@gmail.com> --------- Signed-off-by: SammyOina <sammyoina@gmail.com>
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
// Copyright (c) Ultraviolet
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
mglog "github.com/absmach/magistrala/logger"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
"github.com/ultravioletrs/cocos/cli"
|
|
"github.com/ultravioletrs/cocos/internal/env"
|
|
"github.com/ultravioletrs/cocos/pkg/clients/grpc"
|
|
"github.com/ultravioletrs/cocos/pkg/clients/grpc/agent"
|
|
"github.com/ultravioletrs/cocos/pkg/sdk"
|
|
)
|
|
|
|
const (
|
|
svcName = "cli"
|
|
envPrefixAgentGRPC = "AGENT_GRPC_"
|
|
completion = "completion"
|
|
)
|
|
|
|
type config struct {
|
|
LogLevel string `env:"AGENT_LOG_LEVEL" envDefault:"info"`
|
|
}
|
|
|
|
func main() {
|
|
var cfg config
|
|
if err := env.Parse(&cfg); err != nil {
|
|
log.Fatalf("failed to load %s configuration : %s", svcName, err)
|
|
}
|
|
|
|
logger, err := mglog.New(os.Stdout, cfg.LogLevel)
|
|
if err != nil {
|
|
log.Fatalf("Error creating logger: %s", err)
|
|
}
|
|
|
|
agentGRPCConfig := grpc.Config{}
|
|
if err := env.Parse(&agentGRPCConfig, env.Options{Prefix: envPrefixAgentGRPC}); err != nil {
|
|
logger.Error(fmt.Sprintf("failed to load %s gRPC client configuration : %s", svcName, err))
|
|
return
|
|
}
|
|
|
|
agentGRPCClient, agentClient, err := agent.NewAgentClient(agentGRPCConfig)
|
|
if err != nil {
|
|
logger.Error(err.Error())
|
|
return
|
|
}
|
|
defer agentGRPCClient.Close()
|
|
|
|
agentSDK := sdk.NewAgentSDK(logger, agentClient)
|
|
|
|
cliSVC := cli.New(agentSDK)
|
|
|
|
rootCmd := &cobra.Command{
|
|
Use: "cocos-cli [command]",
|
|
Short: "CLI application for CoCos Service API",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
fmt.Printf("CLI application for CoCos Service API\n\n")
|
|
fmt.Printf("Usage:\n %s [command]\n\n", cmd.CommandPath())
|
|
fmt.Printf("Available Commands:\n")
|
|
|
|
// Filter out "completion" command
|
|
availableCommands := make([]*cobra.Command, 0)
|
|
for _, subCmd := range cmd.Commands() {
|
|
if subCmd.Name() != completion {
|
|
availableCommands = append(availableCommands, subCmd)
|
|
}
|
|
}
|
|
|
|
for _, subCmd := range availableCommands {
|
|
fmt.Printf(" %-15s%s\n", subCmd.Name(), subCmd.Short)
|
|
}
|
|
|
|
fmt.Printf("\nFlags:\n")
|
|
cmd.Flags().VisitAll(func(flag *pflag.Flag) {
|
|
fmt.Printf(" -%s, --%s %s\n", flag.Shorthand, flag.Name, flag.Usage)
|
|
})
|
|
fmt.Printf("\nUse \"%s [command] --help\" for more information about a command.\n", cmd.CommandPath())
|
|
},
|
|
}
|
|
|
|
// Agent Commands
|
|
rootCmd.AddCommand(cliSVC.NewAlgorithmsCmd())
|
|
rootCmd.AddCommand(cliSVC.NewDatasetsCmd())
|
|
rootCmd.AddCommand(cliSVC.NewResultsCmd())
|
|
attestaionCmd := cliSVC.NewAttestationCmd()
|
|
rootCmd.AddCommand(attestaionCmd)
|
|
|
|
// Attestation commands
|
|
attestaionCmd.AddCommand(cliSVC.NewGetAttestationCmd())
|
|
attestaionCmd.AddCommand(cliSVC.NewValidateAttestationValidationCmd())
|
|
|
|
if err := rootCmd.Execute(); err != nil {
|
|
logger.Error(fmt.Sprintf("Command execution failed: %s", err))
|
|
return
|
|
}
|
|
}
|