p2p-rendering-computation/client/GroupTrackContainer.go

package client

import (
	"encoding/json"
	"errors"
	"github.com/Akilan1999/p2p-rendering-computation/config"
	"github.com/google/uuid"
	"io/ioutil"
	"os"
)

// Groups Data Structure type
type Groups struct {
	GroupList []*Group `json:"Groups"`
}

// Group Information about a single group
type Group struct {
	ID                 string            `json:"ID"`
	TrackContainerList []*TrackContainer `json:"TrackContainer"`
	// Sneaky as required only when removing the element
	// Set when GetGroup function is called
	index int
}

// CreateGroup Creates a new group to add a set of track containers
func CreateGroup() (*Group, error) {
	// Creating variable of type new group
	var NewGroup Group
	// Generate new UUID for group ID
	id := uuid.New()
	// Add new group id and prepend with the string "grp"
	// The reason this is done is to differentiate between a
	// group ID and docker container ID
	NewGroup.ID = "grp" + id.String()
	// Adding the new group to the
	// GroupTrackContainer File
	err := NewGroup.AddGroupToFile()
	if err != nil {
		return nil, err
	}

	return &NewGroup, nil
}

// RemoveGroup Removes group based on the group ID provided
func RemoveGroup(GroupID string) error {
	// Read group information from the
	//grouptrackcontainer json file
	groups, err := ReadGroup()
	if err != nil {
		return err
	}
	// Gets Group struct based on group ID
	// provided
	group, err := GetGroup(GroupID)
	if err != nil {
		return err
	}
	// Remove Group struct from the groups variable
	groups.GroupList = append(groups.GroupList[:group.index], groups.GroupList[group.index+1:]...)

	// Write new groups to the grouptrackcontainer json file
	err = groups.WriteGroup()
	if err != nil {
		return err
	}

	return nil
}

// AddContainerToGroup Adds container information to the Group based on the Group ID
func AddContainerToGroup(ContainerID string, GroupID string) (*Group, error) {
	// Gets container information based on container ID provided
	containerInfo, err := GetContainerInformation(ContainerID)
	if err != nil {
		return nil, err
	}
	// Gets group information based on the group ID provided
	group, err := GetGroup(GroupID)
	if err != nil {
		return nil, err
	}
	// Adds container information the group
	group.AddContainer(containerInfo)

	// Get Groups information from reading the grouptrackcontainer.json file
	groups, err := ReadGroup()
	if err != nil {
		return nil, err
	}
	// Updating specific element in the group list with the added container
	groups.GroupList[group.index] = group
	// Write groups information on the grouptrackcontainer.json file
	err = groups.WriteGroup()
	if err != nil {
		return nil, err
	}

	return group, nil
}

// RemoveContainerGroup Remove Container from the group ID specified
func RemoveContainerGroup(ContainerID string, GroupID string) (*Group, error) {
	// Get container information based on container ID provided
	containerInfo, err := GetContainerInformation(ContainerID)
	if err != nil {
		return nil, err
	}
	// Gets group information based on the group ID provided
	group, err := GetGroup(GroupID)
	if err != nil {
		return nil, err
	}
	// Remove container from the appropriate group
	err = group.RemoveContainerGroup(containerInfo)
	if err != nil {
		return nil, err
	}
	// Get Groups information from reading the grouptrackcontainer.json file
	groups, err := ReadGroup()
	if err != nil {
		return nil, err
	}
	// Updating specific element in the group list with the remove container
	groups.GroupList[group.index] = group
	// Write groups information on the grouptrackcontainer.json file
	err = groups.WriteGroup()
	if err != nil {
		return nil, err
	}

	return group, nil
}

// RemoveContainerGroups Remove Container from groups (i.e which ever groups has information
// about that container). This is mostly called when a container is deleted or removed from
// the tracked container list
func RemoveContainerGroups(ContainerID string) error {
	// Get container information based on container ID provided
	containerInfo, err := GetContainerInformation(ContainerID)
	if err != nil {
		return err
	}
	// Get Groups information from reading the grouptrackcontainer.json file
	groups, err := ReadGroup()
	if err != nil {
		return err
	}
	// Removes container information from all groups it is found in
	err = groups.RemoveContainerGroups(containerInfo)
	if err != nil {
		return err
	}
	// Write groups information on the grouptrackcontainer.json file
	err = groups.WriteGroup()
	if err != nil {
		return err
	}

	return nil
}

