p2p-rendering-computation/client/GroupTrackContainer.go

package client

import (
	"encoding/json"
	"errors"
	"git.sr.ht/~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
}

// 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
}