package parser import ( "bufio" "encoding/json" "fmt" "github.com/beevik/etree" "github.com/buger/jsonparser" "github.com/ghodss/yaml" "github.com/magiconair/properties" "github.com/pkg/errors" "github.com/pterodactyl/wings/config" "go.uber.org/zap" "gopkg.in/ini.v1" "io/ioutil" "os" "strings" ) // The file parsing options that are available for a server configuration file. const ( File = "file" Yaml = "yaml" Properties = "properties" Ini = "ini" Json = "json" Xml = "xml" ) type ConfigurationParser string // Defines a configuration file for the server startup. These will be looped over // and modified before the server finishes booting. type ConfigurationFile struct { FileName string `json:"file"` Parser ConfigurationParser `json:"parser"` Replace []ConfigurationFileReplacement `json:"replace"` // Tracks Wings' configuration so that we can quickly get values // out of it when variables request it. configuration []byte } // Defines a single find/replace instance for a given server configuration file. type ConfigurationFileReplacement struct { Match string `json:"match"` Value string `json:"value"` ValueType jsonparser.ValueType `json:"-"` } func (cfr *ConfigurationFileReplacement) UnmarshalJSON(data []byte) error { if m, err := jsonparser.GetString(data, "match"); err != nil { return err } else { cfr.Match = m } if v, dt, _, err := jsonparser.Get(data, "value"); err != nil { return err } else { if dt != jsonparser.String && dt != jsonparser.Number && dt != jsonparser.Boolean { return errors.New( fmt.Sprintf("cannot parse JSON: received unexpected replacement value type: %d", dt), ) } cfr.Value = string(v) cfr.ValueType = dt } return nil } // Parses a given configuration file and updates all of the values within as defined // in the API response from the Panel. func (f *ConfigurationFile) Parse(path string) error { zap.S().Debugw("parsing configuration file", zap.String("path", path), zap.String("parser", string(f.Parser))) mb, _ := json.Marshal(config.Get()) f.configuration = mb var err error switch f.Parser { case Properties: err = f.parsePropertiesFile(path) break case File: err = f.parseTextFile(path) break case Yaml, "yml": err = f.parseYamlFile(path) break case Json: err = f.parseJsonFile(path) break case Ini: err = f.parseIniFile(path) break case Xml: err = f.parseXmlFile(path) break } return err } // Parses an xml file. func (f *ConfigurationFile) parseXmlFile(path string) error { doc := etree.NewDocument() file, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0644) if err != nil { return errors.WithStack(err) } defer file.Close() if _, err := doc.ReadFrom(file); err != nil { return errors.WithStack(err) } // If there is no root we should create a basic start to the file. This isn't required though, // and if it doesn't work correctly I'll just remove the code. if doc.Root() == nil { doc.CreateProcInst("xml", `version="1.0" encoding="utf-8"`) } for i, replacement := range f.Replace { value, _, err := f.LookupConfigurationValue(replacement) if err != nil { return errors.WithStack(err) } // If this is the first item and there is no root element, create that root now and apply // it for future use. if i == 0 && doc.Root() == nil { parts := strings.SplitN(replacement.Match, ".", 2) doc.SetRoot(doc.CreateElement(parts[0])) } path := "./" + strings.Replace(replacement.Match, ".", "/", -1) // If we're not doing a wildcard replacement go ahead and create the // missing element if we cannot find it yet. if !strings.Contains(path, "*") { parts := strings.Split(replacement.Match, ".") // Set the initial element to be the root element, and then work from there. var element = doc.Root() // Iterate over the path to create the required structure for the given element's path. // This does not set a value, only ensures that the base structure exists. We start at index // 1 because an XML document can only contain a single root element, and from there we'll // work our way down the chain. for _, tag := range parts[1:] { if e := element.FindElement(tag); e == nil { element = element.CreateElement(tag) } else { element = e } } } // Iterate over the elements we found and update their values. for _, element := range doc.FindElements(path) { if xmlValueMatchRegex.Match(value) { k := xmlValueMatchRegex.ReplaceAllString(string(value), "$1") v := xmlValueMatchRegex.ReplaceAllString(string(value), "$2") element.CreateAttr(k, v) } else { element.SetText(string(value)) } } } // If you don't truncate the file you'll end up duplicating the data in there (or just appending // to the end of the file. We don't want to do that. if err := file.Truncate(0); err != nil { return errors.WithStack(err) } // Move the cursor to the start of the file to avoid weird spacing issues. file.Seek(0, 0) // Ensure the XML is indented properly. doc.Indent(2) // Write the XML to the file. _, err = doc.WriteTo(file) return errors.WithStack(err) } // Parses an ini file. func (f *ConfigurationFile) parseIniFile(path string) error { // Ini package can't handle a non-existent file, so handle that automatically here // by creating it if not exists. file, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0644); if err != nil { return errors.WithStack(err) } defer file.Close() cfg, err := ini.Load(path) if err != nil { return errors.WithStack(err) } for _, replacement := range f.Replace { path := strings.SplitN(replacement.Match, ".", 2) value, _, err := f.LookupConfigurationValue(replacement) if err != nil { return errors.WithStack(err) } k := path[0] s := cfg.Section("") // Passing a key of foo.bar will look for "bar" in the "[foo]" section of the file. if len(path) == 2 { k = path[1] s = cfg.Section(path[0]) } // If no section was found, create that new section now and then set the // section value we're using to be the new one. if s == nil { s, err = cfg.NewSection(path[0]) if err != nil { return errors.WithStack(err) } } // If the key exists in the file go ahead and set the value, otherwise try to // create it in the section. if s.HasKey(k) { s.Key(k).SetValue(string(value)) } else { if _, err := s.NewKey(k, string(value)); err != nil { return errors.WithStack(err) } } } if _, err := cfg.WriteTo(file); err != nil { return errors.WithStack(err) } return nil } // Parses a json file updating any matching key/value pairs. If a match is not found, the // value is set regardless in the file. See the commentary in parseYamlFile for more details // about what is happening during this process. func (f *ConfigurationFile) parseJsonFile(path string) error { b, err := readFileBytes(path) if err != nil { return errors.WithStack(err) } data, err := f.IterateOverJson(b) if err != nil { return errors.WithStack(err) } output := []byte(data.StringIndent("", " ")) return ioutil.WriteFile(path, output, 0644) } // Parses a yaml file and updates any matching key/value pairs before persisting // it back to the disk. func (f *ConfigurationFile) parseYamlFile(path string) error { b, err := readFileBytes(path) if err != nil { return errors.WithStack(err) } // Unmarshal the yaml data into a JSON interface such that we can work with // any arbitrary data structure. If we don't do this, I can't use gabs which // makes working with unknown JSON signficiantly easier. jsonBytes, err := yaml.YAMLToJSON(b) if err != nil { return errors.WithStack(err) } // Now that the data is converted, treat it just like JSON and pass it to the // iterator function to update values as necessary. data, err := f.IterateOverJson(jsonBytes) if err != nil { return errors.WithStack(err) } // Remarshal the JSON into YAML format before saving it back to the disk. marshaled, err := yaml.JSONToYAML(data.Bytes()) if err != nil { return errors.WithStack(err) } return ioutil.WriteFile(path, marshaled, 0644) } // Parses a text file using basic find and replace. This is a highly inefficient method of // scanning a file and performing a replacement. You should attempt to use anything other // than this function where possible. func (f *ConfigurationFile) parseTextFile(path string) error { file, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0644) if err != nil { return errors.WithStack(err) } defer file.Close() scanner := bufio.NewScanner(file) for scanner.Scan() { hasReplaced := false t := scanner.Text() // Iterate over the potential replacements for the line and check if there are // any matches. for _, replace := range f.Replace { if !strings.HasPrefix(t, replace.Match) { continue } hasReplaced = true t = strings.Replace(t, replace.Match, replace.Value, 1) } // If there was a replacement that occurred on this specific line, do a write to the file // immediately to write that modified content to the disk. if hasReplaced { if _, err := file.WriteAt([]byte(t), int64(len(scanner.Bytes()))); err != nil { return errors.WithStack(err) } } } if err := scanner.Err(); err != nil { return errors.WithStack(err) } return nil } // Parses a properties file and updates the values within it to match those that // are passed. Writes the file once completed. func (f *ConfigurationFile) parsePropertiesFile(path string) error { p, err := properties.LoadFile(path, properties.UTF8) if err != nil { return errors.WithStack(err) } for _, replace := range f.Replace { data, _, err := f.LookupConfigurationValue(replace) if err != nil { return errors.WithStack(err) } if _, _, err := p.Set(replace.Match, string(data)); err != nil { return errors.WithStack(err) } } w, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0644) if err != nil { return errors.WithStack(err) } _, err = p.Write(w, properties.UTF8) return err }