121 lines
2.2 KiB
Go
121 lines
2.2 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"encoding/csv"
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
)
|
||
|
|
||
|
type (
|
||
|
StoryEntry struct {
|
||
|
Author string `json:"author"`
|
||
|
Content string `json:"content"`
|
||
|
ModifiedDate string `json:"modifiedDate"`
|
||
|
StarredCount int `json:"starredCount"`
|
||
|
ReportedCount int `json:"reportedCount"`
|
||
|
}
|
||
|
|
||
|
KanjiEntry struct {
|
||
|
Character string `json:"character"`
|
||
|
Reading string `json:"reading"`
|
||
|
FrameNumber int `json:"frameNumber"`
|
||
|
StrokeCount int `json:"strokeCount"`
|
||
|
Story string `json:"story"`
|
||
|
Stories []StoryEntry `json:"stories"`
|
||
|
}
|
||
|
|
||
|
TableRow = []string
|
||
|
Table = []TableRow
|
||
|
)
|
||
|
|
||
|
func injectStories(table Table, path string) error {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func injectKanjidic(table Table, path string) error {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func loadTable(path string) (Table, error) {
|
||
|
fp, err := os.Open(path)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
defer fp.Close()
|
||
|
|
||
|
reader := csv.NewReader(fp)
|
||
|
reader.Comment = '#'
|
||
|
reader.Comma = '\t'
|
||
|
|
||
|
table, err := reader.ReadAll()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
for i := range table {
|
||
|
table[i] = table[i][:3]
|
||
|
}
|
||
|
|
||
|
return table, nil
|
||
|
}
|
||
|
|
||
|
func saveTable(path string, table Table) error {
|
||
|
fp, err := os.Create(path)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer fp.Close()
|
||
|
|
||
|
writer := csv.NewWriter(fp)
|
||
|
if err := writer.WriteAll(table); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
var (
|
||
|
storiesPath = flag.String("stories", "", "path for stories JSON")
|
||
|
kanjidicPath = flag.String("kanjidic", "", "path for KANJIDIC")
|
||
|
)
|
||
|
|
||
|
flag.Usage = func() {
|
||
|
fmt.Fprintf(os.Stderr, "Usage: %s [options] <notes_in.txt> <notes_out.txt>\n", filepath.Base(os.Args[0]))
|
||
|
fmt.Fprintln(os.Stderr, "Options:")
|
||
|
flag.PrintDefaults()
|
||
|
}
|
||
|
|
||
|
flag.Parse()
|
||
|
|
||
|
args := flag.Args()
|
||
|
if len(args) != 2 {
|
||
|
flag.Usage()
|
||
|
os.Exit(2)
|
||
|
}
|
||
|
|
||
|
table, err := loadTable(os.Args[1])
|
||
|
if err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if len(*storiesPath) > 0 {
|
||
|
if err := injectStories(table, *storiesPath); err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if len(*kanjidicPath) > 0 {
|
||
|
if err := injectKanjidic(table, *kanjidicPath); err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if err := saveTable(os.Args[2], table); err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
}
|