From 9e407afe5f375ca14572374370f08ce81e3ab92c Mon Sep 17 00:00:00 2001
From: spbsoluble <1661003+spbsoluble@users.noreply.github.com>
Date: Mon, 9 Dec 2024 07:31:22 -0800
Subject: [PATCH] feat(store-types): Add ability to create store-types from
 `integration-manifest.json`

Signed-off-by: spbsoluble <1661003+spbsoluble@users.noreply.github.com>
---
 cmd/integration_manifest.go | 30 +++++++++++++++++++
 cmd/storeTypes.go           | 57 +++++++++++++++++++++++++++++--------
 2 files changed, 75 insertions(+), 12 deletions(-)
 create mode 100644 cmd/integration_manifest.go

diff --git a/cmd/integration_manifest.go b/cmd/integration_manifest.go
new file mode 100644
index 0000000..409c37f
--- /dev/null
+++ b/cmd/integration_manifest.go
@@ -0,0 +1,30 @@
+package cmd
+
+import (
+	"github.com/Keyfactor/keyfactor-go-client/v3/api"
+)
+
+type IntegrationManifest struct {
+	Schema          string `json:"$schema"`
+	IntegrationType string `json:"integration_type"`
+	Name            string `json:"name"`
+	Status          string `json:"status"`
+	LinkGithub      bool   `json:"link_github"`
+	UpdateCatalog   bool   `json:"update_catalog"`
+	SupportLevel    string `json:"support_level"`
+	ReleaseDir      string `json:"release_dir"`
+	ReleaseProject  string `json:"release_project"`
+	Description     string `json:"description"`
+	About           About  `json:"about"`
+}
+
+type About struct {
+	Orchestrator Orchestrator `json:"orchestrator"`
+}
+
+type Orchestrator struct {
+	UOFramework              string                     `json:"UOFramework"`
+	PAMSupport               bool                       `json:"pam_support"`
+	KeyfactorPlatformVersion string                     `json:"keyfactor_platform_version"`
+	StoreTypes               []api.CertificateStoreType `json:"store_types"`
+}
diff --git a/cmd/storeTypes.go b/cmd/storeTypes.go
index 068e7ca..9e1ccf1 100644
--- a/cmd/storeTypes.go
+++ b/cmd/storeTypes.go
@@ -133,13 +133,15 @@ var storesTypeCreateCmd = &cobra.Command{
 		}
 
 		if storeTypeConfigFile != "" {
-			createdStore, err := createStoreFromFile(storeTypeConfigFile, kfClient)
+			createdStoreTypes, err := createStoreTypeFromFile(storeTypeConfigFile, kfClient)
 			if err != nil {
 				fmt.Printf("Failed to create store type from file \"%s\"", err)
 				return err
 			}
 
-			fmt.Printf("Created store type called \"%s\"\n", createdStore.Name)
+			for _, v := range createdStoreTypes {
+				fmt.Printf("Created store type \"%s\"\n", v.Name)
+			}
 			return nil
 		}
 
@@ -409,27 +411,58 @@ var fetchStoreTypesCmd = &cobra.Command{
 //			return nil
 //		},
 //	}
-func createStoreFromFile(filename string, kfClient *api.Client) (*api.CertificateStoreType, error) {
+func createStoreTypeFromFile(filename string, kfClient *api.Client) ([]api.CertificateStoreType, error) {
 	// Read the file
+	log.Debug().Str("filename", filename).Msg("Reading store type from file")
 	file, err := os.Open(filename)
+	defer file.Close()
 	if err != nil {
+		log.Error().
+			Str("filename", filename).
+			Err(err).Msg("unable to open file")
 		return nil, err
 	}
 
 	// Compile JSON contents to a api.CertificateStoreType struct
-	var storeType api.CertificateStoreType
+	var sType api.CertificateStoreType
+	var sTypes []api.CertificateStoreType
+
+	log.Debug().Msg("Decoding JSON file as single store type")
 	decoder := json.NewDecoder(file)
-	err = decoder.Decode(&storeType)
-	if err != nil {
-		return nil, err
+	err = decoder.Decode(&sType)
+	if err != nil || (sType.ShortName == "" && sType.Capability == "") {
+		log.Warn().Err(err).Msg("Unable to decode JSON file, attempting to parse an integration manifest")
+		// Attempt to parse as an integration manifest
+		var manifest IntegrationManifest
+		log.Debug().Msg("Decoding JSON file as integration manifest")
+		// Reset the file pointer
+		_, err = file.Seek(0, 0)
+		decoder = json.NewDecoder(file)
+		mErr := decoder.Decode(&manifest)
+		if mErr != nil {
+			return nil, err
+		}
+		log.Debug().Msg("Decoded JSON file as integration manifest")
+		sTypes = manifest.About.Orchestrator.StoreTypes
+	} else {
+		log.Debug().Msg("Decoded JSON file as single store type")
+		sTypes = []api.CertificateStoreType{sType}
 	}
 
-	// Use the Keyfactor client to create the store type
-	createResp, err := kfClient.CreateStoreType(&storeType)
-	if err != nil {
-		return nil, err
+	for _, st := range sTypes {
+		log.Debug().Msgf("Creating certificate store type %s", st.Name)
+		createResp, cErr := kfClient.CreateStoreType(&st)
+		if cErr != nil {
+			log.Error().
+				Str("storeType", st.Name).
+				Err(cErr).Msg("unable to create certificate store type")
+			return nil, cErr
+		}
+		log.Info().Msgf("Certificate store type %s created with ID: %d", st.Name, createResp.StoreType)
 	}
-	return createResp, nil
+	// Use the Keyfactor client to create the store type
+	log.Debug().Msg("Store type created")
+	return sTypes, nil
 }
 
 func formatStoreTypes(sTypesList *[]interface{}) (map[string]interface{}, error) {