Merge branch 'dev_alt'

parser.go

@@ -9,10 +9,10 @@ import (
 )
 
 // ParseFunction -- type
-type ParseFunction func(string, string) (map[string]string, error)
+type ParseFunction func(string, string) (map[int]Pair, error)
 
 // Default parse method for parsing key - value file
-func defaultParse(path, fileName string) (m map[string]string, err error) {
+func defaultParse(path, fileName string) (m map[int]Pair, err error) {
 	absolutePathFile, err := filepath.Abs(filepath.Join(path, fileName))
 	if err != nil {
 		return nil, err
@@ -23,8 +23,9 @@ func defaultParse(path, fileName string) (m map[string]string, err error) {
 	}
 
 	defer file.Close()
-	m = make(map[string]string)
+	m = make(map[int]Pair)
 	reader := bufio.NewReader(file)
+	index := 0
 	for {
 		line, err := reader.ReadString('\n')
 
@@ -37,7 +38,7 @@ func defaultParse(path, fileName string) (m map[string]string, err error) {
 					value = strings.TrimSpace(strings.Replace(line[equal+1:], "\"", "", -1))
 				}
 				// assign the values map
-				m[key] = value
+				m[index] = Pair{key, value}
 			}
 		}
 		if err == io.EOF {
@@ -46,6 +47,7 @@ func defaultParse(path, fileName string) (m map[string]string, err error) {
 		if err != nil {
 			return nil, err
 		}
+		index++
 	}
 	return m, nil
 }

properties.go

@@ -7,11 +7,16 @@ import (
 	"strings"
 )
 
+type Pair struct {
+	First  string
+	Second string
+}
+
 // Properties -- type
 type Properties struct {
 	fileName string
 	path     string
-	values   map[string]string
+	values   map[int]Pair
 	length   int
 }
 
@@ -20,7 +25,7 @@ func New(path, fileName string) Properties {
 	return Properties{
 		fileName: fileName,
 		path:     filepath.Clean(path),
-		values:   make(map[string]string),
+		values:   make(map[int]Pair),
 		length:   0,
 	}
 }
@@ -51,7 +56,7 @@ func (p Properties) Length() int {
 }
 
 // Values -- Getter values of property file
-func (p Properties) Values() map[string]string {
+func (p Properties) Values() map[int]Pair {
 	return p.values
 }
 
@@ -61,7 +66,7 @@ func (p *Properties) Put(key, value string) error {
 		return e.New("Key value is nil")
 	}
 	if p.values != nil {
-		p.values[key] = value
+		p.values[p.length+1] = Pair{key, value}
 		p.length++
 	} else {
 		return e.New("Property values is nil")
@@ -74,10 +79,11 @@ func (p Properties) Get(key string) (string, error) {
 	if key == "" || len(strings.TrimSpace(key)) == 0 {
 		return "", e.New("Key value is nil")
 	}
-	if _, ok := p.values[key]; !ok {
+	index := p.index(key)
+	if _, ok := p.values[index]; !ok {
 		return "", e.New("Key not found")
 	}
-	return p.values[key], nil
+	return p.values[index].Second, nil
 }
 
 // Remove -- Remove property with key
@@ -86,7 +92,8 @@ func (p *Properties) Remove(key string) (string, error) {
 		return "", e.New("Key value is nil")
 	}
 	lenghtBefore := len(p.values)
-	delete(p.values, key)
+	index := p.index(key)
+	delete(p.values, index)
 	if len(p.values) != lenghtBefore {
 		p.length--
 	}
@@ -95,12 +102,24 @@ func (p *Properties) Remove(key string) (string, error) {
 
 // GetProperties -- Get all key value in Properties object
 func (p Properties) GetProperties() (keys []string) {
-	for key := range p.values {
+	for i := range p.values {
+		key := p.values[i].First
 		keys = append(keys, key)
 	}
 	return keys
 }
 
+func (p Properties) index(key string) int {
+	index := -1
+	for i, item := range p.values {
+		if item.First == key {
+			index = i
+			break
+		}
+	}
+	return index
+}
+
 // DefaultLoad -- Load file in Properties object using default parse function
 func (p *Properties) DefaultLoad() (int, error) {
 	return p.Load(defaultParse)

store.go

@@ -1,54 +1,86 @@
 package properties
 
 import (
-	"errors"
+	"bufio"
+	"bytes"
+	"fmt"
+	"log"
 	"os"
 	"path/filepath"
-	"strings"
+	"unicode"
 )
 
 // StoringFunction -- type
 type StoringFunction func(Properties) (*os.File, error)
 
-// FIXME when you remove a property it not work fine
 func defaultStore(p Properties) (*os.File, error) {
 	absolutePathFile, err := filepath.Abs(filepath.Join(p.Path(), p.FileName()))
 	if err != nil {
 		return nil, err
 	}
-	file, err := os.OpenFile(absolutePathFile, os.O_CREATE, 0644)
+	file, err := os.OpenFile(absolutePathFile, os.O_CREATE|os.O_RDWR, 0644)
 	if err != nil {
 		return nil, err
 	}
 
 	// Write this file
 	defer file.Close()
-	for _, key := range p.GetProperties() {
-		value, err := p.Get(key)
-		if err != nil {
-			return nil, err
-		}
-
-		row, err := buildRow(key, value)
-		if err != nil {
-			return nil, err
-		}
-
-		if _, err := file.Write([]byte(row + "\n")); err != nil {
+	writer := bufio.NewWriter(file)
+	for _, pair := range p.values {
+		line := fmt.Sprintf("%s=%s\n", escape(pair.First, true), escape(pair.Second, false))
+		log.Print(line)
+		if _, err := writer.Write([]byte(line)); err != nil {
 			return nil, err
 		}
 	}
 
+	if err := writer.Flush(); err != nil {
+		return nil, err
+	}
 	return file, nil
 }
 
-func buildRow(key, value string) (row string, err error) {
-	if key == "" || value == "" {
-		return "", errors.New("Key or Value param is a empty string")
+// escape returns a string that is safe to use as either a key or value in a
+// property file. Whitespace characters, key separators, and comment markers
+// should always be escaped.
+func escape(s string, key bool) string {
+
+	leading := true
+	var buf bytes.Buffer
+	for _, ch := range s {
+		wasSpace := false
+		if ch == '\t' {
+			buf.WriteString(`\t`)
+		} else if ch == '\n' {
+			buf.WriteString(`\n`)
+		} else if ch == '\r' {
+			buf.WriteString(`\r`)
+		} else if ch == '\f' {
+			buf.WriteString(`\f`)
+		} else if ch == ' ' {
+			if key || leading {
+				buf.WriteString(`\ `)
+				wasSpace = true
+			} else {
+				buf.WriteRune(ch)
+			}
+		} else if ch == ':' {
+			buf.WriteString(`\:`)
+		} else if ch == '=' {
+			buf.WriteString(`\=`)
+		} else if ch == '#' {
+			buf.WriteString(`\#`)
+		} else if ch == '!' {
+			buf.WriteString(`\!`)
+		} else if !unicode.IsPrint(ch) || ch > 126 {
+			buf.WriteString(fmt.Sprintf(`\u%04x`, ch))
+		} else {
+			buf.WriteRune(ch)
 		}
 
-	newKey := strings.Replace(key, "\"", "", -1)
-	newValue := strings.Join([]string{"\"", value, "\""}, "")
-	row = strings.Join([]string{newKey, newValue}, "=")
-	return strings.TrimSpace(row), nil
+		if !wasSpace {
+			leading = false
+		}
+	}
+	return buf.String()
 }