upstream-mattermost/server/cmd/mmctl/commands/license.go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package commands

import (
	"context"
	"errors"
	"os"
	"strconv"
	"time"

	"github.com/mattermost/mattermost/server/v8/cmd/mmctl/client"
	"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"

	"github.com/spf13/cobra"
)

var LicenseCmd = &cobra.Command{
	Use:   "license",
	Short: "Licensing commands",
}

var UploadLicenseCmd = &cobra.Command{
	Use:     "upload [license]",
	Short:   "Upload a license.",
	Long:    "Upload a license. Replaces current license.",
	Example: "  license upload /path/to/license/mylicensefile.mattermost-license",
	RunE:    withClient(uploadLicenseCmdF),
}

var UploadLicenseStringCmd = &cobra.Command{
	Use:     "upload-string [license]",
	Short:   "Upload a license from a string.",
	Long:    "Upload a license from a string. Replaces current license.",
	Example: " license upload-string \"mylicensestring\"",
	RunE:    withClient(uploadLicenseStringCmdF),
}

var RemoveLicenseCmd = &cobra.Command{
	Use:     "remove",
	Short:   "Remove the current license.",
	Long:    "Remove the current license and leave mattermost in Team Edition.",
	Example: "  license remove",
	RunE:    withClient(removeLicenseCmdF),
}

var GetLicenseCmd = &cobra.Command{
	Use:     "get",
	Short:   "Get the current license.",
	Long:    "Get the current server license and print it.",
	Example: "  license get",
	RunE:    withClient(getLicenseCmdF),
}

func init() {
	LicenseCmd.AddCommand(UploadLicenseCmd)
	LicenseCmd.AddCommand(RemoveLicenseCmd)
	LicenseCmd.AddCommand(UploadLicenseStringCmd)
	LicenseCmd.AddCommand(GetLicenseCmd)
	RootCmd.AddCommand(LicenseCmd)
}

func uploadLicenseStringCmdF(c client.Client, cmd *cobra.Command, args []string) error {
	if len(args) != 1 {
		return errors.New("enter one license file to upload")
	}

	licenseBytes := []byte(args[0])

	if _, err := c.UploadLicenseFile(context.TODO(), licenseBytes); err != nil {
		return err
	}

	printer.Print("Uploaded license file")

	return nil
}

func uploadLicenseCmdF(c client.Client, cmd *cobra.Command, args []string) error {
	if len(args) != 1 {
		return errors.New("enter one license file to upload")
	}

	fileBytes, err := os.ReadFile(args[0])
	if err != nil {
		return err
	}

	if _, err := c.UploadLicenseFile(context.TODO(), fileBytes); err != nil {
		return err
	}

	printer.Print("Uploaded license file")

	return nil
}

func removeLicenseCmdF(c client.Client, cmd *cobra.Command, args []string) error {
	if _, err := c.RemoveLicenseFile(context.TODO()); err != nil {
		return err
	}

	printer.Print("Removed license")

	return nil
}

func getLicenseCmdF(c client.Client, cmd *cobra.Command, args []string) error {
	license, _, err := c.GetOldClientLicense(context.TODO(), "")
	if err != nil {
		return err
	}

	if license["IsLicensed"] != "true" {
		printer.Print("No license installed")
		return nil
	}

	// Format timestamps for human-readable output
	license["StartsAtReadable"] = formatLicenseTimestamp(license["StartsAt"])
	license["ExpiresAtReadable"] = formatLicenseTimestamp(license["ExpiresAt"])
	license["IssuedAtReadable"] = formatLicenseTimestamp(license["IssuedAt"])

	printer.PrintT(`License ID: {{.Id}}
Starts At: {{.StartsAtReadable}}
Expires At: {{.ExpiresAtReadable}}
Users: {{.Users}}
SKU: {{.SkuShortName}}
Is Trial: {{.IsTrial}}
Issued At: {{.IssuedAtReadable}}
Name: {{.Name}}
Company: {{.Company}}`, license)

	return nil
}

// formatLicenseTimestamp converts an epoch milliseconds string to a human-readable date.
func formatLicenseTimestamp(epochMs string) string {
	ms, err := strconv.ParseInt(epochMs, 10, 64)
	if err != nil {
		return epochMs
	}
	return time.UnixMilli(ms).UTC().Format(time.RFC3339)
}