-
Notifications
You must be signed in to change notification settings - Fork 208
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add cli command to update onprem token (#2066)
Currently, users cannot update their on-prem token and are limited to reinstalling Odigos in their cluster. This PR introduces the ability to update the odigos-pro secret's data using the odigos cli --------- Co-authored-by: Amir Blum <[email protected]> Co-authored-by: Tamir David <[email protected]>
- Loading branch information
3 people
authored
Jan 8, 2025
1 parent
82a098e
commit a1c17d2
Showing
9 changed files
with
206 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,88 @@ | ||
| package cmd | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "os" | ||
| "time" | ||
|
|
||
| "github.com/odigos-io/odigos/cli/cmd/resources" | ||
| cmdcontext "github.com/odigos-io/odigos/cli/pkg/cmd_context" | ||
| "github.com/odigos-io/odigos/cli/pkg/kube" | ||
| "github.com/odigos-io/odigos/k8sutils/pkg/consts" | ||
| odigosconsts "github.com/odigos-io/odigos/common/consts" | ||
| "github.com/spf13/cobra" | ||
| apierrors "k8s.io/apimachinery/pkg/api/errors" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| ) | ||
|
|
||
| var proCmd = &cobra.Command{ | ||
| Use: "pro", | ||
| Short: "manage odigos pro", | ||
| Long: `The pro command provides various operations and functionalities specifically designed for enterprise users. Use this command to access advanced features and manage your pro account.`, | ||
| Run: func(cmd *cobra.Command, args []string) { | ||
| ctx := cmd.Context() | ||
| client := cmdcontext.KubeClientFromContextOrExit(ctx) | ||
| ns, err := resources.GetOdigosNamespace(client, ctx) | ||
| if resources.IsErrNoOdigosNamespaceFound(err) { | ||
| fmt.Println("\033[31mERROR\033[0m no odigos installation found in the current cluster") | ||
| os.Exit(1) | ||
| } else if err != nil { | ||
| fmt.Printf("\033[31mERROR\033[0m Failed to check if Odigos is already installed: %s\n", err) | ||
| os.Exit(1) | ||
| } | ||
| onPremToken := cmd.Flag("onprem-token").Value.String() | ||
| err = updateOdigosToken(ctx, client, ns, onPremToken) | ||
| if err != nil { | ||
| fmt.Println("\033[31mERROR\033[0m Failed to update token:") | ||
| fmt.Println(err) | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| fmt.Println() | ||
| fmt.Println("\u001B[32mSUCCESS:\u001B[0m Token updated successfully") | ||
| }, | ||
| } | ||
|
|
||
| func updateOdigosToken(ctx context.Context, client *kube.Client, namespace string, onPremToken string) error { | ||
| secret, err := client.CoreV1().Secrets(namespace).Get(ctx, consts.OdigosProSecretName, metav1.GetOptions{}) | ||
| if err != nil { | ||
| if apierrors.IsNotFound(err) { | ||
| return fmt.Errorf("Tokens are not available in the open-source version of Odigos. Please contact Odigos team to inquire about pro version.") | ||
| } | ||
| return err | ||
| } | ||
| secret.Data[consts.OdigosOnpremTokenSecretKey] = []byte(onPremToken) | ||
|
|
||
| _, err = client.CoreV1().Secrets(namespace).Update(ctx, secret, metav1.UpdateOptions{}) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| daemonSet, err := client.AppsV1().DaemonSets(namespace).Get(ctx, "odiglet", metav1.GetOptions{}) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get DaemonSet odiglet in namespace %s: %v", namespace, err) | ||
| } | ||
|
|
||
| // Modify the DaemonSet spec.template to trigger a rollout | ||
| if daemonSet.Spec.Template.Annotations == nil { | ||
| daemonSet.Spec.Template.Annotations = make(map[string]string) | ||
| } | ||
| daemonSet.Spec.Template.Annotations[odigosconsts.RolloutTriggerAnnotation] = time.Now().Format(time.RFC3339) | ||
|
|
||
| _, err = client.AppsV1().DaemonSets(namespace).Update(ctx, daemonSet, metav1.UpdateOptions{}) | ||
| if err != nil { | ||
| fmt.Printf("Failed to restart Odiglets. Reason: %s\n", err) | ||
| fmt.Printf("To trigger a restart manually, run the following command:\n") | ||
| fmt.Printf("kubectl rollout restart daemonset odiglet -n %s\n", daemonSet.Namespace) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func init() { | ||
| rootCmd.AddCommand(proCmd) | ||
|
|
||
| proCmd.Flags().String("onprem-token", "", "On-prem token for Odigos") | ||
| proCmd.MarkFlagRequired("onprem-token") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters