builder/vmware: Start HTTP server to serve files

This commit is contained in:
Mitchell Hashimoto 2013-06-05 14:24:48 -07:00
parent 875951a817
commit eb8278b21c
2 changed files with 59 additions and 0 deletions

View File

@ -18,6 +18,7 @@ type config struct {
ISOUrl string `mapstructure:"iso_url"`
VMName string `mapstructure:"vm_name"`
OutputDir string `mapstructure:"output_directory"`
HTTPDir string `mapstructure:"http_directory"`
}
func (b *Builder) Prepare(raw interface{}) (err error) {
@ -46,6 +47,7 @@ func (b *Builder) Run(ui packer.Ui, hook packer.Hook) packer.Artifact {
&stepPrepareOutputDir{},
&stepCreateDisk{},
&stepCreateVMX{},
&stepHTTPServer{},
}
// Setup the state bag

View File

@ -0,0 +1,57 @@
package vmware
import (
"fmt"
"github.com/mitchellh/multistep"
"github.com/mitchellh/packer/packer"
"net"
"net/http"
)
// This step creates and runs the HTTP server that is serving the files
// specified by the 'http_files` configuration parameter in the template.
//
// Uses:
// config *config
// ui packer.Ui
//
// Produces:
// http_port int - The port the HTTP server started on.
type stepHTTPServer struct{
l net.Listener
}
func (s *stepHTTPServer) Run(state map[string]interface{}) multistep.StepAction {
config := state["config"].(*config)
ui := state["ui"].(packer.Ui)
httpPort := 8080
httpAddr := fmt.Sprintf(":%d", httpPort)
ui.Say(fmt.Sprintf("Starting HTTP server on port %d", httpPort))
// Start the TCP listener
var err error
s.l, err = net.Listen("tcp", httpAddr)
if err != nil {
ui.Error(fmt.Sprintf("Error starting HTTP server: %s", err))
return multistep.ActionHalt
}
// Start the HTTP server and run it in the background
fileServer := http.FileServer(http.Dir(config.HTTPDir))
server := &http.Server{Addr: httpAddr, Handler: fileServer}
go server.Serve(s.l)
// Save the address into the state so it can be accessed in the future
state["http_port"] = httpPort
return multistep.ActionContinue
}
func (s *stepHTTPServer) Cleanup(map[string]interface{}) {
if s.l != nil {
// Close the listener so that the HTTP server stops
s.l.Close()
}
}