Adding Capif provider

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

diff --git a/provider/internal/gentools/commoncollector/commoncollector.go b/provider/internal/gentools/commoncollector/commoncollector.go
new file mode 100644 (file)
index 0000000..ab85e0a
--- /dev/null
+++ b/provider/internal/gentools/commoncollector/commoncollector.go
@@ -0,0 +1,119 @@
+// -
+//   ========================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 (
+       "bufio"
+       "flag"
+       "io/ioutil"
+       "log"
+       "os"
+       "strings"
+
+       "gopkg.in/yaml.v2"
+)
+
+var apiDir *string
+var common = map[interface{}]interface{}{
+       "openapi": "3.0.0",
+       "info": map[interface{}]interface{}{
+               "title":   "Common",
+               "version": "1.0.0",
+       },
+       "components": map[interface{}]interface{}{
+               "schemas": map[interface{}]interface{}{},
+       },
+}
+
+func main() {
+       apiDir = flag.String("apidir", "", "Directory containing API definitions to fix")
+       flag.Parse()
+
+       file, err := os.Open("definitions.txt")
+       if err != nil {
+               log.Fatalf("Error opening file: %v", err)
+       }
+       defer func(file *os.File) {
+               _ = file.Close()
+       }(file)
+
+       scanner := bufio.NewScanner(file)
+       components := common["components"]
+       cMap := components.(map[interface{}]interface{})
+       schemas := cMap["schemas"].(map[interface{}]interface{})
+       for scanner.Scan() {
+               name, data := getDependency(scanner.Text())
+               if name == "EthFlowDescription" {
+                       changeToLocalReference("fDir", "FlowDirection", data)
+               }
+               if name == "ReportingInformation" {
+                       changeToLocalReference("notifMethod", "NotificationMethod", data)
+               }
+               if name == "RelativeCartesianLocation" {
+                       properties := data["properties"].(map[interface{}]interface{})
+                       delete(properties, true)
+                       data["required"] = remove(data["required"].([]interface{}), 1)
+               }
+               schemas[name] = data
+       }
+
+       if err := scanner.Err(); err != nil {
+               log.Fatal(err)
+       }
+
+       modCommon, err := yaml.Marshal(common)
+       if err != nil {
+               log.Fatalf("Marshal: #%v ", err)
+       }
+       err = ioutil.WriteFile(*apiDir+"/"+"CommonData.yaml", modCommon, 0644)
+       if err != nil {
+               log.Fatalf("Error writing yamlFile. #%v ", err)
+       }
+}
+
+func changeToLocalReference(attrname, refName string, data map[interface{}]interface{}) {
+       properties := data["properties"].(map[interface{}]interface{})
+       ref := properties[attrname].(map[interface{}]interface{})
+       ref["$ref"] = "#/components/schemas/" + refName
+}
+
+func getDependency(s string) (string, map[interface{}]interface{}) {
+       info := strings.Split(s, "#")
+       yamlFile, err := ioutil.ReadFile(*apiDir + "/" + info[0])
+       if err != nil {
+               log.Fatalf("Error reading yamlFile. #%v ", err)
+       }
+       m := make(map[string]interface{})
+       err = yaml.Unmarshal(yamlFile, m)
+       if err != nil {
+               log.Fatalf("Unmarshal: %v", err)
+       }
+       components := m["components"]
+       cMap := components.(map[interface{}]interface{})
+       schemas := cMap["schemas"].(map[interface{}]interface{})
+       component := strings.Split(info[1], "/")
+       dep := schemas[component[3]].(map[interface{}]interface{})
+       return component[3], dep
+}
+
+func remove(slice []interface{}, s int) []interface{} {
+       return append(slice[:s], slice[s+1:]...)
+}