121 lines
3.0 KiB
Go
121 lines
3.0 KiB
Go
package yomichan
|
||
|
||
import (
|
||
"fmt"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
|
||
zig "foosoft.net/projects/zero-epwing-go"
|
||
)
|
||
|
||
type epwingExtractor interface {
|
||
extractTerms(entry zig.BookEntry, sequence int) []dbTerm
|
||
extractKanji(entry zig.BookEntry) []dbKanji
|
||
getFontNarrow() map[int]string
|
||
getFontWide() map[int]string
|
||
getRevision() string
|
||
}
|
||
|
||
func epwingExportDb(inputPath, outputPath, language, title string, stride int, pretty bool) error {
|
||
book, err := zig.Load(inputPath)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
translateExp := regexp.MustCompile(`{{([nw])_(\d+)}}`)
|
||
epwingExtractors := map[string]epwingExtractor{
|
||
"三省堂 スーパー大辞林": makeDaijirinExtractor(),
|
||
"大辞泉": makeDaijisenExtractor(),
|
||
"明鏡国語辞典": makeMeikyouExtractor(),
|
||
"故事ことわざの辞典": makeKotowazaExtractor(),
|
||
"研究社 新和英大辞典 第５版": makeWadaiExtractor(),
|
||
"広辞苑第六版": makeKoujienExtractor(),
|
||
"付属資料": makeKoujienExtractor(),
|
||
"学研国語大辞典": makeGakkenExtractor(),
|
||
"古語辞典": makeGakkenExtractor(),
|
||
"故事ことわざ辞典": makeGakkenExtractor(),
|
||
"学研漢和大字典": makeGakkenExtractor(),
|
||
"小学館２": makeShougakukan2Extractor(),
|
||
}
|
||
|
||
var (
|
||
terms dbTermList
|
||
kanji dbKanjiList
|
||
revisions []string
|
||
titles []string
|
||
sequence int
|
||
)
|
||
|
||
for _, subbook := range book.Subbooks {
|
||
if extractor, ok := epwingExtractors[subbook.Title]; ok {
|
||
fontNarrow := extractor.getFontNarrow()
|
||
fontWide := extractor.getFontWide()
|
||
|
||
translate := func(str string) string {
|
||
for _, matches := range translateExp.FindAllStringSubmatch(str, -1) {
|
||
var font map[int]string
|
||
if matches[1] == "n" {
|
||
font = fontNarrow
|
||
} else {
|
||
font = fontWide
|
||
}
|
||
|
||
code, _ := strconv.Atoi(matches[2])
|
||
replacement, ok := font[code]
|
||
if !ok {
|
||
replacement = "<22>"
|
||
}
|
||
|
||
str = strings.Replace(str, matches[0], replacement, -1)
|
||
}
|
||
pattern := regexp.MustCompile("\n+")
|
||
str = pattern.ReplaceAllLiteralString(str, "\n")
|
||
|
||
return str
|
||
}
|
||
|
||
for _, entry := range subbook.Entries {
|
||
entry.Heading = translate(entry.Heading)
|
||
entry.Text = translate(entry.Text)
|
||
|
||
terms = append(terms, extractor.extractTerms(entry, sequence)...)
|
||
kanji = append(kanji, extractor.extractKanji(entry)...)
|
||
|
||
sequence++
|
||
}
|
||
|
||
revisions = append(revisions, extractor.getRevision())
|
||
titles = append(titles, subbook.Title)
|
||
} else {
|
||
return fmt.Errorf("failed to find compatible extractor for '%s'", subbook.Title)
|
||
}
|
||
}
|
||
|
||
if title == "" {
|
||
title = strings.Join(titles, ", ")
|
||
}
|
||
|
||
recordData := map[string]dbRecordList{
|
||
"kanji": kanji.crush(),
|
||
"term": terms.crush(),
|
||
}
|
||
|
||
index := dbIndex{
|
||
Title: title,
|
||
Revision: strings.Join(revisions, ";"),
|
||
Sequenced: true,
|
||
Description: "",
|
||
Attribution: "",
|
||
}
|
||
index.setDefaults()
|
||
|
||
return writeDb(
|
||
outputPath,
|
||
index,
|
||
recordData,
|
||
stride,
|
||
pretty,
|
||
)
|
||
}
|