[check-credentials] add check-credentials command (#12)

Hey, I don't really know Go or how to test this locally, but I think this should do what i'm looking for. Just trying to add a command that I can use to understand if the credentials cache is valid or not. Maybe there's a better way to do this already? --------- Co-authored-by: spalger <[email protected]> Co-authored-by: Harry Bowron <[email protected]>
Metronome-Industries · Dec 7, 2023 · e39126c · e39126c
1 parent 7cd90ad
commit e39126c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/cmd/credentials.go b/cmd/credentials.go
@@ -20,5 +20,7 @@ alias creds="eval \$(quikstrate credentials)"`,
 func init() {
 	credentialsCmd.Flags().StringP("format", "f", "export", "substrate environment")
 	credentialsCmd.Flags().Bool("force", false, "always fetch new credentials")
+	credentialsCmd.Flags().Bool("check", false, "check if credentials are expired, exit 0 if up-to-date otherwise exit 1`")
+	credentialsCmd.MarkFlagsMutuallyExclusive("force", "check")
 	rootCmd.AddCommand(credentialsCmd)
 }
diff --git a/internal/creds/cmd_credentials.go b/internal/creds/cmd_credentials.go
@@ -2,13 +2,19 @@ package creds
 
 import (
 	"log"
+	"os"
 
 	"github.com/spf13/cobra"
 )
 
 func CredentialsCmd(cmd *cobra.Command, args []string) {
 	format := cmd.Flag("format").Value.String()
 	force := cmd.Flag("force").Value.String()
+	check := cmd.Flag("check").Value.String()
+
+	if check == "true" {
+		checkCredentials()
+	}
 
 	var creds Credentials
 	var err error
@@ -26,3 +32,11 @@ func CredentialsCmd(cmd *cobra.Command, args []string) {
 func getDefaultCredentials() (Credentials, error) {
 	return refreshCredentials(RoleData{}, DefaultCredsFile)
 }
+
+func checkCredentials() {
+	creds, err := getCredsFromFile(DefaultCredsFile)
+	if err != nil || creds.needsRefresh() {
+		os.Exit(1)
+	}
+	os.Exit(0)
+}