WIP

jmdict.go

@@ -52,32 +52,35 @@ func findString(needle string, haystack []string) int {
 	return -1
 }
 
-func extractDictEntries(edictEntry jmdict.EdictEntry) []dictEntry {
+func convertEdictEntry(edictEntry jmdict.EdictEntry) []dictEntry {
 	var entries []dictEntry
 
-	for _, kanji := range edictEntry.Kanji {
-		for _, reading := range edictEntry.Reading {
-			if findString(kanji.Expression, reading.Restrictions) != -1 {
-				continue
+	convert := func(reading jmdict.EdictReading, kanji *jmdict.EdictKanji) {
+		if kanji != nil && findString(kanji.Expression, reading.Restrictions) != -1 {
+			return
 		}
 
-			entry := dictEntry{
-				Expression: kanji.Expression,
-				Reading:    reading.Reading,
-				Tags:       make(map[string]bool),
+		entry := dictEntry{Tags: make(map[string]bool)}
+
+		if kanji == nil {
+			entry.Expression = reading.Reading
+		} else {
+			entry.Expression = kanji.Expression
+			entry.Reading = reading.Reading
+
+			entry.addTags(kanji.Information)
+			entry.addTags(kanji.Priority)
 		}
 
 		entry.addTags(reading.Information)
 		entry.addTags(reading.Priority)
-			entry.addTags(kanji.Information)
-			entry.addTags(kanji.Priority)
 
 		for _, sense := range edictEntry.Sense {
 			if findString(reading.Reading, sense.RestrictedReadings) != -1 {
 				continue
 			}
 
-				if findString(kanji.Expression, sense.RestrictedKanji) != -1 {
+			if kanji != nil && findString(kanji.Expression, sense.RestrictedKanji) != -1 {
 				continue
 			}
 
@@ -93,6 +96,17 @@ func extractDictEntries(edictEntry jmdict.EdictEntry) []dictEntry {
 
 		entries = append(entries, entry)
 	}
+
+	if len(edictEntry.Kanji) > 0 {
+		for _, kanji := range edictEntry.Kanji {
+			for _, reading := range edictEntry.Reading {
+				convert(reading, &kanji)
+			}
+		}
+	} else {
+		for _, reading := range edictEntry.Reading {
+			convert(reading, nil)
+		}
 	}
 
 	return entries
@@ -106,7 +120,7 @@ func processEdict(reader io.Reader, writer io.Writer) error {
 
 	var entries []dictEntry
 	for _, edictEntry := range edictEntries {
-		entries = append(entries, extractDictEntries(edictEntry)...)
+		entries = append(entries, convertEdictEntry(edictEntry)...)
 	}
 
 	log.Print(entries[42])