diff --git a/pki/pki.go b/pki/pki.go index 58f256b1..ea2da61e 100644 --- a/pki/pki.go +++ b/pki/pki.go @@ -950,7 +950,11 @@ func (p *PKI) Save(opt ...ConfigOption) error { return errs.FileError(err, p.defaults) } if p.profileDefaults != "" { - if err = fileutil.WriteFile(p.profileDefaults, []byte("{}"), 0644); err != nil { + if _, err := os.Stat(p.profileDefaults); os.IsNotExist(err) { + if err = fileutil.WriteFile(p.profileDefaults, []byte("{}"), 0644); err != nil { + return errs.FileError(err, p.profileDefaults) + } + } else if err != nil { return errs.FileError(err, p.profileDefaults) } } diff --git a/templates/templates.go b/templates/templates.go index e760f0a4..dee7aa5f 100644 --- a/templates/templates.go +++ b/templates/templates.go @@ -20,8 +20,9 @@ type TemplateType string const ( // Snippet will mark a template as a part of a file. Snippet TemplateType = "snippet" - // Line is a template for a single line in a file. - Line TemplateType = "line" + // PrependLine is a template for prepending a single line to a file. If the + // line already exists in the file it will be removed first. + PrependLine TemplateType = "prepend-line" // File will mark a templates as a full file. File TemplateType = "file" // Directory will mark a template as a directory. @@ -120,8 +121,8 @@ func (t *Template) Validate() error { return nil case t.Name == "": return errors.New("template name cannot be empty") - case t.Type != Snippet && t.Type != File && t.Type != Directory && t.Type != Line: - return errors.Errorf("invalid template type %s, it must be %s, %s, %s, or %s", t.Type, Snippet, Line, File, Directory) + case t.Type != Snippet && t.Type != File && t.Type != Directory && t.Type != PrependLine: + return errors.Errorf("invalid template type %s, it must be %s, %s, %s, or %s", t.Type, Snippet, PrependLine, File, Directory) case t.TemplatePath == "" && t.Type != Directory && len(t.Content) == 0: return errors.New("template template cannot be empty") case t.TemplatePath != "" && t.Type == Directory: @@ -267,7 +268,7 @@ func (o *Output) Write() error { return fileutil.WriteFile(path, o.Content, 0600) case Snippet: return fileutil.WriteSnippet(path, o.Content, 0600) - case Line: + case PrependLine: return fileutil.PrependLine(path, o.Content, 0600) default: return errors.Errorf("unexpected output template type %s", string(o.Type)) diff --git a/templates/values.go b/templates/values.go index c59843af..c5e3f291 100644 --- a/templates/values.go +++ b/templates/values.go @@ -22,23 +22,23 @@ type StepSSH struct { var DefaultSSHTemplates = SSHTemplates{ User: []Template{ { - Name: "base_config.tpl", + Name: "config.tpl", Type: Snippet, - TemplatePath: "templates/ssh/base_config.tpl", + TemplatePath: "templates/ssh/config.tpl", Path: "~/.ssh/config", Comment: "#", }, { - Name: "includes.tpl", - Type: Line, - TemplatePath: "templates/ssh/includes.tpl", + Name: "step_includes.tpl", + Type: PrependLine, + TemplatePath: "templates/ssh/step_includes.tpl", Path: "${STEPPATH}/ssh/includes", Comment: "#", }, { - Name: "config.tpl", + Name: "step_config.tpl", Type: File, - TemplatePath: "templates/ssh/config.tpl", + TemplatePath: "templates/ssh/step_config.tpl", Path: "ssh/config", Comment: "#", }, @@ -76,16 +76,16 @@ var DefaultSSHTemplateData = map[string]string{ // Note: on windows `Include C:\...` is treated as a relative path. "base_config.tpl": `Host * {{- if or .User.GOOS "none" | eq "windows" }} -{{- if .User.Authority }} +{{- if .User.StepBasePath }} Include "{{ .User.StepBasePath | replace "\\" "/" | trimPrefix "C:" }}/ssh/includes" {{- else }} - Include "{{ .User.StepPath | replace "\\" "/" | trimPrefix "C:" }}/ssh/config" + Include "{{ .User.StepPath | replace "\\" "/" | trimPrefix "C:" }}/ssh/includes" {{- end }} {{- else }} -{{- if .User.Authority }} +{{- if .User.StepBasePath }} Include "{{.User.StepBasePath}}/ssh/includes" {{- else }} - Include "{{.User.StepPath}}/ssh/config" + Include "{{.User.StepPath}}/ssh/includes" {{- end }} {{- end }}`,