feat: Add method to get ttl time remaining (#31)

* feat: Add method to get ttl time remaining * lint
mirakl · Jul 26, 2024 · b609555 · b609555
1 parent d524ee7
commit b609555
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/internal/libvault/vault.go b/internal/libvault/vault.go
@@ -68,3 +68,12 @@ func CreateClientWithAppRole(roleID, secretID string) (*vault.Client, error) {
 	client.SetToken(resp.Auth.ClientToken)
 	return client, nil
 }
+
+func GetTokenTTLLeft(client *vault.Client) (int, error) {
+	secret, err := client.Auth().Token().LookupSelf()
+	if err != nil {
+		return 0, errors.Wrap(err, "failed to lookup token")
+	}
+
+	return int(secret.Data["ttl"].(float64)), nil
+}
diff --git a/vaultv1/v1.go b/vaultv1/v1.go
@@ -34,6 +34,15 @@ func CreateClientWithAppRole(roleID, secretID string) (*Client, error) {
 	}, nil
 }
 
+func GetTokenTTLLeft(client *Client) (int, error) {
+	timeLeft, err := libvault.GetTokenTTLLeft(client.Client)
+	if err != nil {
+		return 0, errors.Wrapf(err, "")
+	}
+
+	return timeLeft, nil
+}
+
 func (vc *Client) ListSecretPath(path string) ([]string, error) {
 	s, err := vc.Client.Logical().List(path)
 	if err != nil {

diff --git a/vaultv2/v2.go b/vaultv2/v2.go
@@ -35,6 +35,15 @@ func CreateClientWithAppRole(roleID, secretID string) (*Client, error) {
 	}, nil
 }
 
+func GetTokenTTLLeft(client *Client) (int, error) {
+	timeLeft, err := libvault.GetTokenTTLLeft(client.Client)
+	if err != nil {
+		return 0, errors.Wrapf(err, "")
+	}
+
+	return timeLeft, nil
+}
+
 func (vc *Client) ReadSecret(path string, field string) (string, error) {
 	secret, err := vc.GetSecret(path)
 	if err != nil {