From e3c0a13347db23e0ef2c78d4c73e1f3c3af2c159 Mon Sep 17 00:00:00 2001
From: Rogger Vasquez <rvasque3@gmail.com>
Date: Fri, 15 Jul 2022 17:33:53 -0500
Subject: [PATCH] rpk: print license expiration warning

Sending alert in plain format when the license is
about to expire and adding field when license is
expired.
---
 .../rpk/pkg/cli/cmd/cluster/license/info.go   | 27 ++++++++++++++-----
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/src/go/rpk/pkg/cli/cmd/cluster/license/info.go b/src/go/rpk/pkg/cli/cmd/cluster/license/info.go
index 03153aa33f1e..f08d98cd7347 100644
--- a/src/go/rpk/pkg/cli/cmd/cluster/license/info.go
+++ b/src/go/rpk/pkg/cli/cmd/cluster/license/info.go
@@ -3,6 +3,7 @@ package license
 import (
 	"encoding/json"
 	"fmt"
+	"os"
 
 	"github.com/redpanda-data/redpanda/src/go/rpk/pkg/api/admin"
 	"github.com/redpanda-data/redpanda/src/go/rpk/pkg/config"
@@ -40,12 +41,16 @@ func newInfoCommand(fs afero.Fs) *cobra.Command {
 			}
 
 			if info.Properties != (admin.LicenseProperties{}) {
+				expired := info.Properties.Expires < 0
 				if format == "json" {
-					props, err := json.MarshalIndent(info.Properties, "", "  ")
+					props, err := json.MarshalIndent(struct {
+						admin.LicenseProperties
+						Expired bool `json:"license_expired,omitempty"`
+					}{info.Properties, expired}, "", "  ")
 					out.MaybeDie(err, "unable to print license information as json: %v", err)
 					fmt.Printf("%s\n", props)
 				} else {
-					printLicenseInfo(info.Properties)
+					printLicenseInfo(info.Properties, expired)
 				}
 			} else {
 				out.Die("no license loaded")
@@ -57,12 +62,20 @@ func newInfoCommand(fs afero.Fs) *cobra.Command {
 	return command
 }
 
-func printLicenseInfo(p admin.LicenseProperties) {
+func printLicenseInfo(p admin.LicenseProperties, expired bool) {
 	out.Section("LICENSE INFORMATION")
-	licenseFormat := `Organization:    %v
-Type:            %v
-Expires:         %v days
-Version:         %v
+	licenseFormat := `Organization:      %v
+Type:              %v
+Expires:           %v days
+Version:           %v
 `
+	if expired {
+		licenseFormat += `License Expired:   true
+`
+	}
 	fmt.Printf(licenseFormat, p.Organization, p.Type, p.Expires, p.Version)
+	if p.Expires < 30 && p.Expires >= 0 {
+		fmt.Fprintln(os.Stderr, "warning: your license will expire soon")
+		return
+	}
 }