|
45
|
1 /*
|
|
51
|
2 Last Change: 2020-05-01 金 10:46:51.
|
|
45
|
3 */
|
|
|
4
|
|
|
5 package main
|
|
|
6
|
|
|
7 import (
|
|
48
|
8 "bufio"
|
|
47
|
9 "bytes"
|
|
|
10 "compress/gzip"
|
|
45
|
11 "encoding/csv"
|
|
|
12 "fmt"
|
|
|
13 "io"
|
|
|
14 "io/ioutil"
|
|
|
15 "log"
|
|
|
16 "net"
|
|
|
17 "net/http"
|
|
|
18 "os"
|
|
|
19 "path/filepath"
|
|
48
|
20 "sort"
|
|
45
|
21 "strconv"
|
|
|
22 "strings"
|
|
51
|
23 "time"
|
|
45
|
24 )
|
|
|
25
|
|
|
26 type hhs struct {
|
|
|
27 No string
|
|
|
28 //Birth string
|
|
|
29 Name string
|
|
|
30 //Kana string
|
|
|
31 Addr string
|
|
|
32 //Sex string
|
|
|
33 Ccn []string
|
|
|
34 }
|
|
|
35
|
|
|
36 func (h *hhs) GetData() string {
|
|
|
37 s := strings.Join(h.Ccn, "#")
|
|
|
38 s = strings.Join([]string{h.Name, h.Addr, s}, ":")
|
|
|
39 return s
|
|
|
40 }
|
|
|
41
|
|
|
42 func (h *hhs) GetRecent() string {
|
|
|
43 ccn := ""
|
|
|
44 if len(h.Ccn) > 0 {
|
|
|
45 ccn = h.Ccn[0]
|
|
|
46 }
|
|
|
47 return strings.Join([]string{h.No, h.Name, ccn}, ",")
|
|
|
48 }
|
|
|
49
|
|
|
50 var (
|
|
|
51 server string
|
|
|
52 port string
|
|
|
53 hhsdb string
|
|
48
|
54 indexdb string
|
|
49
|
55 pwdb string
|
|
48
|
56 server_root string
|
|
45
|
57 hhash map[string]hhs
|
|
|
58 iymdhash map[string]string
|
|
|
59 iyhash map[string]string
|
|
51
|
60 logfile string
|
|
45
|
61 )
|
|
|
62
|
|
|
63 func init() {
|
|
|
64 port = ":3910"
|
|
|
65 hhsdb = "hhsdb.csv"
|
|
48
|
66 indexdb = "index.csv"
|
|
49
|
67 pwdb = "passwd"
|
|
51
|
68 logfile = "searcher.log"
|
|
45
|
69 }
|
|
|
70
|
|
|
71 func main() {
|
|
48
|
72 server_root = filepath.Dir(os.Args[0])
|
|
|
73 hhsdb = filepath.Join(server_root, hhsdb)
|
|
|
74 indexdb = filepath.Join(server_root, indexdb)
|
|
49
|
75 pwdb = filepath.Join(server_root, pwdb)
|
|
45
|
76
|
|
|
77 // setting IP-Address & Port
|
|
|
78 addrs, err := net.InterfaceAddrs()
|
|
|
79 if err != nil {
|
|
|
80 log.Fatal(err)
|
|
|
81 }
|
|
|
82 for _, a := range addrs {
|
|
|
83 if ipnet, ok := a.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
|
|
|
84 if strings.HasPrefix(ipnet.IP.String(), "169.254") {
|
|
|
85 continue
|
|
|
86 }
|
|
|
87 if ipnet.IP.To4() != nil {
|
|
|
88 server = ipnet.IP.String() + port
|
|
|
89 }
|
|
|
90 }
|
|
|
91 }
|
|
|
92
|
|
|
93 if err := loadDB(); err != nil {
|
|
|
94 log.Fatal(err)
|
|
|
95 }
|
|
|
96
|
|
|
97 // Http-Handler
|
|
48
|
98 http.HandleFunc("/h/", hhs_handler) // Get /h/0800012345 -> name:addr:20200101#20210701#...
|
|
|
99 http.HandleFunc("/hn/", hhsnm_handler) // Get /h/0800012345:0800098765:... -> name1:name2:...
|
|
|
100 http.HandleFunc("/ht/", hhstm_handler) // Get /ht/ -> 2020-03-14 12:34 (2020-04-02)
|
|
|
101 http.HandleFunc("/ha/", hhsdb_handler) // Get /ha/ -> hhsdb.csv for Mover
|
|
|
102 http.HandleFunc("/i/", image_handler) // Get /i/20200110/0800012345.tgz
|
|
|
103 http.HandleFunc("/r/", recent_handler) // Get /r/0800012345:0800067890:0800099999:... -> 0800012345,name1,20200101:0800067890,name2,20210405:...
|
|
|
104 http.HandleFunc("/d/", index_handler) // Get /d/20xx -> 20xx0401:20xx0408:... , /d/20xx0401 -> 0800012345:0800098765:...
|
|
|
105 http.HandleFunc("/dt/", indextm_handler) // Get /dt/ -> 2020-03-14 12:34 (2020-04-02)
|
|
|
106 http.HandleFunc("/ud/", upidx_handler) // Get /ud/20200402
|
|
|
107 http.HandleFunc("/u/", uphhsdb_handler) // POST /u/
|
|
|
108 http.HandleFunc("/ui/", upimage_handler) // POST /ui/20200401/0800012345.tgz
|
|
|
109 http.HandleFunc("/ci/", climage_handler) // Get /ci/20200402
|
|
49
|
110 http.HandleFunc("/pw/", pw_handler) // Get /pw/ -> id1:pw1:id2:pw2:...
|
|
45
|
111
|
|
|
112 log.Fatal(http.ListenAndServe(server, nil))
|
|
|
113 }
|
|
|
114
|
|
48
|
115 /** FUNCTIONS **/
|
|
45
|
116 func loadDB() error {
|
|
48
|
117 hhash = make(map[string]hhs)
|
|
|
118 iymdhash = make(map[string]string)
|
|
|
119 iyhash = make(map[string]string)
|
|
|
120
|
|
45
|
121 b, err := ioutil.ReadFile(hhsdb)
|
|
|
122 if err != nil {
|
|
|
123 return err
|
|
|
124 }
|
|
|
125 r := csv.NewReader(strings.NewReader(string(b)))
|
|
|
126 for {
|
|
|
127 record, err := r.Read()
|
|
|
128 if err == io.EOF {
|
|
|
129 break
|
|
|
130 }
|
|
|
131 if err != nil {
|
|
|
132 return err
|
|
|
133 }
|
|
|
134 h := hhs{
|
|
48
|
135 No: record[0], //Birth: record[1],
|
|
|
136 Name: record[2], //Kana: record[3],
|
|
|
137 Addr: record[4], //Sex: record[5],
|
|
45
|
138 }
|
|
|
139 hhash[record[0]] = h
|
|
|
140 }
|
|
|
141
|
|
48
|
142 b, err = ioutil.ReadFile(indexdb)
|
|
45
|
143 if err != nil {
|
|
|
144 return err
|
|
|
145 }
|
|
|
146 r = csv.NewReader(strings.NewReader(string(b)))
|
|
|
147 for {
|
|
|
148 record, err := r.Read()
|
|
|
149 if err == io.EOF {
|
|
|
150 break
|
|
|
151 }
|
|
|
152 if err != nil {
|
|
|
153 return err
|
|
|
154 }
|
|
|
155 h := hhash[record[0]]
|
|
|
156 ccn := h.Ccn
|
|
|
157 h.Ccn = append(ccn, record[1])
|
|
|
158 hhash[record[0]] = h
|
|
|
159
|
|
|
160 iymdhash[record[1]] += ":" + record[0]
|
|
|
161 }
|
|
|
162
|
|
|
163 for ymd, _ := range iymdhash {
|
|
|
164 y := ymd[0:4]
|
|
|
165 if ymd[4:6] == "01" || ymd[4:6] == "02" || ymd[4:6] == "03" {
|
|
|
166 yy, _ := strconv.Atoi(y)
|
|
|
167 y = fmt.Sprintf("%d", yy - 1)
|
|
|
168 }
|
|
|
169 iyhash[y] += ":" + ymd
|
|
|
170 }
|
|
|
171
|
|
|
172 return nil
|
|
|
173 }
|
|
|
174
|
|
51
|
175 func write_errlog(no int, msg string) {
|
|
|
176 log := filepath.Join(server_root, logfile)
|
|
|
177 t := time.Now().Format("2006-01-02 15:04")
|
|
|
178 msg = fmt.Sprintf("%s [%02d] %s\n", t, no, msg)
|
|
|
179 f, _ := os.OpenFile(log, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
|
180 f.Write([]byte(msg))
|
|
|
181 f.Close()
|
|
|
182 }
|
|
|
183
|
|
48
|
184 /** HTTP HANDLERS **/
|
|
46
|
185 /* Get /h/0800012345 -> name:addr:20200101#20210701#... */
|
|
45
|
186 func hhs_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
187 hno := r.URL.Path[len("/h/"):]
|
|
|
188 s := ""
|
|
|
189 if h, ok := hhash[hno]; ok {
|
|
|
190 s = h.GetData()
|
|
|
191 }
|
|
|
192 w.Write([]byte(s))
|
|
|
193 }
|
|
|
194
|
|
46
|
195 /* Get /hn/0800012345:0800098765:... -> name1:name2:... */
|
|
|
196 func hhsnm_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
197 hnolist := strings.Split(r.URL.Path[len("/hn/"):], ":")
|
|
|
198 var buf []string
|
|
|
199 for _, hno := range hnolist {
|
|
|
200 var n string
|
|
|
201 if h, ok := hhash[hno]; ok {
|
|
|
202 n = h.Name
|
|
|
203 }
|
|
|
204 buf = append(buf, n)
|
|
|
205 }
|
|
|
206 w.Write([]byte(strings.Join(buf, ":")))
|
|
|
207 }
|
|
|
208
|
|
48
|
209 /* Get /ht/ -> 2020-03-14 12:34 (2020-04-02) */
|
|
46
|
210 func hhstm_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
211 date := ""
|
|
47
|
212 if fi, err := os.Stat(hhsdb); err == nil {
|
|
|
213 t := fi.ModTime()
|
|
48
|
214 latest := "0000000001"
|
|
|
215
|
|
|
216 f, _ := os.Open(hhsdb)
|
|
|
217 defer f.Close()
|
|
|
218 input := bufio.NewScanner(f)
|
|
|
219 for input.Scan() {
|
|
|
220 s := strings.Split(input.Text(), ",")
|
|
|
221 if (strings.Compare(s[0], latest) > 0) {
|
|
|
222 latest = s[0]
|
|
|
223 }
|
|
|
224 }
|
|
|
225
|
|
|
226 latest = "(" + latest + ")"
|
|
|
227 date = t.Format("2006-01-02 15:04 ") + latest
|
|
47
|
228 }
|
|
46
|
229 w.Write([]byte(date))
|
|
|
230 }
|
|
|
231
|
|
45
|
232 /* Get /ha/ -> hhsdb.csv for Mover */
|
|
|
233 func hhsdb_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
234 b, _ := ioutil.ReadFile(hhsdb)
|
|
|
235 w.Write(b)
|
|
|
236 }
|
|
|
237
|
|
|
238 /* Get /i/20200110/0800012345.tgz */
|
|
|
239 func image_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
240 file := r.URL.Path[len("/i/"):]
|
|
48
|
241 file = filepath.Join(server_root, "images", filepath.FromSlash(file))
|
|
45
|
242
|
|
|
243 f, err := os.Open(file)
|
|
|
244 if err != nil {
|
|
51
|
245 write_errlog(1, "cannot open " + file)
|
|
45
|
246 http.NotFound(w, r)
|
|
|
247 return
|
|
|
248 }
|
|
|
249 defer f.Close()
|
|
|
250
|
|
|
251 fi, _ := f.Stat()
|
|
|
252
|
|
|
253 w.Header().Set("Content-Type", "rsearcher/octet-stream")
|
|
|
254 w.Header().Set("Content-Length", fmt.Sprintf("%d", fi.Size()))
|
|
|
255 io.Copy(w, f)
|
|
|
256 }
|
|
|
257
|
|
46
|
258 /* Get /r/0800012345:0800067890:0800099999:... -> 0800012345,name1,20200101:0800067890,name2,20210405:... */
|
|
45
|
259 func recent_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
260 hnolist := strings.Split(r.URL.Path[len("/r/"):], ":")
|
|
|
261 var buf []string
|
|
|
262 for _, hno := range hnolist {
|
|
|
263 s := hno + ",,"
|
|
|
264 if h, ok := hhash[hno]; ok {
|
|
|
265 s = h.GetRecent()
|
|
|
266 if h.No == "" {
|
|
|
267 s = hno + s
|
|
|
268 }
|
|
|
269 }
|
|
|
270 buf = append(buf, s)
|
|
|
271 }
|
|
|
272 w.Write([]byte(strings.Join(buf, ":")))
|
|
|
273 }
|
|
|
274
|
|
48
|
275 /* Get /d/20xx -> 20xx0401:20xx0408:... , /d/20xx0401 -> 0800012345:0800098765:... */
|
|
45
|
276 func index_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
277 var buf string
|
|
|
278 ymd := r.URL.Path[len("/d/"):]
|
|
|
279 if len(ymd) == 4 {
|
|
|
280 buf = iyhash[ymd]
|
|
|
281 }
|
|
|
282 if len(ymd) == 8 {
|
|
|
283 buf = iymdhash[ymd]
|
|
|
284 }
|
|
|
285 w.Write([]byte(buf[1:]))
|
|
|
286 }
|
|
|
287
|
|
47
|
288 /* POST /u/ */
|
|
45
|
289 func uphhsdb_handler(w http.ResponseWriter, r *http.Request) {
|
|
47
|
290 if r.Method != http.MethodPost {
|
|
51
|
291 write_errlog(10, "not post method")
|
|
47
|
292 http.NotFound(w, r)
|
|
|
293 return
|
|
|
294 }
|
|
|
295
|
|
51
|
296 f, err := os.Create(hhsdb)
|
|
45
|
297 if err != nil {
|
|
51
|
298 write_errlog(11, "cannot create " + hhsdb)
|
|
45
|
299 http.NotFound(w, r)
|
|
|
300 return
|
|
|
301 }
|
|
47
|
302 defer f.Close()
|
|
|
303
|
|
|
304 b, err := ioutil.ReadAll(r.Body)
|
|
|
305 r.Body.Close()
|
|
|
306 if err != nil {
|
|
51
|
307 write_errlog(12, "cannot read req-body")
|
|
47
|
308 http.NotFound(w, r)
|
|
|
309 return
|
|
|
310 }
|
|
|
311
|
|
|
312 br := bytes.NewReader(b)
|
|
|
313 zr, err := gzip.NewReader(br)
|
|
|
314 if err != nil {
|
|
51
|
315 write_errlog(13, "gzip-reader error")
|
|
47
|
316 http.NotFound(w, r)
|
|
|
317 return
|
|
|
318 }
|
|
|
319 n, err := io.Copy(f, zr)
|
|
|
320
|
|
|
321 if err := zr.Close(); err != nil {
|
|
51
|
322 write_errlog(14, "gzip-reader error")
|
|
47
|
323 http.NotFound(w, r)
|
|
|
324 return
|
|
|
325 }
|
|
|
326
|
|
|
327 w.Write([]byte(fmt.Sprintf("%d bytes are recieved.\n", n)))
|
|
|
328
|
|
|
329 loadDB();
|
|
|
330 }
|
|
|
331
|
|
|
332 /* POST /ui/20200401/0800012345.tgz */
|
|
|
333 func upimage_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
334 if r.Method != http.MethodPost {
|
|
51
|
335 write_errlog(20, "not post method")
|
|
47
|
336 http.NotFound(w, r)
|
|
|
337 return
|
|
|
338 }
|
|
|
339
|
|
|
340 uri := r.URL.Path[len("/ui/"):]
|
|
|
341 ymd := uri[0:8]
|
|
|
342 tgz := uri[9:]
|
|
|
343
|
|
|
344 dir := filepath.Join(server_root, "images", ymd)
|
|
|
345
|
|
|
346 if _, err := os.Stat(dir); os.IsNotExist(err) {
|
|
|
347 os.Mkdir(dir, 0644)
|
|
|
348 }
|
|
|
349
|
|
|
350 file := filepath.Join(server_root, "images", ymd, tgz)
|
|
|
351 f, err := os.Create(file)
|
|
|
352 if err != nil {
|
|
51
|
353 write_errlog(21, "cannot create " + file)
|
|
47
|
354 http.NotFound(w, r)
|
|
|
355 return
|
|
|
356 }
|
|
|
357 defer f.Close()
|
|
|
358
|
|
45
|
359 n, err := io.Copy(f, r.Body)
|
|
|
360 if err != nil {
|
|
51
|
361 write_errlog(22, "cannot copy req-body")
|
|
45
|
362 http.NotFound(w, r)
|
|
|
363 return
|
|
|
364 }
|
|
|
365 w.Write([]byte(fmt.Sprintf("%d bytes are recieved.\n", n)))
|
|
|
366 }
|
|
|
367
|
|
48
|
368 /* Get /dt/ -> 2020-03-14 12:34 (2020-04-02) */
|
|
|
369 func indextm_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
370 date := ""
|
|
|
371 if fi, err := os.Stat(indexdb); err == nil {
|
|
|
372 t := fi.ModTime()
|
|
|
373 latest := "20200101"
|
|
|
374
|
|
|
375 f, _ := os.Open(indexdb)
|
|
|
376 defer f.Close()
|
|
|
377 input := bufio.NewScanner(f)
|
|
|
378 for input.Scan() {
|
|
|
379 s := strings.Split(input.Text(), ",")
|
|
|
380 if (strings.Compare(s[1], latest) > 0) {
|
|
|
381 latest = s[1]
|
|
|
382 }
|
|
|
383 }
|
|
|
384
|
|
|
385 latest = "(" + latest[0:4] + "-" + latest[4:6] + "-" + latest[6:] + ")"
|
|
|
386 date = t.Format("2006-01-02 15:04 ") + latest
|
|
|
387 }
|
|
|
388 w.Write([]byte(date))
|
|
|
389 }
|
|
|
390
|
|
|
391 /* Get /ud/20200402 */
|
|
|
392 func upidx_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
393 date := r.URL.Path[len("/ud/"):]
|
|
|
394 var buf []string
|
|
|
395
|
|
|
396 f, _ := os.Open(indexdb)
|
|
|
397 input := bufio.NewScanner(f)
|
|
|
398 for input.Scan() {
|
|
|
399 if !strings.HasSuffix(input.Text(), date) {
|
|
|
400 buf = append(buf, input.Text())
|
|
|
401 }
|
|
|
402 }
|
|
|
403 f.Close()
|
|
|
404
|
|
|
405 imgdir := filepath.Join(server_root, "images", date)
|
|
|
406 files, _ := ioutil.ReadDir(imgdir)
|
|
|
407 for _, file := range files {
|
|
|
408 i := file.Name()[0:10] + "," + date
|
|
|
409 buf = append(buf, i)
|
|
|
410 }
|
|
|
411 sort.Sort(sort.Reverse(sort.StringSlice(buf)))
|
|
|
412
|
|
|
413 os.Remove(indexdb)
|
|
|
414
|
|
|
415 s := strings.Join(buf, "\n")
|
|
|
416 ioutil.WriteFile(indexdb, []byte(s), 0644)
|
|
|
417
|
|
|
418 loadDB();
|
|
|
419 w.Write([]byte("update index done."))
|
|
|
420 }
|
|
|
421
|
|
|
422 /* Get /ci/20200402 */
|
|
|
423 func climage_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
424 date := r.URL.Path[len("/ci/"):]
|
|
|
425 dir := filepath.Join(server_root, "images", date)
|
|
|
426 os.RemoveAll(dir)
|
|
|
427 }
|
|
|
428
|
|
49
|
429 /* Get /pw/ -> id1:pw1:id2:pw2:... */
|
|
|
430 func pw_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
431 b, err := ioutil.ReadFile(pwdb)
|
|
|
432 if err != nil {
|
|
51
|
433 write_errlog(30, "cannot read passwd-file")
|
|
49
|
434 http.NotFound(w, r)
|
|
|
435 return
|
|
|
436 }
|
|
|
437 w.Write([]byte(strings.ReplaceAll(string(b), "\n", ":")))
|
|
|
438 }
|
|
|
439
|
