132 lines
3.8 KiB
Go
132 lines
3.8 KiB
Go
/*
|
||
* Copyright (c) 2016-2021 Alex Yatskov <alex@foosoft.net>
|
||
*
|
||
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||
* this software and associated documentation files (the "Software"), to deal in
|
||
* the Software without restriction, including without limitation the rights to
|
||
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
||
* the Software, and to permit persons to whom the Software is furnished to do so,
|
||
* subject to the following conditions:
|
||
*
|
||
* The above copyright notice and this permission notice shall be included in all
|
||
* copies or substantial portions of the Software.
|
||
*
|
||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
||
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
||
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
||
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
||
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
*/
|
||
|
||
package yomichan
|
||
|
||
import (
|
||
"fmt"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
|
||
zig "github.com/FooSoft/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(),
|
||
}
|
||
|
||
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)
|
||
}
|
||
|
||
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(),
|
||
}
|
||
|
||
return writeDb(
|
||
outputPath,
|
||
title,
|
||
strings.Join(revisions, ";"),
|
||
true,
|
||
recordData,
|
||
stride,
|
||
pretty,
|
||
)
|
||
}
|