mirror of https://github.com/hashicorp/packer
sangkyu.kim
5 years ago
parent
af865b1591
commit
cd370aaaad
@ -0,0 +1,114 @@
|
||||
package ncloud
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/NaverCloudPlatform/ncloud-sdk-go-v2/ncloud"
|
||||
"io/ioutil"
|
||||
"time"
|
||||
|
||||
"github.com/NaverCloudPlatform/ncloud-sdk-go-v2/services/vserver"
|
||||
"github.com/hashicorp/packer-plugin-sdk/multistep"
|
||||
packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
|
||||
)
|
||||
|
||||
type InitScript struct {
|
||||
KeyName string
|
||||
PrivateKey string
|
||||
}
|
||||
|
||||
type StepCreateInitScript struct {
|
||||
Conn *NcloudAPIClient
|
||||
CreateInitScript func() (string, error)
|
||||
DeleteInitScript func(initScriptNo string) error
|
||||
Say func(message string)
|
||||
Error func(e error)
|
||||
Config *Config
|
||||
}
|
||||
|
||||
func NewStepCreateInitScript(conn *NcloudAPIClient, ui packersdk.Ui, config *Config) *StepCreateInitScript {
|
||||
var step = &StepCreateInitScript{
|
||||
Conn: conn,
|
||||
Say: func(message string) { ui.Say(message) },
|
||||
Error: func(e error) { ui.Error(e.Error()) },
|
||||
Config: config,
|
||||
}
|
||||
|
||||
if config.SupportVPC {
|
||||
step.CreateInitScript = step.createVpcInitScript
|
||||
step.DeleteInitScript = step.deleteVpcInitScript
|
||||
}
|
||||
|
||||
return step
|
||||
}
|
||||
|
||||
func (s *StepCreateInitScript) createVpcInitScript() (string, error) {
|
||||
name := fmt.Sprintf("packer-%d", time.Now().Unix())
|
||||
reqParams := &vserver.CreateInitScriptRequest{
|
||||
RegionCode: &s.Config.RegionCode,
|
||||
InitScriptName: &name,
|
||||
}
|
||||
|
||||
if s.Config.Comm.Type == "winrm" {
|
||||
reqParams.OsTypeCode = ncloud.String("WND")
|
||||
}
|
||||
|
||||
if s.Config.UserData != "" {
|
||||
reqParams.InitScriptContent = &s.Config.UserData
|
||||
}
|
||||
if s.Config.UserDataFile != "" {
|
||||
contents, err := ioutil.ReadFile(s.Config.UserDataFile)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("Problem reading user data file: %s", err)
|
||||
}
|
||||
|
||||
reqParams.InitScriptContent = ncloud.String(string(contents))
|
||||
}
|
||||
|
||||
if reqParams.InitScriptContent == nil {
|
||||
return "", nil
|
||||
}
|
||||
|
||||
resp, err := s.Conn.vserver.V2Api.CreateInitScript(reqParams)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return *resp.InitScriptList[0].InitScriptNo, nil
|
||||
}
|
||||
|
||||
func (s *StepCreateInitScript) deleteVpcInitScript(initScriptNo string) error {
|
||||
reqParams := &vserver.DeleteInitScriptsRequest{
|
||||
RegionCode: &s.Config.RegionCode,
|
||||
InitScriptNoList: []*string{&initScriptNo},
|
||||
}
|
||||
_, err := s.Conn.vserver.V2Api.DeleteInitScripts(reqParams)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StepCreateInitScript) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
||||
s.Say("Create Init script")
|
||||
|
||||
initScriptNo, err := s.CreateInitScript()
|
||||
if err == nil && initScriptNo != "" {
|
||||
state.Put("init_script_no", initScriptNo)
|
||||
s.Say(fmt.Sprintf("Init script[%s] is created", initScriptNo))
|
||||
}
|
||||
|
||||
return processStepResult(err, s.Error, state)
|
||||
}
|
||||
|
||||
func (s *StepCreateInitScript) Cleanup(state multistep.StateBag) {
|
||||
if initScriptNo, ok := state.GetOk("init_script_no"); ok {
|
||||
s.Say("Cleanup Init script")
|
||||
if err := s.DeleteInitScript(initScriptNo.(string)); err != nil {
|
||||
s.Error(err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,63 @@
|
||||
package ncloud
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/hashicorp/packer-plugin-sdk/multistep"
|
||||
)
|
||||
|
||||
func TestStepCreateInitScriptShouldFailIfOperationCreateInitScriptFails(t *testing.T) {
|
||||
var testSubject = &StepCreateInitScript{
|
||||
CreateInitScript: func() (string, error) { return "", fmt.Errorf("!! Unit Test FAIL !!") },
|
||||
Say: func(message string) {},
|
||||
Error: func(e error) {},
|
||||
Config: &Config{
|
||||
Region: "KR",
|
||||
SupportVPC: true,
|
||||
},
|
||||
}
|
||||
|
||||
stateBag := createTestStateBagStepCreateInitScript()
|
||||
|
||||
var result = testSubject.Run(context.Background(), stateBag)
|
||||
|
||||
if result != multistep.ActionHalt {
|
||||
t.Fatalf("Expected the step to return 'ActionHalt', but got '%d'.", result)
|
||||
}
|
||||
|
||||
if _, ok := stateBag.GetOk("error"); ok == false {
|
||||
t.Fatal("Expected the step to set stateBag['Error'], but it was not.")
|
||||
}
|
||||
}
|
||||
|
||||
func TestStepCreateInitScriptShouldPassIfOperationCreateInitScriptPasses(t *testing.T) {
|
||||
var testSubject = &StepCreateInitScript{
|
||||
CreateInitScript: func() (string, error) { return "123", nil },
|
||||
Say: func(message string) {},
|
||||
Error: func(e error) {},
|
||||
Config: &Config{
|
||||
Region: "KR",
|
||||
SupportVPC: true,
|
||||
},
|
||||
}
|
||||
|
||||
stateBag := createTestStateBagStepCreateInitScript()
|
||||
|
||||
var result = testSubject.Run(context.Background(), stateBag)
|
||||
|
||||
if result != multistep.ActionContinue {
|
||||
t.Fatalf("Expected the step to return 'ActionContinue', but got '%d'.", result)
|
||||
}
|
||||
|
||||
if _, ok := stateBag.GetOk("error"); ok == true {
|
||||
t.Fatalf("Expected the step to not set stateBag['Error'], but it was.")
|
||||
}
|
||||
}
|
||||
|
||||
func createTestStateBagStepCreateInitScript() multistep.StateBag {
|
||||
stateBag := new(multistep.BasicStateBag)
|
||||
|
||||
return stateBag
|
||||
}
|
||||
Loading…
Reference in new issue