Fix genders bug and add number

1 files changed, 19 insertions, 4 deletions

diff --git a/setup.go b/setup.go
index 7921f47..23c8bbe 100644
--- a/setup.go
+++ b/setup.go
@@ -141,13 +141,28 @@ func populateDictionary(rawDictionary string, db *sql.DB) error {
 		if len(result.Sounds) > 0 {
 			entry.Sound = result.Sounds[0].IPA
 		}
+
+		var genders, numbers []string
 		for _, r := range result.Tags {
-			var genders []string
-			if r == "masculine" || r == "feminine" {
-				genders = append(genders, r)
+			switch r {
+			case "masculine":
+				genders = append(genders, "masculin")
+			case "feminine":
+				genders = append(genders, "féminin")
+			case "plural":
+				numbers = append(numbers, "pluriel")
+			case "singular":
+				numbers = append(numbers, "singulier")
 			}
-			entry.Gender = strings.Join(genders, " / ")
 		}
+		entry.Gender = strings.Join(
+			[]string{
+				strings.Join(genders, " / "),
+				strings.Join(numbers, " et "),
+			},
+			" ",
+		)
+
 		for _, s := range result.Senses {
 			var example string
 			if len(s.Examples) > 0 {