package xenserver

/* Taken from https://raw.githubusercontent.com/mitchellh/packer/master/builder/qemu/step_prepare_output_dir.go */

import (
	"github.com/mitchellh/multistep"
	"github.com/mitchellh/packer/packer"
	"os"
    "fmt"
    "net/http"
    "crypto/tls"
    "io"
)

type stepShutdownAndExport struct{}

func downloadFile(url, filename string) (err error) {

    // Create the file
    fh, err := os.Create(filename)
    if err != nil {
        return err
    }

    // Define a new transport which allows self-signed certs
    tr := &http.Transport{
        TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
    }

    // Create a client
    client := &http.Client{Transport: tr}

    // Create request and download file

    resp, err := client.Get(url)
    if err != nil {
        return err
    }

    defer resp.Body.Close()
    io.Copy(fh, resp.Body)

    return nil
}


func (stepShutdownAndExport) Run(state multistep.StateBag) multistep.StepAction {
	config := state.Get("config").(config)
	ui := state.Get("ui").(packer.Ui)
    client := state.Get("client").(XenAPIClient)
    instance_uuid := state.Get("instance_uuid").(string)

    instance, _ := client.GetVMByUuid(instance_uuid)

    ui.Say("Step: Shutdown and export VPX")

    // Shutdown the VM
    ui.Say("Shutting down the VM...")
    instance.CleanShutdown()

    //Export the VM

    export_url := fmt.Sprintf("https://%s/export?vm=%s&session_id=%s",
                              client.Host,
                              instance.Ref,
                              client.Session.(string),
                              )

    export_filename := fmt.Sprintf("%s/%s.xva", config.OutputDir, config.InstanceName)
    ui.Say("Getting metadata " + export_url)
    downloadFile(export_url, export_filename)

    disks, _ := instance.GetDisks()
    for _, disk := range disks {
        disk_uuid, _ := disk.GetUuid()

        // Basic auth in URL request is required as session token is not
        // accepted for some reason.
        // @todo: raise with XAPI team.
        disk_export_url := fmt.Sprintf("https://%s:%s@%s/export_raw_vdi?vdi=%s",
                                       client.Username,
                                       client.Password,
                                       client.Host,
                                       disk_uuid,
                                       )

        ui.Say("Getting " + disk_export_url)
        disk_export_filename := fmt.Sprintf("%s/%s.raw", config.OutputDir, disk_uuid)
        ui.Say("Downloading " + disk_uuid)
        downloadFile(disk_export_url, disk_export_filename)
    }


    ui.Say("Download complteded: " + config.OutputDir)

	return multistep.ActionContinue
}

func (stepShutdownAndExport) Cleanup(state multistep.StateBag) {
}