// GetGroup Gets group information based on
// group id provided
func GetGroup(GroupID string) (*Group, error) {
	// Read group information from the
	//grouptrackcontainer json file
	groups, err := ReadGroup()
	if err != nil {
		return nil, err
	}
	// Iterate through the set of groups and
	// if the group ID matches then return it
	for i, group := range groups.GroupList {
		if group.ID == GroupID {
			group.index = i
			return group, nil
		}
	}

	return nil, errors.New("Group not found. ")
}

// AddGroupToFile Adds Group struct to the GroupTrackContainer File
func (grp *Group) AddGroupToFile() error {
	// Gets all group information from the
	// GroupTrackContainer JSON file
	groups, err := ReadGroup()
	if err != nil {
		return err
	}
	// Appending the newly created group
	groups.GroupList = append(groups.GroupList, grp)
	// Writing Group information to the GroupTrackContainer
	// JSON file
	err = groups.WriteGroup()
	if err != nil {
		return err
	}

	return nil
}

// ReadGroup Function reads grouptrackcontainers.json and converts
// result to Groups
func ReadGroup() (*Groups, error) {
	// Get Path from config
	config, err := config.ConfigInit()
	if err != nil {
		return nil, err
	}
	jsonFile, err := os.Open(config.GroupTrackContainersPath)
	// if we os.Open returns an error then handle it
	if err != nil {
		return nil, err
	}
	// defer the closing of our jsonFile so that we can parse it later on
	defer jsonFile.Close()
	// read our opened xmlFile as a byte array.
	byteValue, _ := ioutil.ReadAll(jsonFile)
	// we initialize our Users array
	var groups Groups
	// we unmarshal our byteArray which contains our
	// jsonFile's content into 'users' which we defined above
	json.Unmarshal(byteValue, &groups)
	return &groups, nil
}

// WriteGroup Function to write type Groups to the grouptrackcontainers.json file
func (grp *Groups) WriteGroup() error {
	file, err := json.MarshalIndent(grp, "", " ")
	if err != nil {
		return err
	}
	// Get Path from config
	config, err := config.ConfigInit()
	if err != nil {
		return err
	}
	// Writes to the appropriate file
	err = ioutil.WriteFile(config.GroupTrackContainersPath, file, 0644)
	if err != nil {
		return err
	}
	return nil
}

// AddContainer Adds a container to the Tracked container list of the group
func (grp *Group) AddContainer(Container *TrackContainer) error {
	grp.TrackContainerList = append(grp.TrackContainerList, Container)
	return nil
}

// RemoveContainerGroup Removes container information from the group
func (grp *Group) RemoveContainerGroup(Container *TrackContainer) error {
	// Iterating through all container in the Group of Tracked Container
	for i, container := range grp.TrackContainerList {
		// If the container ID matches then remove the container from the group
		if container.Id == Container.Id {
			grp.TrackContainerList = append(grp.TrackContainerList[:i], grp.TrackContainerList[i+1:]...)
		}
	}
	return nil
}

// RemoveContainerGroups removes container found in all groups
func (grp *Groups) RemoveContainerGroups(Container *TrackContainer) error {
	// Iterating through all groups
	for i, group := range grp.GroupList {
		// Removes the container in the following group
		// if it exists
		err := group.RemoveContainerGroup(Container)
		if err != nil {
			return err
		}
		// Set group information to list groups
		grp.GroupList[i] = group
	}
	return nil
}

// ModifyContainerGroups Modifies container information is all groups
// available
func (TC *TrackContainer) ModifyContainerGroups() error {
	group, err := ReadGroup()
	if err != nil {
		return err
	}
	// Iterate though all groups and modify the container
	// information in groups where the modified container
	// ID matches
	for i, _ := range group.GroupList {
		// Checking in each group if the modified container ID exists
		for j, _ := range group.GroupList[i].TrackContainerList {
			// If there is match then change them
			if group.GroupList[i].TrackContainerList[j].Id == TC.Id {
				group.GroupList[i].TrackContainerList[j] = TC
			}
		}
	}

	// Write modified result to the Groups track container JSON file
	err = group.WriteGroup()
	if err != nil {
		return err
	}

	return nil
}