57
|
1 /*
|
|
2 fwgo.go: Futan-Wariai de Go
|
|
3
|
|
4 Last Change: 2020-07-14 火 15:04:09.
|
|
5 */
|
|
6 package main
|
|
7
|
|
8 /*
|
|
9 #cgo LDFLAGS: -L. -lxdwapi -static
|
|
10 ##### C_SOURCE #####
|
|
11 */
|
|
12 import "C"
|
|
13
|
|
14 import (
|
|
15 "encoding/csv"
|
|
16 "flag"
|
|
17 "fmt"
|
|
18 "io"
|
|
19 "io/ioutil"
|
|
20 "log"
|
|
21 "os"
|
|
22 "regexp"
|
|
23 "strings"
|
|
24 "time"
|
|
25
|
|
26 "golang.org/x/text/encoding/japanese"
|
|
27 "golang.org/x/text/transform"
|
|
28 )
|
|
29
|
|
30 type Data struct {
|
|
31 Hno string
|
|
32 Page string
|
|
33 Zip string
|
|
34 SendTo string
|
|
35 }
|
|
36
|
|
37 func (d *Data) ToCsv() string {
|
|
38 s := []string{d.Hno, d.Page, d.Zip, d.SendTo}
|
|
39 return strings.Join(s, ",")
|
|
40 }
|
|
41
|
|
42 var (
|
|
43 ver = "0.1"
|
|
44
|
|
45 clean bool
|
|
46 skip bool
|
|
47 debug bool
|
|
48 xdwfile string
|
|
49
|
|
50 txtfile = "fwtmp_txt.txt"
|
|
51 infofile = "fwtmp_info.csv"
|
|
52 orderfile = "order.csv" // input
|
|
53 pagefile1 = "fwtmp_page1.txt" // send to sisetsu
|
|
54 pagefile2 = "fwtmp_page2.txt" // send to home
|
|
55 output1 = "output1.xdw"
|
|
56 output2 = "output2.xdw"
|
|
57 )
|
|
58
|
|
59 func init() {
|
|
60 /* INITIALIZE FLAGS */
|
|
61 flag.BoolVar(&clean, "c", false, "clean temporary files & exit")
|
|
62 flag.BoolVar(&skip, "e", false, "use existed files")
|
|
63 flag.BoolVar(&debug, "d", false, "debug mode")
|
|
64 flag.StringVar(&xdwfile, "i", "KBPV016G.xdw", "target xdw file")
|
|
65 }
|
|
66
|
|
67 func main() {
|
|
68 flag.Parse()
|
|
69
|
|
70 /* PRINT HEADER */
|
|
71 fmt.Println("=================================================")
|
|
72 fmt.Println(" 負担割合証を... ")
|
|
73 fmt.Printf(" - fwgo [ver %s] -\n", ver)
|
|
74 fmt.Println("=================================================\n")
|
|
75 print_time("now")
|
|
76 fmt.Println("[0] start ...")
|
|
77
|
|
78 /* CLEAN TEMPORARY DIRECTORY */
|
|
79 os.Remove(output1)
|
|
80 os.Remove(output2)
|
|
81 if !skip {
|
|
82 clean_full()
|
|
83 }
|
|
84 if clean {
|
|
85 os.Exit(0)
|
|
86 }
|
|
87 print_time("check done")
|
|
88
|
|
89 fmt.Println("[1] extract ...")
|
|
90 if !skip {
|
|
91 C.xdw2txt(C.CString(xdwfile), C.CString(txtfile))
|
|
92 }
|
|
93 print_time("extract done.")
|
|
94
|
|
95 c := make(chan int)
|
|
96 fmt.Println("[2] split ... (run background)")
|
|
97 go func() {
|
|
98 if !skip {
|
|
99 C.xdwsplit1(C.CString(xdwfile))
|
|
100 }
|
|
101 print_time("split done.")
|
|
102 c <- 1
|
|
103 }()
|
|
104
|
|
105 fmt.Println("[3] analize ...")
|
|
106 data_hash, err := analize(txtfile, infofile)
|
|
107 if err != nil {
|
|
108 log.Fatal(err)
|
|
109 }
|
|
110 print_time("analize done.")
|
|
111 debug_print(debug, fmt.Sprintf("len = %d", len(data_hash)))
|
|
112
|
|
113 fmt.Println("[4] read order ...")
|
|
114 h_order, n, err := read_order(orderfile)
|
|
115 if err != nil {
|
|
116 log.Fatal(err)
|
|
117 }
|
|
118 print_time("read order done.")
|
|
119 debug_print(debug, fmt.Sprintf("len = %d / %d", len(h_order), n))
|
|
120
|
|
121 fmt.Println("[5] make_list ...")
|
|
122 n, m, l, err := make_list(h_order, data_hash, pagefile1, pagefile2)
|
|
123 if err != nil {
|
|
124 log.Fatal(err)
|
|
125 }
|
|
126 print_time("make list done.")
|
|
127 debug_print(debug, fmt.Sprintf("order, kunai, kugai = %d, %d, %d", n, m, l))
|
|
128
|
|
129 <-c
|
|
130
|
|
131 fmt.Println("[6] merge ...")
|
|
132 clean_mini()
|
|
133 C.xdwmerge(C.CString(pagefile1), C.CString(output1))
|
|
134 print_time("merge 1/2 done.")
|
|
135 clean_mini()
|
|
136 C.xdwmerge(C.CString(pagefile2), C.CString(output2))
|
|
137 print_time("merge 2/2 done.")
|
|
138
|
|
139 fmt.Println("[7] page ...")
|
|
140 C.xdwaddpage(C.CString(output1), C.int(1), C.int(0))
|
|
141 print_time("page 1/2 done.")
|
|
142 C.xdwaddpage(C.CString(output2), C.int(n + 1), C.int(1))
|
|
143 print_time("page 2/2 done.")
|
|
144
|
|
145 if !debug {
|
|
146 fmt.Println("[9] clean ...")
|
|
147 clean_full()
|
|
148 print_time("clean done.")
|
|
149 }
|
|
150 }
|
|
151
|
|
152 func analize(txtfile, infofile string) (map[string]Data, error) {
|
|
153 hash := make(map[string]Data)
|
|
154
|
|
155 c, err := ioutil.ReadFile(txtfile)
|
|
156 if err != nil {
|
|
157 return hash, err
|
|
158 }
|
|
159 r := strings.NewReader(string(c))
|
|
160 tr := transform.NewReader(r, japanese.ShiftJIS.NewDecoder())
|
|
161 b, err := ioutil.ReadAll(tr)
|
|
162 if err != nil {
|
|
163 return hash, err
|
|
164 }
|
|
165
|
|
166 regHid := regexp.MustCompile(`05[0-9]{4}0[1238][0-9]{8}`)
|
|
167 regZip := regexp.MustCompile(`[0123456789]{3}-[0123456789]{4}`)
|
|
168
|
|
169 var csv string
|
|
170
|
|
171 buf := strings.Split(string(b), "生年月日年月日")
|
|
172 for p, v := range buf[:len(buf)-1] {
|
|
173 hid := regHid.FindString(v)
|
|
174 hno := hid[6:16]
|
|
175
|
|
176 zips := regZip.FindAllString(v, -1)
|
|
177 if len(zips) != 2 {
|
|
178 fmt.Println("warnig:", hno, "have not 2 zip-pattern")
|
|
179 }
|
|
180
|
|
181 s := strings.Split(v, zips[1])
|
|
182 s = strings.Split(s[len(s)-1], " 様 ")
|
|
183 send := strings.ReplaceAll(s[0], " ", "")
|
|
184
|
|
185 page := fmt.Sprintf("%05d", p + 1)
|
|
186
|
|
187 data := Data {
|
|
188 Hno: hno,
|
|
189 Page: page,
|
|
190 Zip: zips[1],
|
|
191 SendTo: send,
|
|
192 }
|
|
193 hash[hno] = data
|
|
194
|
|
195 csv += data.ToCsv() + "\n"
|
|
196 }
|
|
197
|
|
198 if err := ioutil.WriteFile(infofile, []byte(csv), 0644); err != nil {
|
|
199 return hash, err
|
|
200 }
|
|
201 return hash, nil;
|
|
202 }
|
|
203
|
|
204 func read_order(csvfile string) ([]string, int, error) {
|
|
205 var order []string
|
|
206
|
|
207 c, err := ioutil.ReadFile(csvfile)
|
|
208 if err != nil {
|
|
209 return order, -1, err
|
|
210 }
|
|
211 r := strings.NewReader(string(c))
|
|
212 tr := transform.NewReader(r, japanese.ShiftJIS.NewDecoder())
|
|
213 b, err := ioutil.ReadAll(tr)
|
|
214 if err != nil {
|
|
215 return order, -1, err
|
|
216 }
|
|
217
|
|
218 n := 0
|
|
219 cr := csv.NewReader(strings.NewReader(string(b)))
|
|
220 for {
|
|
221 record, err := cr.Read()
|
|
222 if err == io.EOF {
|
|
223 break
|
|
224 }
|
|
225 if err != nil {
|
|
226 return order, n, err
|
|
227 }
|
|
228
|
|
229 if strings.HasPrefix(record[0], "0") {
|
|
230 order = append(order, record[0])
|
|
231 }
|
|
232 n++
|
|
233 }
|
|
234 return order, n, nil
|
|
235 }
|
|
236
|
|
237 func make_list(h_order []string, hash map[string]Data, pagefile1, pagefile2 string) (int, int, int, error) {
|
|
238 var n, m, l int
|
|
239 var list1, list2 []string
|
|
240 done := make(map[string]bool)
|
|
241
|
|
242 for _, h := range h_order {
|
|
243 if _, ok := done[h]; !ok {
|
|
244 if data, ok := hash[h]; ok {
|
|
245 list1 = append(list1, data.Page)
|
|
246 done[h] = true
|
|
247 n++
|
|
248 }
|
|
249 }
|
|
250 }
|
|
251
|
|
252 for _, z := range []string{"014-00", "014-01", "014-08", "014-1413"} {
|
|
253 for _, data := range hash {
|
|
254 h := data.Hno
|
|
255 if _, ok := done[h]; !ok {
|
|
256 if strings.HasPrefix(data.Zip, z) {
|
|
257 list2 = append(list2, data.Page)
|
|
258 done[h] = true
|
|
259 m++
|
|
260 }
|
|
261 }
|
|
262 }
|
|
263 }
|
|
264
|
|
265 for _, data := range hash {
|
|
266 h := data.Hno
|
|
267 if _, ok := done[h]; !ok {
|
|
268 list2 = append(list2, data.Page)
|
|
269 done[h] = true
|
|
270 l++
|
|
271 }
|
|
272 }
|
|
273
|
|
274 if err := write_pagefile(pagefile1, list1); err != nil {
|
|
275 return n, m, l, err
|
|
276 }
|
|
277 if err := write_pagefile(pagefile2, list2); err != nil {
|
|
278 return n, m, l, err
|
|
279 }
|
|
280 return n, m, l, nil
|
|
281 }
|
|
282
|
|
283 func write_pagefile(file string, list []string) error {
|
|
284 f, err := os.OpenFile(file, os.O_RDWR|os.O_CREATE, 0755)
|
|
285 if err != nil {
|
|
286 return err
|
|
287 }
|
|
288 for _, p := range list {
|
|
289 fmt.Fprintf(f, "fwtmp_%s.xdw\n", p)
|
|
290 }
|
|
291 if err := f.Close(); err != nil {
|
|
292 return err
|
|
293 }
|
|
294 return nil
|
|
295 }
|
|
296
|
|
297 func print_time(msg string) {
|
|
298 now := time.Now()
|
|
299 fmt.Printf("\t%v # %s\n", now, msg)
|
|
300 }
|
|
301
|
|
302 func debug_print(debug bool, msg string) {
|
|
303 if debug {
|
|
304 fmt.Printf("\t%s\n", msg)
|
|
305 }
|
|
306 }
|
|
307
|
|
308 func clean_full() error {
|
|
309 return clean_file("fwtmp_")
|
|
310 }
|
|
311
|
|
312 func clean_mini() error {
|
|
313 return clean_file("fwtmp_b")
|
|
314 }
|
|
315
|
|
316 func clean_file(prefix string) error {
|
|
317 files, err := ioutil.ReadDir(".")
|
|
318 if err != nil {
|
|
319 return err
|
|
320 }
|
|
321
|
|
322 for _, file := range files {
|
|
323 if strings.HasPrefix(file.Name(), prefix) {
|
|
324 os.Remove(file.Name())
|
|
325 continue
|
|
326 }
|
|
327 }
|
|
328 return err
|
|
329 }
|
|
330
|