fix: yaml parser error when host config not exist

Author: soulteary

internal/parser/yaml.go

@@ -86,23 +86,25 @@ func GroupYAMLConfig(input string) []Define.HostConfig {
 				hostConfig := originConfig
 				hostConfig.Name = hostName
 				hostConfig.Extra.Prefix = prefix
+				if hostConfig.Config != nil {
+					if groupConfig.Common != nil {
+						for key, value := range groupConfig.Common {
 
-				if groupConfig.Common != nil {
-					for key, value := range groupConfig.Common {
-						if _, ok := hostConfig.Config[key]; !ok {
-							hostConfig.Config[key] = value
+							if _, ok := hostConfig.Config[key]; !ok {
+								hostConfig.Config[key] = value
+							}
 						}
 					}
-				}
 
-				if yamlConfig.Default != nil {
-					for key, value := range yamlConfig.Default {
-						if _, ok := hostConfig.Config[key]; !ok {
-							hostConfig.Config[key] = value
+					if yamlConfig.Default != nil {
+						for key, value := range yamlConfig.Default {
+							if _, ok := hostConfig.Config[key]; !ok {
+								hostConfig.Config[key] = value
+							}
 						}
 					}
+					hostConfigs = append(hostConfigs, hostConfig)
 				}
-				hostConfigs = append(hostConfigs, hostConfig)
 			}
 		}
 	}

internal/parser/yaml_test.go

@@ -103,6 +103,28 @@ func TestConvertToYAML(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "Both global and normal config",
+			input: []Define.HostConfig{
+				{
+					Name:   "host1",
+					Notes:  "note1",
+					Config: nil,
+				},
+			},
+			expected: Define.YAMLOutput{
+				Groups: map[string]Define.GroupConfig{
+					"Group host1": {
+						Common: make(map[string]string),
+						Hosts: map[string]Define.HostConfig{
+							"host1": {
+								Notes: "note1",
+							},
+						},
+					},
+				},
+			},
+		},
 	}
 
 	for _, tt := range tests {