Adding enterprise commands support (#8327)

1 files changed, 54 insertions, 0 deletions

diff --git a/cmd/commands/license.go b/cmd/commands/license.go
new file mode 100644
index 000000000..dce257a5d
--- /dev/null
+++ b/cmd/commands/license.go
@@ -0,0 +1,54 @@
+// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package commands
+
+import (
+	"errors"
+	"io/ioutil"
+
+	"github.com/mattermost/mattermost-server/cmd"
+	"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:    uploadLicenseCmdF,
+}
+
+func init() {
+	LicenseCmd.AddCommand(UploadLicenseCmd)
+	cmd.RootCmd.AddCommand(LicenseCmd)
+}
+
+func uploadLicenseCmdF(command *cobra.Command, args []string) error {
+	a, err := cmd.InitDBCommandContextCobra(command)
+	if err != nil {
+		return err
+	}
+
+	if len(args) != 1 {
+		return errors.New("Enter one license file to upload")
+	}
+
+	var fileBytes []byte
+	if fileBytes, err = ioutil.ReadFile(args[0]); err != nil {
+		return err
+	}
+
+	if _, err := a.SaveLicense(fileBytes); err != nil {
+		return err
+	}
+
+	cmd.CommandPrettyPrintln("Uploaded license file")
+
+	return nil
+}