--- /dev/null
+// -
+//
+// ========================LICENSE_START=================================
+// O-RAN-SC
+// %%
+// Copyright (C) 2024 OpenInfra Foundation Europe. All rights reserved.
+// %%
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ========================LICENSE_END===================================
+package main
+
+import (
+ "archive/zip"
+ "encoding/json"
+ "fmt"
+ "io"
+ "os"
+ "path/filepath"
+ "strings"
+
+ cp "github.com/otiai10/copy"
+ "gopkg.in/yaml.v2"
+)
+
+func replaceStringsInDirectory(directory, stringToFind, stringToReplace string) error {
+ files, err := os.ReadDir(directory)
+ checkError(err)
+ for _, file := range files {
+ filePath := filepath.Join(directory, file.Name())
+ if file.IsDir() {
+ if err := replaceStringsInDirectory(filePath, stringToFind, stringToReplace); err != nil {
+ return err
+ }
+ } else {
+ if err := replaceString(filePath, stringToFind, stringToReplace); err != nil {
+ return err
+ }
+ }
+ }
+ return nil
+}
+
+func replaceString(fileLocation, stringToFind, stringToReplace string) error {
+ fileContent, err := os.ReadFile(fileLocation)
+ checkError(err)
+ if !strings.Contains(string(fileContent), stringToFind) {
+ return nil
+ }
+ updatedContent := strings.ReplaceAll(string(fileContent), stringToFind, stringToReplace)
+ err = os.WriteFile(fileLocation, []byte(updatedContent), 0644)
+ checkError(err)
+ return nil
+}
+
+func updateChartMuseumUri(directory string) error {
+ CHART_MUSEUM_GET_URI := "http://localhost:8879/charts"
+ CHART_MUSEUM_POST_URI := "http://localhost:8879/charts/api/charts"
+ err := replaceStringsInDirectory(directory, "UPDATE_THIS_CHART_MUSEUM_POST_CHARTS_URI", CHART_MUSEUM_POST_URI)
+ checkError(err)
+ err = replaceStringsInDirectory(directory, "UPDATE_THIS_CHART_MUSEUM_GET_CHARTS_URI", CHART_MUSEUM_GET_URI)
+ checkError(err)
+ return nil
+}
+
+func generateCsarPackage(directory, fileName string) error {
+ csarFile, err := os.Create(fileName)
+ checkError(err)
+ defer csarFile.Close()
+ csarWriter := zip.NewWriter(csarFile)
+ defer csarWriter.Close()
+ err = filepath.Walk(directory, func(filePath string, fileInfo os.FileInfo, err error) error {
+ if fileInfo.IsDir() {
+ return nil
+ }
+ relPath, err := filepath.Rel(directory, filePath)
+ checkError(err)
+ relPath = strings.ReplaceAll(relPath, "\\", "/")
+ zipFile, err := csarWriter.Create(relPath)
+ checkError(err)
+ fsFile, err := os.Open(filePath)
+ checkError(err)
+ _, err = io.Copy(zipFile, fsFile)
+ checkError(err)
+ defer fsFile.Close()
+ return nil
+ })
+ checkError(err)
+ return nil
+}
+
+func createCsarAndCopy(directory string, paths ...string) error {
+ packageName := directory + ".csar"
+ err := generateCsarPackage(directory, packageName)
+ checkError(err)
+ for _, path := range paths {
+ fmt.Printf("Copying %s to %s \n", packageName, path)
+ copy(packageName, path+packageName)
+ }
+ os.Remove(packageName)
+ os.RemoveAll(directory)
+ return nil
+}
+
+func copy(srcFile string, targetFile string) {
+ data, err := os.ReadFile(srcFile)
+ checkError(err)
+ err = os.WriteFile(targetFile, data, 0644)
+ checkError(err)
+}
+
+func checkError(err error) error {
+ if err != nil {
+ return err
+ }
+ return nil
+}
+
+func removeYamlElement(filePath string, elements ...string) error {
+ fileData, err := os.ReadFile(filePath)
+ checkError(err)
+ var yamlContent map[interface{}]interface{}
+ err = yaml.Unmarshal(fileData, &yamlContent)
+ checkError(err)
+ index := len(elements) - 1
+ dataMap := yamlContent
+ for i := 0; i < index; i++ {
+ resultMap, ok := dataMap[elements[i]].(map[interface{}]interface{})
+ if !ok {
+ return nil
+ }
+ dataMap = resultMap
+ }
+ delete(dataMap, elements[index])
+ updatedYamlContent, err := yaml.Marshal(&yamlContent)
+ checkError(err)
+ err = os.WriteFile(filePath, updatedYamlContent, 0644)
+ checkError(err)
+ return nil
+}
+
+func addJsonElement(filePath string, element string, index string) error {
+ fileData, err := os.ReadFile(filePath)
+ checkError(err)
+ var jsonContent map[string]interface{}
+ err = json.Unmarshal(fileData, &jsonContent)
+ checkError(err)
+ array, ok := jsonContent[index].([]interface{})
+ if !ok {
+ return nil
+ }
+ jsonContent[index] = append(array, element)
+ updatedJsonContent, err := json.Marshal(&jsonContent)
+ checkError(err)
+ err = os.WriteFile(filePath, updatedJsonContent, 0644)
+ checkError(err)
+ return nil
+}
+
+func main() {
+
+ SAMPLE_RAPP_GENERATOR := "sample-rapp-generator"
+ RAPP_BASE_PACKAGE := "rapp-all"
+ TEST_RESOURCES := "/src/test/resources/"
+ ACM_TEST_RESOURCES := "../rapp-manager-acm" + TEST_RESOURCES
+ APPLICATION_TEST_RESOURCES := "../rapp-manager-application" + TEST_RESOURCES
+ MODELS_TEST_RESOURCES := "../rapp-manager-models" + TEST_RESOURCES
+ DME_TEST_RESOURCES := "../rapp-manager-dme" + TEST_RESOURCES
+ SME_TEST_RESOURCES := "../rapp-manager-sme" + TEST_RESOURCES
+
+ if err := os.Chdir("../../../" + SAMPLE_RAPP_GENERATOR); err != nil {
+ fmt.Println("Error changing working directory:", err)
+ return
+ }
+
+ fmt.Println("Generating valid rApp package... ")
+ VALID_RAPP_PACKAGE_FOLDER_NAME := "valid-rapp-package"
+ cp.Copy(RAPP_BASE_PACKAGE, VALID_RAPP_PACKAGE_FOLDER_NAME)
+ updateChartMuseumUri(VALID_RAPP_PACKAGE_FOLDER_NAME)
+ createCsarAndCopy(VALID_RAPP_PACKAGE_FOLDER_NAME, ACM_TEST_RESOURCES, DME_TEST_RESOURCES, SME_TEST_RESOURCES, MODELS_TEST_RESOURCES, APPLICATION_TEST_RESOURCES)
+
+ fmt.Println("Generating valid rApp package without artifacts...")
+ VALID_RAPP_PACKAGE_NO_ARTIFACTS_FOLDER_NAME := "valid-rapp-package-no-artifacts"
+ cp.Copy(RAPP_BASE_PACKAGE, VALID_RAPP_PACKAGE_NO_ARTIFACTS_FOLDER_NAME)
+ updateChartMuseumUri(VALID_RAPP_PACKAGE_NO_ARTIFACTS_FOLDER_NAME)
+ removeYamlElement(VALID_RAPP_PACKAGE_NO_ARTIFACTS_FOLDER_NAME+"/Definitions/asd.yaml", "topology_template", "node_templates", "applicationServiceDescriptor", "artifacts")
+ createCsarAndCopy(VALID_RAPP_PACKAGE_NO_ARTIFACTS_FOLDER_NAME, APPLICATION_TEST_RESOURCES)
+
+ fmt.Println("Generating invalid rApp package...")
+ INVALID_RAPP_PACKAGE_FOLDER_NAME := "invalid-rapp-package"
+ cp.Copy(RAPP_BASE_PACKAGE, INVALID_RAPP_PACKAGE_FOLDER_NAME)
+ os.RemoveAll(INVALID_RAPP_PACKAGE_FOLDER_NAME + "/Files")
+ os.RemoveAll(INVALID_RAPP_PACKAGE_FOLDER_NAME + "/Artifacts")
+ createCsarAndCopy(INVALID_RAPP_PACKAGE_FOLDER_NAME, MODELS_TEST_RESOURCES, APPLICATION_TEST_RESOURCES)
+
+ fmt.Println("Generating invalid rApp package without tosca...")
+ INVALID_RAPP_PACKAGE_NO_TOSCA_FOLDER_NAME := "invalid-rapp-package-no-tosca"
+ cp.Copy(RAPP_BASE_PACKAGE, INVALID_RAPP_PACKAGE_NO_TOSCA_FOLDER_NAME)
+ os.Remove(INVALID_RAPP_PACKAGE_NO_TOSCA_FOLDER_NAME + "/TOSCA-Metadata/TOSCA.meta")
+ createCsarAndCopy(INVALID_RAPP_PACKAGE_NO_TOSCA_FOLDER_NAME, MODELS_TEST_RESOURCES)
+
+ fmt.Println("Generating invalid rApp package without asd yaml...")
+ INVALID_RAPP_PACKAGE_NO_ASD_YAML_FOLDER_NAME := "invalid-rapp-package-no-asd-yaml"
+ cp.Copy(RAPP_BASE_PACKAGE, INVALID_RAPP_PACKAGE_NO_ASD_YAML_FOLDER_NAME)
+ os.Remove(INVALID_RAPP_PACKAGE_NO_ASD_YAML_FOLDER_NAME + "/Definitions/asd.yaml")
+ createCsarAndCopy(INVALID_RAPP_PACKAGE_NO_ASD_YAML_FOLDER_NAME, MODELS_TEST_RESOURCES)
+
+ fmt.Println("Generating invalid rApp package without ACM composition...")
+ INVALID_RAPP_PACKAGE_NO_ACM_COMPOSITION_FOLDER_NAME := "invalid-rapp-package-no-acm-composition"
+ cp.Copy(RAPP_BASE_PACKAGE, INVALID_RAPP_PACKAGE_NO_ACM_COMPOSITION_FOLDER_NAME)
+ os.Remove(INVALID_RAPP_PACKAGE_NO_ACM_COMPOSITION_FOLDER_NAME + "/Files/Acm/definition/compositions.json")
+ createCsarAndCopy(INVALID_RAPP_PACKAGE_NO_ACM_COMPOSITION_FOLDER_NAME, MODELS_TEST_RESOURCES)
+
+ fmt.Println("Generating invalid rApp package without Artifacts...")
+ INVALID_RAPP_PACKAGE_MISSING_ARTIFACT_FOLDER_NAME := "invalid-rapp-package-missing-artifact"
+ cp.Copy(RAPP_BASE_PACKAGE, INVALID_RAPP_PACKAGE_MISSING_ARTIFACT_FOLDER_NAME)
+ os.Remove(INVALID_RAPP_PACKAGE_MISSING_ARTIFACT_FOLDER_NAME + "/Artifacts/Deployment/HELM/orufhrecovery-1.0.0.tgz")
+ createCsarAndCopy(INVALID_RAPP_PACKAGE_MISSING_ARTIFACT_FOLDER_NAME, MODELS_TEST_RESOURCES)
+
+ fmt.Println("Generating invalid rApp package with empty asd yaml...")
+ INVALID_RAPP_PACKAGE_EMPTY_ASD_FOLDER_NAME := "invalid-rapp-package-empty-asd-yaml"
+ cp.Copy(RAPP_BASE_PACKAGE, INVALID_RAPP_PACKAGE_EMPTY_ASD_FOLDER_NAME)
+ os.Truncate(INVALID_RAPP_PACKAGE_EMPTY_ASD_FOLDER_NAME+"/Definitions/asd.yaml", 0)
+ createCsarAndCopy(INVALID_RAPP_PACKAGE_EMPTY_ASD_FOLDER_NAME, MODELS_TEST_RESOURCES)
+
+ fmt.Println("Generating valid rApp package with new dme info type...")
+ VALID_RAPP_PACKAGE_NEW_INFO_TYPE_FOLDER_NAME := "valid-rapp-package-new-info-type"
+ DME_PRODUCER_FILE := VALID_RAPP_PACKAGE_NEW_INFO_TYPE_FOLDER_NAME + "/Files/Dme/infoproducers/json-file-data-producer.json"
+ cp.Copy(RAPP_BASE_PACKAGE, VALID_RAPP_PACKAGE_NEW_INFO_TYPE_FOLDER_NAME)
+ addJsonElement(DME_PRODUCER_FILE, "new-info-type-not-available", "supported_info_types")
+ createCsarAndCopy(VALID_RAPP_PACKAGE_NEW_INFO_TYPE_FOLDER_NAME, DME_TEST_RESOURCES)
+
+}
Code Review / nonrtric / plt / rappmanager.git / blobdiff