package docker

import (
	//log"
	"io/ioutil"
	"io"
	"context"
	"bytes"
	"os"
	"archive/tar"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client"
)


type SSHreturn struct {
	Port     string `bson:port`
	Username string `bson:username`
    Password string `bson:password`
}

// Runs a docker contianer with default settings 
// TODO implement with public keys 
func RunVM()(interface{},error) {
	client, err := client.NewEnvClient()
	if err != nil {
		//log.Fatalf("Unable to create docker client: %s", err)
		return nil,err
	}

	// Client, imagename and Dockerfile location
	tags := []string{"p2p-ubuntu"}
	dockerfile := "./server/docker/containers/docker-ubuntu-sshd/Dockerfile"
	err = BuildImage(client, tags, dockerfile)
	if err != nil {
		//log.Println(err)
		return nil,err
	}

	// TODO Run contianer 

	sshreturn := new(SSHreturn)
	
	sshreturn.Port = "1003"
	sshreturn.Password = "password"
	sshreturn.Username = "master"
	
	return sshreturn, nil

}

// Taken from (https://medium.com/@Frikkylikeme/controlling-docker-with-golang-code-b213d9699998)
func BuildImage(client *client.Client, tags []string, dockerfile string) error{
	ctx := context.Background()

	// Create a buffer 
	buf := new(bytes.Buffer)
	tw := tar.NewWriter(buf)
	defer tw.Close()

	// Create a filereader
	dockerFileReader, err := os.Open(dockerfile)
	if err != nil {
		return err
	}

	// Read the actual Dockerfile 
	readDockerFile, err := ioutil.ReadAll(dockerFileReader)
	if err != nil {
		return err
	}

	// Make a TAR header for the file
	tarHeader := &tar.Header{
		Name: dockerfile,
		Size: int64(len(readDockerFile)),
	}

	// Writes the header described for the TAR file
	err = tw.WriteHeader(tarHeader)
    if err != nil {
		return err
    }

	// Writes the dockerfile data to the TAR file
    _, err = tw.Write(readDockerFile)
    if err != nil {
		return err
    }

    dockerFileTarReader := bytes.NewReader(buf.Bytes())

	// Define the build options to use for the file
	// https://godoc.org/github.com/docker/docker/api/types#ImageBuildOptions
	buildOptions := types.ImageBuildOptions{
        Context:    dockerFileTarReader,
        Dockerfile: dockerfile,
        Remove:     true,
		Tags: 		tags,
	}

	// Build the actual image
	imageBuildResponse, err := client.ImageBuild(
        ctx,
        dockerFileTarReader,
		buildOptions, 
	)	

	if err != nil {
		return err
	}

	// Read the STDOUT from the build process
	defer imageBuildResponse.Body.Close()
	_, err = io.Copy(os.Stdout, imageBuildResponse.Body)
	if err != nil {
		return err
	}

	return nil
}