1 files changed, 256 insertions, 0 deletions
diff --git a/sms-service/src/preload/preload.go b/sms-service/src/preload/preload.go
new file mode 100644
index 0000000..cbf345f
--- /dev/null
+++ b/sms-service/src/preload/preload.go
@@ -0,0 +1,256 @@
+/*
+ * Copyright 2018 Intel Corporation, Inc
+ *
+ * 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.
+ */
+
+package main
+
+import (
+	"bytes"
+	"crypto/tls"
+	"crypto/x509"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"net/url"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"time"
+
+	pkgerrors "github.com/pkg/errors"
+)
+
+//DataJSON stores a list of domains from JSON file
+type DataJSON struct {
+	//Support single domain: {} structure in JSON
+	Domain SecretDomainJSON `json:"domain,omitempty"`
+	//Support plural domains: [{}] structure in JSON
+	Domains []SecretDomainJSON `json:"domains,omitempty"`
+}
+
+//SecretDomainJSON stores a name for the Domain and a list of Secrets
+type SecretDomainJSON struct {
+	Name    string       `json:"name"`
+	Secrets []SecretJSON `json:"secrets"`
+}
+
+//SecretJSON stores a name for the Secret and a list of Values
+type SecretJSON struct {
+	Name   string                 `json:"name"`
+	Values map[string]interface{} `json:"values"`
+}
+
+//Processes the JSON file and returns a DataJSON struct
+func processJSONFile(name string) (DataJSON, error) {
+
+	data, err := ioutil.ReadFile(name)
+	if err != nil {
+		return DataJSON{}, pkgerrors.Cause(err)
+	}
+
+	d := DataJSON{}
+	err = json.Unmarshal(data, &d)
+	if err != nil {
+		return DataJSON{}, pkgerrors.Cause(err)
+	}
+
+	return d, nil
+}
+
+type smsClient struct {
+	BaseURL *url.URL
+	//In seconds
+	Timeout    int
+	CaCertPath string
+
+	httpClient *http.Client
+}
+
+func (c *smsClient) init() error {
+
+	skipVerify := false
+	caCert, err := ioutil.ReadFile(c.CaCertPath)
+	if err != nil {
+		fmt.Println(pkgerrors.Cause(err))
+		fmt.Println("Using Insecure Server Verification")
+		skipVerify = true
+	}
+
+	tlsConfig := &tls.Config{
+		MinVersion: tls.VersionTLS12,
+	}
+
+	tlsConfig.InsecureSkipVerify = skipVerify
+
+	// Add cert information when skipVerify is false
+	if skipVerify == false {
+		caCertPool := x509.NewCertPool()
+		caCertPool.AppendCertsFromPEM(caCert)
+		tlsConfig.RootCAs = caCertPool
+	}
+
+	tr := &http.Transport{
+		TLSClientConfig: tlsConfig,
+	}
+
+	c.httpClient = &http.Client{
+		Transport: tr,
+		Timeout:   time.Duration(c.Timeout) * time.Second,
+	}
+
+	return nil
+}
+
+func (c *smsClient) sendPostRequest(relURL string, message map[string]interface{}) error {
+
+	rel, err := url.Parse(relURL)
+	if err != nil {
+		return pkgerrors.Cause(err)
+	}
+	u := c.BaseURL.ResolveReference(rel)
+
+	body, err := json.Marshal(message)
+	if err != nil {
+		return pkgerrors.Cause(err)
+	}
+
+	resp, err := c.httpClient.Post(u.String(), "application/json", bytes.NewBuffer(body))
+	if err != nil {
+		return pkgerrors.Cause(err)
+	}
+
+	if resp.StatusCode >= 400 && resp.StatusCode < 600 {
+		// Request Failed
+		errText, _ := ioutil.ReadAll(resp.Body)
+		return pkgerrors.Errorf("Request Failed with: %s and Error: %s",
+			resp.Status, string(errText))
+	}
+
+	return nil
+}
+
+func (c *smsClient) createDomain(domain string) error {
+
+	message := map[string]interface{}{
+		"name": domain,
+	}
+	url := "/v1/sms/domain"
+	err := c.sendPostRequest(url, message)
+	if err != nil {
+		return pkgerrors.Cause(err)
+	}
+	return nil
+}
+
+func (c *smsClient) createSecret(domain string, secret string,
+
+	values map[string]interface{}) error {
+	message := map[string]interface{}{
+		"name":   secret,
+		"values": values,
+	}
+
+	url := "/v1/sms/domain/" + strings.TrimSpace(domain) + "/secret"
+	err := c.sendPostRequest(url, message)
+	if err != nil {
+		return pkgerrors.Cause(err)
+	}
+
+	return nil
+}
+
+//uploadToSMS reads through the domain or domains and uploads
+//their corresponding secrets to SMS service
+func (c *smsClient) uploadToSMS(data DataJSON) error {
+
+	var ldata []SecretDomainJSON
+
+	//Check if Domain is empty
+	if strings.TrimSpace(data.Domain.Name) != "" {
+		ldata = append(ldata, data.Domain)
+	} else if len(data.Domains) != 0 {
+		//Check if plural Domains are empty
+		ldata = append(ldata, data.Domains...)
+	} else {
+		return pkgerrors.New("Invalid JSON Data. No domain or domains found")
+	}
+
+	for _, d := range ldata {
+		err := c.createDomain(d.Name)
+		if err != nil {
+			return pkgerrors.Cause(err)
+		}
+
+		for _, s := range d.Secrets {
+			err = c.createSecret(d.Name, s.Name, s.Values)
+			if err != nil {
+				return pkgerrors.Cause(err)
+			}
+		}
+	}
+
+	return nil
+}
+
+func main() {
+
+	cacert := flag.String("cacert", "/sms/certs/aaf_root_ca.cer",
+		"Path to the CA Certificate file")
+	serviceurl := flag.String("serviceurl", "https://aaf-sms.onap",
+		"Url for the SMS Service")
+	serviceport := flag.Int("serviceport", 10443,
+		"Service port if its different than the default")
+	jsondir := flag.String("jsondir", ".",
+		"Folder containing json files to upload")
+
+	flag.Parse()
+
+	files, err := ioutil.ReadDir(*jsondir)
+	if err != nil {
+		log.Fatal(pkgerrors.Cause(err))
+	}
+
+	serviceURL, err := url.Parse(*serviceurl + ":" + strconv.Itoa(*serviceport))
+	if err != nil {
+		log.Fatal(pkgerrors.Cause(err))
+	}
+
+	client := &smsClient{
+		Timeout:    30,
+		BaseURL:    serviceURL,
+		CaCertPath: *cacert,
+	}
+	client.init()
+
+	for _, file := range files {
+		if filepath.Ext(file.Name()) == ".json" {
+			fmt.Println("Processing   ", file.Name())
+			d, err := processJSONFile(file.Name())
+			if err != nil {
+				log.Printf("Error Reading %s : %s", file.Name(), pkgerrors.Cause(err))
+				continue
+			}
+
+			err = client.uploadToSMS(d)
+			if err != nil {
+				log.Printf("Error Uploading %s : %s", file.Name(), pkgerrors.Cause(err))
+				continue
+			}
+		}
+	}
+}