path: root/src/rsync/pkg/context/context.go

/*
 * Copyright 2020 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 context

import (
	"encoding/json"
        "fmt"
        "log"
        "os"
        "sync"
        "strings"
        "github.com/onap/multicloud-k8s/src/orchestrator/pkg/appcontext"
        pkgerrors "github.com/pkg/errors"
        res "github.com/onap/multicloud-k8s/src/rsync/pkg/resource"
        con "github.com/onap/multicloud-k8s/src/rsync/pkg/connector"
        "github.com/onap/multicloud-k8s/src/rsync/pkg/app"
)

type CompositeAppContext struct {
	cid    interface{}
	appsorder      string
	appsdependency string
	appsmap []instMap
}
type clusterInfo struct {
	name   string
	resorder      string
	resdependency string
	ressmap []instMap
}
type instMap struct {
	name    string
	depinfo string
	status  string
	rerr    error
	clusters []clusterInfo
}

const basePath string = "/tmp/rsync/"

func getInstMap(order string, dependency string, level string) ([]instMap, error) {

        if order == "" {
              return nil, pkgerrors.Errorf("Not a valid order value")
        }
        if dependency == "" {
              return nil, pkgerrors.Errorf("Not a valid dependency value")
        }

        if !(level == "app" || level == "res") {
              return nil, pkgerrors.Errorf("Not a valid level name given to create map")
        }


        var aov map[string]interface{}
        json.Unmarshal([]byte(order), &aov)

        s := fmt.Sprintf("%vorder", level)
        appso := aov[s].([]interface{})
        var instmap = make([]instMap, len(appso))

        var adv map[string]interface{}
        json.Unmarshal([]byte(dependency), &adv)
        s = fmt.Sprintf("%vdependency", level)
        appsd := adv[s].(map[string]interface{})
        for i, u := range appso {
                instmap[i] = instMap{u.(string), appsd[u.(string)].(string), "none", nil, nil}
        }

        return instmap, nil
}

func deleteResource(clustername string, resname string, respath string) error {
	k8sClient := app.KubernetesClient{}
	err := k8sClient.Init(clustername, resname)
	if err != nil {
		log.Println("Init failed: " + err.Error())
		return err
	}

	var c con.KubernetesConnector
	c = &k8sClient
	var gp res.Resource
	err = gp.Delete(respath, resname, "default", c)
	if err != nil {
		log.Println("Delete resource failed: " +  err.Error())
		return err
	}
	log.Println("Resource succesfully deleted")
	return nil

}

func createResource(clustername string, resname string, respath string) error {
	k8sClient := app.KubernetesClient{}
	err := k8sClient.Init(clustername, resname)
	if err != nil {
		log.Println("Client init failed: " + err.Error())
		return err
	}

	var c con.KubernetesConnector
	c = &k8sClient
	var gp res.Resource
	_, err = gp.Create(respath,"default", c)
	if err != nil {
		log.Println("Create failed: " +  err.Error())
		return err
	}
	log.Println("Resource succesfully created")
	return nil

}

func terminateResource(ac appcontext.AppContext, resmap instMap, appname string, clustername string) error {
	var resPath string = basePath + appname + "/" + clustername + "/resources/"
	rh, err := ac.GetResourceHandle(appname, clustername, resmap.name)
	if err != nil {
		return err
	}

	resval, err := ac.GetValue(rh)
	if err != nil {
		return err
	}

	if resval != "" {
		if _, err := os.Stat(resPath); os.IsNotExist(err) {
			err = os.MkdirAll(resPath, 0755)
			if err != nil {
				return err
			}
		}
		resPath := resPath + resmap.name + ".yaml"
		f, err := os.Create(resPath)
		defer f.Close()
		if err != nil {
			return err
		}
		_, err = f.WriteString(resval.(string))
		if err != nil {
			return err
		}
		result := strings.Split(resmap.name, "+")
		if result[0] == "" {
			return pkgerrors.Errorf("Resource name is nil")
		}
		err = deleteResource(clustername, result[0], resPath)
		if err != nil {
			return err
		}
		//defer os.Remove(resPath)
	} else {
		return pkgerrors.Errorf("Resource value is nil")
	}

	return nil

}

func instantiateResource(ac appcontext.AppContext, resmap instMap, appname string, clustername string) error {
	var resPath string = basePath + appname + "/" + clustername + "/resources/"
	rh, err := ac.GetResourceHandle(appname, clustername, resmap.name)
	if err != nil {
		return err
	}

	resval, err := ac.GetValue(rh)
	if err != nil {
		return err
	}

	if resval != "" {
		if _, err := os.Stat(resPath); os.IsNotExist(err) {
			err = os.MkdirAll(resPath, 0755)
			if err != nil {
				return err
			}
		}
		resPath := resPath + resmap.name + ".yaml"
		f, err := os.Create(resPath)
		defer f.Close()
		if err != nil {
			return err
		}
		_, err = f.WriteString(resval.(string))
		if err != nil {
			return err
		}
		result := strings.Split(resmap.name, "+")
		if result[0] == "" {
			return pkgerrors.Errorf("Resource name is nil")
		}
		err = createResource(clustername, result[0], resPath)
		if err != nil {
			return err
		}
		//defer os.Remove(resPath)
	} else {
		return pkgerrors.Errorf("Resource value is nil")
	}

	return nil

}

func  terminateResources(ac appcontext.AppContext, ressmap []instMap, appname string, clustername string) error {
        var wg sync.WaitGroup
        var chans = make([]chan int, len(ressmap))
        for l := range chans {
                chans[l] = make(chan int)
        }
        for i:=0; i<len(ressmap); i++ {
                wg.Add(1)
                go func(index int) {
                        if ressmap[index].depinfo == "go" {
                                ressmap[index].status = "start"
                        } else {
                                ressmap[index].status = "waiting"
                                c := <- chans[index]
                                if c != index {
					ressmap[index].status = "error"
                                        ressmap[index].rerr = pkgerrors.Errorf("channel does not match")
					wg.Done()
					return
                                }
                                ressmap[index].status = "start"
                        }
                        ressmap[index].rerr = terminateResource(ac, ressmap[index], appname, clustername)
                        ressmap[index].status = "done"
                        waitstr := fmt.Sprintf("wait on %v",ressmap[index].name)
                        for j:=0; j<len(ressmap); j++ {
                                if ressmap[j].depinfo == waitstr {
                                        chans[j] <- j
                                }
                        }
                        wg.Done()
                }(i)
        }
        wg.Wait()
        for k:=0; k<len(ressmap); k++ {
                if ressmap[k].rerr != nil {
                        return pkgerrors.Errorf("error during resources termination")
                }
        }
        return nil

}

func  instantiateResources(ac appcontext.AppContext, ressmap []instMap, appname string, clustername string) error {
        var wg sync.WaitGroup
        var chans = make([]chan int, len(ressmap))
        for l := range chans {
                chans[l] = make(chan int)
        }
        for i:=0; i<len(ressmap); i++ {
                wg.Add(1)
                go func(index int) {
                        if ressmap[index].depinfo == "go" {
                                ressmap[index].status = "start"
                        } else {
                                ressmap[index].status = "waiting"
                                c := <- chans[index]
                                if c != index {
					ressmap[index].status = "error"
                                        ressmap[index].rerr = pkgerrors.Errorf("channel does not match")
					wg.Done()
					return
                                }
                                ressmap[index].status = "start"
                        }
                        ressmap[index].rerr = instantiateResource(ac, ressmap[index], appname, clustername)
                        ressmap[index].status = "done"
                        waitstr := fmt.Sprintf("wait on %v",ressmap[index].name)
                        for j:=0; j<len(ressmap); j++ {
                                if ressmap[j].depinfo == waitstr {
                                        chans[j] <- j
                                }
                        }
                        wg.Done()
                }(i)
        }
        wg.Wait()
        for k:=0; k<len(ressmap); k++ {
                if ressmap[k].rerr != nil {
                        return pkgerrors.Errorf("error during resources instantiation")
                }
        }
        return nil

}

func terminateApp(ac appcontext.AppContext, appmap instMap) error {

        for i:=0; i<len(appmap.clusters); i++ {
		err := terminateResources(ac, appmap.clusters[i].ressmap, appmap.name,
				appmap.clusters[i].name)
		if err != nil {
			return err
		}
	}
	log.Println("Termination of app done: " + appmap.name)

	return nil

}


func instantiateApp(ac appcontext.AppContext, appmap instMap) error {

        for i:=0; i<len(appmap.clusters); i++ {
		err := instantiateResources(ac, appmap.clusters[i].ressmap, appmap.name,
				appmap.clusters[i].name)
		if err != nil {
			return err
		}
	}
	log.Println("Instantiation of app done: " + appmap.name)
	return nil

}

func  instantiateApps(ac appcontext.AppContext, appsmap []instMap) error {
	var wg sync.WaitGroup
	var chans = make([]chan int, len(appsmap))
	for l := range chans {
		chans[l] = make(chan int)
	}
        for i:=0; i<len(appsmap); i++ {
		wg.Add(1)
                go func(index int) {
                        if appsmap[index].depinfo == "go" {
				appsmap[index].status = "start"
			} else {
				appsmap[index].status = "waiting"
				c := <- chans[index]
				if c != index {
					appsmap[index].status = "error"
                                        appsmap[index].rerr = pkgerrors.Errorf("channel does not match")
					wg.Done()
					return
				}
				appsmap[index].status = "start"
			}
			appsmap[index].rerr = instantiateApp(ac, appsmap[index])
			appsmap[index].status = "done"
			waitstr := fmt.Sprintf("wait on %v",appsmap[index].name)
			for j:=0; j<len(appsmap); j++ {
				if appsmap[j].depinfo == waitstr {
					chans[j] <- j
				}
			}
			wg.Done()
                }(i)
        }
	wg.Wait()
	for k:=0; k<len(appsmap); k++ {
		if appsmap[k].rerr != nil {
			return pkgerrors.Errorf("error during apps instantiation")
		}
	}
	return nil

}

func (instca *CompositeAppContext) InstantiateComApp(cid interface{}) error {
	ac := appcontext.AppContext{}

	_, err := ac.LoadAppContext(cid)
        if err != nil {
                return err
        }
	instca.cid = cid

	appsorder, err := ac.GetAppInstruction("order")
        if err != nil {
                return err
        }
	instca.appsorder = appsorder.(string)
	appsdependency, err := ac.GetAppInstruction("dependency")
        if err != nil {
                return err
        }
	instca.appsdependency = appsdependency.(string)
        instca.appsmap, err = getInstMap(instca.appsorder,instca.appsdependency, "app")
        if err != nil {
                return err
        }

	for j:=0; j<len(instca.appsmap); j++ {
		clusternames, err := ac.GetClusterNames(instca.appsmap[j].name)
		if err != nil {
	                return err
		}
		instca.appsmap[j].clusters = make([]clusterInfo, len(clusternames))
		for k:=0; k<len(clusternames); k++ {
			instca.appsmap[j].clusters[k].name = clusternames[k]
			resorder, err := ac.GetResourceInstruction(
					instca.appsmap[j].name, clusternames[k], "order")
			if err != nil {
		                return err
			}
			instca.appsmap[j].clusters[k].resorder = resorder.(string)

			resdependency, err := ac.GetResourceInstruction(
					instca.appsmap[j].name, clusternames[k], "dependency")
			if err != nil {
		                return err
			}
			instca.appsmap[j].clusters[k].resdependency = resdependency.(string)

			instca.appsmap[j].clusters[k].ressmap, err = getInstMap(
				instca.appsmap[j].clusters[k].resorder,
				instca.appsmap[j].clusters[k].resdependency, "res")
			if err != nil {
		                return err
			}
		}
	}
	err = instantiateApps(ac, instca.appsmap)
        if err != nil {
                return err
        }

	return nil
}

// Delete all the apps
func  terminateApps(ac appcontext.AppContext, appsmap []instMap) error {
	var wg sync.WaitGroup
	var chans = make([]chan int, len(appsmap))
	for l := range chans {
		chans[l] = make(chan int)
	}
        for i:=0; i<len(appsmap); i++ {
		wg.Add(1)
                go func(index int) {
                        if appsmap[index].depinfo == "go" {
				appsmap[index].status = "start"
			} else {
				appsmap[index].status = "waiting"
				c := <- chans[index]
				if c != index {
					appsmap[index].status = "error"
                                        appsmap[index].rerr = pkgerrors.Errorf("channel does not match")
					wg.Done()
					return
				}
				appsmap[index].status = "start"
			}
			appsmap[index].rerr = terminateApp(ac, appsmap[index])
			appsmap[index].status = "done"
			waitstr := fmt.Sprintf("wait on %v",appsmap[index].name)
			for j:=0; j<len(appsmap); j++ {
				if appsmap[j].depinfo == waitstr {
					chans[j] <- j
				}
			}
			wg.Done()
                }(i)
        }
	wg.Wait()
	for k:=0; k<len(appsmap); k++ {
		if appsmap[k].rerr != nil {
			return pkgerrors.Errorf("error during apps instantiation")
		}
	}
	return nil

}
// Delete all the resources for a given context
func (instca *CompositeAppContext) TerminateComApp(cid interface{}) error {
	ac := appcontext.AppContext{}

	_, err := ac.LoadAppContext(cid)
        if err != nil {
                return err
        }
	instca.cid = cid

	appsorder, err := ac.GetAppInstruction("order")
        if err != nil {
                return err
        }
	instca.appsorder = appsorder.(string)
	appsdependency, err := ac.GetAppInstruction("dependency")
        if err != nil {
                return err
        }
	instca.appsdependency = appsdependency.(string)
        instca.appsmap, err = getInstMap(instca.appsorder,instca.appsdependency, "app")
        if err != nil {
                return err
        }

	for j:=0; j<len(instca.appsmap); j++ {
		clusternames, err := ac.GetClusterNames(instca.appsmap[j].name)
		if err != nil {
	                return err
		}
		instca.appsmap[j].clusters = make([]clusterInfo, len(clusternames))
		for k:=0; k<len(clusternames); k++ {
			instca.appsmap[j].clusters[k].name = clusternames[k]
			resorder, err := ac.GetResourceInstruction(
					instca.appsmap[j].name, clusternames[k], "order")
			if err != nil {
		                return err
			}
			instca.appsmap[j].clusters[k].resorder = resorder.(string)

			resdependency, err := ac.GetResourceInstruction(
					instca.appsmap[j].name, clusternames[k], "dependency")
			if err != nil {
		                return err
			}
			instca.appsmap[j].clusters[k].resdependency = resdependency.(string)

			instca.appsmap[j].clusters[k].ressmap, err = getInstMap(
				instca.appsmap[j].clusters[k].resorder,
				instca.appsmap[j].clusters[k].resdependency, "res")
			if err != nil {
		                return err
			}
		}
	}
	err = terminateApps(ac, instca.appsmap)
        if err != nil {
                return err
        }

	return nil

}