added functionality to trigger ansibles from cli

plugin/plugin.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"git.sr.ht/~akilan1999/p2p-rendering-computation/client"
 	"git.sr.ht/~akilan1999/p2p-rendering-computation/config"
 	"gopkg.in/yaml.v2"
 	"io/ioutil"
@@ -230,3 +231,39 @@ func ReadHost(filename string) (*Host, error) {
 
 	return c, nil
 }
+
+// RunPluginCli Runs ansible plugin based on plugin name and contianer name which
+// is derived from the tracked containers file
+func RunPluginCli(PluginName string, ContainerID string) error {
+	// Gets container information based on container ID
+	ContainerInformation, err := client.GetContainerInformation(ContainerID)
+	if err != nil {
+		return err
+	}
+
+	// Setting Up IP's for which the plugins will be executed
+	var ExecuteIPs []*ExecuteIP
+	var ExecuteIP ExecuteIP
+	// Getting port no of SSH port
+	for _,port := range ContainerInformation.Container.Ports.PortSet {
+		if port.PortName == "SSH" {
+			ExecuteIP.SSHPortNo = fmt.Sprint(port.ExternalPort)
+			break
+		}
+	}
+	// Handle error if SSH port is not provided
+	if ExecuteIP.SSHPortNo == "" {
+		return errors.New("SSH port not found")
+	}
+	// IP address of the container 
+	ExecuteIP.IPAddress = ContainerInformation.IpAddress
+    // Append IP to list of executor IP
+	ExecuteIPs = append(ExecuteIPs, &ExecuteIP)
+	// Run plugin to execute plugin
+	_, err = RunPlugin(PluginName,ExecuteIPs)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

plugin/plugin_test.go

@@ -2,6 +2,7 @@ package plugin
 
 import (
 	"fmt"
+	"git.sr.ht/~akilan1999/p2p-rendering-computation/client"
 	"git.sr.ht/~akilan1999/p2p-rendering-computation/config"
 	"git.sr.ht/~akilan1999/p2p-rendering-computation/server/docker"
 	"strconv"
@@ -30,7 +31,7 @@ func TestRunPlugin(t *testing.T) {
 
 	//Test IP 1 configuration
 	testip1.IPAddress = "0.0.0.0"
-	testip1.SSHPortNo = strconv.Itoa(container1.SSHPort)
+	testip1.SSHPortNo = strconv.Itoa(container1.Ports.PortSet[0].ExternalPort)
 
 	// Create docker container and get SSH port
 	container2 ,err := docker.BuildRunContainer(0,"false","")
@@ -40,7 +41,7 @@ func TestRunPlugin(t *testing.T) {
 	}
 	//Test IP 2 configuration
 	testip2.IPAddress = "0.0.0.0"
-	testip2.SSHPortNo = strconv.Itoa(container2.SSHPort)
+	testip2.SSHPortNo = strconv.Itoa(container2.Ports.PortSet[0].ExternalPort)
 
 	testips = append(testips, &testip1)
 	testips = append(testips, &testip2)
@@ -89,4 +90,43 @@ func TestExecuteIP_ModifyHost(t *testing.T) {
 		fmt.Println(err)
 		t.Fail()
 	}
+}
+
+// Test to ensure the cli function runs as intended and executes
+// the test ansible script
+func TestRunPluginCli(t *testing.T) {
+	// Create docker container and get SSH port
+	container1 ,err := docker.BuildRunContainer(0,"false","")
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
+
+	// Ensuring created container is the added to the tracked list
+	err = client.AddTrackContainer(container1, "0.0.0.0")
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
+
+	// Running test Ansible script
+	err = RunPluginCli("TestAnsible", container1.ID)
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
+
+	// Removes container information from the tracker IP addresses
+	err = client.RemoveTrackedContainer(container1.ID)
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
+
+	// Removing container1 after Ansible is executed
+	err = docker.StopAndRemoveContainer(container1.ID)
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
 }