Adding Capif provider

[nonrtric/plt/sme.git] / provider / internal / gentools / enumfixer / enumfixer.go

diff --git a/provider/internal/gentools/enumfixer/enumfixer.go b/provider/internal/gentools/enumfixer/enumfixer.go
new file mode 100644 (file)
index 0000000..e722875
--- /dev/null
+++ b/provider/internal/gentools/enumfixer/enumfixer.go
@@ -0,0 +1,117 @@
+// -
+//   ========================LICENSE_START=================================
+//   O-RAN-SC
+//   %%
+//   Copyright (C) 2022: Nordix Foundation
+//   %%
+//   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 (
+       "flag"
+       "fmt"
+       "io/ioutil"
+       "log"
+       "os"
+       "path/filepath"
+       "reflect"
+       "strconv"
+       "strings"
+
+       "gopkg.in/yaml.v2"
+)
+
+type Enum struct {
+       Enum        []string `yaml:"enum"`
+       Type        string   `yaml:"type"`
+       Description string   `yaml:"description"`
+}
+
+func main() {
+       var apiDir = flag.String("apidir", "", "Directory containing API definitions to fix")
+       flag.Parse()
+       err := filepath.Walk(*apiDir, fixEnums)
+       if err != nil {
+               fmt.Println(err)
+       }
+}
+
+func fixEnums(path string, info os.FileInfo, _ error) error {
+       if !info.IsDir() && strings.HasSuffix(info.Name(), ".yaml") {
+               yamlFile, err := ioutil.ReadFile(path)
+               if err != nil {
+                       log.Printf("yamlFile. Get err   #%v ", err)
+               }
+               m := make(map[string]interface{})
+               err = yaml.Unmarshal(yamlFile, m)
+               if err != nil {
+                       log.Fatalf("Unmarshal: %v", err)
+               }
+               components := m["components"]
+               if components != nil {
+                       cMap := components.(map[interface{}]interface{})
+                       if _, ok := cMap["schemas"].(map[interface{}]interface{}); ok {
+                               schemas := cMap["schemas"].(map[interface{}]interface{})
+                               for typeName, typeDef := range schemas {
+                                       tDMap := typeDef.(map[interface{}]interface{})
+                                       anyOf, ok := tDMap["anyOf"]
+                                       if ok {
+                                               aOSlice := anyOf.([]interface{})
+                                               correctEnum := Enum{}
+                                               mapInterface := aOSlice[0].(map[interface{}]interface{})
+                                               enumInterface := mapInterface["enum"]
+                                               if enumInterface != nil {
+                                                       is := enumInterface.([]interface{})
+                                                       var enumVals []string
+                                                       for i := 0; i < len(is); i++ {
+                                                               if reflect.TypeOf(is[i]).Kind() == reflect.String {
+                                                                       enumVals = append(enumVals, is[i].(string))
+
+                                                               } else if reflect.TypeOf(is[1]).Kind() == reflect.Int {
+                                                                       enumVals = append(enumVals, strconv.Itoa(is[i].(int)))
+                                                               }
+                                                       }
+                                                       correctEnum.Enum = enumVals
+                                                       correctEnum.Type = "string"
+                                                       description := tDMap["description"]
+                                                       if description != nil {
+                                                               correctEnum.Description = description.(string)
+                                                       } else {
+                                                               if aOSlice[1] != nil {
+                                                                       mapInterface = aOSlice[1].(map[interface{}]interface{})
+                                                                       description := mapInterface["description"]
+                                                                       if description != nil {
+                                                                               correctEnum.Description = description.(string)
+                                                                       }
+                                                               }
+                                                       }
+                                                       schemas[typeName] = correctEnum
+                                               }
+                                       }
+                               }
+                               modM, err := yaml.Marshal(m)
+                               if err != nil {
+                                       log.Printf("yamlFile. Get err   #%v ", err)
+                               }
+                               err = ioutil.WriteFile(path, modM, 0644)
+                               if err != nil {
+                                       log.Printf("yamlFile. Get err   #%v ", err)
+                               }
+                       }
+               }
+       }
+       return nil
+}